Add IDEA provisioning

build.gradle.kts

@@ -33,7 +33,7 @@ dependencies {
     implementation("com.github.javaparser:javaparser-core:3.18.0")
     implementation("net.sf.jopt-simple:jopt-simple:5.0.4")
     implementation("org.apache.ant:ant-compress:1.5")
-    implementation("commons-io:commons-io:2.6")
+    implementation(libs.commonIo)
     implementation("org.openjdk.jmc:flightrecorder:8.0.1")
     implementation("com.googlecode.plist:dd-plist:1.23") {
         because("To extract launch details from Android Studio installation")

gradle/libs.versions.toml

@@ -3,6 +3,8 @@ groovy = "3.0.15"
 spock = "2.1-groovy-3.0"
 
 [libraries]
+commonIo = "commons-io:commons-io:2.11.0"
+ideStarter = "com.jetbrains.intellij.tools:ide-starter-squashed:233.13135.103"
 toolingApi = "org.gradle:gradle-tooling-api:7.2"
 
 spock-core = { module = "org.spockframework:spock-core", version.ref = "spock" }

settings.gradle.kts

@@ -14,6 +14,7 @@ include("client-protocol")
 include("instrumentation-support")
 include("studio-agent")
 include("studio-plugin")
+include("ide-provisioning")
 
 rootProject.children.forEach {
     it.projectDir = rootDir.resolve( "subprojects/${it.name}")

src/main/java/org/gradle/profiler/CommandExec.java

@@ -1,5 +1,7 @@
 package org.gradle.profiler;
 
+import org.apache.commons.io.output.TeeOutputStream;
+
 import javax.annotation.Nullable;
 import java.io.ByteArrayOutputStream;
 import java.io.File;

src/main/java/org/gradle/profiler/Logging.java

@@ -1,5 +1,7 @@
 package org.gradle.profiler;
 
+import org.apache.commons.io.output.TeeOutputStream;
+
 import java.io.*;
 
 public class Logging {

src/test/groovy/org/gradle/profiler/AbstractIntegrationTest.groovy

@@ -1,5 +1,6 @@
 package org.gradle.profiler
 
+import org.apache.commons.io.output.TeeOutputStream
 import spock.lang.Specification
 
 class AbstractIntegrationTest extends Specification {

subprojects/ide-provisioning/build.gradle.kts

@@ -0,0 +1,24 @@
+plugins {
+    id("profiler.embedded-library")
+}
+
+java {
+    toolchain {
+        languageVersion.set(JavaLanguageVersion.of(17))
+    }
+}
+
+repositories {
+    maven { url = uri("https://www.jetbrains.com/intellij-repository/releases") }
+    maven { url = uri("https://cache-redirector.jetbrains.com/intellij-dependencies") }
+}
+
+dependencies {
+    implementation(libs.ideStarter) {
+        exclude(group = "io.ktor")
+        exclude(group = "com.jetbrains.infra")
+        exclude(group = "com.jetbrains.intellij.remoteDev")
+    }
+    testImplementation(libs.bundles.testDependencies)
+    testImplementation(libs.commonIo)
+}

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

@@ -0,0 +1,34 @@
+package org.gradle.profiler.ide;
+
+import kotlin.io.FilesKt;
+import org.gradle.profiler.ide.idea.IDEA;
+import org.gradle.profiler.ide.idea.IDEAProvider;
+
+import java.io.File;
+import java.nio.file.Path;
+
+public class DefaultIdeProvider implements IdeProvider<Ide> {
+
+    private final IDEAProvider ideaProvider;
+
+    public DefaultIdeProvider(IDEAProvider ideaProvider) {
+        this.ideaProvider = ideaProvider;
+    }
+
+    @Override
+    public File provideIde(Ide ide, Path homeDir, Path downloadsDir) {
+        File result;
+        if (ide instanceof IDEA) {
+            result = ideaProvider.provideIde((IDEA) ide, homeDir, downloadsDir);
+        } else {
+            throw new IllegalArgumentException("Unknown IDE to provide");
+        }
+
+        cleanup(homeDir, downloadsDir);
+        return result;
+    }
+
+    private void cleanup(Path homeDir, Path downloadsDir) {
+        FilesKt.deleteRecursively(downloadsDir.toFile());
+    }
+}

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

@@ -0,0 +1,9 @@
+package org.gradle.profiler.ide;
+
+import org.jetbrains.annotations.NotNull;
+
+public interface Ide {
+    @NotNull
+    String getVersion();
+}
+

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

@@ -0,0 +1,9 @@
+package org.gradle.profiler.ide;
+
+import java.io.File;
+import java.nio.file.Path;
+
+public interface IdeProvider<T extends Ide> {
+
+    File provideIde(T ide, Path homeDir, Path downloadsDir);
+}

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

@@ -0,0 +1,18 @@
+package org.gradle.profiler.ide.idea;
+
+import org.gradle.profiler.ide.Ide;
+import org.jetbrains.annotations.NotNull;
+
+public class IDEA implements Ide {
+    public static IDEA LATEST = new IDEA("");
+    private final String version;
+    public IDEA(String version) {
+        this.version = version;
+    }
+
+    @NotNull
+    @Override
+    public String getVersion() {
+        return version;
+    }
+}

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

@@ -0,0 +1,74 @@
+package org.gradle.profiler.ide.idea;
+
+import com.intellij.ide.starter.community.PublicIdeDownloader;
+import com.intellij.ide.starter.community.model.BuildType;
+import com.intellij.ide.starter.ide.IdeArchiveExtractor;
+import com.intellij.ide.starter.ide.IdeDownloader;
+import com.intellij.ide.starter.ide.installer.IdeInstallerFile;
+import com.intellij.ide.starter.models.IdeInfo;
+import org.gradle.profiler.ide.IdeProvider;
+import org.jetbrains.annotations.Nullable;
+
+import java.io.File;
+import java.nio.file.Path;
+import java.util.Collections;
+
+public class IDEAProvider implements IdeProvider<IDEA> {
+    private final IdeDownloader ideDownloader;
+    private final IdeArchiveExtractor ideArchiveExtractor;
+
+    public IDEAProvider() {
+        this.ideDownloader = PublicIdeDownloader.INSTANCE;
+        this.ideArchiveExtractor = IdeArchiveExtractor.INSTANCE;
+    }
+
+    @Override
+    public File provideIde(IDEA ide, Path homeDir, Path downloadsDir) {
+        IdeInfo ideInfo = new IdeInfo(
+            "IC",
+            "Idea",
+            "idea",
+            BuildType.EAP.getType(),
+            Collections.emptyList(),
+            "", // latest
+            ide.getVersion(),
+            null,
+            null,
+            "IDEA Community",
+            info -> null
+        );
+
+        String version = ideInfo.getVersion().isEmpty()
+            ? "latest"
+            : ideInfo.getVersion();
+
+        File unpackDir = homeDir
+            .resolve(ideInfo.getInstallerFilePrefix())
+            .resolve(version)
+            .toFile();
+
+        File unpackedIde = getUnpackedIde(unpackDir);
+
+        if (unpackedIde != null) {
+            System.out.println("Downloading is skipped, get " + ideInfo.getFullName() + " from cache");
+            return unpackedIde;
+        }
+
+        IdeInstallerFile installerFile = ideDownloader.downloadIdeInstaller(ideInfo, downloadsDir);
+        ideArchiveExtractor.unpackIdeIfNeeded(installerFile.getInstallerFile().toFile(), unpackDir);
+
+        return unpackDir.listFiles()[0];
+    }
+
+    @Nullable
+    private static File getUnpackedIde(File unpackDir) {
+        File[] unpackedFiles = unpackDir.listFiles();
+        if (unpackedFiles == null || unpackedFiles.length == 0) {
+            return null;
+        }
+        if (unpackedFiles.length == 1) {
+            return unpackedFiles[0];
+        }
+        throw new IllegalStateException("Unexpected content in " + unpackDir);
+    }
+}

subprojects/ide-provisioning/src/test/groovy/org/gradle/profiler/ide/AbstractIdeProvisioningTest.groovy

@@ -0,0 +1,28 @@
+package org.gradle.profiler.ide
+
+import org.apache.commons.io.output.TeeOutputStream
+import org.junit.Rule
+import org.junit.rules.TemporaryFolder
+import spock.lang.Specification
+
+class AbstractIdeProvisioningTest extends Specification {
+
+    private ByteArrayOutputStream outputBuffer
+
+    @Rule
+    protected TemporaryFolder tmpDir = new TemporaryFolder(new File("build/tmp/"))
+
+    def setup() {
+        outputBuffer = new ByteArrayOutputStream()
+        System.out = new PrintStream(new TeeOutputStream(System.out, outputBuffer))
+    }
+
+    protected void outputContains(String content) {
+        assert getOutput().contains(content.trim())
+    }
+
+    protected String getOutput() {
+        System.out.flush()
+        return new String(outputBuffer.toByteArray())
+    }
+}

subprojects/ide-provisioning/src/test/groovy/org/gradle/profiler/ide/IdeProviderTest.groovy

@@ -0,0 +1,33 @@
+package org.gradle.profiler.ide
+
+
+import org.gradle.profiler.ide.idea.IDEA
+import org.gradle.profiler.ide.idea.IDEAProvider
+
+class IdeProviderTest extends AbstractIdeProvisioningTest {
+
+    def "can provide IDEA"() {
+        given:
+        def workDir = tmpDir.newFolder().toPath().toAbsolutePath()
+        def downloadsDir = workDir.resolve("downloads")
+        def ideHomeDir = workDir.resolve("ide")
+        def ideProvider = new DefaultIdeProvider(new IDEAProvider())
+
+        when:
+        def ide = ideProvider.provideIde(IDEA.LATEST, ideHomeDir, downloadsDir)
+
+        then:
+        outputContains("Downloading https://")
+        ide.exists()
+
+        when:
+        def ide2 = ideProvider.provideIde(IDEA.LATEST, ideHomeDir, downloadsDir)
+
+        then:
+        outputContains("Downloading is skipped, get IDEA Community from cache")
+        ide == ide2
+
+        and:
+        !downloadsDir.toFile().exists()
+    }
+}