kotlinx.rpc.codegen.extension.RPCIrServiceProcessor.kt Maven / Gradle / Ivy

Go to download

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

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

There is a newer version: 2.0.21-0.4.0

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.extension

import org.jetbrains.kotlin.cli.common.messages.MessageCollector
import org.jetbrains.kotlin.ir.IrStatement
import org.jetbrains.kotlin.ir.declarations.IrClass
import org.jetbrains.kotlin.ir.types.defaultType
import org.jetbrains.kotlin.ir.util.isInterface
import org.jetbrains.kotlin.ir.visitors.IrElementTransformer

internal class RPCIrServiceProcessor(
    @Suppress("unused")
    private val logger: MessageCollector,
) : IrElementTransformer {
    @Suppress("PARAMETER_NAME_CHANGED_ON_OVERRIDE")
    override fun visitClass(declaration: IrClass, context: RPCIrContext): IrStatement {
        if (declaration.isInterface && declaration.superTypes.contains(context.rpc.defaultType)) {
            processService(declaration, context)
        }

        return super.visitClass(declaration, context)
    }

    private fun processService(service: IrClass, context: RPCIrContext) {
        val declaration = RPCDeclarationScanner.scanServiceDeclaration(service, context)
        RPCStubGenerator(declaration, context, logger).generate()
    }
}