claw-subagent-service 0.0.37 → 0.0.39

package/service/worker.js

@@ -3,6 +3,7 @@ const fs = require('fs');
 const path = require('path');
 const os = require('os');
 const { execSync } = require('child_process');
+const axios = require('axios');
 const { createLogger } = require('./logger');
 const { RongCloudClient, MessageHandler, ensurePluginsAllow } = require('./rongcloud');
 const { RongyunMessageHandler } = require('./modules/rongyun-message-handler');
@@ -14,6 +15,7 @@ const { startOpencodeService, stopOpencodeService } = require('./modules/opencod
 
 const log = createLogger('worker');
 const PORT = process.env.SILENT_SERVICE_PORT ? parseInt(process.env.SILENT_SERVICE_PORT, 10) : 28765;
+const HOST = process.env.SILENT_SERVICE_HOST || '127.0.0.1';
 
 
 // 捕获所有异常，强制打印到控制台（绕过 logger）
@@ -53,9 +55,20 @@ function findPidOnPort(port) {
         }
       }
     } else {
-      const out = execSync(`lsof -i :${port} -t 2>/dev/null`, { encoding: 'utf8', timeout: 5000 });
-      const pid = parseInt(out.trim(), 10);
-      if (!isNaN(pid)) return pid;
+      // 优先尝试 lsof，再兜底 ss / fuser / netstat（适配精简 Docker 镜像）
+      const commands = [
+        `lsof -i :${port} -t 2>/dev/null`,
+        `fuser ${port}/tcp 2>/dev/null`,
+        `ss -tlnp 2>/dev/null | grep ":${port}" | sed -n 's/.*pid=\\([0-9]*\\).*/\\1/p'`,
+        `netstat -tlnp 2>/dev/null | grep ":${port}" | sed -n 's/.*\\/\\([0-9]*\\).*/\\1/p'`,
+      ];
+      for (const cmd of commands) {
+        try {
+          const out = execSync(cmd, { encoding: 'utf8', timeout: 5000 }).trim();
+          const pid = parseInt(out.split('\n')[0], 10);
+          if (!isNaN(pid) && pid > 0) return pid;
+        } catch { continue; }
+      }
     }
   } catch { /* port is free */ }
   return null;
@@ -198,6 +211,53 @@ rongcloudConfig = loadRongCloudConfig();
 let rongcloudClient = null;
 let messageHandler = null;
 
