Fix extension: remove module type, add 45s timeout for HF cold start, robust error handling

extension/background.js

@@ -1,162 +1,151 @@
 /**
  * ClauseGuard — Background Service Worker (Manifest V3)
- * Handles: API calls, auth token management, side panel control, usage tracking
  */
 
 const API_BASE = "https://gaurv007-clauseguard-api.hf.space";
 const FREE_SCANS_PER_MONTH = 10;
+const API_TIMEOUT_MS = 45000; // 45s — HF free tier can take 30s to wake
 
 // ─── Side Panel ───
-chrome.sidePanel.setPanelBehavior({ openPanelOnActionClick: false }).catch(() => {});
+try { chrome.sidePanel.setPanelBehavior({ openPanelOnActionClick: false }); } catch(e) {}
+
+// ─── Fetch with timeout ───
+async function fetchWithTimeout(url, options, timeoutMs) {
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), timeoutMs);
+  try {
+    const resp = await fetch(url, { ...options, signal: controller.signal });
+    clearTimeout(timer);
+    return resp;
+  } catch (err) {
+    clearTimeout(timer);
+    throw err;
+  }
+}
 
 // ─── Message Router ───
 chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
-  switch (message.type) {
-    case "ANALYZE_TEXT":
-      handleAnalyze(message.payload, sender.tab?.id).then(sendResponse);
-      return true; // keep channel open for async
-
-    case "GET_AUTH":
-      getAuth().then(sendResponse);
-      return true;
-
-    case "CHECK_USAGE":
-      checkUsage().then(sendResponse);
-      return true;
-
-    case "OPEN_SIDEPANEL":
-      if (sender.tab?.id) {
-        chrome.sidePanel.open({ tabId: sender.tab.id });
-      }
-      break;
-
-    case "GET_RESULTS":
-      getStoredResults(sender.tab?.id || message.tabId).then(sendResponse);
-      return true;
-  }
+  const handle = async () => {
+    switch (message.type) {
+      case "ANALYZE_TEXT":
+        return await handleAnalyze(message.payload, sender.tab?.id);
+      case "GET_AUTH":
+        return await getAuth();
+      case "CHECK_USAGE":
+        return await checkUsage();
+      case "OPEN_SIDEPANEL":
+        if (sender.tab?.id) chrome.sidePanel.open({ tabId: sender.tab.id });
+        return { ok: true };
+      case "GET_RESULTS":
+        return await getStoredResults(sender.tab?.id || message.tabId);
+      default:
+        return null;
+    }
+  };
+  handle().then(sendResponse).catch(err => {
+    console.error("ClauseGuard handler error:", err);
+    sendResponse({ error: err.message });
+  });
+  return true;
 });
 
-// ─── External message from website (auth token) ───
+// ─── External messages from website ───
 chrome.runtime.onMessageExternal.addListener((message, sender, sendResponse) => {
   const allowed = ["https://app.clauseguard.com", "https://clauseguard.com"];
-  if (!allowed.some(origin => sender.origin?.startsWith(origin))) return;
-
+  if (!allowed.some(o => sender.origin?.startsWith(o))) return;
   if (message.type === "SET_AUTH") {
-    chrome.storage.sync.set({
-      authToken: message.token,
-      plan: message.plan || "free",
-      email: message.email || "",
-    });
+    chrome.storage.sync.set({ authToken: message.token, plan: message.plan || "free", email: message.email || "" });
     sendResponse({ success: true });
   }
-
   if (message.type === "LOGOUT") {
     chrome.storage.sync.remove(["authToken", "plan", "email"]);
     sendResponse({ success: true });
   }
 });
 
