Add option to auto rebuild resources before F3+T

2 files changed, 59 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/firmament/features/FeatureManager.kt b/src/main/kotlin/moe/nea/firmament/features/FeatureManager.kt
index 5f76c07..dfbb5ed 100644
--- a/src/main/kotlin/moe/nea/firmament/features/FeatureManager.kt
+++ b/src/main/kotlin/moe/nea/firmament/features/FeatureManager.kt
@@ -21,6 +21,7 @@ package moe.nea.firmament.features
 import kotlinx.serialization.Serializable
 import kotlinx.serialization.serializer
 import moe.nea.firmament.Firmament
+import moe.nea.firmament.features.debug.DeveloperFeatures
 import moe.nea.firmament.features.fishing.FishingWarning
 import moe.nea.firmament.features.inventory.SlotLocking
 import moe.nea.firmament.features.world.FairySouls
@@ -48,6 +49,8 @@ object FeatureManager : DataHolder<FeatureManager.Config>(serializer(), "feature
             loadFeature(FairySouls)
             loadFeature(FishingWarning)
             loadFeature(SlotLocking)
+            if (Firmament.DEBUG)
+                loadFeature(DeveloperFeatures)
             hasAutoloaded = true
         }
     }
diff --git a/src/main/kotlin/moe/nea/firmament/features/debug/DeveloperFeatures.kt b/src/main/kotlin/moe/nea/firmament/features/debug/DeveloperFeatures.kt
new file mode 100644
index 0000000..ff69cd1
--- /dev/null
+++ b/src/main/kotlin/moe/nea/firmament/features/debug/DeveloperFeatures.kt
@@ -0,0 +1,56 @@
+package moe.nea.firmament.features.debug
+
+import java.nio.file.Path
+import java.util.concurrent.CompletableFuture
+import kotlin.io.path.absolute
+import kotlin.io.path.exists
+import net.minecraft.client.MinecraftClient
+import net.minecraft.text.Text
+import moe.nea.firmament.features.FirmamentFeature
+import moe.nea.firmament.gui.config.ManagedConfig
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.TimeMark
+import moe.nea.firmament.util.iterate
+
+object DeveloperFeatures : FirmamentFeature {
+    override val name: String
+        get() = "developer"
+    override val identifier: String
+        get() = "developer"
+    override val config: TConfig
+        get() = TConfig
+    override val defaultEnabled: Boolean
+        get() = false
+
+    val gradleDir by lazy {
+        Path.of(".").absolute()
+            .iterate { it.parent }
+            .find { it.resolve("settings.gradle.kts").exists() }!!
+    }
+
+    object TConfig : ManagedConfig("developer") {
+        val autoRebuildResources by toggle("auto-rebuild") { false }
+    }
+
+    @JvmStatic
+    fun hookOnBeforeResourceReload(client: MinecraftClient): CompletableFuture<Void> {
+        val reloadFuture = if (TConfig.autoRebuildResources && isEnabled) {
+            val builder = ProcessBuilder("./gradlew", ":processResources")
+            builder.directory(gradleDir.toFile())
+            builder.inheritIO()
+            val process = builder.start()
+            MC.player?.sendMessage(Text.translatable("firmament.dev.resourcerebuild.start"))
+            val startTime = TimeMark.now()
+            process.toHandle().onExit().thenApply {
+                MC.player?.sendMessage(Text.translatable("firmament.dev.resourcerebuild.done", startTime.passedTime()))
+                Unit
+            }
+        } else {
+            CompletableFuture.completedFuture(Unit)
+        }
+        return reloadFuture.thenCompose { client.reloadResources() }
+    }
+
+    override fun onLoad() {
+    }
+}