392 lines
9.2 KiB
Lua
392 lines
9.2 KiB
Lua
requireInjector(getfenv(1))
|
|
|
|
local Config = require('config')
|
|
local Event = require('event')
|
|
local itemDB = require('itemDB')
|
|
local Socket = require('socket')
|
|
local Terminal = require('terminal')
|
|
local UI = require('ui')
|
|
local Util = require('util')
|
|
|
|
multishell.setTitle(multishell.getCurrent(), 'Turtles')
|
|
UI.Button.defaults.focusIndicator = ' '
|
|
UI:configure('Turtles', ...)
|
|
|
|
local config = { }
|
|
Config.load('Turtles', config)
|
|
|
|
local options = {
|
|
turtle = { arg = 'i', type = 'number', value = config.id or -1,
|
|
desc = 'Turtle ID' },
|
|
tab = { arg = 's', type = 'string', value = config.tab or 'Sel',
|
|
desc = 'Selected tab to display' },
|
|
help = { arg = 'h', type = 'flag', value = false,
|
|
desc = 'Displays the options' },
|
|
}
|
|
|
|
local SCRIPTS_PATH = 'usr/etc/scripts'
|
|
|
|
local nullTerm = Terminal.getNullTerm(term.current())
|
|
local turtles = { }
|
|
local socket
|
|
local policies = {
|
|
{ label = 'none' },
|
|
{ label = 'digOnly' },
|
|
{ label = 'attackOnly' },
|
|
{ label = 'digAttack' },
|
|
{ label = 'turtleSafe' },
|
|
}
|
|
|
|
local page = UI.Page {
|
|
coords = UI.Window {
|
|
backgroundColor = colors.black,
|
|
height = 4,
|
|
},
|
|
tabs = UI.Tabs {
|
|
x = 1, y = 5, ey = -2,
|
|
scripts = UI.Grid {
|
|
tabTitle = 'Run',
|
|
backgroundColor = UI.TabBar.defaults.selectedBackgroundColor,
|
|
columns = {
|
|
{ heading = '', key = 'label' },
|
|
},
|
|
disableHeader = true,
|
|
sortColumn = 'label',
|
|
autospace = true,
|
|
},
|
|
turtles = UI.Grid {
|
|
tabTitle = 'Select',
|
|
backgroundColor = UI.TabBar.defaults.selectedBackgroundColor,
|
|
columns = {
|
|
{ heading = 'label', key = 'label' },
|
|
{ heading = 'Dist', key = 'distance' },
|
|
{ heading = 'Status', key = 'status' },
|
|
{ heading = 'Fuel', key = 'fuel' },
|
|
},
|
|
disableHeader = true,
|
|
sortColumn = 'label',
|
|
autospace = true,
|
|
},
|
|
inventory = UI.Grid {
|
|
backgroundColor = UI.TabBar.defaults.selectedBackgroundColor,
|
|
tabTitle = 'Inv',
|
|
columns = {
|
|
{ heading = '', key = 'index', width = 2 },
|
|
{ heading = '', key = 'qty', width = 2 },
|
|
{ heading = 'Inventory', key = 'id', width = UI.term.width - 7 },
|
|
},
|
|
disableHeader = true,
|
|
sortColumn = 'index',
|
|
},
|
|
--[[
|
|
policy = UI.Grid {
|
|
tabTitle = 'Mod',
|
|
backgroundColor = UI.TabBar.defaults.selectedBackgroundColor,
|
|
columns = {
|
|
{ heading = 'label', key = 'label' },
|
|
},
|
|
values = policies,
|
|
disableHeader = true,
|
|
sortColumn = 'label',
|
|
autospace = true,
|
|
},
|
|
]]
|
|
action = UI.Window {
|
|
tabTitle = 'Action',
|
|
backgroundColor = UI.TabBar.defaults.selectedBackgroundColor,
|
|
moveUp = UI.Button {
|
|
x = 5, y = 2,
|
|
text = 'up',
|
|
fn = 'turtle.up',
|
|
},
|
|
moveDown = UI.Button {
|
|
x = 5, y = 4,
|
|
text = 'dn',
|
|
fn = 'turtle.down',
|
|
},
|
|
moveForward = UI.Button {
|
|
x = 9, y = 3,
|
|
text = 'f',
|
|
fn = 'turtle.forward',
|
|
},
|
|
moveBack = UI.Button {
|
|
x = 2, y = 3,
|
|
text = 'b',
|
|
fn = 'turtle.back',
|
|
},
|
|
turnLeft = UI.Button {
|
|
x = 2, y = 6,
|
|
text = 'lt',
|
|
fn = 'turtle.turnLeft',
|
|
},
|
|
turnRight = UI.Button {
|
|
x = 8, y = 6,
|
|
text = 'rt',
|
|
fn = 'turtle.turnRight',
|
|
},
|
|
info = UI.TextArea {
|
|
x = 2, y = 9
|
|
}
|
|
},
|
|
},
|
|
statusBar = UI.StatusBar(),
|
|
notification = UI.Notification(),
|
|
accelerators = {
|
|
q = 'quit',
|
|
},
|
|
}
|
|
|
|
function page:enable(turtle)
|
|
self.turtle = turtle
|
|
UI.Page.enable(self)
|
|
end
|
|
|
|
function page:runFunction(script, nowrap)
|
|
for i = 1, 2 do
|
|
if not socket then
|
|
socket = Socket.connect(self.turtle.id, 161)
|
|
end
|
|
|
|
if socket then
|
|
if not nowrap then
|
|
script = 'turtle.run(' .. script .. ')'
|
|
end
|
|
if socket:write({ type = 'scriptEx', args = script }) then
|
|
local t = socket:read(3)
|
|
if t then
|
|
return table.unpack(t)
|
|
end
|
|
return false, 'Socket timeout'
|
|
end
|
|
end
|
|
socket = nil
|
|
end
|
|
self.notification:error('Unable to connect')
|
|
end
|
|
|
|
function page:runScript(scriptName)
|
|
if self.turtle then
|
|
self.notification:info('Connecting')
|
|
self:sync()
|
|
|
|
local cmd = string.format('Script %d %s', self.turtle.id, scriptName)
|
|
local ot = term.redirect(nullTerm)
|
|
pcall(function() shell.run(cmd) end)
|
|
term.redirect(ot)
|
|
self.notification:success('Sent')
|
|
end
|
|
end
|
|
|
|
function page.coords:draw()
|
|
local t = self.parent.turtle
|
|
self:clear()
|
|
if t then
|
|
self:setCursorPos(2, 2)
|
|
local ind = 'GPS'
|
|
if not t.point.gps then
|
|
ind = 'REL'
|
|
end
|
|
self:print(string.format('%s : %d,%d,%d\n Fuel: %s\n',
|
|
ind, t.point.x, t.point.y, t.point.z, Util.toBytes(t.fuel)))
|
|
end
|
|
end
|
|
|
|
--[[ Inventory Tab ]]--
|
|
function page.tabs.inventory:getRowTextColor(row, selected)
|
|
if page.turtle and row.selected then
|
|
return colors.yellow
|
|
end
|
|
return UI.Grid.getRowTextColor(self, row, selected)
|
|
end
|
|
|
|
function page.tabs.inventory:draw()
|
|
local t = page.turtle
|
|
Util.clear(self.values)
|
|
if t then
|
|
for _,v in ipairs(t.inventory) do
|
|
if v.qty > 0 then
|
|
table.insert(self.values, v)
|
|
if v.index == t.slotIndex then
|
|
v.selected = true
|
|
end
|
|
if v.id then
|
|
-- local item = itemDB:get({ v.id, v.dmg })
|
|
-- if item then
|
|
-- v.id = item.displayName
|
|
-- else
|
|
-- v.id = v.id:gsub('.*:(.*)', '%1')
|
|
-- end
|
|
v.id = itemDB:getName(v)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
self:adjustWidth()
|
|
self:update()
|
|
UI.Grid.draw(self)
|
|
end
|
|
|
|
function page.tabs.inventory:eventHandler(event)
|
|
if event.type == 'grid_select' then
|
|
local fn = string.format('turtle.select(%d)', event.selected.index)
|
|
page:runFunction(fn)
|
|
else
|
|
return UI.Grid.eventHandler(self, event)
|
|
end
|
|
return true
|
|
end
|
|
|
|
function page.tabs.scripts:draw()
|
|
|
|
Util.clear(self.values)
|
|
local files = fs.list(SCRIPTS_PATH)
|
|
for _,path in pairs(files) do
|
|
table.insert(self.values, { label = path, path = fs.combine(SCRIPTS_PATH, path) })
|
|
end
|
|
self:update()
|
|
UI.Grid.draw(self)
|
|
end
|
|
|
|
function page.tabs.scripts:eventHandler(event)
|
|
if event.type == 'grid_select' then
|
|
page:runScript(event.selected.label)
|
|
else
|
|
return UI.Grid.eventHandler(self, event)
|
|
end
|
|
return true
|
|
end
|
|
|
|
function page.tabs.turtles:getDisplayValues(row)
|
|
row = Util.shallowCopy(row)
|
|
if row.fuel then
|
|
row.fuel = Util.toBytes(row.fuel)
|
|
end
|
|
if row.distance then
|
|
row.distance = Util.round(row.distance, 1)
|
|
end
|
|
return row
|
|
end
|
|
|
|
function page.tabs.turtles:draw()
|
|
Util.clear(self.values)
|
|
for _,v in pairs(network) do
|
|
if v.fuel then
|
|
table.insert(self.values, v)
|
|
end
|
|
end
|
|
self:update()
|
|
UI.Grid.draw(self)
|
|
end
|
|
|
|
function page.tabs.turtles:eventHandler(event)
|
|
if event.type == 'grid_select' then
|
|
page.turtle = event.selected
|
|
config.id = event.selected.id
|
|
Config.update('Turtles', config)
|
|
multishell.setTitle(multishell.getCurrent(), page.turtle.label)
|
|
if socket then
|
|
socket:close()
|
|
socket = nil
|
|
end
|
|
else
|
|
return UI.Grid.eventHandler(self, event)
|
|
end
|
|
return true
|
|
end
|
|
|
|
function page.statusBar:draw()
|
|
local t = self.parent.turtle
|
|
if t then
|
|
local status = string.format('%s [ %s ]', t.status, Util.round(t.distance, 2))
|
|
self:setStatus(status, true)
|
|
end
|
|
UI.StatusBar.draw(self)
|
|
end
|
|
|
|
function page.tabs.tabBar:selectTab(tabTitle)
|
|
if tabTitle then
|
|
config.tab = tabTitle
|
|
Config.update('Turtles', config)
|
|
return UI.TabBar.selectTab(self, tabTitle)
|
|
end
|
|
end
|
|
|
|
function page:showBlocks()
|
|
|
|
local script = [[
|
|
local function inspect(direction)
|
|
local s,b = turtle['inspect' .. (direction or '')]()
|
|
if not s then
|
|
return 'minecraft:air:0'
|
|
end
|
|
return string.format('%s:%d', b.name, b.metadata)
|
|
end
|
|
|
|
local bu, bf, bd = inspect('Up'), inspect(), inspect('Down')
|
|
return string.format('%s\n%s\n%s', bu, bf, bd)
|
|
]]
|
|
|
|
local s, m = self:runFunction(script, true)
|
|
self.tabs.action.info:setText(s or m)
|
|
end
|
|
|
|
function page:eventHandler(event)
|
|
if event.type == 'quit' then
|
|
UI:exitPullEvents()
|
|
elseif event.type == 'button_press' then
|
|
if event.button.fn then
|
|
self:runFunction(event.button.fn, event.button.nowrap)
|
|
self:showBlocks()
|
|
elseif event.button.script then
|
|
self:runScript(event.button.script)
|
|
end
|
|
else
|
|
return UI.Page.eventHandler(self, event)
|
|
end
|
|
return true
|
|
end
|
|
|
|
function page:enable()
|
|
UI.Page.enable(self)
|
|
-- self.tabs:activateTab(page.tabs.turtles)
|
|
end
|
|
|
|
if not Util.getOptions(options, { ... }, true) then
|
|
return
|
|
end
|
|
|
|
if options.turtle.value >= 0 then
|
|
for i = 1, 10 do
|
|
page.turtle = _G.network[options.turtle.value]
|
|
if page.turtle then
|
|
break
|
|
end
|
|
os.sleep(1)
|
|
end
|
|
end
|
|
|
|
Event.onInterval(1, function()
|
|
if page.turtle then
|
|
local t = _G.network[page.turtle.id]
|
|
page.turtle = t
|
|
page:draw()
|
|
page:sync()
|
|
end
|
|
end)
|
|
|
|
UI:setPage(page)
|
|
|
|
local lookup = {
|
|
Run = page.tabs.scripts,
|
|
Select = page.tabs.turtles,
|
|
Inv = page.tabs.inventory,
|
|
-- Mod = page.tabs.policy,
|
|
Action = page.tabs.action,
|
|
}
|
|
|
|
if lookup[options.tab.value] then
|
|
page.tabs:activateTab(lookup[options.tab.value])
|
|
end
|
|
|
|
UI:pullEvents()
|