-// ─── Core: Analyze Text ───
+// ─── Core: Analyze ───
 async function handleAnalyze(payload, tabId) {
-  try {
-    // Check usage limits
-    const usage = await checkUsage();
-    if (!usage.allowed) {
-      return { error: "limit_reached", message: "Free scan limit reached. Upgrade to Pro for unlimited scans.", usage };
-    }
-
-    const { text, url } = payload;
+  const usage = await checkUsage();
+  if (!usage.allowed) {
+    return { error: "limit_reached", message: "Free scan limit reached.", usage };
+  }
 
-    // Split text into clauses
-    const clauses = splitIntoClauses(text);
-    if (clauses.length === 0) {
-      return { error: "no_clauses", message: "No analyzable clauses found on this page." };
-    }
+  const { text, url } = payload;
+  const clauses = splitIntoClauses(text);
+  if (clauses.length === 0) {
+    return { error: "no_clauses", message: "No analyzable clauses found." };
+  }
 
-    // Call API
+  // Try API first, fall back to local
+  let results;
+  try {
     const auth = await getAuth();
-    const response = await fetch(`${API_BASE}/api/analyze`, {
+    const resp = await fetchWithTimeout(`${API_BASE}/api/analyze`, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
         ...(auth.token ? { Authorization: `Bearer ${auth.token}` } : {}),
       },
       body: JSON.stringify({ clauses, source_url: url }),
-    });
+    }, API_TIMEOUT_MS);
 
-    if (!response.ok) {
-      throw new Error(`API error: ${response.status}`);
-    }
-
-    const results = await response.json();
+    if (!resp.ok) throw new Error(`HTTP ${resp.status}`);
+    results = await resp.json();
+  } catch (err) {
+    console.warn("API unavailable, using local analysis:", err.message);
+    results = localAnalyze(text, url);
+  }
 
-    // Store results for this tab
+  // Store + badge
+  if (tabId) {
     await chrome.storage.local.set({ [`results_${tabId}`]: results });
-
-    // Increment usage
-    await incrementUsage();
-
-    // Update badge
     const flagged = results.results?.filter(r => r.categories?.length > 0).length || 0;
     if (flagged > 0) {
       chrome.action.setBadgeText({ text: String(flagged), tabId });
       chrome.action.setBadgeBackgroundColor({ color: flagged > 3 ? "#ef4444" : "#f59e0b", tabId });
+    } else {
+      chrome.action.setBadgeText({ text: "", tabId });
     }
-
-    return results;
-  } catch (err) {
-    console.error("ClauseGuard analyze error:", err);
-    // Fallback: use local pattern matching
-    return localAnalyze(payload.text, payload.url);
   }
+
+  await incrementUsage();
+  return results;
 }
 
