295 lines
7.2 KiB
Lua
295 lines
7.2 KiB
Lua
_G.requireInjector(_ENV)
|
|
|
|
local Event = require('event')
|
|
local Socket = require('socket')
|
|
local sync = require('sync')
|
|
local UI = require('ui')
|
|
local Util = require('util')
|
|
|
|
local colors = _G.colors
|
|
local device = _G.device
|
|
local socket
|
|
|
|
local SHIELD_SLOT = 2
|
|
|
|
local options = {
|
|
user = { arg = 'u', type = 'string',
|
|
desc = 'User name associated with bound manipulator' },
|
|
slot = { arg = 's', type = 'number',
|
|
desc = 'Optional inventory slot to use to transfer to milo' },
|
|
shield = { arg = 'e', type = 'flag',
|
|
desc = 'Use shield slot to use to transfer to milo' },
|
|
server = { arg = 'm', type = 'number',
|
|
desc = 'ID of Milo server' },
|
|
help = { arg = 'h', type = 'flag', value = false,
|
|
desc = 'Displays the options' },
|
|
}
|
|
|
|
local args = { ... }
|
|
if not Util.getOptions(options, args) then
|
|
print()
|
|
error('Invalid arguments')
|
|
end
|
|
|
|
if not options.user.value or not options.server.value then
|
|
Util.showOptions(options)
|
|
print()
|
|
error('Invalid arguments')
|
|
end
|
|
|
|
if (options.slot.value or options.shield.value) and
|
|
not (device.neuralInterface and device.neuralInterface.getInventory) then
|
|
error('Introspection module is required for transferring items')
|
|
end
|
|
|
|
local page = UI.Page {
|
|
menuBar = UI.MenuBar {
|
|
buttons = {
|
|
{ text = 'Craft', event = 'craft' },
|
|
{ text = 'Refresh', event = 'refresh', x = -9 },
|
|
},
|
|
},
|
|
grid = UI.Grid {
|
|
y = 2, ey = -2,
|
|
columns = {
|
|
{ heading = ' Qty', key = 'count' , width = 4, justify = 'right' },
|
|
{ heading = 'Name', key = 'displayName' },
|
|
},
|
|
sortColumn = 'displayName',
|
|
},
|
|
statusBar = UI.StatusBar {
|
|
filter = UI.TextEntry {
|
|
x = 1, ex = -4,
|
|
limit = 50,
|
|
shadowText = 'filter',
|
|
shadowTextColor = colors.gray,
|
|
backgroundColor = colors.cyan,
|
|
backgroundFocusColor = colors.cyan,
|
|
accelerators = {
|
|
[ 'enter' ] = 'craft',
|
|
},
|
|
},
|
|
display = UI.Button {
|
|
x = -3,
|
|
event = 'toggle_display',
|
|
value = 0,
|
|
text = 'A',
|
|
},
|
|
},
|
|
notification = UI.Notification(),
|
|
accelerators = {
|
|
r = 'refresh',
|
|
q = 'quit',
|
|
[ 'control-e' ] = 'eject',
|
|
[ 'control-r' ] = 'refresh',
|
|
[ 'control-s' ] = 'eject_stack',
|
|
[ 'control-1' ] = 'eject_1',
|
|
[ 'control-2' ] = 'eject_1',
|
|
[ 'control-3' ] = 'eject_1',
|
|
[ 'control-4' ] = 'eject_1',
|
|
[ 'control-5' ] = 'eject_1',
|
|
[ 'control-6' ] = 'eject_1',
|
|
[ 'control-7' ] = 'eject_1',
|
|
[ 'control-8' ] = 'eject_1',
|
|
[ 'control-9' ] = 'eject_1',
|
|
[ 'control-0' ] = 'eject_1',
|
|
[ 'control-m' ] = 'machines',
|
|
[ 'control-l' ] = 'resume',
|
|
},
|
|
displayMode = 0,
|
|
}
|
|
|
|
local function filterItems(t, filter, displayMode)
|
|
if filter or displayMode > 0 then
|
|
local r = { }
|
|
if filter then
|
|
filter = filter:lower()
|
|
end
|
|
for _,v in pairs(t) do
|
|
if not filter or string.find(v.lname, filter, 1, true) then
|
|
if not displayMode or
|
|
displayMode == 0 or
|
|
displayMode == 1 and v.count > 0 or
|
|
displayMode == 2 and v.has_recipe then
|
|
table.insert(r, v)
|
|
end
|
|
end
|
|
end
|
|
return r
|
|
end
|
|
return t
|
|
end
|
|
|
|
function page:sendRequest(data)
|
|
local response
|
|
|
|
sync(self, function()
|
|
local msg
|
|
for _ = 1, 2 do
|
|
if not socket or not socket.connected then
|
|
socket, msg = Socket.connect(options.server.value, 4242)
|
|
if socket then
|
|
socket:write(options.user.value)
|
|
end
|
|
end
|
|
if socket then
|
|
if socket:write(data) then
|
|
response = socket:read(2)
|
|
if response then
|
|
return
|
|
end
|
|
end
|
|
socket:close()
|
|
end
|
|
end
|
|
self.notification:error(msg or 'Failed to connect')
|
|
end)
|
|
|
|
return response
|
|
end
|
|
|
|
function page.statusBar:draw()
|
|
return UI.Window.draw(self)
|
|
end
|
|
|
|
function page.grid:getRowTextColor(row, selected)
|
|
if row.is_craftable then
|
|
return colors.yellow
|
|
end
|
|
if row.has_recipe then
|
|
return colors.cyan
|
|
end
|
|
return UI.Grid:getRowTextColor(row, selected)
|
|
end
|
|
|
|
function page.grid:getDisplayValues(row)
|
|
row = Util.shallowCopy(row)
|
|
row.count = row.count > 0 and Util.toBytes(row.count) or ''
|
|
if row.low then
|
|
row.low = Util.toBytes(row.low)
|
|
end
|
|
if row.limit then
|
|
row.limit = Util.toBytes(row.limit)
|
|
end
|
|
return row
|
|
end
|
|
|
|
function page:eventHandler(event)
|
|
if event.type == 'quit' then
|
|
UI:exitPullEvents()
|
|
|
|
elseif event.type == 'eject' then
|
|
local item = self.grid:getSelected()
|
|
if item then
|
|
local response = self:sendRequest({ request = 'transfer', item = item, count = 1 })
|
|
item.count = item.count - response.count
|
|
self.grid:draw()
|
|
end
|
|
|
|
elseif event.type == 'eject_stack' then
|
|
local item = self.grid:getSelected()
|
|
if item then
|
|
local response = self:sendRequest({ request = 'transfer', item = item, count = 64 })
|
|
item.count = item.count - response.count
|
|
self.grid:draw()
|
|
end
|
|
|
|
elseif event.type == 'refresh' then
|
|
self:refresh()
|
|
self.grid:draw()
|
|
self.statusBar.filter:focus()
|
|
|
|
elseif event.type == 'toggle_display' then
|
|
local values = {
|
|
[0] = 'A',
|
|
[1] = 'I',
|
|
[2] = 'C',
|
|
}
|
|
|
|
event.button.value = (event.button.value + 1) % 3
|
|
self.displayMode = event.button.value
|
|
event.button.text = values[event.button.value]
|
|
event.button:draw()
|
|
self:applyFilter()
|
|
self.grid:draw()
|
|
|
|
elseif event.type == 'text_change' then
|
|
self.filter = event.text
|
|
if #self.filter == 0 then
|
|
self.filter = nil
|
|
end
|
|
self:applyFilter()
|
|
self.grid:draw()
|
|
self.statusBar.filter:focus()
|
|
|
|
else
|
|
UI.Page.eventHandler(self, event)
|
|
end
|
|
return true
|
|
end
|
|
|
|
function page:enable()
|
|
self:refresh()
|
|
self:setFocus(self.statusBar.filter)
|
|
UI.Page.enable(self)
|
|
end
|
|
|
|
function page:refresh()
|
|
local items = self:sendRequest({ request = 'list' })
|
|
|
|
if items then
|
|
self.items = items
|
|
self:applyFilter()
|
|
end
|
|
end
|
|
|
|
function page:applyFilter()
|
|
local t = filterItems(self.items, self.filter, self.displayMode)
|
|
self.grid:setValues(t)
|
|
end
|
|
|
|
if options.slot.value then
|
|
debug('Transfer items initialized')
|
|
Event.onInterval(2, function()
|
|
local neural = device.neuralInterface
|
|
if neural and neural.getInventory then
|
|
local item = neural.getInventory().getItem(options.slot.value)
|
|
if item then
|
|
debug('depositing')
|
|
page:sendRequest({ request = 'deposit', slot = options.slot.value })
|
|
-- local item =
|
|
-- TODO: update count for this one item
|
|
-- page.grid:draw() page:sync()
|
|
end
|
|
else
|
|
debug('missing Introspection module')
|
|
end
|
|
end)
|
|
end
|
|
|
|
if options.slot.value or options.eslot.value then
|
|
debug('Transfer items initialized')
|
|
Event.onInterval(2, function()
|
|
local neural = device.neuralInterface
|
|
if neural and neural.getEquipment then
|
|
local item = neural.getEquipment().getItem(SHIELD_SLOT)
|
|
if item then
|
|
debug('depositing')
|
|
page:sendRequest({ request = 'deposit', slot = 'shield' })
|
|
-- local item =
|
|
-- TODO: update count for this one item
|
|
-- page.grid:draw() page:sync()
|
|
end
|
|
else
|
|
debug('missing Introspection module')
|
|
end
|
|
end)
|
|
end
|
|
|
|
UI:setPage(page)
|
|
UI:pullEvents()
|
|
|
|
if socket then
|
|
socket:close()
|
|
end
|