Is it possible to get Class<T> from KSClassDeclaration ?

Hello, I have a question while using ksp. Is it possible to get Class<T> from KSClassDeclaration ? I want to know the Class<T> annotated with @CustomAnnotation.

For example, the following situation exists.

@Retention(AnnotationRetention.SOURCE)
annotation class CustomAnnotation

@CustomAnnotation
enum class MyEnum {
   DEMO
}

I want to get Class<MyEnum> in CustomAnnotationVisitor.visitClassDeclaration.

class CustomAnnotationProcessor(
    private val codeGenerator: CodeGenerator,
    private val logger: KSPLogger,
): SymbolProcessor {

    override fun process(resolver: Resolver): List<KSAnnotated> {
        val targetAnnotation = CustomAnnotation::class.java
        val elements = resolver.getSymbolsWithAnnotation("${targetAnnotation.packageName}.${targetAnnotation.simpleName}")

        val ret = elements.filter { !it.validate() }.toList()

        elements
            .filter { it is KSClassDeclaration && it.validate() }
            .forEach {
                it.accept(CustomAnnotationVisitor(), Unit)
            }

        return ret
    }

    inner class CustomAnnotationVisitor : KSVisitorVoid() {

        override fun visitClassDeclaration(classDeclaration: KSClassDeclaration, data: Unit) {
            // HERE!!!!!
        }
    }
}

I tried this but got an error.

class CustomAnnotationProcessor(
    private val codeGenerator: CodeGenerator,
    private val logger: KSPLogger,
): SymbolProcessor {

    override fun process(resolver: Resolver): List<KSAnnotated> {
        val targetAnnotation = CustomAnnotation::class.java
        val elements = resolver.getSymbolsWithAnnotation("${targetAnnotation.packageName}.${targetAnnotation.simpleName}")

        val ret = elements.filter { !it.validate() }.toList()

        elements
            .filter { it is KSClassDeclaration && it.validate() }
            .forEach {
                it.accept(CustomAnnotationVisitor(), Unit)
            }

        return ret
    }

    inner class CustomAnnotationVisitor : KSVisitorVoid() {

        override fun visitClassDeclaration(classDeclaration: KSClassDeclaration, data: Unit) {
             Class.forName(classDeclaration.qualifiedName!!.asString())
        }
    }
}

thank you.

[DHosseiny]

For Kotlin poet users: This library has an Extention to convert KSClassDeclaration to com.squareup.kotlinpoet.ClassName. For example for classDeclaration with classKind of annotation you can convert KSClassDeclaration to com.squareup.kotlinpoet.ClassName and use it to add as annotation:

val classDeclaration: KSClassDeclaration
TypeSpec.enumBuilder(enumClassName)
            .addAnnotation(
                classDeclaration.toClassName()
            )
            .build()

[sangyongchoi]

@neetopia thank you. I’ll check.