refactor: Proper task for muzzle printReferences

Author: bric3

buildSrc/src/main/kotlin/datadog/gradle/plugin/muzzle/MuzzlePlugin.kt

@@ -5,6 +5,8 @@ import datadog.gradle.plugin.muzzle.MuzzleMavenRepoUtils.muzzleDirectiveToArtifa
 import datadog.gradle.plugin.muzzle.MuzzleMavenRepoUtils.resolveVersionRange
 import datadog.gradle.plugin.muzzle.MuzzleReportUtils.dumpVersionRanges
 import datadog.gradle.plugin.muzzle.MuzzleReportUtils.mergeReports
+import datadog.gradle.plugin.muzzle.tasks.MuzzleTask
+import datadog.gradle.plugin.muzzle.tasks.MuzzlePrintReferencesTask
 import org.eclipse.aether.artifact.Artifact
 import org.gradle.api.NamedDomainObjectProvider
 import org.gradle.api.Plugin
@@ -89,11 +91,7 @@ class MuzzlePlugin : Plugin<Project> {
       dependsOn(compileMuzzle)
     }
 
-    project.tasks.register<MuzzleTask>("printReferences") {
-      description = "Print references created by instrumentation muzzle"
-      doLast {
-        printMuzzle(project)
-      }
+    project.tasks.register<MuzzlePrintReferencesTask>("printReferences") {
       dependsOn(compileMuzzle)
     }
 

buildSrc/src/main/kotlin/datadog/gradle/plugin/muzzle/tasks/AbstractMuzzleTask.kt

@@ -0,0 +1,9 @@
+package datadog.gradle.plugin.muzzle.tasks
+
+import org.gradle.api.DefaultTask
+
+abstract class AbstractMuzzleTask : DefaultTask() {
+  init {
+    group = "Muzzle"
+  }
+}

buildSrc/src/main/kotlin/datadog/gradle/plugin/muzzle/tasks/MuzzlePrintReferencesTask.kt

@@ -0,0 +1,28 @@
+package datadog.gradle.plugin.muzzle.tasks
+
+import datadog.gradle.plugin.muzzle.mainSourceSet
+import org.gradle.api.Project
+import org.gradle.api.tasks.TaskAction
+import java.lang.reflect.Method
+import java.net.URLClassLoader
+import javax.inject.Inject
+
+abstract class MuzzlePrintReferencesTask @Inject constructor(
+  private val instrumentationProject: Project
+) : AbstractMuzzleTask() {
+  init {
+    description = "Print references created by instrumentation muzzle"
+  }
+
+  @TaskAction
+  fun printMuzzle() {
+    val cp = instrumentationProject.mainSourceSet.runtimeClasspath
+    val cl = URLClassLoader(cp.map { it.toURI().toURL() }.toTypedArray(), null)
+    val printMethod: Method = cl.loadClass("datadog.trace.agent.tooling.muzzle.MuzzleVersionScanPlugin")
+      .getMethod(
+        "printMuzzleReferences",
+        ClassLoader::class.java
+      )
+    printMethod.invoke(null, cl)
+  }
+}

…tadog/gradle/plugin/muzzle/MuzzleTask.kt …gradle/plugin/muzzle/tasks/MuzzleTask.ktbuildSrc/src/main/kotlin/datadog/gradle/plugin/muzzle/MuzzleTask.kt renamed to buildSrc/src/main/kotlin/datadog/gradle/plugin/muzzle/tasks/MuzzleTask.kt buildSrc/src/main/kotlin/datadog/gradle/plugin/muzzle/MuzzleTask.kt renamed to buildSrc/src/main/kotlin/datadog/gradle/plugin/muzzle/tasks/MuzzleTask.kt

@@ -1,26 +1,23 @@
-package datadog.gradle.plugin.muzzle
+package datadog.gradle.plugin.muzzle.tasks
 
+import datadog.gradle.plugin.muzzle.MuzzleAction
+import datadog.gradle.plugin.muzzle.MuzzleDirective
+import datadog.gradle.plugin.muzzle.allMainSourceSet
+import datadog.gradle.plugin.muzzle.mainSourceSet
 import org.gradle.api.DefaultTask
 import org.gradle.api.NamedDomainObjectProvider
 import org.gradle.api.Project
 import org.gradle.api.artifacts.Configuration
 import org.gradle.api.file.FileCollection
 import org.gradle.api.invocation.BuildInvocationDetails
-import org.gradle.api.tasks.SourceSet
-import org.gradle.api.tasks.SourceSetContainer
 import org.gradle.jvm.toolchain.JavaLanguageVersion
 import org.gradle.jvm.toolchain.JavaToolchainService
-import org.gradle.kotlin.dsl.getByType
 import org.gradle.workers.WorkerExecutor
 import java.lang.reflect.Method
 import java.net.URLClassLoader
 import javax.inject.Inject
 
-abstract class MuzzleTask : DefaultTask() {
-  init {
-    group = "Muzzle"
-  }
-
+abstract class MuzzleTask : AbstractMuzzleTask() {
   @get:Inject
   abstract val javaToolchainService: JavaToolchainService
 
@@ -63,17 +60,6 @@ abstract class MuzzleTask : DefaultTask() {
     }
   }
 
-  fun printMuzzle(instrumentationProject: Project) {
-    val cp = instrumentationProject.mainSourceSet.runtimeClasspath
-    val cl = URLClassLoader(cp.map { it.toURI().toURL() }.toTypedArray(), null)
-    val printMethod: Method = cl.loadClass("datadog.trace.agent.tooling.muzzle.MuzzleVersionScanPlugin")
-      .getMethod(
-        "printMuzzleReferences",
-        ClassLoader::class.java
-      )
-    printMethod.invoke(null, cl)
-  }
-
   private fun createAgentClassPath(project: Project): FileCollection {
     project.logger.info("Creating agent classpath for $project")
     val cp = project.files()