add dependency dump for JVM & Android tasks

Author: zsqw123

gradle.properties

@@ -30,5 +30,5 @@ junitVersion=5.10.0
 GROUP=io.github.tiktok.knit
 localPublish=true
 remoteKnitVersion=0.1.3
-knitVersion=0.1.4
+knitVersion=0.1.5
 projectGitUrl=https://github.com/tiktok/knit

knit-asm/src/main/java/tiktok/knit/plugin/writer/GlobalProvidesWriter.kt

@@ -35,6 +35,7 @@ import tiktok.knit.plugin.monitorOut
 import tiktok.knit.plugin.new
 import tiktok.knit.plugin.putStatic
 import tiktok.knit.plugin.sameFrame
+import tiktok.knit.plugin.sameFrame1Throwable
 import tiktok.knit.plugin.typedLoad
 import tiktok.knit.plugin.unaryPlus
 
@@ -186,6 +187,7 @@ class GlobalProvidesWriter(private val context: KnitContext) {
 
             if (singleton.threadSafe) {
                 +handlerNode // catch {
+                sameFrame1Throwable()
                 astore(argCount + 1)
                 ldc(lockType)
                 monitorOut()

knit-plugin/src/main/java/tiktok/knit/plugin/KnitExtension.kt

@@ -0,0 +1,19 @@
+// Copyright (c) 2025 by TikTok Ltd., All rights reserved.
+// Licensed under the Apache License Version 2.0 that can be found in the
+// LICENSE file in the root directory of this source tree.
+
+package tiktok.knit.plugin
+
+import org.gradle.api.provider.Property
+
+/**
+ * Created by yuejunyu on 2025/8/14
+ * Extension for configuring Knit plugin settings.
+ */
+interface KnitExtension {
+    /**
+     * The output path for the dependency tree JSON file.
+     * Defaults to "build/knit/dependency-tree.json"
+     */
+    val dependencyTreeOutputPath: Property<String>
+}

knit-plugin/src/main/java/tiktok/knit/plugin/KnitGradlePlugin.kt

@@ -21,14 +21,16 @@ import java.io.File
  */
 abstract class KnitGradlePlugin : Plugin<Project> {
     override fun apply(target: Project) {
+        val extension = target.extensions.create("KnitExtension", KnitExtension::class.java)
+        extension.dependencyTreeOutputPath.convention("build/knit/dependency-tree.json")
         if (KnitAndroidConfig.tryConfigAndroid(target)) return
         if (target.tryConfigJvm()) return
         System.err.println("cannot found any knit target available.")
     }
 
     private fun Project.tryConfigJvm(): Boolean {
         val jarTasks = tasks.withType(Jar::class.java)
-        if (jarTasks.isNullOrEmpty()) return false
+        if (jarTasks.isEmpty()) return false
         for (originJarTask in jarTasks) {
             val originOutput = originJarTask.archiveFile
             val originOutputFile = originOutput.get().asFile
@@ -55,9 +57,15 @@ abstract class KnitGradlePlugin : Plugin<Project> {
 
         @TaskAction
         fun taskAction() {
+            val dumpOutputFile = project.file(
+                project.extensions.getByType(KnitExtension::class.java).dependencyTreeOutputPath,
+            )
             val allJars = listOf(originJar.get().asFile)
             val outputJarFile = output.get().asFile
-            val knitTask = KnitTask(allJars, emptyList(), outputJarFile, true)
+            val knitTask = KnitTask(
+                allJars, emptyList(), outputJarFile, true,
+                dumpOutput = dumpOutputFile,
+            )
             knitTask.execute()
         }
     }

knit-plugin/src/main/java/tiktok/knit/plugin/KnitPipeline.kt

@@ -6,6 +6,7 @@ package tiktok.knit.plugin
 
 import org.objectweb.asm.Type
 import org.objectweb.asm.tree.ClassNode
+import tiktok.knit.plugin.dump.KnitDumper
 import tiktok.knit.plugin.element.BoundComponentClass
 import tiktok.knit.plugin.element.BoundComponentMapping
 import tiktok.knit.plugin.element.ComponentClass
@@ -19,13 +20,15 @@ import tiktok.knit.plugin.injection.InjectionBinder
 import tiktok.knit.plugin.injection.InjectionFactoryContext
 import tiktok.knit.plugin.writer.ComponentWriter
 import tiktok.knit.plugin.writer.GlobalProvidesWriter
+import java.io.File
 
 /**
  * Created by yuejunyu on 2025/4/15
  * @author yuejunyu.0
  */
 class KnitPipeline(
     private val useJrt: Boolean,
+    private val dumpOutputFile: File,
 ) {
     private class KnitContextImpl : KnitContext {
         override val componentMap: MutableMap<InternalName, ComponentClass> = mutableMapOf()
@@ -132,4 +135,8 @@ class KnitPipeline(
             it.attach2BoundMapping(mapping, map)
         }
     }
+
+    fun finish() {
+        KnitDumper().dumpContext(knitContextImpl, dumpOutputFile)
+    }
 }

knit-plugin/src/main/java/tiktok/knit/plugin/KnitTask.kt

@@ -25,13 +25,14 @@ class KnitTask(
     private val dirInputs: List<File>,
     private val outputJar: File,
     private val useJrt: Boolean,
+    private val dumpOutput: File,
 ) {
     fun execute() {
         val jarOutput = JarOutputStream(
             BufferedOutputStream(FileOutputStream(outputJar)),
         )
         val graphPipeline = GraphPipeline()
-        val knitPipeline = KnitPipeline(useJrt)
+        val knitPipeline = KnitPipeline(useJrt, dumpOutput)
 
         val container = ContentContainer()
         val allClasses = container.getAllClasses(jarInputs, dirInputs, jarOutput)
@@ -53,6 +54,8 @@ class KnitTask(
             jarOutput.closeEntry()
         }
 
+        knitPipeline.finish()
+
         container.closeAll()
         jarOutput.close()
     }

knit-plugin/src/main/java/tiktok/knit/plugin/android/KnitAndroidConfig.kt

@@ -13,6 +13,7 @@ import org.gradle.api.provider.ListProperty
 import org.gradle.api.tasks.InputFiles
 import org.gradle.api.tasks.OutputFile
 import org.gradle.api.tasks.TaskAction
+import tiktok.knit.plugin.KnitExtension
 import tiktok.knit.plugin.KnitTask
 import java.io.File
 
@@ -58,10 +59,13 @@ internal object KnitAndroidConfig {
         @TaskAction
         fun taskAction() {
             val androidJarFile = findAndroidJar(project)
+            val dumpOutputFile = project.file(
+                project.extensions.getByType(KnitExtension::class.java).dependencyTreeOutputPath,
+            )
             val allJars = allJars.get().map { it.asFile } + androidJarFile
             val allDirs = allDirectories.get().map { it.asFile }
             val outputJarFile = output.get().asFile
-            val knitTask = KnitTask(allJars, allDirs, outputJarFile, false)
+            val knitTask = KnitTask(allJars, allDirs, outputJarFile, false, dumpOutputFile)
             knitTask.execute()
         }
     }