Added support for Well-Known Types

Author: Mr3zee

gradle-plugin/src/main/kotlin/kotlinx/rpc/buf/tasks/BufGenerateTask.kt

@@ -56,6 +56,8 @@ public abstract class BufGenerateTask : BufExecTask() {
     /**
      * Whether to include Well-Known Types.
      *
+     * Automatically sets [includeImports] to `true`.
+     *
      * @see <a href="https://buf.build/docs/reference/cli/buf/generate/#include-wkt">buf generate --include-wkt</a>
      * @see [BufGenerateExtension.includeWkt]
      */
@@ -91,7 +93,7 @@ public abstract class BufGenerateTask : BufExecTask() {
             buildList {
                 add("--output"); add(outputDirectory.get().absolutePath)
 
-                if (includeImports.get()) {
+                if (includeImports.get() || includeWkt.get()) {
                     add("--include-imports")
                 }
 

gradle-plugin/src/main/kotlin/kotlinx/rpc/internal/configureLocalProtocGenDevelopmentDependency.kt

@@ -14,7 +14,9 @@ import org.gradle.kotlin.dsl.provideDelegate
 import org.gradle.kotlin.dsl.withType
 
 @InternalRpcApi
-public fun Project.configureLocalProtocGenDevelopmentDependency() {
+public fun Project.configureLocalProtocGenDevelopmentDependency(
+    vararg sourceSetSuffix: String = arrayOf("Test"),
+) {
     val globalRootDir: String by extra
 
     rpcExtension().protoc.plugins {
@@ -32,7 +34,7 @@ public fun Project.configureLocalProtocGenDevelopmentDependency() {
     }
 
     tasks.withType<BufGenerateTask>().configureEach {
-        if (name.endsWith("Test")) {
+        if (sourceSetSuffix.any { name.endsWith(it) }) {
             val includedBuild = gradle.includedBuild("protoc-gen")
             dependsOn(includedBuild.task(":grpc:jar"))
             dependsOn(includedBuild.task(":protobuf:jar"))

protobuf/protobuf-core/build.gradle.kts

@@ -4,6 +4,8 @@
 
 @file:OptIn(InternalRpcApi::class)
 
+import org.jetbrains.kotlin.gradle.dsl.KotlinVersion
+import kotlinx.rpc.buf.tasks.BufGenerateTask
 import kotlinx.rpc.internal.InternalRpcApi
 import kotlinx.rpc.internal.configureLocalProtocGenDevelopmentDependency
 import util.configureCLibCInterop
@@ -14,19 +16,23 @@ plugins {
 }
 
 kotlin {
+    compilerOptions {
+        apiVersion = KotlinVersion.KOTLIN_2_1
+    }
+
     sourceSets {
         commonMain {
             dependencies {
                 api(projects.utils)
                 api(projects.protobuf.protobufInputStream)
+                api(projects.grpc.grpcCodec)
+
                 api(libs.kotlinx.io.core)
             }
         }
 
         commonTest {
             dependencies {
-                implementation(projects.grpc.grpcCodec)
-
                 implementation(libs.kotlin.test)
                 implementation(libs.coroutines.test)
             }
@@ -66,4 +72,17 @@ protoSourceSets {
     }
 }
 
-configureLocalProtocGenDevelopmentDependency()
+configureLocalProtocGenDevelopmentDependency("Main", "Test")
+
+val generatedCodeDir = layout.projectDirectory
+    .dir("src")
+    .dir("commonMain")
+    .dir("generated-code")
+    .asFile
+
+tasks.withType<BufGenerateTask>().configureEach {
+    if (name.contains("Main")) {
+        includeWkt = true
+        outputDirectory = generatedCodeDir
+    }
+}

protobuf/protobuf-core/src/commonMain/generated-code/kotlin-multiplatform/com/google/protobuf/kotlin/Any.kt

@@ -0,0 +1,13 @@
+@file:OptIn(ExperimentalRpcApi::class, InternalRpcApi::class)
+package com.google.protobuf.kotlin
+
+import kotlinx.rpc.internal.utils.*
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.AnyInternal.CODEC::class)
+public interface Any { 
+    public val typeUrl: String
+    public val value: ByteArray
+
+    public companion object
+}
+

protobuf/protobuf-core/src/commonMain/generated-code/kotlin-multiplatform/com/google/protobuf/kotlin/Api.kt

@@ -0,0 +1,39 @@
+@file:OptIn(ExperimentalRpcApi::class, InternalRpcApi::class)
+package com.google.protobuf.kotlin
+
+import kotlinx.rpc.internal.utils.*
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.ApiInternal.CODEC::class)
+public interface Api { 
+    public val name: String
+    public val methods: List<com.google.protobuf.kotlin.Method>
+    public val options: List<com.google.protobuf.kotlin.Option>
+    public val version: String
+    public val sourceContext: com.google.protobuf.kotlin.SourceContext
+    public val mixins: List<com.google.protobuf.kotlin.Mixin>
+    public val syntax: com.google.protobuf.kotlin.Syntax
+
+    public companion object
+}
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.MethodInternal.CODEC::class)
+public interface Method { 
+    public val name: String
+    public val requestTypeUrl: String
+    public val requestStreaming: Boolean
+    public val responseTypeUrl: String
+    public val responseStreaming: Boolean
+    public val options: List<com.google.protobuf.kotlin.Option>
+    public val syntax: com.google.protobuf.kotlin.Syntax
+
+    public companion object
+}
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.MixinInternal.CODEC::class)
+public interface Mixin { 
+    public val name: String
+    public val root: String
+
+    public companion object
+}
+

protobuf/protobuf-core/src/commonMain/generated-code/kotlin-multiplatform/com/google/protobuf/kotlin/Duration.kt

@@ -0,0 +1,13 @@
+@file:OptIn(ExperimentalRpcApi::class, InternalRpcApi::class)
+package com.google.protobuf.kotlin
+
+import kotlinx.rpc.internal.utils.*
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.DurationInternal.CODEC::class)
+public interface Duration { 
+    public val seconds: Long
+    public val nanos: Int
+
+    public companion object
+}
+

protobuf/protobuf-core/src/commonMain/generated-code/kotlin-multiplatform/com/google/protobuf/kotlin/Empty.kt

@@ -0,0 +1,10 @@
+@file:OptIn(ExperimentalRpcApi::class, InternalRpcApi::class)
+package com.google.protobuf.kotlin
+
+import kotlinx.rpc.internal.utils.*
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.EmptyInternal.CODEC::class)
+public interface Empty { 
+    public companion object
+}
+

protobuf/protobuf-core/src/commonMain/generated-code/kotlin-multiplatform/com/google/protobuf/kotlin/FieldMask.kt

@@ -0,0 +1,12 @@
+@file:OptIn(ExperimentalRpcApi::class, InternalRpcApi::class)
+package com.google.protobuf.kotlin
+
+import kotlinx.rpc.internal.utils.*
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.FieldMaskInternal.CODEC::class)
+public interface FieldMask { 
+    public val paths: List<kotlin.String>
+
+    public companion object
+}
+

protobuf/protobuf-core/src/commonMain/generated-code/kotlin-multiplatform/com/google/protobuf/kotlin/SourceContext.kt

@@ -0,0 +1,12 @@
+@file:OptIn(ExperimentalRpcApi::class, InternalRpcApi::class)
+package com.google.protobuf.kotlin
+
+import kotlinx.rpc.internal.utils.*
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.SourceContextInternal.CODEC::class)
+public interface SourceContext { 
+    public val fileName: String
+
+    public companion object
+}
+

protobuf/protobuf-core/src/commonMain/generated-code/kotlin-multiplatform/com/google/protobuf/kotlin/Struct.kt

@@ -0,0 +1,63 @@
+@file:OptIn(ExperimentalRpcApi::class, InternalRpcApi::class)
+package com.google.protobuf.kotlin
+
+import kotlin.jvm.JvmInline
+import kotlinx.rpc.internal.utils.*
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.StructInternal.CODEC::class)
+public interface Struct { 
+    public val fields: Map<kotlin.String, com.google.protobuf.kotlin.Value>
+
+    public companion object
+}
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.ValueInternal.CODEC::class)
+public interface Value { 
+    public val kind: com.google.protobuf.kotlin.Value.Kind?
+
+    public sealed interface Kind { 
+        @JvmInline
+        public value class NullValue(
+            public val value: com.google.protobuf.kotlin.NullValue,
+        ): Kind
+
+        @JvmInline
+        public value class NumberValue(public val value: Double): Kind
+
+        @JvmInline
+        public value class StringValue(public val value: String): Kind
+
+        @JvmInline
+        public value class BoolValue(public val value: Boolean): Kind
+
+        @JvmInline
+        public value class StructValue(
+            public val value: com.google.protobuf.kotlin.Struct,
+        ): Kind
+
+        @JvmInline
+        public value class ListValue(
+            public val value: com.google.protobuf.kotlin.ListValue,
+        ): Kind
+    }
+
+    public companion object
+}
+
+@kotlinx.rpc.grpc.codec.WithCodec(com.google.protobuf.kotlin.ListValueInternal.CODEC::class)
+public interface ListValue { 
+    public val values: List<com.google.protobuf.kotlin.Value>
+
+    public companion object
+}
+
+public sealed class NullValue(public open val number: Int) { 
+    public object NULL_VALUE: NullValue(number = 0)
+
+    public data class UNRECOGNIZED(override val number: Int): NullValue(number)
+
+    public companion object { 
+        public val entries: List<NullValue> by lazy { listOf(NULL_VALUE) }
+    }
+}
+