Residual Decontaminant Degasser Purification Unit

-- Credits -- __cats_____ (cat.)
local component = require("component")
local sides = require("sides")
local event = require("event")
local math = require("math")

-- ========= USER CONFIG =========
-- Redstone receiver used for degasser sensor | side options: {north, east, south, west, top, bottom}
local degassSensorSide = sides.north
local activityDetectorSide = sides.east
-- These sides must be configured for this script to work as expected.

-- The setup requires exactly one Redstone I/O and ME controller. It will not function if there is more than one connected to the OC computer.
-- Its proxy is used for both the 4-bit signal and Activity Detector.

--====== END USER CONFIG ======

-- ====== DECLARATIONS ======
local redstoneIO = component.redstone
local meController = component.me_controller

local fakePatternLabels = {"supercon", "neutronium", "helium", "neon", "krypton", "xenon", "coolant"}
local fakePatterns = {}

local gases = {
  [0] = "helium",
  [1] = "neon",
  [2] = "krypton",
  [3] = "xenon"}

-- ====== END DECLARATIONS ======

local function decimalToBits(value)
    local bits = {}
    for i = 0, 3 do
      bits[i + 1] = math.floor(value / (2 ^ i)) % 2
    end
    return bits
end

local function generatePatternMap()
  local craftables = meController.getCraftables()
  local stack = nil
  local label = nil
  for i, pattern in ipairs(craftables) do
    local success
    success, stack = pcall(pattern.getStack)
    
    if not success then success, stack = pcall(pattern.getItemStack) end
    if not success then error("Item Stack retrieval failed. This has likely occured due to an unexpected API change. Reminder that this script is intended for 2.8.x and 2.9 only.") end
    
    label = stack.label
    print("[Grade 7 Water] Pattern for [" .. label .. "] detected. Adding to fakePatterns...")
    fakePatterns[label] = pattern
  end
end

local function checkBits()
  local degassReading = redstoneIO.getInput(degassSensorSide)
  local bits = decimalToBits(degassReading)
  return bits
end

local function determineActions(bits)
  local actions = {}

  -- Bit 4 overrides EVERYTHING - Override/No injection
  if bits[4] == 1 then
    print("[Grade 7 Water] Bit 4: No fluid will be injected this cycle")
    return nil
  end

  -- No bits enabled - Supercoolant injection
  if bits[1] == 0 and bits[2] == 0 and bits[3] == 0 then
    actions["coolant"] = fakePatterns["coolant"]
    print("[Grade 7 Water] No bits enabled. Injecting super coolant")
    return actions
  end

  -- Bit 1 = Inert Gas Injection 
  if bits[1] == 1 then
    local gas = bits[2] + bits[3]*2
    local gasName = gases[gas]
    
    if gasName then
      actions[gasName] = fakePatterns[gasName]
      print("[Grade 7 Water] Bit 1: Queuing " .. gasName .. " injection")
    end
  end

  -- Bit 2 - Superconductor Injection
  if bits[2] == 1 then
    actions["supercon"] = fakePatterns["supercon"]
    print("[Grade 7 Water] Bit 2: Queuing superconductor injection")
  end

  -- Bit 3 - Neutronium Injection
  if bits[3] == 1 then
    actions["neutronium"] = fakePatterns["neutronium"]
    print("[Grade 7 Water] Bit 3: Queuing neutronium injection")
  end

  return actions
end

local function verifyCraftCompletion(trackers)
  for source, tracker in pairs(trackers) do
    if tracker.isCanceled() then
      print("[Grade 7 Water] Crafting request canceled for pattern: " .. tostring(source))
      print("[Grade 7 Water] Please ensure the ME network has valid crafting storage and enough fluids for a cycle")
      trackers[source] = nil
    elseif tracker.isDone() then
      trackers[source] = nil
    end
  end
end

local function processCycle()
  local bits = checkBits()
  print("[Grade 7 Water] Signal: " .. table.concat(bits))
  local actions = determineActions(bits)
  
  if actions ~= nil then
    local ongoing_crafts = {}
    for source, action in pairs(actions) do
      local progress = action.request(1)
      ongoing_crafts[source] = progress
      os.sleep(0.1)
    end
    
    for i = 1,5 do
      verifyCraftCompletion(ongoing_crafts)
      os.sleep(0.2)
    end
  end
  print("[Grade 7 Water] Current cycle finished. Waiting for next cycle...")
  print("[Grade 7 Water] -------------------------")
end

local function verifyHardware()
  if meController == nil then 
    error("[Grade 7 Water] Missing ME controller")
  end
  
  if redstoneIO == nil then
    error("[Grade 7 Water] Missing Redstone I/O")
  end

  for _, label in ipairs(fakePatternLabels) do
    if fakePatterns[label] == nil then
      error("[Grade 7 Water] Missing pattern detected: " .. label)
    end
  end
  
  print("[Grade 7 Water] Hardware/patterns successfully verified.")
end

local function waitForNewCycle()
  event.pull("redstone_changed", redstoneIO.address, activityDetectorSide)
end

local function startup()
  generatePatternMap()
  verifyHardware()
end

local function main()
  print("[Grade 7 Water] Starting main loop...")
  print("[Grade 7 Water] -------------------------")
  while true do
    waitForNewCycle()
    os.sleep(1)
    processCycle()
  end
end

startup()
main()