os.loadAPI("button")
m = peripheral.wrap("right")
m.clear()

local time = 15
local enabled = false
local count = 0
local dir = 0

function pulse(color)
    rs.setBundledOutput("back", color)
    sleep(1)
    rs.setBundledOutput("back", 0)
end

local function getDir()
  if dir == 1 then
    return "Forward"
  elseif dir == 2 then
    return "Backward"
  elseif dir == 3 then
    return "Left"
  elseif dir == 4 then
    return "Right"
  else
    return "None"
  end
end

function status()
  if enabled then m.setBackgroundColor(colors.green)
  else m.setBackgroundColor(colors.red) end

  m.setCursorPos(2,2)
  m.write("                           ")
  m.setCursorPos(2,3)
  m.write("                           ")
  m.setCursorPos(2,4)
  m.write("                           ")

  w,h = m.getSize()

  if enabled then
    m.setCursorPos((w-string.len("Status : Enabled"))/2+1, 2)
    m.write("Status : Enabled")
  else
    m.setCursorPos((w-string.len("Status : Disabled"))/2+1, 2)
    m.write("Status : Disabled")
  end

  m.setCursorPos((w-string.len("Count : "..count))/2+1, 3)
  m.write("Count : "..count)

  m.setCursorPos((w-string.len("Direction : "..getDir(dir)))/2+1, 4)
  m.write("Direction : "..getDir(dir))
end

function fillTable()
   button.setTable("Forward", forward, 10,20,6,6)
   button.setTable("Left", left, 2,12,8,8)
   button.setTable("Right", right, 18,28,8,8)
   button.setTable("Back", back, 10,20,10,10)
   button.screen()
end

function getClick()
   event,side,x,y = os.pullEvent("monitor_touch")
   button.checkxy(x,y)
end

function forward()
   button.toggleButton("Forward")
   enabled = true
   dir = 1
   count = 10

   while count > 0 do
     pulse(colors.red)

     count = count - 1
     status()
     sleep(time)
   end

   button.toggleButton("Forward")
   enabled = false
   dir = 0

   status()
end

function back()
  button.toggleButton("Back")
  enabled = true
  dir = 2
  count = 100

  while count > 0 do
    pulse(colors.green)
    count = count - 1
    status()
    sleep(time)
  end

  button.toggleButton("Back")
  enabled = false
  dir = 0

  status()
end

function left()
  button.toggleButton("Left")
  enabled = true
  dir = 3
  count = 5

  while count > 0 do
    pulse(colors.brown)
    count = count - 1
    status()
    sleep(3)
  end

  button.toggleButton("Left")
  enabled = false
  dir = 0

  status()
end

function right()
  button.toggleButton("Right")
  enabled = true
  dir = 4
  count = 5

  while count > 0 do
    pulse(colors.blue)
    count = count - 1
    status()
    sleep(3)
  end

  button.toggleButton("Right")
  enabled = false
  dir = 0

  status()
end

fillTable()
status()

while true do
   getClick()
end