add pages service

Author: shiena

src/main/resources/update/gitbucket-page_1.1.sql

@@ -0,0 +1,8 @@
+INSERT INTO PAGES
+    (USER_NAME, REPOSITORY_NAME, SOURCE)
+SELECT
+    A.USER_NAME
+    , A.REPOSITORY_NAME
+    , 'gh-pages'
+FROM
+    REPOSITORY A;

src/main/resources/update/gitbucket-page_1.1.xml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<changeSet>
+    <createTable tableName="PAGES">
+        <column name="USER_NAME" type="varchar(100)" nullable="false"/>
+        <column name="REPOSITORY_NAME" type="varchar(100)" nullable="false"/>
+        <column name="SOURCE" type="varchar(100)" nullable="false"/>
+    </createTable>
+
+    <addPrimaryKey constraintName="IDX_PAGES_PK" tableName="PAGES" columnNames="USER_NAME, REPOSITORY_NAME"/>
+</changeSet>

src/main/scala/Plugin.scala

@@ -1,5 +1,6 @@
 import gitbucket.core.controller.ControllerBase
 import gitbucket.plugin.pages.PagesController
+import io.github.gitbucket.solidbase.migration.{ SqlMigration, LiquibaseMigration }
 import io.github.gitbucket.solidbase.model.Version
 
 class Plugin extends gitbucket.core.plugin.Plugin {
@@ -16,7 +17,12 @@ class Plugin extends gitbucket.core.plugin.Plugin {
     new Version("0.7"),
     new Version("0.8"),
     new Version("0.9"),
-    new Version("1.0")
+    new Version("1.0"),
+    new Version(
+      "1.1",
+      new LiquibaseMigration("update/gitbucket-page_1.1.xml"),
+      new SqlMigration("update/gitbucket-page_1.1.sql")
+    )
   )
 
   override val controllers: Seq[(String, ControllerBase)] = Seq(

src/main/scala/gitbucket/plugin/model/PageProfile.scala

@@ -0,0 +1,6 @@
+package gitbucket.plugin.model
+
+import gitbucket.core.model._
+
+object Profile extends CoreProfile
+  with PagesComponent

src/main/scala/gitbucket/plugin/model/Pages.scala

@@ -0,0 +1,40 @@
+package gitbucket.plugin.model
+
+trait PagesComponent { self: gitbucket.core.model.Profile =>
+  import profile.api._
+
+  implicit val psColumnType = MappedColumnType.base[PageSourceType, String](ps => ps.code, code => PageSourceType.valueOf(code))
+
+  lazy val Pages = TableQuery[Pages]
+
+  class Pages(tag: Tag) extends Table[Page](tag, "PAGES") {
+    val userName = column[String]("USER_NAME")
+    val repositoryName = column[String]("REPOSITORY_NAME")
+    val source = column[PageSourceType]("SOURCE")
+    def * = (userName, repositoryName, source) <> ((Page.apply _).tupled, Page.unapply)
+  }
+}
+
+abstract sealed case class PageSourceType(code: String)
+
+object PageSourceType {
+  object NONE extends PageSourceType("none")
+  object MASTER_DOCS extends PageSourceType("master /docs")
+  object MASTER extends PageSourceType("master")
+  object GH_PAGES extends PageSourceType("gh-pages")
+
+  val values: Vector[PageSourceType] = Vector(NONE, MASTER_DOCS, MASTER, GH_PAGES)
+
+  private val map: Map[String, PageSourceType] = values.map(enum => enum.code -> enum).toMap
+
+  def apply(code: String): PageSourceType = map(code)
+
+  def valueOf(code: String): PageSourceType = map(code)
+  def valueOpt(code: String): Option[PageSourceType] = map.get(code)
+}
+
+case class Page(
+  userName: String,
+  repositoryName: String,
+  source: PageSourceType
+)

src/main/scala/gitbucket/plugin/service/PagesService.scala

@@ -0,0 +1,36 @@
+package gitbucket.plugin.service
+
+import gitbucket.plugin.model.{ Page, PageSourceType }
+import gitbucket.plugin.model.Profile._
+import gitbucket.plugin.model.Profile.profile.blockingApi._
+
+trait PagesService {
+
+  def getPageOptions(userName: String, repositoryName: String)(implicit s: Session): Option[Page] =
+    Pages.filter(t => (t.userName === userName.bind) && (t.repositoryName === repositoryName.bind)).firstOption
+
+  def registerPageOptions(userName: String, repositoryName: String, source: PageSourceType)(implicit s: Session): Unit =
+    Pages.insert(Page(userName, repositoryName, source))
+
+  def updatePageOptions(userName: String, repositoryName: String, source: PageSourceType)(implicit s: Session): Unit =
+    Pages
+      .filter(t => (t.userName === userName.bind) && (t.repositoryName === repositoryName.bind))
+      .map(t => t.source)
+      .update(source)
+
+  def renameRepository(userName: String, oldRepositoryName: String, newRepositoryName: String)(implicit s: Session): Unit =
+    Pages
+      .filter(t => (t.userName === userName.bind) && (t.repositoryName === oldRepositoryName.bind))
+      .map(t => t.repositoryName)
+      .update(newRepositoryName)
+
+  def renameUserName(oldUserName: String, newUserName: String, repositoryName: String)(implicit s: Session): Unit =
+    Pages
+      .filter(t => (t.userName === oldUserName.bind) && (t.repositoryName === repositoryName.bind))
+      .map(t => t.userName)
+      .update(newUserName)
+
+  def deletePageOptions(userName: String, repositoryName: String)(implicit s: Session): Unit = {
+    Pages.filter(t => (t.userName === userName.bind) && (t.repositoryName === repositoryName.bind)).delete
+  }
+}