startup

monitor_textScale = 0.5

Style = {
    CDefault = colors.white,
    BGDefault = colors.blue,

    CTitle = colors.black,
    BGTitle = colors.cyan,

    CWarning = colors.white,
    BGWarning = colors.red,

    CSuccess = colors.white,
    BGSuccess = colors.lime,

    CDisabled = colors.gray,
    BGDisabled = colors.blue,

    CRadarmap = colors.gray,
    BGRadarmap = colors.green,

    CRadarborder = colors.white,
    BGRadarborder = colors.black,

    CRadarself = colors.white,
    BGRadarself = colors.lime,
    TextRadarself = "R",

    CRadarother = colors.black,
    BGRadarother = colors.red,
    TextRadarother = "#"
}

if not term.isColor() then
    print("Advanced computer required")
    exit()
end

----------- Monitor support

function SetMonitorColorFrontBack(frontColor, backgroundColor)
    term.setBackgroundColor(backgroundColor)
    term.setTextColor(frontColor)
end

function Write(text)
    term.write(text)
end

function SetCursorPos(x, y)
    term.setCursorPos(x, y)
end

function SetColorDefault()
    SetMonitorColorFrontBack(Style.CDefault, Style.BGDefault)
end

function SetColorTitle()
    SetMonitorColorFrontBack(Style.CTitle, Style.BGTitle)
end

function SetColorWarning()
    SetMonitorColorFrontBack(Style.CWarning, Style.BGWarning)
end

function SetColorSuccess()
    SetMonitorColorFrontBack(Style.CSuccess, Style.BGSuccess)
end

function SetColorDisabled()
    SetMonitorColorFrontBack(Style.CDisabled, Style.BGDisabled)
end

function SetColorRadarmap()
    SetMonitorColorFrontBack(Style.CRadarmap, Style.BGRadarmap)
end

function SetColorRadarborder()
    SetMonitorColorFrontBack(Style.CRadarborder, Style.BGRadarborder)
end

function Clear()
    clearWarningTick = -1
    SetColorDefault()
    term.clear()
    SetCursorPos(1, 1)
end

function ClearLine()
    SetColorDefault()
    term.clearLine()
    SetCursorPos(1, 1)
end

function WriteLn(Text)
    Write(Text)
    local x, y = term.getCursorPos()
    local width, height = term.getSize()
    if y > height - 1 then
        y = 1
    end
    SetCursorPos(1, y + 1)
end

function WriteCentered(y, text)
    SetCursorPos((51 - text:len()) / 2, y)
    term.write(text)
    local xt, yt = term.getCursorPos()
    SetCursorPos(1, yt + 1)
end

function ShowTitle(text)
    Clear()
    SetColorTitle()
    WriteCentered(1, text)
    SetColorDefault()
end

function ShowMenu(text)
    Write(text)
    local xt, yt = term.getCursorPos()
    for i = xt, 51 do
        Write(" ")
    end
    SetCursorPos(1, yt + 1)
end

local clearWarningTick = -1
function ShowWarning(text)
    SetColorWarning()
    SetCursorPos((51 - text:len() - 2) / 2, 19)
    Write(" " .. text .. " ")
    SetColorDefault()
    clearWarningTick = 5
end
function ClearWarning(text)
    if clearWarningTick > 0 then
        clearWarningTick = clearWarningTick - 1
    elseif clearWarningTick == 0 then
        SetColorDefault()
        SetCursorPos(1, 19)
        ClearLine()
    end
end

----------- Formatting & popups

function FormatFloat(value, nbchar)
    local str = "?"
    if value ~= nil then
        str = string.format("%f", value)
    end
    if nbchar ~= nil then
        str = string.sub("               " .. str, -nbchar)
    end
    return str
end
function FormatInteger(value, nbchar)
    local str = "?"
    if value ~= nil then
        str = string.format("%d", value)
    end
    if nbchar ~= nil then
        str = string.sub("               " .. str, -nbchar)
    end
    return str
end

function boolToYesNo(bool)
    if bool then
        return "YES"
    else
        return "no"
    end
end

