Added WIP driver for computer-craft flavors of devices.
This commit is contained in:
parent
dc9ca7c2af
commit
2771e934c9
|
|
@ -0,0 +1,235 @@
|
|||
local VERSION = "1.0.2"
|
||||
|
||||
-- Global component states
|
||||
local components = {}
|
||||
|
||||
-- Global signal-indexed state maps.
|
||||
local lastTrainOverheadStates = {}
|
||||
local lastTrainOverheadDataObjs = {}
|
||||
local trainScanTimers = {}
|
||||
|
||||
-- Determines if a train is currently traversing the given signal.
|
||||
local function trainOverhead(signal)
|
||||
return redstone.getInput(signal.redstoneInputSide)
|
||||
end
|
||||
|
||||
-- Determines if a train was over the given signal the last time we checked.
|
||||
local function wasTrainOverheadPreviously(signal)
|
||||
return lastTrainOverheadStates[signal.id] == true
|
||||
end
|
||||
|
||||
local function getTrainData(signal)
|
||||
local det = peripheral.wrap(signal.detector)
|
||||
if det == nil then
|
||||
print("Error: Signal " .. signal.id .. "'s detector could not be connected. Please fix the config and restart.")
|
||||
else
|
||||
return det.consist()
|
||||
end
|
||||
end
|
||||
|
||||
-- Returns the from, to branches for a signal when a train travels in the given direction.
|
||||
local function getBranches(dir, signal)
|
||||
local branches = signal.branches
|
||||
if string.upper(branches[1].direction) == string.upper(dir) then
|
||||
return branches[2].id, branches[1].id
|
||||
elseif string.upper(branches[2].direction) == string.upper(dir) then
|
||||
return branches[1].id, branches[2].id
|
||||
end
|
||||
end
|
||||
|
||||
-- Sends an update to the RailSignal API.
|
||||
local function sendSignalUpdate(ws, signal, data, msgType)
|
||||
local from, to = getBranches(string.upper(data.direction), signal)
|
||||
local msg = textutils.serializeJSON({
|
||||
signalId = signal.id,
|
||||
fromBranchId = from,
|
||||
toBranchId = to,
|
||||
type = msgType
|
||||
})
|
||||
print("-> S: " .. signal.id .. ", from: " .. from .. ", to: " .. to .. ", T: " .. msgType)
|
||||
ws.send(msg)
|
||||
end
|
||||
|
||||
local function updateBranchStatusIndicator(branch, status, config)
|
||||
if branch.monitor ~= nil then
|
||||
local mon = peripheral.wrap(branch.monitor)
|
||||
if mon ~= nil then
|
||||
local c = config.statusColors[status]
|
||||
if c == nil then c = config.statusColors.ERROR end
|
||||
mon.setBackgroundColor(c)
|
||||
mon.clear()
|
||||
else
|
||||
print("Error! Could not connect to monitor " .. branch.monitor .. " for branch " .. branch.id .. ". Check and fix config.")
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function updateBranchStatus(signal, branchId, status, config)
|
||||
for _, branch in pairs(signal.branches) do
|
||||
if branch.id == branchId then
|
||||
updateBranchStatusIndicator(branch, status, config)
|
||||
return
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- Manually set the status indicator for all branch monitors.
|
||||
local function setAllBranchStatus(config, status)
|
||||
for _, signal in pairs(config.signals) do
|
||||
for _, branch in pairs(signal.branches) do
|
||||
updateBranchStatusIndicator(branch, status, config)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function initMonitorColors(config)
|
||||
for _, signal in pairs(config.signals) do
|
||||
for _, branch in pairs(signal.branches) do
|
||||
if branch.monitor ~= nil then
|
||||
local mon = peripheral.wrap(branch.monitor)
|
||||
if mon ~= nil then
|
||||
for c, v in pairs(config.paletteColors) do
|
||||
mon.setPaletteColor(c, v)
|
||||
end
|
||||
else
|
||||
print("Error! Could not connect to monitor " .. branch.monitor .. " for branch " .. branch.id .. ". Check and fix config.")
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- Checks if all signals managed by this controller are reporting an "online" status.
|
||||
local function checkSignalOnlineStatus(config)
|
||||
for _, signal in pairs(config.signals) do
|
||||
local resp = http.get(config.apiUrl .. "/railSystems/" .. config.rsId .. "/signals/" .. signal.id)
|
||||
if resp == nil or resp.getResponseCode() ~= 200 then
|
||||
return false
|
||||
else
|
||||
local signalData = textutils.unserializeJSON(resp.readAll())
|
||||
if not signalData.online then
|
||||
return false
|
||||
end
|
||||
end
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
local function handleRedstoneEvent(ws, config)
|
||||
for _, signal in pairs(config.signals) do
|
||||
if trainOverhead(signal) and not wasTrainOverheadPreviously(signal) then
|
||||
local data = getTrainData(signal)
|
||||
if data == nil then
|
||||
print("Got redstone event but could not obtain train data on signal " .. signal.id)
|
||||
else
|
||||
lastTrainOverheadDataObjs[signal.id] = data
|
||||
lastTrainOverheadStates[signal.id] = true
|
||||
sendSignalUpdate(ws, signal, data, "BEGIN")
|
||||
trainScanTimers[signal.id] = os.startTimer(config.trainScanInterval)
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function handleTrainScanTimerEvent(ws, config, timerId)
|
||||
for k, signal in pairs(config.signals) do
|
||||
if trainScanTimers[signal.id] == timerId then
|
||||
if trainOverhead(signal) then -- The train is still overhead.
|
||||
local data = getTrainData(signal)
|
||||
if data ~= nil then
|
||||
lastTrainOverheadDataObjs[signal.id] = data
|
||||
end
|
||||
trainScanTimers[signal.id] = os.startTimer(config.trainScanInterval)
|
||||
else -- The train has left the signal so send an update.
|
||||
sendSignalUpdate(ws, signal, lastTrainOverheadDataObjs[signal.id], "END")
|
||||
lastTrainOverheadDataObjs[signal.id] = nil
|
||||
lastTrainOverheadStates[signal.id] = nil
|
||||
trainScanTimers[signal.id] = nil
|
||||
end
|
||||
return
|
||||
end
|
||||
end
|
||||
print("Warn: Train scan timer was ignored.")
|
||||
end
|
||||
|
||||
local function handleWebSocketMessage(msg, config)
|
||||
local data = textutils.unserializeJSON(msg)
|
||||
local branchId = data["branchId"]
|
||||
local status = data["status"]
|
||||
print("<- B: " .. branchId .. ", Status: " .. status)
|
||||
for _, signal in pairs(config.signals) do
|
||||
updateBranchStatus(signal, branchId, status, config)
|
||||
end
|
||||
end
|
||||
|
||||
local function loadConfig(file)
|
||||
local f = io.open(file, "r")
|
||||
if f == nil then return createConfig(file) end
|
||||
local text = f:read("*a")
|
||||
f:close()
|
||||
return textutils.unserialize(text)
|
||||
end
|
||||
|
||||
-- Connects to the RailSignal API websocket. Will block indefinitely until a connection can be obtained.
|
||||
local function connectToWebSocket(config)
|
||||
local signalIds = {}
|
||||
for _, signal in pairs(config.signals) do
|
||||
table.insert(signalIds, tostring(signal.id))
|
||||
end
|
||||
local signalIdsStr = table.concat(signalIds, ",")
|
||||
while true do
|
||||
local ws, err = http.websocket(config.wsUrl, {[config.wsHeader] = signalIdsStr})
|
||||
if ws == false then
|
||||
print("Error connecting to RailSignal websocket:\n\tError: " .. err .. "\n\tTrying again in 3 seconds.")
|
||||
os.sleep(3)
|
||||
else
|
||||
print("Successfully connected to RailSignal websocket at " .. config.wsUrl .. " for managing signals: " .. signalIdsStr)
|
||||
return ws
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
-- Main Script
|
||||
term.clear()
|
||||
print("RailSignal Signal Controller V" .. VERSION)
|
||||
print(" By Andrew Lalis. For more info, check here: https://github.com/andrewlalis/RailSignalAPI")
|
||||
print(" To update, execute \"sig update\" in the command line.")
|
||||
local w, h = term.getSize()
|
||||
print(string.rep("-", w))
|
||||
if arg[1] ~= nil and string.lower(arg[1]) == "update" then
|
||||
print("Updating to the latest version of RailSignal signal program.")
|
||||
fs.delete("sig.lua")
|
||||
shell.execute("pastebin", "get", "erA3mSfd", "sig.lua")
|
||||
print("Download complete. Restarting...")
|
||||
os.sleep(1)
|
||||
os.reboot()
|
||||
end
|
||||
local config = loadConfig("sig_config.tbl")
|
||||
initMonitorColors(config)
|
||||
setAllBranchStatus(config, "NONE")
|
||||
local ws = connectToWebSocket(config)
|
||||
local refreshWebSocketAlarm = os.setAlarm((os.time() + math.random(1, 23)) % 24)
|
||||
while true do
|
||||
local eventData = {os.pullEvent()}
|
||||
local event = eventData[1]
|
||||
if event == "redstone" then
|
||||
handleRedstoneEvent(ws, config)
|
||||
elseif event == "timer" then
|
||||
handleTrainScanTimerEvent(ws, config, eventData[2])
|
||||
elseif event == "websocket_message" then
|
||||
handleWebSocketMessage(eventData[3], config)
|
||||
elseif event == "websocket_closed" then
|
||||
setAllBranchStatus(config, "ERROR")
|
||||
print("! RailSignal websocket closed. Attempting to reconnect.")
|
||||
os.sleep(0.5)
|
||||
ws = connectToWebSocket(config)
|
||||
elseif event == "alarm" and eventData[2] == refreshWebSocketAlarm then
|
||||
print("! Checking signal online status.")
|
||||
if not checkSignalOnlineStatus(config) then
|
||||
print("Not all signals are reporting an online status. Reconnecting to the websocket.")
|
||||
ws.close()
|
||||
ws = connectToWebSocket(config)
|
||||
end
|
||||
end
|
||||
end
|
||||
ws.close()
|
||||
Loading…
Reference in New Issue