From c89b663acad487caeb15f7521be3dd14342dd4e7 Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Fri, 18 Oct 2024 00:41:25 +0200
Subject: Add slot binding

---
 src/main/kotlin/util/mc/ScreenUtil.kt | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 src/main/kotlin/util/mc/ScreenUtil.kt

(limited to 'src/main/kotlin/util/mc/ScreenUtil.kt')

diff --git a/src/main/kotlin/util/mc/ScreenUtil.kt b/src/main/kotlin/util/mc/ScreenUtil.kt
new file mode 100644
index 0000000..36feb6b
--- /dev/null
+++ b/src/main/kotlin/util/mc/ScreenUtil.kt
@@ -0,0 +1,26 @@
+package moe.nea.firmament.util.mc
+
+import net.minecraft.client.gui.screen.Screen
+import net.minecraft.client.gui.screen.ingame.HandledScreen
+import net.minecraft.entity.player.PlayerInventory
+import net.minecraft.screen.slot.Slot
+
+object ScreenUtil {
+	private var lastScreen: Screen? = null
+	private var slotsByIndex: Map<SlotIndex, Slot> = mapOf()
+
+	data class SlotIndex(val index: Int, val isPlayerInventory: Boolean)
+
+	fun Screen.getSlotsByIndex(): Map<SlotIndex, Slot> {
+		if (this !is HandledScreen<*>) return mapOf()
+		if (lastScreen === this) return slotsByIndex
+		lastScreen = this
+		slotsByIndex = this.screenHandler.slots.associate {
+			SlotIndex(it.index, it.inventory is PlayerInventory) to it
+		}
+		return slotsByIndex
+	}
+
+	fun Screen.getSlotByIndex( index: Int, isPlayerInventory: Boolean): Slot? =
+		getSlotsByIndex()[SlotIndex(index, isPlayerInventory)]
+}
-- 
cgit