[CC][1.0] ControlMe (Pocket Computer)

--[[
HPWebcamAble presents...
ControlMe

--=== Description ===--
This program lets you control a turtle from a pocket computer!

This version of the program is for the POCKET COMPUTER
Get the Turtle version here: pastebin.com/zjn3E5LS


--=== Installation ===--
Pastebin Code: mTGccYbM

To download a file from pastebin, run this command in a computer:
pastebin get <code> <file name>


--=== Update History ===--
The pastebin will always have the most recent version

|1.0|
-Release
]]

--=== Variables ===--
local args = {...}
local protocalName = "controlme"
local turtleID
local w,h = term.getSize()
local tools = {}
local knownTools = {
  ["modem"] = {text = "Modem", upAndDown = false},
  ["minecraft:diamond_pickaxe"] = {text = "Mine", upAndDown = true, action = "turtle.dig"},
  ["minecraft:diamond_sword"] = {text = "Attack", upAndDown = true, action = "turtle.attack"},
  ["minecraft:diamond_hoe"] = {text = "Hoe", upAndDown = true, action = "turtle.dig"},
  ["minecraft:diamond_axe"] = {text = "Chop", upAndDown = true, action = "turtle.dig"},
  ["minecraft:diamond_shovel"] = {text = "Dig", upAndDown = true, action = "turtle.dig"}
}
local turtleImage = [[ffffffffff
f77777777f
f77ffff77f
f77777777f
f88888888f
f88888888f
ffffffffff
]]
local inventoryImage = [[77777777777777777
78887888788878887
78887888788878887
77777777777777777
78887888788878887
78887888788878887
77777777777777777
78887888788878887
78887888788878887
77777777777777777
78887888788878887
78887888788878887
77777777777777777
]]

-- Load Images
do
  local f = fs.open("temp","w") f.write(turtleImage) f.close()
  turtleImage = paintutils.loadImage("temp")
  f = fs.open("temp","w") f.write(inventoryImage) f.close()
  inventoryImage = paintutils.loadImage("temp") fs.delete("temp")
end

--=== Functions ===--
local function color(text,back)
  local temp = text and term.setTextColor(text) or nil
  temp = back and term.setBackgroundColor(back) or nil
end

local function printC(text,y)
  if type(text) ~= "string" or type(y) ~= "number" then error("expected string,number, got "..type(text)..","..type(y),2) end
  local lenght = #text
  local start = math.floor((w-lenght)/2)+1
  term.setCursorPos(start,y)
  term.write(text)
  return start,start+lenght
end

-- These functions are from my Screen API
-- http://pastebin.com/4j4mJsWw
local default = {
  object = {
    text = nil,
    name = "default",
    minX = 1,minY = 1,
    maxX = 7,maxY = 3,
    states = {
      on = {text = colors.white, back = colors.lime},
      off = {text = colors.white, back = colors.red}
    },
    clickable = false,
    visible = true,
    state = "off",
    action = nil
  },
  clickArea = {
    name = "default",
    minX = 1,minY = 1,
    maxX = 5,maxY = 3,
    clickable = true,
    action = nil
  }
}

local function fillTable(toFill,fillWith) --Used by the API
  if toFill == nil then toFill = {} end
  for a,b in pairs(fillWith) do
    if type(b) == "table" then
      toFill[a] = fillTable(toFill[a],b)
    else
      toFill[a] = toFill[a]==nil and b or toFill[a]
    end
  end
  return toFill
end

local function countIndexes(tbl) --Also used by API
  local total = 0
  for a,b in pairs(tbl) do total = total+1 end
  return total
end

local function assert(check,err,lvl)
  lvl = lvl or 2
  if not check then error(err,lvl+1) end
  return check
end

