From c4de2248ab6628b98494eccd87cf8f3bdd0d8c42 Mon Sep 17 00:00:00 2001
From: DBotThePony <dbotthepony@yandex.ru>
Date: Sun, 16 Oct 2022 19:46:08 +0700
Subject: [PATCH] Disallow to quickly pickup items from equipment slots on
 doubleclick

---
 .../mc/otm/compat/cos/CosmeticArmorCompat.kt         |  4 ++++
 .../dbotthepony/mc/otm/compat/curios/CuriosCompat.kt | 12 ++++++++++++
 .../kotlin/ru/dbotthepony/mc/otm/menu/MatteryMenu.kt |  8 ++++++++
 src/main/kotlin/ru/dbotthepony/mc/otm/menu/Slots.kt  |  4 ++++
 4 files changed, 28 insertions(+)

diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/compat/cos/CosmeticArmorCompat.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/compat/cos/CosmeticArmorCompat.kt
index 67e487169..30d41b08a 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/compat/cos/CosmeticArmorCompat.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/compat/cos/CosmeticArmorCompat.kt
@@ -58,6 +58,10 @@ private class CosmeticSlot(container: Container, private val slot: EquipmentSlot
 		return 1
 	}
 
+	override fun canTakeItemForPickAll(): Boolean {
+		return false
+	}
+
 	override fun getNoItemIcon(): com.mojang.datafixers.util.Pair<ResourceLocation, ResourceLocation>? {
 		return when (slot) {
 			EquipmentSlot.FEET -> com.mojang.datafixers.util.Pair.of(InventoryMenu.BLOCK_ATLAS, InventoryMenu.EMPTY_ARMOR_SLOT_BOOTS)
diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/compat/curios/CuriosCompat.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/compat/curios/CuriosCompat.kt
index 168555593..ddda5ccb1 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/compat/curios/CuriosCompat.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/compat/curios/CuriosCompat.kt
@@ -113,3 +113,15 @@ fun Player.curiosAwareStream(includeCosmetics: Boolean = true): Stream<out Aware
 
 	return curiosAwareStreamImpl(includeCosmetics)
 }
+
+val Slot.isCurioSlot: Boolean get() {
+	if (!isCuriosLoaded) {
+		return false
+	}
+
+	return isCurioSlotImpl
+}
+
+private val Slot.isCurioSlotImpl: Boolean get() {
+	return this is CurioSlot || this is CosmeticCurioSlot
+}
diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/menu/MatteryMenu.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/menu/MatteryMenu.kt
index b3a675162..b125c3536 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/menu/MatteryMenu.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/menu/MatteryMenu.kt
@@ -19,6 +19,7 @@ import ru.dbotthepony.mc.otm.client.screen.panels.Dock
 import ru.dbotthepony.mc.otm.client.screen.panels.EditablePanel
 import ru.dbotthepony.mc.otm.client.screen.panels.SlotPanel
 import ru.dbotthepony.mc.otm.compat.cos.cosmeticArmorSlots
+import ru.dbotthepony.mc.otm.compat.curios.isCurioSlot
 import ru.dbotthepony.mc.otm.container.ItemFilter
 import ru.dbotthepony.mc.otm.container.ItemFilterNetworkSlot
 import ru.dbotthepony.mc.otm.menu.widget.AbstractWidget
@@ -313,6 +314,13 @@ abstract class MatteryMenu @JvmOverloads protected constructor(
 		return moveItemStackToSlots(stack, _playerInventorySlots, simulate = simulate)
 	}
 
+	override fun canTakeItemForPickAll(p_38908_: ItemStack, p_38909_: Slot): Boolean {
+		if (p_38909_ is EquipmentSlot)
+			return false
+
+		return super.canTakeItemForPickAll(p_38908_, p_38909_) && (p_38909_ !is MatterySlot || p_38909_.canTakeItemForPickAll()) && !p_38909_.isCurioSlot
+	}
+
 	override fun moveItemStackTo(
 		item: ItemStack,
 		initialSlot: Int,
diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/menu/Slots.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/menu/Slots.kt
index 16e93f35d..263564309 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/menu/Slots.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/menu/Slots.kt
@@ -20,6 +20,10 @@ open class MatterySlot @JvmOverloads constructor(container: Container, index: In
 	override fun mayPlace(itemStack: ItemStack): Boolean {
 		return super.mayPlace(itemStack) && (!ignoreSpectators || runOnClient(true) { minecraft.player?.isSpectator != true })
 	}
+
+	open fun canTakeItemForPickAll(): Boolean {
+		return false
+	}
 }
 
 open class MachineOutputSlot @JvmOverloads constructor(container: Container, index: Int, x: Int = 0, y: Int = 0) : MatterySlot(container, index, x, y) {