PotatOS OmniDisk

--[[
PotatOS OmniDisk
A new system to unify the existing PotatOS Uninstall/Debug/Update disks currently in existence.
Comes with a flexible, modular design, centralized licensing, possibly neater code, and a menu.

This is designed to be executed by the OmniDisk Loader (https://pastebin.com/S1RS76pv) but may run on its own, though this is NOT a supported configuration.

This is NOT usable simply by copying it onto a disk due to PotatOS signing requirements.
You must use the dcopy (https://pastebin.com/TfNgRUKC) program or manually generate a hex-format ECC signature and write it to "disk/signature" (PotatOS will, however, not run it unless this signature is from the PDSK).
]]

local function try_report_incident(...)
    if _G.report_incident then
        _G.report_incident(...)
        print "This incident has been reported."
    end
end

local r = process.get_running()
local sandbox = process.info "sandbox"
if sandbox then
    for _, p in pairs(process.list()) do
        if p.parent == sandbox and p.ID ~= r.ID then
            process.signal(p.ID, process.signals.KILL)
        end
    end
end
pcall(process.signal, "sandbox", process.signals.KILL)
os.queueEvent "stop"

local function fetch(URL)
    local h, e = http.get(URL)
    if not h then error(e) end
    local o = h.readAll()
    h.close()
    return o
end

local UUID = "@UUID@" -- Populated by dcopy utility, in some setups

local args = ...

if type(args) == "table" and args.UUID then UUID = args.UUID end

local json
if _G.json_for_disks_and_such then json = _G.json_for_disks_and_such
elseif textutils.unserialiseJSON then
    json = { encode = textutils.serialiseJSON, decode = textutils.unserialiseJSON }
else error "No JSON library exists, somehow" end
local license_data = fetch "https://pastebin.com/raw/viz0spjb"
local licenses = json.decode(license_data)
local license = licenses[UUID]

local disk_ID
local disk_loader_args = args.arguments
if type(disk_loader_args) == "table" and disk_loader_args.ID then
    disk_ID = disk_loader_args.ID
end

local function runfile(program, ...)
    local ok, err = loadfile(program)
    if not ok then error(err) end
    ok(...)
end

local features = {
    test = {
        fn = function() print "Hello, World!" end,
        description = "Test function."
    },
    exit = {
        fn = function() os.reboot() end,
        description = "Leave OmniDisk, return to PotatOS.",
        always_permitted = true
    },
    UUID = {
        fn = function() print("UUID:", UUID) print("Disk ID:", disk_ID or "[???]") end,
        description = "Print this OmniDisk's Licensing UUID.",
        always_permitted = true
    },
    uninstall = {
        fn = function() print "Uninstalling..." _G.uninstall "omnidisk" end,
        description = "Uninstall potatOS"
    },
    REPL = {
        fn = function() runfile("/rom/programs/shell.lua", "lua") end,
        description = "Open a Lua REPL for debugging."
    },
    shell = {
        fn = function()
            printError "WARNING!"
            print "Do not attempt to modify the code of this PotatOS OmniDisk. Unauthorized attempts to do so will invalidate the signature and make the disk unusable. All code beyond a limited core is stored in an online file to which you do not have write access. Probably. Contact gollark for further information."
            runfile "/rom/programs/shell.lua"
        end,
        description = "Open an unsandboxed shell."
    },
    update = {
        fn = function() runfile("autorun", "update") end,
        description = "Update PotatOS."
    },
    dump_license = {
        fn = function() print(UUID) textutils.pagedPrint(textutils.serialise(license)) end,
        description = "Dump license information."
    },
    primes = {
        fn = function()
            if not _G.findprime or not _G.isprime then
                error "findprime/isprime not available. Update potatOS."
            end
            write "Difficulty? (1-16) "
            local difficulty = tonumber(read())
            if type(difficulty) ~= "number" then error "ERR_PEBKAC\nThat's not a number." end
            local maxrand = math.pow(10, difficulty)
            local p1 = findprime(math.random(2, maxrand))
            local p2 = findprime(math.random(2, maxrand))

            local num = p1 * p2
            print("Please find the prime factors of the following number:", num)
            write "Factor 1: "
            local f1 = tonumber(read())
            write "Factor 2: "
            local f2 = tonumber(read())
            if (f1 == p1 and f2 == p2) or (f2 == p1 and f1 == p2) then
                print "Yay! You got it right!"
            else
                print("Factors", f1, f2, "invalid.", p1, p2, "expected.")
            end
        end,
        description = "Bored? You can factor some semiprimes!"
    },
    potatoplex = {
        fn = function()
            write "Run potatoplex with arguments: "
            local args = read()
            runfile("rom/programs/http/pastebin.lua", "run", "wYBZjQhN", args)
        end,
        description = "Potatoplex your life!"
    },
    chronometer = {
        fn = function()
            runfile("rom/programs/http/pastebin.lua", "run", "r24VMWk4")
        end,
        description = "Tell the time with Chronometer!"
    },
    latest_paste = {
        fn = function()
            write "WARNING: This views the latest paste on Pastebin. Exposure to the raw output of the Internet may be detrimental to your mental health. Do you want to continue (y/n)? "
            local yn = read()
            if not yn:lower():match "y" then return end
            local html = fetch "https://pastebin.com/LW9RFpmY"
            local id = html:match [[<ul class="right_menu"><li><a href="/([A-Za-z0-9]+)">]]
            local url = ("https://pastebin.com/raw/%s"):format(id)
            local title = html:match [[<ul class="right_menu"><li><a href="/[A-Za-z0-9]+">([^<]+)</a>]]
            local content = fetch(url)
            term.clear()
            term.setCursorPos(1, 1)
            textutils.pagedPrint(title .. "\n" .. url .. "\n\n" .. content)
        end,
        description = "View latest paste on Pastebin."
    }
}

local function wait()
    write "Press Any key to continue."
    os.pullEvent "key"
    local timer = os.startTimer(0)
    while true do
        local e, arg = os.pullEvent()
        if (e == "timer" and arg == timer) or e == "char" then return end
    end
end

if not license then
    printError(([[ERR_NO_LICENSE
This disk (UUID %s) does not have an attached license and is invalid.
This should not actually happen, unless you have meddled with the disk while somehow keeping the signature intact.
Please contact gollark.]]):format(tostring(UUID)))
    try_report_incident(("OmniDisk UUID %s has no license data"):format(tostring(UUID)), {"security", "omnidisk"}, {
        extra_meta = {
            disk_ID = disk_ID,
            omnidisk_UUID = UUID
        }
    })
    wait()
    os.reboot()
end

if disk_ID then
    local license_ID = license.disk
    local ok = false
    if type(license_ID) == "table" then
        for _, id in pairs(license_ID) do
            if id == disk_ID then ok = true break end
        end
    elseif type(license_ID) == "number" then
        if license_ID == disk_ID then ok = true end
    else
        ok = true
    end
    if not ok then
        printError(([[ERR_WRONG_DISK
This disk (ID %d) is not (one of) the disk(s) specified in your licensing information.
This license (UUID %s) allows use of this/these disk(s): %s.
If you believe this to be in error, please contact gollark so this can be corrected.
Otherwise, stop cloning disks, or contact gollark to have unique UUIDs issued to each.]]):format(disk_ID, UUID, json.encode(license_ID)))
        try_report_incident(("Disk ID mismatch: %d used with license %s"):format(disk_ID, UUID, json.encode(license_ID)), {"security", "omnidisk"}, {
            extra_meta = {
                permitted_disk_IDs = license_ID,
                disk_ID = disk_ID,
                omnidisk_UUID = UUID
            }
        })
        wait()
        os.reboot()
    end
end

local permitted_feature_lookup = {}
for _, feature in pairs(license.features) do
    permitted_feature_lookup[feature] = true
end

while true do
    term.setCursorPos(1, 1)
    term.clear()

    local usable = {}
    local i = 0

    print [[Welcome to PotatOS OmniDisk!
Available options:]]

    for name, feature in pairs(features) do
        if permitted_feature_lookup["*"] or permitted_feature_lookup[name] or feature.always_permitted then
            textutils.pagedPrint(("%d. %s - %s"):format(i, name, feature.description or "[no description available]"))
            usable[i] = feature.fn
            usable[name] = feature.fn
            i = i + 1
        end
    end

    write "Select an option: "
    local option = read()
    local fn
    local as_num = tonumber(option)

    if as_num then fn = usable[as_num] else fn = usable[option] end
    if not fn then
        printError(("ERR_ID_10T\nPlease select an option which actually exists.\n'%s' doesn't."):format(tostring(option)))
        wait()
    else
        local ok, res = pcall(fn)
        if not ok then
            printError(res)
            wait()
        else
            wait()
        end
    end
end