dependency graph

Author: dosukoi-android

build.gradle.kts

@@ -75,4 +75,12 @@ allprojects {
     }
 }
 
-apply(from = file("gradle/projectDependencyGraph.gradle"))
+val projectDependencyGraph by tasks.registering(ProjectDependencyGraphTask::class) {
+    doLast {
+        copy {
+            from(rootProject.buildDir.resolve("reports/dependency-graph/project.dot.png"))
+            into(rootProject.projectDir)
+        }
+    }
+}
+

buildSrc/src/main/kotlin/ProjectDependencyGraphTask.kt

@@ -0,0 +1,154 @@
+import org.gradle.api.DefaultTask
+import org.gradle.api.Project
+import org.gradle.api.artifacts.ProjectDependency
+import org.gradle.api.tasks.TaskAction
+import java.util.Locale
+
+@Suppress("unused")
+open class ProjectDependencyGraphTask : DefaultTask() {
+    @TaskAction
+    fun run() {
+        val dot = project.rootDir.resolve("gradle/dependency-graph/project.dot")
+        dot.parentFile.mkdirs()
+        dot.delete()
+
+        dot.appendText(
+            """
+            |digraph {
+            |  graph [label="${project.rootProject.name}\n ",labelloc=t,fontsize=30,ranksep=1.4];
+            |  node [style=filled, fillcolor="#bbbbbb"];
+            |  rankdir=TB;
+            |
+        """.trimMargin()
+        )
+
+        val rootProjects = mutableListOf<Project>()
+        val queue = mutableListOf(project.rootProject)
+        while (queue.isNotEmpty()) {
+            val project = queue.removeAt(0)
+            rootProjects.add(project)
+            queue.addAll(project.childProjects.values)
+        }
+
+        val projects = LinkedHashSet<Project>()
+        val dependencies = LinkedHashMap<Pair<Project, Project>, MutableList<String>>()
+        val multiplatformProjects = mutableListOf<Project>()
+        val jsProjects = mutableListOf<Project>()
+        val androidProjects = mutableListOf<Project>()
+        val javaProjects = mutableListOf<Project>()
+        val rankAndroid = mutableListOf<Project>()
+        val rankDomain = mutableListOf<Project>()
+        val rankRepository = mutableListOf<Project>()
+
+        queue.clear()
+        queue.add(project.rootProject)
+        while (queue.isNotEmpty()) {
+            val project = queue.removeAt(0)
+            queue.addAll(project.childProjects.values)
+
+            if (project.plugins.hasPlugin("org.jetbrains.kotlin.multiplatform")) {
+                multiplatformProjects.add(project)
+            }
+            if (project.plugins.hasPlugin("kotlin2js")) {
+                jsProjects.add(project)
+            }
+            if (project.plugins.hasPlugin("com.android.library") || project.plugins.hasPlugin("com.android.application")) {
+                androidProjects.add(project)
+            }
+            if (project.plugins.hasPlugin("java-library") || project.plugins.hasPlugin("java")) {
+                javaProjects.add(project)
+            }
+
+            if (!project.path.startsWith(":core:") && project.path.endsWith(":android")) {
+                rankAndroid.add(project)
+            }
+            if (!project.path.startsWith(":core:") && project.path.endsWith(":repository")) {
+                rankRepository.add(project)
+            }
+
+            project.configurations.all {
+                getDependencies()
+                    .filterIsInstance<ProjectDependency>()
+                    .filter { project != it.dependencyProject }
+                    .map { it.dependencyProject }
+                    .forEach { dependency ->
+                        projects.add(project)
+                        projects.add(dependency)
+                        rootProjects.remove(dependency)
+
+                        val graphKey = Pair(project, dependency)
+                        val traits = dependencies.computeIfAbsent(graphKey) { mutableListOf() }
+
+                        if (name.toLowerCase(Locale.getDefault()).endsWith("implementation")) {
+                            traits.add("style=dotted")
+                        }
+                    }
+            }
+        }
+
+        projects.sortedBy { it.path }.also {
+            projects.clear()
+            projects.addAll(it)
+        }
+
+        dot.appendText("\n  # Projects\n\n")
+        for (project in projects) {
+            val traits = mutableListOf<String>()
+
+            if (rootProjects.contains(project)) {
+                traits.add("shape=box")
+            }
+
+            if (multiplatformProjects.contains(project)) {
+                if (androidProjects.contains(project)) {
+                    traits.add("fillcolor=\"#f7ffad\"")
+                } else {
+                    traits.add("fillcolor=\"#ffd2b3\"")
+                }
+            } else if (jsProjects.contains(project)) {
+                traits.add("fillcolor=\"#ffffba\"")
+            } else if (androidProjects.contains(project)) {
+                traits.add("fillcolor=\"#baffc9\"")
+            } else if (javaProjects.contains(project)) {
+                traits.add("fillcolor=\"#ffb3ba\"")
+            } else {
+                traits.add("fillcolor=\"#eeeeee\"")
+            }
+
+            dot.appendText("  \"${project.path}\" [${traits.joinToString(", ")}];\n")
+        }
+
+        dot.appendText("\n  {rank = same;")
+        for (project in projects) {
+            if (rootProjects.contains(project)) {
+                dot.appendText(" \"${project.path}\";")
+            }
+        }
+        dot.appendText("}\n")
+
+        for (sameRank in listOf(rankAndroid, rankDomain, rankRepository)) {
+            dot.appendText("\n  {rank = same;")
+            for (project in sameRank) {
+                dot.appendText(" \"${project.path}\";")
+            }
+            dot.appendText("}\n")
+        }
+
+        dot.appendText("\n  # Dependencies\n\n")
+        dependencies.forEach { (key, traits) ->
+            dot.appendText("  \"${key.first.path}\" -> \"${key.second.path}\"")
+            if (traits.isNotEmpty()) {
+                dot.appendText(" [${traits.joinToString(", ")}]")
+            }
+            dot.appendText("\n")
+        }
+
+        dot.appendText("}\n")
+
+        project.exec {
+            commandLine = listOf("sh", "-c", "cd \"${dot.parentFile}\"; dot -Tpng -O project.dot")
+        }
+        println("Project module dependency graph created at ${dot.absolutePath}.png")
+    }
+}
+