--init
os.loadAPI("controllerDir/json")

--Monitor
mon = peripheral.wrap("top")

--Shortcuts
r = "right"
l = "left"

--other vars
tracks = {}
before = 1

--functions
function arg(arg, _type)

  if type(arg) ~= _type then
    term.setTextColor(colors.red)
    error("Bad argument: Expected ".._type.." but got "..type(arg).."!")
  end
end

function trainOnTrack(track)
  arg(track, "number")
  return rs.testBundledInput("right", track) 
end

function placeholder() 
  
  w,h = term.getSize()
  
  for i=1, w, 1 do
    write("-")
  end
  
end

function regTrack(color)

  arg(color, "number")
  table.insert(tracks, color)

end

function openRednet()
  
  for _,side in ipairs(rs.getSides()) do
    if peripheral.isPresent(side) and peripheral.getType(side) == "modem" then
      rednet.open(side)
      return side
    end
  end
  error("No modem connected!")
end

try = 0

function sendTrain()

  r = math.random(1, #tracks)
  color = tracks[r]
  train = trainOnTrack(colors.red)
  
  if not train then
    try = try +1
    if try == 14 then
      print("Found 14 times no train, sleeping 2 sec")
      sleep(2)
    end
    
    if try == 20 then
      print("Found 20 times no train, giving up...")
    else
      sendTrain()
    end
  else
    rs.setBundledOutput("left", color)
    sleep(2)
    rs.setBundledOutput("left", 0)
    try = 0
    print("Train send!")
  end
end

--Code
print("Starting controller...")
placeholder()
print("MC-V:".. _MC_VERSION)
print("CC-V:".. _CC_VERSION)
print("LuaJ-V:".. _LUAJ_VERSION)
placeholder()
openRednet()
rednet.host("json","train.base.red")

regTrack(colors.blue)

function handle()
  id,msgraw = rednet.receive("json")
  print("Recived")
  term.setTextColor(colors.orange)
  print(msgraw)
  term.setTextColor(colors.white)
  write("from ")
  term.setTextColor(colors.blue)
  print(id)
  term.setTextColor(colors.white)
  --#META_SETUP#--
  msg = {}
  msg_meta = {}
  setmetatable(msg, msg_meta)
  msg_meta.__index = 
  function(table, key)
    print("Invalid message recived! (Value for " .. key .. "is nil!")
    placeholder()
    handle()
  end
  --#META_SETUP#--    
  
  msg = json.decode(msgraw)
  if msg == nil then
    print("Invalid message recived! (No JSON)")
    placeholder()
    handle()
  end
  
  if msg.command == "send" then
    print("Remote send request recived!")
    sendTrain()
  end
  
  placeholder()
end

while true do 
  handle()
end

  if msg == nil then
    print("Invalid message recived! (No JSON)")
    placeholder()
    handle()
  end