4 files changed, 114 insertions, 0 deletions
diff --git a/build-src/src/main/kotlin/GenerateItemIds.kt b/build-src/src/main/kotlin/GenerateItemIds.kt
new file mode 100644
index 0000000..24f2f62
--- /dev/null
+++ b/build-src/src/main/kotlin/GenerateItemIds.kt
@@ -0,0 +1,72 @@
+import com.google.gson.Gson
+import org.gradle.api.DefaultTask
+import org.gradle.api.file.DirectoryProperty
+import org.gradle.api.provider.Property
+import org.gradle.api.tasks.Input
+import org.gradle.api.tasks.InputDirectory
+import org.gradle.api.tasks.Internal
+import org.gradle.api.tasks.OutputDirectory
+import org.gradle.api.tasks.TaskAction
+import java.io.File
+
+abstract class GenerateItemIds : DefaultTask() {
+	@get: OutputDirectory
+	abstract val outputDirectory: DirectoryProperty
+
+	@get: InputDirectory
+	abstract val repoFiles: DirectoryProperty
+
+	@get: Input
+	abstract val repoHash: Property<String>
+
+	@get: Input
+	abstract val packageName: Property<String>
+
+	@get:Internal
+	val outputFile get() = outputDirectory.asFile.get().resolve(packageName.get().replace(".", "/") + "/ItemIds.java")
+
+	init {
+		repoHash.convention("unknown-repo-git-hash")
+	}
+
+	@TaskAction
+	fun generateItemIds() {
+		val nonIdName = "[^A-Z0-9_]".toRegex()
+
+		data class Item(val id: String, val file: File) {
+			val javaName get() = id.replace(nonIdName, { "__" + it.value.single().code })
+		}
+
+		val items = mutableListOf<Item>()
+		for (listFile in repoFiles.asFile.get().resolve("items").listFiles() ?: emptyArray()) {
+			listFile ?: continue
+			if (listFile.extension != "json") {
+				error("Unknown file $listFile")
+			}
+			items.add(Item(listFile.nameWithoutExtension, listFile))
+		}
+		items.sortedBy { it.id }
+		outputFile.parentFile.mkdirs()
+		val writer = outputFile.writer().buffered()
+		writer.appendLine("// @generated from " + repoHash.get())
+		writer.appendLine("package " + packageName.get() + ";")
+		writer.appendLine()
+		writer.appendLine("import moe.nea.ledger.ItemId;")
+		writer.appendLine()
+		writer.appendLine("/**")
+		writer.appendLine(" * Automatically generated {@link ItemId} list.")
+		writer.appendLine(" */")
+		writer.appendLine("@org.jspecify.annotations.NullMarked")
+		writer.appendLine("public interface ItemIds {")
+		val gson = Gson()
+		for (item in items) {
+			writer.appendLine("\t/**")
+			writer.appendLine("\t * @see <a href=${gson.toJson("https://github.com/NotEnoughUpdates/NotEnoughUpdates-REPO/blob/${repoHash.get()}/items/${item.id}.json")}>JSON definition</a>")
+			writer.appendLine("\t */")
+			writer.appendLine("\tItemId ${item.javaName} =" +
+					                  " ItemId.forName(${gson.toJson(item.id)});")
+		}
+		writer.appendLine("}")
+		writer.close()
+	}
+}
diff --git a/build-src/src/main/kotlin/RepoDownload.kt b/build-src/src/main/kotlin/RepoDownload.kt
new file mode 100644
index 0000000..182c66f
--- /dev/null
+++ b/build-src/src/main/kotlin/RepoDownload.kt
@@ -0,0 +1,41 @@
+import org.gradle.api.DefaultTask
+import org.gradle.api.file.DirectoryProperty
+import org.gradle.api.provider.Property
+import org.gradle.api.tasks.Input
+import org.gradle.api.tasks.OutputDirectory
+import org.gradle.api.tasks.TaskAction
+import java.net.URI
+import java.util.zip.ZipInputStream
+
+abstract class RepoDownload : DefaultTask() {
+	@get:Input
+	abstract val hash: Property<String>
+
+	@get:OutputDirectory
+	abstract val outputDirectory: DirectoryProperty
+
+	init {
+		outputDirectory.convention(project.layout.buildDirectory.dir("extracted-test-repo"))
+	}
+
+	@TaskAction
+	fun performDownload() {
+		val outputDir = outputDirectory.asFile.get().absoluteFile
+		outputDir.mkdirs()
+		URI("https://github.com/notEnoughUpdates/notEnoughUpdates-rEPO/archive/${hash.get()}.zip").toURL().openStream()
+			.let(::ZipInputStream)
+			.use { zipInput ->
+				while (true) {
+					val entry = zipInput.nextEntry ?: break
+					val destination = outputDir.resolve(
+						entry.name.substringAfter('/')).absoluteFile
+					require(outputDir in generateSequence(destination) { it.parentFile })
+					if (entry.isDirectory) continue
+					destination.parentFile.mkdirs()
+					destination.outputStream().use { output ->
+						zipInput.copyTo(output)
+					}
+				}
+			}
+	}
+}
+\ No newline at end of file
diff --git a/build-src/src/main/kotlin/ledger-marker.gradle.kts b/build-src/src/main/kotlin/ledger-marker.gradle.kts
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/build-src/src/main/kotlin/ledger-marker.gradle.kts
diff --git a/build-src/src/main/kotlin/ledger-repo.gradle.kts b/build-src/src/main/kotlin/ledger-repo.gradle.kts
new file mode 100644
index 0000000..1a9be63
--- /dev/null
+++ b/build-src/src/main/kotlin/ledger-repo.gradle.kts
@@ -0,0 +1 @@
+tasks.register("downloadRepo", RepoDownload::class)