grpc-native: First message generation

Signed-off-by: Johannes Zottele <[email protected]>

Author: Jozott00

gradle-plugin/src/main/kotlin/kotlinx/rpc/proto/DefaultProtoSourceSet.kt

@@ -8,11 +8,7 @@ import kotlinx.rpc.buf.tasks.BufGenerateTask
 import kotlinx.rpc.util.findOrCreate
 import kotlinx.rpc.util.withKotlinJvmExtension
 import kotlinx.rpc.util.withKotlinKmpExtension
-import org.gradle.api.Action
-import org.gradle.api.GradleException
-import org.gradle.api.NamedDomainObjectFactory
-import org.gradle.api.NamedDomainObjectProvider
-import org.gradle.api.Project
+import org.gradle.api.*
 import org.gradle.api.file.SourceDirectorySet
 import org.gradle.api.provider.ListProperty
 import org.gradle.api.provider.Property
@@ -104,7 +100,7 @@ internal fun Project.createProtoExtensions() {
         findOrCreateAndConfigure("jvmTest", null)
 
         sourceSets.configureEach {
-            if (name == "jvmMain" || name == "jvmTest") {
+            if (name == "jvmMain" || name == "jvmTest" || name == "nativeTest") {
                 findOrCreateAndConfigure(name, this)
             }
         }

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/internal/KTag.kt renamed to grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/pb/KTag.kt

@@ -2,11 +2,9 @@
  * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
  */
 
-package kotlinx.rpc.grpc.internal
+package kotlinx.rpc.grpc.pb
 
-import kotlinx.rpc.grpc.internal.KTag.Companion.K_TAG_TYPE_BITS
-
-internal enum class WireType {
+public enum class WireType {
     VARINT, // 0
     FIXED64, // 1
     LENGTH_DELIMITED, // 2
@@ -15,13 +13,13 @@ internal enum class WireType {
     FIXED32, // 5
 }
 
-internal data class KTag(val fieldNr: Int, val wireType: WireType) {
+public data class KTag(val fieldNr: Int, val wireType: WireType) {
 
     init {
         check(isValidFieldNr(fieldNr)) { "Invalid field number: $fieldNr" }
     }
 
-    companion object {
+    internal companion object {
         // Number of bits in a tag which identify the wire type.
         const val K_TAG_TYPE_BITS: Int = 3;
 
@@ -31,7 +29,7 @@ internal data class KTag(val fieldNr: Int, val wireType: WireType) {
 }
 
 internal fun KTag.toRawKTag(): UInt {
-    return (fieldNr.toUInt() shl K_TAG_TYPE_BITS) or wireType.ordinal.toUInt()
+    return (fieldNr.toUInt() shl KTag.Companion.K_TAG_TYPE_BITS) or wireType.ordinal.toUInt()
 }
 
 internal fun KTag.Companion.fromOrNull(rawKTag: UInt): KTag? {

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/pb/WireDecoder.kt

@@ -0,0 +1,87 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.pb
+
+import kotlinx.io.Buffer
+import kotlinx.rpc.internal.utils.InternalRpcApi
+
+/**
+ * A platform-specific decoder for wire format data.
+ *
+ * This decoder is used by first calling [readTag], than looking up the field based on the field number in the returned,
+ * tag and then calling the actual `read*()` method to read the value to the corresponding field.
+ *
+ * [hadError] indicates an error during decoding. While calling `read*()` is safe, the returned values
+ * are meaningless if [hadError] returns `true`.
+ *
+ * NOTE: If the [hadError] after a call to `read*()` returns `false`, it doesn't mean that the
+ * value is correctly decoded. E.g., the following test will pass:
+ * ```kt
+ * val fieldNr = 1
+ * val buffer = Buffer()
+ *
+ * val encoder = WireEncoder(buffer)
+ * assertTrue(encoder.writeInt32(fieldNr, 12312))
+ * encoder.flush()
+ *
+ * WireDecoder(buffer).use { decoder ->
+ *     decoder.readTag()
+ *     decoder.readBool()
+ *     assertFalse(decoder.hasError())
+ * }
+ * ```
+ */
+@InternalRpcApi
+public interface WireDecoder : AutoCloseable {
+    public fun hadError(): Boolean
+
+    /**
+     * When the read tag is null, it indicates EOF and the parse may stop at this point.
+     */
+    public fun readTag(): KTag?
+    public fun readBool(): Boolean
+    public fun readInt32(): Int
+    public fun readInt64(): Long
+    public fun readUInt32(): UInt
+    public fun readUInt64(): ULong
+    public fun readSInt32(): Int
+    public fun readSInt64(): Long
+    public fun readFixed32(): UInt
+    public fun readFixed64(): ULong
+    public fun readSFixed32(): Int
+    public fun readSFixed64(): Long
+    public fun readFloat(): Float
+    public fun readDouble(): Double
+
+    public fun readEnum(): Int
+    public fun readString(): String
+    public fun readBytes(): ByteArray
+    public fun readPackedBool(): List<Boolean>
+    public fun readPackedInt32(): List<Int>
+    public fun readPackedInt64(): List<Long>
+    public fun readPackedSInt32(): List<Int>
+    public fun readPackedSInt64(): List<Long>
+    public fun readPackedUInt32(): List<UInt>
+    public fun readPackedUInt64(): List<ULong>
+    public fun readPackedFixed32(): List<UInt>
+    public fun readPackedFixed64(): List<ULong>
+    public fun readPackedSFixed32(): List<Int>
+    public fun readPackedSFixed64(): List<Long>
+    public fun readPackedFloat(): List<Float>
+    public fun readPackedDouble(): List<Double>
+    public fun readPackedEnum(): List<Int>
+}
+
+/**
+ * Creates a platform-specific [WireDecoder].
+ *
+ * This constructor takes a [Buffer] instead of a [kotlinx.io.Source] because
+ * the native implementation (`WireDecoderNative`) depends on [Buffer]'s internal structure.
+ *
+ * NOTE: Do not use the [source] buffer while the [WireDecoder] is still open.
+ *
+ * @param source The buffer containing the encoded wire-format data.
+ */
+internal expect fun WireDecoder(source: Buffer): WireDecoder

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/pb/WireEncoder.kt

@@ -0,0 +1,56 @@
+/*
+ * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
+ */
+
+package kotlinx.rpc.grpc.pb
+
+import kotlinx.io.Sink
+import kotlinx.rpc.internal.utils.InternalRpcApi
+
+/**
+ * A platform-specific class that encodes values into protobuf's wire format.
+ *
+ * If one `write*()` method returns false, the encoding of the value failed
+ * and no further encodings can be performed on this [WireEncoder].
+ *
+ * [flush] must be called to ensure that all data is written to the [Sink].
+ */
+@InternalRpcApi
+@OptIn(ExperimentalUnsignedTypes::class)
+public interface WireEncoder {
+    public fun flush()
+    public fun writeBool(field: Int, value: Boolean): Boolean
+    public fun writeInt32(fieldNr: Int, value: Int): Boolean
+    public fun writeInt64(fieldNr: Int, value: Long): Boolean
+    public fun writeUInt32(fieldNr: Int, value: UInt): Boolean
+    public fun writeUInt64(fieldNr: Int, value: ULong): Boolean
+    public fun writeSInt32(fieldNr: Int, value: Int): Boolean
+    public fun writeSInt64(fieldNr: Int, value: Long): Boolean
+    public fun writeFixed32(fieldNr: Int, value: UInt): Boolean
+    public fun writeFixed64(fieldNr: Int, value: ULong): Boolean
+    public fun writeSFixed32(fieldNr: Int, value: Int): Boolean
+    public fun writeSFixed64(fieldNr: Int, value: Long): Boolean
+    public fun writeFloat(fieldNr: Int, value: Float): Boolean
+    public fun writeDouble(fieldNr: Int, value: Double): Boolean
+    public fun writeEnum(fieldNr: Int, value: Int): Boolean
+    public fun writeBytes(fieldNr: Int, value: ByteArray): Boolean
+    public fun writeString(fieldNr: Int, value: String): Boolean
+    public fun writePackedBool(fieldNr: Int, value: List<Boolean>, fieldSize: Int): Boolean
+    public fun writePackedInt32(fieldNr: Int, value: List<Int>, fieldSize: Int): Boolean
+    public fun writePackedInt64(fieldNr: Int, value: List<Long>, fieldSize: Int): Boolean
+    public fun writePackedUInt32(fieldNr: Int, value: List<UInt>, fieldSize: Int): Boolean
+    public fun writePackedUInt64(fieldNr: Int, value: List<ULong>, fieldSize: Int): Boolean
+    public fun writePackedSInt32(fieldNr: Int, value: List<Int>, fieldSize: Int): Boolean
+    public fun writePackedSInt64(fieldNr: Int, value: List<Long>, fieldSize: Int): Boolean
+    public fun writePackedFixed32(fieldNr: Int, value: List<UInt>): Boolean
+    public fun writePackedFixed64(fieldNr: Int, value: List<ULong>): Boolean
+    public fun writePackedSFixed32(fieldNr: Int, value: List<Int>): Boolean
+    public fun writePackedSFixed64(fieldNr: Int, value: List<Long>): Boolean
+    public fun writePackedFloat(fieldNr: Int, value: List<Float>): Boolean
+    public fun writePackedDouble(fieldNr: Int, value: List<Double>): Boolean
+    public fun writePackedEnum(fieldNr: Int, value: List<Int>, fieldSize: Int): Boolean =
+        writePackedInt32(fieldNr, value, fieldSize)
+}
+
+
+internal expect fun WireEncoder(sink: Sink): WireEncoder

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/internal/WireSize.kt renamed to grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/pb/WireSize.kt

@@ -2,7 +2,7 @@
  * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
  */
 
-package kotlinx.rpc.grpc.internal
+package kotlinx.rpc.grpc.pb
 
 internal object WireSize
 

grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/internal/readPacked.kt renamed to grpc/grpc-core/src/commonMain/kotlin/kotlinx/rpc/grpc/pb/readPacked.kt

@@ -5,6 +5,8 @@
 package kotlinx.rpc.grpc.internal
 
 import kotlinx.io.Buffer
+import kotlinx.rpc.grpc.pb.*
+import kotlin.experimental.ExperimentalNativeApi
 import kotlin.test.*
 
 enum class TestPlatform {

grpc/grpc-core/src/commonTest/kotlin/kotlinx/rpc/grpc/internal/WireCodecTest.kt

@@ -2,10 +2,9 @@
  * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
  */
 
-package kotlinx.rpc.grpc.internal
+package kotlinx.rpc.grpc.pb
 
 import kotlinx.io.Buffer
-import kotlinx.io.Source
 
 internal actual fun WireDecoder(source: Buffer): WireDecoder {
     TODO("Not yet implemented")

grpc/grpc-core/src/jsMain/kotlin/kotlinx/rpc/grpc/internal/WireDecoder.js.kt renamed to grpc/grpc-core/src/jsMain/kotlin/kotlinx/rpc/grpc/pb/WireDecoder.js.kt

@@ -2,7 +2,7 @@
  * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
  */
 
-package kotlinx.rpc.grpc.internal
+package kotlinx.rpc.grpc.pb
 
 import kotlinx.io.Sink
 

grpc/grpc-core/src/jsMain/kotlin/kotlinx/rpc/grpc/internal/WireEncoder.js.kt renamed to grpc/grpc-core/src/jsMain/kotlin/kotlinx/rpc/grpc/pb/WireEncoder.js.kt

@@ -2,7 +2,7 @@
  * Copyright 2023-2025 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
  */
 
-package kotlinx.rpc.grpc.internal
+package kotlinx.rpc.grpc.pb
 
 internal actual fun WireSize.int32(value: Int): UInt {
     TODO("Not yet implemented")