1 files changed, 149 insertions, 0 deletions
diff --git a/src/main/kotlin/gui/config/KeyBindingHandler.kt b/src/main/kotlin/gui/config/KeyBindingHandler.kt
new file mode 100644
index 0000000..c389cc9
--- /dev/null
+++ b/src/main/kotlin/gui/config/KeyBindingHandler.kt
@@ -0,0 +1,149 @@
+
+
+package moe.nea.firmament.gui.config
+
+import io.github.notenoughupdates.moulconfig.common.IMinecraft
+import io.github.notenoughupdates.moulconfig.common.MyResourceLocation
+import io.github.notenoughupdates.moulconfig.deps.libninepatch.NinePatch
+import io.github.notenoughupdates.moulconfig.gui.GuiImmediateContext
+import io.github.notenoughupdates.moulconfig.gui.KeyboardEvent
+import io.github.notenoughupdates.moulconfig.gui.component.TextComponent
+import org.lwjgl.glfw.GLFW
+import kotlinx.serialization.json.Json
+import kotlinx.serialization.json.JsonElement
+import kotlinx.serialization.json.decodeFromJsonElement
+import kotlinx.serialization.json.encodeToJsonElement
+import net.minecraft.text.Text
+import net.minecraft.util.Formatting
+import moe.nea.firmament.gui.FirmButtonComponent
+import moe.nea.firmament.keybindings.FirmamentKeyBindings
+import moe.nea.firmament.keybindings.SavedKeyBinding
+
+class KeyBindingHandler(val name: String, val managedConfig: ManagedConfig) :
+    ManagedConfig.OptionHandler<SavedKeyBinding> {
+
+    override fun initOption(opt: ManagedOption<SavedKeyBinding>) {
+        FirmamentKeyBindings.registerKeyBinding(name, opt)
+    }
+
+    override fun toJson(element: SavedKeyBinding): JsonElement? {
+        return Json.encodeToJsonElement(element)
+    }
+
+    override fun fromJson(element: JsonElement): SavedKeyBinding {
+        return Json.decodeFromJsonElement(element)
+    }
+
+    override fun emitGuiElements(opt: ManagedOption<SavedKeyBinding>, guiAppender: GuiAppender) {
+        var editing = false
+        var lastPressed = 0
+        var lastPressedNonModifier = 0
+        var label: String = ""
+        var button: FirmButtonComponent? = null
+        fun updateLabel() {
+            var stroke = opt.value.format()
+            if (editing) {
+                stroke = Text.literal("")
+                val (shift, alt, ctrl) = SavedKeyBinding.getMods(SavedKeyBinding.getModInt())
+                if (shift) {
+                    stroke.append("SHIFT + ")
+                }
+                if (alt) {
+                    stroke.append("ALT + ")
+                }
+                if (ctrl) {
+                    stroke.append("CTRL + ")
+                }
+                stroke.append("???")
+                stroke.styled { it.withColor(Formatting.YELLOW) }
+            }
+            label = (stroke).string
+            managedConfig.save()
+        }
+        button = object : FirmButtonComponent(
+            TextComponent(
+                IMinecraft.instance.defaultFontRenderer,
+                { label },
+                130,
+                TextComponent.TextAlignment.LEFT,
+                false,
+                false
+            ), action = {
+                if (editing) {
+                    button!!.blur()
+                } else {
+                    editing = true
+                    button!!.requestFocus()
+                    updateLabel()
+                }
+            }) {
+            override fun keyboardEvent(event: KeyboardEvent, context: GuiImmediateContext): Boolean {
+                if (event is KeyboardEvent.KeyPressed) {
+                    return if (event.pressed) onKeyPressed(event.keycode, SavedKeyBinding.getModInt())
+                    else onKeyReleased(event.keycode, SavedKeyBinding.getModInt())
+                }
+                return super.keyboardEvent(event, context)
+            }
+
+            override fun getBackground(context: GuiImmediateContext): NinePatch<MyResourceLocation> {
+                if (editing) return activeBg
+                return super.getBackground(context)
+            }
+
+            fun onKeyPressed(ch: Int, modifiers: Int): Boolean {
+                if (!editing) {
+                    return false
+                }
+                if (ch == GLFW.GLFW_KEY_ESCAPE) {
+                    lastPressedNonModifier = 0
+                    editing = false
+                    lastPressed = 0
+                    opt.value = SavedKeyBinding(GLFW.GLFW_KEY_UNKNOWN)
+                    updateLabel()
+                    blur()
+                    return true
+                }
+                if (ch == GLFW.GLFW_KEY_LEFT_SHIFT || ch == GLFW.GLFW_KEY_RIGHT_SHIFT
+                    || ch == GLFW.GLFW_KEY_LEFT_ALT || ch == GLFW.GLFW_KEY_RIGHT_ALT
+                    || ch == GLFW.GLFW_KEY_LEFT_CONTROL || ch == GLFW.GLFW_KEY_RIGHT_CONTROL
+                ) {
+                    lastPressed = ch
+                } else {
+                    opt.value = SavedKeyBinding(
+                        ch, modifiers
+                    )
+                    editing = false
+                    blur()
+                    lastPressed = 0
+                    lastPressedNonModifier = 0
+                }
+                updateLabel()
+                return true
+            }
+
+            override fun onLostFocus() {
+                lastPressedNonModifier = 0
+                editing = false
+                lastPressed = 0
+                updateLabel()
+            }
+
+            fun onKeyReleased(ch: Int, modifiers: Int): Boolean {
+                if (!editing)
+                    return false
+                if (lastPressedNonModifier == ch || (lastPressedNonModifier == 0 && ch == lastPressed)) {
+                    opt.value = SavedKeyBinding(ch, modifiers)
+                    editing = false
+                    blur()
+                    lastPressed = 0
+                    lastPressedNonModifier = 0
+                }
+                updateLabel()
+                return true
+            }
+        }
+        updateLabel()
+        guiAppender.appendLabeledRow(opt.labelText, button)
+    }
+
+}