total-recall-brain 3.2.2 → 3.3.0

package/extension/background.js

@@ -1,7 +1,7 @@
 // Total Recall — Background Service Worker
 // Handles context menus, message routing, and passive browsing buffer.
 
-importScripts('lib/brain-client.js');
+importScripts('lib/preconfigured.js', 'lib/brain-client.js');
 
 // Destructure for convenience after importScripts populates self.BrainClient
 const { share, search, healthCheck } = self.BrainClient;

package/extension/content-script.js

@@ -131,4 +131,19 @@
 
   // Wait 2 seconds after page load to avoid interfering with page rendering
   setTimeout(queryBrain, 2000);
+
+  // ---- Listen for page context requests ----
+  chrome.runtime.onMessage.addListener((msg, _sender, sendResponse) => {
+    if (msg.type === 'GET_PAGE_TEXT') {
+      const selection = window.getSelection().toString().trim();
+      const pageText = document.body.innerText.slice(0, 10000); // limit to 10k chars to avoid token blowout
+      sendResponse({
+        url: location.href,
+        title: document.title,
+        selection,
+        pageText
+      });
+    }
+    return true;
+  });
 })();

package/extension/lib/brain-client.js

@@ -2,46 +2,85 @@
 // Shared by background.js, popup, sidepanel, and options page.
 // Exposed as `self.BrainClient` namespace (no ES-module bundler required).
 
