1 files changed, 46 insertions, 122 deletions
diff --git a/src/main/kotlin/gui/config/KeyBindingHandler.kt b/src/main/kotlin/gui/config/KeyBindingHandler.kt
index c389cc9..7ec7e81 100644
--- a/src/main/kotlin/gui/config/KeyBindingHandler.kt
+++ b/src/main/kotlin/gui/config/KeyBindingHandler.kt
@@ -1,5 +1,3 @@
-
-
 package moe.nea.firmament.gui.config
 
 import io.github.notenoughupdates.moulconfig.common.IMinecraft
@@ -13,137 +11,63 @@ 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)
-    }
+	ManagedConfig.OptionHandler<SavedKeyBinding> {
 
-    override fun toJson(element: SavedKeyBinding): JsonElement? {
-        return Json.encodeToJsonElement(element)
-    }
+	override fun initOption(opt: ManagedOption<SavedKeyBinding>) {
+		FirmamentKeyBindings.registerKeyBinding(name, opt)
+	}
 
-    override fun fromJson(element: JsonElement): SavedKeyBinding {
-        return Json.decodeFromJsonElement(element)
-    }
+	override fun toJson(element: SavedKeyBinding): JsonElement? {
+		return Json.encodeToJsonElement(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 fromJson(element: JsonElement): SavedKeyBinding {
+		return Json.decodeFromJsonElement(element)
+	}
 
-            override fun getBackground(context: GuiImmediateContext): NinePatch<MyResourceLocation> {
-                if (editing) return activeBg
-                return super.getBackground(context)
-            }
+	override fun emitGuiElements(opt: ManagedOption<SavedKeyBinding>, guiAppender: GuiAppender) {
+		lateinit var button: FirmButtonComponent
+		val sm = KeyBindingStateManager(
+			{ opt.value },
+			{ opt.value = it },
+			{ button.blur() },
+			{ button.requestFocus() }
+		)
+		button = object : FirmButtonComponent(
+			TextComponent(
+				IMinecraft.instance.defaultFontRenderer,
+				{ sm.label.string },
+				130,
+				TextComponent.TextAlignment.LEFT,
+				false,
+				false
+			), action = {
+				sm.onClick()
+			}) {
+			override fun keyboardEvent(event: KeyboardEvent, context: GuiImmediateContext): Boolean {
+				if (event is KeyboardEvent.KeyPressed) {
+					return sm.keyboardEvent(event.keycode, event.pressed)
+				}
+				return super.keyboardEvent(event, 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 getBackground(context: GuiImmediateContext): NinePatch<MyResourceLocation> {
+				if (sm.editing) return activeBg
+				return super.getBackground(context)
+			}
 
-            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)
-    }
+			override fun onLostFocus() {
+				sm.onLostFocus()
+			}
+		}
+		sm.updateLabel()
+		guiAppender.appendLabeledRow(opt.labelText, button)
+	}
 
 }