Refactor package list to be a preprocessor

1 files changed, 23 insertions, 0 deletions

diff --git a/plugins/jekyll/src/main/kotlin/JekyllPlugin.kt b/plugins/jekyll/src/main/kotlin/JekyllPlugin.kt
index c4f7f2c7..b9f5e1a5 100644
--- a/plugins/jekyll/src/main/kotlin/JekyllPlugin.kt
+++ b/plugins/jekyll/src/main/kotlin/JekyllPlugin.kt
@@ -1,24 +1,47 @@
 package org.jetbrains.dokka.jekyll
 
 import org.jetbrains.dokka.CoreExtensions
+import org.jetbrains.dokka.base.renderers.PackageListCreator
+import org.jetbrains.dokka.base.renderers.RootCreator
 import org.jetbrains.dokka.gfm.CommonmarkRenderer
 import org.jetbrains.dokka.pages.*
 import org.jetbrains.dokka.plugability.DokkaContext
 import org.jetbrains.dokka.plugability.DokkaPlugin
+import org.jetbrains.dokka.plugability.plugin
+import org.jetbrains.dokka.plugability.query
+import org.jetbrains.dokka.transformers.pages.PageTransformer
 import java.lang.StringBuilder
 
 
 class JekyllPlugin : DokkaPlugin() {
 
+    val jekyllPreprocessors by extensionPoint<PageTransformer>()
+
     val renderer by extending {
         CoreExtensions.renderer providing { JekyllRenderer(it) } applyIf { format == "jekyll" }
     }
+
+    val rootCreator by extending {
+        jekyllPreprocessors with RootCreator
+    }
+
+    val packageListCreator by extending {
+        jekyllPreprocessors providing {
+            PackageListCreator(
+                it,
+                "jekyll",
+                "md"
+            )
+        } order { after(rootCreator) }
+    }
 }
 
 class JekyllRenderer(
     context: DokkaContext
 ) : CommonmarkRenderer(context) {
 
+    override val preprocessors = context.plugin<JekyllPlugin>().query { jekyllPreprocessors }
+
     override fun buildPage(page: ContentPage, content: (StringBuilder, ContentPage) -> Unit): String {
         val builder = StringBuilder()
         builder.append("---\n")