1 files changed, 103 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..9dadb80
--- /dev/null
+++ b/src/main/kotlin/features/macros/ComboProcessor.kt
@@ -0,0 +1,103 @@
+package moe.nea.firmament.features.macros
+
+import kotlin.time.Duration.Companion.seconds
+import net.minecraft.network.chat.Component
+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>()
+
+	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.pose().pushMatrix()
+		val width = 120
+		event.context.pose().translate(
+			(MC.window.guiScaledWidth - width) / 2F,
+			(MC.window.guiScaledHeight) / 2F + 8
+		)
+		val breadCrumbText = breadCrumbs.joinToString(" > ")
+		event.context.drawString(
+			MC.font,
+			tr("firmament.combo.active", "Current Combo: ").append(breadCrumbText),
+			0,
+			0,
+			-1,
+			true
+		)
+		event.context.pose().translate(0F, MC.font.lineHeight + 2F)
+		for ((key, value) in activeTrie.nodes) {
+			event.context.drawString(
+				MC.font,
+				Component.literal("$breadCrumbText > $key: ").append(value.label),
+				0,
+				0,
+				-1,
+				true
+			)
+			event.context.pose().translate(0F, MC.font.lineHeight + 1F)
+		}
+		event.context.pose().popMatrix()
+	}
+
+	@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 { }
+	}
+}