From 07ad79f0b38ffe2b2137a8d689cdcf88fd1c4f58 Mon Sep 17 00:00:00 2001 From: InsanusMokrassar Date: Wed, 12 Feb 2025 21:19:40 +0600 Subject: [PATCH] add including of annotations from source function on variation generation --- ksp/variations/generator/src/main/kotlin/Processor.kt | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ksp/variations/generator/src/main/kotlin/Processor.kt b/ksp/variations/generator/src/main/kotlin/Processor.kt index 43305fa14ef..60869f4088f 100644 --- a/ksp/variations/generator/src/main/kotlin/Processor.kt +++ b/ksp/variations/generator/src/main/kotlin/Processor.kt @@ -9,6 +9,7 @@ import com.google.devtools.ksp.processing.SymbolProcessor import com.google.devtools.ksp.symbol.* import com.squareup.kotlinpoet.* import com.squareup.kotlinpoet.ParameterizedTypeName.Companion.parameterizedBy +import com.squareup.kotlinpoet.ksp.toAnnotationSpec import com.squareup.kotlinpoet.ksp.toClassName import com.squareup.kotlinpoet.ksp.toKModifier import com.squareup.kotlinpoet.ksp.toTypeName @@ -90,6 +91,9 @@ class Processor( val baseFunctionDefaults = mutableMapOf() val funSpec = FunSpec.builder(ksFunctionDeclaration.simpleName.asString()).apply { modifiers.addAll(ksFunctionDeclaration.modifiers.mapNotNull { it.toKModifier() }) + ksFunctionDeclaration.annotations.forEach { + addAnnotation(it.toAnnotationSpec(omitDefaultValues = false)) + } ksFunctionDeclaration.extensionReceiver ?.let { receiver(it.toTypeName()) }