Set Main-Class attr for KMP 2.1.0 or above

docs/changes/README.md

@@ -7,6 +7,7 @@
 
 - Add Kotlin DSL examples in docs. ([#1306](https://github.com/GradleUp/shadow/pull/1306))
 - Support using type-safe dependency accessors in `ShadowJar.dependencies`. ([#1322](https://github.com/GradleUp/shadow/pull/1322))
+- Set `Main-Class` attr for KMP 2.1.0 or above. ([#1337](https://github.com/GradleUp/shadow/pull/1337))
 
 **Changed**
 

docs/kmp-plugin/README.md

@@ -15,7 +15,10 @@ configure additional tasks for bundling the shadowed JAR for its `jvm` target.
     val ktorVersion = "3.1.0"
 
     kotlin {
-      jvm()
+      jvm().mainRun {
+        // Optionally, set the main class for `runJvm`, it's available from Kotlin 2.1.0
+        mainClass = "myapp.MainKt"
+      }
       sourceSets {
         val commonMain by getting {
           dependencies {
@@ -49,7 +52,10 @@ configure additional tasks for bundling the shadowed JAR for its `jvm` target.
     def ktorVersion = "3.1.0"
 
     kotlin {
-      jvm()
+      jvm().mainRun {
+        // Optionally, set the main class for `runJvm`, it's available from Kotlin 2.1.0
+        it.mainClass.set('myapp.MainKt')
+      }
       sourceSets {
         commonMain {
           dependencies {

src/functionalTest/kotlin/com/github/jengelman/gradle/plugins/shadow/KmpPluginTest.kt

@@ -1,7 +1,10 @@
 package com.github.jengelman.gradle.plugins.shadow
 
 import assertk.assertThat
+import assertk.assertions.isEqualTo
+import com.github.jengelman.gradle.plugins.shadow.internal.mainClassAttributeKey
 import com.github.jengelman.gradle.plugins.shadow.util.containsEntries
+import com.github.jengelman.gradle.plugins.shadow.util.getMainAttr
 import kotlin.io.path.appendText
 import kotlin.io.path.writeText
 import org.junit.jupiter.api.BeforeEach
@@ -21,11 +24,14 @@ class KmpPluginTest : BasePluginTest() {
 
   @Test
   fun compatKmpJvmTarget() {
+    val mainClassName = "my.MainKt"
     val mainClassEntry = writeClass(sourceSet = "jvmMain", isJava = false)
     projectScriptPath.appendText(
       """
         kotlin {
-          jvm()
+          jvm().mainRun {
+            it.mainClass.set('$mainClassName')
+          }
           sourceSets {
             commonMain {
               dependencies {
@@ -49,6 +55,7 @@ class KmpPluginTest : BasePluginTest() {
         mainClassEntry,
         *entriesInAB,
       )
+      getMainAttr(mainClassAttributeKey).isEqualTo(mainClassName)
     }
   }
 }

src/main/kotlin/com/github/jengelman/gradle/plugins/shadow/ShadowKmpPlugin.kt

@@ -1,15 +1,21 @@
 package com.github.jengelman.gradle.plugins.shadow
 
 import com.github.jengelman.gradle.plugins.shadow.ShadowJavaPlugin.Companion.registerShadowJarCommon
+import com.github.jengelman.gradle.plugins.shadow.internal.mainClassAttributeKey
+import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar
+import kotlin.collections.contains
 import org.gradle.api.Plugin
 import org.gradle.api.Project
+import org.jetbrains.kotlin.gradle.ExperimentalKotlinGradlePluginApi
 import org.jetbrains.kotlin.gradle.dsl.KotlinMultiplatformExtension
+import org.jetbrains.kotlin.gradle.dsl.KotlinVersion as KgpVersion
 
 public abstract class ShadowKmpPlugin : Plugin<Project> {
+  private lateinit var kmpExtension: KotlinMultiplatformExtension
 
   override fun apply(project: Project) {
     with(project) {
-      val kmpExtension = extensions.getByType(KotlinMultiplatformExtension::class.java)
+      kmpExtension = extensions.getByType(KotlinMultiplatformExtension::class.java)
       val kotlinJvmMain = kmpExtension.jvm().compilations.named("main")
       registerShadowJarCommon { task ->
         task.from(kotlinJvmMain.map { it.output.allOutputs })
@@ -18,6 +24,24 @@ public abstract class ShadowKmpPlugin : Plugin<Project> {
             listOf(configurations.getByName(kotlinJvmMain.get().runtimeDependencyConfigurationName))
           },
         )
+        configureMainClass(task)
+      }
+    }
+  }
+
+  private fun Project.configureMainClass(task: ShadowJar) {
+    if (KgpVersion.DEFAULT < KgpVersion.KOTLIN_2_1) return
+
+    @OptIn(ExperimentalKotlinGradlePluginApi::class)
+    kmpExtension.jvm().mainRun {
+      // Fix cannot serialize object of type 'org.jetbrains.kotlin.gradle.targets.jvm.tasks.KotlinJvmRun'.
+      val mainClassName = provider { mainClass }
+      task.inputs.property("mainClassName", mainClassName)
+      task.doFirst {
+        // Inject the attribute if it is not already present.
+        if (!task.manifest.attributes.contains(mainClassAttributeKey)) {
+          task.manifest.attributes[mainClassAttributeKey] = mainClassName.get().get()
+        }
       }
     }
   }