Advanced and secure Doggy OS Recovery

-- Doggy OS Recovery

local function clearScreen()
    term.clear()
    term.setCursorPos(1, 1)
end

local function setRecoveryPassword(password)
    local file = fs.open("/config.txt", "w")
    file.write(password)
    file.close()
end

local function getRecoveryPassword()
    local file = fs.open("/config.txt", "r")
    local password = file.readAll()
    file.close()
    return password
end

local function checkPassword()
    local password = getRecoveryPassword()

    if password ~= "false" then
        while true do
            clearScreen()
            term.setCursorPos(1, 1)
            print("Enter Recovery Password:")
            local inputPassword = io.read("*l") or ""  -- Use default value to handle possible EOF errors
            if inputPassword == password then
                break
            else
                print("Incorrect password. Try again.")
                sleep(2)
            end
        end
    end
end

local function resetRecoveryPassword()
    local newPassword = io.read("Enter new recovery password: ")
    setRecoveryPassword(newPassword)
    print("Recovery password updated.")
    sleep(2)
end

local function removeRecoveryPassword()
    setRecoveryPassword("false")
    print("Recovery password removed.")
    sleep(2)
end

local function fullSystemRecovery()
    clearScreen()
    shell.run("delete /disk")
    shell.run("copy /recovery /disk")
    print("Full System Recovery completed.")
    sleep(2)
end

local function refreshRecoveryPartition()
    clearScreen()
    print("Please insert Doggy OS disk!")

    while not fs.exists("/disk2/") do
        sleep(1)
    end

    shell.run("delete /recovery")
    shell.run("copy /disk2 /recovery")
    print("Recovery partition refreshed.")
    sleep(2)
end

local function installUpdateFromDisk()
    clearScreen()
    print("Please insert Doggy OS disk!")

    while not fs.exists("/disk2/") do
        sleep(1)
    end

    shell.run("delete /disk")
    shell.run("copy /disk2 /disk")
    print("Update installed from disk.")
    sleep(2)
end

-- Main loop
while true do
    checkPassword()

    clearScreen()
    print("Doggy OS Recovery Menu:")
    print("1. Recovery Options")
    print("2. Power Options")
    print("3. Security")

    local mainChoice = tonumber(io.read())

    if mainChoice == 1 then
        while true do
            clearScreen()
            term.setCursorPos(1, 1)
            print("Recovery Options:")
            print("1. Full System Recovery")
            print("2. Refresh Recovery Partition")
            print("3. Install Update from Disk")
            print("4. Back")

            local recoveryChoice = tonumber(io.read())

            if recoveryChoice == 1 then
                fullSystemRecovery()
            elseif recoveryChoice == 2 then
                refreshRecoveryPartition()
            elseif recoveryChoice == 3 then
                installUpdateFromDisk()
            elseif recoveryChoice == 4 then
                break  -- Go back to the main menu
            else
                print("Invalid choice.")
                sleep(2)
            end
        end
    elseif mainChoice == 2 then
        clearScreen()
        term.setCursorPos(1, 1)
        print("Power Options:")
        print("1. Reboot system")
        print("2. Shutdown system")

        local powerChoice = tonumber(io.read())

        if powerChoice == 1 then
            os.reboot()
        elseif powerChoice == 2 then
            os.shutdown()
        else
            print("Invalid choice.")
            sleep(2)
        end
    elseif mainChoice == 3 then
        while true do
            clearScreen()
            term.setCursorPos(1, 1)
            print("Security Options:")
            print("1. Reset Recovery Password")
            print("2. Remove Recovery Password")
            print("3. Back")

            local securityChoice = tonumber(io.read())

            if securityChoice == 1 then
                resetRecoveryPassword()
            elseif securityChoice == 2 then
                removeRecoveryPassword()
            elseif securityChoice == 3 then
                break  -- Go back to the main menu
            else
                print("Invalid choice.")
                sleep(2)
            end
        end
    else
        print("Invalid choice. Please try again.")
        sleep(2)
    end
end