Morfly

class FunctionProcessor(...) : SymbolProcessor {

    override fun process(resolver: Resolver): List<KSAnnotated> {
        val symbols = resolver
            .getSymbolsWithAnnotation("com.morfly.Function")
            .filterIsInstance<KSClassDeclaration>()
            
        if (!symbols.iterator().hasNext()) return emptyList()
    }
}

Morfly

class FunctionProcessor(...) : SymbolProcessor {

    override fun process(resolver: Resolver): List<KSAnnotated> {
        val symbols = resolver
            .getSymbolsWithAnnotation("com.morfly.Function")
            .filterIsInstance<KSClassDeclaration>()

        if (!symbols.iterator().hasNext()) return emptyList()

        val file: OutputStream = codeGenerator.createNewFile(

Morfly

package com.morfly

class FunctionProcessorProvider : SymbolProcessorProvider {

    override fun create(environment: SymbolProcessorEnvironment): SymbolProcessor {
        return FunctionProcessor(
            codeGenerator = environment.codeGenerator,
            logger = environment.logger,
            options = environment.options
        )

Morfly

package com.morfly

@Target(AnnotationTarget.CLASS)
annotation class Function(val name: String)

Morfly

interface KSVisitor<D, R> {
    fun visitNode(node: KSNode, data: D): R

    fun visitClassDeclaration(classDeclaration: KSClassDeclaration, data: D): R
    
    fun visitPropertyDeclaration(property: KSPropertyDeclaration, data: D): R

    // ~20 more functions
}

Morfly

interface KSNode {
    ...
    
    fun <D, R> accept(visitor: KSVisitor<D, R>, data: D): R
}

Morfly

class KSPropertyDeclarationImpl(...) : KSPropertyDeclaration, ... {
    ...
    
    override fun <D, R> accept(visitor: KSVisitor<D, R>, data: D): R {
        return visitor.visitPropertyDeclaration(this, data)
    }
}

Morfly

...

kotlin {
    sourceSets {
        main {
            kotlin.srcDirs(file("$buildDir/generated/ksp/main/kotlin"))
        }
    }
}

Morfly

class FunctionProcessor(...) : SymbolProcessor {
    ...

    inner class Visitor(private val file: OutputStream) : KSVisitorVoid() {
        
        override fun visitClassDeclaration(classDeclaration: KSClassDeclaration, data: Unit) {
        }
        
        override fun visitPropertyDeclaration(property: KSPropertyDeclaration, data: Unit) {
        }

Morfly

package com.morfly

@Function(name = "myAmazingFunction")
interface MyAmazingFunction

fun main() {
    println("Hello World!")
}