-- Function to draw the title bar with the red "X"
local function drawTitleBar()
    local width, _ = term.getSize()
    term.setBackgroundColor(colors.blue)
    term.setTextColor(colors.white)
    term.setCursorPos(1, 1)
    term.write("Doggy OS Browser")
    term.setCursorPos(width, 1)
    term.write("X")
    term.setBackgroundColor(colors.black)
    term.setTextColor(colors.white)
end

-- Function to check if a mouse click is on the red "X" in the title bar
local function isCloseButtonClicked(x, y)
    local width, _ = term.getSize()
    return x == width and y == 1
end

-- Function to wait for a mouse click event
local function waitForClick()
    while true do
        local event, button, x, y = os.pullEvent("mouse_click")
        if isCloseButtonClicked(x, y) then
            return true
        end
    end
end

-- Function to display an error message and wait
local function displayError(message)
    term.clear()
    drawTitleBar()
    print("Error: " .. message)
    print("Press any key to return to the main menu.")
    os.pullEvent("key")  -- Wait for any key press
end

-- Function to get URL from user
local function getURL()
    term.clear()
    drawTitleBar()
    print("Enter URL:")
    return read()
end

-- Automatically find and open the first available modem
local function openModem()
    local sides = {"top", "bottom", "left", "right", "front", "back"}
    for _, side in ipairs(sides) do
        if peripheral.isPresent(side) and peripheral.getType(side) == "modem" then
            rednet.open(side)
            return side
        end
    end
    displayError("No modem found.")
    return nil
end

-- Load DNS Provider ID from file
local function loadDNSProviderID()
    local filePath = "DNS-Provider.txt"
    if not fs.exists(filePath) then
        displayError("DNS-Provider.txt not found.")
        return nil
    end
    local file = fs.open(filePath, "r")
    local id = file.readAll()
    file.close()
    return tonumber(id)
end

-- Save DNS Provider ID to file
local function saveDNSProviderID(id)
    local filePath = "DNS-Provider.txt"
    local file = fs.open(filePath, "w")
    file.write(id)
    file.close()
    print("DNS Provider ID updated.")
end

-- Connect to DNS and get IP address
local function connectToDNS(url, dnsID)
    local modemSide = openModem()
    if not modemSide then return nil end

    term.clear()
    drawTitleBar()
    print("Connecting to DNS...")

    rednet.send(dnsID, url)
    while true do
        local event, p1, p2 = os.pullEvent()
        if event == "rednet_message" and p1 == dnsID then
            if p2 == "PASSWORD_REQUIRED" then
                -- Prompt for password
                term.clear()
                drawTitleBar()
                print("Enter DNS Password:")
                local password = read("*")
                rednet.send(dnsID, password)
            elseif p2 == "AUTH_SUCCESS" then
                print("Authenticated successfully!")
            elseif p2 == "AUTH_FAILED" then
                displayError("Authentication failed. Incorrect password.")
                return nil
            else
                rednet.close(modemSide)
                return p2
            end
        elseif event == "mouse_click" and isCloseButtonClicked(p1, p2) then
            rednet.close(modemSide)
            return nil
        end
    end
end

-- Connect to webserver and get website code
local function connectToWebserver(ip)
    local modemSide = openModem()
    if not modemSide then return nil end

    term.clear()
    drawTitleBar()
    print("Connecting to Webserver...")

    rednet.send(tonumber(ip), "GET")
    local timer = os.startTimer(5)
    while true do
        local event, p1, p2 = os.pullEvent()
        if event == "rednet_message" and p1 == tonumber(ip) then
            rednet.close(modemSide)
            return p2
        elseif event == "timer" and p1 == timer then
            rednet.close(modemSide)
            displayError("Connection timed out.")
            return nil
        elseif event == "mouse_click" and isCloseButtonClicked(p1, p2) then
            rednet.close(modemSide)
            return nil
        end
    end
end

-- Execute website code and render it
local function runWebsite(code)
    local func, err = load(code, "website", "t", _ENV)
    if func then
        term.clear()
        drawTitleBar()
        func()  -- Run the website code
        while not waitForClick() do end  -- Keep checking for click events to close the browser
        term.clear()
    else
        displayError("Error loading website: " .. err)
    end
end

-- Main menu function
local function mainMenu()
    term.clear()
    drawTitleBar()
    print("Select an option:")
    print("1. Enter URL")
    print("2. Edit DNS Provider ID")
    print("3. Exit")
    local choice = tonumber(read())

    if choice == 1 then
        while true do
            local url = getURL()
            if url == nil then break end

            local dnsID = loadDNSProviderID()
            if not dnsID then
                displayError("Failed to load DNS Provider ID.")
                return
            end

            local ip = connectToDNS(url, dnsID)
            if ip == "404_NOT_FOUND" then
                displayError("Error 404: Webpage cannot be found!")
                return
            elseif not ip then
                displayError("No response from DNS.")
                return
            end

            local site = connectToWebserver(ip)
            if site then
                runWebsite(site)
            end
        end
    elseif choice == 2 then
        print("Enter new DNS Provider ID:")
        local newID = tonumber(read())
        if newID then
            saveDNSProviderID(newID)
        else
            displayError("Invalid input. Please enter a number.")
        end
    elseif choice == 3 then
        return
    else
        displayError("Invalid choice.")
    end
end

-- Run the main menu loop
while true do
    mainMenu()
end