function readInputNumber(currentValue)
    local inputAbort = false
    local input = string.format(currentValue)
    if input == "0" then
        input = ""
    end
    local x, y = term.getCursorPos()
    repeat
        ClearWarning()
        SetColorDefault()
        SetCursorPos(x, y)
        Write(input .. "            ")
        input = string.sub(input, -9)

        local params = { os.pullEventRaw() }
        local eventName = params[1]
        local side = params[2]
        if side == nil then side = "none" end
        if eventName == "key" then
            local keycode = params[2]
            if keycode >= 2 and keycode <= 10 then -- 1 to 9
                input = input .. string.format(keycode - 1)
            elseif keycode == 11 or keycode == 82 then -- 0 & keypad 0
                input = input .. "0"
            elseif keycode >= 79 and keycode <= 81 then -- keypad 1 to 3
                input = input .. string.format(keycode - 78)
            elseif keycode >= 75 and keycode <= 87 then -- keypad 4 to 6
                input = input .. string.format(keycode - 71)
            elseif keycode >= 71 and keycode <= 73 then -- keypad 7 to 9
                input = input .. string.format(keycode - 64)
            elseif keycode == 14 then -- Backspace
                input = string.sub(input, 1, string.len(input) - 1)
            elseif keycode == 211 then -- Delete
                input = ""
            elseif keycode == 28 then -- Enter
                inputAbort = true
            else
                ShowWarning("Key " .. keycode .. " is invalid")
            end
        elseif eventName == "char" then
            -- drop it
        elseif eventName == "key_up" then
            -- drop it
        elseif eventName == "terminate" then
            inputAbort = true
        elseif not common_event(eventName, params[2]) then
            ShowWarning("Event '" .. eventName .. "', " .. side .. " is unsupported")
        end
    until inputAbort
    SetCursorPos(1, y + 1)
    if input == "" then
        return currentValue
    else
        return tonumber(input)
    end
end

function readInputText(currentValue)
    local inputAbort = false
    local input = string.format(currentValue)
    local x, y = term.getCursorPos()
    repeat
        ClearWarning()
        SetColorDefault()
        SetCursorPos(x, y)
        Write(input .. "                              ")
        input = string.sub(input, -30)

        local params = { os.pullEventRaw() }
        local eventName = params[1]
        local side = params[2]
        if side == nil then side = "none" end
        if eventName == "key" then
            local keycode = params[2]
            if keycode == 14 then -- Backspace
                input = string.sub(input, 1, string.len(input) - 1)
            elseif keycode == 211 then -- Delete
                input = ""
            elseif keycode == 28 then -- Enter
                inputAbort = true
            else
                -- ShowWarning("Key " .. keycode .. " is invalid")
            end
        elseif eventName == "char" then
            local char = params[2]
            if char >= ' ' and char <= '~' then -- 1 to 9
                input = input .. char
            else
                ShowWarning("Char #" .. string.byte(char) .. " is invalid")
            end
        elseif eventName == "key_up" then
            -- drop it
        elseif eventName == "terminate" then
            inputAbort = true
        elseif not common_event(eventName, params[2]) then
            ShowWarning("Event '" .. eventName .. "', " .. side .. " is unsupported")
        end
    until inputAbort
    SetCursorPos(1, y + 1)
    if input == "" then
        return currentValue
    else
        return input
    end
end

function readConfirmation()
    ShowWarning("Are you sure? (y/n)")
    repeat
        local params = { os.pullEventRaw() }
        local eventName = params[1]
        local side = params[2]
        if side == nil then side = "none" end
        if eventName == "key" then
            local keycode = params[2]
            if keycode == 21 then -- Y
                return true
            else
                return false
            end
        elseif eventName == "char" then
            -- drop it
        elseif eventName == "key_up" then
            -- drop it
        elseif eventName == "terminate" then
            return false
        elseif not common_event(eventName, params[2]) then
            ShowWarning("Event '" .. eventName .. "', " .. side .. " is unsupported")
        end
    until false
end

----------- commons: menu, event handlers, etc.

function common_event(eventName, param)
    if eventName == "redstone" then
        redstone_event(param)
    elseif eventName == "timer" then
    elseif eventName == "reactorPulse" then
        reactor_pulse(param)
--  elseif eventName == "reactorDeactivation" then
--      ShowWarning("Reactor deactivated")
--  elseif eventName == "reactorActivation" then
--      ShowWarning("Reactor activated")
    else
        return false
    end
    return true
end

function menu_common()
    SetCursorPos(1, 18)
    SetColorTitle()
    ShowMenu("0 Connections, 1 Reactor, X Exit")
end

----------- Redstone support

