total-recall-brain 3.3.0 → 3.5.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/bin/antigravity.mjs

@@ -83,10 +83,9 @@ if (!process.env.GOOGLE_API_KEY) {
   }
 }
 
-const apiKey = process.env.GOOGLE_API_KEY;
+const apiKey = process.env.GOOGLE_API_KEY || '';
 if (!apiKey) {
-  console.error('Error: GOOGLE_API_KEY environment variable is not configured.');
-  process.exit(1);
+  console.warn('Warning: GOOGLE_API_KEY environment variable is not configured. Attempting connection via GCP Application Default Credentials or ambient environment...');
 }
 
 // Resolve the model dynamically from the API registry based on the category/alias selected
@@ -109,7 +108,13 @@ async function resolveGenerativeModel(selectedModel) {
   }
 
   try {
-    const res = await fetch(`https://generativelanguage.googleapis.com/v1beta/models?key=${apiKey}`, {
+    const keyParam = apiKey ? `key=${apiKey}` : '';
+    const headers = {};
+    if (!apiKey && process.env.GCLOUD_TOKEN) {
+      headers['Authorization'] = `Bearer ${process.env.GCLOUD_TOKEN}`;
+    }
+    const res = await fetch(`https://generativelanguage.googleapis.com/v1beta/models?${keyParam}`, {
+      headers,
       signal: AbortSignal.timeout(5000)
     });
     if (res.ok) {
@@ -154,13 +159,19 @@ async function resolveGenerativeModel(selectedModel) {
 async function main() {
   try {
     const resolvedModel = await resolveGenerativeModel(requestedModel);
-    const url = `https://generativelanguage.googleapis.com/v1beta/models/${resolvedModel}:generateContent?key=${apiKey}`;
+    const keyParam = apiKey ? `?key=${apiKey}` : '';
+    const url = `https://generativelanguage.googleapis.com/v1beta/models/${resolvedModel}:generateContent${keyParam}`;
+
+    const headers = {
+      'Content-Type': 'application/json',
+    };
+    if (!apiKey && process.env.GCLOUD_TOKEN) {
+      headers['Authorization'] = `Bearer ${process.env.GCLOUD_TOKEN}`;
+    }
 
     const res = await fetch(url, {
       method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-      },
+      headers,
       body: JSON.stringify({
         contents: [{
           parts: [{ text: prompt }]

package/extension/background.js

@@ -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

@@ -6,8 +6,21 @@
   const DEFAULT_BRAIN_URL = 'http://127.0.0.1:3000';
 
   async function getConfig() {
-    const config = await chrome.storage.sync.get(['brainUrl', 'pat']);
+    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
+    };
+
+    if (syncConfig.pat && !localConfig.pat) {
+      await chrome.storage.local.set({ pat: syncConfig.pat });
+      await chrome.storage.sync.remove('pat');
+    }
+
     return {
       brainUrl: config.brainUrl || pre.brainUrl || DEFAULT_BRAIN_URL,
       pat: config.pat || pre.pat || ''
@@ -67,13 +80,13 @@
     return res.json();
   }
 
-  // Self-heal stale storage: if preconfigured values exist, automatically write them to chrome.storage.sync
+  // Self-heal stale storage: if preconfigured values exist, write them to extension-local storage.
   try {
-    if (typeof chrome !== 'undefined' && chrome.storage && chrome.storage.sync) {
-      chrome.storage.sync.get(['brainUrl', 'pat']).then((config) => {
+    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.sync.set({ brainUrl: pre.brainUrl, pat: pre.pat });
+          chrome.storage.local.set({ brainUrl: pre.brainUrl, pat: pre.pat });
         }
       });
     }

package/extension/options/options.js

@@ -13,12 +13,14 @@
 
   // ---- 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 };
 
     const pre = self.PreConfigured || {};
     brainUrlInput.value = settings.brainUrl || pre.brainUrl || 'http://127.0.0.1:3000';
@@ -48,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/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),
   })
@@ -686,3 +686,22 @@ export async function fetchExtensionStatus(): Promise<{ available: boolean; conn
   return res.json()
 }
 
+// ─── Auto-Update APIs ──────────────────────────────────────────────────────────
+
+export interface UpdateCheckResult {
+  currentVersion: string
+  latestVersion: string
+  updateAvailable: boolean
+}
+
+export async function checkUpdate(): Promise<UpdateCheckResult> {
+  const res = await apiFetch(`${API_BASE}/api/update/check`)
+  if (!res.ok) throw new Error(`Check update API error: ${res.status}`)
+  return res.json()
+}
+
+export async function runUpdate(): Promise<{ success: boolean; message: string }> {
+  const res = await apiFetch(`${API_BASE}/api/update/run`, { method: 'POST' })
+  if (!res.ok) throw new Error(`Run update API error: ${res.status}`)
+  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');
         }
       })
 

package/frontend/src/pages/HealthPage.tsx

@@ -1,6 +1,6 @@
 import { useState, useEffect } from 'react'
-import { fetchHealth } from '../api'
-import type { HealthData } from '../types'
+import { fetchHealth, checkUpdate, runUpdate } from '../api'
+import type { HealthData, UpdateCheckResult } from '../types'
 
 function formatUptime(seconds: number): string {
   const d = Math.floor(seconds / 86400)
@@ -15,6 +15,12 @@ export default function HealthPage() {
   const [health, setHealth] = useState<HealthData | null>(null)
   const [error, setError] = useState('')
   const [lastPoll, setLastPoll] = useState<Date | null>(null)
+  
+  // Auto-Update States
+  const [updateInfo, setUpdateInfo] = useState<UpdateCheckResult | null>(null)
+  const [isUpdating, setIsUpdating] = useState(false)
+  const [showConfirmModal, setShowConfirmModal] = useState(false)
+  const [updateError, setUpdateError] = useState('')
 
   useEffect(() => {
     let active = true
@@ -33,17 +39,166 @@ export default function HealthPage() {
 
     poll()
     const interval = setInterval(poll, 5000)
+
+    // Check for updates
+    const checkVersion = async () => {
+      try {
+        const data = await checkUpdate()
+        if (active) setUpdateInfo(data)
+      } catch (e) {
+        console.error('Failed to check updates', e)
+      }
+    }
+    checkVersion()
+
     return () => { active = false; clearInterval(interval) }
   }, [])
 
+  const handleStartUpdate = async () => {
+    setShowConfirmModal(false)
+    setIsUpdating(true)
+    setUpdateError('')
+    
+    try {
+      await runUpdate()
+      
+      // Keep polling health to check when it goes down and recovers
+      let serverDown = false
+      const pollInterval = setInterval(async () => {
+        try {
+          const res = await fetchHealth()
+          if (serverDown && res.status === 'healthy') {
+            clearInterval(pollInterval)
+            window.location.reload()
+          }
+        } catch {
+          // Connection failed = server is offline / restarting
+          serverDown = true
+        }
+      }, 2000)
+
+    } catch (e) {
+      setUpdateError((e as Error).message)
+      setIsUpdating(false)
+    }
+  }
+
   return (
-    <div className="page">
-      <div className="page-header">
-        <h1>System Health</h1>
-        <p>Live monitoring of your sovereign brain {lastPoll && <span>· Last poll: {lastPoll.toLocaleTimeString()}</span>}</p>
+    <div className="page" style={{ position: 'relative' }}>
+      {isUpdating && (
+        <div style={{
+          position: 'fixed',
+          inset: 0,
+          background: 'rgba(13, 17, 23, 0.9)',
+          zIndex: 1000,
+          display: 'flex',
+          flexDirection: 'column',
+          alignItems: 'center',
+          justifyContent: 'center',
+          backdropFilter: 'blur(8px)',
+          animation: 'fadeIn 0.3s ease'
+        }}>
+          <div className="spinner" style={{
+            width: 50,
+            height: 50,
+            border: '4px solid var(--border)',
+            borderTopColor: 'var(--accent)',
+            borderRadius: '50%',
+            animation: 'spin 1s linear infinite',
+            marginBottom: 20
+          }} />
+          <h2 style={{ color: 'var(--text-primary)', marginBottom: 8 }}>Updating Total Recall...</h2>
+          <p style={{ color: 'var(--text-muted)', fontSize: 14 }}>Applying git changes, installing npm modules, and restarting daemon.</p>
+          <p style={{ color: 'var(--accent)', fontSize: 13, marginTop: 10 }}>Please do not close this window.</p>
+        </div>
+      )}
+
+      {showConfirmModal && (
+        <div style={{
+          position: 'fixed',
+          inset: 0,
+          background: 'rgba(0, 0, 0, 0.7)',
+          zIndex: 900,
+          display: 'flex',
+          alignItems: 'center',
+          justifyContent: 'center',
+          backdropFilter: 'blur(4px)',
+          padding: 20
+        }}>
+          <div className="card" style={{
+            maxWidth: 450,
+            width: '100%',
+            backgroundColor: 'var(--bg-surface)',
+            border: '1px solid var(--border)',
+            borderRadius: 12,
+            padding: 24,
+            boxShadow: '0 20px 40px rgba(0, 0, 0, 0.5)'
+          }}>
+            <h3 style={{ marginBottom: 12, color: 'var(--text-primary)' }}>Confirm Core Self-Update</h3>
+            <p style={{ color: 'var(--text-muted)', marginBottom: 20, fontSize: 14, lineHeight: 1.5 }}>
+              This will pull the latest version of Total Recall ({updateInfo?.latestVersion}) from the git repository, reinstall dependencies, rebuild frontend bundles, and reboot the system kernel.
+            </p>
+            <div style={{ display: 'flex', gap: 12, justifyContent: 'flex-end' }}>
+              <button 
+                className="btn btn-secondary" 
+                onClick={() => setShowConfirmModal(false)}
+                style={{ padding: '8px 16px', borderRadius: 8 }}
+              >
+                Cancel
+              </button>
+              <button 
+                className="btn btn-primary" 
+                onClick={handleStartUpdate}
+                style={{ padding: '8px 16px', borderRadius: 8, background: 'var(--accent)' }}
+              >
+                Update Now
+              </button>
+            </div>
+          </div>
+        </div>
+      )}
+
+      <div className="page-header" style={{ display: 'flex', justifyContent: 'space-between', alignItems: 'center' }}>
+        <div>
+          <h1>System Health</h1>
+          <p>Live monitoring of your sovereign brain {lastPoll && <span>· Last poll: {lastPoll.toLocaleTimeString()}</span>}</p>
+        </div>
+        {updateInfo?.updateAvailable && (
+          <div className="badge badge-success" style={{
+            display: 'flex',
+            alignItems: 'center',
+            gap: 12,
+            padding: '12px 18px',
+            borderRadius: 12,
+            background: 'rgba(63, 185, 80, 0.1)',
+            border: '1px solid rgba(63, 185, 80, 0.3)',
+            animation: 'pulse 2s infinite'
+          }}>
+            <span style={{ fontSize: 14, color: 'var(--text-primary)', fontWeight: 600 }}>
+              New version available: {updateInfo.latestVersion}
+            </span>
+            <button 
+              className="btn btn-primary" 
+              onClick={() => setShowConfirmModal(true)}
+              style={{
+                fontSize: 12,
+                padding: '6px 12px',
+                borderRadius: 6,
+                backgroundColor: '#3fb950',
+                color: '#fff',
+                border: 'none',
+                cursor: 'pointer',
+                fontWeight: 600
+              }}
+            >
+              Update Now
+            </button>
+          </div>
+        )}
       </div>
 
       {error && <div className="badge badge-error" style={{ marginBottom: 16 }}>⚠️ {error}</div>}
+      {updateError && <div className="badge badge-error" style={{ marginBottom: 16 }}>⚠️ Update failed: {updateError}</div>}
 
       {health ? (
         <>
@@ -55,7 +210,9 @@ export default function HealthPage() {
               <div className="label">Status</div>
             </div>
             <div className="stat-card">
-              <div className="value" style={{ color: 'var(--accent)' }}>{health.version}</div>
+              <div className="value" style={{ color: 'var(--accent)' }}>
+                {health.version} {updateInfo && !updateInfo.updateAvailable && <span style={{ fontSize: 12, color: 'var(--text-muted)' }}>(Latest)</span>}
+              </div>
               <div className="label">Version</div>
             </div>
             <div className="stat-card">
@@ -76,6 +233,10 @@ export default function HealthPage() {
                 <tr><th>Version</th><td>{health.version}</td></tr>
                 <tr><th>Uptime</th><td>{formatUptime(health.uptime_seconds)}</td></tr>
                 <tr><th>Timestamp</th><td>{health.timestamp}</td></tr>
+                <tr><th>Daemon Status</th><td>{health.daemon || 'unknown'}</td></tr>
+                {health.cli_agents && (
+                  <tr><th>CLI Agents</th><td>{health.cli_agents.join(', ')}</td></tr>
+                )}
                 <tr><th>Polling</th><td>Every 5 seconds</td></tr>
               </tbody>
             </table>
@@ -86,6 +247,17 @@ export default function HealthPage() {
           {[1, 2, 3, 4].map(i => <div key={i} className="skeleton" style={{ height: 100 }} />)}
         </div>
       ) : null}
+
+      <style>{`
+        @keyframes spin {
+          0% { transform: rotate(0deg); }
+          100% { transform: rotate(360deg); }
+        }
+        @keyframes fadeIn {
+          from { opacity: 0; }
+          to { opacity: 1; }
+        }
+      `}</style>
     </div>
   )
 }

package/frontend/src/pages/IntegrationsPage.tsx

@@ -391,10 +391,12 @@ export default function IntegrationsPage() {
           )
 
           if (matchingKey) {
-            if (matchingKey.hit_count > 0) {
+            if (matchingKey.hit_count > 0 || preset.id === 'antigravity') {
               const lastUsedStr = matchingKey.last_used_at 
                 ? ` (Active ${new Date(matchingKey.last_used_at).toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' })})`
-                : ''
+                : preset.id === 'antigravity'
+                  ? ' (Active Now)'
+                  : ''
               statusBadge = (
                 <span style={{
                   display: 'inline-flex',