Add the exampel project to integration tests

Author: ting-yuan

integration-tests/src/test/kotlin/com/google/devtools/ksp/test/PlaygroundIT.kt

@@ -0,0 +1,30 @@
+package com.google.devtools.ksp.test
+
+import TestsCommon
+import org.gradle.testkit.runner.GradleRunner
+import org.gradle.testkit.runner.TaskOutcome
+import org.junit.Assert
+import org.junit.Test
+import java.io.File
+import java.util.jar.*
+
+class PlaygroundIT : TestsCommon() {
+    override val testProjectName = "playground"
+
+    @Test
+    fun testPlayground() {
+        val gradleRunner = GradleRunner.create().withProjectDir(testProjectDir)
+
+        val resultCleanBuild = gradleRunner.withArguments("clean", "build").build()
+
+        Assert.assertEquals(TaskOutcome.SUCCESS, resultCleanBuild.task(":workload:build")?.outcome)
+
+        val artifact = File(testProjectDir, "workload/build/libs/workload-1.0-SNAPSHOT.jar")
+        Assert.assertTrue(artifact.exists())
+
+        val jarFile = JarFile(artifact)
+        Assert.assertTrue(jarFile.getEntry("TestProcessor.log").size > 0)
+        Assert.assertTrue(jarFile.getEntry("HELLO.class").size > 0)
+        Assert.assertTrue(jarFile.getEntry("com/example/AClassBuilder.class").size > 0)
+    }
+}

integration-tests/src/test/resources/playground/settings.gradle.kts

@@ -0,0 +1,19 @@
+pluginManagement {
+    val kotlinVersion: String by settings
+    val kspVersion: String by settings
+    val testRepo: String by settings
+    plugins {
+        id("com.google.devtools.ksp") version kspVersion
+        kotlin("jvm") version kotlinVersion
+    }
+    repositories {
+        maven(testRepo)
+        gradlePluginPortal()
+        google()
+    }
+}
+
+rootProject.name = "playground"
+
+include(":workload")
+include(":test-processor")

integration-tests/src/test/resources/playground/test-processor/build.gradle.kts

@@ -0,0 +1,26 @@
+val kspVersion: String by project
+val testRepo: String by project
+
+plugins {
+    kotlin("jvm")
+}
+
+group = "com.example"
+version = "1.0-SNAPSHOT"
+
+repositories {
+    maven(testRepo)
+    mavenCentral()
+    google()
+}
+
+dependencies {
+    implementation(kotlin("stdlib"))
+    implementation("com.squareup:javapoet:1.12.1")
+    implementation("com.google.devtools.ksp:symbol-processing-api:$kspVersion")
+}
+
+sourceSets.main {
+    java.srcDirs("src/main/kotlin")
+}
+

integration-tests/src/test/resources/playground/test-processor/src/main/kotlin/Builder.kt

@@ -0,0 +1,4 @@
+package com.example.annotation
+
+annotation class Builder {
+}

integration-tests/src/test/resources/playground/test-processor/src/main/kotlin/BuilderProcessor.kt

@@ -0,0 +1,83 @@
+import com.google.devtools.ksp.processing.CodeGenerator
+import com.google.devtools.ksp.processing.Dependencies
+import com.google.devtools.ksp.processing.KSPLogger
+import com.google.devtools.ksp.processing.Resolver
+import com.google.devtools.ksp.processing.SymbolProcessor
+import com.google.devtools.ksp.symbol.*
+import java.io.File
+import java.io.OutputStream
+
+fun OutputStream.appendText(str: String) {
+    this.write(str.toByteArray())
+}
+class BuilderProcessor : SymbolProcessor {
+    lateinit var codeGenerator: CodeGenerator
+
+    override fun finish() {
+
+    }
+
+    override fun init(options: Map<String, String>, kotlinVersion: KotlinVersion, codeGenerator: CodeGenerator, logger: KSPLogger) {
+        this.codeGenerator = codeGenerator
+    }
+
+    override fun process(resolver: Resolver) {
+        val symbols = resolver.getSymbolsWithAnnotation("com.example.annotation.Builder")
+        println(symbols)
+        symbols
+            .filter { it is KSClassDeclaration }
+            .map { it.accept(BuilderVisitor(), Unit) }
+    }
+
+    inner class BuilderVisitor : KSVisitorVoid() {
+        override fun visitClassDeclaration(classDeclaration: KSClassDeclaration, data: Unit) {
+            classDeclaration.primaryConstructor!!.accept(this, data)
+        }
+
+        override fun visitFunctionDeclaration(function: KSFunctionDeclaration, data: Unit) {
+            val parent = function.parentDeclaration as KSClassDeclaration
+            val packageName = parent.containingFile!!.packageName.asString()
+            val className = "${parent.simpleName.asString()}Builder"
+            val file = codeGenerator.createNewFile(Dependencies(true, function.containingFile!!), packageName , className)
+            file.appendText("package $packageName\n\n")
+            file.appendText("class $className{\n")
+            function.parameters.forEach {
+                val name = it.name!!.asString()
+                val typeName = StringBuilder(it.type.resolve().declaration.qualifiedName?.asString() ?: "<ERROR>")
+                if (typeName.toString() == "<ERROR>") {
+                    File("/tmp/BuilderLog").appendText("${(it.name as KSName).asString()}:${function.containingFile?.fileName}")
+                }
+                val typeArgs = it.type.element!!.typeArguments
+                if (it.type.element!!.typeArguments.isNotEmpty()) {
+                    typeName.append("<")
+                    typeName.append(
+                            typeArgs.map {
+                                val type = it.type?.resolve()
+                                "${it.variance.label} ${type?.declaration?.qualifiedName?.asString()}" +
+                                        if (type?.nullability == Nullability.NULLABLE) "?" else ""
+                            }.joinToString(", ")
+                    )
+                    typeName.append(">")
+                }
+//                typeName.append(if ((it.type?.type as? KSUserType)?.isNullable == true) "?" else "")
+                file.appendText("    private var $name: $typeName? = null\n")
+                file.appendText("    internal fun with${name.capitalize()}($name: $typeName): $className {\n")
+                file.appendText("        this.$name = $name\n")
+                file.appendText("        return this\n")
+                file.appendText("    }\n\n")
+            }
+            file.appendText("    internal fun build(): ${parent.qualifiedName!!.asString()} {\n")
+            file.appendText("        return ${parent.qualifiedName!!.asString()}(")
+            file.appendText(
+                function.parameters.map {
+                    "${it.name!!.asString()}!!"
+                }.joinToString(", ")
+            )
+            file.appendText(")\n")
+            file.appendText("    }\n")
+            file.appendText("}\n")
+            file.close()
+        }
+    }
+
+}