5 files changed, 394 insertions, 0 deletions
diff --git a/src/main/kotlin/features/macros/ComboProcessor.kt b/src/main/kotlin/features/macros/ComboProcessor.kt
new file mode 100644
index 0000000..5c5ac0e
--- /dev/null
+++ b/src/main/kotlin/features/macros/ComboProcessor.kt
@@ -0,0 +1,114 @@
+package moe.nea.firmament.features.macros
+
+import kotlin.time.Duration.Companion.seconds
+import net.minecraft.client.util.InputUtil
+import net.minecraft.text.Text
+import moe.nea.firmament.annotations.Subscribe
+import moe.nea.firmament.events.HudRenderEvent
+import moe.nea.firmament.events.TickEvent
+import moe.nea.firmament.events.WorldKeyboardEvent
+import moe.nea.firmament.keybindings.SavedKeyBinding
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.TimeMark
+import moe.nea.firmament.util.tr
+
+object ComboProcessor {
+
+	var rootTrie: Branch = Branch(mapOf())
+		private set
+
+	var activeTrie: Branch = rootTrie
+		private set
+
+	var isInputting = false
+	var lastInput = TimeMark.farPast()
+	val breadCrumbs = mutableListOf<SavedKeyBinding>()
+
+	init {
+		val f = SavedKeyBinding(InputUtil.GLFW_KEY_F)
+		val one = SavedKeyBinding(InputUtil.GLFW_KEY_1)
+		val two = SavedKeyBinding(InputUtil.GLFW_KEY_2)
+		setActions(
+			MacroData.DConfig.data.comboActions
+		)
+	}
+
+	fun setActions(actions: List<ComboKeyAction>) {
+		rootTrie = KeyComboTrie.fromComboList(actions)
+		reset()
+	}
+
+	fun reset() {
+		activeTrie = rootTrie
+		lastInput = TimeMark.now()
+		isInputting = false
+		breadCrumbs.clear()
+	}
+
+	@Subscribe
+	fun onTick(event: TickEvent) {
+		if (isInputting && lastInput.passedTime() > 3.seconds)
+			reset()
+	}
+
+
+	@Subscribe
+	fun onRender(event: HudRenderEvent) {
+		if (!isInputting) return
+		if (!event.isRenderingHud) return
+		event.context.matrices.push()
+		val width = 120
+		event.context.matrices.translate(
+			(MC.window.scaledWidth - width) / 2F,
+			(MC.window.scaledHeight) / 2F + 8,
+			0F
+		)
+		val breadCrumbText = breadCrumbs.joinToString(" > ")
+		event.context.drawText(
+			MC.font,
+			tr("firmament.combo.active", "Current Combo: ").append(breadCrumbText),
+			0,
+			0,
+			-1,
+			true
+		)
+		event.context.matrices.translate(0F, MC.font.fontHeight + 2F, 0F)
+		for ((key, value) in activeTrie.nodes) {
+			event.context.drawText(
+				MC.font,
+				Text.literal("$breadCrumbText > $key: ").append(value.label),
+				0,
+				0,
+				-1,
+				true
+			)
+			event.context.matrices.translate(0F, MC.font.fontHeight + 1F, 0F)
+		}
+		event.context.matrices.pop()
+	}
+
+	@Subscribe
+	fun onKeyBinding(event: WorldKeyboardEvent) {
+		val nextEntry = activeTrie.nodes.entries
+			.find { event.matches(it.key) }
+		if (nextEntry == null) {
+			reset()
+			return
+		}
+		event.cancel()
+		breadCrumbs.add(nextEntry.key)
+		lastInput = TimeMark.now()
+		isInputting = true
+		val value = nextEntry.value
+		when (value) {
+			is Branch -> {
+				activeTrie = value
+			}
+
+			is Leaf -> {
+				value.execute()
+				reset()
+			}
+		}.let { }
+	}
+}
diff --git a/src/main/kotlin/features/macros/HotkeyAction.kt b/src/main/kotlin/features/macros/HotkeyAction.kt
new file mode 100644
index 0000000..011f797
--- /dev/null
+++ b/src/main/kotlin/features/macros/HotkeyAction.kt
@@ -0,0 +1,40 @@
+package moe.nea.firmament.features.macros
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+import net.minecraft.text.Text
+import moe.nea.firmament.util.MC
+
+@Serializable
+sealed interface HotkeyAction {
+	// TODO: execute
+	val label: Text
+	fun execute()
+}
+
+@Serializable
+@SerialName("command")
+data class CommandAction(val command: String) : HotkeyAction {
+	override val label: Text
+		get() = Text.literal("/$command")
+
+	override fun execute() {
+		MC.sendCommand(command)
+	}
+}
+
+// Mit onscreen anzeige:
+// F -> 1 /equipment
+// F -> 2 /wardrobe
+// Bei Combos: Keys buffern! (für wardrobe hotkeys beispielsweiße)
+
+// Radial menu
+// Hold F
+// Weight (mach eins doppelt so groß)
+// /equipment
+// /wardrobe
+
+// Bei allen: Filter!
+// - Nur in Dungeons / andere Insel
+// - Nur wenn ich Item X im inventar habe (fishing rod)
+
diff --git a/src/main/kotlin/features/macros/KeyComboTrie.kt b/src/main/kotlin/features/macros/KeyComboTrie.kt
new file mode 100644
index 0000000..57ff289
--- /dev/null
+++ b/src/main/kotlin/features/macros/KeyComboTrie.kt
@@ -0,0 +1,68 @@
+package moe.nea.firmament.features.macros
+
+import kotlinx.serialization.Serializable
+import net.minecraft.text.Text
+import moe.nea.firmament.keybindings.SavedKeyBinding
+import moe.nea.firmament.util.ErrorUtil
+
+sealed interface KeyComboTrie {
+	val label: Text
+
+	companion object {
+		fun fromComboList(
+			combos: List<ComboKeyAction>,
+		): Branch {
+			val root = Branch(mutableMapOf())
+			for (combo in combos) {
+				var p = root
+				if (combo.keys.isEmpty()) {
+					ErrorUtil.softUserError("Key Combo for ${combo.action.label.string} is empty")
+					continue
+				}
+				for ((index, key) in combo.keys.withIndex()) {
+					val m = (p.nodes as MutableMap)
+					if (index == combo.keys.lastIndex) {
+						if (key in m) {
+							ErrorUtil.softUserError("Overlapping actions found for ${combo.keys.joinToString(" > ")} (another action ${m[key]} already exists).")
+							break
+						}
+
+						m[key] = Leaf(combo.action)
+					} else {
+						val c = m.getOrPut(key) { Branch(mutableMapOf()) }
+						if (c !is Branch) {
+							ErrorUtil.softUserError("Overlapping actions found for ${combo.keys} (final node exists at index $index) through another action already")
+							break
+						} else {
+							p = c
+						}
+					}
+				}
+			}
+			return root
+		}
+	}
+}
+
+
+@Serializable
+data class ComboKeyAction(
+	val action: HotkeyAction,
+	val keys: List<SavedKeyBinding>,
+)
+
+data class Leaf(val action: HotkeyAction) : KeyComboTrie {
+	override val label: Text
+		get() = action.label
+
+	fun execute() {
+		action.execute()
+	}
+}
+
+data class Branch(
+	val nodes: Map<SavedKeyBinding, KeyComboTrie>
+) : KeyComboTrie {
+	override val label: Text
+		get() = Text.literal("...") // TODO: better labels
+}
diff --git a/src/main/kotlin/features/macros/MacroData.kt b/src/main/kotlin/features/macros/MacroData.kt
new file mode 100644
index 0000000..78a5948
--- /dev/null
+++ b/src/main/kotlin/features/macros/MacroData.kt
@@ -0,0 +1,11 @@
+package moe.nea.firmament.features.macros
+
+import kotlinx.serialization.Serializable
+import moe.nea.firmament.util.data.DataHolder
+
+@Serializable
+data class MacroData(
+    var comboActions: List<ComboKeyAction> = listOf(),
+){
+	object DConfig : DataHolder<MacroData>(kotlinx.serialization.serializer(), "macros", ::MacroData)
+}
diff --git a/src/main/kotlin/features/macros/MacroUI.kt b/src/main/kotlin/features/macros/MacroUI.kt
new file mode 100644
index 0000000..17fdd0a
--- /dev/null
+++ b/src/main/kotlin/features/macros/MacroUI.kt
@@ -0,0 +1,161 @@
+package moe.nea.firmament.features.macros
+
+import io.github.notenoughupdates.moulconfig.gui.CloseEventListener
+import io.github.notenoughupdates.moulconfig.observer.ObservableList
+import io.github.notenoughupdates.moulconfig.xml.Bind
+import moe.nea.firmament.annotations.Subscribe
+import moe.nea.firmament.commands.thenExecute
+import moe.nea.firmament.events.CommandEvent
+import moe.nea.firmament.gui.config.AllConfigsGui.toObservableList
+import moe.nea.firmament.gui.config.KeyBindingStateManager
+import moe.nea.firmament.keybindings.SavedKeyBinding
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.MoulConfigUtils
+import moe.nea.firmament.util.ScreenUtil
+
+class MacroUI {
+
+
+	companion object {
+		@Subscribe
+		fun onCommands(event: CommandEvent.SubCommand) {
+			// TODO: add button in config
+			event.subcommand("macros") {
+				thenExecute {
+					ScreenUtil.setScreenLater(MoulConfigUtils.loadScreen("config/macros/index", MacroUI(), null))
+				}
+			}
+		}
+
+	}
+
+	@field:Bind("combos")
+	val combos = Combos()
+
+	class Combos {
+		@field:Bind("actions")
+		val actions: ObservableList<ActionEditor> = ObservableList(
+			MacroData.DConfig.data.comboActions.mapTo(mutableListOf()) {
+				ActionEditor(it, this)
+			}
+		)
+
+		var dontSave = false
+
+		@Bind
+		fun beforeClose(): CloseEventListener.CloseAction {
+			if (!dontSave)
+				save()
+			return CloseEventListener.CloseAction.NO_OBJECTIONS_TO_CLOSE
+		}
+
+		@Bind
+		fun addCommand() {
+			actions.add(
+				ActionEditor(
+					ComboKeyAction(
+						CommandAction("ac Hello from a Firmament Hotkey"),
+						listOf()
+					),
+					this
+				)
+			)
+		}
+
+		@Bind
+		fun discard() {
+			dontSave = true
+			MC.screen?.close()
+		}
+
+		@Bind
+		fun saveAndClose() {
+			save()
+			MC.screen?.close()
+		}
+
+		@Bind
+		fun save() {
+			MacroData.DConfig.data.comboActions = actions.map { it.asSaveable() }
+			MacroData.DConfig.markDirty()
+			ComboProcessor.setActions(MacroData.DConfig.data.comboActions) // TODO: automatically reload those from the config on startup
+		}
+	}
+
+	class KeyBindingEditor(var binding: SavedKeyBinding, val parent: ActionEditor) {
+		val sm = KeyBindingStateManager(
+			{ binding },
+			{ binding = it },
+			::blur,
+			::requestFocus
+		)
+
+		@field:Bind
+		val button = sm.createButton()
+
+		init {
+			sm.updateLabel()
+		}
+
+		fun blur() {
+			button.blur()
+		}
+
+		@Bind
+		fun delete() {
+			parent.combo.removeIf { it === this }
+			parent.combo.update()
+		}
+
+		fun requestFocus() {
+			button.requestFocus()
+		}
+	}
+
+	class ActionEditor(val action: ComboKeyAction, val parent: MacroUI.Combos) {
+		fun asSaveable(): ComboKeyAction {
+			return ComboKeyAction(
+				CommandAction(command),
+				combo.map { it.binding }
+			)
+		}
+
+		@field:Bind("command")
+		var command: String = (action.action as CommandAction).command
+
+		@field:Bind("combo")
+		val combo = action.keys.map { KeyBindingEditor(it, this) }.toObservableList()
+
+		@Bind
+		fun formattedCombo() =
+			combo.joinToString(" > ") { it.binding.toString() }
+
+		@Bind
+		fun addStep() {
+			combo.add(KeyBindingEditor(SavedKeyBinding.unbound(), this))
+		}
+
+		@Bind
+		fun back() {
+			MC.screen?.close()
+		}
+
+		@Bind
+		fun delete() {
+			parent.actions.removeIf { it === this }
+			parent.actions.update()
+		}
+		@Bind
+		fun edit() {
+			MC.screen = MoulConfigUtils.loadScreen("config/macros/editor", this, MC.screen)
+		}
+	}
+}
+
+private fun <T> ObservableList<T>.setAll(ts: Collection<T>) {
+	val observer = this.observer
+	this.clear()
+	this.addAll(ts)
+	this.observer = observer
+	this.update()
+}