diff options
Diffstat (limited to 'symbols/src')
| -rw-r--r-- | symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt | 99 | ||||
| -rw-r--r-- | symbols/src/main/kotlin/process/SubscribeAnnotationProcessor.kt | 17 |
2 files changed, 113 insertions, 3 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 + } + + +} diff --git a/symbols/src/main/kotlin/process/SubscribeAnnotationProcessor.kt b/symbols/src/main/kotlin/process/SubscribeAnnotationProcessor.kt index 0999878..6d88b69 100644 --- a/symbols/src/main/kotlin/process/SubscribeAnnotationProcessor.kt +++ b/symbols/src/main/kotlin/process/SubscribeAnnotationProcessor.kt @@ -30,6 +30,7 @@ class SubscribeAnnotationProcessor( val codeGenerator: CodeGenerator, ) : SymbolProcessor { override fun finish() { + subscriptions.sort() val subscriptionSet = subscriptions.mapTo(mutableSetOf()) { it.parent.containingFile!! } val dependencies = Dependencies( aggregating = true, @@ -39,8 +40,7 @@ class SubscribeAnnotationProcessor( .createNewFile(dependencies, "moe.nea.firmament.annotations.generated", "AllSubscriptions") .bufferedWriter() subscriptionsFile.apply { - appendLine("// This file is @generated by SubscribeAnnotationProcessor at ${SimpleDateFormat("yyyy-MM-dd HH:mm:ss z").format( - Date())}") + appendLine("// This file is @generated by SubscribeAnnotationProcessor") appendLine("// Do not edit") for (file in subscriptionSet) { appendLine("// Dependency: ${file.filePath}") @@ -70,7 +70,18 @@ class SubscribeAnnotationProcessor( val parent: KSClassDeclaration, val child: KSFunctionDeclaration, val type: KSType, - ) + ) : Comparable<Subscription> { + override fun compareTo(other: Subscription): Int { + var compare = parent.qualifiedName!!.asString().compareTo(other.parent.qualifiedName!!.asString()) + if (compare != 0) return compare + compare = other.child.simpleName.asString().compareTo(child.simpleName.asString()) + if (compare != 0) return compare + compare = other.type.declaration.qualifiedName!!.asString() + .compareTo(type.declaration.qualifiedName!!.asString()) + if (compare != 0) return compare + return 0 + } + } val subscriptions = mutableListOf<Subscription>() |