aboutsummaryrefslogtreecommitdiff
path: root/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/DeclarationOrderProcessor.kt
diff options
context:
space:
mode:
Diffstat (limited to 'test-utils/src/main/kotlin/com/google/devtools/ksp/processor/DeclarationOrderProcessor.kt')
-rw-r--r--test-utils/src/main/kotlin/com/google/devtools/ksp/processor/DeclarationOrderProcessor.kt44
1 files changed, 44 insertions, 0 deletions
diff --git a/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/DeclarationOrderProcessor.kt b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/DeclarationOrderProcessor.kt
new file mode 100644
index 00000000..6d9b9f61
--- /dev/null
+++ b/test-utils/src/main/kotlin/com/google/devtools/ksp/processor/DeclarationOrderProcessor.kt
@@ -0,0 +1,44 @@
+package com.google.devtools.ksp.processor
+
+import com.google.devtools.ksp.KspExperimental
+import com.google.devtools.ksp.getClassDeclarationByName
+import com.google.devtools.ksp.processing.Resolver
+import com.google.devtools.ksp.symbol.KSAnnotated
+import com.google.devtools.ksp.symbol.KSDeclaration
+import com.google.devtools.ksp.symbol.KSFunctionDeclaration
+import com.google.devtools.ksp.symbol.KSPropertyDeclaration
+
+@KspExperimental
+class DeclarationOrderProcessor : AbstractTestProcessor() {
+ private val result = mutableListOf<String>()
+ override fun toResult() = result
+
+ override fun process(resolver: Resolver): List<KSAnnotated> {
+ val classNames = listOf(
+ "lib.KotlinClass", "lib.JavaClass",
+ "KotlinClass", "JavaClass"
+ )
+ classNames.map {
+ checkNotNull(resolver.getClassDeclarationByName(it)) {
+ "cannot find $it"
+ }
+ }.forEach { klass ->
+ result.add(klass.qualifiedName!!.asString())
+ result.addAll(
+ resolver.getDeclarationsInSourceOrder(klass).filterIsInstance<KSPropertyDeclaration>().map {
+ it.toSignature(resolver)
+ }
+ )
+ result.addAll(
+ resolver.getDeclarationsInSourceOrder(klass).filterIsInstance<KSFunctionDeclaration>().map {
+ it.toSignature(resolver)
+ }
+ )
+ }
+ return emptyList()
+ }
+
+ private fun KSDeclaration.toSignature(
+ resolver: Resolver
+ ) = "${simpleName.asString()}:${resolver.mapToJvmSignature(this)}"
+}