initial commit

Author: yaroot

.gitignore

@@ -0,0 +1,3 @@
+*.iml
+.idea/
+target/

build.sbt

@@ -0,0 +1,18 @@
+organization  := "gitbucket"
+
+name          := "pages-plugin"
+
+version       := "0.1"
+
+scalaVersion  := "2.11.7"
+
+scalacOptions := Seq("-unchecked", "-deprecation", "-encoding", "utf8", "-feature")
+
+resolvers += Resolver.jcenterRepo
+
+resolvers += "amateras-repo" at "http://amateras.sourceforge.jp/mvn/"
+
+libraryDependencies ++= Seq(
+  "gitbucket"          % "gitbucket-assembly" % "3.9.0" % "provided",
+  "javax.servlet"      % "javax.servlet-api"  % "3.1.0" % "provided"
+)

project/build.properties

@@ -0,0 +1 @@
+sbt.version=0.13.9

project/plugins.sbt

@@ -0,0 +1 @@
+addSbtPlugin("org.scalariform" % "sbt-scalariform" % "1.6.0")

src/main/scala/Plugin.scala

@@ -0,0 +1,15 @@
+
+import gitbucket.core.controller.ControllerBase
+import gitbucket.core.util.Version
+import gitbucket.plugin.pages.PagesController
+
+class Plugin extends gitbucket.core.plugin.Plugin {
+  override val pluginId: String = "pages"
+  override val pluginName: String = "Pages Plugin"
+  override val versions: Seq[Version] = Seq(Version(3, 9))
+  override val description: String = "Provides Pages feature on GitBucket"
+
+  override val controllers: Seq[(String, ControllerBase)] = Seq(
+    "/*" -> new PagesController
+  )
+}

src/main/scala/gitbucket/plugin/pages/pages.scala

@@ -0,0 +1,89 @@
+package gitbucket.plugin.pages
+
+import gitbucket.core.controller.ControllerBase
+import gitbucket.core.service.{ AccountService, RepositoryService }
+import gitbucket.core.util.ControlUtil._
+import gitbucket.core.util.{ Directory, JGitUtil, ReferrerAuthenticator }
+import org.eclipse.jgit.api.Git
+import org.eclipse.jgit.lib.ObjectId
+import org.eclipse.jgit.revwalk.RevCommit
+import org.eclipse.jgit.treewalk.TreeWalk
+
+import scala.language.implicitConversions
+
+class PagesController
+  extends ControllerBase
+  with AccountService
+  with RepositoryService
+  with ReferrerAuthenticator
+
+trait PagesControllerBase {
+  self: ControllerBase with AccountService with RepositoryService with ReferrerAuthenticator =>
+
+  val PAGES_BRANCH = "gh-pages"
+
+  get("/:owner/:repository/pages/*")(referrersOnly { repository =>
+    val path = params("splat")
+    using(Git.open(Directory.getRepositoryDir(repository.owner, repository.name))) { git =>
+      val objectId = Option(git.getRepository.resolve(PAGES_BRANCH))
+        .map(JGitUtil.getRevCommitFromId(git, _))
+        .flatMap { revCommit =>
+          getPathIndexObjectId(git, path, revCommit)
+        }
+
+      objectId match {
+        case Some((path0, objId)) =>
+          // redirect [owner/repo/pages/path] -> [owner/repo/pages/path/]
+          if (path0 != path && path0.startsWith(path) && !path.endsWith("/")) {
+            redirect(s"/${repository.owner}/${repository.name}/pages/$path/")
+          } else {
+            JGitUtil.getObjectLoaderFromId(git, objId) { loader =>
+              contentType = Option(servletContext.getMimeType(path0)).getOrElse("application/octet-stream")
+              response.setContentLength(loader.getSize.toInt)
+              loader.copyTo(response.getOutputStream)
+            }
+            ()
+          }
+        case None =>
+          NotFound()
+      }
+    }
+  })
+
+  get("/:owner/:repository/pages")(referrersOnly { repository =>
+    redirect(s"/${repository.owner}/${repository.name}/pages/")
+  })
+
+  def getPathIndexObjectId(git: Git, path: String, revCommit: RevCommit) = {
+    getPathObjectId0(git, path, revCommit)
+      .orElse(getPathObjectId0(git, appendPath(path, "index.html"), revCommit))
+      .orElse(getPathObjectId0(git, appendPath(path, "index.htm"), revCommit))
+  }
+
+  def getPathObjectId0(git: Git, path: String, revCommit: RevCommit) = {
+    getPathObjectId(git, path, revCommit).map(path -> _)
+  }
+
+  def appendPath(base: String, suffix: String): String = {
+    if (base == "") suffix
+    else if (base.endsWith("/")) base + suffix
+    else base + "/" + suffix
+  }
+
+  // copy&paste from RepositoryViewerControllerBase
+  private def getPathObjectId(git: Git, path: String, revCommit: RevCommit): Option[ObjectId] = {
+    @scala.annotation.tailrec
+    def _getPathObjectId(path: String, walk: TreeWalk): Option[ObjectId] = walk.next match {
+      case true if walk.getPathString == path => Some(walk.getObjectId(0))
+      case true => _getPathObjectId(path, walk)
+      case false => None
+    }
+
+    using(new TreeWalk(git.getRepository)) { treeWalk =>
+      treeWalk.addTree(revCommit.getTree)
+      treeWalk.setRecursive(true)
+      _getPathObjectId(path, treeWalk)
+    }
+  }
+}
+