Updated descriptor to use RpcType instead of KType directly

Mr3zee · Mr3zee · commit a53708fa4c22 · 2024-11-25T20:24:58.000+01:00
diff --git a/compiler-plugin/compiler-plugin-backend/src/main/core/kotlinx/rpc/codegen/extension/RpcIrContext.kt b/compiler-plugin/compiler-plugin-backend/src/main/core/kotlinx/rpc/codegen/extension/RpcIrContext.kt
@@ -105,6 +105,10 @@ internal class RpcIrContext(
         getRpcIrClassSymbol("RpcServiceDescriptor", "descriptor")
     }
 
+    val rpcType by lazy {
+        getRpcIrClassSymbol("RpcType", "descriptor")
+    }
+
     val rpcCallable by lazy {
         getRpcIrClassSymbol("RpcCallable", "descriptor")
     }
diff --git a/compiler-plugin/compiler-plugin-backend/src/main/core/kotlinx/rpc/codegen/extension/RpcStubGenerator.kt b/compiler-plugin/compiler-plugin-backend/src/main/core/kotlinx/rpc/codegen/extension/RpcStubGenerator.kt
@@ -976,6 +976,11 @@ internal class RpcStubGenerator(
      * Where:
      *  - `<field-name>` - the name of the field
      */
+    @Suppress(
+        "detekt.NestedBlockDepth",
+        "detekt.LongMethod",
+        "detekt.CyclomaticComplexMethod",
+    )
     private fun IrClass.generateInvokator(i: Int, callable: ServiceDeclaration.Callable) {
         invokators[callable.name] = addProperty {
             name = Name.identifier("${callable.name}Invokator")
@@ -1205,13 +1210,13 @@ internal class RpcStubGenerator(
      * ```kotlin
      * RpcCallable<MyService>(
      *     name = "<callable-name>",
-     *     dataType = typeOf<<callable-data-type>>(),
-     *     returnType = typeOf<<callable-return-type>>(),
+     *     dataType = RpcCall(typeOf<<callable-data-type>>()),
+     *     returnType = RpcCall(typeOf<<callable-return-type>>()),
      *     invokator = <callable-invokator>,
      *     parameters = arrayOf( // or emptyArray()
      *         RpcParameter(
      *             "<method-parameter-name-1>",
-     *             typeOf<<method-parameter-type-1>>(),
+     *             RpcCall(typeOf<<method-parameter-type-1>>())
      *         ),
      *         ...
      *     ),
@@ -1247,14 +1252,14 @@ internal class RpcStubGenerator(
                 is ServiceDeclaration.FlowField -> ctx.fieldDataObject.defaultType
             }
 
-            putValueArgument(1, irTypeOfCall(dataType))
+            putValueArgument(1, irRpcTypeCall(dataType))
 
             val returnType = when (callable) {
                 is ServiceDeclaration.Method -> callable.function.returnType
                 is ServiceDeclaration.FlowField -> callable.property.getterOrFail.returnType
             }
 
-            putValueArgument(2, irTypeOfCall(returnType))
+            putValueArgument(2, irRpcTypeCall(returnType))
 
             val invokator = invokators[callable.name]
                 ?: error("Expected invokator for ${callable.name} in ${declaration.service.name}")
@@ -1309,7 +1314,7 @@ internal class RpcStubGenerator(
                             )
                         }.apply {
                             putValueArgument(0, stringConst(parameter.value.name.asString()))
-                            putValueArgument(1, irTypeOfCall(parameter.type))
+                            putValueArgument(1, irRpcTypeCall(parameter.type))
                         }
                     },
                 )
@@ -1521,6 +1526,25 @@ internal class RpcStubGenerator(
         }
     }
 
+    /**
+     * IR call of the `RpcType(KType, Array<Annotation>)` function
+     */
+    private fun irRpcTypeCall(type: IrType): IrConstructorCallImpl {
+        return vsApi {
+            IrConstructorCallImplVS(
+                startOffset = UNDEFINED_OFFSET,
+                endOffset = UNDEFINED_OFFSET,
+                type = ctx.rpcType.defaultType,
+                symbol = ctx.rpcType.constructors.single(),
+                typeArgumentsCount = 0,
+                valueArgumentsCount = 1,
+                constructorTypeArgumentsCount = 0,
+            )
+        }.apply {
+            putValueArgument(0, irTypeOfCall(type))
+        }
+    }
+
     /**
      * IR call of the `typeOf<...>()` function
      */
diff --git a/core/api/core.api b/core/api/core.api
@@ -56,12 +56,12 @@ public abstract interface annotation class kotlinx/rpc/annotations/Rpc : java/la
 }
 
 public final class kotlinx/rpc/descriptor/RpcCallable {
-	public fun <init> (Ljava/lang/String;Lkotlin/reflect/KType;Lkotlin/reflect/KType;Lkotlinx/rpc/descriptor/RpcInvokator;[Lkotlinx/rpc/descriptor/RpcParameter;)V
-	public final fun getDataType ()Lkotlin/reflect/KType;
+	public fun <init> (Ljava/lang/String;Lkotlinx/rpc/descriptor/RpcType;Lkotlinx/rpc/descriptor/RpcType;Lkotlinx/rpc/descriptor/RpcInvokator;[Lkotlinx/rpc/descriptor/RpcParameter;)V
+	public final fun getDataType ()Lkotlinx/rpc/descriptor/RpcType;
 	public final fun getInvokator ()Lkotlinx/rpc/descriptor/RpcInvokator;
 	public final fun getName ()Ljava/lang/String;
 	public final fun getParameters ()[Lkotlinx/rpc/descriptor/RpcParameter;
-	public final fun getReturnType ()Lkotlin/reflect/KType;
+	public final fun getReturnType ()Lkotlinx/rpc/descriptor/RpcType;
 }
 
 public abstract interface class kotlinx/rpc/descriptor/RpcInvokator {
@@ -76,9 +76,9 @@ public abstract interface class kotlinx/rpc/descriptor/RpcInvokator$Method : kot
 }
 
 public final class kotlinx/rpc/descriptor/RpcParameter {
-	public fun <init> (Ljava/lang/String;Lkotlin/reflect/KType;)V
+	public fun <init> (Ljava/lang/String;Lkotlinx/rpc/descriptor/RpcType;)V
 	public final fun getName ()Ljava/lang/String;
-	public final fun getType ()Lkotlin/reflect/KType;
+	public final fun getType ()Lkotlinx/rpc/descriptor/RpcType;
 }
 
 public abstract interface class kotlinx/rpc/descriptor/RpcServiceDescriptor {
@@ -92,3 +92,9 @@ public final class kotlinx/rpc/descriptor/RpcServiceDescriptorKt {
 	public static final fun serviceDescriptorOf (Lkotlin/reflect/KType;)Lkotlinx/rpc/descriptor/RpcServiceDescriptor;
 }
 
+public final class kotlinx/rpc/descriptor/RpcType {
+	public fun <init> (Lkotlin/reflect/KType;)V
+	public final fun getKType ()Lkotlin/reflect/KType;
+	public fun toString ()Ljava/lang/String;
+}
+
diff --git a/core/src/commonMain/kotlin/kotlinx/rpc/descriptor/RpcServiceDescriptor.kt b/core/src/commonMain/kotlin/kotlinx/rpc/descriptor/RpcServiceDescriptor.kt
@@ -54,8 +54,8 @@ public interface RpcServiceDescriptor<T : RemoteService> {
 @ExperimentalRpcApi
 public class RpcCallable<T : RemoteService>(
     public val name: String,
-    public val dataType: KType,
-    public val returnType: KType,
+    public val dataType: RpcType,
+    public val returnType: RpcType,
     public val invokator: RpcInvokator<T>,
     public val parameters: Array<RpcParameter>,
 )
@@ -74,4 +74,11 @@ public sealed interface RpcInvokator<T : RemoteService> {
 }
 
 @ExperimentalRpcApi
-public class RpcParameter(public val name: String, public val type: KType)
+public class RpcParameter(public val name: String, public val type: RpcType)
+
+@ExperimentalRpcApi
+public class RpcType(public val kType: KType) {
+    override fun toString(): String {
+        return return kType.toString()
+    }
+}
diff --git a/gradle-conventions/latest-only/src/main/kotlin/conventions-gradle-doctor.gradle.kts b/gradle-conventions/latest-only/src/main/kotlin/conventions-gradle-doctor.gradle.kts
@@ -9,6 +9,6 @@ plugins {
 doctor {
     enableTestCaching = false
     warnWhenNotUsingParallelGC = true
-    disallowMultipleDaemons = true
+    disallowMultipleDaemons = false
     GCFailThreshold = 0.5f
 }
diff --git a/krpc/krpc-core/src/commonMain/kotlin/kotlinx/rpc/krpc/internal/SerializationUtils.kt b/krpc/krpc-core/src/commonMain/kotlin/kotlinx/rpc/krpc/internal/SerializationUtils.kt
@@ -7,6 +7,7 @@ package kotlinx.rpc.krpc.internal
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.SharedFlow
 import kotlinx.coroutines.flow.StateFlow
+import kotlinx.rpc.descriptor.RpcType
 import kotlinx.rpc.internal.utils.InternalRpcApi
 import kotlinx.serialization.*
 import kotlinx.serialization.modules.SerializersModule
@@ -27,6 +28,11 @@ internal fun SerializersModule.buildContextual(type: KType): KSerializer<Any?> {
     return result as? KSerializer<Any?> ?: error("No serializer found for $type")
 }
 
+@InternalRpcApi
+public fun SerializersModule.rpcSerializerForType(type: RpcType): KSerializer<Any?> {
+    return rpcSerializerForType(type.kType)
+}
+
 @InternalRpcApi
 public fun SerializersModule.rpcSerializerForType(type: KType): KSerializer<Any?> {
     return when (type.classifier) {
@@ -35,6 +41,7 @@ public fun SerializersModule.rpcSerializerForType(type: KType): KSerializer<Any?
     }
 }
 
+
 @InternalRpcApi
 public fun unsupportedSerialFormatError(serialFormat: SerialFormat): Nothing {
     error("Unsupported serial format ${serialFormat::class}, only StringFormat and BinaryFormats are supported")
diff --git a/krpc/krpc-test/src/jvmMain/kotlin/kotlinx/rpc/krpc/test/KrpcTestService.kt b/krpc/krpc-test/src/jvmMain/kotlin/kotlinx/rpc/krpc/test/KrpcTestService.kt
@@ -35,7 +35,7 @@ interface KrpcTestService : RemoteService {
     suspend fun nonSerializableClass(localDate: @Contextual LocalDate): LocalDate
     suspend fun nonSerializableClassWithSerializer(
         localDateTime: @Serializable(LocalDateTimeSerializer::class) LocalDateTime,
-    ): @Serializable(LocalDateTimeSerializer::class) LocalDateTime
+    ): String
 
     suspend fun incomingStreamSyncCollect(arg1: Flow<String>): Int
     suspend fun incomingStreamAsyncCollect(arg1: Flow<String>): Int
diff --git a/krpc/krpc-test/src/jvmMain/kotlin/kotlinx/rpc/krpc/test/KrpcTransportTestBase.kt b/krpc/krpc-test/src/jvmMain/kotlin/kotlinx/rpc/krpc/test/KrpcTransportTestBase.kt
@@ -157,7 +157,6 @@ abstract class KrpcTransportTestBase {
     }
 
     @Test
-    @Ignore // todo will fix in next PR
     fun nonSerializableParameter() {
         runBlocking {
             val localDate = LocalDate.of(2001, 8, 23)

Original file line number	Diff line number	Diff line change
`@@ -105,6 +105,10 @@ internal class RpcIrContext(`
`105`	`105`	`getRpcIrClassSymbol("RpcServiceDescriptor", "descriptor")`
`106`	`106`	`}`
`107`	`107`
	`108`	`+ val rpcType by lazy {`
	`109`	`+ getRpcIrClassSymbol("RpcType", "descriptor")`
	`110`	`+ }`
	`111`	`+`
`108`	`112`	`val rpcCallable by lazy {`
`109`	`113`	`getRpcIrClassSymbol("RpcCallable", "descriptor")`
`110`	`114`	`}`
Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,6 @@ plugins {`
`9`	`9`	`doctor {`
`10`	`10`	`enableTestCaching = false`
`11`	`11`	`warnWhenNotUsingParallelGC = true`
`12`		`- disallowMultipleDaemons = true`
	`12`	`+ disallowMultipleDaemons = false`
`13`	`13`	`GCFailThreshold = 0.5f`
`14`	`14`	`}`
Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,6 @@ abstract class KrpcTransportTestBase {`
`157`	`157`	`}`
`158`	`158`
`159`	`159`	`@Test`
`160`		`- @Ignore // todo will fix in next PR`
`161`	`160`	`fun nonSerializableParameter() {`
`162`	`161`	`runBlocking {`
`163`	`162`	`val localDate = LocalDate.of(2001, 8, 23)`