1 files changed, 225 insertions, 0 deletions
diff --git a/apps/extension/src/background.ts b/apps/extension/src/background.ts
new file mode 100644
index 00000000..21179f86
--- /dev/null
+++ b/apps/extension/src/background.ts
@@ -0,0 +1,225 @@
+import { getBaseURL } from "@/lib/utils";
+import {
+  messageListener,
+  MessageType,
+  registerMessageHandler,
+} from "./helpers";
+import { handleExportXBookmarks, setupTwitterHeaderListener } from "./twitter";
+
+type TabState = {
+  isActive: boolean;
+};
+
+const tabStates = new Map<number, TabState>();
+
+const checkIfLoggedIn = async () => {
+  const baseURL = await getBaseURL();
+  const response = await fetch(`${baseURL}/api/session`);
+
+  return response.status == 200;
+};
+
+// When extension is installed
+chrome.runtime.onInstalled.addListener(async (details) => {
+  const isLoggedIn = await checkIfLoggedIn();
+
+  const baseURL = await getBaseURL();
+  if (!isLoggedIn) {
+    chrome.tabs.create({ url: `${baseURL}/signin` });
+  }
+
+  // TODO: show extension help page
+});
+
+// Clean up tab state when tab is closed
+chrome.tabs.onRemoved.addListener((tabId) => {
+  tabStates.delete(tabId);
+});
+
+// communication with content script
+chrome.action.onClicked.addListener(async (tab) => {
+  if (!tab.id) return;
+
+  try {
+    // Check if we can inject into this tab
+    if (
+      !tab.url ||
+      tab.url.startsWith("chrome://") ||
+      tab.url.startsWith("edge://") ||
+      tab.url.startsWith("about:")
+    ) {
+      alert("Cannot modify Chrome system pages");
+      return;
+    }
+
+    const baseURL = await getBaseURL();
+
+    const isLoggedIn = await checkIfLoggedIn();
+    if (!isLoggedIn) {
+      chrome.tabs.create({ url: `${baseURL}/signin` });
+      return;
+    }
+
+    const currentState = tabStates.get(tab.id) || { isActive: false };
+    const newState = { isActive: !currentState.isActive };
+
+    chrome.scripting.executeScript({
+      target: { tabId: tab.id },
+      files: ["scripts/content.js"],
+    });
+
+    // Update state
+    tabStates.set(tab.id, newState);
+
+    // Update icon title
+    await chrome.action.setTitle({
+      tabId: tab.id,
+      title: newState.isActive ? "Disable SuperMemory" : "Enable SuperMemory",
+    });
+  } catch (error) {
+    console.error("Failed to toggle content:", error);
+  }
+});
+
+chrome.runtime.onMessage.addListener(messageListener);
+
+registerMessageHandler<MessageType>(
+  "GET_SPACES",
+  async (message, sender, sendResponse) => {
+    // Handle getting spaces
+    const baseURL = await getBaseURL();
+    const response = await fetch(`${baseURL}/backend/api/spaces`);
+    const data = await response.json();
+    sendResponse(data);
+  }
+);
+
+registerMessageHandler<MessageType>(
+  "EXPORT_TWITTER_BOOKMARKS",
+  async (message, sender, sendResponse) => {
+    handleExportXBookmarks();
+    sendResponse({ status: "started" });
+  }
+);
+
+registerMessageHandler<MessageType>(
+  "SAVE_PAGE",
+  async (message, sender, sendResponse) => {
+    if (!message.payload) {
+      sendResponse({ error: "No payload" });
+      return;
+    }
+
+    const baseURL = await getBaseURL();
+
+    const isLoggedIn = await checkIfLoggedIn();
+    if (!isLoggedIn) {
+      sendResponse({ error: "Not logged in" });
+      chrome.tabs.create({ url: `${baseURL}/signin` });
+      return;
+    }
+
+    console.log(message.payload);
+
+    const response = await fetch(`${baseURL}/backend/api/add`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({
+        spaces: message.payload.spaces || [],
+        content: message.payload.url,
+        prefetched: message.payload.prefetched,
+      }),
+    });
+    if (response.status !== 200) {
+      sendResponse({ error: "Failed to save page", status: response.status });
+      return;
+    }
+    const data = await response.json();
+    // Handle saving highlight
+    sendResponse({ success: true, data, status: response.status });
+  }
+);
+
+registerMessageHandler<MessageType>(
+  "ACTIVATE_CONTENT",
+  async (message, sender, sendResponse) => {
+    console.log("Activating content");
+    chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => {
+      const currentTab = tabs[0];
+      if (currentTab.id) {
+        try {
+          await chrome.scripting.executeScript({
+            target: { tabId: currentTab.id },
+            files: ["scripts/content.js"],
+          });
+
+        } catch (error) {
+          console.error("Error injecting content script:", error);
+        }
+      }
+    });
+    sendResponse({ success: true });
+  }
+);
+
+registerMessageHandler<MessageType>(
+  "SYNC_CHROME_BOOKMARKS",
+  async (message, sender, sendResponse) => {
+    // activate the chrome bookmarks syncing.
+
+    // TODO: We probably want to sync bookmarks from the extension to the web app.
+    chrome.bookmarks.onCreated.addListener((id, bookmark) => {
+      console.log("Bookmark created:", bookmark);
+    });
+  }
+);
+
+registerMessageHandler<MessageType>(
+  "IMPORT_CHROME_BOOKMARKS",
+  async (message, sender, sendResponse) => {
+    // activate the chrome bookmarks importing.
+    // first get all chrome bookmarks
+    chrome.bookmarks.getRecent(100, (bookmarks) => {
+      console.log("Bookmarks:", bookmarks);
+    });
+  }
+);
+
+// External message listener
+chrome.runtime.onMessageExternal.addListener(
+  async (request, sender, sendResponse) => {
+    if (request.action === "exportBookmarks") {
+      handleExportXBookmarks();
+      sendResponse({ status: "exported" });
+      return true;
+    }
+    if (request.action === "importBookmarks") {
+      const baseURL = await getBaseURL();
+      chrome.bookmarks.getRecent(100, async (bookmarks) => {
+        for (const { url } of bookmarks) {
+          console.log("Importing bookmark:", url);
+          const r = await fetch(`${baseURL}/backend/api/add`, {
+            method: "POST",
+            headers: {
+              "Content-Type": "application/json",
+            },
+            body: JSON.stringify({ content: url, spaces: [] }),
+          });
+
+          const response = await r.json();
+          console.log("Response:", response);
+        }
+        sendResponse({ status: "imported", bookmarks });
+      });
+      return true;
+    }
+    if (request.action === "ping") {
+      sendResponse({ status: "pong" });
+      return true;
+    }
+  }
+);
+
+setupTwitterHeaderListener();