PassKeeper/extension/background.js

/**
 * extension/background.js — Manifest V3 service worker.
 *
 * Responsibilities:
 *  - Update the action badge (number of matching vault items) for the active tab.
 *  - Bridge SAVE_CREDENTIALS messages from content script → chrome.storage.session
 *    so the popup can pick them up as a save-prompt.
 *  - Re-update badges when the vault cache changes.
 */

// ── Badge helpers ────────────────────────────────────────────────────────────

async function updateBadgeForTab(tabId, url) {
  try {
    const { vault_items } = await chrome.storage.session.get("vault_items");
    if (!vault_items?.length) {
      chrome.action.setBadgeText({ text: "", tabId });
      return;
    }

    let hostname;
    try {
      hostname = new URL(url).hostname.replace(/^www\./, "");
    } catch {
      chrome.action.setBadgeText({ text: "", tabId });
      return;
    }

    const matches = vault_items.filter((item) => {
      if (item.item_type !== "password" || !item.plain?.url) return false;
      try {
        const h = new URL(item.plain.url).hostname.replace(/^www\./, "");
        return (
          h === hostname ||
          h.endsWith(`.${hostname}`) ||
          hostname.endsWith(`.${h}`)
        );
      } catch {
        return false;
      }
    });

    if (matches.length > 0) {
      chrome.action.setBadgeText({ text: String(matches.length), tabId });
      chrome.action.setBadgeBackgroundColor({ color: "#1a73e8", tabId });
    } else {
      chrome.action.setBadgeText({ text: "", tabId });
    }
  } catch {
    /* tab may have closed */
  }
}

async function refreshAllBadges() {
  const tabs = await chrome.tabs.query({});
  for (const tab of tabs) {
    if (tab.url?.startsWith("http")) updateBadgeForTab(tab.id, tab.url);
  }
}

// ── Tab events ───────────────────────────────────────────────────────────────

chrome.tabs.onActivated.addListener(async ({ tabId }) => {
  try {
    const tab = await chrome.tabs.get(tabId);
    if (tab.url?.startsWith("http")) updateBadgeForTab(tabId, tab.url);
  } catch {}
});

chrome.tabs.onUpdated.addListener((tabId, changeInfo, tab) => {
  if (changeInfo.status === "complete" && tab.url?.startsWith("http")) {
    updateBadgeForTab(tabId, tab.url);
  }
});

// ── Message handler ──────────────────────────────────────────────────────────

chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
  // No-op ping from popup — keeps the service worker alive while the browser
  // is open so chrome.storage.session is not wiped between popup openings.
  if (msg.type === "KEEPALIVE") {
    sendResponse({ ok: true });
    return false;
  }

  // Popup signals vault cache was refreshed → re-check all tab badges.
  if (msg.type === "VAULT_UPDATED") {
    refreshAllBadges();
    sendResponse({ ok: true });
    return false;
  }

  // Content script detected credentials on form submit → store for save-prompt.
  if (msg.type === "SAVE_CREDENTIALS") {
    chrome.storage.session.set({ pending_save: msg.data });
    sendResponse({ ok: true });
    return false;
  }

  // Bridge: web app logged in → store tokens in extension storage.
  if (msg.type === "WEB_SESSION_SYNC") {
    chrome.storage.session.set({
      access_token: msg.access_token,
      enc_key_salt: msg.enc_key_salt || "",
    });
    // Also persist enc_key_salt locally so the unlock-only view survives browser restarts
    chrome.storage.local.set({
      refresh_token: msg.refresh_token,
      ...(msg.enc_key_salt ? { enc_key_salt: msg.enc_key_salt } : {}),
    });
    sendResponse({ ok: true });
    return false;
  }

  // Bridge: web app logged out → clear extension session.
  if (msg.type === "WEB_SESSION_CLEAR") {
    chrome.storage.session.remove([
      "access_token",
      "vault_key_jwk",
      "vault_items",
      "enc_key_salt",
    ]);
    chrome.storage.local.remove("refresh_token");
    sendResponse({ ok: true });
    return false;
  }

  // Popup logged in via extension → inject session into open vault tabs.
  if (msg.type === "EXT_SESSION_SYNC") {
    chrome.tabs.query(
      { url: "https://pwkeeper.ngodanguyen.tech/*" },
      (tabs) => {
        tabs.forEach((tab) => {
          chrome.tabs
            .sendMessage(tab.id, {
              type: "INJECT_SESSION",
              access_token: msg.access_token,
              refresh_token: msg.refresh_token,
              enc_key_salt: msg.enc_key_salt,
            })
            .catch(() => {});
        });
      },
    );
    sendResponse({ ok: true });
    return false;
  }

  return false;
});