-const DEFAULT_BRAIN_URL = 'http://127.0.0.1:3000';
-
-async function getConfig() {
-  const config = await chrome.storage.sync.get(['brainUrl', 'pat']);
-  return {
-    brainUrl: config.brainUrl || DEFAULT_BRAIN_URL,
-    pat: config.pat || ''
-  };
-}
-
-async function brainFetch(path, options = {}) {
-  const { brainUrl, pat } = await getConfig();
-  const url = `${brainUrl}${path}`;
-  const headers = { 'Content-Type': 'application/json', ...options.headers };
-  if (pat) headers['Authorization'] = `Bearer ${pat}`;
-
-  const res = await fetch(url, { ...options, headers });
-  if (res.status === 401) {
-    chrome.action.setBadgeText({ text: '!' });
-    chrome.action.setBadgeBackgroundColor({ color: '#ef4444' });
-    throw new Error('Authentication failed — check your PAT token');
+(function () {
+  const DEFAULT_BRAIN_URL = 'http://127.0.0.1:3000';
+
+  async function getConfig() {
+    const config = await chrome.storage.sync.get(['brainUrl', 'pat']);
+    const pre = self.PreConfigured || {};
+    return {
+      brainUrl: config.brainUrl || pre.brainUrl || DEFAULT_BRAIN_URL,
+      pat: config.pat || pre.pat || ''
+    };
+  }
+
+  async function brainFetch(path, options = {}) {
+    const { brainUrl, pat } = await getConfig();
+    const url = `${brainUrl}${path}`;
+    const headers = { 'Content-Type': 'application/json', ...options.headers };
+    if (pat) headers['Authorization'] = `Bearer ${pat}`;
+
+    const res = await fetch(url, { ...options, headers });
+    if (res.status === 401) {
+      chrome.action.setBadgeText({ text: '!' });
+      chrome.action.setBadgeBackgroundColor({ color: '#ef4444' });
+      throw new Error('Authentication failed — check your PAT token');
+    }
+    if (!res.ok) throw new Error(`Brain API error: ${res.status}`);
+    return res.json();
+  }
+
+  async function share(data) {
+    return brainFetch('/api/share', { method: 'POST', body: JSON.stringify(data) });
+  }
+
+  async function search(query, topK = 3) {
+    return brainFetch('/api/memory/search/semantic', {
+      method: 'POST',
+      body: JSON.stringify({ query, top_k: topK })
+    });
   }
-  if (!res.ok) throw new Error(`Brain API error: ${res.status}`);
-  return res.json();
-}
-
-async function share(data) {
-  return brainFetch('/api/share', { method: 'POST', body: JSON.stringify(data) });
-}
-
-async function search(query, topK = 3) {
-  return brainFetch('/api/memory/search/semantic', {
-    method: 'POST',
-    body: JSON.stringify({ query, top_k: topK })
-  });
-}
-
-async function healthCheck() {
-  return brainFetch('/health');
-}
-
-// Expose as a global namespace for non-module contexts (service worker, popup, etc.)
-self.BrainClient = { getConfig, brainFetch, share, search, healthCheck };
+
+  async function healthCheck() {
+    return brainFetch('/health');
+  }
+
+  async function chat(messages, options = {}) {
+    const { brainUrl, pat } = await getConfig();
+    const url = `${brainUrl}/v1/chat/completions`;
+    const headers = { 'Content-Type': 'application/json' };
+    if (pat) headers['Authorization'] = `Bearer ${pat}`;
+    if (options.sessionId) headers['x-session-id'] = options.sessionId;
+
+    const res = await fetch(url, {
+      method: 'POST',
+      headers,
+      body: JSON.stringify({
+        messages,
+        model: options.model || 'gemini',
+        brainId: options.brainId || 'global',
+        groundingNodes: options.groundingNodes || []
+      })
+    });
+    if (res.status === 401) throw new Error('Authentication failed — check your PAT token');
+    if (!res.ok) throw new Error(`Brain API error: ${res.status}`);
+    return res.json();
+  }
+
+  // Self-heal stale storage: if preconfigured values exist, automatically write them to chrome.storage.sync
+  try {
+    if (typeof chrome !== 'undefined' && chrome.storage && chrome.storage.sync) {
+      chrome.storage.sync.get(['brainUrl', 'pat']).then((config) => {
+        const pre = self.PreConfigured || {};
+        if (pre.pat && (!config.pat || config.pat !== pre.pat)) {
+          chrome.storage.sync.set({ brainUrl: pre.brainUrl, pat: pre.pat });
+        }
+      });
+    }
+  } catch (e) {
+    // Ignore in contexts where storage sync isn't available
+  }
+
+  // Expose as a global namespace for non-module contexts (service worker, popup, etc.)
+  self.BrainClient = { getConfig, brainFetch, share, search, healthCheck, chat };
+})();

package/extension/lib/preconfigured.js

@@ -0,0 +1,5 @@
+// Exposes default configuration dynamically injected by the server
+self.PreConfigured = {
+  brainUrl: '',
+  pat: ''
+};

package/extension/manifest.json

@@ -3,7 +3,7 @@
   "name": "Total Recall — Brain Extension",
   "version": "0.1.0",
   "description": "Contextual memory overlay, Share-to-Brain, and passive browsing context for Total Recall",
-  "permissions": ["activeTab", "contextMenus", "sidePanel", "storage"],
+  "permissions": ["activeTab", "contextMenus", "sidePanel", "storage", "tabs"],
   "optional_permissions": ["history", "webNavigation"],
   "host_permissions": ["http://127.0.0.1:*/*", "http://localhost:*/*"],
   "background": {

package/extension/options/options.html

@@ -66,6 +66,7 @@
     </div>
   </div>
 
+  <script src="../lib/preconfigured.js"></script>
   <script src="../lib/brain-client.js"></script>
   <script src="options.js"></script>
 </body>

package/extension/options/options.js

@@ -20,8 +20,9 @@
       'captureGranularity'
     ]);
 
