User setup

-- WARNING: Use this script at your own risk.
-- Ensure you have a backup before running it.

-- Doggy OS First Admin Account Setup Script with Clean UI

-- Function to clear the screen and set up UI
local function setupUI(title, subtitle)
    term.clear()
    term.setCursorPos(1, 1)
    term.setTextColor(colors.white)
    term.setBackgroundColor(colors.black)
    term.clear()

    -- Center the title
    local w, h = term.getSize()
    local x = math.floor((w - #title) / 2)
    term.setCursorPos(x, 2)
    term.setTextColor(colors.cyan)
    print(title)

    -- Subtitle below the title
    local x_subtitle = math.floor((w - #subtitle) / 2)
    term.setCursorPos(x_subtitle, 4)
    term.setTextColor(colors.lightGray)
    print(subtitle)

    -- Draw horizontal line for separation
    term.setCursorPos(1, 6)
    term.setTextColor(colors.gray)
    print(string.rep("-", w))
end

-- Function to draw an input box with grey shading from [ to ]
local function drawInputBox(x, y, width)
    term.setBackgroundColor(colors.gray)  -- Set background color to grey
    term.setTextColor(colors.lightBlue)  -- Set text color to light blue

    -- Draw the left bracket and shaded area
    term.setCursorPos(x, y)
    term.write("[")
    term.write(string.rep(" ", width - 2))  -- Fill with spaces for the shaded area

    -- Draw the right bracket
    term.write("]")

    -- Shadow effect (bottom-right)
    term.setTextColor(colors.lightGray)
    term.setCursorPos(x + width, y)
    term.write(" ")
end

-- Function to get input with a clean input box
local function readWithInputBox(x, y, width, isPassword)
    -- Draw the input box
    drawInputBox(x, y, width)

    -- Move cursor inside the box for typing
    term.setCursorPos(x + 1, y)

    -- Capture input (either password or regular)
    if isPassword then
        return read("*")
    else
        return read()
    end
end

-- Function to delete all directories in /disk/users/ except "root"
local function deleteDirectories()
    local usersPath = "/disk/users/"

    -- Get a list of all directories
    local directories = fs.list(usersPath)

    setupUI("Doggy OS Setup", "Cleaning up old user directories")

    for _, dir in pairs(directories) do
        if dir ~= "root" then
            fs.delete(usersPath .. dir)
        end
    end

    print("\nUnnecessary directories deleted.")
    os.sleep(2)
end

-- Function to create /disk/users/ directory
local function createUsersDirectory()
    local usersPath = "/disk/users/"

    -- Recreate /disk/users/ directory
    fs.makeDir(usersPath)

    setupUI("Doggy OS Setup", "Setting up the users directory")
    print("\nCreated /disk/users/ directory.")
    os.sleep(2)
end

-- Function to generate a recovery key
local function generateRecoveryKey()
    setupUI("Doggy OS Recovery Key Setup", "Generating recovery key")

    -- Generate a random recovery key
    local chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
    local recoveryKey = ""
    for i = 1, 16 do
        local index = math.random(1, #chars)
        recoveryKey = recoveryKey .. chars:sub(index, index)
    end

    -- Display the recovery key to the user
    term.setCursorPos(3, 8)
    print("Recovery Key:")
    term.setCursorPos(3, 9)
    print(recoveryKey)

    -- Save recovery key to the secure file
    local recoveryDir = "/disk/security/"
    fs.makeDir(recoveryDir) -- Ensure the directory exists
    local recoveryFile = fs.open(recoveryDir .. "RecoveryKEY.TXT", "w")
    recoveryFile.write(recoveryKey)
    recoveryFile.close()

    print("\nPlease save this key securely. It will not be shown again.")
    print("\nPress any key to continue...")
    os.pullEvent("key") -- Wait for any key press before proceeding
end

-- Function to set up the first admin account
local function setupFirstAdminAccount()
    setupUI("Doggy OS First Admin Setup", "Please provide Admin credentials")

    -- Enter username
    term.setCursorPos(3, 8)
    print("Admin Username:")
    local username = readWithInputBox(3, 9, 20, false)

    -- Enter password
    term.setCursorPos(3, 12)
    print("Admin Password:")
    local password = readWithInputBox(3, 13, 20, true)

    -- Confirm password
    term.setCursorPos(3, 16)
    print("Confirm Password:")
    local confirmPassword = readWithInputBox(3, 17, 20, true)

    -- Check if passwords match
    while password ~= confirmPassword do
        setupUI("Doggy OS First Admin Setup", "Passwords do not match. Try again.")

        term.setCursorPos(3, 12)
        print("Admin Password:")
        password = readWithInputBox(3, 13, 20, true)

        term.setCursorPos(3, 16)
        print("Confirm Password:")
        confirmPassword = readWithInputBox(3, 17, 20, true)
    end

    -- Store the user details in the directory
    local userPath = "/disk/users/" .. username

    -- Create user directory
    fs.makeDir(userPath)

    -- Store password in a file
    local passwordFile = fs.open(userPath .. "/password.txt", "w")
    passwordFile.write(password)
    passwordFile.close()

    -- Create admin.txt file
    local adminFile = fs.open(userPath .. "/admin.txt", "w")
    adminFile.close()

    setupUI("Doggy OS Setup", "Configuring admin account")
    print("Admin account for '" .. username .. "' has been successfully created.")
    os.sleep(3)
end

-- Function to check for and enable root account if needed
local function enableRootAccount()
    local rootPath = "/disk/users/root/"
    local disabledFile = rootPath .. "disabled.txt"

    if fs.exists(disabledFile) then
        setupUI("Doggy OS Root Access", "Would you like to enable the root account?")
        print("Press 'Y' to enable, 'N' to skip.")
        local input = read()

        if input:lower() == "y" then
            fs.delete(disabledFile)
            setupUI("Doggy OS Root Access", "Root account enabled.")
            print("Root account is now enabled.")

            -- Ask if user wants to reset root password
            setupUI("Doggy OS Root Access", "Would you like to reset the root password?")
            print("Press 'Y' to reset, 'N' to keep the current password.")
            local resetInput = read()

            if resetInput:lower() == "y" then
                -- Prompt for a new root password
                term.setCursorPos(3, 8)
                print("Enter new root password:")
                local newPassword = readWithInputBox(3, 9, 20, true)

                term.setCursorPos(3, 12)
                print("Confirm new root password:")
                local confirmNewPassword = readWithInputBox(3, 13, 20, true)

                -- Confirm password match
                while newPassword ~= confirmNewPassword do
                    setupUI("Doggy OS Root Access", "Passwords do not match. Try again.")
                    term.setCursorPos(3, 8)
                    print("Enter new root password:")
                    newPassword = readWithInputBox(3, 9, 20, true)

                    term.setCursorPos(3, 12)
                    print("Confirm new root password:")
                    confirmNewPassword = readWithInputBox(3, 13, 20, true)
                end

                -- Save the new root password
                local rootPasswordFile = fs.open(rootPath .. "/password.txt", "w")
                rootPasswordFile.write(newPassword)
                rootPasswordFile.close()

                setupUI("Doggy OS Root Access", "Root password reset successfully.")
                print("Root password has been reset.")
                os.sleep(2)
            else
                setupUI("Doggy OS Root Access", "Root password remains unchanged.")
                print("Root password remains unchanged.")
                os.sleep(2)
            end
        else
            setupUI("Doggy OS Root Access", "Root account remains disabled.")
            print("Root account is still disabled.")
            os.sleep(2)
        end
    else
        setupUI("Doggy OS Root Access", "Root account is already enabled.")
        print("Root account is already enabled.")
        os.sleep(2)
    end
end

-- Function to label the computer
local function labelComputer()
    setupUI("Doggy OS Setup", "Labeling your computer")

    term.setCursorPos(3, 8)
    print("Enter a name for this computer:")
    local label = readWithInputBox(3, 9, 20, false)

    -- Set the computer label
    os.setComputerLabel(label)

    setupUI("Doggy OS Setup", "Finalizing setup")
    print("Computer labeled as \"" .. label .. "\".")
    os.sleep(2)
end

-- Main program
deleteDirectories()
createUsersDirectory()
generateRecoveryKey()
setupFirstAdminAccount()
enableRootAccount()  -- Root access enabled before labeling the computer
labelComputer()

-- Run /disk/boot/error instead of /disk/boot/BIOS
setupUI("Running Doggy OS", "Please wait...")
shell.run("/disk/boot/error")