@yeaft/webchat-agent 0.0.3 → 0.0.5

package/connection.js

@@ -1,4 +1,5 @@
 import WebSocket from 'ws';
+import { execSync } from 'child_process';
 import ctx from './context.js';
 import { encrypt, decrypt, isEncrypted, decodeKey } from './encryption.js';
 import { handleTerminalCreate, handleTerminalInput, handleTerminalResize, handleTerminalClose } from './terminal.js';
@@ -264,6 +265,48 @@ async function handleMessage(msg) {
         process.exit(1);
       }, 500);
       break;
+
+    case 'upgrade_agent':
+      console.log('[Agent] Upgrade requested, checking for updates...');
+      try {
+        const pkgName = ctx.pkgName || '@yeaft/webchat-agent';
+        // Check latest version before installing
+        const latestVersion = execSync(`npm view ${pkgName} version`, { stdio: 'pipe' }).toString().trim();
+        if (latestVersion === ctx.agentVersion) {
+          console.log(`[Agent] Already at latest version (${ctx.agentVersion}), skipping upgrade.`);
+          sendToServer({ type: 'upgrade_agent_ack', success: true, alreadyLatest: true, version: ctx.agentVersion });
+          break;
+        }
+        console.log(`[Agent] Upgrading from ${ctx.agentVersion} to ${latestVersion}...`);
+        execSync(`npm install -g ${pkgName}@latest`, { stdio: 'pipe' });
+        console.log('[Agent] Upgrade successful, restarting...');
+        sendToServer({ type: 'upgrade_agent_ack', success: true, version: latestVersion });
+        // Restart after upgrade (same as restart_agent)
+        setTimeout(() => {
+          for (const [, term] of ctx.terminals) {
+            if (term.pty) { try { term.pty.kill(); } catch {} }
+            if (term.timer) clearTimeout(term.timer);
+          }
+          ctx.terminals.clear();
+          for (const [, state] of ctx.conversations) {
+            if (state.abortController) state.abortController.abort();
+            if (state.inputStream) state.inputStream.done();
+          }
+          ctx.conversations.clear();
+          stopAgentHeartbeat();
+          if (ctx.ws) {
+            ctx.ws.removeAllListeners('close');
+            ctx.ws.close();
+          }
+          clearTimeout(ctx.reconnectTimer);
+          console.log('[Agent] Cleanup done, exiting for auto-restart...');
+          process.exit(1);
+        }, 500);
+      } catch (e) {
+        console.error('[Agent] Upgrade failed:', e.message);
+        sendToServer({ type: 'upgrade_agent_ack', success: false, error: e.message });
+      }
+      break;
   }
 }
 

package/index.js

@@ -16,6 +16,7 @@ const __dirname = dirname(fileURLToPath(import.meta.url));
 // Load package version
 const pkg = JSON.parse(readFileSync(join(__dirname, 'package.json'), 'utf-8'));
 ctx.agentVersion = pkg.version;
+ctx.pkgName = pkg.name;
 
 // 配置文件路径（向后兼容：先查当前目录 .claude-agent.json）
 const LOCAL_CONFIG_FILE = join(process.cwd(), '.claude-agent.json');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yeaft/webchat-agent",
-  "version": "0.0.3",
+  "version": "0.0.5",
   "description": "Remote agent for Yeaft WebChat — connects worker machines to the central server",
   "main": "index.js",
   "type": "module",