diff --git a/src/main/scala/app/ControllerBase.scala b/src/main/scala/app/ControllerBase.scala index 3ebad37..da60646 100644 --- a/src/main/scala/app/ControllerBase.scala +++ b/src/main/scala/app/ControllerBase.scala @@ -1,9 +1,9 @@ package app +import model.Account import org.scalatra._ import org.scalatra.json._ import org.json4s._ -import org.json4s.jackson._ import jp.sf.amateras.scalatra.forms._ /** @@ -13,11 +13,35 @@ implicit val jsonFormats = DefaultFormats - implicit def context: Context = Context(servletContext.getContextPath, LoginUser) + implicit def context: Context = Context(servletContext.getContextPath, LoginAccount) - // TODO get from session - private val LoginUser = "admin" //System.getProperty("user.name") - + private def LoginAccount: Option[Account] = { + session.get("LOGIN_ACCOUNT") match { + case Some(x: Account) => Some(x) + case _ => None + } + } + + protected def usersOnly(action: => Any) = { + { + context.loginAccount match { + case Some(x) => action + case None => redirect("/signin") + } + } + } + + protected def usersOnly[T](action: T => Any) = { + (form: T) => { + context.loginAccount match { + case Some(x) => action(form) + case None => redirect("/signin") + } + } + } + +// protected def adminOnly() + } -case class Context(path: String, loginUser: String) \ No newline at end of file +case class Context(path: String, loginAccount: Option[Account]) \ No newline at end of file diff --git a/src/main/scala/app/CreateRepositoryController.scala b/src/main/scala/app/CreateRepositoryController.scala index 850253c..1a650a5 100644 --- a/src/main/scala/app/CreateRepositoryController.scala +++ b/src/main/scala/app/CreateRepositoryController.scala @@ -8,12 +8,13 @@ import org.apache.commons.io._ import jp.sf.amateras.scalatra.forms._ -class CreateRepositoryController extends CreateRepositoryControllerBase with ProjectService with AccountService +class CreateRepositoryController extends CreateRepositoryControllerBase + with ProjectService with AccountService with WikiService /** * Creates new repository. */ -trait CreateRepositoryControllerBase extends ControllerBase { self: ProjectService => +trait CreateRepositoryControllerBase extends ControllerBase { self: ProjectService with WikiService => case class RepositoryCreationForm(name: String, description: String) // TODO Option @@ -25,19 +26,21 @@ /** * Show the new repository form. */ - get("/new") { + get("/new")(usersOnly { html.newrepo() - } + }) /** * Create new repository. */ - post("/new", form) { form => + post("/new", form)(usersOnly { form => + val loginUserName = context.loginAccount.get.userName + // Insert to the database at first - createProject(form.name, context.loginUser, Some(form.description)) + createProject(form.name, loginUserName, Some(form.description)) // Create the actual repository - val gitdir = getRepositoryDir(context.loginUser, form.name) + val gitdir = getRepositoryDir(loginUserName, form.name) val repository = new RepositoryBuilder().setGitDir(gitdir).setBare.build repository.create @@ -46,7 +49,7 @@ config.setBoolean("http", null, "receivepack", true) config.save - val tmpdir = getInitRepositoryDir(context.loginUser, form.name) + val tmpdir = getInitRepositoryDir(loginUserName, form.name) try { // Clone the repository Git.cloneRepository.setURI(gitdir.toURI.toString).setDirectory(tmpdir).call @@ -67,9 +70,12 @@ FileUtils.deleteDirectory(tmpdir) } + // Create Wiki repository + createWikiRepository(context.loginAccount.get, form.name) + // redirect to the repository - redirect("/%s/%s".format(context.loginUser, form.name)) - } + redirect("/%s/%s".format(loginUserName, form.name)) + }) /** * Constraint for the repository name. @@ -78,7 +84,7 @@ def validate(name: String, value: String): Option[String] = { if(!value.matches("^[a-zA-Z0-9\\-_]+$")){ Some("Repository name contains invalid character.") - } else if(getRepositories(context.loginUser, servletContext).contains(value)){ + } else if(getRepositories(context.loginAccount.get.userName, servletContext).contains(value)){ Some("Repository already exists.") } else { None diff --git a/src/main/scala/app/SignInController.scala b/src/main/scala/app/SignInController.scala index a6fc748..8a71579 100644 --- a/src/main/scala/app/SignInController.scala +++ b/src/main/scala/app/SignInController.scala @@ -1,8 +1,11 @@ package app +import service._ import jp.sf.amateras.scalatra.forms._ -class SignInController extends ControllerBase { +class SignInController extends SignInControllerBase with AccountService + +trait SignInControllerBase extends ControllerBase { self: AccountService => case class SignInForm(email: String, password: String) @@ -16,7 +19,18 @@ } post("/signin", form){ form => - // TODO check email and password - redirect("/%s".format(context.loginUser)) + val account = getAccountByUserName(form.email) + if(account.isEmpty || account.get.password != form.password){ + redirect("/signin") + } else { + session.setAttribute("LOGIN_ACCOUNT", account.get) + redirect("/%s".format(account.get.userName)) + } } + + get("/signout"){ + session.invalidate + redirect("/signin") + } + } \ No newline at end of file diff --git a/src/main/scala/app/WikiController.scala b/src/main/scala/app/WikiController.scala index c307ecb..dac5d30 100644 --- a/src/main/scala/app/WikiController.scala +++ b/src/main/scala/app/WikiController.scala @@ -94,7 +94,7 @@ val repository = params("repository") saveWikiPage(owner, repository, form.currentPageName, form.pageName, - form.content, context.loginUser, form.message.getOrElse("")) + form.content, context.loginAccount.get, form.message.getOrElse("")) redirect("%s/%s/wiki/%s".format(owner, repository, form.pageName)) } @@ -111,7 +111,7 @@ val repository = params("repository") saveWikiPage(owner, repository, form.currentPageName, form.pageName, - form.content, context.loginUser, form.message.getOrElse("")) + form.content, context.loginAccount.get, form.message.getOrElse("")) redirect("%s/%s/wiki/%s".format(owner, repository, form.pageName)) } @@ -121,7 +121,7 @@ val repository = params("repository") val page = params("page") - deleteWikiPage(owner, repository, page, context.loginUser, "Delete %s".format(page)) + deleteWikiPage(owner, repository, page, context.loginAccount.get.userName, "Delete %s".format(page)) redirect("%s/%s/wiki".format(owner, repository)) } diff --git a/src/main/scala/service/WikiService.scala b/src/main/scala/service/WikiService.scala index f33bb63..4944399 100644 --- a/src/main/scala/service/WikiService.scala +++ b/src/main/scala/service/WikiService.scala @@ -34,17 +34,16 @@ } trait WikiService { - import WikiService._ // TODO synchronized? - def createWikiRepository(owner: String, repository: String): Unit = { - val dir = Directory.getWikiRepositoryDir(owner, repository) + def createWikiRepository(owner: model.Account, repository: String): Unit = { + val dir = Directory.getWikiRepositoryDir(owner.userName, repository) if(!dir.exists){ val repo = new RepositoryBuilder().setGitDir(dir).setBare.build try { repo.create - saveWikiPage(owner, repository, "Home", "Home", "Welcome to the %s wiki!!".format(repository), owner, "Initial Commit") + saveWikiPage(owner.userName, repository, "Home", "Home", "Welcome to the %s wiki!!".format(repository), owner, "Initial Commit") } finally { repo.close } @@ -55,9 +54,6 @@ * Returns the wiki page. */ def getWikiPage(owner: String, repository: String, pageName: String): Option[WikiPageInfo] = { - // TODO create wiki repository in the repository setting changing. - createWikiRepository(owner, repository) - JGitUtil.withGit(Directory.getWikiRepositoryDir(owner, repository)){ git => try { JGitUtil.getFileList(git, "master", ".").find(_.name == pageName + ".md").map { file => @@ -82,11 +78,8 @@ * Save the wiki page. */ def saveWikiPage(owner: String, repository: String, currentPageName: String, newPageName: String, - content: String, committer: String, message: String): Unit = { - - // TODO create wiki repository in the repository setting changing. - createWikiRepository(owner, repository) - + content: String, committer: model.Account, message: String): Unit = { + val workDir = Directory.getWikiWorkDir(owner, repository) // clone @@ -118,8 +111,7 @@ // commit and push if(added || deleted){ - // TODO committer's mail address - git.commit.setAuthor(committer, committer + "@devnull").setMessage(message).call + git.commit.setCommitter(committer.userName, committer.mailAddress).setMessage(message).call git.push.call } } @@ -129,8 +121,6 @@ * Delete the wiki page. */ def deleteWikiPage(owner: String, repository: String, pageName: String, committer: String, message: String): Unit = { - // TODO create wiki repository in the repository setting changing. - createWikiRepository(owner, repository) val workDir = Directory.getWikiWorkDir(owner, repository) diff --git a/src/main/twirl/index.scala.html b/src/main/twirl/index.scala.html index d6f9b38..4de76a1 100644 --- a/src/main/twirl/index.scala.html +++ b/src/main/twirl/index.scala.html @@ -3,7 +3,7 @@ @main("GitBucket"){

GitBucket

} \ No newline at end of file diff --git a/src/main/twirl/main.scala.html b/src/main/twirl/main.scala.html index a0e7a3f..16897cc 100644 --- a/src/main/twirl/main.scala.html +++ b/src/main/twirl/main.scala.html @@ -31,10 +31,14 @@ GitBucket diff --git a/src/main/twirl/user.scala.html b/src/main/twirl/user.scala.html index d881c7d..b9d2124 100644 --- a/src/main/twirl/user.scala.html +++ b/src/main/twirl/user.scala.html @@ -25,7 +25,7 @@ @repositories.map { repository =>
-
@repository.name
+
@repository.name
@repository.project.description
Last updated: @repository.project.lastActivityDate