agentgui 1.0.744 → 1.0.746

package/database.js

@@ -595,6 +595,21 @@ try {
   console.error('[Migration] Backfill error:', err.message);
 }
 
+try {
+  const hasFts = db.prepare("SELECT name FROM sqlite_master WHERE type='table' AND name='messages_fts'").get();
+  if (!hasFts) {
+    db.exec("CREATE VIRTUAL TABLE messages_fts USING fts5(content, conversationId UNINDEXED, role UNINDEXED, content_rowid='rowid')");
+    const msgs = db.prepare("SELECT rowid, content, conversationId, role FROM messages").all();
+    if (msgs.length > 0) {
+      const ins = db.prepare("INSERT INTO messages_fts(rowid, content, conversationId, role) VALUES (?, ?, ?, ?)");
+      const tx = db.transaction(() => { for (const m of msgs) ins.run(m.rowid, m.content, m.conversationId, m.role); });
+      tx();
+      console.log(`[Migration] FTS5 index created with ${msgs.length} messages`);
+    }
+  }
+} catch (err) {
+  console.error('[Migration] FTS5 error:', err.message);
+}
 
 const stmtCache = new Map();
 function prep(sql) {
@@ -774,6 +789,8 @@ export const queries = {
     );
     stmt.run(id, conversationId, role, storedContent, now);
 
+    try { prep('INSERT INTO messages_fts(rowid, content, conversationId, role) VALUES ((SELECT rowid FROM messages WHERE id = ?), ?, ?, ?)').run(id, storedContent, conversationId, role); } catch (_) {}
+
     const updateConvStmt = prep('UPDATE conversations SET updated_at = ? WHERE id = ?');
     updateConvStmt.run(now, conversationId);
 
@@ -1924,6 +1941,21 @@ export const queries = {
     return stmt.run(toolId, toolId, keepCount).changes;
   },
 
+  searchMessages(query, limit = 50) {
+    try {
+      const stmt = prep(`
+        SELECT m.id, m.conversationId, m.role, m.content, m.created_at,
+               c.title as conversationTitle, c.agentType
+        FROM messages_fts fts
+        JOIN messages m ON m.rowid = fts.rowid
+        JOIN conversations c ON c.id = m.conversationId
+        WHERE messages_fts MATCH ?
+        ORDER BY m.created_at DESC LIMIT ?
+      `);
+      return stmt.all(query, limit);
+    } catch (_) { return []; }
+  },
+
   // ============ ACP-COMPATIBLE QUERIES ============
   ...createACPQueries(db, prep)
 };

package/lib/ws-handlers-conv.js

@@ -135,6 +135,53 @@ export function register(router, deps) {
     return { markdown: md, title: conv.title };
   });
 
+  router.handle('conv.sync', (p) => {
+    const conv = queries.getConversation(p.id);
+    if (!conv) notFound();
+    const machineSnap = execMachine.snapshot(p.id);
+    const executionState = machineSnap?.value || 'idle';
+    const latestSession = queries.getLatestSession(p.id);
+    const sinceSeq = parseInt(p.sinceSeq || '-1');
+    const since = parseInt(p.since || '0');
+    let missedChunks = [];
+    if (latestSession && sinceSeq >= 0) {
+      missedChunks = queries.getChunksSinceSeq(latestSession.id, sinceSeq);
+    } else if (latestSession && since > 0) {
+      missedChunks = queries.getConversationChunksSince(p.id, since);
+    }
+    return {
+      conversation: conv,
+      executionState,
+      isActivelyStreaming: execMachine.isActive(p.id),
+      latestSession,
+      missedChunks,
+      missedCount: missedChunks.length,
+      queueLength: machineSnap?.context?.queue?.length || 0
+    };
+  });
+
+  router.handle('conv.search', (p) => {
+    if (!p.query || typeof p.query !== 'string' || p.query.trim().length < 2) return { results: [] };
+    const limit = Math.min(parseInt(p.limit || '50'), 200);
+    return { results: queries.searchMessages(p.query.trim(), limit) };
+  });
+
+  router.handle('conv.prune', (p) => {
+    const conv = queries.getConversation(p.id);
+    if (!conv) notFound();
+    const keep = Math.max(parseInt(p.keep || '500'), 100);
+    const sessions = queries.getConversationSessions(p.id);
+    if (!sessions || sessions.length === 0) return { pruned: 0 };
+    const latestSessionId = sessions[0]?.id;
+    let pruned = 0;
+    for (const sess of sessions) {
+      if (sess.id === latestSessionId) continue;
+      const count = queries.getSessionChunks(sess.id)?.length || 0;
+      if (count > 0) { queries.deleteSessionChunks(sess.id); pruned += count; }
+    }
+    return { pruned, kept: latestSessionId, sessionsProcessed: sessions.length };
+  });
+
   router.handle('conv.cancel', (p) => {
     if (!execMachine.isActive(p.id)) notFound('No active execution to cancel');
     const ctx = execMachine.getContext(p.id);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentgui",
-  "version": "1.0.744",
+  "version": "1.0.746",
   "description": "Multi-agent ACP client with real-time communication",
   "type": "module",
   "main": "electron/main.js",

package/server.js

@@ -2116,6 +2116,28 @@ const server = http.createServer(async (req, res) => {
       return;
     }
 
+    if (pathOnly === '/api/restore' && req.method === 'POST') {
+      const dbPath = path.join(os.homedir(), '.gmgui', 'data.db');
+      const backupPath = dbPath + '.bak-' + Date.now();
+      const chunks = [];
+      req.on('data', (chunk) => chunks.push(chunk));
+      req.on('end', () => {
+        try {
+          const body = Buffer.concat(chunks);
+          if (body.length < 100 || body.slice(0, 16).toString() !== 'SQLite format 3\0') {
+            sendJSON(req, res, 400, { error: 'Invalid SQLite database file' });
+            return;
+          }
+          fs.copyFileSync(dbPath, backupPath);
+          fs.writeFileSync(dbPath, body);
+          sendJSON(req, res, 200, { success: true, backupPath, size: body.length });
+        } catch (e) {
+          sendJSON(req, res, 500, { error: e.message });
+        }
+      });
+      return;
+    }
+
     if (pathOnly === '/api/debug/machines' && req.method === 'GET' && process.env.DEBUG) {
       const toolSnap = {};
       for (const [id, actor] of toolInstallMachine.getMachineActors()) {

package/static/js/client.js

@@ -915,12 +915,14 @@ class AgentGUIClient {
     }
 
     this._streamStartedAt = Date.now();
+    this._sessionCost = 0;
     if (this._elapsedTimer) clearInterval(this._elapsedTimer);
     this._elapsedTimer = setInterval(() => {
       const label = streamingDiv?.querySelector('.streaming-indicator-label');
       if (label) {
         const sec = ((Date.now() - this._streamStartedAt) / 1000) | 0;
-        label.textContent = sec < 60 ? sec + 's' : Math.floor(sec / 60) + 'm ' + (sec % 60) + 's';
+        const time = sec < 60 ? sec + 's' : Math.floor(sec / 60) + 'm ' + (sec % 60) + 's';
+        label.textContent = this._sessionCost > 0 ? time + ' · $' + this._sessionCost.toFixed(4) : time;
       }
     }, 1000);
 
@@ -1040,6 +1042,10 @@ class AgentGUIClient {
       return;
     }
 
+    if (block.type === 'result' && block.total_cost_usd) {
+      this._sessionCost = (this._sessionCost || 0) + block.total_cost_usd;
+    }
+
     const el = this.renderer.renderBlock(block, data, blocksEl);
     if (el) {
       blocksEl.appendChild(el);