-- Doggy OS UEFI

-- 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 the UEFI message
local function showUEFIMessage()
    drawBorder()
    local _, h = term.getSize()
    centerText("Doggy OS UEFI", math.floor(h / 2) - 1)
    centerText("Press Enter for boot options", math.floor(h / 2) + 1)
end

-- Function to load the boot animation
local function loadBootAnimation()
    if fs.exists("/disk/boot/boot-animation") then
        shell.run("/disk/boot/boot-animation")
    else
        print("Boot animation not found!")
    end
end

-- Function to load the boot options
local function loadBootOptions()
    if fs.exists("/disk/boot/boot-options") then
        shell.run("/disk/boot/boot-options")
    else
        print("Boot options not found!")
    end
end

-- Function to show the Doggy OS Authentication screen
local function showAuthScreen()
    drawBorder()
    local _, h = term.getSize()
    centerText("Doggy OS Authentication", math.floor(h / 2) - 1)
    centerText("F10 has been pressed", math.floor(h / 2) + 1)
end

-- Function to check the password
local function checkPassword()
    local passwordFile = "/disk/security/poweron.pass"
    if fs.exists(passwordFile) then
        local file = fs.open(passwordFile, "r")
        local correctPassword = file.readAll()
        file.close()

        while true do
            drawBorder()
            local _, h = term.getSize()
            centerText("Embedded Security Password", math.floor(h / 2) - 2)
            centerText("Enter Password:", math.floor(h / 2))
            term.setCursorPos(math.floor((term.getSize() / 2)), math.floor(h / 2) + 1)
            local password = read("*")  -- Read the password, hiding input

            if password == correctPassword then
                return true
            else
                local timeoutFile = "/FW-Security/BIOS/Boot/TIMEOUT"
                local f = fs.open(timeoutFile, "w")
                f.close()
                return false
            end
        end
    end
    return true
end

-- Function to handle timeout
local function handleTimeout()
    local timeoutFile = "/FW-Security/BIOS/Boot/TIMEOUT"
    if fs.exists(timeoutFile) then
        local timeout = 30
        while timeout > 0 do
            drawBorder()
            local _, h = term.getSize()
            centerText("Incorrect password.", math.floor(h / 2) - 1)
            centerText("System locked for " .. timeout .. " seconds.", math.floor(h / 2))
            sleep(1)
            timeout = timeout - 1
        end
        fs.delete(timeoutFile)
        os.reboot()
    end
end

-- Main function to handle UEFI message and timeout
local function main()
    handleTimeout()

    if not checkPassword() then
        handleTimeout()
        return
    end

    showUEFIMessage()
    local timer = os.startTimer(2)  -- Set a timer for 2 seconds

    while true do
        local event, param1, param2, param3 = os.pullEvent()
        if event == "key" and param1 == keys.enter then
            -- Load boot options if Enter is pressed
            loadBootOptions()
            return
        elseif event == "key" and param1 == keys.f10 then
            -- Show Authentication screen if F10 is pressed
            showAuthScreen()
            return
        elseif event == "timer" and param1 == timer then
            -- Timer expired, load boot animation
            loadBootAnimation()
            return
        end
    end
end

-- Start the main function
main()