local class = require('class') local Util = require('util') local InventoryAdapter = require('inventoryAdapter') local Peripheral = require('peripheral') local NetworkedAdapter = class() function NetworkedAdapter:init(args) local defaults = { name = 'Networked Adapter', remotes = { }, remoteDefaults = { }, } Util.merge(self, defaults) Util.merge(self, args) if not self.side or self.side == 'network' then self.modem = Peripheral.get('wired_modem') if self.modem and self.modem.getNameLocal then self.localName = self.modem.getNameLocal() for _, v in pairs(self.modem.getNamesRemote()) do local remote = Peripheral.get({ name = v }) if remote and remote.size and remote.size() >= 27 and remote.list and not (self.remoteDefaults[v] and self.remoteDefaults[v].ignore) then local adapter = InventoryAdapter.wrap({ side = v, direction = self.localName }) if adapter then table.insert(self.remotes, adapter) end end end end for _, remote in pairs(self.remotes) do Util.merge(remote, self.remoteDefaults[remote.side]) end table.sort(self.remotes, function(a, b) if not a.priority then return false elseif not b.priority then return true end return a.priority < b.priority end) end _G._p = self --------------------------------------------- DEBUG end function NetworkedAdapter:isValid() return #self.remotes > 0 end function NetworkedAdapter:refresh(throttle) return self:listItems(throttle) end -- provide a consolidated list of items function NetworkedAdapter:listItems(throttle) local cache = { } local items = { } throttle = throttle or Util.throttle() for _, remote in pairs(self.remotes) do remote:listItems(throttle) local rcache = remote.cache or { } -- TODO: add a method in each adapter that only updates a passed cache for key,v in pairs(rcache) do if v.count > 0 then local entry = cache[key] if not entry then entry = Util.shallowCopy(v) entry.count = v.count cache[key] = entry table.insert(items, entry) else entry.count = entry.count + v.count end throttle() end end end if not Util.empty(items) then self.cache = cache return items end end function NetworkedAdapter:getItemInfo(item) if not self.cache then self:listItems() end local key = table.concat({ item.name, item.damage, item.nbtHash }, ':') local items = self.cache or { } return items[key] end function NetworkedAdapter:getPercentUsed() if self.cache and self.getDrawerCount then return math.floor(Util.size(self.cache) / self.getDrawerCount() * 100) end return 0 end function NetworkedAdapter:provide(item, qty, slot, direction) local total = 0 for _, remote in ipairs(self.remotes) do debug('%s -> slot %d: %d %s', remote.side, slot or -1, qty, item.name) local amount = remote:provide(item, qty, slot, direction) qty = qty - amount total = total + amount if qty <= 0 then break end end return total end function NetworkedAdapter:extract(slot, qty, toSlot) error('extract not supported') local total = 0 for _, remote in pairs(self.remotes) do debug('extract %d slot:%d', qty, slot) local amount = remote:extract(slot, qty, toSlot) qty = qty - amount total = total + amount if qty <= 0 then break end end return total end function NetworkedAdapter:insert(slot, qty, toSlot, item) local total = 0 -- toSlot is not really valid with this adapter if toSlot then error('NetworkedAdapter: toSlot is not valid') end local key = table.concat({ item.name, item.damage, item.nbtHash }, ':') if not self.cache then self:listItems() end local function insert(remote) debug('slot %d -> %s: %s', slot, remote.side, qty) local amount = remote:insert(slot, qty, toSlot) qty = qty - amount total = total + amount end if self.cache[key] then -- is this item in some chest -- low to high priority if the chest already contains that item for _, remote in Util.rpairs(self.remotes) do if qty <= 0 then break end if remote.cache and remote.cache[key] then insert(remote) end end end -- high to low priority for _, remote in ipairs(self.remotes) do if qty <= 0 then break end insert(remote) end return total end return NetworkedAdapter