wyvern/IO.lua

51 lines
1.6 KiB
Lua

local d = require "luadash"
local w = require "lib"
local conf = w.load_config({
"chest",
"items",
"sleep_time"
}, {
sleep_time = 1,
items = {}
})
w.init()
local chest = peripheral.wrap(conf.chest)
local function get_num_stacks(total_items)
return math.ceil(total_items / 64)
end
while true do
local stacks_stored = d.map(chest.list(), w.to_wyvern_item)
local items_stored = w.collate_stacks(stacks_stored)
for item_name, quantity_desired in pairs(conf.items) do
local quantity_stocked = items_stored[item_name].count or 0
if quantity_desired > quantity_stocked then -- if we have fewer items than are desired, extract some from store
local request = w.string_to_item(item_name)
request.type = "extract"
request.destination_inventory = conf.chest
local result = w.unwrap(w.query_by_type("storage", request), "extracting items")
print("Moved", result.moved, item_name, "from storage.")
end
end
for slot, item in pairs(stacks_stored) do
local ii = w.get_internal_identifier(item)
local stored = (items_stored[ii] or {count = 0}).count
local wanted = conf.items[ii] or 0
if (get_num_stacks(stored) * 64) >= wanted then -- if item is not in want list or we have too many, send it back to storage
local result = w.unwrap(w.query_by_type("storage", {
type = "insert",
from_inventory = conf.chest,
from_slot = slot
}), "inserting items")
print("Moved", result.moved, ii, "to storage.")
end
end
sleep(conf.sleep_time)
end