+/**
+ * 向服务端刷新融云 token
+ */
+async function refreshRongCloudToken() {
+  const nodeId = rongcloudConfig?.accountId;
+  if (!nodeId) {
+    log.error('[WORKER] 无法刷新 token: 缺少 nodeId');
+    return false;
+  }
+
+  const serverUrl = process.env.DM_SERVER_URL || 'https://newsradar.dreamdt.cn/im';
+  try {
+    log.info(`[WORKER] 正在向服务端刷新 token, nodeId=${nodeId}`);
+    const resp = await axios.get(`${serverUrl}/api/claw/token/${nodeId}`, { timeout: 15000 });
+    if (resp.data?.code === 200) {
+      const newToken = resp.data.data?.token || resp.data.token || '';
+      if (!newToken) {
+        log.error('[WORKER] 服务端返回了空 token');
+        return false;
+      }
+      log.info('[WORKER] token 刷新成功');
+
+      // 更新内存配置
+      rongcloudConfig.token = newToken;
+
+      // 保存到 config.json
+      try {
+        if (fs.existsSync(clawBridgeConfigPath)) {
+          const clawConfig = JSON.parse(fs.readFileSync(clawBridgeConfigPath, 'utf8'));
+          clawConfig.token = newToken;
+          clawConfig.expiresAt = Date.now() + 7 * 24 * 60 * 60 * 1000; // 7天
+          fs.writeFileSync(clawBridgeConfigPath, JSON.stringify(clawConfig, null, 2));
+          log.info('[WORKER] 新 token 已保存到 config.json');
+        }
+      } catch (err) {
+        log.error(`[WORKER] 保存新 token 失败: ${err.message}`);
+      }
+      return true;
+    }
+    log.error(`[WORKER] 刷新 token 失败: ${resp.data?.message || '未知错误'}`);
+    return false;
+  } catch (err) {
+    log.error(`[WORKER] 刷新 token 异常: ${err.message}`);
+    return false;
+  }
+}
+
 async function initRongCloud() {
   if (!rongcloudConfig) return;
 
@@ -233,22 +293,22 @@ async function initRongCloud() {
 
   // 包装 MessageHandler.handleMessage 以处理结构化消息
   const originalHandleMessage = messageHandler.handleMessage.bind(messageHandler);
-  
+
   messageHandler.handleMessage = async (msg) => {
     // 检查是否是结构化消息
     if (msg.content && typeof msg.content === 'string') {
       try {
         const parsed = JSON.parse(msg.content);
-        
+
         if (parsed.msg_type) {
           // 这是结构化消息，使用 RongyunMessageHandler 处理
           log.info(`[WORKER] 收到结构化消息: type=${parsed.msg_type}, from=${parsed.source_im_id || msg.senderUserId}`);
-          
+
           // 忽略自己发送的消息
           if (parsed.source_im_id === rongcloudConfig.accountId) {
             return;
           }
-          
+
           // Timestamp 校验（5分钟有效期）
           const msgTimestamp = parsed.timestamp;
           if (msgTimestamp) {
@@ -262,7 +322,7 @@ async function initRongCloud() {
               return;
             }
           }
-          
+
           // 解析 content 字段（它本身可能是 JSON 字符串）
           let innerContent = parsed.content;
           if (typeof innerContent === 'string') {
@@ -272,7 +332,7 @@ async function initRongCloud() {
               // 保持字符串
             }
           }
-          
+
           // 构建消息数据
           // 注意：后端发送的 command 消息中，command/command_id/request_id 在 content 字段内
           // 保留原始 content（用户消息内容），同时展开其他字段
@@ -284,7 +344,7 @@ async function initRongCloud() {
             targetId: msg.targetId,
             conversationType: msg.conversationType,
           };
-          
+
           // 使用 RongyunMessageHandler 处理
           try {
             await rongyunMessageHandler.handle(messageData);
@@ -297,18 +357,30 @@ async function initRongCloud() {
         // 不是 JSON，是普通消息，继续传给原始 handler
       }
     }
-    
+
     // 调用原始的 handleMessage（处理普通消息）
     return originalHandleMessage(msg);
   };
-  
+
   // 添加调试日志：确认替换后的方法
   log.info('[WORKER-DEBUG] 替换后 messageHandler.handleMessage 类型: ' + typeof messageHandler.handleMessage);
 
-  const connected = await rongcloudClient.connect(messageHandler);
+  let connected = await rongcloudClient.connect(messageHandler);
+
+  // 连接失败时尝试刷新 token 并重试一次
+  if (!connected) {
+    log.warn('[WORKER] 首次融云连接失败，尝试刷新 token...');
+    const refreshed = await refreshRongCloudToken();
+    if (refreshed) {
+      // 使用新 token 重新创建客户端并连接
+      rongcloudClient = new RongCloudClient(rongcloudConfig, log);
+      connected = await rongcloudClient.connect(messageHandler);
+    }
+  }
+
   if (connected) {
     log.info('[WORKER] 融云连接成功');
-    
+
     // 发送 CLIENT_CONNECTED
     try {
       await messageSender.sendClientConnected();
@@ -316,21 +388,21 @@ async function initRongCloud() {
     } catch (err) {
       log.error(`[WORKER] 发送 CLIENT_CONNECTED 失败: ${err.message}`);
     }
-    
+
     // 启动心跳管理器
     const heartbeatManager = new HeartbeatManager(rongcloudClient, rongcloudConfig, log);
     heartbeatManager.start(getMacAddress, getOpenClawStatus);
-    
+
     // 启动仪表盘上报
     const dashboardReporter = new DashboardReporter(rongcloudClient, rongcloudConfig, log);
     dashboardReporter.start(getMacAddress);
-    
+
     // 保存引用以便关闭时停止
     global.heartbeatManager = heartbeatManager;
     global.dashboardReporter = dashboardReporter;
-    
+
   } else {
-    log.error('[WORKER] 融云连接失败');
+    log.error('[WORKER] 融云连接失败，token 刷新后仍无法连接');
   }
 }
 
@@ -415,15 +487,15 @@ server.on('error', (err) => {
     setTimeout(() => {
       log.info(`[WORKER] 重新尝试监听端口 ${PORT}...`);
       server.close(() => {});
-      server.listen(PORT, '127.0.0.1');
+      server.listen(PORT, HOST);
     }, 2000);
     return;
   }
   log.error(`[WORKER] HTTP 服务错误: ${err.message}`);
 });
 
-server.listen(PORT, '127.0.0.1', () => {
-  log.info(`[WORKER] HTTP 服务已启动: http://127.0.0.1:${PORT}/health`);
+server.listen(PORT, HOST, () => {
+  log.info(`[WORKER] HTTP 服务已启动: http://${HOST}:${PORT}/health`);
 });
 
 process.on('message', (msg) => {