1 files changed, 68 insertions, 0 deletions
diff --git a/src/main/kotlin/moe/nea/firmament/features/inventory/SlotLocking.kt b/src/main/kotlin/moe/nea/firmament/features/inventory/SlotLocking.kt
new file mode 100644
index 0000000..9f75541
--- /dev/null
+++ b/src/main/kotlin/moe/nea/firmament/features/inventory/SlotLocking.kt
@@ -0,0 +1,68 @@
+package moe.nea.firmament.features.inventory
+
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.serializer
+import net.minecraft.client.gui.DrawableHelper
+import net.minecraft.entity.player.PlayerInventory
+import moe.nea.firmament.events.HandledScreenKeyPressedEvent
+import moe.nea.firmament.events.IsSlotProtectedEvent
+import moe.nea.firmament.events.SlotRenderEvents
+import moe.nea.firmament.features.FirmamentFeature
+import moe.nea.firmament.keybindings.FirmamentKeyBindings
+import moe.nea.firmament.mixins.accessor.AccessorHandledScreen
+import moe.nea.firmament.util.CommonSoundEffects
+import moe.nea.firmament.util.MC
+import moe.nea.firmament.util.data.ProfileSpecificDataHolder
+
+object SlotLocking : FirmamentFeature {
+    override val name: String
+        get() = "Slot Locking"
+    override val identifier: String
+        get() = "slot-locking"
+
+    @Serializable
+    data class Data(
+        val lockedSlots: MutableSet<Int> = mutableSetOf(),
+    )
+
+    object DConfig : ProfileSpecificDataHolder<Data>(serializer(), "locked-slots", ::Data)
+
+    val keyBinding by FirmamentKeyBindings::SLOT_LOCKING
+    val lockedSlots get() = DConfig.data?.lockedSlots
+    override fun onLoad() {
+        HandledScreenKeyPressedEvent.subscribe {
+            if (!it.matches(keyBinding)) return@subscribe
+            val inventory = MC.handledScreen ?: return@subscribe
+            inventory as AccessorHandledScreen
+
+            val slot = inventory.focusedSlot_NEU ?: return@subscribe
+            val lockedSlots = lockedSlots ?: return@subscribe
+            if (slot.inventory is PlayerInventory) {
+                if (slot.index in lockedSlots) {
+                    lockedSlots.remove(slot.index)
+                } else {
+                    lockedSlots.add(slot.index)
+                }
+                DConfig.markDirty()
+                CommonSoundEffects.playSuccess()
+            }
+        }
+        IsSlotProtectedEvent.subscribe {
+            if (it.slot.inventory is PlayerInventory && it.slot.index in (lockedSlots ?: setOf())) {
+                it.protect()
+            }
+        }
+        SlotRenderEvents.Before.subscribe {
+            if (it.slot.inventory is PlayerInventory && it.slot.index in (lockedSlots ?: setOf())) {
+                DrawableHelper.fill(
+                    it.matrices,
+                    it.slot.x,
+                    it.slot.y,
+                    it.slot.x + 16,
+                    it.slot.y + 16,
+                    0xFFFF0000.toInt()
+                )
+            }
+        }
+    }
+}