kotlinx.rpc.codegen.RPCSymbolProcessorContext.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlinx-rpc-ksp-plugin Show documentation

kotlinx.rpc, a Kotlin library for adding asynchronous RPC services to your applications.

There is a newer version: 2.0.10-0.2.4

Show newest version

/*
 * Copyright 2023-2024 JetBrains s.r.o and contributors. Use of this source code is governed by the Apache 2.0 license.
 */

package kotlinx.rpc.codegen

import com.google.devtools.ksp.getClassDeclarationByName
import com.google.devtools.ksp.processing.Resolver
import com.google.devtools.ksp.symbol.KSType
import com.google.devtools.ksp.symbol.KSTypeArgument

class RPCSymbolProcessorContext(private val resolver: Resolver) {
    val rpcType by lazy { resolveClass("kotlinx.rpc.RPC") }

    val plainFlowType by lazy { resolveClass("kotlinx.coroutines.flow.Flow") }

    val sharedFlowType by lazy { resolveClass("kotlinx.coroutines.flow.SharedFlow") }

    val stateFlowType by lazy { resolveClass("kotlinx.coroutines.flow.StateFlow") }

    val rpcEagerProperty by lazy { resolveClass("kotlinx.rpc.RPCEagerField") }

    private fun resolveClass(name: String, typeArguments: List = emptyList()): KSType {
        return resolver.getClassDeclarationByName(name)?.asType(typeArguments)
            ?: codegenError("Could not find $name declaration")
    }
}