background.js

chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
  if (msg.action === "update") {
    console.log("Received update request:", msg);
    const sites = msg.enabled ? msg.sites : [];

    syncRules(sites)
      .then(() => {
        console.log("Rules synced successfully");
        sendResponse({ success: true });
      })
      .catch((err) => {
        console.error("Rule sync failed:", err);
        sendResponse({ success: false, error: err.message });
      });

    return true;
  }
});

chrome.runtime.onInstalled.addListener(async () => {
  console.log("Extension installed/updated.");
  try {
    const data = await chrome.storage.local.get(["sites", "enabled"]);
    const isEnabled = data.enabled !== false;
    const sites = data.sites || [];

    await syncRules(isEnabled ? sites : []);
  } catch (e) {
    console.error("Initialization failed:", e);
  }
});

async function syncRules(sites) {
  const currentRules = await chrome.declarativeNetRequest.getDynamicRules();
  const removeRuleIds = currentRules.map((r) => r.id);

  console.log(`Clearing ${removeRuleIds.length} old rules...`);

  if (!sites || !sites.length) {
    await chrome.declarativeNetRequest.updateDynamicRules({
      removeRuleIds: removeRuleIds,
      addRules: [],
    });
    return;
  }

  const addRules = sites.map((site, index) => ({
    id: index + 1,
    priority: 1,
    action: {
      type: "redirect",
      redirect: { extensionPath: "/blocked.html" },
    },
    condition: {
      urlFilter: `||${site}`,
      resourceTypes: ["main_frame"],
    },
  }));

  console.log(`Applying ${addRules.length} new rules...`);

  await chrome.declarativeNetRequest.updateDynamicRules({
    removeRuleIds: removeRuleIds,
    addRules: addRules,
  });
}