stack replenish module

Author: beanbag44

src/main/kotlin/com/lambda/module/modules/player/StackReplenish.kt

@@ -0,0 +1,76 @@
+/*
+ * Copyright 2025 Lambda
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.lambda.module.modules.player
+
+import com.lambda.config.AutomationConfig.Companion.setDefaultAutomationConfig
+import com.lambda.config.applyEdits
+import com.lambda.context.SafeContext
+import com.lambda.event.events.TickEvent
+import com.lambda.event.listener.SafeListener.Companion.listen
+import com.lambda.interaction.managers.inventory.InventoryRequest.Companion.inventoryRequest
+import com.lambda.module.Module
+import com.lambda.module.tag.ModuleTag
+import com.lambda.util.item.ItemStackUtils.slotId
+import com.lambda.util.player.SlotUtils.hotbar
+import com.lambda.util.player.SlotUtils.storage
+import net.minecraft.item.ItemStack
+import net.minecraft.item.Items
+
+object StackReplenish : Module(
+	name = "StackReplenish",
+	description = "Automatically refills stacks from your inventory",
+	tag = ModuleTag.PLAYER
+) {
+	private val minStackPercent by setting("Min Stack Percentage", 30, 0..100, 1, "Minimum percentage of a complete stack before refilling", "%")
+	private val offhand by setting("Offhand", false, "Replenishes the players offhand stack")
+
+	init {
+		setDefaultAutomationConfig {
+			applyEdits {
+				hideAllGroupsExcept(inventoryConfig)
+				inventoryConfig.apply {
+					::immediateAccessOnly.edit { defaultValue(false) }
+					hide(::disposables, ::swapWithDisposables, ::providerPriority, ::storePriority)
+				}
+			}
+		}
+
+		listen<TickEvent.Pre> {
+			if (player.currentScreenHandler.cursorStack.item !== Items.AIR) return@listen
+			player.hotbar.forEach { stack -> checkReplenish(stack) }
+			if (offhand) checkReplenish(player.offHandStack)
+		}
+	}
+
+	private fun SafeContext.checkReplenish(stack: ItemStack) {
+		if (stack.count.toFloat() / stack.maxCount >= (minStackPercent.toFloat() / 100)) return
+		if (!stack.isStackable) return
+
+		player.storage.forEach { invStack ->
+			if (invStack.item !== stack.item) return@forEach
+			val invId = invStack.slotId
+			val completing = stack.count + invStack.count >= stack.maxCount
+			val tooMany = invStack.count + stack.count > stack.maxCount
+			inventoryRequest {
+				moveSlot(invId, stack.slotId)
+				if (tooMany) pickup(invId)
+			}.submit()
+			if (completing) return
+		}
+	}
+}

src/main/kotlin/com/lambda/util/item/ItemStackUtils.kt

@@ -80,7 +80,7 @@ object ItemStackUtils {
     val List<ItemStack>.copy: List<ItemStack> get() = map { it.copy() }
 
     context(safeContext: SafeContext)
-    val ItemStack.slotId get() = safeContext.player.currentScreenHandler.slots.find { it.stack.equal(this) }?.id ?: -1
+    val ItemStack.slotId get() = safeContext.player.currentScreenHandler.slots.find { it.stack === this }?.id ?: -1
 	context(safeContext: SafeContext)
 	val ItemStack.inventoryIndex get() = safeContext.player.inventory.getSlotWithStack(this)
 	context(safeContext: SafeContext)