jvmMain.com.caesarealabs.rpc4k.processor.ApiDefinitionUtils.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rpc4k-processor-jvm Show documentation

RPC Utility

The newest version!

package com.caesarealabs.rpc4k.processor

import com.caesarealabs.rpc4k.processor.utils.poet.FormattedString
import com.caesarealabs.rpc4k.processor.utils.poet.toSerializerString
import com.caesarealabs.rpc4k.processor.utils.poet.withArgumentList
import com.squareup.kotlinpoet.AnnotationSpec
import com.squareup.kotlinpoet.ClassName
import com.squareup.kotlinpoet.FileSpec
import com.squareup.kotlinpoet.MemberName

internal object ApiDefinitionUtils {
     const val FactoryName: String = "Factory"

     const val NetworkClientSuffix: String = "NetworkClient"
     const val ClientInterfaceSuffix: String = "Client"

     const val ServerSuffix: String = "Router"

     const val Group: String = "com.caesarealabs"

     const val Package: String = "${Group}.rpc4k.generated"


    val listOfFunction = MemberName("kotlin.collections", "listOf")
    // Kotlin doesn't let us reference these annotations directly sadly so we need to resort to strings
     val experimentalUnsignedTypes = ClassName("kotlin", "ExperimentalUnsignedTypes")
    private val experimentalSerializationApi = ClassName("kotlinx.serialization", "ExperimentalSerializationApi")
     val optIn = ClassName("kotlin", "OptIn")

    fun listOfSerializers(rpc: RpcFunction): FormattedString {
        return listOfFunction.withArgumentList(rpc.parameters.map { it.type.toSerializerString() })
    }

    fun listOfEventSubSerializers(rpc: RpcEventEndpoint): FormattedString {
        // We don't need to deserialize the target either because we have it as a normal kotlin value coming from the invoker.
        return listOfFunction.withArgumentList(rpc.parameters.filter { !it.isDispatch && !it.isTarget }.map { it.value.type.toSerializerString() })
    }
    fun FileSpec.Builder.ignoreExperimentalWarnings() {
        addAnnotation(AnnotationSpec.builder(optIn).addMember("%T::class, %T::class", experimentalUnsignedTypes, experimentalSerializationApi).build())
    }
}