package management

core/shapes.lua

@@ -0,0 +1,153 @@
+_G.requireInjector()
+
+local GPS    = require('gps')
+local Socket = require('socket')
+local UI     = require('ui')
+local Util   = require('util')
+
+local multishell = _ENV.multishell
+local textutils  = _G.textutils
+
+multishell.setTitle(multishell.getCurrent(), 'Shapes')
+
+local args = { ... }
+local turtleId = args[1] or error('Supply turtle ID')
+turtleId = tonumber(turtleId)
+
+local levelScript = [[
+  requireInjector(getfenv(1))
+
+  local Util  = require('util')
+
+  local s, m = turtle.run(function()
+    turtle.addFeatures('level')
+    turtle.setStatus('Leveling')
+
+    if turtle.enableGPS() then
+      local pt = Util.shallowCopy(turtle.point)
+      local s, m = pcall(function()
+        turtle.level(data.startPt, data.endPt, data.firstPt)
+      end)
+
+      turtle.pathfind(pt)
+
+      if not s and m then
+        error(m)
+      end
+    end
+  end)
+
+  if not s then
+    error(m)
+  end
+]]
+
+local data = Util.readTable('/usr/config/shapes') or { }
+
+local page = UI.Page {
+  titleBar   = UI.TitleBar { title = 'Shapes' },
+  info       = UI.Window {  x =  5, y =  3, height = 1 },
+  startCoord = UI.Button {  x =  2, y =  6, text = 'Start   ', event = 'startCoord' },
+  endCoord   = UI.Button {  x =  2, y =  8, text = 'End     ', event = 'endCoord'   },
+  supplies   = UI.Button {  x =  2, y = 10, text = 'Supplies', event = 'supplies'   },
+  first      = UI.Button {  x =  2, y = 11, text = 'First',    event = 'firstCoord' },
+  cancel     = UI.Button {  x =  2, y = -3, text = 'Abort',    event = 'cancel'     },
+  begin      = UI.Button {  x = -8, y = -3, text = 'Begin',    event = 'begin'      },
+  accelerators = { q = 'quit' },
+  notification = UI.Notification(),
+  statusBar = UI.StatusBar(),
+}
+
+function page.info:draw()
+  local function size(a, b)
+    return (math.abs(a.x - b.x) + 1) *
+           (math.abs(a.y - b.y) + 1) *
+           (math.abs(a.z - b.z) + 1)
+  end
+
+  self:clear()
+  if not data.startPt then
+    self:write(1, 1, 'Set starting corner')
+  elseif not data.endPt then
+    self:write(1, 1, 'Set ending corner')
+  else
+    self:write(1, 1, 'Blocks: ' .. size(data.startPt, data.endPt))
+  end
+end
+
+function page:getPoint()
+  local pt = GPS.getPoint()
+  if not pt then
+    self.notification:error('GPS not available')
+  end
+  return pt
+end
+
+function page:runFunction(id, script)
+--Util.writeFile('script.tmp', script)
+  self.notification:info('Connecting')
+  local fn, msg = loadstring(script, 'script')
+  if not fn then
+    self.notification:error('Error in script')
+    --debug(msg)
+    return
+  end
+
+  local socket = Socket.connect(id, 161)
+  if not socket then
+    self.notification:error('Unable to connect')
+    return
+  end
+  socket:write({ type = 'script', args = script })
+  socket:close()
+
+  self.notification:success('Sent')
+end
+
+function page:eventHandler(event)
+  if event.type == 'startCoord' then
+    data.startPt = self:getPoint()
+    if data.startPt then
+      self.statusBar:setStatus('starting corner set')
+      Util.writeTable('/usr/config/shapes', data)
+    end
+    self:draw()
+  elseif event.type == 'endCoord' then
+    data.endPt = self:getPoint()
+    if data.endPt then
+      self.statusBar:setStatus('ending corner set')
+      Util.writeTable('/usr/config/shapes', data)
+    end
+    self:draw()
+  elseif event.type == 'firstCoord' then
+    data.firstPt = self:getPoint()
+    if data.firstPt then
+      self.statusBar:setStatus('first point set')
+      Util.writeTable('/usr/config/shapes', data)
+    end
+    self:draw()
+  elseif event.type == 'supplies' then
+    data.suppliesPt = self:getPoint()
+    if data.suppliesPt then
+      self.statusBar:setStatus('supplies location set')
+      Util.writeTable('/usr/config/shapes', data)
+    end
+  elseif event.type == 'begin' then
+    if data.startPt and data.endPt then
+      local s = 'local data = ' .. textutils.serialize(data) .. levelScript
+      self:runFunction(turtleId, s)
+    else
+      self.notification:error('Corners not set')
+    end
+    self.statusBar:setStatus('')
+  elseif event.type == 'cancel' then
+    self:runFunction(turtleId, 'turtle.abort(true)')
+    self.statusBar:setStatus('')
+  else
+    return UI.Page.eventHandler(self, event)
+  end
+  return true
+end
+
+UI:setPage(page)
+UI:pullEvents()