Fusion Reactor Setup

-- pastebin get fFxDjyhu Setup.lua && Setup -f -r -p -s

--Importing various APIs
local filesystem = require("filesystem");
local component = require("component");
local term = require("term");
local shell = require("shell");
local computer = require("computer");
local version = "0.2";

--Iterating Through startup parameters
local args = {...};

local uninstall = false;
local forceInstall = false;
local repair = false;
local pastebin = false;
local stayOn = false;

for i,arg in ipairs(args) do
    if arg == "-u" then uninstall = true; end
    if arg == "-f" then forceInstall = true; end
    if arg == "-r" then repair = true; end
    if arg == "-p" then pastebin = true; end
    if arg == "-s" then stayOn = true; end
end

local function pastebinInstall() --Alternative Install through pastebin instead of github
    print("Starting Install...");
    shell.execute("mkdir /home/FusionController")
    shell.execute("pastebin get wJCXfem8 /bin/controller.lua"); --Download Controller
    shell.execute("pastebin get KAR6jmr8 /home/FusionController/API.lua"); --Download Controller API
end

local function run_uninstall() --Uninstall the program.
    print("Uninstalling Reactor Controller...");
    os.sleep(1);
    filesystem.remove("/bin/controller.lua");
    filesystem.remove("/home/FusionController/");
end

term.clear();
print(string.rep("═", 80));
if uninstall then --If uninstall parameter is detected, only uninstall.
    run_uninstall()
    shell.execute("rm Setup.lua");
    print("Finished Uninstalling. Press any key to restart.");
    io.read(); --Wait for keyboard input
    computer.shutdown(true);
else
    if component.isAvailable("nc_fusion_reactor") or forceInstall then --Check if fusion reactor is connected to computer, or if force install is enabled.
        print("Starting Reactor Controller Setup | " .. version);
        if filesystem.exists("/bin/controller.lua") then --Check if program already exists on computer.
            if repair == false then --If repair is enabled, skip asking and uninstall, else ask if they want to uninstall.
                print("Detected controller already on system...");
                print("Would you like to uninstall the Controller first? [Y/n]");
                local answer = io.read();
            else
                answer = "Y"
            end
            if answer == "n" or answer == "N" then --If they do not want to re-install, then close setup.
                print("Closing Setup...");
                os.sleep(1);
                term.clear();
                os.exit();
            else
                run_uninstall();
            end
        end
        if pastebin then --if pastebin parameter, then install through pastebin.
            if not forceInstall then
                print("[WARNING] Pastebin parameter detected. Using pastebin is not the recommended form, since pastes aren't permanent. Would you like to continue [Y/n]");
                local pastebinAnswer = io.read();
            else
                local pastebinAnswer = "y";
            end
            if pastebinAnswer == "n" or pastebinAnswer == "N" then
                print("Closing Setup...");
                os.sleep(1);
                term.clear();
                os.exit();
            end
            pastebinInstall();
        else --if not pastebin install, install through github.
            print("Starting Install...");
            shell.execute("mkdir /home/FusionController")
            print("Downloading Controller...");
            shell.execute("wget https://raw.githubusercontent.com/Corbinhol/Nuclear-Crafter-fusion-reactor-controller/main/Controller.lua /bin/controller.lua -Q");
            print("Downloading API...")
            shell.execute("wget https://raw.githubusercontent.com/Corbinhol/Nuclear-Crafter-fusion-reactor-controller/main/Api.lua /home/FusionController/Api.lua -Q");
        end
        shell.execute("rm Setup.lua"); --remove setup after install (github only.)
        if stayOn then
            print("Closing Setup");
        else
            print("Finished Installing. Press any key to restart.");
            io.read();
            computer.shutdown(true); --Restart computer afterwards.
        end
    else --If Fusion Reactor doesn't exist, then don't run installer (unless force parameter is used)
        print("Error: No fusion reactor found, [Use -f to force install without it]");
        os.exit();
    end
end