(refs #22)Add discover action

Author: takezoe

src/main/scala/gitbucket/gist/controller/GistController.scala

@@ -24,6 +24,7 @@ import org.apache.commons.io.FileUtils
 import org.eclipse.jgit.api.Git
 import org.eclipse.jgit.lib._
 import org.scalatra.Ok
+import play.twirl.api.Html
 
 class GistController extends GistControllerBase with GistService with GistCommentService with AccountService
   with GistEditorAuthenticator with UsersAuthenticator
@@ -47,25 +48,7 @@ trait GistControllerBase extends ControllerBase {
     if(context.loginAccount.isDefined){
       val gists = getRecentGists(context.loginAccount.get.userName, 0, 4)
       html.edit(gists, None, Seq(("", JGitUtil.ContentInfo("text", None, Some("UTF-8")))))
-    } else {
-      val page = request.getParameter("page") match {
-        case ""|null => 1
-        case s => s.toInt
-      }
-      val result = getPublicGists((page - 1) * Limit, Limit)
-      val count  = countPublicGists()
-
-      val gists: Seq[(Gist, GistInfo)] = result.map { gist =>
-        val userName = gist.userName
-        val repoName = gist.repositoryName
-        val files = getGistFiles(userName, repoName)
-        val (fileName, source) = files.head
-
-        (gist, GistInfo(fileName, getLines(source), files.length, getForkedCount(userName, repoName), getCommentCount(userName, repoName)))
-      }
-
-      html.list(None, gists, page, page * Limit < count)
-    }
+    } else _discoverGists()
   }
 
   get("/gist/:userName/:repoName"){
@@ -292,23 +275,8 @@ trait GistControllerBase extends ControllerBase {
     html.editor(count, "", JGitUtil.ContentInfo("text", None, Some("UTF-8")))
   }
 
-  get("/gist/_public"){
-    val page = request.getParameter("page") match {
-      case ""|null => 1
-      case s => s.toInt
-    }
-    val result = getPublicGists((page - 1) * Limit, Limit)
-    val count  = countPublicGists()
-
-    val gists: Seq[(Gist, GistInfo)] = result.map { gist =>
-      val userName = gist.userName
-      val repoName = gist.repositoryName
-      val files = getGistFiles(userName, repoName)
-      val (fileName, source) = files.head
-
-      (gist, GistInfo(fileName, source, files.length, getForkedCount(userName, repoName), getCommentCount(userName, repoName)))
-    }
-      html.list(None, gists, page, page * Limit < count)
+  get("/gist/discover"){
+    _discoverGists()
   }
 
   ////////////////////////////////////////////////////////////////////////////////
@@ -422,9 +390,8 @@ trait GistControllerBase extends ControllerBase {
         } getOrElse {
           contentType = formats("json")
           org.json4s.jackson.Serialization.write(
-            Map("content" -> gitbucket.core.view.Markdown.toHtml(comment.content,
-              gist.toRepositoryInfo, false, true, true, true) // TODO isEditableこれでいいのか？
-            ))
+            Map("content" -> gitbucket.core.view.Markdown.toHtml(comment.content, gist.toRepositoryInfo, false, true, true, true))
+          )
         }
       }
     } getOrElse NotFound
@@ -446,8 +413,27 @@ trait GistControllerBase extends ControllerBase {
   // Private Methods
   //
   ////////////////////////////////////////////////////////////////////////////////
+  private def _discoverGists(): Html = {
+    val page = request.getParameter("page") match {
+      case ""|null => 1
+      case s => s.toInt
+    }
+    val result = getVisibleGists((page - 1) * Limit, Limit, None)
+    val count  = countPublicGists()
+
+    val gists: Seq[(Gist, GistInfo)] = result.map { gist =>
+      val userName = gist.userName
+      val repoName = gist.repositoryName
+      val files = getGistFiles(userName, repoName)
+      val (fileName, source) = files.head
+
+      (gist, GistInfo(fileName, getLines(source), files.length, getForkedCount(userName, repoName), getCommentCount(userName, repoName)))
+    }
+
+    html.list(None, gists, page, page * Limit < count)
+  }
 
-  private def _gist(userName: String, repoName: Option[String] = None, revision: String = "master") = {
+  private def _gist(userName: String, repoName: Option[String] = None, revision: String = "master"): Html = {
     repoName match {
       case None => {
         val page = params.get("page") match {

src/main/scala/gitbucket/gist/service/GistService.scala

@@ -1,5 +1,6 @@
 package gitbucket.gist.service
 
+import gitbucket.core.model.Account
 import gitbucket.gist.model.Gist
 import gitbucket.gist.model.Profile._
 import profile.simple._
@@ -9,8 +10,14 @@ trait GistService {
   def getRecentGists(userName: String, offset: Int, limit: Int)(implicit s: Session): Seq[Gist] =
     Gists.filter(_.userName === userName.bind).sortBy(_.registeredDate desc).drop(offset).take(limit).list
 
-  def getPublicGists(offset: Int, limit: Int)(implicit s: Session): Seq[Gist] =
-    Gists.filter(_.isPrivate === false.bind).sortBy(_.registeredDate desc).drop(offset).take(limit).list
+  def getVisibleGists(offset: Int, limit: Int, account: Option[Account])(implicit s: Session): Seq[Gist] = {
+    val query = account.map { x =>
+      Gists.filter { t => (t.isPrivate === false.bind) || (t.userName === x.userName.bind) }
+    } getOrElse {
+      Gists.filter { t => (t.isPrivate === false.bind) }
+    }
+    query.sortBy(_.registeredDate desc).drop(offset).take(limit).list
+  }
 
   def countPublicGists()(implicit s: Session): Int =
     Query(Gists.filter(_.isPrivate === false.bind).length).first