term.clear()

-- Function to get the terminal dimensions
function getTerminalSize()
    local width, height = term.getSize()
    return width, height
end

-- Function to draw the loading bar centered
function drawLoadingBar(current, total, message)
    local width, _ = getTerminalSize()
    local barLength = 30 -- Length of the loading bar
    local filledLength = math.floor((current / total) * barLength)
    local emptyLength = barLength - filledLength
    local bar = "[" .. string.rep("=", filledLength) .. string.rep(" ", emptyLength) .. "]"

    -- Calculate the position to center the bar
    local barPos = math.floor((width - barLength - 2) / 2)

    -- Move the UI down by adjusting the row numbers
    local barRow = 8
    local messageRow = 10

    -- Clear the previous loading bar and message
    term.setCursorPos(1, barRow)
    term.write(string.rep(" ", width)) -- Clear the previous bar
    term.setCursorPos(1, messageRow)
    term.write(string.rep(" ", width)) -- Clear the previous message

    term.setCursorPos(barPos, barRow)
    term.write(bar .. " " .. math.floor((current / total) * 100) .. "%")

    -- Print the current message centered below the bar
    if message then
        term.setCursorPos(math.floor((width - #message) / 2), messageRow)
        term.write(message)
    end
end

-- Function to center and stylize text
function printCenteredText(text, row)
    local width, _ = getTerminalSize()
    local textLength = #text
    local pos = math.floor((width - textLength) / 2)
    term.setCursorPos(pos, row)
    term.write(text)
end

-- Function to display an error message
function showError(message)
    term.clear()
    local width, height = getTerminalSize()
    local errorMessage = "Error: " .. message
    local midHeight = math.floor(height / 2)

    term.setCursorPos(1, midHeight)
    term.write(string.rep(" ", width)) -- Clear the row
    term.setCursorPos(math.floor((width - #errorMessage) / 2), midHeight)
    term.write(errorMessage)

    term.setCursorPos(1, midHeight + 1)
    term.write("Press any key to exit...")
    os.pullEvent("key") -- Wait for any key press
    term.clear()
end

-- Print the initial status messages
printCenteredText("Installing...", 4)  -- Moved down to row 4
printCenteredText("DO NOT Interrupt Install", 6)  -- Moved down to row 6

-- Define installation steps with updated messages
local steps = {
    { message = "Setup copying files...", action = function() fs.copy("/disk/", "/root") end },
    { message = "Unmounting installation disk...", action = function()
        disk.eject("top")
        disk.eject("bottom")
        disk.eject("left")
        disk.eject("right")
    end },
    { message = "Formatting file system...", action = function() shell.run("rename", "/root", "/disk") end },
    { message = "Creating boot recovery directory...", action = function() fs.copy("/disk/", "recovery") end },
    { action = function() shell.run("/disk/install-assist") end }  -- No message for this step
}

-- Simulate the installation process with updates
local totalSteps = #steps
for i, step in ipairs(steps) do
    local success, err = pcall(function()
        if step.message then
            drawLoadingBar(i, totalSteps, step.message)
        else
            drawLoadingBar(i, totalSteps)
        end
        step.action()  -- Execute the installation step
        sleep(4) -- Increased sleep duration to slow down the process
    end)

    if not success then
        showError(err)
        return
    end
end

-- Final message indicating completion
local _, height = getTerminalSize()
term.setCursorPos(1, height - 2)
term.write(string.rep(" ", term.getSize()))
term.setCursorPos(math.floor((term.getSize() - #("Installation complete!")) / 2), height - 2)
term.write("Installation complete!")