-- Get args
local arg1, arg2, arg3 = ...
-- Startup
print('ReactorControl Engaged. View Monitor.')
local w, h = term.getSize()
-- Vars
emptyflag = 0
offlineflag = 0
flashflag = 0

reactorNameInput = "BigReactors-Reactor_" .. arg1
reactor = peripheral.wrap(reactorNameInput)
if reactor == nil then
   print("Reactor null")
end

monitorSideInput = arg2
monitor = peripheral.wrap(monitorSideInput)
if monitor == nil then
   print("Monitor null")
end

controlRodNumber = arg3
if controlRodNumber == nil then
   print("controlRodNumber null")
end

-- Monitor Settings
monitor.setTextScale(0.5)
monitor.setBackgroundColor(colors.black)

if fs.exists("ProgressBar") then
    print("You have the ProgressBar API")
 else
    print("Downloading API")
    shell.run("pastebin get Rzxj0KKM ProgressBar")
    print("Restarting Program")
    restart()
end

--Load ProgressBar API
os.loadAPI('ProgressBar')
ProgressBar.SetPeripheral(monitorSideInput)

function MakeFuelInfo ()
    fuelMax = reactor.getFuelAmountMax()
    fuelAmount = reactor.getFuelAmount()
    fuelAmountText = math.floor(((reactor.getFuelAmount()/reactor.getFuelAmountMax())*100)+0.5)
    monitor.setCursorPos(6, 1)
    monitor.setTextColor(colors.yellow)
    monitor.write(fuelAmountText..'%')
end

function MakeWasteInfo()
    wasteMax = reactor.getFuelAmountMax()
    wasteAmount = reactor.getWasteAmount()
    wasteAmountText = math.floor(((reactor.getWasteAmount()/reactor.getFuelAmountMax())*100)+0.5)
    monitor.setCursorPos(8, 2)
    monitor.setTextColor(colors.cyan)
    monitor.write(wasteAmountText..'%')
end

function ControlRods(numberOfRods)
local cursorPos = 6
local rodLevelAvg = 0
    --Run for each rod
        for i=1, numberOfRods do
            rodLevelAvg = rodLevelAvg + (reactor.getControlRodLevel(0))
        end
        rodLevelAvg = math.floor(((rodLevelAvg / numberOfRods))+0.5)

        monitor.setTextColor(colors.yellow)
        monitor.setCursorPos(1,3)
        monitor.write('Rod Depths: ')
        monitor.setCursorPos(12,3)
        monitor.write(rodLevelAvg..'%')

end

function GetCasingHeatinfo()

        if reactor.getCasingTemperature()>=650 then
            monitor.setTextColor(colors.purple)
            else if reactor.getCasingTemperature()>=950 then
                monitor.setTextColor(colors.red)
            else
            monitor.setTextColor(colors.green)
            end
        end
        casingTemp = reactor.getCasingTemperature()
        monitor.setCursorPos(9, 4)
        monitor.write(math.floor((casingTemp)+0.5)..' C')

end

 function GetFuelHeatInfo()

        if reactor.getFuelTemperature()>=650 then
          monitor.setTextColor(colors.purple)
          else if reactor.getFuelTemperature()>=950 then
            monitor.setTextColor(colors.red)
          else
         monitor.setTextColor(colors.green)
          end
        end
        fuelTemp = reactor.getFuelTemperature()
        monitor.setCursorPos(8,5)
        monitor.write(math.floor((fuelTemp)+0.5)..' C')
 end

 function GetFluxInfo()
        -- RF Stored
        rfStored = reactor.getEnergyStored()
        rfStoredFormat = rfStored / 1000
        monitor.setCursorPos(9,6)
        monitor.write(math.floor((rfStoredFormat)+0.5)..'K')
        --------------------------------------------------
        -- RF Per Tick
        monitor.setCursorPos(1,7)
        monitor.setTextColor(colors.orange)
        rfPerTick = reactor.getEnergyProducedLastTick()
        --ProgressBar.SetTable("RFPerTick", 10000000, rfPerTick, 2,w-11,15)

        monitor.setCursorPos(1,7)
        monitor.setTextColor(colors.pink)
        monitor.write('RF/t: '..(math.floor(reactor.getEnergyProducedLastTick()+0.5)))
        --------------------------------------------------

 end

