mirror of
https://github.com/InsanusMokrassar/TelegramBotAPI.git
synced 2025-12-28 08:59:27 +00:00
update classcasts
This commit is contained in:
@@ -1,9 +1,13 @@
|
||||
package dev.inmo.tgbotapi.ksp.processor
|
||||
|
||||
import com.google.devtools.ksp.KspExperimental
|
||||
import com.google.devtools.ksp.getAnnotationsByType
|
||||
import com.google.devtools.ksp.isAnnotationPresent
|
||||
import com.google.devtools.ksp.symbol.*
|
||||
import com.squareup.kotlinpoet.*
|
||||
import com.squareup.kotlinpoet.ParameterizedTypeName.Companion.parameterizedBy
|
||||
import com.squareup.kotlinpoet.ksp.*
|
||||
import dev.inmo.tgbotapi.utils.internal.ClassCastsIncluded
|
||||
|
||||
private fun FileSpec.Builder.addTopLevelImport(className: ClassName) {
|
||||
className.topLevelClassName().let {
|
||||
@@ -27,6 +31,24 @@ private fun FileSpec.Builder.createTypeDefinition(ksClassDeclaration: KSClassDec
|
||||
}
|
||||
}
|
||||
|
||||
@OptIn(KspExperimental::class)
|
||||
private fun KSClassDeclaration.buildPrefix(sourceDeclaration: KSClassDeclaration): String {
|
||||
val ownName = if (isAnnotationPresent(ClassCastsIncluded.ExcludeSubName::class)) {
|
||||
""
|
||||
} else {
|
||||
simpleName.asString()
|
||||
}
|
||||
when (val parentDeclaration = parentDeclaration) {
|
||||
is KSClassDeclaration -> if (parentDeclaration === sourceDeclaration) {
|
||||
return ownName
|
||||
} else {
|
||||
return "${parentDeclaration.buildPrefix(sourceDeclaration)}$ownName"
|
||||
}
|
||||
}
|
||||
return ownName
|
||||
}
|
||||
|
||||
@OptIn(KspExperimental::class)
|
||||
fun FileSpec.Builder.fill(
|
||||
sourceKSClassDeclaration: KSClassDeclaration,
|
||||
subtypesMap: Map<KSClassDeclaration, Set<KSClassDeclaration>>,
|
||||
@@ -40,8 +62,10 @@ fun FileSpec.Builder.fill(
|
||||
val sourceClassName = sourceKSClassDeclaration.toClassName()
|
||||
val targetClassClassName = targetClassDeclaration.toClassName()
|
||||
val targetClassTypeDefinition = createTypeDefinition(targetClassDeclaration)
|
||||
val simpleName = targetClassDeclaration.simpleName.asString()
|
||||
val withFirstLowerCase = simpleName.replaceFirstChar { it.lowercase() }
|
||||
// val simpleName = targetClassDeclaration.simpleName.asString()
|
||||
// val additionalPrefix = targetClassDeclaration.buildPrefix()
|
||||
val resultPrefix = targetClassDeclaration.buildPrefix(sourceKSClassDeclaration)
|
||||
val withFirstLowerCase = resultPrefix.replaceFirstChar { it.lowercase() }
|
||||
val castedOrNullName = "${withFirstLowerCase}OrNull"
|
||||
|
||||
addTopLevelImport(targetClassClassName)
|
||||
@@ -68,7 +92,7 @@ fun FileSpec.Builder.fill(
|
||||
}.build()
|
||||
)
|
||||
addFunction(
|
||||
FunSpec.builder("if$simpleName").apply {
|
||||
FunSpec.builder("if$resultPrefix").apply {
|
||||
val genericType = TypeVariableName("T", null)
|
||||
addTypeVariable(genericType)
|
||||
receiver(sourceClassName)
|
||||
|
||||
Reference in New Issue
Block a user