aboutsummaryrefslogtreecommitdiff
path: root/symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt
diff options
context:
space:
mode:
authorLinnea Gräf <nea@nea.moe>2024-07-20 15:10:49 +0200
committerLinnea Gräf <nea@nea.moe>2024-07-20 16:20:56 +0200
commit3ad44b992fbb6859b01a99eae1a317ea67754e4d (patch)
treedaf1d3b317d51af547c69d13fe36322de7425cca /symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt
parent9b1b1446814f971e914c034aab00c95fdae87163 (diff)
downloadFirmament-3ad44b992fbb6859b01a99eae1a317ea67754e4d.tar.gz
Firmament-3ad44b992fbb6859b01a99eae1a317ea67754e4d.tar.bz2
Firmament-3ad44b992fbb6859b01a99eae1a317ea67754e4d.zip
Make builds reproducible allowing for verifying builds
[no changelog]
Diffstat (limited to 'symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt')
-rw-r--r--symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt99
1 files changed, 99 insertions, 0 deletions
diff --git a/symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt b/symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt
new file mode 100644
index 0000000..ccfbdd7
--- /dev/null
+++ b/symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt
@@ -0,0 +1,99 @@
+/*
+ * SPDX-FileCopyrightText: 2024 Linnea Gräf <nea@nea.moe>
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+package moe.nea.firmament.annotations.process
+
+import com.google.auto.service.AutoService
+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.Origin
+import com.google.gson.Gson
+import com.google.gson.JsonArray
+import com.google.gson.JsonObject
+
+
+class MixinAnnotationProcessor(
+ val codeGenerator: CodeGenerator,
+ val logger: KSPLogger
+) : SymbolProcessor {
+ @AutoService(SymbolProcessorProvider::class)
+ class Provider : SymbolProcessorProvider {
+ override fun create(environment: SymbolProcessorEnvironment): SymbolProcessor {
+ return MixinAnnotationProcessor(environment.codeGenerator, environment.logger)
+ }
+ }
+
+ val mixinPackage = "moe.nea.firmament.mixins"
+ val refmapName = "Firmament-refmap.json"
+ val mixinPlugin = "moe.nea.firmament.init.MixinPlugin"
+ val scaffold = """
+{
+ "required": true,
+ "plugin": "moe.nea.firmament.init.MixinPlugin",
+ "package": "{mixinPackage}",
+ "compatibilityLevel": "JAVA_17",
+ "injectors": {
+ "defaultRequire": 1
+ },
+ "refmap": "{refmapName}",
+ "client": {mixins}
+}
+"""
+ var rounds = mutableListOf<KSClassDeclaration>()
+
+ override fun process(resolver: Resolver): List<KSAnnotated> {
+ return resolver.getSymbolsWithAnnotation("org.spongepowered.asm.mixin.Mixin")
+ .filter { processElement(it, resolver) }.toList()
+ }
+
+ override fun finish() {
+ val output = codeGenerator.createNewFile(
+ Dependencies(
+ aggregating = true,
+ *rounds.map { it.containingFile!! }.toTypedArray()),
+ "", "firmament.mixins",
+ extensionName = "json")
+ val writer = output.writer()
+ val gson = Gson()
+ val mixinJson = JsonObject()
+ mixinJson.addProperty("required", true)
+ mixinJson.addProperty("plugin", mixinPlugin)
+ mixinJson.addProperty("package", mixinPackage)
+ mixinJson.addProperty("compatibilityLevel", "JAVA_21")
+ mixinJson.addProperty("refmap", refmapName)
+ val mixinArray = JsonArray()
+ rounds.map { it.qualifiedName!!.asString().removePrefix("$mixinPackage.") }
+ .sorted()
+ .forEach(mixinArray::add)
+ mixinJson.add("client", mixinArray)
+ gson.toJson(mixinJson, writer)
+ writer.close()
+ rounds
+ }
+
+ private fun processElement(decl: KSAnnotated, resolver: Resolver): Boolean {
+ if (decl !is KSClassDeclaration) {
+ logger.error("@Mixin only allowed on class declarations", decl)
+ return true
+ }
+ decl.qualifiedName ?: logger.error("@Mixin only allowed on classes with a proper name")
+ if (decl.origin != Origin.JAVA) logger.error("@Mixin only allowed in java code")
+ val packageName = decl.packageName.asString()
+ if (packageName != mixinPackage && !packageName.startsWith("$mixinPackage."))
+ logger.error("@Mixin outside of mixin package", decl)
+ rounds.add(decl)
+ return true
+ }
+
+
+}