1 files changed, 148 insertions, 0 deletions
diff --git a/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotatedUtilProcessor.kt b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotatedUtilProcessor.kt
new file mode 100644
index 00000000..954feb43
--- /dev/null
+++ b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/AnnotatedUtilProcessor.kt
@@ -0,0 +1,148 @@
+/*
+ * Copyright 2020 Google LLC
+ * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.google.devtools.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.processing.Resolver
+import com.google.devtools.ksp.symbol.KSAnnotated
+import com.google.devtools.ksp.symbol.KSNode
+import com.google.devtools.ksp.visitor.KSTopDownVisitor
+import kotlin.reflect.KClass
+
+class AnnotatedUtilProcessor : AbstractTestProcessor() {
+    val results = mutableListOf<String>()
+    private val annotationKClasses = listOf(
+        ParametersTestAnnotation::class,
+        ParameterArraysTestAnnotation::class,
+        ParametersTestWithNegativeDefaultsAnnotation::class,
+        OuterAnnotation::class
+    )
+    private val visitors = listOf(IsAnnotationPresentVisitor(), GetAnnotationsByTypeVisitor())
+
+    override fun process(resolver: Resolver): List<KSAnnotated> {
+        resolver.getSymbolsWithAnnotation("com.google.devtools.ksp.processor.Test", true).forEach {
+            results.add("Test: $it")
+            visitors.forEach { visitor -> it.accept(visitor, results) }
+        }
+        return emptyList()
+    }
+
+    override fun toResult(): List<String> {
+        return results
+    }
+
+    inner class IsAnnotationPresentVisitor : KSTopDownVisitor<MutableCollection<String>, Unit>() {
+        @OptIn(KspExperimental::class)
+        override fun visitAnnotated(annotated: KSAnnotated, data: MutableCollection<String>) {
+            annotationKClasses.forEach { clazz ->
+                if (annotated.isAnnotationPresent(clazz)) {
+                    data.add("IsPresent: $clazz")
+                }
+            }
+        }
+
+        override fun defaultHandler(node: KSNode, data: MutableCollection<String>) {
+        }
+    }
+
+    inner class GetAnnotationsByTypeVisitor : KSTopDownVisitor<MutableCollection<String>, Unit>() {
+        @OptIn(KspExperimental::class)
+        override fun visitAnnotated(annotated: KSAnnotated, data: MutableCollection<String>) {
+            annotationKClasses.forEach { clazz ->
+                annotated.getAnnotationsByType(clazz).forEach { data.add("ByType: ${it.asString()}") }
+            }
+        }
+
+        override fun defaultHandler(node: KSNode, data: MutableCollection<String>) {
+        }
+    }
+}
+
+@Suppress("LongParameterList")
+annotation class ParametersTestAnnotation(
+    val booleanValue: Boolean = false,
+    val byteValue: Byte = 2,
+    val shortValue: Short = 3,
+    val charValue: Char = 'b',
+    val doubleValue: Double = 4.0,
+    val floatValue: Float = 5.0f,
+    val intValue: Int = 6,
+    val longValue: Long = 7L,
+    val stringValue: String = "emptystring",
+    val kClassValue: KClass<*> = ParametersTestAnnotation::class,
+    val enumValue: TestEnum = TestEnum.NONE,
+)
+
+fun ParameterArraysTestAnnotation.asString(): String =
+    "ParameterArraysTestAnnotation" +
+        "[booleanArrayValue=${booleanArrayValue.asList()}," +
+        "byteArrayValue=${byteArrayValue.asList()}," +
+        "shortArrayValue=${shortArrayValue.asList()}," +
+        "charArrayValue=${charArrayValue.asList()}," +
+        "doubleArrayValue=${doubleArrayValue.asList()}," +
+        "floatArrayValue=${floatArrayValue.asList()}," +
+        "intArrayValue=${intArrayValue.asList()}," +
+        "longArrayValue=${longArrayValue.asList()}," +
+        "stringArrayValue=${stringArrayValue.asList()}," +
+        "kClassArrayValue=${kClassArrayValue.asList()}," +
+        "enumArrayValue=${enumArrayValue.asList()}]"
+
+@Suppress("LongParameterList")
+annotation class ParameterArraysTestAnnotation(
+    val booleanArrayValue: BooleanArray = booleanArrayOf(),
+    val byteArrayValue: ByteArray = byteArrayOf(),
+    val shortArrayValue: ShortArray = shortArrayOf(),
+    val charArrayValue: CharArray = charArrayOf(),
+    val doubleArrayValue: DoubleArray = doubleArrayOf(),
+    val floatArrayValue: FloatArray = floatArrayOf(),
+    val intArrayValue: IntArray = intArrayOf(),
+    val longArrayValue: LongArray = longArrayOf(),
+    val stringArrayValue: Array<String> = emptyArray(),
+    val kClassArrayValue: Array<KClass<*>> = emptyArray(),
+    val enumArrayValue: Array<TestEnum> = emptyArray(),
+)
+
+annotation class ParametersTestWithNegativeDefaultsAnnotation(
+    val byteValue: Byte = -2,
+    val shortValue: Short = -3,
+    val doubleValue: Double = -4.0,
+    val floatValue: Float = -5.0f,
+    val intValue: Int = -6,
+    val longValue: Long = -7L,
+)
+
+enum class TestEnum {
+    NONE, VALUE1, VALUE2
+}
+
+annotation class Test
+
+annotation class InnerAnnotation(val value: String = "default")
+
+annotation class OuterAnnotation(
+    val innerAnnotation: InnerAnnotation = InnerAnnotation(),
+)
+
+fun Annotation.asString(): String {
+    return when (this) {
+        is ParameterArraysTestAnnotation -> asString()
+        else -> toString()
+    }
+}