mirror of https://github.com/kepler155c/opus
104 lines
2.5 KiB
Lua
104 lines
2.5 KiB
Lua
local class = require('class')
|
|
local Logger = require('logger')
|
|
|
|
local ChestProvider = class()
|
|
|
|
function ChestProvider:init(args)
|
|
|
|
args = args or { }
|
|
|
|
self.stacks = {}
|
|
self.name = 'chest'
|
|
self.direction = args.direction or 'up'
|
|
self.wrapSide = args.wrapSide or 'bottom'
|
|
self.p = peripheral.wrap(self.wrapSide)
|
|
end
|
|
|
|
function ChestProvider:isValid()
|
|
return self.p and self.p.getAllStacks
|
|
end
|
|
|
|
function ChestProvider:refresh()
|
|
if self.p then
|
|
self.p.condenseItems()
|
|
self.stacks = self.p.getAllStacks(false)
|
|
local t = { }
|
|
for _,s in ipairs(self.stacks) do
|
|
local key = s.id .. ':' .. s.dmg
|
|
if t[key] and t[key].qty < 64 then
|
|
t[key].max_size = t[key].qty
|
|
else
|
|
t[key] = {
|
|
qty = s.qty
|
|
}
|
|
end
|
|
end
|
|
for _,s in ipairs(self.stacks) do
|
|
local key = s.id .. ':' .. s.dmg
|
|
if t[key].max_size then
|
|
s.max_size = t[key].qty
|
|
else
|
|
s.max_size = 64
|
|
end
|
|
end
|
|
end
|
|
return self.stacks
|
|
end
|
|
|
|
function ChestProvider:getItemInfo(id, dmg)
|
|
local item = { id = id, dmg = dmg, qty = 0, max_size = 64 }
|
|
for _,stack in pairs(self.stacks) do
|
|
if stack.id == id and stack.dmg == dmg then
|
|
item.name = stack.display_name
|
|
item.qty = item.qty + stack.qty
|
|
item.max_size = stack.max_size
|
|
end
|
|
end
|
|
return item
|
|
end
|
|
|
|
function ChestProvider:craft(id, dmg, qty)
|
|
return false
|
|
end
|
|
|
|
function ChestProvider:craftItems(items)
|
|
end
|
|
|
|
function ChestProvider:provide(item, qty, slot)
|
|
if self.p then
|
|
self.stacks = self.p.getAllStacks(false)
|
|
for key,stack in pairs(self.stacks) do
|
|
if stack.id == item.id and stack.dmg == item.dmg then
|
|
local amount = math.min(qty, stack.qty)
|
|
self.p.pushItemIntoSlot(self.direction, key, amount, slot)
|
|
qty = qty - amount
|
|
if qty <= 0 then
|
|
break
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function ChestProvider:insert(slot, qty)
|
|
if self.p then
|
|
local s, m = pcall(function() self.p.pullItem(self.direction, slot, qty) end)
|
|
if not s and m then
|
|
print('chestProvider:pullItem')
|
|
print(m)
|
|
Logger.log('chestProvider', 'Insert failed, trying again')
|
|
sleep(1)
|
|
s, m = pcall(function() self.p.pullItem(self.direction, slot, qty) end)
|
|
if not s and m then
|
|
print('chestProvider:pullItem')
|
|
print(m)
|
|
Logger.log('chestProvider', 'Insert failed again')
|
|
else
|
|
Logger.log('chestProvider', 'Insert successful')
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
return ChestProvider
|