drmon.lua

local reactorSide, igateName, ofluxSide, monName, oFlow, iFlow, mon, monitor, monX, monY, reactor, outflux, influx, ri, monType, modem, message

local targetStrength = 50
local maxTemperature = 8000
local safeTemperature = 7000
local lowestFieldPercent = 25

local activateOnCharged = 1
local identify = false

-- please leave things untouched from here on
os.loadAPI("lib/f")

local version = "1.0.0"
-- toggleable via the monitor, use our algorithm to achieve our target field strength or let the user tweak it
local autoInputGate = 1

-- last performed action
local action = "None since reboot"
local emergencyCharge = false
local emergencyTemp = false

monitor = f.periphSearch("monitor")
influx = f.periphSearch("flux_gate")
outflux = peripheral.wrap(ofluxSide)
reactor = peripheral.wrap(reactorSide)

monX, monY = monitor.getSize()
mon = {}
mon.monitor,mon.X, mon.Y = monitor, monX, monY
monitor.setBackgroundColor(colors.black)
monitor.clear()

--write settings to config file
function save_config()
  sw = fs.open("config.txt", "w")
  sw.writeLine(version)
  sw.writeLine(monType)
  sw.writeLine(reactorSide)
  sw.writeLine(igateName)
  sw.writeLine(ofluxSide)
  sw.writeLine(monName)
  sw.writeLine(oFlow)
  sw.writeLine(iFlow)
  sw.writeLine(autoInputGate)
  sw.close()
end

--read settings from file
function load_config()
  sr = fs.open("config.txt", "r")
  version = sr.readLine()
  monType = sr.readLine()
  reactorSide = sr.readLine()
  igateName = sr.readLine()
  ofluxSide = sr.readLine()
  monName = sr.readLine()
  oFlow = tonumber(sr.readLine())
  iFlow = tonumber(sr.readLine())
  autoInputGate = tonumber(sr.readLine())
  sr.close()
end

-- 1st time? save our settings, if not, load our settings
if fs.exists("config.txt") == false then
  save_config()
else
  load_config()
end

function buttons()

  while true do
    load_config()
    outflux.setSignalLowFlow(oFlow)
    influx.setSignalLowFlow(iFlow)
    -- button handler
    event, side, xPos, yPos = os.pullEvent("monitor_touch")

    -- output gate controls
    -- 2-4 = -1000, 6-9 = -10000, 10-12,8 = -100000
    -- 17-19 = +1000, 21-23 = +10000, 25-27 = +100000
    if yPos == 8 then
      if xPos >= 2 and xPos <= 4 then
        oFlow = oFlow-1000
      elseif xPos >= 6 and xPos <= 9 then
        oFlow = oFlow-10000
      elseif xPos >= 10 and xPos <= 12 then
        oFlow = oFlow-100000
      elseif xPos >= 17 and xPos <= 19 then
        oFlow = oFlow+100000
      elseif xPos >= 21 and xPos <= 23 then
        oFlow = oFlow+10000
      elseif xPos >= 25 and xPos <= 27 then
        oFlow = oFlow+1000
      end
      outflux.setSignalLowFlow(oFlow)
      save_config()
    end

    -- input gate controls
    -- 2-4 = -1000, 6-9 = -10000, 10-12,8 = -100000
    -- 17-19 = +1000, 21-23 = +10000, 25-27 = +100000
    if yPos == 10 and autoInputGate == 0 then
      if xPos >= 2 and xPos <= 4 then
        iFlow = iFlow-1000
      elseif xPos >= 6 and xPos <= 9 then
        iFlow = iFlow-10000
      elseif xPos >= 10 and xPos <= 12 then
        iFlow = iFlow-100000
      elseif xPos >= 17 and xPos <= 19 then
        iFlow = iFlow+100000
      elseif xPos >= 21 and xPos <= 23 then
        iFlow = iFlow+10000
      elseif xPos >= 25 and xPos <= 27 then
        iFlow = iFlow+1000
      end
      influx.setSignalLowFlow(iFlow)
      save_config()
    end

    -- input gate toggle
    if yPos == 10 and ( xPos == 14 or xPos == 15) then
      autoInputGate = 1
      save_config()
    end
  end
end