function new(nBackground)
  local api = {}
  local objects = {}
  local clickAreas = {}
  local background = nBackground

  function api.checkPos(x,y)
    x,y = tonumber(x),tonumber(y)
    assert(x and y,"expected number,number")
    for name,data in pairs(clickAreas) do
      if x >= data.minX and x <= data.maxX and y >= data.minY and y <= data.maxY and data.clickable then
        if type(data.action)=="function" then return true,true,data.action()
        else return true,false,"click_area",name end
      end
    end
    for name,data in pairs(objects) do
      if data.clickable and data.visible and x >= data.minX and x <= data.maxX and y >= data.minY and y <= data.maxY then
        if type(data.action)=="function" then return true,true,data.action()
        else return true,false,"object",name end
      end
    end
    return false
  end

  function api.handleEvents(bUseRaw)
    local pull = bUseRaw and os.pullEventRaw or os.pullEvent
    local event = {pull()}
    if event[1] == "mouse_click" then
      local wasElement,hadFunction,elementType,name = api.checkPos(event[3],event[4])
      if wasElement then
        if not hadFunction then
          return elementType,name,event[2]
        else return nil
        end
      end
    end
    return unpack(event)
  end

  function api.setDefaultObject(newDefaultObject)
    assert(type(newDefaultObject)=="table","expected table, got "..type(newDefaultObject))
    default.object = fillTable(newDefaultObject,default.object)
  end

  function api.setDefaultClickArea(newDefaultClickArea)
    assert(type(newDefaultClickArea)=="table","expected table, got "..type(newDefaultClickArea))
    default.clickArea = fillTable(newDefaultClickArea,default.clickArea)
  end

  function api.draw()
    if background then term.setBackgroundColor(background) term.clear() end
    for name,data in pairs(objects) do
      api.drawObject(name)
    end
  end

  function api.addClickArea(clickAreaInfo)
    assert(type(clickAreaInfo)=="table","expected table, got "..type(clickAreaInfo))
    fillTable(clickAreaInfo,default.clickArea)
    assert(clickAreas[clickAreaInfo.name]==nil,"an object with the name '"..clickAreaInfo.name.."' already exists")
    clickAreas[clickAreaInfo.name] = clickAreaInfo
  end

  function api.toggleClickArea(name)
    assert(clickAreas[name]~=nil,"Click Area '"..name.."' doesn't exist")
    clickAreas[name].clickable = not clickAreas[name].clickable
    return clickAreas[name].clickable
  end

  function api.getClickArea(name)
    assert(clickAreas[name]~=nil,"Click Area '"..name.."' doesn't exist")
    return clickAreas[name]
  end

  function api.addObject(objectInfo)
    assert(type(objectInfo)=="table","expected table, got "..type(objectInfo))
    objectInfo = fillTable(objectInfo,default.object)
    assert(objects[objectInfo.name]==nil,"an object with the name '"..objectInfo.name.."' already exists")
    objects[objectInfo.name] = objectInfo
  end

  function api.drawObject(name)
    assert(objects[name]~=nil,"Object '"..name.."' doesn't exsist")
    local objData = objects[name]
    if objData.visible == false then return end
    assert(objData.states~=nil,"Object '"..name.."' has no states!")
    assert(objData.states[objData.state],"Object '"..name.."' doesn't have state '"..objData.state.."'")
    term.setBackgroundColor(objData.states[objData.state].back)
    term.setTextColor(objData.states[objData.state].text)
    for i = 0, objData.maxY-objData.minY do
      term.setCursorPos(objData.minX,objData.minY+i)
      term.write(string.rep(" ",objData.maxX-objData.minX+1))
    end
    if objData.text then
      local xPos = objData.minX+math.floor(((objData.maxX-objData.minX+1)-#objData.text)/2)
      local yPos = objData.minY+(objData.maxY-objData.minY)/2
      term.setCursorPos(xPos,yPos) term.write(objData.text)
    end
  end

  function api.toggleObjectState(name) -- Only works if an object has two states
    assert(objects[name]~=nil,"Object '"..name.."' doesn't exist")
    assert(countIndexes(objects[name].states)==2,"Object '"..name.."' can't be toggled, it doesn't have two states")
    curState = objects[name].state
    for a,b in pairs(objects[name].states) do
      if a ~= curState then
        objects[name].state = a
      end
    end
    return objects[name].state
  end

  function api.setObjectState(name,state)
    assert(objects[name] ~= nil,"Object '"..name.."' doesn't exist")
    assert(objects[name].states[state],"Object '"..name.."' doesn't have state '"..state.."'")
    objects[name].state = state
  end

  function api.getObject(name)
    assert(objects[name] ~= nil,"Object '"..name.."' doesn't exist")
    return objects[name]
  end

  function api.toggleObjectVisible(name)
    assert(objects[name] ~= nil,"Object '"..name.."' doesn't exist")
    objects[name].visible = not objects[name].visible
    return objects[name].visible
  end

  function api.toggleObjectClickable(name)
    assert(objects[name] ~= nil,"Object '"..name.."' doesn't exist")
    objects[name].clickable = not objects[name].clickable
    return objects[name].clickable
  end

  return api
end
-- End Screen API

local function checkMessage(event,skipTableCheck,skipIDCheck)
  return event[4] == protocalName and (skipTableCheck or type(event[3]) == "table") and (skipIDCheck or event[2] == turtleID)
end

local function message(message)
  rednet.send(turtleID,message,protocalName)
end

local function broadcast(message)
  rednet.broadcast(message,protocalName)
end

local function execute(func,waitTime)
  message({action="execute",func=func})
  local timer = waitTime and os.startTimer(waitTime) or "done"
  local completed = false
  while true do
    local event = {os.pullEvent()}
    if event[1] == "rednet_message" and checkMessage(event) and event[3].action == "completed" then
      if timer == "done" then return event[3].result
      else completed = true end
    elseif event[1] == "timer" and timer == event[2] then
      if completed then break
      else timer = "done" end
    end
  end
end

local function getModem()
  for a,b in pairs(rs.getSides()) do
    if peripheral.getType(b) == "modem" and peripheral.call(b,"isWireless") then
      return b
    end
  end
end

local function waitForMessage(skipTableCheck,skipIDCheck)
  while true do
    local event = {os.pullEvent("rednet_message")}
    if checkMessage(event,skipTableCheck,skipIDCheck) then
      return event[3]
    end
  end
end

-- Create the screen
local mainScreen = new()

local function action(buttonName,func,waitTime)
  mainScreen.getObject(buttonName).state = "active" mainScreen.drawObject(buttonName)
  execute(func,waitTime or 0.5)
  mainScreen.getObject(buttonName).state = "on" mainScreen.drawObject(buttonName)
end

local function forward()
  action("forward","turtle.forward()")
end

local function turnLeft()
  action("turnLeft","turtle.turnLeft()")
end

local function turnRight()
  action("turnRight","turtle.turnRight()")
end

local function back()
  action("back","turtle.back()")
end

local function up()
  action("up","turtle.up()")
end

local function down()
  action("down","turtle.down()")
end

local drawMainScreen

local function options()
  -- Silly animition that I love so please don't question it
  for i = 0, 12 do
    color(colors.white,colors.blue) term.clear()
    paintutils.drawImage(turtleImage,9,7-(i/2))
    color(colors.white,colors.green) term.setCursorPos(1,h-i) term.clearLine() printC("Options",h-i)
    color(colors.white,colors.brown)
    for a = 0, i-1 do
      term.setCursorPos(1,h-a) term.clearLine()
    end
    sleep(0.1)
  end

  color(colors.black,colors.lightGray)
  printC("              ",10)
  printC("  Disconnect  ",11)
  printC("   and Exit   ",12)
  printC("              ",13)
  printC("              ",15)
  printC("     Back     ",16)
  printC("              ",17)

  while true do
    local event = {os.pullEvent("mouse_click")}
    if event[4] == 10 or event[4] == 11 or event[4] == 12 or event[4] == 1 then
      error()
    elseif event[4] == 16 or event[4] == 15 or event[4] == 17 then
      break
    end
  end

  for i = 12, 0, -1 do
    color(colors.white,colors.blue) term.clear()
    paintutils.drawImage(turtleImage,9,7-(i/2))
    color(colors.white,colors.green) term.setCursorPos(1,h-i) term.clearLine() printC("Options",h-i)
    color(colors.white,colors.brown)
    for a = 0, i-1 do
      term.setCursorPos(1,h-a) term.clearLine()
    end
    sleep(0.1)
  end
  drawMainScreen()
end

-- Add buttons
--mainScreen.addObject({name="inventory", minX=10, maxY=12, state="default", maxX=17, clickable=false, visible=true, states={default={text=1,back=256,},}, minY=11, text="View Inv", })
mainScreen.addObject({name="turnRight", minX=20, maxY=18, state="on", maxX=25, clickable=true, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, minY=16, text="Turn R", action=turnRight})
mainScreen.addObject({name="actionLeft", minX=20, maxY=11, state="on", maxX=26, clickable=false, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, text="No Tool", minY=9, })
mainScreen.addObject({name="turnLeft", minX=2, maxY=18, state="on", maxX=7, clickable=true, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, minY=16, text="Turn L", action=turnLeft})
mainScreen.addObject({name="actionRightDown", minX=3, maxY=13, state="on", maxX=7, clickable=false, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, minY=13, text="v", })
mainScreen.addObject({name="actionRight", minX=1, maxY=11, state="on", maxX=7, clickable=false, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, minY=9, text="No Tool", })
mainScreen.addObject({name="back", minX=9, maxY=17, state="on", maxX=18, clickable=true, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, minY=15, text="Backward", action=back})
mainScreen.addObject({minY=20, minX=1, state="default", maxY=20, maxX=26, clickable=true, visible=true, states={default={text=1,back=8192,},}, name="options", text="Options", action = options})
mainScreen.addObject({name="actionRightUp", minX=3, maxY=7, state="on", maxX=7, clickable=true, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, minY=7, text="^", })
mainScreen.addObject({name="actionLeftDown", minX=20, maxY=13, state="on", maxX=24, clickable=false, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, text="v", minY=13, })
mainScreen.addObject({name="down", minX=20, maxY=4, state="on", maxX=25, clickable=true, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, text="Down", minY=2, action=down})
mainScreen.addObject({name="up", minX=2, maxY=4, state="on", maxX=7, clickable=true, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, text="Up", minY=2, action=up})
mainScreen.addObject({name="actionLeftUp", minX=20, maxY=7, state="on", maxX=24, clickable=false, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, text="^", minY=7, })
mainScreen.addObject({name="forward", minX=9, maxY=5, state="on", maxX=18, clickable=true, visible=true, states={off={text=1,back=256,},on={text=32768,back=1,},active={text=1,back=32,},}, minY=3, text="Forward", action=forward})
-- Function to draw the screen
drawMainScreen = function()
  color(nil,colors.blue) term.clear()
  paintutils.drawImage(turtleImage,9,7)
  mainScreen.draw()
end

local function getFreeSlot()
  for i = 1, 16 do
    local result = execute("turtle.getItemCount("..i..")")
    if result == 0 then
      return i
    end
  end
  return false
end

local function showToolName(name,side)
  local obj = mainScreen.getObject("action"..side)
  local up = mainScreen.getObject("action"..side.."Up")
  local down = mainScreen.getObject("action"..side.."Down")
  if knownTools[name] then
    obj.text = knownTools[name].text
    obj.state = "on"
    obj.clickable = true
    if knownTools[name].action then
      obj.action = function() action("action"..side,knownTools[ tools.left ].action.."()",0.2) end
    end
    mainScreen.drawObject("action"..side)
    if knownTools[name].upAndDown then
      up.state = "on"
      up.clickable = true
      up.action = function() action("action"..side.."Up",knownTools[name].action.."Up()",0.2) end
      down.state = "on"
      down.clickable = true
      down.action = function() action("action"..side.."Down",knownTools[name].action.."Down()",0.2) end
      mainScreen.drawObject("action"..side.."Up")
      mainScreen.drawObject("action"..side.."Down")
    else
      up.state = "off"
      up.clickable = false
      down.state = "off"
      down.clickable = false
      mainScreen.drawObject("action"..side.."Up")
      mainScreen.drawObject("action"..side.."Down")
    end
  else
    obj.text = name=="none" and "No Tool"  or "?"
    obj.state ="off"
    obj.clickable = false
    mainScreen.drawObject("action"..side)
    up.state = "off"
    up.clickable = false
    down.state = "off"
    down.clickable = false
    mainScreen.drawObject("action"..side.."Up")
    mainScreen.drawObject("action"..side.."Down")
  end
end

local function getLeftTool()
  message({action="getLeftTool"})
  while true do
    local msg = waitForMessage()
    if msg.action == "getLeftTool" then
      tools.left = msg.result
      return msg.result
    end
  end
end

local function getRightTool()
  message({action="getRightTool"})
  while true do
    local msg = waitForMessage()
    if msg.action == "getRightTool" then
      tools.right = msg.result
      return msg.result
    end
  end
end

local function detectTools()
  showToolName(getRightTool(),"Right")
  showToolName(getLeftTool(),"Left")
end

--=== Program ===--
if not pocket then
  printError("This program requires a pocket computer!")
  return
elseif args[1] == "help" then
  print("ControlMe")
  print("By HPWebcamAble")
  print("")
  print("Use this program to remotly control your turtle!")
  print("Start this program on your pockect computer and the corresponding program on a turtle and control away!")
end

do
  local modemSide = getModem()
  if modemSide then
    rednet.open(modemSide)
  else
    printError("This program requires a wireless modem")
    return
  end
end

local continue = true
while continue do
  color(colors.white,colors.black) term.clear()
  printC("Enter your Turtle's ID:",5)
  term.setCursorPos(math.floor(w/2)-2,6)
  local input = tonumber(read())

  if input then
    color(colors.white,colors.black) term.clear()
    printC("Connecting to",5)
    printC("turtle ID "..input,6)
    printC("(*)",8)
    local state = 1

    turtleID = input
    message({action="connect",id=input})
    local timer = os.startTimer(1)
    while true do
      local event = {os.pullEventRaw()}
      if event[1] == "terminate" then
        term.clear() term.setCursorPos(1,1) print("Program terminated")
        return
      elseif event[1] == "rednet_message" and checkMessage(event,true) then
        if event[3] == "connected" then
          continue = false
          break
        end
      elseif event[1] == "timer" and event[2] == timer then
        state = (state == 4 and 1 or state+1)
        term.clearLine() printC(string.rep("(",state).."*"..string.rep(")",state),8)
        message({action="connect",id=input})
        timer = os.startTimer(1)
      end
    end
  end
end

local function heartbeat()

  message("ping")
  local timer = os.startTimer(2)
  while true do
    local event = {os.pullEvent()}
    if event[1] == "rednet_message" then
      if checkMessage(event,true) and event[3] == "pong" then
        sleep(3)
        message("ping")
        timer = os.startTimer(2)
      end
    elseif event[1] == "timer" and event[2] == timer then
      color(colors.white,colors.black) term.setCursorPos(1,1) term.clear() print("Lost contact with turtle!")
      error("FORCEQUIT")
    end
  end

end

local function main()

  drawMainScreen()
  detectTools()

  while true do
    local event = {mainScreen.handleEvents()}
    if event[1] == "rednet_message" and checkMessage(event) then
      local msg = event[3]
      if msg.action == "ping" then
        message("pong")
      end
    elseif event[1] == "key" then
      if event[2] == keys.up or event[2] == keys.w then
        forward()
      elseif event[2] == keys.down or event[2] == keys.s then
        back()
      elseif event[2] == keys.right or event[2] == keys.d then
        turnRight()
      elseif event[2] == keys.left or event[2] == keys.a then
        turnLeft()
      elseif event[2] == keys.space then
        up()
      elseif event[2] == keys.leftShift then
        down()
      elseif event[2] == keys.q then

      end
    elseif event[1] == "object" then

    end
  end

end

local function run()
  parallel.waitForAny(main,heartbeat)
end

local state,err = pcall(run)

if err then
  if not err:find("Terminated") and not err:find("FORCEQUIT") then
    color(colors.white,colors.black)
    term.clear()
    printC("An error occured:",1)
    term.setCursorPos(1,3)
    print(err)
  elseif not err:find("FORCEQUIT") then
    color(colors.white,colors.black)
    term.clear()
    term.setCursorPos(1,1)
  end
else
  color(colors.white,colors.black)
  term.clear()
  term.setCursorPos(1,1)
end