-- Function to draw a border
local function drawBorder()
    term.clear()
    local w, h = term.getSize()
    for y = 1, h do
        term.setCursorPos(1, y)
        term.write("|")
        term.setCursorPos(w, y)
        term.write("|")
    end
    for x = 1, w do
        term.setCursorPos(x, 1)
        term.write("-")
        term.setCursorPos(x, h)
        term.write("-")
    end
    term.setCursorPos(1, 1)
    term.write("+")
    term.setCursorPos(w, 1)
    term.write("+")
    term.setCursorPos(1, h)
    term.write("+")
    term.setCursorPos(w, h)
    term.write("+")
end

-- Function to center text on the screen
local function centerText(text, y)
    local w, _ = term.getSize()
    local x = math.floor((w - #text) / 2) + 1
    term.setCursorPos(x, y)
    term.write(text)
end

-- Function to show UEFI options menu
local function showMenu()
    drawBorder()
    local _, h = term.getSize()
    centerText("Doggy OS UEFI Options", math.floor(h / 2) - 3)
    centerText("1. Recovery", math.floor(h / 2) - 1)
    centerText("2. Bootable devices", math.floor(h / 2))
    centerText("3. Device Info", math.floor(h / 2) + 1)
    centerText("4. Shutdown", math.floor(h / 2) + 2)
end

-- Function to list bootable devices
local function listBootableDevices()
    drawBorder()
    local _, h = term.getSize()
    centerText("Select Bootable Device", math.floor(h / 2) - 2)

    local devices = {}
    for _, side in ipairs({"top", "bottom", "left", "right", "front", "back"}) do
        if disk.isPresent(side) then
            local mountPath = disk.getMountPath(side)
            if fs.exists(mountPath) then
                for _, file in ipairs(fs.list(mountPath)) do
                    if file:match("^startup") then
                        table.insert(devices, side)
                        break
                    end
                end
            end
        end
    end

    for i, device in ipairs(devices) do
        centerText(i .. ". " .. device, math.floor(h / 2) - 1 + i)
    end

    if #devices == 0 then
        centerText("No bootable devices found", math.floor(h / 2))
        os.sleep(2)
        showMenu()
    else
        term.setCursorPos(1, math.floor(h / 2) + #devices + 1)
        term.write("Select a device number: ")
        local choice = tonumber(read())
        if choice and devices[choice] then
            shell.run(disk.getMountPath(devices[choice]) .. "/startup.lua")
        else
            listBootableDevices()
        end
    end
end

-- Function to show device information
local function showDeviceInfo()
    term.setBackgroundColor(colors.black)
    term.clear()
    term.setCursorPos(1, 1)

    -- Define colors
    local bgColor = colors.black
    local textColor = colors.white
    local headerColor = colors.blue
    local infoColor = colors.green

    -- Function to print a header with a background color
    local function printHeader(text)
        term.setTextColor(headerColor)
        term.setBackgroundColor(bgColor)
        print(text)
        term.setTextColor(textColor)
    end

    -- Function to print info with colors
    local function printInfo(label, value)
        term.setTextColor(infoColor)
        term.write(label .. ": ")
        term.setTextColor(textColor)
        print(value)
    end

    -- Display ASCII art
    term.setTextColor(headerColor)
    print("     |\\_/|")
    print("     | @ @   DEVICE INFORMATION ")
    print("     |   <>              _  ")
    print("     |  _/\\------____ ((| |))")
    print("     |               `--' |   ")
    print(" ____|_       ___|   |___.' ")
    print("/_/_____/____/_______|")

    -- Function to read config file using Lua's loadfile
    local function readConfig()
        local configFilePath = "/disk/bootloader/system.config"
        if fs.exists(configFilePath) then
            local config = loadfile(configFilePath)()
            if type(config) == "table" then
                return config
            else
                return {}  -- Return an empty table if the format is incorrect
            end
        end
        return {}  -- Return an empty table if the file does not exist
    end

    -- Function to get the device info
    local function getDeviceInfo()
        local config = readConfig()  -- Read config from Lua file
        local id = os.getComputerID() or "Unknown ID"
        local label = os.getComputerLabel() or "No Label"
        local osVersion = config.os_version or "Unknown OS Version"
        local bootloaderVersion = config.bootloader_version or "Unknown Bootloader Version"
        local securityPatch = config.security_patch or "Unknown Security Patch"
        local recoveryPartition = fs.exists("/recovery/") and "Recovery partition exists" or "No Recovery partition"

        return {
            id = id,
            label = label,
            osVersion = osVersion,
            bootloaderVersion = bootloaderVersion,
            securityPatch = securityPatch,
            recoveryPartition = recoveryPartition
        }
    end

    -- Get device info
    local info = getDeviceInfo()

    -- Display device info with better UI
    printHeader("Computer Information:")
    printInfo("Computer ID", info.id)
    printInfo("Computer Label", info.label)
    printHeader("Software Information:")
    printInfo("Doggy OS Version", info.osVersion)
    printInfo("VA11-ILLA Bootloader Version", info.bootloaderVersion)
    printInfo("Security Patch Version", info.securityPatch)
    printHeader("System Information:")
    printInfo("Recovery Partition Status", info.recoveryPartition)

    -- Prompt and wait for Enter key to reboot
    term.setTextColor(colors.yellow)
    term.setTextColor(textColor)
    local input = read()
    if input == "" then
        -- Reboot the computer
        os.reboot()
    end
end

-- Function to handle menu selection
local function handleSelection()
    while true do
        local event, key = os.pullEvent("key")
        if key == keys.one then
            shell.run("/disk/boot/Recovery.lua")
        elseif key == keys.two then
            listBootableDevices()
        elseif key == keys.three then
            showDeviceInfo()
        elseif key == keys.four then
            os.shutdown()
        end
    end
end

-- Function to verify user credentials
local function verifyCredentials(username, password)
    local userPath = "/disk/users/" .. username
    if fs.exists(userPath .. "/admin.txt") and fs.exists(userPath .. "/password.txt") then
        local file = fs.open(userPath .. "/password.txt", "r")
        local storedPassword = file.readAll()
        file.close()
        return storedPassword == password
    end
    return false
end

-- Function to authenticate user
local function authenticate()
    term.clear()
    drawBorder()
    local _, h = term.getSize()
    centerText("Doggy OS UEFI Authentication", 3)

    local w, _ = term.getSize()
    local usernameX = math.floor(w / 2) - 10

    term.setCursorPos(usernameX, 5)
    term.write("Username: ")
    term.setCursorPos(usernameX + 10, 5)
    local username = read()

    term.setCursorPos(usernameX, 7)
    term.write("Password: ")
    term.setCursorPos(usernameX + 10, 7)
    local password = read("*") -- Masked input for password

    if verifyCredentials(username, password) then
        showMenu()
        handleSelection()
    else
        centerText("Invalid credentials. Access denied.", 10)
        os.sleep(2)
        os.reboot()
    end
end

-- Start authentication process
authenticate()