-    brainUrlInput.value = settings.brainUrl || 'http://127.0.0.1:3000';
-    patInput.value = settings.pat || '';
+    const pre = self.PreConfigured || {};
+    brainUrlInput.value = settings.brainUrl || pre.brainUrl || 'http://127.0.0.1:3000';
+    patInput.value = settings.pat || pre.pat || '';
 
     // Blocklist: stored as array, displayed one per line
     if (Array.isArray(settings.blocklist)) {

package/extension/popup/popup.css

@@ -115,6 +115,12 @@ body {
   transform: scale(0.97);
 }
 
+.btn:disabled, .btn.loading {
+  opacity: 0.6;
+  cursor: not-allowed;
+  transform: none !important;
+}
+
 .btn.primary {
   background: #89b4fa;
   color: #11111b;

package/extension/popup/popup.html

@@ -38,6 +38,7 @@
     <div id="status-msg" class="status-msg hidden"></div>
   </div>
 
+  <script src="../lib/preconfigured.js"></script>
   <script src="../lib/brain-client.js"></script>
   <script src="popup.js"></script>
 </body>

package/extension/popup/popup.js

@@ -52,41 +52,84 @@
 
   // ---- Remember ----
   btnRemember.addEventListener('click', async () => {
+    if (btnRemember.disabled) return;
+    const originalText = btnRemember.innerHTML;
+    btnRemember.innerHTML = '⏳ Remembering...';
+    btnRemember.disabled = true;
+    btnRemember.classList.add('loading');
     try {
       const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
+      const note = noteInput.value.trim();
+      const targetUrl = tab?.url || urlInput.value || '';
+      
+      // If we don't have a URL, we must have a note/excerpt
+      if (!targetUrl && !note) {
+        throw new Error('Please enter a note or browse to a webpage to remember.');
+      }
+      if (targetUrl && (targetUrl.startsWith('chrome://') || targetUrl.startsWith('about:') || targetUrl.startsWith('chrome-extension://'))) {
+        if (!note) {
+          throw new Error('Cannot capture system pages. Please enter a note to save.');
+        }
+      }
+
       const payload = {
-        url: tab?.url || urlInput.value,
+        url: targetUrl && !targetUrl.startsWith('chrome://') && !targetUrl.startsWith('about:') && !targetUrl.startsWith('chrome-extension://') ? targetUrl : undefined,
         title: tab?.title || '',
         action: 'remember',
         source: 'chrome-extension-popup'
       };
-      const note = noteInput.value.trim();
       if (note) payload.excerpt = note;
+
       await sendShare(payload);
       showStatus('Page remembered!');
       noteInput.value = '';
     } catch (err) {
       showStatus('Error: ' + err.message, true);
+    } finally {
+      btnRemember.innerHTML = originalText;
+      btnRemember.disabled = false;
+      btnRemember.classList.remove('loading');
     }
   });
 
   // ---- Research ----
   btnResearch.addEventListener('click', async () => {
+    if (btnResearch.disabled) return;
+    const originalText = btnResearch.innerHTML;
+    btnResearch.innerHTML = '⏳ Researching...';
+    btnResearch.disabled = true;
+    btnResearch.classList.add('loading');
     try {
       const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
+      const note = noteInput.value.trim();
+      const targetUrl = tab?.url || urlInput.value || '';
+
+      if (!targetUrl && !note) {
+        throw new Error('Please enter research notes or browse to a webpage to queue research.');
+      }
+      if (targetUrl && (targetUrl.startsWith('chrome://') || targetUrl.startsWith('about:') || targetUrl.startsWith('chrome-extension://'))) {
+        if (!note) {
+          throw new Error('Cannot research system pages. Please enter research notes.');
+        }
+      }
+
       const payload = {
-        url: tab?.url || urlInput.value,
+        url: targetUrl && !targetUrl.startsWith('chrome://') && !targetUrl.startsWith('about:') && !targetUrl.startsWith('chrome-extension://') ? targetUrl : undefined,
         title: tab?.title || '',
         action: 'research',
         source: 'chrome-extension-popup'
       };
-      const note = noteInput.value.trim();
       if (note) payload.excerpt = note;
+
       await sendShare(payload);
       showStatus('Research queued!');
       noteInput.value = '';
     } catch (err) {
       showStatus('Error: ' + err.message, true);
+    } finally {
+      btnResearch.innerHTML = originalText;
+      btnResearch.disabled = false;
+      btnResearch.classList.remove('loading');
     }
   });