diff options
author | Linnea Gräf <nea@nea.moe> | 2024-07-20 15:10:49 +0200 |
---|---|---|
committer | Linnea Gräf <nea@nea.moe> | 2024-07-20 16:20:56 +0200 |
commit | 3ad44b992fbb6859b01a99eae1a317ea67754e4d (patch) | |
tree | daf1d3b317d51af547c69d13fe36322de7425cca /symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt | |
parent | 9b1b1446814f971e914c034aab00c95fdae87163 (diff) | |
download | Firmament-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.kt | 99 |
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 + } + + +} |