aboutsummaryrefslogtreecommitdiff
path: root/symbols/src
diff options
context:
space:
mode:
Diffstat (limited to 'symbols/src')
-rw-r--r--symbols/src/main/kotlin/process/GameTestContainingClassProcessor.kt80
1 files changed, 80 insertions, 0 deletions
diff --git a/symbols/src/main/kotlin/process/GameTestContainingClassProcessor.kt b/symbols/src/main/kotlin/process/GameTestContainingClassProcessor.kt
new file mode 100644
index 0000000..4fcf91f
--- /dev/null
+++ b/symbols/src/main/kotlin/process/GameTestContainingClassProcessor.kt
@@ -0,0 +1,80 @@
+package moe.nea.firmament.annotations.process
+
+import com.google.auto.service.AutoService
+import com.google.devtools.ksp.containingFile
+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.processing.SymbolProcessorEnvironment
+import com.google.devtools.ksp.processing.SymbolProcessorProvider
+import com.google.devtools.ksp.symbol.KSAnnotated
+import com.google.devtools.ksp.symbol.KSClassDeclaration
+import com.google.devtools.ksp.symbol.KSFile
+import com.google.gson.Gson
+import com.google.gson.JsonArray
+import com.google.gson.JsonObject
+import java.io.OutputStreamWriter
+import java.nio.charset.StandardCharsets
+import java.util.TreeSet
+
+class GameTestContainingClassProcessor(
+ val logger: KSPLogger,
+ val codeGenerator: CodeGenerator,
+ val sourceSetName: String,
+) : SymbolProcessor {
+
+
+ @AutoService(SymbolProcessorProvider::class)
+ class Provider : SymbolProcessorProvider {
+ override fun create(environment: SymbolProcessorEnvironment): SymbolProcessor {
+ return GameTestContainingClassProcessor(
+ environment.logger,
+ environment.codeGenerator,
+ environment.options["firmament.sourceset"] ?: "main")
+ }
+ }
+
+ val allClasses: MutableSet<String> = TreeSet()
+ val allSources = mutableSetOf<KSFile>()
+
+ override fun process(resolver: Resolver): List<KSAnnotated> {
+ val annotated = resolver.getSymbolsWithAnnotation("net.minecraft.test.GameTest").toList()
+ annotated.forEach {
+ val containingClass = it.parent as KSClassDeclaration
+ allClasses.add(containingClass.qualifiedName!!.asString())
+ allSources.add(it.containingFile!!)
+ }
+ return emptyList()
+ }
+
+ fun createJson(): JsonObject {
+ return JsonObject().apply {
+ addProperty("schemaVersion", 1)
+ addProperty("id", "firmament-gametest")
+ addProperty("name", "Firmament Gametest")
+ addProperty("version", "1.0.0")
+ addProperty("environment", "*")
+ add("entrypoints", JsonObject().apply {
+ add("fabric-gametest", JsonArray().apply {
+ allClasses.forEach {
+ add(it)
+ }
+ })
+ })
+ }
+ }
+
+ override fun finish() {
+ if (allClasses.isEmpty()) return
+ val stream = codeGenerator.createNewFile(Dependencies(aggregating = true, *allSources.toTypedArray()),
+ "",
+ "fabric.mod",
+ "json")
+ val output = OutputStreamWriter(stream, StandardCharsets.UTF_8)
+ Gson().toJson(createJson(), output)
+ output.close()
+ }
+
+}