document.addEventListener('DOMContentLoaded', async () => {
    const toggleBtn = document.getElementById('toggleBtn');
    
    const tabs = await browser.tabs.query({ active: true, currentWindow: true });
    const tab = tabs[0];
    if (!tab?.url) return;
    
    const url = new URL(tab.url);
    const domain = url.hostname;

    const data = await browser.storage.local.get({ disabledSites: [] });
    let disabledSites = data.disabledSites;
    const isDisabled = disabledSites.includes(domain);

    toggleBtn.textContent = isDisabled ? 'Enable' : 'Disable';

    toggleBtn.addEventListener('click', async () => {
        if (isDisabled) {
            disabledSites = disabledSites.filter(site => site !== domain);
        } else {
            disabledSites.push(domain);
        }

        await browser.storage.local.set({ disabledSites });
        browser.tabs.reload(tab.id);
        window.close();
    });
});