aboutsummaryrefslogtreecommitdiff
path: root/symbols/src/main/kotlin
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
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')
-rw-r--r--symbols/src/main/kotlin/process/MixinAnnotationProcessor.kt99
-rw-r--r--symbols/src/main/kotlin/process/SubscribeAnnotationProcessor.kt17
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>()