Update structure of buildSrc

build.gradle

@@ -1,4 +1,4 @@
-import com.jraska.github.client.*
+import com.jraska.github.client.tasks.*
 
 buildscript {
   ext.kotlin_version = '1.3.50'

buildSrc/src/main/java/com/jraska/github/client/DependencyTreeFactory.kt → buildSrc/src/main/java/com/jraska/github/client/GradleDependencyGraphFactory.kt

@@ -1,19 +1,14 @@
 package com.jraska.github.client
 
+import com.jraska.github.client.graph.DependencyGraph
 import org.gradle.api.Project
 import org.gradle.api.internal.artifacts.dependencies.DefaultProjectDependency
 
-object DependencyTreeFactory {
+object GradleDependencyGraphFactory {
 
   fun create(project: Project): DependencyGraph {
     val dependencies = project.listDependencyPairs()
-
-    val modulesTree = DependencyGraph()
-    dependencies.forEach {
-      modulesTree.addEdge(it.first, it.second)
-    }
-
-    return modulesTree
+    return DependencyGraph.create(dependencies)
   }
 
   private fun Project.listDependencyPairs(): List<Pair<String, String>> {

buildSrc/src/main/java/com/jraska/github/client/DependencyGraph.kt → buildSrc/src/main/java/com/jraska/github/client/graph/DependencyGraph.kt

@@ -1,4 +1,4 @@
-package com.jraska.github.client
+package com.jraska.github.client.graph
 
 class DependencyGraph() {
   private val nodes = mutableMapOf<String, Node>()
@@ -47,10 +47,6 @@ class DependencyGraph() {
     )
   }
 
-  fun addEdge(from: String, to: String) {
-    getOrCreate(from).dependsOn.add(getOrCreate(to))
-  }
-
   fun subTree(key: String): DependencyGraph {
     val dependencyTree = DependencyGraph()
 
@@ -66,6 +62,10 @@ class DependencyGraph() {
     }
   }
 
+  private fun addEdge(from: String, to: String) {
+    getOrCreate(from).dependsOn.add(getOrCreate(to))
+  }
+
   private fun countEdges(): Int {
     return nodes().flatMap { node -> node.dependsOn }.count()
   }
@@ -100,4 +100,12 @@ class DependencyGraph() {
       }
     }
   }
+
+  companion object {
+    fun create(dependencies: List<Pair<String, String>>): DependencyGraph {
+      val graph = DependencyGraph()
+      dependencies.forEach { graph.addEdge(it.first, it.second) }
+      return graph
+    }
+  }
 }

buildSrc/src/main/java/com/jraska/github/client/GraphStatistics.kt → buildSrc/src/main/java/com/jraska/github/client/graph/GraphStatistics.kt

@@ -1,4 +1,4 @@
-package com.jraska.github.client
+package com.jraska.github.client.graph
 
 data class GraphStatistics(
   val modulesCount: Int,

buildSrc/src/main/java/com/jraska/github/client/GraphvizGenerator.kt → buildSrc/src/main/java/com/jraska/github/client/graph/GraphvizWriter.kt

@@ -1,6 +1,6 @@
-package com.jraska.github.client
+package com.jraska.github.client.graph
 
-object GraphvizGenerator {
+object GraphvizWriter {
   fun toGraphviz(dependencyGraph: DependencyGraph, groups: Set<String> = emptySet()): String {
 
     val longestPathConnections = dependencyGraph.longestPath()

buildSrc/src/main/java/com/jraska/github/client/LongestPath.kt → buildSrc/src/main/java/com/jraska/github/client/graph/LongestPath.kt

@@ -1,4 +1,4 @@
-package com.jraska.github.client
+package com.jraska.github.client.graph
 
 class LongestPath(
   val nodeNames: List<String>

buildSrc/src/main/java/com/jraska/github/client/AssertCrossLayerDependencies.kt → buildSrc/src/main/java/com/jraska/github/client/tasks/AssertCrossLayerDependencies.kt

@@ -1,5 +1,7 @@
-package com.jraska.github.client
+package com.jraska.github.client.tasks
 
+import com.jraska.github.client.GradleDependencyGraphFactory
+import com.jraska.github.client.graph.DependencyGraph
 import org.gradle.api.DefaultTask
 import org.gradle.api.GradleException
 import org.gradle.api.tasks.Input
@@ -11,7 +13,7 @@ open class AssertCrossLayerDependencies : DefaultTask() {
 
   @TaskAction
   fun run() {
-    val modulesTree = DependencyTreeFactory.create(project)
+    val modulesTree = GradleDependencyGraphFactory.create(project)
 
     verifyAllLayersHaveModule(modulesTree)
 
@@ -27,7 +29,7 @@ open class AssertCrossLayerDependencies : DefaultTask() {
   private fun buildErrorMessage(againstLayerDependencies: List<Pair<DependencyGraph.Node, DependencyGraph.Node>>): String {
     val errorsMessage = againstLayerDependencies.joinToString("\n") { " Module '${it.first.key}' cannot depend on '${it.second.key}'." }
 
-    return "Dependencies agaist direction of layers '${layersDependencyString()}' are not allowed. The violating dependencies are: \n$errorsMessage"
+    return "Dependencies against direction of layers '${layersDependencyString()}' are not allowed. The violating dependencies are: \n$errorsMessage"
   }
 
   private fun layersDependencyString(): String {
@@ -38,10 +40,7 @@ open class AssertCrossLayerDependencies : DefaultTask() {
     val nodes = modulesTree.nodes()
 
     for (layerPrefix in layersFromTheTop) {
-      val someNodeInLayer = nodes.find { it.key.startsWith(layerPrefix) }
-      if (someNodeInLayer == null) {
-        throw GradleException("There is no module, which belongs to layer '$layerPrefix'")
-      }
+      nodes.find { it.key.startsWith(layerPrefix) } ?: throw GradleException("There is no module, which belongs to layer '$layerPrefix'")
     }
   }
 

buildSrc/src/main/java/com/jraska/github/client/AssertInLayerDependencies.kt → buildSrc/src/main/java/com/jraska/github/client/tasks/AssertInLayerDependencies.kt

@@ -1,5 +1,7 @@
-package com.jraska.github.client
+package com.jraska.github.client.tasks
 
+import com.jraska.github.client.GradleDependencyGraphFactory
+import com.jraska.github.client.graph.DependencyGraph
 import org.gradle.api.DefaultTask
 import org.gradle.api.GradleException
 import org.gradle.api.tasks.Input
@@ -11,7 +13,7 @@ open class AssertInLayerDependencies : DefaultTask() {
 
   @TaskAction
   fun run() {
-    val modulesTree = DependencyTreeFactory.create(project)
+    val modulesTree = GradleDependencyGraphFactory.create(project)
 
     val inLayerDependencies = modulesTree.nodes()
       .filter { it.key.startsWith(layerPrefix) }

buildSrc/src/main/java/com/jraska/github/client/AssertModuleTreeHeightTask.kt → buildSrc/src/main/java/com/jraska/github/client/tasks/AssertModuleTreeHeightTask.kt

@@ -1,5 +1,6 @@
-package com.jraska.github.client
+package com.jraska.github.client.tasks
 
+import com.jraska.github.client.GradleDependencyGraphFactory
 import org.gradle.api.DefaultTask
 import org.gradle.api.GradleException
 import org.gradle.api.tasks.Input
@@ -14,7 +15,7 @@ open class AssertModuleTreeHeightTask : DefaultTask() {
 
   @TaskAction
   fun run() {
-    val modulesTree = DependencyTreeFactory.create(project)
+    val modulesTree = GradleDependencyGraphFactory.create(project)
 
     val height = modulesTree.heightOf(moduleName)
     if (height > maxHeight) {

buildSrc/src/main/java/com/jraska/github/client/tasks/GenerateModulesGraphTask.kt

@@ -0,0 +1,20 @@
+package com.jraska.github.client.tasks
+
+import com.jraska.github.client.GradleDependencyGraphFactory
+import com.jraska.github.client.graph.GraphvizWriter
+import org.gradle.api.DefaultTask
+import org.gradle.api.tasks.Input
+import org.gradle.api.tasks.TaskAction
+
+open class GenerateModulesGraphTask : DefaultTask() {
+  @Input
+  var layers: Array<String> = emptyArray()
+
+  @TaskAction
+  fun run() {
+    val allModulesTree = GradleDependencyGraphFactory.create(project)
+
+    println(allModulesTree.statistics())
+    println(GraphvizWriter.toGraphviz(allModulesTree, layers.toSet()))
+  }
+}

buildSrc/src/main/java/com/jraska/github/client/GenerateOneModuleGraphTask.kt → buildSrc/src/main/java/com/jraska/github/client/tasks/GenerateOneModuleGraphTask.kt

@@ -1,5 +1,7 @@
-package com.jraska.github.client
+package com.jraska.github.client.tasks
 
+import com.jraska.github.client.GradleDependencyGraphFactory
+import com.jraska.github.client.graph.GraphvizWriter
 import org.gradle.api.DefaultTask
 import org.gradle.api.tasks.Input
 import org.gradle.api.tasks.TaskAction
@@ -10,10 +12,10 @@ open class GenerateOneModuleGraphTask : DefaultTask() {
 
   @TaskAction
   fun run() {
-    val allModulesTree = DependencyTreeFactory.create(project)
+    val allModulesTree = GradleDependencyGraphFactory.create(project)
     val moduleTree = allModulesTree.subTree(moduleName)
 
     println(moduleTree.statistics())
-    println(GraphvizGenerator.toGraphviz(moduleTree))
+    println(GraphvizWriter.toGraphviz(moduleTree))
   }
 }

buildSrc/src/test/java/com/jraska/github/client/graph/DependencyGraphTest.kt

@@ -0,0 +1,87 @@
+package com.jraska.github.client.graph
+
+import org.junit.Test
+
+class DependencyGraphTest {
+  @Test
+  fun correctHeightIsMaintained() {
+    val dependencyTree = DependencyGraph.create(
+      listOf(
+        "app" to "feature",
+        "app" to "lib",
+        "feature" to "lib",
+        "lib" to "core"
+      )
+    )
+
+    assert(dependencyTree.heightOf("app") == 3)
+  }
+
+  @Test
+  fun findsProperLongestPath() {
+    val dependencyTree = DependencyGraph.create(
+      listOf(
+        "app" to "feature",
+        "app" to "lib",
+        "app" to "core",
+        "feature" to "lib",
+        "lib" to "core"
+      )
+    )
+
+    assert(dependencyTree.longestPath("app").nodeNames == listOf("app", "feature", "lib", "core"))
+  }
+
+  @Test
+  fun findsProperRoot() {
+    val dependencyTree = DependencyGraph.create(
+      listOf(
+        "feature" to "lib",
+        "lib" to "core",
+        "app" to "feature",
+        "app" to "lib",
+        "app" to "core"
+      )
+    )
+
+    assert(dependencyTree.findRoot().key == "app")
+  }
+
+  @Test
+  fun createSubtreeProperly() {
+    val dependencyTree = DependencyGraph.create(
+      listOf(
+        "feature" to "lib",
+        "lib" to "core",
+        "app" to "feature",
+        "feature" to "core",
+        "app" to "core"
+      )
+    )
+
+    val subTree = dependencyTree.subTree("feature")
+
+    assert(subTree.findRoot().key == "feature")
+    assert(subTree.heightOf("feature") == 2)
+    assert(subTree.longestPath("feature").nodeNames == listOf("feature", "lib", "core"))
+  }
+
+  @Test
+  fun createCountStatisticsWell() {
+    val dependencyTree = DependencyGraph.create(
+      listOf(
+        "feature" to "lib",
+        "lib" to "core",
+        "app" to "feature",
+        "feature" to "core",
+        "app" to "core"
+      )
+    )
+
+    val statistics = dependencyTree.statistics()
+
+    assert(statistics.height == 3)
+    assert(statistics.modulesCount == 4)
+    assert(statistics.edgesCount == 5)
+  }
+}