使用 mermaid 替代 dot 用于生成可视化文件

b7woreo · May 23, 2023 · e3d46fb · e3d46fb
1 parent 95322b1
commit e3d46fb
Show file tree

Hide file tree

Showing 12 changed files with 187 additions and 257 deletions.
diff --git a/example/android-application/build.gradle b/example/android-application/build.gradle
@@ -1,5 +1,6 @@
 plugins {
     id("com.android.application")
+    id('gradle.assistant')
 }
 
 android {
@@ -15,4 +16,6 @@ android {
 
 dependencies {
     implementation project(":example:android-library")
+    implementation project(":example:java-library")
+    implementation libs.rxjava
 }
diff --git a/example/android-library/build.gradle b/example/android-library/build.gradle
@@ -15,4 +15,5 @@ android {
 
 dependencies{
     implementation project(":example:java-library")
+    implementation libs.rxjava
 }
diff --git a/gradlew b/gradlew
diff --git a/plugin/build.gradle b/plugin/build.gradle
@@ -10,7 +10,7 @@ gradlePlugin {
     plugins{
         gdr {
             id = "gradle.assistant"
-            implementationClass = "io.github.knownitwhy.gdr.GdrPlugin"
+            implementationClass = "gradle.assistant.GradleAssistant"
         }
     }
 }

diff --git a/plugin/src/main/kotlin/gradle/assistant/GradleAssistant.kt b/plugin/src/main/kotlin/gradle/assistant/GradleAssistant.kt
@@ -12,10 +12,10 @@ import org.gradle.api.plugins.JavaPlugin
 import org.gradle.api.tasks.SourceSetContainer
 import org.gradle.configurationcache.extensions.capitalized
 
-class GradleAssistant : Plugin<Project> {
+abstract class GradleAssistant : Plugin<Project> {
 
     companion object {
-        private const val TASK_GROUP = "GradleAssistant"
+        private const val TASK_GROUP = "assistant"
     }
 
     override fun apply(project: Project) {

diff --git a/plugin/src/main/kotlin/gradle/assistant/dot/Dot.kt b/plugin/src/main/kotlin/gradle/assistant/dot/Dot.kt
diff --git a/plugin/src/main/kotlin/gradle/assistant/dot/Graphviz.kt b/plugin/src/main/kotlin/gradle/assistant/dot/Graphviz.kt
diff --git a/plugin/src/main/kotlin/gradle/assistant/graphic/Graphic.kt b/plugin/src/main/kotlin/gradle/assistant/graphic/Graphic.kt
@@ -0,0 +1,83 @@
+package gradle.assistant.graphic
+
+import java.io.File
+
+interface Graphic {
+    fun render(output: File, builder: Builder.() -> Unit)
+
+    interface Builder {
+        fun node(content: String, shape: Shape)
+        fun edge(from: String, to: String)
+    }
+
+    sealed class Shape {
+        object Box : Shape()
+        object Oval : Shape()
+        object Diamond : Shape()
+    }
+}
+
+class MermaidGraphic : Graphic {
+    private fun template(content: String): String {
+        return """
+        <!DOCTYPE html>
+        <html lang="en">
+        <body>
+          <pre id="mermaid">$content</pre>
+          <script type="module">
+            import mermaid from 'https://cdn.jsdelivr.net/npm/mermaid@10/dist/mermaid.esm.min.mjs';
+            import he from 'https://cdn.jsdelivr.net/npm/he@1.2.0/+esm';
+            const { svg, _ } = await mermaid.render('mermaid', he.decode(document.getElementById('mermaid').innerHTML));
+            document.body.innerHTML = svg;
+            document.getElementById('mermaid').setAttribute('width', '99999%');
+            document.getElementById('mermaid').setAttribute('height', '99999%');
+          </script>
+        </body>
+        </html>
+        """.trimIndent()
+    }
+
+    override fun render(output: File, builder: Graphic.Builder.() -> Unit) {
+        val content = MermaidGraphicBuilder().apply {
+            builder()
+        }.build()
+        output.writeText(template(content))
+    }
+}
+
+private class MermaidGraphicBuilder : Graphic.Builder {
+
+    private var currentId = 0
+    private val ids = mutableMapOf<String, Int>()
+    private val statements = mutableSetOf<String>()
+
+    override fun node(content: String, shape: Graphic.Shape) {
+        if (ids[content] != null) return
+
+        ids[content] = currentId++
+
+        when (shape) {
+            is Graphic.Shape.Box -> {
+                statements.add("${ids[content]}[\"$content\"]")
+            }
+
+            is Graphic.Shape.Oval -> {
+                statements.add("${ids[content]}([\"$content\"])")
+            }
+
+            is Graphic.Shape.Diamond -> {
+                statements.add("${ids[content]}{{\"$content\"}}")
+            }
+        }
+    }
+
+    override fun edge(from: String, to: String) {
+        val fromId = ids[from] ?: return
+        val toId = ids[to] ?: return
+        statements.add("$fromId --> $toId")
+    }
+
+    fun build(): String {
+        return "flowchart TD\n${statements.joinToString(separator = "\n")}"
+    }
+}