agentgui 1.0.720 → 1.0.721

package/lib/jsonl-watcher.js

@@ -12,7 +12,7 @@ export class JsonlWatcher {
     this._owned = ownedSessionIds;
     this._tails = new Map();
     this._convMap = new Map();
-    this._emitted = new Map();
+    this._frags = new Map();
     this._timers = new Map();
     this._seqs = new Map();
     this._streaming = new Set();
@@ -42,7 +42,7 @@ export class JsonlWatcher {
       this._convMap.delete(sid);
       this._seqs.delete(sid);
       this._streaming.delete(sid);
-      for (const key of [...this._emitted.keys()]) if (key.startsWith(`${sid}:`)) this._emitted.delete(key);
+      for (const key of [...this._frags.keys()]) if (key.startsWith(`${sid}:`)) this._frags.delete(key);
       for (const [fp, s] of this._tails.entries()) {
         if (!fp.includes(sid)) continue;
         if (s.fd !== null) try { fs.closeSync(s.fd); } catch (_) {}
@@ -56,7 +56,7 @@ export class JsonlWatcher {
   removeAllConversations() {
     for (const s of this._tails.values()) if (s.fd !== null) try { fs.closeSync(s.fd); } catch (_) {}
     for (const t of this._timers.values()) clearTimeout(t);
-    this._tails.clear(); this._convMap.clear(); this._emitted.clear();
+    this._tails.clear(); this._convMap.clear(); this._frags.clear();
     this._timers.clear(); this._seqs.clear(); this._streaming.clear();
   }
 
@@ -114,8 +114,7 @@ export class JsonlWatcher {
     if (this._convMap.has(sid)) return this._convMap.get(sid);
     const found = this._q.getConversations().find(c => c.claudeSessionId === sid);
     if (found) { this._convMap.set(sid, found.id); return found.id; }
-    if (e.type === 'queue-operation' || e.type === 'last-prompt') return null;
-    if (e.type === 'user' && e.isMeta) return null;
+    if (e.type !== 'user' || e.isMeta) return null;
     const cwd = e.cwd || process.cwd();
     const branch = e.gitBranch || '';
     const base = path.basename(cwd);
@@ -133,13 +132,6 @@ export class JsonlWatcher {
     this._bc({ type: 'streaming_progress', sessionId: sid, conversationId: cid, block, blockRole: role, seq: this._seq(sid), timestamp: Date.now(), ...extra });
   }
 
-  _emitBlocks(cid, sid, blocks, role) {
-    for (const b of blocks) {
-      if (!b || !b.type) continue;
-      this._emit(cid, sid, b, role);
-    }
-  }
-
   _startStreaming(cid, sid) {
     if (this._streaming.has(sid)) return;
     this._streaming.add(sid);
@@ -153,17 +145,11 @@ export class JsonlWatcher {
   }
 
   _route(cid, sid, e) {
-    if (e.type === 'queue-operation' || e.type === 'last-prompt' || (e.type === 'user' && e.isMeta)) return;
-
-    if (e.isApiErrorMessage && e.error === 'rate_limit') {
-      this._bc({ type: 'streaming_error', sessionId: sid, conversationId: cid, error: 'Rate limit hit', recoverable: true, timestamp: Date.now() });
-      return;
-    }
+    if (e.type === 'queue-operation' || (e.type === 'user' && e.isMeta)) return;
 
     if (e.type === 'system') {
       if (e.subtype === 'init') { this._startStreaming(cid, sid); return; }
       if (e.subtype === 'turn_duration' || e.subtype === 'stop_hook_summary') { this._endStreaming(cid, sid); return; }
-      this._startStreaming(cid, sid);
       this._emit(cid, sid, { type: 'system', subtype: e.subtype, model: e.model, cwd: e.cwd, tools: e.tools, preTokens: e.compactMetadata?.preTokens }, 'system');
       return;
     }
@@ -171,14 +157,11 @@ export class JsonlWatcher {
     if (e.type === 'assistant' && e.message?.content) {
       this._startStreaming(cid, sid);
       const key = `${sid}:${e.message.id}`;
-      const prev = this._emitted.get(key) || 0;
-      const content = e.message.content;
-      const newBlocks = content.slice(prev);
-      if (newBlocks.length > 0) {
-        this._emitted.set(key, content.length);
-        this._emitBlocks(cid, sid, newBlocks, 'assistant');
-      }
-      if (e.message.stop_reason) this._emitted.delete(key);
+      const prevCount = this._frags.get(key)?.message?.content?.length || 0;
+      const newBlocks = e.message.content.slice(prevCount);
+      if (e.message.stop_reason === null || e.message.stop_reason === undefined) this._frags.set(key, e);
+      else this._frags.delete(key);
+      for (const b of newBlocks) this._emit(cid, sid, b, 'assistant');
       return;
     }
 
@@ -192,25 +175,14 @@ export class JsonlWatcher {
 
     if (e.type === 'progress') {
       this._startStreaming(cid, sid);
-      const d = e.data || {};
-      if (d.type === 'agent_progress') {
-        const inner = d.message?.message || d.message;
-        const content = inner?.content;
-        if (Array.isArray(content)) {
-          const role = inner.role === 'user' ? 'tool_result' : 'assistant';
-          this._emitBlocks(cid, sid, content, role);
-        }
-        return;
-      }
-      if (d.type === 'hook_progress') {
-        this._emit(cid, sid, { type: 'hook_progress', hookEvent: d.hookEvent, hookName: d.hookName }, 'system');
-        return;
-      }
-      if (d.type === 'mcp_progress') {
-        this._emit(cid, sid, { type: 'mcp_progress', status: d.status, serverName: d.serverName, toolName: d.toolName }, 'system');
-        return;
-      }
-      this._emit(cid, sid, { type: d.type || e.subtype || 'progress', content: e.content || d }, 'progress');
+      const dataType = e.data?.type || e.subtype || 'progress';
+      const content = dataType === 'agent_progress' ? e.data?.message : (e.data || e.content);
+      this._emit(cid, sid, { type: dataType, agentId: e.agentId, content }, 'progress');
+      return;
+    }
+
+    if (e.isApiErrorMessage && e.error === 'rate_limit') {
+      this._bc({ type: 'streaming_error', sessionId: sid, conversationId: cid, error: 'Rate limit hit', recoverable: true, timestamp: Date.now() });
       return;
     }
 

package/lib/plugin-loader.js

@@ -1,188 +1,162 @@
-// Plugin loader - manages registry, dependencies, hot reload, error isolation
-
-import fs from 'fs';
-import path from 'path';
-import { EventEmitter } from 'events';
-
-class PluginLoader extends EventEmitter {
-  constructor(pluginDir) {
-    super();
-    this.pluginDir = pluginDir;
-    this.registry = new Map(); // name => plugin module
-    this.instances = new Map(); // name => initialized plugin state
-    this.states = new Map(); // name => { routes, wsHandlers, api, ... }
-    this.watchers = new Map(); // name => file watcher
-    this.errorCounts = new Map(); // name => { count, firstTime }
-  }
-
-  // Load plugin module from disk
-  async loadPlugin(name) {
-    const filePath = path.join(this.pluginDir, `${name}.js`);
-    if (!fs.existsSync(filePath)) {
-      throw new Error(`Plugin file not found: ${filePath}`);
-    }
-    // Clear module cache for hot reload (ES modules use import cache differently)
-    const fileUrl = `file://${filePath}?v=${Date.now()}`;
-    try {
-      const plugin = await import(fileUrl);
-      this.registry.set(name, plugin.default || plugin);
-      return plugin.default || plugin;
-    } catch (error) {
-      console.error(`Failed to load plugin ${name}:`, error.message);
-      throw error;
-    }
-  }
-
-  // Initialize plugin and all dependencies
-  async initializePlugin(name, config) {
-    const plugin = this.registry.get(name);
-    if (!plugin) {
-      throw new Error(`Plugin ${name} not found in registry`);
-    }
-
-    // Check if already initialized
-    if (this.instances.has(name)) {
-      return this.instances.get(name);
-    }
-
-    // Initialize dependencies first
-    for (const depName of (plugin.dependencies || [])) {
-      if (!this.instances.has(depName)) {
-        await this.initializePlugin(depName, config);
-      }
-    }
-
-    // Initialize this plugin
-    try {
-      const result = await plugin.init(config, this.instances);
-      this.instances.set(name, result);
-      return result;
-    } catch (error) {
-      console.error(`[PluginLoader] Error initializing ${name}:`, error.message);
-      throw error;
-    }
-  }
-
-  // Get initialized plugin result
-  get(name) {
-    return this.instances.get(name);
-  }
-
-  // Hot reload a plugin
-  async reloadPlugin(name) {
-    const plugin = this.registry.get(name);
-    if (!plugin) {
-      console.warn(`[PluginLoader] Cannot reload ${name}: not found`);
-      return;
-    }
-
-    const state = this.instances.get(name);
-    if (!state) {
-      console.warn(`[PluginLoader] Cannot reload ${name}: not initialized`);
-      return;
-    }
-
-    try {
-      // Stop old instance
-      if (state.stop) await state.stop();
-
-      // Reload plugin module
-      this.loadPlugin(name);
-      const reloadedPlugin = this.registry.get(name);
-
-      // Reinitialize with preserved state
-      const newState = await reloadedPlugin.reload(state);
-      this.instances.set(name, newState);
-      this.emit('reload', { name, success: true });
-      console.log(`[PluginLoader] Reloaded plugin: ${name}`);
-    } catch (error) {
-      console.error(`[PluginLoader] Error reloading ${name}:`, error.message);
-      this.emit('reload', { name, success: false, error: error.message });
-    }
-  }
-
-  // Watch a plugin file for changes
-  watchPlugin(name, callback) {
-    const filePath = path.join(this.pluginDir, `${name}.js`);
-    if (this.watchers.has(name)) {
-      return; // Already watching
-    }
-
-    const watcher = fs.watch(filePath, async (eventType) => {
-      if (eventType === 'change') {
-        setTimeout(() => callback(name), 100); // Debounce
-      }
-    });
-
-    this.watchers.set(name, watcher);
-  }
-
-  // Stop watching a plugin
-  unwatchPlugin(name) {
-    const watcher = this.watchers.get(name);
-    if (watcher) {
-      watcher.close();
-      this.watchers.delete(name);
-    }
-  }
-
-  // Load all plugins from directory
-  async loadAllPlugins(config) {
-    if (!fs.existsSync(this.pluginDir)) {
-      fs.mkdirSync(this.pluginDir, { recursive: true });
-      return;
-    }
-
-    const files = fs.readdirSync(this.pluginDir).filter(f => f.endsWith('.js'));
-    for (const file of files) {
-      const name = file.replace('.js', '');
-      try {
-        await this.loadPlugin(name);
-      } catch (error) {
-        console.error(`[PluginLoader] Failed to load ${name}:`, error.message);
-      }
-    }
-
-    // Initialize in dependency order
-    const sorted = this.topologicalSort();
-    for (const name of sorted) {
-      try {
-        await this.initializePlugin(name, config);
-      } catch (error) {
-        console.error(`[PluginLoader] Failed to initialize ${name}:`, error.message);
-      }
-    }
-  }
-
-  // Topological sort by dependencies
-  topologicalSort() {
-    const visited = new Set(), result = [];
-    const visit = (name) => {
-      if (visited.has(name)) return;
-      visited.add(name);
-      for (const dep of (this.registry.get(name)?.dependencies || [])) { if (this.registry.has(dep)) visit(dep); }
-      result.push(name);
-    };
-    for (const name of this.registry.keys()) visit(name);
-    return result;
-  }
-
-  // Graceful shutdown
-  async shutdown() {
-    const sorted = this.topologicalSort().reverse();
-    for (const name of sorted) {
-      const state = this.instances.get(name);
-      if (state && state.stop) {
-        try {
-          await state.stop();
-        } catch (error) {
-          console.error(`[PluginLoader] Error stopping ${name}:`, error.message);
-        }
-      }
-      this.unwatchPlugin(name);
-    }
-    this.instances.clear();
-    this.registry.clear();
-  }
-}
-export default PluginLoader;
+import fs from 'fs';
+import path from 'path';
+import { EventEmitter } from 'events';
+
+class PluginLoader extends EventEmitter {
+  constructor(pluginDir) {
+    super();
+    this.pluginDir = pluginDir;
+    this.registry = new Map();
+    this.instances = new Map();
+    this.states = new Map();
+    this.watchers = new Map();
+    this.errorCounts = new Map();
+    this.fileMap = new Map();
+  }
+
+  async loadPlugin(fileName) {
+    const filePath = path.join(this.pluginDir, `${fileName}.js`);
+    if (!fs.existsSync(filePath)) {
+      throw new Error(`Plugin file not found: ${filePath}`);
+    }
+    const fileUrl = `file://${filePath}?v=${Date.now()}`;
+    try {
+      const plugin = await import(fileUrl);
+      const mod = plugin.default || plugin;
+      const regName = mod.name || fileName;
+      this.registry.set(regName, mod);
+      this.fileMap.set(regName, fileName);
+      return mod;
+    } catch (error) {
+      console.error(`Failed to load plugin ${fileName}:`, error.message);
+      throw error;
+    }
+  }
+
+  async initializePlugin(name, config) {
+    const plugin = this.registry.get(name);
+    if (!plugin) {
+      throw new Error(`Plugin ${name} not found in registry`);
+    }
+    if (this.instances.has(name)) {
+      return this.instances.get(name);
+    }
+    for (const depName of (plugin.dependencies || [])) {
+      if (!this.instances.has(depName)) {
+        await this.initializePlugin(depName, config);
+      }
+    }
+    try {
+      const result = await plugin.init(config, this.instances);
+      this.instances.set(name, result);
+      return result;
+    } catch (error) {
+      console.error(`[PluginLoader] Error initializing ${name}:`, error.message);
+      throw error;
+    }
+  }
+
+  get(name) {
+    return this.instances.get(name);
+  }
+
+  async reloadPlugin(name) {
+    const plugin = this.registry.get(name);
+    if (!plugin) {
+      console.warn(`[PluginLoader] Cannot reload ${name}: not found`);
+      return;
+    }
+    const state = this.instances.get(name);
+    if (!state) {
+      console.warn(`[PluginLoader] Cannot reload ${name}: not initialized`);
+      return;
+    }
+    try {
+      if (state.stop) await state.stop();
+      const fileName = this.fileMap.get(name) || name;
+      await this.loadPlugin(fileName);
+      const reloadedPlugin = this.registry.get(name);
+      const newState = await reloadedPlugin.reload(state);
+      this.instances.set(name, newState);
+      this.emit('reload', { name, success: true });
+      console.log(`[PluginLoader] Reloaded plugin: ${name}`);
+    } catch (error) {
+      console.error(`[PluginLoader] Error reloading ${name}:`, error.message);
+      this.emit('reload', { name, success: false, error: error.message });
+    }
+  }
+
+  watchPlugin(name, callback) {
+    const fileName = this.fileMap.get(name) || name;
+    const filePath = path.join(this.pluginDir, `${fileName}.js`);
+    if (this.watchers.has(name)) return;
+    const watcher = fs.watch(filePath, async (eventType) => {
+      if (eventType === 'change') {
+        setTimeout(() => callback(name), 100);
+      }
+    });
+    this.watchers.set(name, watcher);
+  }
+
+  unwatchPlugin(name) {
+    const watcher = this.watchers.get(name);
+    if (watcher) {
+      watcher.close();
+      this.watchers.delete(name);
+    }
+  }
+
+  async loadAllPlugins(config) {
+    if (!fs.existsSync(this.pluginDir)) {
+      fs.mkdirSync(this.pluginDir, { recursive: true });
+      return;
+    }
+    const files = fs.readdirSync(this.pluginDir).filter(f => f.endsWith('.js'));
+    for (const file of files) {
+      const fileName = file.replace('.js', '');
+      try {
+        await this.loadPlugin(fileName);
+      } catch (error) {
+        console.error(`[PluginLoader] Failed to load ${fileName}:`, error.message);
+      }
+    }
+    const sorted = this.topologicalSort();
+    for (const name of sorted) {
+      try {
+        await this.initializePlugin(name, config);
+      } catch (error) {
+        console.error(`[PluginLoader] Failed to initialize ${name}:`, error.message);
+      }
+    }
+  }
+
+  topologicalSort() {
+    const visited = new Set(), result = [];
+    const visit = (name) => {
+      if (visited.has(name)) return;
+      visited.add(name);
+      for (const dep of (this.registry.get(name)?.dependencies || [])) { if (this.registry.has(dep)) visit(dep); }
+      result.push(name);
+    };
+    for (const name of this.registry.keys()) visit(name);
+    return result;
+  }
+
+  async shutdown() {
+    const sorted = this.topologicalSort().reverse();
+    for (const name of sorted) {
+      const state = this.instances.get(name);
+      if (state && state.stop) {
+        try {
+          await state.stop();
+        } catch (error) {
+          console.error(`[PluginLoader] Error stopping ${name}:`, error.message);
+        }
+      }
+      this.unwatchPlugin(name);
+    }
+    this.instances.clear();
+    this.registry.clear();
+  }
+}
+export default PluginLoader;

package/lib/plugins/agents-plugin.js

@@ -49,10 +49,9 @@ export default {
               if (!agent) return res.status(404).json({ error: 'Agent not found' });
 
               const session = stream.api.createSession(id);
-              // Use runClaudeWithStreaming instead
               activeExecutions.set(id, { sessionId: session.id });
 
-              res.json({ sessionId: session.id, pid: proc.pid });
+              res.json({ sessionId: session.id });
             } catch (e) {
               res.status(500).json({ error: e.message });
             }

package/lib/plugins/database-plugin.js

@@ -1,6 +1,6 @@
 // Database plugin - SQLite init, schema, checkpoint recovery
 
-import * as dbModule from '../database.js';
+import * as dbModule from '../../database.js';
 
 export default {
   name: 'database',

package/lib/plugins/stream-plugin.js

@@ -8,7 +8,8 @@ export default {
   dependencies: ['database'],
 
   async init(config, plugins) {
-    const db = plugins.get('database');
+    const dbPlugin = plugins.get('database');
+    const db = dbPlugin?.api || {};
     const activeSessions = new Map();
     const pendingMessages = new Map();
     const rateLimitState = new Map();

package/lib/tool-provisioner.js

@@ -38,7 +38,7 @@ export async function autoProvision(TOOLS, statusCache, install, broadcast) {
           broadcast({ type: 'tool_update_failed', toolId: tool.id, data: result });
         }
       } else {
-        log(`${tool.id} v${status.installedVersion} up-to-date`);
+        log(`${tool.id} v${status.installedVersion || 'unknown'} up-to-date`);
         broadcast({ type: 'tool_status_update', toolId: tool.id, data: { installed: true, isUpToDate: true, installedVersion: status.installedVersion, status: 'installed' } });
       }
     } catch (err) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentgui",
-  "version": "1.0.720",
+  "version": "1.0.721",
   "description": "Multi-agent ACP client with real-time communication",
   "type": "module",
   "main": "server.js",

package/server.js

@@ -4479,7 +4479,7 @@ const BROADCAST_TYPES = new Set([
   'streaming_start', 'streaming_progress', 'streaming_complete', 'streaming_error',
   'tool_install_started', 'tool_install_progress', 'tool_install_complete', 'tool_install_failed',
   'tool_update_progress', 'tool_update_complete', 'tool_update_failed',
-  'tool_status_update',
+  'tool_status_update', 'tool_update_available',
   'tools_update_started', 'tools_update_complete', 'tools_refresh_complete',
   'pm2_monit_update', 'pm2_monitoring_started', 'pm2_monitoring_stopped',
   'pm2_list_response', 'pm2_start_response', 'pm2_stop_response',
@@ -5257,7 +5257,7 @@ import PluginLoader from './lib/plugin-loader.js';
 const pluginLoader = new PluginLoader(path.join(__dirname, 'lib', 'plugins'));
 async function loadPluginExtensions() {
   try {
-    await pluginLoader.loadAllPlugins({ router: app, baseUrl: BASE_URL, logger: console, env: process.env });
+    await pluginLoader.loadAllPlugins({ router: expressApp, baseUrl: BASE_URL, logger: console, env: process.env });
     const names = Array.from(pluginLoader.registry.keys());
     if (names.length > 0) {
       for (const name of names) {
@@ -5266,7 +5266,7 @@ async function loadPluginExtensions() {
         for (const route of state.routes) {
           const fullPath = BASE_URL + route.path;
           const method = (route.method || 'GET').toLowerCase();
-          if (app[method]) app[method](fullPath, route.handler);
+          if (expressApp[method]) expressApp[method](fullPath, route.handler);
         }
       }
       console.log(`[PLUGINS] Loaded extensions: ${names.join(', ')}`);