CC: Nuclear Generator Controller

--[[--

    Wireless modem: back
    Wired modem: front

    Relay id: 131
    Monitor name: monitor_1
    MFE name: batbox_1
    Self name: computer_4

    Controller nickname: controller
    Relay nickname: relay

--]]--

--(( Variables ))--

local running = true

local controllerid = os.getComputerID()
local controllername = "controller"
local relayid = -1
local relayname = "relay"

local loading = {"|","/","-","\\"}
local subs = {
    getid = "get id",
    gotid = "got id",
    gotidack = "got id, acknowledge",
    setnuke = "set nuke state",
    getoutput = "get nuke output",
    gotoutput = "got nuke output",
    getnukestate = "get nuke state",
    gotnukestate = "got nuke state",
}

local wireless = "back" -- side
local cable = "front" -- side
local channel = 61234

local refreshrate = .2 -- seconds

local modem = peripheral.wrap(wireless) or error("Unable to wrap wireless modem!",0)
if not modem.isWireless() then error("Wireless modem is required! ("..wireless.." side)",0) end

local wired = peripheral.wrap(cable) or error("Unable to wrap wired modem!",0)
if wired.isWireless() then error("Wired modem is required! ("..cable.." side)",0) end

-- nuke
local nukestate = false -- false=off, true=on
local nukeoutput = 0 -- EU/t

-- mfe
local mfepowerstored = 0 -- EU
local mfepowermax = 0 -- EU
local mfeoutput = 0 -- EU/t
local mfemaxoutput = 0 -- EU
local mfeside = "batbox_1"
local mfe = peripheral.wrap(mfeside) or error("Unable to wrap storage device!",0)

-- monitor
local monitorside = "monitor_1"
local monitor = peripheral.wrap(monitorside) or error("Unable to wrap monitor display!",0)
local monitorw,monitorh = monitor.getSize()

-- computer
local w,h = term.getSize()

--(( Functions ))--

local function setColor(fg,bg,obj)
    local obj = obj or term
    if obj.isColor then
        if obj.isColor() then
            if type(fg) == "number" then
                obj.setTextColor(fg)
            end if type(bg) == "number" then
                obj.setBackground(bg)
            end
        end
    end
end

local function sendMsg(from,to,subject,info)
    info = info or ""
    local message = {
        from = from,
        to = to,
        subject = subject,
        info = info,
    }
    message = textutils.serialize(message)
    modem.transmit(channel,channel,message)
end

local function init()
    modem.open(channel)

    local timeout = 10 -- seconds
    local timeouttimer = os.startTimer(timeout)
    local refreshtimer = os.startTimer(refreshrate)
    local refreshcount = 0

    local x,y = term.getCursorPos()
    local relayknows = false

    repeat
        local ev,p1,p2,p3,p4,p5 = os.pullEvent()

        -- Loading message
        term.setCursorPos(x,y)
        setColor(colors.yellow)
        term.write("Initializing ")
        setColor(colors.orange)
        term.write(loading[(refreshcount%#loading)+1])

        -- Event handling
        if ev == "modem_message" then
            -- Message handling
            local msg = textutils.unserialize(p4)
            if type(msg) == "table" then
                if msg.from == relayname
                and msg.to == controllername then
                    if msg.subject == subs.getid then
                        -- Reply with id
                        sendMsg(controllername,relayname,subs.gotid,controllerid)
                    elseif msg.subject == subs.gotid then
                        if relayid == -1 then
                            -- Save id (into variable)
                            if type(msg.info) == "number" then
                                relayid = msg.info
                                -- Reply with acknowledge confirm
                                sendMsg(controllername,relayname,subs.gotidack)
                            end
                        end
                    elseif msg.subject == subs.gotidack then
                        -- Relay got my id
                        relayknows = true
                    end
                end
            end
        elseif ev == "timer" then
            -- Timer handling
            local timer = p1
            if timer == refreshtimer then
                -- Loading count
                refreshtimer = os.startTimer(refreshrate)
                refreshcount = refreshcount + 1
                -- Ask for id
                if relayid == -1 then
                    sendMsg(controllername,relayname,subs.getid)
                end
            elseif timer == timeouttimer then
                -- Timeout
                term.setCursorPos(x,y)
                setColor(colors.yellow)
                term.write("Initializing ")
                setColor(colors.orange)
                term.write("X")
                print()
                setColor(colors.red)
                error("Operation timed out.",0)
            end
        end
    until relayid ~= -1 and relayknows

    term.setCursorPos(x,y)
    setColor(colors.yellow)
    term.write("Initializing ")
    setColor(colors.orange)
    term.write("X")
    setColor(colors.lime)
    term.write(" Done!")
    print()
end

local function setNuke(state)
    sendMsg(controllername,relayname,subs.setnuke,state)
end
local function getNukeOutput()
    sendMsg(controllername,relayname,subs.getoutput)
end

local function updateMFE()
    -- output
    mfeoutput = mfe.getOutput()
    mfemaxoutput = mfe.getMaxEnergyOutput()

    -- storage
    mfepowerstored = mfe.getStored()
    mfepowermax = mfe.getCapacity()
end

local function drawMonitor()
    -- for debugging, raw info
    local info = {
        mfeoutput = tostring(mfeoutput) .. "/" .. tostring(mfemaxoutput),
        mfepower = tostring(mfepowerstored) .. "/" .. tostring(mfepowermax),
        nukeoutput = tostring(nukeoutput),
    }
    if nukestate then
        info.nukestate = "ON"
    else
        info.nukestate = "OFF"
    end

    local count = 0
    for index,value in pairs(info) do
        monitor.setCursorPos(2,1+count)
        setColor(colors.orange,_,monitor)
        monitor.write(index)
        setColor(colors.red,_,monitor)
        monitor.write(" = ")
        setColor(colors.yellow,_,monitor)
        monitor.write(tostring(value))
        count = count + 1
    end
end

--(( Main program ))--

init()
print("Initiating complete!")

local refreshtimer = os.startTimer(refreshrate)
local refreshcount = 0

while running do
    local ev,p1,p2,p3,p4,p5 = os.pullEvent()

    if ev == "modem_message" then
        -- Message handling
        local msg = textutils.unserialize(p4)
        if type(msg) == "table" then
            if msg.to == controllername then
                if msg.from == relayname then
                    -- Receive nuke state
                    if msg.subject == subs.gotnukestate then
                        if type(msg.info) == "boolean" then
                            nukestate = msg.info
                        end
                    end
                    -- Receive nuke output
                    if msg.subject == subs.gotoutput then
                        if type(msg.info) == "number" then
                            nukeoutput = msg.info
                        end
                    end
                end
            end
        end
    elseif ev == "timer" then
        if p1 == refreshtimer then
            refreshtimer = os.startTimer(refreshrate)
            refreshcount = refreshcount + 1
            setNuke(nukestate)
            getNukeOutput()
            updateMFE()
            drawMonitor()
        end
    end
end

--(( EOF ))--