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

    const data = await chrome.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 chrome.storage.local.set({ disabledSites });
        chrome.tabs.reload(tab.id);
        window.close();
    });
});