popup.js

document.addEventListener('DOMContentLoaded', () => {
  const enabledEl = document.getElementById('enabled');
  const applyBtn = document.getElementById('apply');
  const ACTION_RUN = 'runreplacements';

  function sendRunMessageToGovTabs() {
    chrome.tabs.query({ url: '*://*.gov/*' }, (tabs) => {
      for (const t of tabs) chrome.tabs.sendMessage(t.id, { action: ACTION_RUN });
    });
  }

  // Load current enabled state
  chrome.storage.sync.get(['enabled'], (res) => {
    enabledEl.checked = res.enabled === true;
  });

  enabledEl.addEventListener('change', () => {
    const enabled = enabledEl.checked;
    // Save enabled state
    chrome.storage.sync.set({ enabled });
    if (enabled) {
      // If there are no custom phrases saved, populate defaults so enabling immediately works
      chrome.storage.sync.get(['customPhrases'], (res) => {
        const has = Array.isArray(res.customPhrases) && res.customPhrases.length;
        if (!has) {
          const url = chrome.runtime.getURL('defaults.json');
          fetch(url).then(r => r.json()).then(defs => {
            if (Array.isArray(defs) && defs.length) {
              chrome.storage.sync.set({ customPhrases: defs }, () => {
                // notify tabs to re-run replacements
                sendRunMessageToGovTabs();
              });
            }
          }).catch(() => {
            // Still notify tabs to run if any custom phrases exist
            sendRunMessageToGovTabs();
          });
        } else {
          // already have custom phrases, just notify tabs
          sendRunMessageToGovTabs();
        }
      });
    } else {
      // If disabling, notify tabs to clear replacements if needed (no-op here)
    }
  });

  applyBtn.addEventListener('click', () => {
    const url = chrome.runtime.getURL('defaults.json');
    fetch(url).then(r => r.json()).then(defs => {
      if (Array.isArray(defs)) {
        chrome.storage.sync.set({ customPhrases: defs }, () => {
          const lbl = applyBtn.querySelector('.btn-label');
          if (lbl) { lbl.textContent = 'Applied'; setTimeout(() => lbl.textContent = 'Apply defaults', 1200); }
        });
      }
    }).catch(() => {
      const lbl = applyBtn.querySelector('.btn-label');
      if (lbl) { lbl.textContent = 'Failed'; setTimeout(() => lbl.textContent = 'Apply defaults', 1200); }
    });
  });
  const openLink = document.getElementById('openOptions');
  if (openLink) {
    openLink.addEventListener('click', (e) => {
      e.preventDefault();
      try { chrome.runtime.openOptionsPage(); } catch (err) { window.open('options.html'); }
      window.close();
    });
  }
});