Make G work with open containers too

This commit is contained in:
2025-04-01 00:02:03 +02:00
parent c8eab8dc4d
commit d4b4e2c003

View File

@@ -3,6 +3,7 @@
if not CLIENT then return end
local utils = require("Cyka.utils")
local dump = require("Cyka.dump")
---@class ItemLocation
---@field inventory Barotrauma.ItemInventory
@@ -124,7 +125,9 @@ local function tryMoveItem(item, itemTree, force)
-- MyModGlobal.debugPrint(string.format("Can be put in slot %d: %s", itemLocation.slotIndex, tostring(canBePut)))
if itemLocation.maxFits > 0 and canBePut then
moved = moved or itemLocation.inventory.TryPutItem(item, itemLocation.slotIndex, false, true, Character.Controlled, true)
moved = moved or
itemLocation.inventory.TryPutItem(item, itemLocation.slotIndex, false, true, Character.Controlled,
true)
if moved then
itemLocation.maxFits = itemLocation.inventory.HowManyCanBePut(item.Prefab, itemLocation.slotIndex)
end
@@ -152,7 +155,9 @@ local function tryMoveItem(item, itemTree, force)
if maxFits > 0 then
-- MyModGlobal.debugPrint(string.format("Trying to move item to empty slot at index: %d", itemLocation.slotIndex))
moved = moved or itemLocation.inventory.TryPutItem(item, itemLocation.slotIndex, true, false, Character.Controlled, true)
moved = moved or
itemLocation.inventory.TryPutItem(item, itemLocation.slotIndex, true, false, Character.Controlled,
true)
if moved then
itemLocation.maxFits = itemLocation.inventory.HowManyCanBePut(item.Prefab, itemLocation.slotIndex)
end
@@ -225,7 +230,9 @@ end
-- Function to quickly stack items from inventory to containers
-- 6 and 7 are hands
-- 9..18 are main slots
local inventorySlotsToStack = { 6, 7, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18 }
local inventorySlotsToStack = { 6, 7, }
-- local inventorySlotsToStack = { 6, 7, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18 }
---@param character Barotrauma.Character
local function quickStackItems(character)
if not character then
MyModGlobal.debugPrint("No character found")
@@ -257,30 +264,37 @@ local function quickStackItems(character)
--DumpTable(itemTree)
local toMove = {}
for _, slotid in ipairs(inventorySlotsToStack) do
MyModGlobal.debugPrint(string.format("Processing inventory slot: %d", slotid))
local slot = inventory.slots[slotid]
if #slot.items > 0 then
local item = slot.items[1]
local tags = item.Prefab.Tags
local shouldSuss = true
for tag in tags do
if tag.value:find("tool") or tag.value:find("weapon") then
MyModGlobal.debugPrint(string.format("Item '%s' is a tool or weapon, skipping", item.Name))
shouldSuss = false
break
for item in character.HeldItems do
MyModGlobal.debugPrint(string.format("Item: %s", item.Prefab.Identifier.Value))
if item.OwnInventory then
toMove = utils.enqueueInventory(item.OwnInventory, toMove)
end
end
if shouldSuss then
local before = #toMove
toMove = utils.enqueueSlot(slot, toMove)
local after = #toMove
MyModGlobal.debugPrint(string.format("Enqueued %d items from the inventory slot %d", after - before,
slotid))
end
end
end
-- for _, slotid in ipairs(inventorySlotsToStack) do
-- MyModGlobal.debugPrint(string.format("Processing inventory slot: %d", slotid))
-- local slot = inventory.slots[slotid]
-- if #slot.items > 0 then
-- -- local item = slot.items[1]
-- -- local tags = item.Prefab.Tags
-- local shouldSuss = true
-- -- for tag in tags do
-- -- if tag.value:find("tool") or tag.value:find("weapon") then
-- -- MyModGlobal.debugPrint(string.format("Item '%s' is a tool or weapon, skipping", item.Name))
-- -- shouldSuss = false
-- -- break
-- -- end
-- -- end
-- if shouldSuss then
-- local before = #toMove
-- toMove = utils.enqueueSlot(slot, toMove)
-- local after = #toMove
-- MyModGlobal.debugPrint(string.format("Enqueued %d items from the inventory slot %d", after - before,
-- slotid))
-- end
-- end
-- end
local openContainers = utils.getOpenContainers()
for _, container in ipairs(openContainers) do
@@ -308,16 +322,8 @@ local function stackToCursor()
return
end
for _, slot in ipairs(slots) do
local item
if not slot.slot.items or #slot.slot.items == 0 then
MyModGlobal.debugPrint("No items in slot")
goto continue
end
item = slot.slot.items[1]
MyModGlobal.debugPrint(string.format("Stacking all player items to %s", item.Prefab.Identifier.Value))
utils.enqueueAllPlayerItems({}, function(ititem)
local item, slot
local function predicate(ititem)
if ititem.Prefab.Identifier.Value == item.Prefab.Identifier.Value then
if item == ititem then return false end
-- We are moving items in the predicate because we expect to only
@@ -328,7 +334,8 @@ local function stackToCursor()
-- UPDATE: OK well that was a stupid idea, it returns an error for other shit as well
-- What other shit? Wish I knew
-- So we'll use HowManyCanBePut instead...
local moved = slot.inventory.TryPutItem(ititem, slot.slotIndex - 1, false, true, Character.Controlled, true)
local moved = slot.inventory.TryPutItem(ititem, slot.slotIndex - 1, false, true, Character.Controlled,
true)
if not moved then
MyModGlobal.debugPrint(string.format("Failed to move item %s to slot %d", ititem.Name, slot
.slotIndex - 1))
@@ -341,8 +348,22 @@ local function stackToCursor()
return false, true
end
end
end)
end
-- We gotta do a little juggling...
for _, sslot in ipairs(slots) do
slot = sslot
local items
if not slot.slot.items or #slot.slot.items == 0 then
MyModGlobal.debugPrint("No items in slot")
goto continue
end
item = slot.slot.items[1]
MyModGlobal.debugPrint(string.format("Stacking all player items to %s", item.Prefab.Identifier.Value))
items = {}
utils.enqueueAllPlayerItems(items, predicate)
utils.enqueueOpenContainers(items, predicate)
::continue::
end
end
@@ -374,7 +395,8 @@ local function stackAllToCursor()
-- UPDATE: OK well that was a stupid idea, it returns an error for other shit as well
-- What other shit? Wish I knew
-- So we'll use HowManyCanBePut instead...
local moved = slot.inventory.TryPutItem(ititem, slot.slotIndex - 1, false, true, Character.Controlled, true)
local moved = slot.inventory.TryPutItem(ititem, slot.slotIndex - 1, false, true, Character.Controlled,
true)
if not moved then
MyModGlobal.debugPrint(string.format("Failed to move item %s to slot %d", ititem.Name, slot
.slotIndex - 1))