Register shadow distribution

Author: Goooler

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

@@ -12,7 +12,10 @@ import org.gradle.api.Action
 import org.gradle.api.GradleException
 import org.gradle.api.Plugin
 import org.gradle.api.Project
+import org.gradle.api.distribution.Distribution
+import org.gradle.api.file.CopySpec
 import org.gradle.api.plugins.ApplicationPlugin
+import org.gradle.api.provider.Provider
 import org.gradle.api.tasks.JavaExec
 import org.gradle.api.tasks.Sync
 import org.gradle.api.tasks.TaskContainer
@@ -87,21 +90,13 @@ public abstract class ShadowApplicationPlugin : Plugin<Project> {
   }
 
   protected open fun Project.configureDistribution() {
-    distributions.register(DISTRIBUTION_NAME) {
-      it.contents { shadowDist ->
-        shadowDist.from(file("src/dist"))
-        shadowDist.into("lib") { lib ->
-          lib.from(tasks.shadowJar)
-          // Reflects the value of the `Class-Path` attribute in the JAR manifest.
-          lib.from(configurations.shadow)
-        }
-        // Defaults to bin dir.
-        shadowDist.into(applicationExtension.executableDir) { bin ->
-          bin.from(tasks.startShadowScripts)
-          bin.filePermissions { permissions -> permissions.unix(UNIX_SCRIPT_PERMISSIONS) }
-        }
-        shadowDist.with(applicationExtension.applicationDistribution)
+    registerShadowDistributionCommon { dist ->
+      // Defaults to bin dir.
+      dist.into(applicationExtension.executableDir) { bin ->
+        bin.from(tasks.startShadowScripts)
+        bin.filePermissions { permissions -> permissions.unix(UNIX_SCRIPT_PERMISSIONS) }
       }
+      dist.with(applicationExtension.applicationDistribution)
     }
   }
 
@@ -115,7 +110,7 @@ public abstract class ShadowApplicationPlugin : Plugin<Project> {
     /**
      * Reflects the number of 755.
      */
-    private const val UNIX_SCRIPT_PERMISSIONS = "rwxr-xr-x"
+    internal const val UNIX_SCRIPT_PERMISSIONS = "rwxr-xr-x"
 
     public const val DISTRIBUTION_NAME: String = SHADOW
 
@@ -152,5 +147,21 @@ public abstract class ShadowApplicationPlugin : Plugin<Project> {
         action.execute(task)
       }
     }
+
+    internal fun Project.registerShadowDistributionCommon(
+      action: Action<CopySpec>,
+    ): Provider<Distribution> {
+      return distributions.register(DISTRIBUTION_NAME) {
+        it.contents { shadowDist ->
+          shadowDist.from(file("src/dist"))
+          shadowDist.into("lib") { lib ->
+            lib.from(tasks.shadowJar)
+            // Reflects the value of the `Class-Path` attribute in the JAR manifest.
+            lib.from(configurations.shadow)
+          }
+          action.execute(shadowDist)
+        }
+      }
+    }
   }
 }

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

@@ -1,7 +1,10 @@
 package com.github.jengelman.gradle.plugins.shadow
 
+import com.github.jengelman.gradle.plugins.shadow.ShadowApplicationPlugin.Companion.UNIX_SCRIPT_PERMISSIONS
 import com.github.jengelman.gradle.plugins.shadow.ShadowApplicationPlugin.Companion.registerRunShadowCommon
+import com.github.jengelman.gradle.plugins.shadow.ShadowApplicationPlugin.Companion.registerShadowDistributionCommon
 import com.github.jengelman.gradle.plugins.shadow.ShadowApplicationPlugin.Companion.registerStartShadowScriptsCommon
+import com.github.jengelman.gradle.plugins.shadow.ShadowApplicationPlugin.Companion.startShadowScripts
 import com.github.jengelman.gradle.plugins.shadow.internal.isAtLeastKgpVersion
 import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar.Companion.SHADOW_JAR_TASK_NAME
 import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar.Companion.registerShadowJarCommon
@@ -49,6 +52,7 @@ public abstract class ShadowKmpPlugin : Plugin<Project> {
       (tasks.getByName("startScriptsFor$targetNameCap") as CreateStartScripts).let {
         addCreateScriptsTask(it)
       }
+      configureDistribution()
     }
   }
 
@@ -99,4 +103,16 @@ public abstract class ShadowKmpPlugin : Plugin<Project> {
       }
     }
   }
+
+  private fun Project.configureDistribution() {
+    registerShadowDistributionCommon { dist ->
+      // Should use KotlinJvmBinaryDsl.applicationDistribution instead.
+      dist.into("bin") { bin ->
+        bin.from(tasks.startShadowScripts)
+        bin.filePermissions { permissions -> permissions.unix(UNIX_SCRIPT_PERMISSIONS) }
+      }
+      // TODO: we can't access KotlinJvmBinaryDsl instance for now.
+//      dist.with(KotlinJvmBinaryDsl.applicationDistribution)
+    }
+  }
 }