Introduce ide-provisioning-api and load implementation via custom classloader

Author: asodja

build.gradle.kts

@@ -21,6 +21,7 @@ description = "A tool to profile and benchmark Gradle builds"
 
 val gradleRuntime by configurations.creating
 val profilerPlugins by configurations.creating
+val ideImplementation by configurations.creating
 
 dependencies {
     implementation(libs.toolingApi)
@@ -42,6 +43,7 @@ dependencies {
         because("To write JSON output")
     }
     implementation(project(":client-protocol"))
+    implementation(project(":ide-provisioning-api"))
 
 
     gradleRuntime(gradleApi())
@@ -52,6 +54,7 @@ dependencies {
     profilerPlugins(project(":studio-agent"))
     profilerPlugins(project(":heap-dump"))
     profilerPlugins(project(":studio-plugin"))
+    ideImplementation(project(":ide-provisioning"))
 
     runtimeOnly("org.slf4j:slf4j-simple:1.7.10")
     testImplementation(libs.bundles.testDependencies)
@@ -87,13 +90,29 @@ val generateHtmlReportJavaScript = tasks.register<NpxTask>("generateHtmlReportJa
     args.addAll(source.absolutePath, "--outfile", output.get().asFile.absolutePath)
 }
 
+val listIdeProvisioningDependencies = tasks.register("listIdeProvisioningDependencies") {
+    val input = ideImplementation.minus(gradleRuntime)
+    val output = project.layout.buildDirectory.file("ide-provisioning/ide-provisioning.txt")
+    inputs.files(input)
+    outputs.file(output)
+    doLast {
+        output.get().asFile.writeText(input.joinToString("\n") { it.name })
+    }
+}
+
 tasks.processResources {
     into("META-INF/jars") {
         from(profilerPlugins.minus(gradleRuntime)) {
             // Removing the version from the JARs here, since they are referenced by name in production code.
             rename("""(.*)-\d\.\d.*\.jar""", "${'$'}1.jar")
         }
     }
+    into("META-INF/jars") {
+        from(ideImplementation.minus(gradleRuntime))
+    }
+    into("META-INF/classpath") {
+        from(listIdeProvisioningDependencies)
+    }
     from(generateHtmlReportJavaScript)
 }
 

buildSrc/src/main/kotlin/profiler.java-library.gradle.kts

@@ -14,6 +14,13 @@ repositories {
         }
     }
     mavenCentral()
+    maven {
+        url = uri("https://www.jetbrains.com/intellij-repository/releases")
+    }
+    maven {
+        url = uri("https://cache-redirector.jetbrains.com/intellij-dependencies")
+    }
+
 }
 
 java {

settings.gradle.kts

@@ -14,6 +14,7 @@ include("client-protocol")
 include("instrumentation-support")
 include("studio-agent")
 include("studio-plugin")
+include("ide-provisioning-api")
 include("ide-provisioning")
 
 rootProject.children.forEach {

src/main/java/org/gradle/profiler/instrument/GradleInstrumentation.java

@@ -4,13 +4,18 @@
 import org.gradle.profiler.GeneratedInitScript;
 import org.gradle.profiler.GradleArgsCalculator;
 
+import java.io.BufferedReader;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.InputStreamReader;
 import java.io.PrintWriter;
+import java.io.UncheckedIOException;
+import java.net.URL;
 import java.nio.file.Files;
 import java.nio.file.StandardCopyOption;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * Represents some instrumentation that uses Gradle APIs and that is injected by gradle-profiler.
@@ -56,4 +61,20 @@ public static File unpackPlugin(String jarName) {
             throw UncheckedException.throwAsUncheckedException(e);
         }
     }
+
+    public static URL[] getClasspath(String classPathName) {
+        List<String> classpathJars = readLines("/META-INF/classpath/" + classPathName + ".txt");
+        return classpathJars.stream()
+            .map(jar -> GradleInstrumentation.class.getResource("/META-INF/jars/" + jar))
+            .toArray(URL[]::new);
+    }
+
+    private static List<String> readLines(String resourcePath) {
+        try (BufferedReader reader = new BufferedReader(new InputStreamReader(GradleInstrumentation.class.getResourceAsStream(resourcePath)))) {
+            return reader.lines()
+                .collect(Collectors.toList());
+        } catch (IOException e) {
+            throw new UncheckedIOException(e);
+        }
+    }
 }

src/main/java/org/gradle/profiler/studio/AndroidStudioSyncAction.java

@@ -2,10 +2,16 @@
 
 import org.gradle.profiler.BuildAction;
 import org.gradle.profiler.GradleClient;
+import org.gradle.profiler.ide.RunIde;
+import org.gradle.profiler.instrument.GradleInstrumentation;
 import org.gradle.profiler.result.BuildActionResult;
-import org.gradle.profiler.studio.StudioGradleClient;
 
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.Arrays;
 import java.util.List;
+import java.util.ServiceLoader;
 
 /**
  * A mock-up of Android studio sync.

subprojects/ide-provisioning-api/build.gradle.kts

@@ -0,0 +1,13 @@
+plugins {
+    id("profiler.embedded-library")
+    id("profiler.publication")
+}
+
+description = "Api for IDE provisioning capabilities for Gradle profiler"
+
+
+java {
+    toolchain {
+        languageVersion.set(JavaLanguageVersion.of(8))
+    }
+}

subprojects/ide-provisioning-api/src/main/java/org/gradle/profiler/ide/RunIde.java

@@ -0,0 +1,5 @@
+package org.gradle.profiler.ide;
+
+public interface RunIde {
+    void run();
+}

subprojects/ide-provisioning/build.gradle.kts

@@ -1,23 +1,18 @@
 plugins {
     id("profiler.embedded-library")
-    id("profiler.publication")
     kotlin("jvm") version "1.9.22"
 }
 
 description = "IDE provisioning capabilities for Gradle profiler"
 
-repositories {
-    maven { url = uri("https://www.jetbrains.com/intellij-repository/releases") }
-    maven { url = uri("https://cache-redirector.jetbrains.com/intellij-dependencies") }
-}
-
 java {
     toolchain {
         languageVersion.set(JavaLanguageVersion.of(17))
     }
 }
 
 dependencies {
+    implementation(project(":ide-provisioning-api"))
     implementation(libs.ideStarter) {
         exclude(group = "io.ktor")
     }

subprojects/ide-provisioning/src/main/java/org/gradle/profiler/ide/RunIdeImpl.java

@@ -0,0 +1,8 @@
+package org.gradle.profiler.ide;
+
+public class RunIdeImpl implements RunIde {
+    @Override
+    public void run() {
+        System.out.println("Running IDE");
+    }
+}

subprojects/ide-provisioning/src/main/resources/META-INF/services/org.gradle.profiler.ide.RunIde

@@ -0,0 +1 @@
+org.gradle.profiler.ide.RunIdeImpl