package dev.inmo.tgbotapi.ksp.processor import com.google.devtools.ksp.getAllSuperTypes import com.google.devtools.ksp.processing.* import com.google.devtools.ksp.symbol.KSAnnotated import com.google.devtools.ksp.symbol.KSClassDeclaration import com.squareup.kotlinpoet.AnnotationSpec import com.squareup.kotlinpoet.FileSpec import com.squareup.kotlinpoet.ksp.writeTo import dev.inmo.tgbotapi.ksp.lib.ClassCastsIncluded import java.io.File class TelegramBotAPISymbolProcessor( private val codeGenerator: CodeGenerator, private val targetPackage: String = "", private val outputFile: String = "Output", private val outputFolder: String? = null ) : SymbolProcessor { override fun process(resolver: Resolver): List { val classes = resolver.getSymbolsWithAnnotation(ClassCastsIncluded::class.qualifiedName!!).filterIsInstance() val classesSubtypes = mutableMapOf>() resolver.getAllFiles().forEach { it.declarations.forEach { potentialSubtype -> if (potentialSubtype is KSClassDeclaration) { val allSupertypes = potentialSubtype.getAllSuperTypes().map { it.declaration } classes.forEach { if (it in allSupertypes) { classesSubtypes.getOrPut(it) { mutableSetOf() }.add(potentialSubtype) } } } } } fun fillWithSealeds(source: KSClassDeclaration, current: KSClassDeclaration = source) { current.getSealedSubclasses().forEach { classesSubtypes.getOrPut(source) { mutableSetOf() }.add(it) fillWithSealeds(source, it) } } classes.forEach { fillWithSealeds(it) } val fileSpec = FileSpec.builder( targetPackage, outputFile ).apply { addAnnotation( AnnotationSpec.builder(Suppress::class).apply { addMember("\"unused\"") addMember("\"RemoveRedundantQualifierName\"") addMember("\"RedundantVisibilityModifier\"") addMember("\"NOTHING_TO_INLINE\"") addMember("\"UNCHECKED_CAST\"") addMember("\"OPT_IN_USAGE\"") useSiteTarget(AnnotationSpec.UseSiteTarget.FILE) }.build() ) classes.forEach { fill( it, classesSubtypes.toMap() ) } }.build() runCatching { outputFolder ?.also { File(it).apply { delete() runCatching { mkdirs() } fileSpec.writeTo(this) } } ?: fileSpec.writeTo(codeGenerator, false) } return emptyList() } }