total-recall-brain 3.2.3 → 3.4.0

package/README.md

@@ -1,6 +1,6 @@
 # Total Recall — SSSS Sovereign AI OS
 
-[![Version](https://img.shields.io/badge/version-3.0.0-indigo.svg)](package.json)
+[![Version](https://img.shields.io/badge/version-3.3.0-indigo.svg)](package.json)
 [![License](https://img.shields.io/badge/license-MIT-emerald.svg)](LICENSE)
 [![Continuous Integration](https://img.shields.io/badge/CI-passing-teal.svg)](.github/workflows/test.yml)
 

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;
@@ -75,7 +75,7 @@ function showBadge(text, color) {
 // ---------------------------------------------------------------------------
 // Message listener (popup, content-script, sidepanel → background)
 // ---------------------------------------------------------------------------
-chrome.runtime.onMessage.addListener((msg, _sender, sendResponse) => {
+chrome.runtime.onMessage.addListener((msg, sender, sendResponse) => {
   if (msg.type === 'QUERY_BRAIN') {
     search(msg.query, msg.topK || 3)
       .then(data => sendResponse({ memories: data.results || [] }))
@@ -96,4 +96,16 @@ chrome.runtime.onMessage.addListener((msg, _sender, sendResponse) => {
       .catch(() => sendResponse({ connected: false }));
     return true;
   }
+
+  if (msg.type === 'OPEN_SIDE_PANEL') {
+    const tabId = sender.tab?.id;
+    if (!chrome.sidePanel?.open || !tabId) {
+      sendResponse({ success: false, error: 'Side panel is unavailable for this tab.' });
+      return false;
+    }
+    chrome.sidePanel.open({ tabId })
+      .then(() => sendResponse({ success: true }))
+      .catch(err => sendResponse({ success: false, error: err.message }));
+    return true;
+  }
 });

package/extension/content-script.js

@@ -112,8 +112,25 @@
     document.body.appendChild(pillHost);
   }
 
+  function isBlocked(blocklist) {
+    if (!Array.isArray(blocklist) || blocklist.length === 0) return false;
+    let host = '';
+    try {
+      host = new URL(pageContext.url).hostname.toLowerCase();
+    } catch {
+      return false;
+    }
+    return blocklist.some((entry) => {
+      const pattern = String(entry || '').trim().toLowerCase();
+      return pattern && (host === pattern || host.endsWith(`.${pattern}`) || pageContext.url.toLowerCase().includes(pattern));
+    });
+  }
+
   // ---- Query brain after delay ----
-  function queryBrain() {
+  async function queryBrain() {
+    const { passiveTracking = false, blocklist = [] } = await chrome.storage.sync.get(['passiveTracking', 'blocklist']);
+    if (!passiveTracking || isBlocked(blocklist)) return;
+
     const query = pageContext.title || pageContext.url;
     if (!query) return;
 

package/extension/lib/brain-client.js

@@ -2,64 +2,98 @@
 // 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';
+(function () {
+  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 getConfig() {
+    const [localConfig, syncConfig] = await Promise.all([
+      chrome.storage.local.get(['brainUrl', 'pat']),
+      chrome.storage.sync.get(['brainUrl', 'pat'])
+    ]);
+    const pre = self.PreConfigured || {};
+    const config = {
+      brainUrl: localConfig.brainUrl || syncConfig.brainUrl,
+      pat: localConfig.pat || syncConfig.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}`;
+    if (syncConfig.pat && !localConfig.pat) {
+      await chrome.storage.local.set({ pat: syncConfig.pat });
+      await chrome.storage.sync.remove('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');
+    return {
+      brainUrl: config.brainUrl || pre.brainUrl || DEFAULT_BRAIN_URL,
+      pat: config.pat || pre.pat || ''
+    };
   }
-  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 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}`;
 
-async function search(query, topK = 3) {
-  return brainFetch('/api/memory/search/semantic', {
-    method: 'POST',
-    body: JSON.stringify({ query, top_k: topK })
-  });
-}
+    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 })
+    });
+  }
 
-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;
+  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();
-}
+    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, write them to extension-local storage.
+  try {
+    if (typeof chrome !== 'undefined' && chrome.storage && chrome.storage.local) {
+      chrome.storage.local.get(['brainUrl', 'pat']).then((config) => {
+        const pre = self.PreConfigured || {};
+        if (pre.pat && (!config.pat || config.pat !== pre.pat)) {
+          chrome.storage.local.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 };
+  // 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

@@ -13,15 +13,18 @@
 
   // ---- Load saved settings ----
   async function loadSettings() {
-    const settings = await chrome.storage.sync.get([
-      'brainUrl',
-      'pat',
+    const [localSettings, syncSettings] = await Promise.all([
+      chrome.storage.local.get(['brainUrl', 'pat']),
+      chrome.storage.sync.get([
       'blocklist',
       'captureGranularity'
+      ])
     ]);
+    const settings = { ...syncSettings, ...localSettings };
 
-    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)) {
@@ -47,9 +50,12 @@
       const granularityRadio = document.querySelector('input[name="granularity"]:checked');
       const captureGranularity = granularityRadio ? granularityRadio.value : 'minimal';
 
-      await chrome.storage.sync.set({
+      await chrome.storage.local.set({
         brainUrl: brainUrlInput.value.trim() || 'http://127.0.0.1:3000',
         pat: patInput.value.trim(),
+      });
+      await chrome.storage.sync.remove('pat');
+      await chrome.storage.sync.set({
         blocklist,
         captureGranularity
       });

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

@@ -59,14 +59,27 @@
     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 = '';
@@ -88,14 +101,26 @@
     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 = '';

package/extension/sidepanel/sidepanel.html

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

package/extension/sidepanel/sidepanel.js

@@ -183,9 +183,15 @@
     btnRemember.classList.add('loading');
     try {
       const tab = await getCurrentTab();
+      if (!tab || !tab.url) {
+        throw new Error('No active webpage found to capture. Make sure you are on a webpage tab.');
+      }
+      if (tab.url.startsWith('chrome://') || tab.url.startsWith('about:') || tab.url.startsWith('chrome-extension://')) {
+        throw new Error('Cannot capture system pages.');
+      }
       await sendShare({
         url: tab.url,
-        title: tab.title,
+        title: tab.title || 'Untitled Page',
         action: 'remember',
         source: 'chrome-extension-sidepanel'
       });
@@ -208,9 +214,15 @@
     btnResearch.classList.add('loading');
     try {
       const tab = await getCurrentTab();
+      if (!tab || !tab.url) {
+        throw new Error('No active webpage found to capture. Make sure you are on a webpage tab.');
+      }
+      if (tab.url.startsWith('chrome://') || tab.url.startsWith('about:') || tab.url.startsWith('chrome-extension://')) {
+        throw new Error('Cannot capture system pages.');
+      }
       await sendShare({
         url: tab.url,
-        title: tab.title,
+        title: tab.title || 'Untitled Page',
         action: 'research',
         source: 'chrome-extension-sidepanel'
       });
@@ -245,13 +257,19 @@
     btnSaveNote.disabled = true;
     btnSaveNote.classList.add('loading');
     try {
-      const tab = await getCurrentTab();
+      let tab = null;
+      try {
+        tab = await getCurrentTab();
+      } catch {
+        // ignore
+      }
+      const hasValidUrl = tab && tab.url && !tab.url.startsWith('chrome://') && !tab.url.startsWith('about:') && !tab.url.startsWith('chrome-extension://');
       await sendShare({
-        url: tab.url,
-        title: `Note: ${text.slice(0, 50)}`,
+        url: hasValidUrl ? tab.url : undefined,
+        title: hasValidUrl ? `Note: ${text.slice(0, 50)}` : undefined,
         excerpt: text,
         action: 'remember',
-        category: 'fact',
+        category: 'facts',
         source: 'chrome-extension-sidepanel'
       });
       noteInput.value = '';

package/frontend/index.html

@@ -5,6 +5,7 @@
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <meta name="description" content="Total Recall — Sovereign AI Brain Dashboard. Memory explorer, chat interface, code sandbox, and system health monitoring." />
     <meta name="theme-color" content="#0a0a0f" />
+    <link rel="manifest" href="/manifest.json" />
     <link rel="preconnect" href="https://fonts.googleapis.com" />
     <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin />
     <link href="https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&family=JetBrains+Mono:wght@400;500&display=swap" rel="stylesheet" />
@@ -13,5 +14,12 @@
   <body>
     <div id="root"></div>
     <script type="module" src="/src/main.tsx"></script>
+    <script>
+      if ('serviceWorker' in navigator) {
+        window.addEventListener('load', () => {
+          navigator.serviceWorker.register('/sw.js').catch(() => {});
+        });
+      }
+    </script>
   </body>
 </html>

package/frontend/src/api.ts

@@ -563,7 +563,7 @@ export async function resolveConflict(
 
 export async function saveMemory(slug: string, node: Partial<MemoryNode>): Promise<MemoryNode> {
   const res = await apiFetch(`${API_BASE}/api/memory/${slug}`, {
-    method: "PUT",
+    method: "PATCH",
     headers: { "Content-Type": "application/json" },
     body: JSON.stringify(node),
   })
@@ -685,4 +685,3 @@ export async function fetchExtensionStatus(): Promise<{ available: boolean; conn
   if (!res.ok) return { available: false, connected: false }
   return res.json()
 }
-

package/frontend/src/pages/ChatPage.tsx

@@ -57,7 +57,10 @@ export default function ChatPage({ activeBrainId, onBrainChange }: { activeBrain
 
   // Model Selector state
   const [availableModels, setAvailableModels] = useState<string[]>([])
-  const [selectedModel, setSelectedModel] = useState<string>(() => localStorage.getItem('selectedModel') || '')
+  const [selectedModel, setSelectedModel] = useState<string>(() => {
+    const saved = localStorage.getItem('selectedModel');
+    return (saved && saved !== 'gemini') ? saved : 'antigravity';
+  })
   const [selectedSubModel, setSelectedSubModel] = useState<string>(() => localStorage.getItem('selectedSubModel') || '')
   const [geminiModels, setGeminiModels] = useState<{ id: string; displayName: string }[]>([])
   const [showModelDropdown, setShowModelDropdown] = useState(false)
@@ -104,27 +107,27 @@ export default function ChatPage({ activeBrainId, onBrainChange }: { activeBrain
       .then(health => {
         const agents = health.cli_agents && health.cli_agents.length > 0
           ? health.cli_agents
-          : ['antigravity', 'gemini', 'claude', 'codex'];
+          : ['antigravity', 'claude', 'codex'];
         setAvailableModels(agents);
         
         const saved = localStorage.getItem('selectedModel');
         if (saved && agents.includes(saved)) {
           setSelectedModel(saved);
         } else {
-          const defaultAgent = agents.includes('gemini') ? 'gemini' : agents[0];
+          const defaultAgent = agents.includes('antigravity') ? 'antigravity' : agents[0];
           setSelectedModel(defaultAgent);
           localStorage.setItem('selectedModel', defaultAgent);
         }
       })
       .catch(() => {
-        const agents = ['antigravity', 'gemini', 'claude', 'codex'];
+        const agents = ['antigravity', 'claude', 'codex'];
         setAvailableModels(agents);
         const saved = localStorage.getItem('selectedModel');
         if (saved && agents.includes(saved)) {
           setSelectedModel(saved);
         } else {
-          setSelectedModel('gemini');
-          localStorage.setItem('selectedModel', 'gemini');
+          setSelectedModel('antigravity');
+          localStorage.setItem('selectedModel', 'antigravity');
         }
       })