local tblRedstoneState = {-- Remember redstone state on each side
    ["top"] = rs.getInput("top"),
    ["front"] = rs.getInput("front"),
    ["left"] = rs.getInput("left"),
    ["right"] = rs.getInput("right"),
    ["back"] = rs.getInput("back"),
    ["bottom"] = rs.getInput("bottom"),
}
local tblSides = {-- list all sides and offset coordinates
    ["top"   ] = { 3, 1},
    ["front" ] = { 1, 3},
    ["left"  ] = { 3, 3},
    ["right" ] = { 5, 3},
    ["back"  ] = { 5, 5},
    ["bottom"] = { 3, 5},
}

function redstone_event()
    -- Event only returns nil so we need to check sides manually
    local message = ""
    for side, state in pairs(tblRedstoneState) do
        if rs.getInput(side) ~= state then
            -- print(side .. " is now " .. tostring(rs.getInput(side)))
            message = message .. side .. " "
            tblRedstoneState[side] = rs.getInput(side)
        end
    end
    if message ~= "" then
        message = "Redstone changed on " .. message
        showWarning(message)
    end
end

----------- Configuration

function data_save()
    local file = fs.open("shipdata.txt", "w")
    file.writeLine(textutils.serialize(data))
    file.close()
end

function data_read()
    if fs.exists("shipdata.txt") then
        local file = fs.open("shipdata.txt", "r")
        data = textutils.unserialize(file.readAll())
        file.close()
    else
        data = { }
    end
    if data.reactor_mode == nil then data.reactor_mode = 0; end
    if data.reactor_rate == nil then data.reactor_rate = 100; end
    if data.reactor_targetStability == nil then data.reactor_targetStability = 50; end
    if data.reactor_laserAmount == nil then data.reactor_laserAmount = 10000; end
end

function data_setName()
    ShowTitle("<==== Set name ====>")

    SetCursorPos(1, 2)
    Write("Enter ship name: ")
    label = readInputText(label)
    os.setComputerLabel(label)
    os.reboot()
end

----------- Reactor support

reactor_output = 0

function reactor_boot()
    if reactor ~= nil then
        WriteLn("Booting Reactor...")
        local isActive, strMode, releaseRate = reactor.active()
        if strMode == "OFF" then
            data.reactor_mode = 0
        elseif strMode == "MANUAL" then
            data.reactor_mode = 1
        elseif strMode == "ABOVE" then
            data.reactor_mode = 2
        elseif strMode == "RATE" then
            data.reactor_mode = 3
        else
            data.reactor_mode = 0
        end
    end
end

function reactor_key(keycode)
    if keycode == 31 then -- S
        reactor_start()
        return true
    elseif keycode == 25 then -- P
        reactor_stop()
        return true
    elseif keycode == 38 then -- L
        reactor_laser()
        return true
    elseif keycode == 24 then -- O
        data.reactor_mode = (data.reactor_mode + 1) % 4
        reactor_setMode()
        data_save()
        return true
    elseif keycode == 34 then -- G
        data.reactor_rate = data.reactor_rate / 10
        reactor_setMode()
        data_save()
        return true
    elseif keycode == 20 then -- T
        data.reactor_rate = data.reactor_rate * 10
        reactor_setMode()
        data_save()
        return true
    elseif keycode == 36 then -- J
        data.reactor_laserAmount = data.reactor_laserAmount / 10
        reactor_setLaser()
        data_save()
        return true
    elseif keycode == 22 then -- U
        data.reactor_laserAmount = data.reactor_laserAmount * 10
        reactor_setLaser()
        data_save()
        return true
    elseif keycode == 74 then -- -
        data.reactor_targetStability = data.reactor_targetStability - 1
        reactor_setTargetStability()
        data_save()
        return true
    elseif keycode == 78 then -- +
        data.reactor_targetStability = data.reactor_targetStability + 1
        reactor_setTargetStability()
        data_save()
        return true
    elseif keycode == 46 then -- C
        reactor_config()
        data_save()
        return true
    end
    return false
end

