npm - agentgui - Versions diffs - 1.0.763 → 1.0.764 - Mend

agentgui 1.0.763 → 1.0.764

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (3) hide show

package/package.json +1 -1
package/server.js +24 -26
package/static/js/websocket-manager.js +3 -1

package/package.json CHANGED Viewed

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

package/server.js CHANGED Viewed

@@ -527,6 +527,10 @@ const server = http.createServer(async (req, res) => {
     }
     if (pathOnly === '/api/health' && req.method === 'GET') {
+      let dbStatus = { ok: true };
+      try { queries._db.prepare('SELECT 1').get(); } catch (e) { dbStatus = { ok: false, error: e.message }; }
+      const queueSizes = {};
+      for (const [k, v] of messageQueues) queueSizes[k] = v.length;
       sendJSON(req, res, 200, {
         status: 'ok',
         version: PKG_VERSION,
@@ -535,7 +539,9 @@ const server = http.createServer(async (req, res) => {
         activeExecutions: activeExecutions.size,
         wsClients: wss.clients.size,
         memory: process.memoryUsage(),
-        acp: getACPStatus()
+        acp: getACPStatus(),
+        db: dbStatus,
+        queueSizes
       });
       return;
     }
@@ -1189,7 +1195,7 @@ const server = http.createServer(async (req, res) => {
     }
     if (pathOnly === '/api/agents' && req.method === 'GET') {
-      console.log(`[API /api/agents] Returning ${discoveredAgents.length} agents:`, discoveredAgents.map(a => a.id).join(', '));
+      debugLog(`[API /api/agents] Returning ${discoveredAgents.length} agents`);
             sendJSON(req, res, 200, { agents: discoveredAgents });
       return;
     }
@@ -1199,21 +1205,6 @@ const server = http.createServer(async (req, res) => {
       return;
     }
-    if (pathOnly === '/api/health' && req.method === 'GET') {
-      let dbStatus = { ok: true };
-      try { queries._db.prepare('SELECT 1').get(); } catch (e) { dbStatus = { ok: false, error: e.message }; }
-      const queueSizes = {};
-      for (const [k, v] of messageQueues) queueSizes[k] = v.length;
-      sendJSON(req, res, 200, {
-        uptime: process.uptime(),
-        db: dbStatus,
-        activeExecutionCount: activeExecutions.size,
-        queueSizes,
-        wsClientCount: syncClients.size,
-        memory: process.memoryUsage()
-      });
-      return;
-    }
     if (pathOnly === '/api/debug' && req.method === 'GET') {
       const execSnap = {};
@@ -2093,7 +2084,8 @@ function serveFile(filePath, res, req) {
     fs.readFile(filePath, (err2, data) => {
       if (err2) { res.writeHead(500); res.end('Server error'); return; }
       let content = data.toString();
-      const baseTag = `<script>window.__BASE_URL='${BASE_URL}';window.__SERVER_VERSION='${PKG_VERSION}';</script>`;
+      const wsToken = process.env.PASSWORD ? `window.__WS_TOKEN='${process.env.PASSWORD.replace(/'/g, "\\'")}';` : '';
+      const baseTag = `<script>window.__BASE_URL='${BASE_URL}';window.__SERVER_VERSION='${PKG_VERSION}';${wsToken}</script>`;
       content = content.replace('<head>', `<head>\n  <base href="${BASE_URL}/">\n  ` + baseTag);
       content = content.replace(/(href|src)="vendor\//g, `$1="${BASE_URL}/vendor/`);
       content = content.replace(/(src)="\/gm\/js\//g, `$1="${BASE_URL}/js/`);
@@ -2821,12 +2813,18 @@ const subscriptionIndex = new Map();
 const pm2Subscribers = new Set();
 wss.on('connection', (ws, req) => {
-  // req.url in WebSocket is just the path (e.g., '/gm/sync'), not a full URL
-  const wsPath = req.url.startsWith(BASE_URL) ? req.url.slice(BASE_URL.length) : req.url;
-  if (wsPath === '/hot-reload') {
+  const _pwd = process.env.PASSWORD;
+  if (_pwd) {
+    const url = new URL(req.url, 'http://localhost');
+    const token = url.searchParams.get('token');
+    if (token !== _pwd) { ws.close(4001, 'Unauthorized'); return; }
+  }
+  const wsPath = req.url.split('?')[0];
+  const wsRoute = wsPath.startsWith(BASE_URL) ? wsPath.slice(BASE_URL.length) : wsPath;
+  if (wsRoute === '/hot-reload') {
     hotReloadClients.push(ws);
     ws.on('close', () => { const i = hotReloadClients.indexOf(ws); if (i > -1) hotReloadClients.splice(i, 1); });
-  } else if (wsPath === '/sync') {
+  } else if (wsRoute === '/sync') {
     syncClients.add(ws);
     ws.isAlive = true;
     ws.subscriptions = new Set();
@@ -2857,7 +2855,7 @@ wss.on('connection', (ws, req) => {
        if (ws.pm2Subscribed) {
          pm2Subscribers.delete(ws);
        }
-       console.log(`[WebSocket] Client ${ws.clientId} disconnected`);
+       debugLog(`[WebSocket] Client ${ws.clientId} disconnected`);
      });
   }
 });
@@ -2935,13 +2933,13 @@ registerMsgHandlers(wsRouter, {
 registerQueueHandlers(wsRouter, { queries, messageQueues, broadcastSync });
-console.log('[INIT] About to call registerSessionHandlers, discoveredAgents.length:', discoveredAgents.length);
+debugLog('[INIT] registerSessionHandlers, agents: ' + discoveredAgents.length);
 registerSessionHandlers(wsRouter, {
   db: queries, discoveredAgents, modelCache,
   getAgentDescriptor, activeScripts, broadcastSync,
   startGeminiOAuth: (req) => startGeminiOAuth(req, { PORT, BASE_URL, rootDir }), geminiOAuthState: getGeminiOAuthState
 });
-console.log('[INIT] registerSessionHandlers completed');
+debugLog('[INIT] registerSessionHandlers completed');
 registerRunHandlers(wsRouter, {
   queries, discoveredAgents, activeExecutions, activeProcessesByRunId,
@@ -3014,7 +3012,7 @@ wsRouter.onLegacy((data, ws) => {
     if (data.conversationId && checkpointManager.hasPendingCheckpoint(data.conversationId)) {
       const checkpoint = checkpointManager.getPendingCheckpoint(data.conversationId);
       if (checkpoint) {
-        console.log(`[checkpoint] Injecting ${checkpoint.events.length} events to client for ${data.conversationId}`);
+        debugLog(`[checkpoint] Injecting ${checkpoint.events.length} events to client for ${data.conversationId}`);
         const latestSession = queries.getLatestSession(data.conversationId);
         if (latestSession) {

package/static/js/websocket-manager.js CHANGED Viewed

@@ -113,7 +113,9 @@ class WebSocketManager {
   getWebSocketURL() {
     const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
     const baseURL = window.__BASE_URL || '/gm';
-    return `${protocol}//${window.location.host}${baseURL}/sync`;
+    let url = `${protocol}//${window.location.host}${baseURL}/sync`;
+    if (window.__WS_TOKEN) url += `?token=${encodeURIComponent(window.__WS_TOKEN)}`;
+    return url;
   }
   async connect() {