Add Gradle plugin to aggregate all graph tasks

app/build.gradle

@@ -132,6 +132,14 @@ dependencies {
   androidTestImplementation project(':core-testing')
 }
 
+apply plugin: 'com.jraska.module.graph'
+
+moduleGraphRules {
+  maxHeight = 4
+  moduleLayersFromTheTop = [":feature", ":lib", ":core"]
+  restrinctInLayerDependencies = [":feature", ":lib"]
+}
+
 apply plugin: 'com.google.gms.google-services'
 repositories {
   mavenCentral()

build.gradle

@@ -1,7 +1,5 @@
-import com.jraska.github.client.tasks.*
-
 buildscript {
-  ext.kotlin_version = '1.3.50'
+  ext.kotlin_version = '1.3.61'
   repositories {
     google()
     maven { url 'https://maven.fabric.io/public' }
@@ -30,25 +28,6 @@ task clean(type: Delete) {
   delete rootProject.buildDir
 }
 
-task generateModulesGraph(type: GenerateModulesGraphTask)
-
-task generateOneModuleGraph(type: GenerateOneModuleGraphTask) {
-  moduleName = ":app"
-}
-
-task assertDependencyTreeHeight(type: AssertModuleTreeHeightTask) {
-  moduleName = ":app"
-  maxHeight = 4
-}
-
-task assertNoFeatureDependencies(type: AssertInLayerDependencies) {
-  layerPrefix = ":feature"
-}
-
-task assertCorrectLayerDependencies(type: AssertCrossLayerDependencies) {
-  layersFromTheTop = [":feature", ":lib", ":core"]
-}
-
 subprojects {
   apply plugin: "org.jlleitschuh.gradle.ktlint"
   project.plugins.whenPluginAdded { plugin ->

buildSrc/build.gradle

@@ -1,22 +1,22 @@
 buildscript {
   repositories {
     jcenter()
-//        maven { url "https://plugins.gradle.org/m2/" }
   }
   dependencies {
-    classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:1.3.50"
+    classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:1.3.61"
   }
 }
 
 apply plugin: 'kotlin'
+apply plugin: 'java-gradle-plugin'
 
 repositories {
   jcenter()
 }
 
 dependencies {
   implementation gradleApi()
-  implementation "org.jetbrains.kotlin:kotlin-stdlib-jdk8:1.3.50"
+  implementation "org.jetbrains.kotlin:kotlin-stdlib-jdk8:1.3.61"
 
   testImplementation 'junit:junit:4.12'
 }
@@ -31,3 +31,12 @@ compileTestKotlin {
     jvmTarget = "1.8"
   }
 }
+
+gradlePlugin {
+  plugins {
+    moduleGraphAssertions {
+      id = 'com.jraska.module.graph'
+      implementationClass = 'com.jraska.module.graph.ModuleGraphAssertionsPlugin'
+    }
+  }
+}

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

@@ -7,7 +7,7 @@ import org.gradle.api.GradleException
 import org.gradle.api.tasks.Input
 import org.gradle.api.tasks.TaskAction
 
-open class AssertCrossLayerDependencies : DefaultTask() {
+open class AssertLayersOrderTask : DefaultTask() {
   @Input
   lateinit var layersFromTheTop: Array<String>
 

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

@@ -7,7 +7,7 @@ import org.gradle.api.GradleException
 import org.gradle.api.tasks.Input
 import org.gradle.api.tasks.TaskAction
 
-open class AssertInLayerDependencies : DefaultTask() {
+open class AssertNoInLayerDependencies : DefaultTask() {
   @Input
   lateinit var layerPrefix: String
 

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

@@ -17,6 +17,5 @@ open class GenerateModulesGraphTask : DefaultTask() {
 
     println(allModulesTree.statistics())
     println(GraphvizWriter.toGraphviz(allModulesTree, layers.toSet()))
-    println(KotlinCodeWriter.toKotlinCode(allModulesTree))
   }
 }

buildSrc/src/main/java/com/jraska/module/graph/Api.kt

@@ -0,0 +1,17 @@
+package com.jraska.module.graph
+
+object Api {
+  object Tasks {
+    const val GENERATE_GRAPHVIZ = "generateModulesGrapvizText"
+
+    const val ASSERT_ALL = "assertModulesGraph"
+    const val ASSERT_MAX_HEIGHT = "assertMaxHeight"
+    const val ASSERT_LAYER_ORDER = "assertModuleLayersOrder"
+    const val ASSERT_NO_IN_LAYER_PREFIX = "assertNoDependenciesWithin"
+  }
+
+  const val CHECK_TASK = "check"
+
+  const val EXTENSION_ROOT = "moduleGraphRules"
+}
+

buildSrc/src/main/java/com/jraska/module/graph/GraphRulesExtension.kt

@@ -0,0 +1,8 @@
+package com.jraska.module.graph
+
+open class GraphRulesExtension {
+  var appModuleName = ":app"
+  var maxHeight: Int = 0
+  var moduleLayersFromTheTop = emptyArray<String>()
+  var restrinctInLayerDependencies = emptyArray<String>()
+}

buildSrc/src/main/java/com/jraska/module/graph/ModuleGraphAssertionsPlugin.kt

@@ -0,0 +1,80 @@
+package com.jraska.module.graph
+
+import com.jraska.github.client.tasks.AssertLayersOrderTask
+import com.jraska.github.client.tasks.AssertModuleTreeHeightTask
+import com.jraska.github.client.tasks.AssertNoInLayerDependencies
+import com.jraska.github.client.tasks.GenerateModulesGraphTask
+import com.jraska.module.graph.Api.Tasks
+import org.gradle.api.Plugin
+import org.gradle.api.Project
+import org.gradle.api.Task
+import org.gradle.language.base.plugins.LifecycleBasePlugin.VERIFICATION_GROUP
+import java.util.Locale
+
+@Suppress("unused") // Used as plugin
+class ModuleGraphAssertionsPlugin : Plugin<Project> {
+
+  override fun apply(project: Project) {
+    val graphRules = project.extensions.create(GraphRulesExtension::class.java, Api.EXTENSION_ROOT, GraphRulesExtension::class.java)
+
+    project.afterEvaluate {
+      addModulesAssertions(project, graphRules)
+    }
+  }
+
+  private fun addModulesAssertions(project: Project, graphRules: GraphRulesExtension) {
+    project.addModuleGraphGeneration()
+
+    val allAssertionsTask = project.tasks.create(Tasks.ASSERT_ALL)
+    allAssertionsTask.group = VERIFICATION_GROUP
+    project.tasks.find { it.name == Api.CHECK_TASK }?.dependsOn(allAssertionsTask)
+
+    project.addMaxHeightTasks(graphRules).forEach { allAssertionsTask.dependsOn(it) }
+    project.addModuleLayersTasks(graphRules).forEach { allAssertionsTask.dependsOn(it) }
+    project.addInLayerDependencyTasks(graphRules).forEach { allAssertionsTask.dependsOn(it) }
+  }
+
+  private fun Project.addModuleGraphGeneration() {
+    tasks.create(Tasks.GENERATE_GRAPHVIZ, GenerateModulesGraphTask::class.java)
+  }
+
+  private fun Project.addMaxHeightTasks(graphRules: GraphRulesExtension): List<Task> {
+    if (graphRules.maxHeight <= 0) {
+      return emptyList()
+    }
+
+    val task = tasks.create(Tasks.ASSERT_MAX_HEIGHT, AssertModuleTreeHeightTask::class.java)
+    task.maxHeight = graphRules.maxHeight
+    task.moduleName = graphRules.appModuleName
+    task.group = VERIFICATION_GROUP
+
+    return listOf(task)
+  }
+
+  private fun Project.addModuleLayersTasks(graphRules: GraphRulesExtension): List<Task> {
+    if (graphRules.moduleLayersFromTheTop.isEmpty()) {
+      return emptyList()
+    }
+
+    val task = tasks.create(Tasks.ASSERT_LAYER_ORDER, AssertLayersOrderTask::class.java)
+    task.layersFromTheTop = graphRules.moduleLayersFromTheTop
+    task.group = VERIFICATION_GROUP
+
+    return listOf(task)
+  }
+
+  private fun Project.addInLayerDependencyTasks(graphRules: GraphRulesExtension): List<Task> {
+    return graphRules.restrinctInLayerDependencies.map { layerPrefix ->
+      val taskNameSuffix = layerPrefix.replace(":", "").capitalizeFirst()
+      val task = tasks.create("${Tasks.ASSERT_NO_IN_LAYER_PREFIX}$taskNameSuffix", AssertNoInLayerDependencies::class.java)
+      task.layerPrefix = layerPrefix
+      task.group = VERIFICATION_GROUP
+
+      return@map task
+    }
+  }
+
+  private fun String.capitalizeFirst(): String {
+    return this.substring(0, 1).toUpperCase(Locale.US).plus(this.substring(1))
+  }
+}