local MODEM_SIDE = "bottom"
local MONITOR_SIDE = "left"
local DB_FILE = "switchmap.db"
local PROTOCOL = "rtm_switch"
local native = term.current()

local map = {
version = 1,
gridW = 20,
gridH = 12,
tiles = {},
}

local selectedX, selectedY = 1, 1
local currentTool = "rail_h"
local status = "Indulas..."
local monitor = nil
local lastTouch = nil
local running = true

local tileTypes = {
"empty",
"rail_h",
"rail_v",
"curve_ne",
"curve_nw",
"curve_se",
"curve_sw",
"cross",
"switch_cross",
"signal",
"label",
}

local tileChars = {
empty = " ",
rail_h = "=",
rail_v = "|",
curve_ne = "/",
curve_nw = "\\",
curve_se = "\\",
curve_sw = "/",
cross = "X",
switch_cross = "S",
signal = "!",
label = "L",
}

local function center(text, y, txt, bg)
local w, _ = term.getSize()
term.setCursorPos(math.max(1, math.floor((w - #text) / 2) + 1), y)
if bg then term.setBackgroundColor(bg) end
if txt then term.setTextColor(txt) end
term.write(text)
end

local function setStatus(msg)
status = msg
end

local function openNet()
if peripheral.getType(MODEM_SIDE) ~= "modem" then
setStatus("Nincs modem a bottom oldalon")
return false
end
if not rednet.isOpen(MODEM_SIDE) then
rednet.open(MODEM_SIDE)
end
return true
end

local function getTile(x, y)
for i = 1, #map.tiles do
local t = map.tiles[i]
if t.x == x and t.y == y then return t, i end
end
return nil, nil
end

local function setTile(x, y, data)
local _, idx = getTile(x, y)
if data.type == "empty" then
if idx then table.remove(map.tiles, idx) end
return
end
if idx then
map.tiles[idx] = data
else
table.insert(map.tiles, data)
end
end

local function saveDb()
local h = fs.open(DB_FILE, "w")
h.write(textutils.serialize(map))
h.close()
setStatus("Mentve: " .. DB_FILE)
end

local function loadDb()
if not fs.exists(DB_FILE) then
setStatus("Nincs adatbazis, uj keszul")
return
end
local h = fs.open(DB_FILE, "r")
local raw = h.readAll()
h.close()
local t = textutils.unserialize(raw)
if type(t) == "table" then
map = t
setStatus("Betoltve: " .. DB_FILE)
else
setStatus("Hibas adatbazis")
end
end

local function tileColor(tile)
if not tile then return colors.black, colors.gray end
if tile.type == "rail_h" or tile.type == "rail_v" or tile.type == "curve_ne" or tile.type == "curve_nw" or tile.type == "curve_se" or tile.type == "curve_sw" or tile.type == "cross" then
return colors.black, colors.lightGray
elseif tile.type == "switch_cross" then
if tile.offline then return colors.black, colors.red end
if tile.state == "cross" then return colors.black, colors.orange end
return colors.black, colors.lime
elseif tile.type == "signal" then
return colors.black, colors.red
elseif tile.type == "label" then
return colors.black, colors.cyan
end
return colors.black, colors.white
end

local function drawMap()
if not monitor then return end
monitor.setBackgroundColor(colors.black)
monitor.clear()
local mw, mh = monitor.getSize()
local ox, oy = 2, 2

monitor.setCursorPos(2, 1)
monitor.setTextColor(colors.cyan)
monitor.write("RTM TERKEP")

for y = 1, map.gridH do
for x = 1, map.gridW do
local sx = ox + (x - 1) * 2
local sy = oy + (y - 1)
if sx <= mw - 1 and sy <= mh - 2 then
local tile = getTile(x, y)
local bg = colors.black
local txt = colors.gray
local ch = "."
if tile then
bg, txt = tileColor(tile)
ch = tileChars[tile.type] or "?"
end
if x == selectedX and y == selectedY then
monitor.setBackgroundColor(colors.blue)
monitor.setTextColor(colors.white)
else
monitor.setBackgroundColor(bg)
monitor.setTextColor(txt)
end
monitor.setCursorPos(sx, sy)
monitor.write(ch .. " ")
end
end
end

monitor.setBackgroundColor(colors.black)
monitor.setTextColor(colors.white)
monitor.setCursorPos(2, mh)
monitor.write("Erintes: valto/kurzor")
end

local function prompt(text, default)
term.setBackgroundColor(colors.black)
term.setTextColor(colors.white)
write(text)
local v = read(nil, nil, nil, default)
return v
end

local function editTileAt(x, y)
term.redirect(native)
term.setBackgroundColor(colors.black)
term.setTextColor(colors.white)
term.clear()
term.setCursorPos(1,1)
local tile = getTile(x, y) or {x = x, y = y, type = currentTool}

print("Szerkesztes: " .. x .. "," .. y)
print("Tipusok:")
for i = 1, #tileTypes do
print(i .. " - " .. tileTypes[i])
end
write("Tipus szama [" .. tile.type .. "]: ")
local choice = tonumber(read())
if choice and tileTypes[choice] then tile.type = tileTypes[choice] end

if tile.type == "empty" then
setTile(x, y, {x = x, y = y, type = "empty"})
setStatus("Torolve: " .. x .. "," .. y)
return
end

tile.x = x
tile.y = y

if tile.type == "switch_cross" then
tile.name = prompt("Nev [" .. (tile.name or "Valto") .. "]: ", tile.name or "Valto")
tile.slaveId = tonumber(prompt("Slave ID [" .. tostring(tile.slaveId or "") .. "]: ", tile.slaveId and tostring(tile.slaveId) or ""))
tile.state = prompt("Allapot (straight/cross) [" .. (tile.state or "straight") .. "]: ", tile.state or "straight")
tile.protocol = PROTOCOL
tile.offline = false
elseif tile.type == "label" then
tile.text = prompt("Felirat [" .. (tile.text or "") .. "]: ", tile.text or "")
elseif tile.type == "signal" then
tile.name = prompt("Jelzo neve [" .. (tile.name or "Jelzo") .. "]: ", tile.name or "Jelzo")
end

setTile(x, y, tile)
setStatus("Elem mentve: " .. tile.type .. " @ " .. x .. "," .. y)
end

local function sendSwitch(tile, newState)
if not tile or tile.type ~= "switch_cross" then return end
if not tile.slaveId then
setStatus("Nincs slave ID")
return
end
if not openNet() then return end

rednet.send(tile.slaveId, {cmd = "set", state = newState}, tile.protocol or PROTOCOL)
local timer = os.startTimer(2)
while true do
local e, a, b, c = os.pullEvent()
if e == "rednet_message" then
local senderId, msg, protocol = a, b, c
if senderId == tile.slaveId and protocol == (tile.protocol or PROTOCOL) and type(msg) == "table" and msg.ok then
tile.state = msg.state or newState
tile.offline = false
setStatus("Valasz: " .. (tile.name or "valt") .. " -> " .. tile.state)
return
end
elseif e == "timer" and a == timer then
tile.offline = true
setStatus("Timeout slave #" .. tile.slaveId)
return
end
end
end

local function pingAllSwitches()
if not openNet() then return end
for _, tile in ipairs(map.tiles) do
if tile.type == "switch_cross" and tile.slaveId then
rednet.send(tile.slaveId, {cmd = "ping"}, tile.protocol or PROTOCOL)
tile.offline = true
end
end

local timer = os.startTimer(1.5)
while true do
local e, a, b, c = os.pullEvent()
if e == "rednet_message" then
local senderId, msg, protocol = a, b, c
if type(msg) == "table" and msg.ok and protocol == PROTOCOL then
for _, tile in ipairs(map.tiles) do
if tile.type == "switch_cross" and tile.slaveId == senderId then
tile.offline = false
tile.state = msg.state or tile.state
end
end
end
elseif e == "timer" and a == timer then
setStatus("Ping kesz")
return
end
end
end

local function handleMonitorTouch(side, x, y)
if side ~= MONITOR_SIDE then return end
local gx = math.floor((x - 2) / 2) + 1
local gy = (y - 2) + 1
if gx >= 1 and gx <= map.gridW and gy >= 1 and gy <= map.gridH then
selectedX, selectedY = gx, gy
lastTouch = {x = gx, y = gy}
local tile = getTile(gx, gy)
if tile and tile.type == "switch_cross" then
local ns = tile.state == "straight" and "cross" or "straight"
sendSwitch(tile, ns)
else
setStatus("Kijelolve: " .. gx .. "," .. gy)
end
end
end

local function uiLoop()
while running do
term.redirect(native)
term.setBackgroundColor(colors.black)
term.setTextColor(colors.white)
term.clear()
term.setCursorPos(1,1)
print("RTM HALOZATI MANAGER")
print("====================")
print("Kijelolt tile: " .. selectedX .. "," .. selectedY)
local tile = getTile(selectedX, selectedY)
if tile then
print("Tipus: " .. tile.type)
if tile.type == "switch_cross" then
print("Slave: " .. tostring(tile.slaveId or "nincs"))
print("Allapot: " .. tostring(tile.state or "?"))
print("Offline: " .. tostring(tile.offline or false))
end
else
print("Tipus: ures")
end
print("")
print("1 - Tile szerkesztes")
print("2 - Kurzor mozgas")
print("3 - Valto valtasa")
print("4 - Osszes slave ping")
print("5 - Mentes")
print("6 - Betoltes")
print("7 - Palya meret")
print("8 - Kilepes")
print("")
print("Status: " .. status)
write("> ")
local cmd = read()

if cmd == "1" then
editTileAt(selectedX, selectedY)
elseif cmd == "2" then
local nx = tonumber(prompt("X: ", tostring(selectedX))) or selectedX
local ny = tonumber(prompt("Y: ", tostring(selectedY))) or selectedY
selectedX = math.max(1, math.min(map.gridW, nx))
selectedY = math.max(1, math.min(map.gridH, ny))
setStatus("Kurzor mozgatva")
elseif cmd == "3" then
local t = getTile(selectedX, selectedY)
if t and t.type == "switch_cross" then
sendSwitch(t, t.state == "straight" and "cross" or "straight")
else
setStatus("Itt nincs valto")
end
elseif cmd == "4" then
pingAllSwitches()
elseif cmd == "5" then
saveDb()
elseif cmd == "6" then
loadDb()
elseif cmd == "7" then
local gw = tonumber(prompt("Grid W: ", tostring(map.gridW))) or map.gridW
local gh = tonumber(prompt("Grid H: ", tostring(map.gridH))) or map.gridH
map.gridW = math.max(5, gw)
map.gridH = math.max(5, gh)
setStatus("Palya meret frissitve")
elseif cmd == "8" then
running = false
break
else
setStatus("Ismeretlen parancs")
end
drawMap()
end
end

local function monitorLoop()
while running do
drawMap()
local e, side, x, y = os.pullEvent()
if e == "monitor_touch" then
handleMonitorTouch(side, x, y)
drawMap()
elseif e == "monitor_resize" then
drawMap()
end
end
end

local function netLoop()
while running do
local senderId, msg, protocol = rednet.receive(PROTOCOL)
if type(msg) == "table" and msg.ok then
for _, tile in ipairs(map.tiles) do
if tile.type == "switch_cross" and tile.slaveId == senderId then
tile.offline = false
tile.state = msg.state or tile.state
setStatus("Frissult: " .. (tile.name or ("#" .. senderId)))
end
end
drawMap()
end
end
end

loadDb()
openNet()
if peripheral.getType(MONITOR_SIDE) == "monitor" then
monitor = peripheral.wrap(MONITOR_SIDE)
monitor.setTextScale(0.5)
else
setStatus("Nincs monitor a left oldalon")
end

if monitor then
parallel.waitForAny(uiLoop, monitorLoop, netLoop)
else
parallel.waitForAny(uiLoop, netLoop)
end

term.redirect(native)
term.setBackgroundColor(colors.black)
term.setTextColor(colors.white)
term.clear()
term.setCursorPos(1,1)
print("Manager leallva")