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(
+    sourceSetSuffix: String = "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 (name.endsWith(sourceSetSuffix)) {
             val includedBuild = gradle.includedBuild("protoc-gen")
             dependsOn(includedBuild.task(":grpc:jar"))
             dependsOn(includedBuild.task(":protobuf:jar"))

protobuf/protobuf-core/build.gradle.kts

@@ -19,6 +19,7 @@ kotlin {
             dependencies {
                 api(projects.utils)
                 api(projects.protobuf.protobufInputStream)
+                api(projects.protobuf.protobufWellKnownTypes)
                 api(libs.kotlinx.io.core)
             }
         }

protobuf/protobuf-well-known-types/build.gradle.kts

@@ -0,0 +1,48 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+@file:OptIn(InternalRpcApi::class)
+
+import kotlinx.rpc.buf.tasks.BufGenerateTask
+import kotlinx.rpc.internal.InternalRpcApi
+import kotlinx.rpc.internal.configureLocalProtocGenDevelopmentDependency
+import org.jetbrains.kotlin.gradle.dsl.KotlinVersion
+
+plugins {
+    alias(libs.plugins.conventions.kmp)
+    alias(libs.plugins.kotlinx.rpc)
+}
+
+configureLocalProtocGenDevelopmentDependency(sourceSetSuffix = "Main")
+
+val generatedCodeDir = layout.projectDirectory
+    .dir("src")
+    .dir("commonMain")
+    .dir("generated-code")
+    .asFile
+
+tasks.withType<BufGenerateTask>().configureEach {
+    if (name.contains("Main")) {
+        includeWkt = true
+        outputDirectory = generatedCodeDir
+    }
+}
+
+kotlin {
+    // for timestamp
+    compilerOptions {
+        apiVersion = KotlinVersion.KOTLIN_2_1
+    }
+
+    sourceSets {
+        commonMain {
+            dependencies {
+                api(projects.protobuf.protobufCore)
+                api(projects.grpc.grpcCodec)
+
+                api(libs.kotlinx.io.core)
+            }
+        }
+    }
+}

protobuf/protobuf-well-known-types/gradle.properties

@@ -0,0 +1,21 @@
+#
+# Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+#
+kotlinx.rpc.exclude.wasmWasi=true
+kotlinx.rpc.exclude.js=true
+kotlinx.rpc.exclude.wasmJs=true
+kotlinx.rpc.exclude.iosArm64=true
+kotlinx.rpc.exclude.iosX64=true
+kotlinx.rpc.exclude.iosSimulatorArm64=true
+kotlinx.rpc.exclude.linuxArm64=true
+kotlinx.rpc.exclude.linuxX64=true
+kotlinx.rpc.exclude.macosX64=true
+kotlinx.rpc.exclude.mingwX64=true
+kotlinx.rpc.exclude.tvosArm64=true
+kotlinx.rpc.exclude.tvosSimulatorArm64=true
+kotlinx.rpc.exclude.tvosX64=true
+kotlinx.rpc.exclude.watchosArm32=true
+kotlinx.rpc.exclude.watchosArm64=true
+kotlinx.rpc.exclude.watchosDeviceArm64=true
+kotlinx.rpc.exclude.watchosSimulatorArm64=true
+kotlinx.rpc.exclude.watchosX64=true

protobuf/protobuf-well-known-types/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-well-known-types/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-well-known-types/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-well-known-types/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-well-known-types/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
+}
+