1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import com.google.devtools.ksp.processing.*
import com.google.devtools.ksp.symbol.*
import java.io.OutputStreamWriter
class TestProcessor(
val codeGenerator: CodeGenerator,
val logger: KSPLogger
) : SymbolProcessor {
// FIXME: use getSymbolsWithAnnotation after it is fixed.
var rounds = 0
override fun process(resolver: Resolver): List<KSAnnotated> {
rounds++
logger.warn("$rounds: ${resolver.getNewFiles().toList().sortedBy { it.fileName }}")
// Would fail if "Bar.kt" isn't dirty.
val barKt = resolver.getAllFiles().single { it.fileName == "Bar.kt" }
val bazKt = resolver.getAllFiles().single { it.fileName == "Baz.kt" }
if (rounds == 1) {
codeGenerator.createNewFile(Dependencies(false), "", "Foo", "kt").use { output ->
OutputStreamWriter(output).use { writer ->
writer.write("package com.example\n\n")
writer.write("open class Foo\n")
}
}
}
if (rounds == 2) {
val fooKt = resolver.getAllFiles().single { it.fileName == "Foo.kt" }
codeGenerator.createNewFile(Dependencies(false, fooKt, barKt), "", "FooBar", "kt").use { output ->
OutputStreamWriter(output).use { writer ->
writer.write("package com.example\n\n")
writer.write("open class FooBar\n")
}
}
codeGenerator.createNewFile(Dependencies(false, fooKt, bazKt), "", "FooBaz", "kt").use { output ->
OutputStreamWriter(output).use { writer ->
writer.write("package com.example\n\n")
writer.write("open class FooBaz\n")
}
}
}
return emptyList()
}
}
class TestProcessorProvider : SymbolProcessorProvider {
override fun create(
environment: SymbolProcessorEnvironment
): SymbolProcessor {
return TestProcessor(environment.codeGenerator, environment.logger)
}
}
|