Factor out fabricator stack
This commit is contained in:
@@ -182,178 +182,6 @@ LuaUserData.MakeMethodAccessible(Descriptors["Barotrauma.CargoManager"], "GetCon
|
||||
----------------------------------------------------------------------------------------------------
|
||||
----------------------------------------------------------------------------------------------------
|
||||
|
||||
---@return {item: Barotrauma.Item, fabricator: Barotrauma.FabricatorComponent}, string?
|
||||
local function getOpenFabricator()
|
||||
-- Get the controlled character
|
||||
local controlledCharacter = Character.Controlled
|
||||
if not controlledCharacter then return nil, "No controlled character found" end
|
||||
|
||||
-- Check if the character has a selected item
|
||||
local selectedItem = controlledCharacter.SelectedItem
|
||||
if not selectedItem then return nil, "No selected item found" end
|
||||
|
||||
-- Check if the selected item has a Fabricator component
|
||||
local fabricator = Game.GetFabricatorComponent(selectedItem)
|
||||
if not fabricator then return nil, "No fabricator component found" end
|
||||
|
||||
return {
|
||||
item = selectedItem,
|
||||
fabricator = fabricator
|
||||
}
|
||||
end
|
||||
|
||||
--- Recipes can have multiple inputs, for example ammo
|
||||
--- Can be made either out of copper iron or steel, 1 of either
|
||||
---@class RecipeInfo
|
||||
---@field targetItem {identifier: string, name: string, amount: number}
|
||||
---@field requiredItems {amount: number, minCondition: number, maxCondition: number, prefabs: string[]}[]
|
||||
|
||||
---@param fabricator Barotrauma.FabricatorComponent
|
||||
---@return RecipeInfo, string?
|
||||
local function getSelectedRecipeRequirements(fabricator)
|
||||
-- local openFabricator, err = getOpenFabricator()
|
||||
-- if err then return nil, err end
|
||||
-- local fabricator = openFabricator.fabricator
|
||||
|
||||
local selectedRecipe = fabricator.SelectedItem
|
||||
if not selectedRecipe then return nil, "No selected recipe found" end
|
||||
|
||||
local requiredItems = {}
|
||||
for requiredItem in selectedRecipe.RequiredItems do
|
||||
local itemInfo = {
|
||||
amount = tonumber(requiredItem.Amount),
|
||||
minCondition = tonumber(requiredItem.MinCondition),
|
||||
maxCondition = tonumber(requiredItem.MaxCondition),
|
||||
prefabs = {}
|
||||
}
|
||||
|
||||
for prefab in requiredItem.ItemPrefabs do
|
||||
itemInfo.prefabs[#itemInfo.prefabs + 1] = prefab.Identifier.Value
|
||||
end
|
||||
|
||||
requiredItems[#requiredItems + 1] = itemInfo
|
||||
end
|
||||
|
||||
return {
|
||||
targetItem = {
|
||||
identifier = selectedRecipe.TargetItem.Identifier,
|
||||
name = selectedRecipe.TargetItem.Name,
|
||||
amount = selectedRecipe.Amount
|
||||
},
|
||||
requiredItems = requiredItems
|
||||
}
|
||||
end
|
||||
|
||||
-- Hook into player control to listen for key press
|
||||
Hook.Patch("Barotrauma.Character", "ControlLocalPlayer", function(instance, ptable)
|
||||
if not PlayerInput.KeyHit(CONFIG.FABRICATOR_KEY) then return end
|
||||
|
||||
-- TODO: Maybe get items from entire sub...?
|
||||
-- There's no point getting recipes if we don't have all of this bullshit
|
||||
---@type Barotrauma.Character
|
||||
local character = instance
|
||||
if not character then
|
||||
debugPrint("Character instance is nil.")
|
||||
return
|
||||
end
|
||||
---@type Barotrauma.CharacterInventory
|
||||
local inventory = character.Inventory
|
||||
if not inventory then
|
||||
debugPrint("Character inventory is nil.")
|
||||
return
|
||||
end
|
||||
---@type Barotrauma.ItemInventory.Slot
|
||||
local bagSlot = inventory.slots[BAG_SLOT]
|
||||
if not bagSlot then
|
||||
debugPrint("Bag slot not found.")
|
||||
return
|
||||
end
|
||||
if #bagSlot.items == 0 then
|
||||
debugPrint("Bag slot is empty.")
|
||||
return
|
||||
end
|
||||
---@type Barotrauma.Item
|
||||
local bagItem = bagSlot.items[1]
|
||||
if not bagItem then
|
||||
debugPrint("Bag item not found.")
|
||||
return
|
||||
end
|
||||
|
||||
local fabricator, err = getOpenFabricator()
|
||||
if err then
|
||||
print(string.format("Error getting open fabricator: %s", err))
|
||||
return
|
||||
end
|
||||
|
||||
local recipe
|
||||
recipe, err = getSelectedRecipeRequirements(fabricator.fabricator)
|
||||
if err then
|
||||
print(string.format("Error getting selected recipe requirements: %s", err))
|
||||
return
|
||||
end
|
||||
-- DumpTable(recipe)
|
||||
|
||||
-- TODO: Maybe make it so every press cycles the input
|
||||
-- For recipes that have multiple prefabs
|
||||
-- But then again what if it has 3 items with 4 prefabs each..
|
||||
-- Is that 4 iterations or 3*4 iterations?
|
||||
-- We can not use #toFind because we can remove 0th item
|
||||
-- Which means it's no longer contiguous
|
||||
-- Which means #toFind returns 0
|
||||
local toFind = recipe.requiredItems
|
||||
local remaining = #toFind
|
||||
---@type Barotrauma.Item[]
|
||||
local toGet = {}
|
||||
---@type fun(item: Barotrauma.Item): boolean, boolean
|
||||
local filter = function(item)
|
||||
local found = false
|
||||
-- DumpTable(toFind)
|
||||
-- toFind are all items we need to find
|
||||
for i, itemInfo in pairs(toFind) do
|
||||
-- prefabs are all items that satisfy the requirements
|
||||
for _, prefab in ipairs(itemInfo.prefabs) do
|
||||
-- debugPrint(string.format("Checking %s against %s", item.Prefab.Identifier.Value, prefab))
|
||||
if item.Prefab.Identifier.Value == prefab then
|
||||
-- debugPrint(string.format("That'll do %s %s", item.Prefab.Identifier.Value, prefab))
|
||||
toGet[#toGet + 1] = item
|
||||
itemInfo.amount = itemInfo.amount - 1
|
||||
found = true
|
||||
break
|
||||
end
|
||||
end
|
||||
if itemInfo.amount <= 0 then
|
||||
-- debugPrint(string.format("Removing %s from toFind", itemInfo.prefabs[1]))
|
||||
toFind[i] = nil
|
||||
remaining = remaining - 1
|
||||
end
|
||||
if found then break end
|
||||
end
|
||||
-- DumpTable(toFind)
|
||||
-- debugPrint(string.format("Found %s %s", item.Prefab.Identifier.Value, tostring(remaining)))
|
||||
return found, remaining == 0
|
||||
end
|
||||
-- DumpTable(toGet)
|
||||
|
||||
local items = enqueueInventory(bagItem.OwnInventory, {}, filter)
|
||||
-- DumpTable(items)
|
||||
-- TODO: This might explode... Oh well?
|
||||
local inputInventory = fabricator.item.OwnInventories[1]
|
||||
for iinventory in fabricator.item.OwnInventories do
|
||||
if #iinventory.slots > 1 then
|
||||
inputInventory = iinventory
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
local slot = -1
|
||||
local previous = nil
|
||||
for _, item in ipairs(items) do
|
||||
if previous ~= item.Prefab.Identifier then slot = slot + 1 end
|
||||
inputInventory.TryPutItem(item, slot, false, true, nil)
|
||||
previous = item.Prefab.Identifier
|
||||
end
|
||||
DumpTable(items)
|
||||
end, Hook.HookMethodType.After)
|
||||
|
||||
----------------------------------------------------------------------------------------------------
|
||||
----------------------------------------------------------------------------------------------------
|
||||
|
||||
174
QuickStackToBag/Lua/Cyka/fabricatorstack.lua
Normal file
174
QuickStackToBag/Lua/Cyka/fabricatorstack.lua
Normal file
@@ -0,0 +1,174 @@
|
||||
local utils = require("Cyka.utils")
|
||||
|
||||
---@return {item: Barotrauma.Item, fabricator: Barotrauma.FabricatorComponent}, string?
|
||||
local function getOpenFabricator()
|
||||
-- Get the controlled character
|
||||
local controlledCharacter = Character.Controlled
|
||||
if not controlledCharacter then return nil, "No controlled character found" end
|
||||
|
||||
-- Check if the character has a selected item
|
||||
local selectedItem = controlledCharacter.SelectedItem
|
||||
if not selectedItem then return nil, "No selected item found" end
|
||||
|
||||
-- Check if the selected item has a Fabricator component
|
||||
local fabricator = Game.GetFabricatorComponent(selectedItem)
|
||||
if not fabricator then return nil, "No fabricator component found" end
|
||||
|
||||
return {
|
||||
item = selectedItem,
|
||||
fabricator = fabricator
|
||||
}
|
||||
end
|
||||
|
||||
--- Recipes can have multiple inputs, for example ammo
|
||||
--- Can be made either out of copper iron or steel, 1 of either
|
||||
---@class RecipeInfo
|
||||
---@field targetItem {identifier: string, name: string, amount: number}
|
||||
---@field requiredItems {amount: number, minCondition: number, maxCondition: number, prefabs: string[]}[]
|
||||
|
||||
---@param fabricator Barotrauma.FabricatorComponent
|
||||
---@return RecipeInfo, string?
|
||||
local function getSelectedRecipeRequirements(fabricator)
|
||||
-- local openFabricator, err = getOpenFabricator()
|
||||
-- if err then return nil, err end
|
||||
-- local fabricator = openFabricator.fabricator
|
||||
|
||||
local selectedRecipe = fabricator.SelectedItem
|
||||
if not selectedRecipe then return nil, "No selected recipe found" end
|
||||
|
||||
local requiredItems = {}
|
||||
for requiredItem in selectedRecipe.RequiredItems do
|
||||
local itemInfo = {
|
||||
amount = tonumber(requiredItem.Amount),
|
||||
minCondition = tonumber(requiredItem.MinCondition),
|
||||
maxCondition = tonumber(requiredItem.MaxCondition),
|
||||
prefabs = {}
|
||||
}
|
||||
|
||||
for prefab in requiredItem.ItemPrefabs do
|
||||
itemInfo.prefabs[#itemInfo.prefabs + 1] = prefab.Identifier.Value
|
||||
end
|
||||
|
||||
requiredItems[#requiredItems + 1] = itemInfo
|
||||
end
|
||||
|
||||
return {
|
||||
targetItem = {
|
||||
identifier = selectedRecipe.TargetItem.Identifier,
|
||||
name = selectedRecipe.TargetItem.Name,
|
||||
amount = selectedRecipe.Amount
|
||||
},
|
||||
requiredItems = requiredItems
|
||||
}
|
||||
end
|
||||
|
||||
-- Hook into player control to listen for key press
|
||||
Hook.Patch("Barotrauma.Character", "ControlLocalPlayer", function(instance, ptable)
|
||||
if not PlayerInput.KeyHit(MyModGlobal.CONFIG.FABRICATOR_KEY) then return end
|
||||
|
||||
-- TODO: Maybe get items from entire sub...?
|
||||
-- There's no point getting recipes if we don't have all of this bullshit
|
||||
---@type Barotrauma.Character
|
||||
local character = instance
|
||||
if not character then
|
||||
MyModGlobal.debugPrint("Character instance is nil.")
|
||||
return
|
||||
end
|
||||
---@type Barotrauma.CharacterInventory
|
||||
local inventory = character.Inventory
|
||||
if not inventory then
|
||||
MyModGlobal.debugPrint("Character inventory is nil.")
|
||||
return
|
||||
end
|
||||
---@type Barotrauma.ItemInventory.Slot
|
||||
local bagSlot = inventory.slots[BAG_SLOT]
|
||||
if not bagSlot then
|
||||
MyModGlobal.debugPrint("Bag slot not found.")
|
||||
return
|
||||
end
|
||||
if #bagSlot.items == 0 then
|
||||
MyModGlobal.debugPrint("Bag slot is empty.")
|
||||
return
|
||||
end
|
||||
---@type Barotrauma.Item
|
||||
local bagItem = bagSlot.items[1]
|
||||
if not bagItem then
|
||||
MyModGlobal.debugPrint("Bag item not found.")
|
||||
return
|
||||
end
|
||||
|
||||
local fabricator, err = getOpenFabricator()
|
||||
if err then
|
||||
print(string.format("Error getting open fabricator: %s", err))
|
||||
return
|
||||
end
|
||||
|
||||
local recipe
|
||||
recipe, err = getSelectedRecipeRequirements(fabricator.fabricator)
|
||||
if err then
|
||||
print(string.format("Error getting selected recipe requirements: %s", err))
|
||||
return
|
||||
end
|
||||
-- MyModGlobal.DumpTable(recipe)
|
||||
|
||||
-- TODO: Maybe make it so every press cycles the input
|
||||
-- For recipes that have multiple prefabs
|
||||
-- But then again what if it has 3 items with 4 prefabs each..
|
||||
-- Is that 4 iterations or 3*4 iterations?
|
||||
-- We can not use #toFind because we can remove 0th item
|
||||
-- Which means it's no longer contiguous
|
||||
-- Which means #toFind returns 0
|
||||
local toFind = recipe.requiredItems
|
||||
local remaining = #toFind
|
||||
---@type Barotrauma.Item[]
|
||||
local toGet = {}
|
||||
---@type fun(item: Barotrauma.Item): boolean, boolean
|
||||
local filter = function(item)
|
||||
local found = false
|
||||
-- MyModGlobal.DumpTable(toFind)
|
||||
-- toFind are all items we need to find
|
||||
for i, itemInfo in pairs(toFind) do
|
||||
-- prefabs are all items that satisfy the requirements
|
||||
for _, prefab in ipairs(itemInfo.prefabs) do
|
||||
-- MyModGlobal.debugPrint(string.format("Checking %s against %s", item.Prefab.Identifier.Value, prefab))
|
||||
if item.Prefab.Identifier.Value == prefab then
|
||||
-- MyModGlobal.debugPrint(string.format("That'll do %s %s", item.Prefab.Identifier.Value, prefab))
|
||||
toGet[#toGet + 1] = item
|
||||
itemInfo.amount = itemInfo.amount - 1
|
||||
found = true
|
||||
break
|
||||
end
|
||||
end
|
||||
if itemInfo.amount <= 0 then
|
||||
-- MyModGlobal.debugPrint(string.format("Removing %s from toFind", itemInfo.prefabs[1]))
|
||||
toFind[i] = nil
|
||||
remaining = remaining - 1
|
||||
end
|
||||
if found then break end
|
||||
end
|
||||
-- MyModGlobal.DumpTable(toFind)
|
||||
-- MyModGlobal.debugPrint(string.format("Found %s %s", item.Prefab.Identifier.Value, tostring(remaining)))
|
||||
return found, remaining == 0
|
||||
end
|
||||
-- MyModGlobal.DumpTable(toGet)
|
||||
|
||||
local items = utils.enqueueInventory(bagItem.OwnInventory, {}, filter)
|
||||
-- MyModGlobal.DumpTable(items)
|
||||
-- TODO: This might explode... Oh well?
|
||||
local inputInventory = fabricator.item.OwnInventories[1]
|
||||
for iinventory in fabricator.item.OwnInventories do
|
||||
if #iinventory.slots > 1 then
|
||||
inputInventory = iinventory
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
local slot = -1
|
||||
local previous = nil
|
||||
for _, item in ipairs(items) do
|
||||
if previous ~= item.Prefab.Identifier then slot = slot + 1 end
|
||||
inputInventory.TryPutItem(item, slot, false, true, nil)
|
||||
previous = item.Prefab.Identifier
|
||||
end
|
||||
MyModGlobal.DumpTable(items)
|
||||
end, Hook.HookMethodType.After)
|
||||
Reference in New Issue
Block a user