175 lines
3.5 KiB
Lua
175 lines
3.5 KiB
Lua
local Event = require('event')
|
|
local itemDB = require('itemDB')
|
|
local Milo = require('milo')
|
|
local Socket = require('socket')
|
|
|
|
local device = _G.device
|
|
|
|
local SHIELD_SLOT = 2
|
|
|
|
local context = Milo:getContext()
|
|
|
|
local function getNameSafe(v)
|
|
local name
|
|
local s, m = pcall(function()
|
|
name = v.getName()
|
|
end)
|
|
if not s then
|
|
_G._debug(m)
|
|
end
|
|
return name
|
|
end
|
|
|
|
local function getManipulatorForUser(user)
|
|
for _,v in pairs(device) do
|
|
if v.type == 'manipulator' and v.getName and getNameSafe(v) == user then
|
|
return v
|
|
end
|
|
end
|
|
end
|
|
|
|
local function compactList(list)
|
|
local c = { }
|
|
for k,v in pairs(list) do
|
|
c[k]= table.concat({ v.has_recipe and 1 or 0, v.count, v.displayName }, ':')
|
|
end
|
|
return c
|
|
end
|
|
|
|
local function client(socket)
|
|
_G._debug('REMOTE: connection from ' .. socket.dhost)
|
|
|
|
local user = socket:read(2)
|
|
if not user then
|
|
return
|
|
end
|
|
|
|
local manipulator = getManipulatorForUser(user)
|
|
if not manipulator then
|
|
_G._debug('REMOTE: Manipulator with introspection module bound with user not found. Closing connection.')
|
|
socket:write({
|
|
msg = 'Manipulator not found'
|
|
})
|
|
socket:close()
|
|
return
|
|
end
|
|
|
|
_G._debug('REMOTE: all good')
|
|
socket:write({
|
|
data = 'ok',
|
|
})
|
|
|
|
local function makeNode(devType)
|
|
local devName = user .. ':' .. devType
|
|
local adapter = device[devName]
|
|
if adapter then
|
|
return {
|
|
adapter = adapter,
|
|
name = devName,
|
|
}
|
|
end
|
|
end
|
|
|
|
repeat
|
|
local data = socket:read()
|
|
if not data then
|
|
break
|
|
end
|
|
|
|
if data.request == 'scan' then -- full scan of all inventories
|
|
local items = Milo:mergeResources(Milo:listItems(true))
|
|
socket:write(compactList(items))
|
|
|
|
elseif data.request == 'list' then
|
|
local items = Milo:mergeResources(Milo:listItems())
|
|
socket:write(compactList(items))
|
|
|
|
elseif data.request == 'deposit' then
|
|
local function deposit()
|
|
local devType = 'inventory'
|
|
local slotNo = data.slot
|
|
if data.slot == 'shield' then
|
|
slotNo = SHIELD_SLOT
|
|
devType = 'equipment'
|
|
end
|
|
|
|
local node = makeNode(devType)
|
|
if node then
|
|
local slot = node.adapter.getItemMeta(slotNo)
|
|
if slot then
|
|
context.storage:import(node, slotNo, slot.count, slot)
|
|
end
|
|
end
|
|
end
|
|
|
|
local list = Milo:listItems()
|
|
local current = list[data.key] and list[data.key].count or 0
|
|
|
|
socket:write({
|
|
key = data.key,
|
|
current = current,
|
|
})
|
|
|
|
Milo:queueRequest({ }, deposit)
|
|
|
|
elseif data.request == 'transfer' then
|
|
local count = data.count
|
|
|
|
if count == 'stack' then
|
|
count = itemDB:getMaxCount(data.item)
|
|
elseif count == 'all' then
|
|
local item = Milo:getItem(Milo:listItems(), data.item)
|
|
count = item and item.count or 0
|
|
end
|
|
|
|
local function transfer(request)
|
|
local target = makeNode('inventory')
|
|
if target then
|
|
context.storage:export(
|
|
target,
|
|
nil,
|
|
request.requested,
|
|
data.item)
|
|
end
|
|
end
|
|
|
|
local request = Milo:makeRequest(data.item, count, transfer)
|
|
|
|
socket:write(request)
|
|
end
|
|
until not socket.connected
|
|
|
|
_G._debug('REMOTE: disconnected from ' .. socket.dhost)
|
|
end
|
|
|
|
local handler
|
|
|
|
local function listen()
|
|
if device.wireless_modem then
|
|
handler = Event.addRoutine(function()
|
|
_G._debug('REMOTE: listening on port 4242')
|
|
while true do
|
|
local socket = Socket.server(4242)
|
|
Event.addRoutine(function()
|
|
client(socket)
|
|
socket:close()
|
|
end)
|
|
end
|
|
end)
|
|
end
|
|
end
|
|
|
|
Event.on({ 'device_attach', 'device_detach' }, function(_, name)
|
|
if name == 'wireless_modem' then
|
|
if handler then
|
|
handler:terminate()
|
|
handler = nil
|
|
_debug('REMOTE: wireless modem disconnected')
|
|
else
|
|
listen()
|
|
end
|
|
end
|
|
end)
|
|
|
|
listen()
|