function reactor_page()
    ShowTitle(label .. " - Reactor status")

    SetCursorPos(1, 2)
    if reactor == nil then
        SetColorDisabled()
        Write("Reactor not detected")
    else
        SetColorDefault()
        Write("Reactor stability")
        instabilities = { reactor.instability() }
        average = 0
        for key,instability in pairs(instabilities) do
            SetCursorPos(12, 2 + key)
            stability = math.floor((100.0 - instability) * 10) / 10
            if stability >= data.reactor_targetStability then
                SetColorSuccess()
            else
                SetColorWarning()
            end
            Write(FormatFloat(stability, 5) .. " %")
            average = average + instability
        end
        average = average / #instabilities

        SetColorDefault()
        local energy = { reactor.energy() }
        SetCursorPos(1, 7)
        Write("Energy   : ")
        if energy[1] ~= nil then
            Write(FormatInteger(energy[1], 10) .. " / " .. energy[2] .. " RF +" .. FormatInteger(reactor_output, 5) .. " RF/t")
        else
            Write("???")
        end
        SetCursorPos(1, 8)
        Write("Outputing: ")
        if energy[1] ~= nil then
            Write(energy[3] .. " RF/t")
        end

        SetColorDefault()
        SetCursorPos(1, 9)
        Write("Activated: ")
        isActive = reactor.active()
        if isActive then SetColorSuccess() else SetColorDefault() end
        Write(boolToYesNo(isActive))
    end

    if #reactorlasers == 0 then
        SetColorDisabled()
        SetCursorPos(30, 2)
        Write("Lasers not detected")
    else
        SetColorDefault()
        SetCursorPos(30, 2)
        Write("Lasers")

        for key,reactorlaser in pairs(reactorlasers) do
            local side = reactorlaser.side()
            if side ~= nil then
                side = side % 4
                SetColorDefault()
                SetCursorPos(4, 3 + side)
                Write("Side " .. side .. ":")
                SetCursorPos(30, 3 + side)
                local energy = reactorlaser.energy()
                if not reactorlaser.hasReactor() then
                    SetColorDisabled()
                elseif energy > 3 * data.reactor_laserAmount then
                    SetColorSuccess()
                else
                    SetColorWarning()
                end
                Write(FormatInteger(reactorlaser.energy(), 6))
            end
        end
    end

    SetColorDefault()
    SetCursorPos(1, 10)
    Write("  -----------------------------------------------")
    SetCursorPos(1, 11)
    Write("Output mode     : ")
    if data.reactor_mode == 0 then
        SetColorDisabled()
        Write("hold")
    elseif data.reactor_mode == 1 then
        Write("manual/unlimited")
    elseif data.reactor_mode == 2 then
        Write("surplus above " .. data.reactor_rate .. " RF")
    else
        Write("rated at " .. data.reactor_rate .. " RF")
    end
    SetColorDefault()
    SetCursorPos( 1, 12)
    Write("Target stability: " .. data.reactor_targetStability .. "%")
    SetCursorPos(30, 12)
    Write("Laser amount: " .. data.reactor_laserAmount)

    SetColorTitle()
    SetCursorPos(1, 14)
    ShowMenu("S - Start reactor, P - Stop reactor, L - Use lasers")
    SetCursorPos(1, 15)
    ShowMenu("O - Output mode, C - Configuration")
    SetCursorPos(1, 16)
    ShowMenu("+/- - Target stability, U/J - Laser amount")
    SetCursorPos(1, 17)
    ShowMenu("G/T - Output rate/threshold")
end

function reactor_setMode()
    if data.reactor_rate < 1 then
        data.reactor_rate = 1
    elseif data.reactor_rate > 100000 then
        data.reactor_rate = 100000
    end
    if reactor ~= nil then
        if data.reactor_mode == 0 then
            reactor.release(false)
        elseif data.reactor_mode == 1 then
            reactor.release(true)
        elseif data.reactor_mode == 2 then
            reactor.releaseAbove(data.reactor_rate)
        else
            reactor.releaseRate(data.reactor_rate)
        end
    end
end

function reactor_setLaser()
    if data.reactor_laserAmount < 1 then
        data.reactor_laserAmount = 1
    elseif data.reactor_laserAmount > 100000 then
        data.reactor_laserAmount = 100000
    end
end

function reactor_setTargetStability()
    if data.reactor_targetStability < 1 then
        data.reactor_targetStability = 1
    elseif data.reactor_targetStability > 100 then
        data.reactor_targetStability = 100
    end
end

function reactor_start()
    if reactor ~= nil then
        reactor_setMode()
        reactor.active(true)
    end
end

function reactor_stop()
    if reactor ~= nil then
        reactor.active(false)
    end
end

function reactor_laser(side)
    for key,reactorlaser in pairs(reactorlasers) do
        if (side == nil) or (reactorlaser.side() == side) then
            reactorlaser.stabilize(data.reactor_laserAmount)
        end
    end
end

function reactor_pulse(output)
    reactor_output = output
    if reactor == nil then
        os.reboot()
    end
    local instabilities = { reactor.instability() }
    for key,instability in pairs(instabilities) do
        local stability = math.floor((100.0 - instability) * 10) / 10
        if stability < data.reactor_targetStability then
            reactor_laser(key - 1)
        end
    end
