Try to figure out how to shade protobuf (doesn't work yet...)

Author: slinkydeveloper

buildSrc/src/main/kotlin/library-publishing-conventions.gradle.kts

@@ -7,11 +7,68 @@ project.afterEvaluate {
     publishing {
         publications {
             create<MavenPublication>("maven") {
+                afterEvaluate {
+                    val shadowJar = tasks.findByName("shadowJar")
+                    if (shadowJar == null) {
+                        from(components["java"])
+                    }
+                    else {
+                        apply(plugin = "com.gradleup.shadow")
+
+                        from(components["shadow"])
+                        artifact(tasks["sourcesJar"]!!)
+                        artifact(tasks["javadocJar"]!!)
+
+                        afterEvaluate {
+                            // Fix for avoiding inclusion of runtime dependencies marked as 'shadow' in MANIFEST Class-Path.
+                            // https://github.com/johnrengelman/shadow/issues/324
+                            pom.withXml {
+                                val rootNode = asElement()
+                                val doc = rootNode.ownerDocument
+
+                                val dependenciesNode =
+                                    if (rootNode.getElementsByTagName("dependencies").length != 0) {
+                                        rootNode.getElementsByTagName("dependencies").item(0)
+                                    } else {
+                                        rootNode.appendChild(
+                                            doc.createElement("dependencies")
+                                        )
+                                    }
+
+                                project.configurations["shade"].allDependencies.forEach { dep ->
+                                    dependenciesNode.appendChild(
+                                        doc.createElement("dependency").apply {
+                                            appendChild(
+                                                doc.createElement("groupId").apply {
+                                                    textContent = dep.group
+                                                }
+                                            )
+                                            appendChild(
+                                                doc.createElement("artifactId").apply {
+                                                    textContent = dep.name
+                                                }
+                                            )
+                                            appendChild(
+                                                doc.createElement("version").apply {
+                                                    textContent = dep.version
+                                                }
+                                            )
+                                            appendChild(
+                                                doc.createElement("scope").apply {
+                                                    textContent = "runtime"
+                                                }
+                                            )
+                                        }
+                                    )
+                                }
+                            }
+                        }
+                    }
+                }
+
                 groupId = "dev.restate"
                 artifactId = project.name
 
-                from(components["java"])
-
                 pom {
                     name = "Restate SDK :: ${project.name}"
                     description = project.description!!

sdk-core/build.gradle.kts

@@ -1,4 +1,6 @@
+import org.gradle.kotlin.dsl.allDependencies
 import org.jetbrains.dokka.gradle.AbstractDokkaTask
+import org.jetbrains.kotlin.gradle.utils.extendsFrom
 
 plugins {
   `java-library`
@@ -8,19 +10,26 @@ plugins {
   `library-publishing-conventions`
   alias(libs.plugins.jsonschema2pojo)
   alias(libs.plugins.protobuf)
+  alias(libs.plugins.shadow)
 
   // https://github.com/gradle/gradle/issues/20084#issuecomment-1060822638
   id(libs.plugins.spotless.get().pluginId) apply false
 }
 
 description = "Restate SDK Core"
 
+val shade by configurations.creating
+val implementation by configurations.getting
+implementation.extendsFrom(shade)
+val api by configurations.getting
+api.extendsFrom(shade)
+
 dependencies {
   compileOnly(libs.jspecify)
 
   implementation(project(":sdk-common"))
 
-  implementation(libs.protobuf.java)
+  shade(libs.protobuf.java)
   implementation(libs.log4j.api)
 
   // We need this for the manifest
@@ -33,6 +42,7 @@ dependencies {
   api(libs.opentelemetry.api)
 
   testCompileOnly(libs.jspecify)
+  testImplementation(libs.protobuf.java)
   testImplementation(libs.mutiny)
   testImplementation(libs.junit.jupiter)
   testImplementation(libs.assertj)
@@ -75,6 +85,24 @@ tasks {
     dependsOn(generateJsonSchema2Pojo, generateProto)
   }
   withType<AbstractDokkaTask>().configureEach { dependsOn(generateJsonSchema2Pojo, generateProto) }
+
+  getByName("jar") {
+    enabled = false
+    dependsOn(shadowJar)
+  }
+
+  shadowJar {
+    configurations = listOf(shade)
+    enableRelocation = true
+    archiveClassifier = null
+    relocate("com.google.protobuf", "dev.restate.shaded.com.google.protobuf")
+    dependencies {
+      project.configurations["shadow"].allDependencies.forEach {
+        exclude(dependency(it))
+      }
+      exclude("**/google/protobuf/*.proto")
+    }
+  }
 }
 
 // spotless configuration for protobuf