function drawButtons(y)

  -- 2-4 = -1000, 6-9 = -10000, 10-12,8 = -100000
  -- 17-19 = +1000, 21-23 = +10000, 25-27 = +100000

  f.draw_text(mon, 2, y, "<", colors.white, colors.gray)
  f.draw_text(mon, 6, y, "<<", colors.white, colors.gray)
  f.draw_text(mon, 10, y, "<<<", colors.white, colors.gray)

  f.draw_text(mon, 17, y, ">>>", colors.white, colors.gray)
  f.draw_text(mon, 21, y, ">>", colors.white, colors.gray)
  f.draw_text(mon, 25, y, ">", colors.white, colors.gray)
end

function pad(str, len, char)
    if char == nil then char = ' ' end
    return string.rep(char, len - #str) .. str
end

function update()
  while true do 
    term.clear()
    term.setCursorPos(1,1)
    ri = reactor.getReactorInfo()
    -- print out all the infos from .getReactorInfo() to term
    if ri == nil then
      error("reactor has an invalid setup")
    end
    for k, v in pairs (ri) do
      if k == "failSafe" then
        print(k.. ": ".. tostring(v))
      else
        print(k.. ": ".. v)
      end
    end
    print("Output Gate: ", outflux.getSignalLowFlow())
    print("Input Gate: ", influx.getSignalLowFlow())
    print("Last Message: ", message)
    -- monitor output
    local statusColor
    statusColor = colors.red
    if ri.status == "running" then
      statusColor = colors.green
    elseif ri.status == "offline" then
      statusColor = colors.gray
    elseif ri.status == "warming_up" then
      statusColor = colors.orange
    end
    f.draw_text_lr(mon, 2, 2, 1, "Reactor Status", pad(string.upper(ri.status),12," "), colors.white, statusColor, colors.black)
    f.draw_text_lr(mon, 2, 4, 1, "Generation", pad(f.format_int(ri.generationRate), 10, " ") .. " rf/t", colors.white, colors.lime, colors.black)
    local tempColor = colors.red
    if ri.temperature <= 5000 then tempColor = colors.green end
    if ri.temperature >= 5000 and ri.temperature <= 6500 then tempColor = colors.orange end
    f.draw_text_lr(mon, 2, 6, 1, "Temperature", pad(f.format_int(ri.temperature),13," ") .. " C", colors.white, tempColor, colors.black)
    f.draw_text_lr(mon, 2, 7, 1, "Output Gate", pad(f.format_int(outflux.getSignalLowFlow()),10," ") .. " rf/t", colors.white, colors.blue, colors.black)
    -- buttons
    drawButtons(8)
    f.draw_text_lr(mon, 2, 9, 1, "Input Gate", pad(f.format_int(influx.getSignalLowFlow()),11," ") .. " rf/t", colors.white, colors.blue, colors.black)
    if autoInputGate == 1 then
      f.draw_text(mon, 14, 10, "AU", colors.green, colors.gray)
    else
      f.draw_text(mon, 14, 10, "MA", colors.white, colors.gray)
      drawButtons(10)
    end
    local satPercent
    satPercent = math.ceil(ri.energySaturation / ri.maxEnergySaturation * 10000)*.01
    f.draw_text_lr(mon, 2, 11, 1, "Energy Saturation", pad(tostring(satPercent),8," ") .. "%", colors.white, colors.white, colors.black)
    f.progress_bar(mon, 2, 12, mon.X-2, satPercent, 100, colors.blue, colors.gray)
    local fieldPercent, fieldColor
    fieldPercent = math.ceil(ri.fieldStrength / ri.maxFieldStrength * 10000)*.01
    fieldColor = colors.red
    if fieldPercent >= 50 then fieldColor = colors.green end
    if fieldPercent < 50 and fieldPercent > 30 then fieldColor = colors.orange end
    if autoInputGate == 1 then 
      f.draw_text_lr(mon, 2, 14, 1, "Field Strength T:" .. targetStrength, fieldPercent .. "%", colors.white, fieldColor, colors.black)
    else
      f.draw_text_lr(mon, 2, 14, 1, "Field Strength", pad(tostring(fieldPercent),6," ") .. "%", colors.white, fieldColor, colors.black)
    end
    f.progress_bar(mon, 2, 15, mon.X-2, fieldPercent, 100, fieldColor, colors.gray)

    local fuelPercent, fuelColor
    fuelPercent = 100 - math.ceil(ri.fuelConversion / ri.maxFuelConversion * 10000)*.01
    fuelColor = colors.red
    if fuelPercent >= 70 then fuelColor = colors.green end
    if fuelPercent < 70 and fuelPercent > 30 then fuelColor = colors.orange end
    f.draw_text_lr(mon, 2, 17, 1, "Fuel ", pad(tostring(fuelPercent),10," ") .. "%", colors.white, fuelColor, colors.black)
    f.progress_bar(mon, 2, 18, mon.X-2, fuelPercent, 100, fuelColor, colors.gray)
    f.draw_text_lr(mon, 2, 19, 1, "Action ", pad(action,20," "), colors.gray, colors.gray, colors.black)
    -- actual reactor interaction
    --
    if emergencyCharge == true then
      reactor.chargeReactor()
    end
    -- are we charging? open the floodgates
    if ri.status == "warming_up" then
      influx.setSignalLowFlow(900000)
      emergencyCharge = false
    end
    -- are we stopping from a shutdown and our temp is better? activate
    if emergencyTemp == true and ri.status == "stopping" and ri.temperature < safeTemperature then
      reactor.activateReactor()
      emergencyTemp = false
    end
    -- are we charged? lets activate
    if ri.status == "warming_up" and activateOnCharged == 1 then
      reactor.activateReactor()
    end
    -- are we on? regulate the input fludgate to our target field strength
    -- or set it to our saved setting since we are on manual
    if ri.status == "running" then
      if autoInputGate == 1 then 
        if ri.fieldStrength < (targetStrength * 1000000) then
          fluxval = ((targetStrength * 1000000) - ri.fieldStrength) + ri.fieldDrainRate * 10  -- Charge ! 
          influx.setSignalLowFlow(fluxval)
        else
          influx.setSignalLowFlow(ri.fieldDrainRate - 1)
        end
      else
        influx.setSignalLowFlow(iFlow)
      end
      save_config()
    end
    -- safeguards
    --
    -- out of fuel, kill it
    if fuelPercent <= 10 then
      reactor.stopReactor()
      action = "Fuel below 10%, refuel"
    end
    -- field strength is too dangerous, kill and it try and charge it before it blows
    if fieldPercent <= lowestFieldPercent and ri.status == "running" then
      action = "Field Str < " ..lowestFieldPercent.."%"
      reactor.stopReactor()
      reactor.chargeReactor()
      emergencyCharge = true
    end
    -- temperature too high, kill it and activate it when its cool
    if ri.temperature > maxTemperature then
      reactor.stopReactor()
      action = "Temp > " .. maxTemperature
      emergencyTemp = true
    end
    sleep(0.1)
  end
end

function patch()
  local installURL = "https://raw.githubusercontent.com/aedan/drmon/master/install.lua"
  install = http.get(installURL)
  installFile = install.readAll()
  local file = fs.open("startup", "w")
  file.write(installFile)
  file.close()
end

function wireless()
  modem = "none"
  local list = peripheral.getNames()
  for i = 1, #list do
    check = peripheral.getMethods(list[i])
    for a = 1, #check do
      if check[a] == "isWireless" then
        test = peripheral.wrap(list[i])
        if test.isWireless() then
          modem = list[i]
        end
      end
    end
  end
  if modem ~= "none" then
    while true do
      if not rednet.isOpen(modem) then
          rednet.open(modem)
      end
      id,message = rednet.receive()
      if message == "reboot" then
        os.reboot()
      end
      if message == "shutdown" then
        reactor.stopReactor()
      end
      if message == "startup" then
        reactor.chargeReactor()
        reactor.activateReactor()
      end
      if message == "checkin" then
        rednet.send(id, "hello")
      end
      if message == "status" then
        rednet.send(id,ri)
      end
      if message == "patch" then
        patch()
        os.reboot()
      end
      if message == "identify" then
        if identify == true then
          monitor.setBackgroundColor(colors.lightBlue)
          monitor.clear()
          identify = false
        else
          monitor.setBackgroundColor(colors.black)
          monitor.clear()
          identify = true
        end
      end
    end
  end
end

parallel.waitForAll(update, buttons, wireless)