end

function reactor_config()
    ShowTitle(label .. " - Reactor configuration")

    SetCursorPos(1, 2)
    if reactor == nil then
        SetColorDisabled()
        Write("Reactor not detected")
    else
        SetColorDefault()
        SetCursorPos(1, 4)
        Write("Reactor output rate (" .. data.reactor_rate .. " RF): ")
        data.reactor_rate = readInputNumber(data.reactor_rate)
        reactor_setMode()
        SetCursorPos(1, 5)
        Write("Reactor output rate set")

        SetCursorPos(1, 7)
        Write("Laser energy level (" .. data.reactor_laserAmount .. "): ")
        data.reactor_laserAmount = readInputNumber(data.reactor_laserAmount)
        reactor_setLaser()
        SetCursorPos(1, 8)
        Write("Laser energy level set")

        SetCursorPos(1, 10)
        Write("Reactor target stability (" .. data.reactor_targetStability .. "%): ")
        data.reactor_targetStability = readInputNumber(data.reactor_targetStability)
        reactor_setTargetStability()
        SetCursorPos(1, 11)
        Write("Reactor target stability set")
    end
end

----------- Boot sequence
label = os.getComputerLabel()
if not label then
    label = "" .. os.getComputerID()
end

-- read configuration
data_read()

-- initial scanning
monitors = {}
ShowTitle(label .. " - Connecting...")
WriteLn("")

sides = peripheral.getNames()
reactor = nil
reactorlasers = {}
for key,side in pairs(sides) do
    sleep(0)
    if peripheral.getType(side) == "warpdriveEnanReactorCore" then
        WriteLn("Wrapping " .. side)
        reactor =  peripheral.wrap(side)
    elseif peripheral.getType(side) == "warpdriveEnanReactorLaser" then
        WriteLn("Wrapping " .. side)
        table.insert(reactorlasers, peripheral.wrap(side))
    end
end
-- sleep(1)

if not os.getComputerLabel() and reactor ~= nil then
    data_setName()
end

-- peripherals status
function connections_page()
    ShowTitle(label .. " - Connections")

    WriteLn("")
    if #monitors == 0 then
        SetColorDisabled()
        WriteLn("No Monitor detected")
    end

    if reactor == nil then
        SetColorDisabled()
        WriteLn("No Enantiomorphic reactor detected")
    else
        SetColorSuccess()
        WriteLn("Enantiomorphic reactor detected")
    end

    if #reactorlasers == 0 then
        SetColorDisabled()
        WriteLn("No reactor stabilisation laser detected")
    elseif #reactorlasers == 1 then
        SetColorSuccess()
        WriteLn("1 reactor stabilisation laser detected")
    else
        SetColorSuccess()
        WriteLn(#reactorlasers .. " reactor stabilisation lasers detected")
    end
end

-- peripheral boot up
Clear()
connections_page()
SetColorDefault()
WriteLn("")
sleep(0.5)
reactor_boot()

-- main loop
abort = false
refresh = false
page = reactor_page
keyHandler = reactor_key
tickRefresh = 10
repeat
    ClearWarning()
    if refresh and tickRefresh == 0 then
        Clear()
        page()
        menu_common()
        refresh = false
        tickRefresh = 10
    else
        tickRefresh = tickRefresh - 1
    end
    eventName, var = os.pullEventRaw()

    if side == nil then side = "none" end
    if eventName == "key" then
        keycode = var
        if keycode == 45 then -- x for eXit
            os.pullEventRaw()
            abort = true
        elseif keycode == 11 or keycode == 82 then -- 0
            page = connections_page
            keyHandler = nil
            refresh = true
        elseif keycode == 2 or keycode == 79 then -- 1
            page = reactor_page
            keyHandler = reactor_key
            refresh = true
        elseif keyHandler ~= nil and keyHandler(keycode) then
            refresh = true
            os.sleep(0.2)
        else
            ShowWarning("Key " .. keycode .. " is invalid")
            os.sleep(0.2)
        end
    elseif eventName == "reactorPulse" then
        reactor_pulse(var)
        refresh = (page == reactor_page)
    elseif eventName == "terminate" then
        abort = true
    elseif not common_event(eventName, var) then
        ShowWarning("Event '" .. eventName .. "', " .. side .. " is unsupported")
        refresh = true
        os.sleep(0.2)
    end
until abort

-- clear screens on exit
SetMonitorColorFrontBack(colors.white, colors.black)
term.clear()
SetCursorPos(1, 1)
Write("")