refactor: Enable background file processing and popup persistence

Author: scott-the-programmer

src/background.ts

@@ -81,5 +81,9 @@ export async function handleFileProcessing(
     });
 
     return errorResult;
+  } finally {
+    chrome.runtime.sendMessage({
+      type: "processComplete",
+    });
   }
 }

src/popup.ts

@@ -1,50 +1,10 @@
 import TokenStore from "./tokenStore";
 import { ProcessResult, FileDetails } from "./types";
 
-document.addEventListener("DOMContentLoaded", async () => {
-  try {
-    const tabs = await chrome.tabs.query({ active: true, currentWindow: true });
-    const tab = tabs[0];
-
-    if (tab.id) {
-      const result = await chrome.scripting.executeScript({
-        target: { tabId: tab.id },
-        func: () => {
-          const items: Record<string, string> = {};
-          for (let i = 0; i < localStorage.length; i++) {
-            const key = localStorage.key(i);
-            if (
-              key &&
-              (key.includes("chatsvcagg") || key.includes("ic3.teams.office"))
-            ) {
-              items[key] = localStorage.getItem(key) || "";
-            }
-          }
-          return items;
-        },
-      });
-
-      if (result && result[0].result) {
-        for (const [key, value] of Object.entries(result[0].result)) {
-          try {
-            const parsedValue = JSON.parse(value as string);
-            await chrome.storage.local.set({ [key]: parsedValue });
-            console.log("Stored token:", key);
-          } catch (e) {
-            console.error("Failed to parse token:", key, e);
-          }
-        }
-        console.log("Teams tokens captured from page");
-      }
-    }
-  } catch (err) {
-    console.error("Error capturing tokens:", err);
-  }
-});
-
 function updateStatus(
   message: string,
   type: "ready" | "success" | "error" | "processing" = "ready",
+  persist: boolean = true,
 ) {
   const statusDiv = document.getElementById("status")!;
   const statusText = document.getElementById("statusText")!;
@@ -61,6 +21,20 @@ function updateStatus(
   statusDiv.classList.add(statusClasses[type][0]);
   statusIcon.className = statusClasses[type][1];
   statusText.textContent = message;
+
+  if (persist) {
+    localStorage.setItem("uploadStatus", JSON.stringify({ message, type }));
+  }
+}
+
+function restoreStatus() {
+  const status = localStorage.getItem("uploadStatus");
+  if (status) {
+    const { message, type } = JSON.parse(status);
+    updateStatus(message, type, false);
+  } else {
+    updateStatus("Ready to process", "ready", false);
+  }
 }
 
 async function handleFileProcessing() {
@@ -94,45 +68,11 @@ async function handleFileProcessing() {
 
     const tokens = await TokenStore.collectTokensFromStorage();
 
-    const result = await chrome.runtime.sendMessage({
+    chrome.runtime.sendMessage({
       action: "processFiles",
       files: filesJson,
       tokens: tokens,
     });
-
-    const { success, error, status } = result as ProcessResult;
-    if (error) {
-      updateStatus(error, "error");
-    } else {
-      updateStatus(status || "", success ? "success" : "processing");
-
-      await chrome.browsingData.remove(
-        {
-          origins: ["https://teams.microsoft.com"],
-        },
-        {
-          cacheStorage: true,
-          cookies: true,
-          localStorage: true,
-          serviceWorkers: true,
-          indexedDB: true,
-          cache: true,
-          appcache: true,
-        },
-      );
-
-      await chrome.storage.local.clear();
-
-      const tabs = await chrome.tabs.query({
-        active: true,
-        currentWindow: true,
-      });
-      const tab = tabs[0];
-
-      if (tab.id) {
-        await chrome.tabs.reload(tab.id);
-      }
-    }
   } catch (error) {
     let errorMessage = "";
 
@@ -166,12 +106,16 @@ function updateSelectedFiles() {
   }
 }
 
-document
-  .getElementById("fileInput")
-  ?.addEventListener("change", updateSelectedFiles);
-document
-  .getElementById("processButton")
-  ?.addEventListener("click", () => handleFileProcessing());
+document.addEventListener("DOMContentLoaded", () => {
+  restoreStatus();
+
+  document
+    .getElementById("fileInput")
+    ?.addEventListener("change", updateSelectedFiles);
+  document
+    .getElementById("processButton")
+    ?.addEventListener("click", () => handleFileProcessing());
+});
 
 chrome.runtime.onMessage.addListener(
   (message: { type: string; error: any; status: any; success: any }) => {
@@ -181,5 +125,8 @@ chrome.runtime.onMessage.addListener(
         message.success ? "success" : "error",
       );
     }
+    if (message.type === "processComplete") {
+      localStorage.removeItem("uploadStatus");
+    }
   },
 );