add pages hook

Author: shiena

src/main/scala/Plugin.scala

@@ -1,5 +1,5 @@
 import gitbucket.core.controller.ControllerBase
-import gitbucket.plugin.pages.PagesController
+import gitbucket.plugin.pages.{ PagesController, PagesHook }
 import io.github.gitbucket.solidbase.migration.{ SqlMigration, LiquibaseMigration }
 import io.github.gitbucket.solidbase.model.Version
 
@@ -28,5 +28,9 @@ class Plugin extends gitbucket.core.plugin.Plugin {
   override val controllers: Seq[(String, ControllerBase)] = Seq(
     "/*" -> new PagesController
   )
+
+  override val repositoryHooks = Seq(
+    new PagesHook
+  )
 }
 

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

@@ -0,0 +1,34 @@
+package gitbucket.plugin.pages
+
+import gitbucket.core.plugin.RepositoryHook
+import gitbucket.plugin.model.PageSourceType
+import gitbucket.plugin.model.Profile.profile.blockingApi._
+import gitbucket.plugin.service.PagesService
+
+class PagesHook
+  extends PagesHookBase
+  with PagesService
+
+trait PagesHookBase extends RepositoryHook {
+  self: PagesService =>
+
+  override def created(owner: String, repository: String)(implicit session: Session): Unit =
+    registerPageOptions(owner, repository, PageSourceType.GH_PAGES)
+
+  override def deleted(owner: String, repository: String)(implicit session: Session): Unit =
+    deletePageOptions(owner, repository)
+
+  override def renamed(owner: String, oldRepository: String, newRepository: String)(implicit session: Session): Unit =
+    renameRepository(owner, oldRepository, newRepository)
+
+  override def transferred(oldOwner: String, newOwner: String, repository: String)(implicit session: Session): Unit =
+    renameUserName(oldOwner, newOwner, repository)
+
+  override def forked(owner: String, newOwner: String, repository: String)(implicit session: Session): Unit = {
+    val source = getPageOptions(owner, repository) match {
+      case Some(p) => p.source
+      case None => PageSourceType.GH_PAGES
+    }
+    registerPageOptions(newOwner, repository, source)
+  }
+}