-- Main Function
function main()
    while true do
    -- Initial Setup
        monitor.clear()
        monitor.setCursorPos(1,1)
        monitor.setTextColor(colors.white)
    -----------------------------------------------

    -- Make Fuel info
        --Set POS, Color, Title
        monitor.setCursorPos(1,1)
        monitor.setTextColor(colors.yellow)
        monitor.write('Fuel ')
        MakeFuelInfo()
    -----------------------------------------------

    -- Waste Bar
        --Set POS, Color, Title
        monitor.setCursorPos(1,2)
        monitor.setTextColor(colors.lightBlue)
        monitor.write('Waste ')
        MakeWasteInfo()
    -----------------------------------------------

    -- Control Rod
        --Set POS, Color, Title
        monitor.setCursorPos(1,3)
        ControlRods(controlRodNumber)
    -----------------------------------------------

    -- Temps and Heat
        monitor.setCursorPos(1,4)
        monitor.setTextColor(colors.lightGray)
        monitor.write('Casing: ')
        GetCasingHeatinfo()
        --------------------------------------------------
        monitor.setCursorPos(1,5)
        monitor.setTextColor(colors.yellow)
        monitor.write('Fuel: ')
        GetFuelHeatInfo()
    --------------------------------------------------

    -- Flux Information
        monitor.setCursorPos(1,6)
        monitor.setTextColor(colors.yellow)
        monitor.write('Stored: ')
        monitor.setCursorPos(1,6)
        monitor.setTextColor(colors.green)
        GetFluxInfo()
    --------------------------------------------------

        if flashflag==0 then
          flashflag=1
          if offlineflag==1 then
            monitor.setCursorPos(1,8)
            monitor.setTextColor(colors.lightGray)
            monitor.write('OFFLINE [Manual]')
            monitor.setCursorPos(1,9)
            monitor.write('[Manual]')
          end
          if emptyflag==1 then
            monitor.setCursorPos(1,8)
            monitor.setTextColor(colors.pink)
            monitor.write('OFFLINE')
            monitor.setCursorPos(1,9)
            monitor.write('[No Fuel]')
          end
          if emptyflag==0 and offlineflag==0 and reactor.getControlRodLevel(0)>75 then
            monitor.setCursorPos(1,8)
            monitor.setTextColor(colors.yellow)
            monitor.write('ONLINE')
            monitor.setCursorPos(1,9)
            monitor.write('[Low Power]')
          end
          if emptyflag==0 and offlineflag==0 and reactor.getControlRodLevel(0)<=75 then
            monitor.setCursorPos(1,8)
            monitor.setTextColor(colors.orange)
            monitor.write('ONLINE')
            monitor.setCursorPos(1,9)
            monitor.write('[High Power]')
          end
        else
          flashflag=0
          monitor.setCursorPos(1,17)
          monitor.clearLine()
        end

        if reactor.getEnergyStored()<=10000000 and reactor.getEnergyStored()>100 then
            reactor.setAllControlRodLevels(0+(math.floor(reactor.getEnergyStored()/100000)))
        else
            reactor.setAllControlRodLevels(0)
        end

        if reactor.getFuelAmount()<=100 and offlineflag==0 then
            reactor.setAllControlRodLevels(100)
            reactor.setActive(false)
            emptyflag=1
        else
            emptyflag=0
        end

        if rs.getInput('bottom')==false and emptyflag==0 then
            reactor.setActive(true)
            offlineflag=0
        end

        if rs.getInput('bottom')==true and emptyflag==0 then
            reactor.setActive(false)
            reactor.setAllControlRodLevels(100)
            offlineflag=1
        end
    sleep(1)
    end
end

function restart()
    main()
end

-- Reactor Connection Checl
if (reactor.getConnected) then
    main()
end