-// ─── Local Fallback Analyzer (regex patterns) ───
-function localAnalyze(text, url) {
+// ─── Local fallback ───
+function localAnalyze(text) {
   const clauses = splitIntoClauses(text);
   const patterns = {
-    0: [/arbitrat/i, /binding arbitration/i, /waive.*right.*court/i, /class action waiver/i],
-    1: [/sole discretion/i, /reserves? the right to (modify|change|update|amend)/i, /at any time.*without (prior )?notice/i],
-    2: [/remove.*content.*without/i, /right to remove/i, /we may.*remove/i],
-    3: [/exclusive jurisdiction/i, /courts? of.*(california|delaware|new york|ireland|england)/i, /submit to.*jurisdiction/i],
-    4: [/governed by.*laws? of/i, /shall be governed/i, /laws of the state of/i],
+    0: [/arbitrat/i, /binding arbitration/i, /waive.*right.*court/i],
+    1: [/sole discretion/i, /reserves? the right to (modify|change|update)/i, /at any time.*without.*notice/i],
+    2: [/remove.*content.*without/i, /right to remove/i],
+    3: [/exclusive jurisdiction/i, /courts? of.*(california|delaware|new york|ireland)/i, /submit to.*jurisdiction/i],
+    4: [/governed by.*laws? of/i, /shall be governed/i],
     5: [/not liable/i, /shall not be (liable|responsible)/i, /in no event.*liable/i, /limitation of liability/i, /without warranty/i],
-    6: [/terminat.*at any time/i, /suspend.*account.*without/i, /we may (terminat|suspend|discontinu)/i],
-    7: [/by (using|accessing).*you agree/i, /continued use.*constitutes? acceptance/i],
+    6: [/terminat.*at any time/i, /suspend.*account.*without/i, /we may (terminat|suspend)/i],
+    7: [/by (using|accessing).*you agree/i, /continued use.*constitutes/i],
   };
-
-  const labelNames = [
-    "Limitation of liability", "Unilateral termination", "Unilateral change",
-    "Content removal", "Contract by using", "Choice of law", "Jurisdiction", "Arbitration",
-  ];
-  const highRisk = new Set([0, 1, 6]); // liability, termination, arbitration (indices in labelNames mapping)
-  const severityMap = { 0: "HIGH", 1: "HIGH", 2: "MEDIUM", 3: "MEDIUM", 4: "MEDIUM", 5: "HIGH", 6: "HIGH", 7: "LOW" };
+  const names = ["Arbitration","Unilateral change","Content removal","Jurisdiction","Choice of law","Limitation of liability","Unilateral termination","Contract by using"];
+  const sevMap = {0:"HIGH",1:"MEDIUM",2:"MEDIUM",3:"MEDIUM",4:"MEDIUM",5:"HIGH",6:"HIGH",7:"LOW"};
 
   const results = clauses.map(clause => {
     const categories = [];
     for (const [id, pats] of Object.entries(patterns)) {
       if (pats.some(p => p.test(clause))) {
-        categories.push({
-          id: Number(id),
-          name: labelNames[id],
-          severity: severityMap[id],
-        });
+        categories.push({ name: names[id], severity: sevMap[id] });
       }
     }
     return { text: clause, categories };
   });
 
   const flagged = results.filter(r => r.categories.length > 0);
-  const sevCounts = { HIGH: 0, MEDIUM: 0, LOW: 0 };
-  flagged.forEach(r => r.categories.forEach(c => sevCounts[c.severity]++));
-
-  const riskScore = Math.min(100, Math.round(
-    (sevCounts.HIGH * 20 + sevCounts.MEDIUM * 10 + sevCounts.LOW * 5) / Math.max(1, clauses.length) * 100
-  ));
+  const sev = { HIGH: 0, MEDIUM: 0, LOW: 0 };
+  flagged.forEach(r => r.categories.forEach(c => sev[c.severity]++));
+  const risk = Math.min(100, Math.round((sev.HIGH*20 + sev.MEDIUM*10 + sev.LOW*5) / Math.max(1, clauses.length) * 100));
 
   return {
-    risk_score: riskScore,
-    grade: riskScore >= 60 ? "F" : riskScore >= 40 ? "D" : riskScore >= 20 ? "C" : riskScore >= 10 ? "B" : "A",
+    risk_score: risk,
+    grade: risk >= 60 ? "F" : risk >= 40 ? "D" : risk >= 20 ? "C" : risk >= 10 ? "B" : "A",
     total_clauses: clauses.length,
     flagged_count: flagged.length,
     results,
@@ -164,68 +153,40 @@ function localAnalyze(text, url) {
   };
 }
 
-// ─── Clause Splitter ───
+// ─── Utils ───
 function splitIntoClauses(text) {
-  const cleaned = text.replace(/\n{2,}/g, "\n").trim();
-  const clauses = cleaned.split(/(?<=[.!?])\s+(?=[A-Z0-9(])|(?:\n)(?=\d+[.)]\s|\([a-z]\)\s|•\s|-\s)/);
-  return clauses.map(c => c.trim()).filter(c => c.length > 30);
+  return text.replace(/\n{2,}/g, "\n").trim()
+    .split(/(?<=[.!?])\s+(?=[A-Z0-9(])|(?:\n)(?=\d+[.)]\s|\([a-z]\)\s)/)
+    .map(c => c.trim()).filter(c => c.length > 30);
 }
 
-// ─── Auth Helpers ───
 async function getAuth() {
-  return new Promise(resolve => {
-    chrome.storage.sync.get(["authToken", "plan", "email"], data => {
-      resolve({
-        token: data.authToken || null,
-        plan: data.plan || "free",
-        email: data.email || null,
-        isLoggedIn: !!data.authToken,
-        isPro: data.plan === "pro" || data.plan === "team",
-      });
-    });
-  });
+  return new Promise(r => chrome.storage.sync.get(["authToken","plan","email"], d => r({
+    token: d.authToken||null, plan: d.plan||"free", email: d.email||null,
+    isLoggedIn: !!d.authToken, isPro: d.plan==="pro"||d.plan==="team",
+  })));
 }
 
-// ─── Usage Tracking ───
 async function checkUsage() {
-  return new Promise(resolve => {
-    chrome.storage.sync.get(["plan", "scansThisMonth", "monthResetAt"], data => {
-      const now = new Date();
-      const resetAt = data.monthResetAt ? new Date(data.monthResetAt) : null;
-
-      // Reset counter on new month
-      if (!resetAt || now.getMonth() !== resetAt.getMonth() || now.getFullYear() !== resetAt.getFullYear()) {
-        chrome.storage.sync.set({ scansThisMonth: 0, monthResetAt: now.toISOString() });
-        resolve({ allowed: true, used: 0, limit: FREE_SCANS_PER_MONTH, plan: data.plan || "free" });
-        return;
-      }
-
-      const used = data.scansThisMonth || 0;
-      const plan = data.plan || "free";
-      const allowed = plan !== "free" || used < FREE_SCANS_PER_MONTH;
-
-      resolve({ allowed, used, limit: plan === "free" ? FREE_SCANS_PER_MONTH : Infinity, plan });
-    });
-  });
+  return new Promise(r => chrome.storage.sync.get(["plan","scansThisMonth","monthResetAt"], d => {
+    const now = new Date(), reset = d.monthResetAt ? new Date(d.monthResetAt) : null;
+    if (!reset || now.getMonth() !== reset.getMonth() || now.getFullYear() !== reset.getFullYear()) {
+      chrome.storage.sync.set({ scansThisMonth: 0, monthResetAt: now.toISOString() });
+      r({ allowed: true, used: 0, limit: FREE_SCANS_PER_MONTH, plan: d.plan||"free" });
+      return;
+    }
+    const used = d.scansThisMonth||0, plan = d.plan||"free";
+    r({ allowed: plan!=="free"||used<FREE_SCANS_PER_MONTH, used, limit: plan==="free"?FREE_SCANS_PER_MONTH:Infinity, plan });
+  }));
 }
 
 async function incrementUsage() {
-  return new Promise(resolve => {
-    chrome.storage.sync.get(["scansThisMonth"], data => {
-      chrome.storage.sync.set({ scansThisMonth: (data.scansThisMonth || 0) + 1 }, resolve);
-    });
-  });
+  return new Promise(r => chrome.storage.sync.get(["scansThisMonth"], d =>
+    chrome.storage.sync.set({ scansThisMonth: (d.scansThisMonth||0)+1 }, r)));
 }
 
 async function getStoredResults(tabId) {
-  return new Promise(resolve => {
-    chrome.storage.local.get([`results_${tabId}`], data => {
-      resolve(data[`results_${tabId}`] || null);
-    });
-  });
+  return new Promise(r => chrome.storage.local.get([`results_${tabId}`], d => r(d[`results_${tabId}`]||null)));
 }
 
-// ─── Tab cleanup ───
-chrome.tabs.onRemoved.addListener(tabId => {
-  chrome.storage.local.remove([`results_${tabId}`]);
-});
+chrome.tabs.onRemoved.addListener(tabId => chrome.storage.local.remove([`results_${tabId}`]));

extension/manifest.json

@@ -2,7 +2,7 @@
   "manifest_version": 3,
   "name": "ClauseGuard — AI Fine Print Scanner",
   "version": "1.0.0",
-  "description": "Instantly highlights unfair clauses in Terms of Service, contracts, and lease agreements. Powered by Legal AI.",
+  "description": "Highlights unfair clauses in Terms of Service, contracts, and lease agreements.",
   "permissions": [
     "activeTab",
     "storage",
@@ -13,8 +13,7 @@
     "https://gaurv007-clauseguard-api.hf.space/*"
   ],
   "background": {
-    "service_worker": "background.js",
-    "type": "module"
+    "service_worker": "background.js"
   },
   "action": {
     "default_popup": "popup.html",
@@ -44,8 +43,5 @@
     "32": "icons/icon32.png",
     "48": "icons/icon48.png",
     "128": "icons/icon128.png"
-  },
-  "content_security_policy": {
-    "extension_pages": "script-src 'self'; object-src 'self'"
   }
 }