Do not repeat @OptIn annotation for generated class

Foso · Feb 5, 2025 · 3cd31c1 · 3cd31c1
1 parent 737ab5c
commit 3cd31c1
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 15 deletions.
diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md
@@ -23,6 +23,7 @@ ktorfit{
 
 ## Fixed
 - @Headers annotation produces unexpected newline in generated code by ksp plugin #752
+- Generated code containing repeated @OptIn annotation #767
 
 # [2.2.0]()
 * Supported Kotlin version: 2.0.0; 2.0.10; 2.0.20, 2.1.0-Beta1; 2.0.21-RC, 2.0.21, 2.1.0-RC, 2.1.0-RC2, 2.1.0, 2.1.10

diff --git a/ktorfit-ksp/src/main/kotlin/de/jensklingenberg/ktorfit/poetspec/ImplClassSpec2.kt b/ktorfit-ksp/src/main/kotlin/de/jensklingenberg/ktorfit/poetspec/ImplClassSpec2.kt
@@ -1,7 +1,9 @@
 package de.jensklingenberg.ktorfit.poetspec
 
 import com.google.devtools.ksp.processing.Resolver
+import com.google.devtools.ksp.symbol.KSAnnotation
 import com.google.devtools.ksp.symbol.KSPropertyDeclaration
+import com.google.devtools.ksp.symbol.KSType
 import com.google.devtools.ksp.symbol.KSTypeReference
 import com.squareup.kotlinpoet.AnnotationSpec
 import com.squareup.kotlinpoet.ClassName
@@ -10,7 +12,7 @@ import com.squareup.kotlinpoet.FunSpec
 import com.squareup.kotlinpoet.KModifier
 import com.squareup.kotlinpoet.PropertySpec
 import com.squareup.kotlinpoet.TypeSpec
-import com.squareup.kotlinpoet.ksp.toAnnotationSpec
+import com.squareup.kotlinpoet.ksp.toClassName
 import com.squareup.kotlinpoet.ksp.toTypeName
 import de.jensklingenberg.ktorfit.KtorfitOptions
 import de.jensklingenberg.ktorfit.model.ClassData
@@ -56,25 +58,16 @@ private fun createImplClassTypeSpec(
     implClassProperties: List<PropertySpec>,
     funSpecs: List<FunSpec>
 ): TypeSpec {
-    val optInAnnotations =
-        classData.annotations.filter { it.shortName.getShortName() == "OptIn" }.map { it.toAnnotationSpec() }
     val helperProperty =
         PropertySpec
             .builder(converterHelper.objectName, converterHelper.toClassName())
             .initializer("${converterHelper.name}(${ktorfitClass.objectName})")
             .addModifiers(KModifier.PRIVATE)
             .build()
 
-    val internalApiAnnotation =
-        AnnotationSpec
-            .builder(ClassName("kotlin", "OptIn"))
-            .addMember(
-                "%T::class",
-                internalApi,
-            ).build()
     return TypeSpec
         .classBuilder(implClassName)
-        .addAnnotations(optInAnnotations + internalApiAnnotation)
+        .addAnnotation(getOptInAnnotation(classData.annotations))
         .addModifiers(classData.modifiers)
         .primaryConstructor(
             FunSpec
@@ -94,6 +87,27 @@ private fun createImplClassTypeSpec(
         .build()
 }
 
+private fun getOptInAnnotation(annotations: List<KSAnnotation>): AnnotationSpec {
+    val markerClasses =
+        annotations
+            .filter { it.shortName.getShortName() == "OptIn" }
+            .flatMap { annotation ->
+                @Suppress("UNCHECKED_CAST")
+                (annotation.arguments[0].value as? List<KSType>)
+                    ?.map { it.toClassName() }
+                    .orEmpty()
+            }
+            .plus(internalApi)
+            .toTypedArray<Any>()
+
+    val format = (1..markerClasses.size).joinToString { "%T::class" }
+
+    return AnnotationSpec
+        .builder(ClassName("kotlin", "OptIn"))
+        .addMember(format, *markerClasses)
+        .build()
+}
+
 private fun propertySpec(property: KSPropertyDeclaration): PropertySpec {
     val propBuilder =
         PropertySpec

diff --git a/ktorfit-ksp/src/test/kotlin/de/jensklingenberg/ktorfit/OptInTest.kt b/ktorfit-ksp/src/test/kotlin/de/jensklingenberg/ktorfit/OptInTest.kt
@@ -30,8 +30,7 @@ suspend fun test(@Header("testHeader") testParameterNonNullable: String?, @Heade
             )
 
         val expectedHeadersArgumentText =
-            """@OptIn(ExperimentalCompilerApi::class)
-@OptIn(InternalKtorfitApi::class)
+            """@OptIn(ExperimentalCompilerApi::class, InternalKtorfitApi::class)
 public class _TestServiceImpl(
   private val _ktorfit: Ktorfit,
 ) : TestService {
@@ -41,9 +40,8 @@ public class _TestServiceImpl(
   override suspend fun test("""
 
         val compilation = getCompilation(listOf(source))
-        val result = compilation.compile()
 
-        val generatedSourcesDir = compilation.kspSourcesDir
+        val generatedSourcesDir = compilation.apply { compile() }.kspSourcesDir
         val generatedFile =
             File(
                 generatedSourcesDir,