Support @composable functions (#83)

alonalbert · web-flow · commit e667b589d26a · 2025-05-08T15:56:45.000-07:00
Download required runtime lib and compiler plugin if code is Compose.
diff --git a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/DependencyCache.kt b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/DependencyCache.kt
@@ -0,0 +1,92 @@
+/*
+ * Copyright (C) 2025 Romain Guy
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package dev.romainguy.kotlin.explorer
+
+import java.io.FileNotFoundException
+import java.net.URL
+import java.nio.file.Path
+import java.nio.file.StandardOpenOption.*
+import java.util.zip.ZipInputStream
+import kotlin.io.path.createParentDirectories
+import kotlin.io.path.notExists
+import kotlin.io.path.outputStream
+
+private const val MAVEN_REPO = "https://repo1.maven.org/maven2"
+private const val GOOGLE_REPO = "https://maven.google.com"
+private val repos = listOf(MAVEN_REPO, GOOGLE_REPO)
+
+private const val BASE_PATH = "%1\$s/%2\$s/%3\$s/%2\$s-%3\$s"
+
+class DependencyCache(private val root: Path) {
+
+    fun getDependency(group: String, name: String, version: String, onOutput: (String) -> Unit): Path {
+        val basePath = BASE_PATH.format(group.replace('.', '/'), name, version)
+        val dst = root.resolve("$basePath.jar")
+        if (dst.notExists()) {
+            onOutput("Downloading artifact $group:$name:$version")
+            repos.forEach {
+                if (getDependency(it, basePath, dst, onOutput)) {
+                    return dst
+                }
+            }
+            onOutput("Could not find artifact $group:$name:$version")
+        }
+        return dst
+    }
+
+    /**
+     * Try to download a jar from a repo.
+     *
+     * First tries to download the `jar` file directly. If the `jar` is not found, will try to download an `aar` and
+     * extract the `classes.ja` file from it.
+     *
+     * @return true if the repo owns the artifact.
+     */
+    private fun getDependency(repo: String, basePath: String, dst: Path, onOutput: (String) -> Unit): Boolean {
+        try {
+            // Does the artifact exist in repo?
+            URL("$repo/$basePath.pom").openStream().reader().close()
+        } catch (_: FileNotFoundException) {
+            return false
+        }
+        dst.createParentDirectories()
+        dst.outputStream(CREATE, WRITE, TRUNCATE_EXISTING).use { outputStream ->
+            try {
+                URL("$repo/$basePath.jar").openStream().use {
+                    it.copyTo(outputStream)
+                }
+            } catch (_: FileNotFoundException) {
+                val aar = "$repo/$basePath.aar"
+
+                ZipInputStream(URL(aar).openStream()).use {
+                    while (true) {
+                        val entry = it.nextEntry
+                        if (entry == null) {
+                            onOutput("Could not find 'classes.jar' in $aar")
+                            return true
+                        }
+                        if (entry.name == "classes.jar") {
+                            it.copyTo(outputStream)
+                            break
+                        }
+                    }
+                }
+
+            }
+        }
+        return true
+    }
+}
diff --git a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/Disassembly.kt b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/Disassembly.kt
@@ -47,6 +47,8 @@ suspend fun buildAndRun(
     toolPaths: ToolPaths,
     kotlinOnlyConsumers: Boolean,
     compilerFlags: String,
+    settingsDirectory: Path,
+    composeVersion: String,
     source: String,
     onLogs: (AnnotatedString) -> Unit,
     onStatusUpdate: (String, Float) -> Unit
@@ -65,7 +67,7 @@ suspend fun buildAndRun(
             Files.writeString(path, source)
             writeSupportFiles(directory)
 
-            val kotlinc = KotlinCompiler(toolPaths, directory).compile(kotlinOnlyConsumers, compilerFlags, path)
+            val kotlinc = KotlinCompiler(toolPaths, settingsDirectory, directory).compile(kotlinOnlyConsumers, compilerFlags, composeVersion, path)
 
             if (kotlinc.exitCode != 0) {
                 withContext(ui) {
@@ -102,6 +104,8 @@ suspend fun buildAndDisassemble(
     compilerFlags: String,
     r8rules: String,
     minApi: Int,
+    settingsDirectory: Path,
+    composeVersion: String,
     instructionSets: Map<ISA, Boolean>,
     onByteCode: (CodeContent) -> Unit,
     onDex: (CodeContent) -> Unit,
@@ -127,7 +131,7 @@ suspend fun buildAndDisassemble(
             Files.writeString(path, source)
             writeSupportFiles(directory)
 
-            val kotlinc = KotlinCompiler(toolPaths, directory).compile(kotlinOnlyConsumers, compilerFlags, path)
+            val kotlinc = KotlinCompiler(toolPaths, settingsDirectory, directory).compile(kotlinOnlyConsumers, compilerFlags, composeVersion, path)
 
             if (kotlinc.exitCode != 0) {
                 updater.addJob(launch(ui) {
diff --git a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/KotlinExplorer.kt b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/KotlinExplorer.kt
@@ -532,6 +532,8 @@ private fun FrameWindowScope.MainMenu(
                 explorerState.compilerFlags,
                 explorerState.r8Rules,
                 explorerState.minApi,
+                explorerState.directory,
+                explorerState.composeVersion,
                 instructionSets,
                 onByteCodeUpdate,
                 onDexUpdate,
@@ -549,6 +551,8 @@ private fun FrameWindowScope.MainMenu(
                 explorerState.toolPaths,
                 explorerState.kotlinOnlyConsumers,
                 explorerState.compilerFlags,
+                explorerState.directory,
+                explorerState.composeVersion,
                 sourceTextArea.text,
                 onLogsUpdate,
                 onStatusUpdate
diff --git a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/Settings.kt b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/Settings.kt
@@ -44,6 +44,7 @@ fun Settings(
     val kotlinOnlyConsumers = remember { mutableStateOf(state.kotlinOnlyConsumers) }
     val compilerFlags = rememberTextFieldState(state.compilerFlags)
     val r8rules = rememberTextFieldState(state.r8Rules)
+    val composeVersion = rememberTextFieldState(state.composeVersion)
     val minApi = rememberTextFieldState(state.minApi.toString())
     val indent = rememberTextFieldState(state.indent.toString())
     val lineNumberWidth = rememberTextFieldState(state.lineNumberWidth.toString())
@@ -55,6 +56,7 @@ fun Settings(
             kotlinOnlyConsumers.value,
             compilerFlags.text.toString(),
             r8rules.text.toString(),
+            composeVersion.text.toString(),
             minApi.text.toString(),
             indent.text.toString(),
             lineNumberWidth.text.toString(),
@@ -71,6 +73,7 @@ fun Settings(
         IntSetting("Line number column width: ", lineNumberWidth, minValue = 1)
         StringSetting("Kotlin compiler flags: ", compilerFlags)
         MultiLineStringSetting("R8 rules: ", r8rules)
+        StringSetting("Compose version: ", composeVersion) {composeVersion.text.isNotEmpty()}
         IntSetting("Min API: ", minApi, minValue = 1)
         BooleanSetting("Kotlin only consumers", kotlinOnlyConsumers)
         BooleanSetting("Decompile hidden instruction sets", decompileHiddenIsa)
@@ -101,6 +104,7 @@ private fun ExplorerState.saveState(
     kotlinOnlyConsumers: Boolean,
     compilerFlags: String,
     r8Rules: String,
+    composeVersion: String,
     minApi: String,
     indent: String,
     lineNumberWidth: String,
@@ -111,6 +115,7 @@ private fun ExplorerState.saveState(
     this.kotlinOnlyConsumers = kotlinOnlyConsumers
     this.compilerFlags = compilerFlags
     this.r8Rules = r8Rules
+    this.composeVersion = composeVersion
     this.minApi = minApi.toIntOrNull() ?: 21
     this.indent = indent.toIntOrNull() ?: 4
     this.lineNumberWidth = lineNumberWidth.toIntOrNull() ?: 4
diff --git a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/State.kt b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/State.kt
@@ -32,6 +32,7 @@ private const val KeepEverything = "KEEP_EVERYTHING"
 private const val KotlinOnlyConsumers = "KOTLIN_ONLY_CONSUMERS"
 private const val CompilerFlags = "COMPILER_FLAGS"
 private const val R8Rules = "R8_RULES"
+private const val ComposeVersion = "COMPOSE_VERSION"
 private const val MinApi = "MIN_API"
 private const val AutoBuildOnStartup = "AUTO_BUILD_ON_STARTUP"
 private const val Presentation = "PRESENTATION"
@@ -63,6 +64,7 @@ class ExplorerState {
     var kotlinOnlyConsumers by BooleanState(KotlinOnlyConsumers, true)
     var compilerFlags by StringState(CompilerFlags, "")
     var r8Rules by StringState(R8Rules, "")
+    var composeVersion by StringState(ComposeVersion, "1.8.1")
     var minApi by IntState(MinApi, 21)
     var autoBuildOnStartup by BooleanState(AutoBuildOnStartup, false)
     var presentationMode by BooleanState(Presentation, false)
diff --git a/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/build/KolinCompiler.kt b/src/jvmMain/kotlin/dev/romainguy/kotlin/explorer/build/KolinCompiler.kt
@@ -16,28 +16,72 @@
 
 package dev.romainguy.kotlin.explorer.build
 
+import dev.romainguy.kotlin.explorer.DependencyCache
+import dev.romainguy.kotlin.explorer.ProcessResult
 import dev.romainguy.kotlin.explorer.ToolPaths
 import dev.romainguy.kotlin.explorer.process
 import java.io.File
 import java.nio.file.Path
+import kotlin.io.path.useLines
 
 private val BuiltInFiles = listOf(
     "Keep.kt",
     "NeverInline.kt"
 )
 
-class KotlinCompiler(private val toolPaths: ToolPaths, private val outputDirectory: Path) {
-    suspend fun compile(kotlinOnlyConsumers: Boolean, compilerFlags: String, source: Path) =
-        process(*buildCompileCommand(kotlinOnlyConsumers, compilerFlags, source), directory = outputDirectory)
+class KotlinCompiler(
+    private val toolPaths: ToolPaths,
+    settingsDirectory: Path,
+    private val outputDirectory: Path,
+) {
+    private val dependencyCache = DependencyCache(settingsDirectory.resolve("dependency-cache"))
 
-    private fun buildCompileCommand(kotlinOnlyConsumers: Boolean, compilerFlags: String, file: Path): Array<String> {
-        val classpath = (toolPaths.kotlinLibs + listOf(toolPaths.platform)).joinToString(File.pathSeparator) { jar -> jar.toString() }
-        val command = mutableListOf(
-            toolPaths.kotlinc.toString(),
-            "-Xmulti-platform",
-            "-classpath",
-            "\"$classpath\""
-        ).apply {
+    suspend fun compile(
+        kotlinOnlyConsumers: Boolean,
+        compilerFlags: String,
+        composeVersion: String,
+        source: Path
+    ): ProcessResult {
+        val sb = StringBuilder()
+        val result = process(
+            *buildCompileCommand(
+                kotlinOnlyConsumers,
+                compilerFlags,
+                composeVersion,
+                source
+            ) { sb.appendLine(it) }, directory = outputDirectory
+        )
+        return ProcessResult(result.exitCode, "$sb\n${result.output}")
+    }
+
+    private suspend fun buildCompileCommand(
+        kotlinOnlyConsumers: Boolean,
+        compilerFlags: String,
+        composeVersion: String,
+        file: Path,
+        onOutput: (String) -> Unit
+    ): Array<String> {
+        val isCompose = file.isCompose()
+        val classpath = buildList {
+            addAll(toolPaths.kotlinLibs)
+            add(toolPaths.platform)
+            if (isCompose) {
+                add(
+                    dependencyCache.getDependency(
+                        "androidx.compose.runtime",
+                        "runtime-android",
+                        composeVersion,
+                        onOutput,
+                    )
+                )
+            }
+        }.joinToString((File.pathSeparator)) { it.toString() }
+
+        val command = buildList {
+            add(toolPaths.kotlinc.toString())
+            add("-Xmulti-platform")
+            add("-classpath")
+            add(classpath)
             if (kotlinOnlyConsumers) {
                 this += "-Xno-param-assertions"
                 this += "-Xno-call-assertions"
@@ -47,6 +91,15 @@ class KotlinCompiler(private val toolPaths: ToolPaths, private val outputDirecto
                 // TODO: Do something smarter in case a flag looks like -foo="something with space"
                 addAll(compilerFlags.split(' '))
             }
+            if (isCompose) {
+                val composePlugin = dependencyCache.getDependency(
+                    "org.jetbrains.kotlin",
+                    "kotlin-compose-compiler-plugin",
+                    getKotlinVersion(),
+                    onOutput,
+                )
+                add("-Xplugin=$composePlugin")
+            }
             // Source code to compile
             this += file.toString()
             for (fileName in BuiltInFiles) {
@@ -56,4 +109,19 @@ class KotlinCompiler(private val toolPaths: ToolPaths, private val outputDirecto
 
         return command.toTypedArray()
     }
+
+    private suspend fun getKotlinVersion(): String {
+        val command = mutableListOf(
+            toolPaths.kotlinc.toString(),
+            "-version",
+        ).toTypedArray()
+        return process(*command).output.substringAfter("kotlinc-jvm ").substringBefore(" ")
+    }
 }
+
+private fun Path.isCompose(): Boolean {
+    return useLines { lines ->
+        lines.filter { it.trim().startsWith("import") }
+            .any { it.split(" ").last() == "androidx.compose.runtime.Composable" }
+    }
+}
diff --git a/src/jvmTest/kotlin/dev/romainguy/kotlin/explorer/testing/Builder.kt b/src/jvmTest/kotlin/dev/romainguy/kotlin/explorer/testing/Builder.kt
@@ -54,7 +54,7 @@ class LocalBuilder(private val outputDirectory: Path) : Builder {
     private val cwd = Path.of(System.getProperty("user.dir"))
     private val testData = cwd.resolve("src/jvmTest/kotlin/testData")
     private val toolPaths = createToolsPath()
-    private val kotlinCompiler = KotlinCompiler(toolPaths, outputDirectory)
+    private val kotlinCompiler = KotlinCompiler(toolPaths, outputDirectory, outputDirectory)
     private val byteCodeDecompiler = ByteCodeDecompiler()
 
     override fun generateByteCode(testFile: String): String {
@@ -82,7 +82,7 @@ class LocalBuilder(private val outputDirectory: Path) : Builder {
     }
 
     private suspend fun kotlinCompile(path: Path) {
-        val result = kotlinCompiler.compile(true, "", path)
+        val result = kotlinCompiler.compile(true, "", "not-used", path)
         if (result.exitCode != 0) {
             System.err.println(result.output)
             fail("kotlinc error")
