hvip-mcp-server 0.4.3 → 0.6.0

package/dist/hub-worker.js

@@ -3717,6 +3717,56 @@ var import_websocket_server = __toESM(require_websocket_server(), 1);
 
 // src/hub-worker.ts
 var import_node_child_process = require("node:child_process");
+
+// src/utils/logger.ts
+var LEVEL_EMOJI = {
+  [0 /* DEBUG */]: "\u{1F50D}",
+  [1 /* INFO */]: "\u2139\uFE0F",
+  [2 /* WARN */]: "\u26A0\uFE0F",
+  [3 /* ERROR */]: "\u274C",
+  [4 /* SILENT */]: ""
+};
+function resolveLevel() {
+  const raw = (process.env.LOG_LEVEL || "").toUpperCase();
+  if (raw === "DEBUG") return 0 /* DEBUG */;
+  if (raw === "WARN" || raw === "WARNING") return 2 /* WARN */;
+  if (raw === "ERROR") return 3 /* ERROR */;
+  if (raw === "SILENT" || raw === "OFF") return 4 /* SILENT */;
+  return 1 /* INFO */;
+}
+var currentLevel = resolveLevel();
+var LoggerImpl = class {
+  constructor(tag) {
+    this.tag = tag;
+  }
+  tag;
+  debug(msg) {
+    this.write(0 /* DEBUG */, msg);
+  }
+  info(msg) {
+    this.write(1 /* INFO */, msg);
+  }
+  warn(msg) {
+    this.write(2 /* WARN */, msg);
+  }
+  error(msg) {
+    this.write(3 /* ERROR */, msg);
+  }
+  write(level, msg) {
+    if (level < currentLevel) return;
+    const ts = (/* @__PURE__ */ new Date()).toISOString().slice(11, 23);
+    const emoji = LEVEL_EMOJI[level];
+    const tag = this.tag ? `[${this.tag}]` : "";
+    const prefix = emoji ? `${emoji} ${tag}` : tag;
+    process.stderr.write(`${ts} ${prefix} ${msg}
+`);
+  }
+};
+function logger(tag) {
+  return new LoggerImpl(tag);
+}
+
+// src/hub-worker.ts
 var argv = process.argv.slice(2);
 function flag(name) {
   const i = argv.indexOf("--" + name);
@@ -3732,8 +3782,12 @@ var AGENT_ID = `worker-${TASK_ID}-${Date.now()}`;
 var AGENT_NAME = `Worker\xB7${TASK_ID}`;
 var CLAUDE_CLI = process.env.CLAUDE_CLI || "claude";
 var WORKER_TIMEOUT_MS = parseInt(process.env.HUB_WORKER_TIMEOUT || "600000", 10);
+var log = logger(`Worker-${TASK_ID}`);
 var ws;
 var taskReceived = false;
+var heartbeatTimer;
+var taskDoneSent = false;
+var taskTitle = "";
 function connect() {
   ws = new import_websocket.default(HUB_URL);
   ws.on("open", () => {
@@ -3746,6 +3800,11 @@ function connect() {
       version: "0.3.0",
       capabilities: [TASK_ID]
     }));
+    heartbeatTimer = setInterval(() => {
+      if (ws.readyState === import_websocket.default.OPEN) {
+        ws.send(JSON.stringify({ type: "agent:status" }));
+      }
+    }, 3e4);
   });
   ws.on("message", (raw) => {
     try {
@@ -3757,6 +3816,7 @@ function connect() {
     }
   });
   ws.on("close", () => {
+    clearInterval(heartbeatTimer);
     if (!taskReceived) {
       process.stderr.write("[Worker] \u8FDE\u63A5\u65AD\u5F00\uFF0C5s \u540E\u91CD\u8FDE...\n");
       setTimeout(connect, 5e3);
@@ -3767,6 +3827,19 @@ function connect() {
 `);
   });
 }
+function sendTaskDone(taskId, result) {
+  if (taskDoneSent) return;
+  taskDoneSent = true;
+  try {
+    ws.send(JSON.stringify({
+      type: "task:done",
+      taskId,
+      agentId: AGENT_ID,
+      result
+    }));
+  } catch {
+  }
+}
 function handleMessage(msg) {
   switch (msg.type) {
     case "agent:registered":
@@ -3804,8 +3877,8 @@ function detectTaskType(title) {
   return "code";
 }
 function doTask(taskId, title, url, promptB64) {
-  process.stderr.write(`[Worker] \u{1F680} \u5F00\u59CB\u6267\u884C ${taskId}: ${title}
-`);
+  taskTitle = title;
+  log.info(`\u{1F680} \u5F00\u59CB\u6267\u884C: ${title}`);
   ws.send(JSON.stringify({ type: "task:claim", taskId, agentId: AGENT_ID }));
   let prompt;
   let mode = "code";
@@ -3822,7 +3895,7 @@ function doTask(taskId, title, url, promptB64) {
   }
   try {
     const memUrl = `http://127.0.0.1:${WEB_PORT}/api/memory/for-task?q=${encodeURIComponent(title)}`;
-    const kbRaw = (0, import_node_child_process.execSync)(`curl -s "${memUrl}"`, { timeout: 5e3, stdio: ["ignore", "pipe", "ignore"] });
+    const kbRaw = (0, import_node_child_process.execSync)(`curl -s "${memUrl}"`, { timeout: 5e3, stdio: ["ignore", "pipe", "ignore"], windowsHide: true });
     if (kbRaw && kbRaw.length > 0) {
       const kb = JSON.parse(kbRaw.toString());
       if (kb.hit > 0) {
@@ -3845,7 +3918,8 @@ Worker \u5DF2\u542F\u52A8\uFF0CClaude Code \u6B63\u5728\u521D\u59CB\u5316...`
   const child = (0, import_node_child_process.spawn)(CLAUDE_CLI, ["-p", prompt], {
     cwd: REPO_PATH,
     stdio: ["pipe", "pipe", "pipe"],
-    env: { ...process.env, NO_COLOR: "1" }
+    env: { ...process.env, NO_COLOR: "1" },
+    windowsHide: true
   });
   child.stdin.end();
   let output = "";
@@ -3864,6 +3938,14 @@ ${text}`
   const timeoutTimer = setTimeout(() => {
     process.stderr.write(`[Worker] \u23F0 \u8D85\u65F6 (${WORKER_TIMEOUT_MS / 1e3}s)\uFF0C\u5F3A\u5236\u7EC8\u6B62
 `);
+    sendTaskDone(taskId, `\u274C (\u8D85\u65F6 ${WORKER_TIMEOUT_MS / 1e3}s) ${title}
+${output.slice(-3e3)}`);
+    ws.send(JSON.stringify({
+      type: "room:message",
+      roomId: "#review",
+      text: `\u274C ${taskId} \u8D85\u65F6 ${WORKER_TIMEOUT_MS / 1e3}s\uFF0C\u5DF2\u7EC8\u6B62\u3002\u8F93\u51FA:
+${output.slice(-1e3)}`
+    }));
     try {
       child.kill("SIGKILL");
     } catch {
@@ -3879,8 +3961,7 @@ ${text}`
     clearInterval(progressTimer);
     clearTimeout(timeoutTimer);
     if (code === 0) {
-      process.stderr.write(`[Worker] \u2705 Claude Code \u5B8C\u6210 (exit ${code})
-`);
+      log.info(`\u2705 Claude Code \u5B8C\u6210 (exit ${code})`);
       if (mode === "market" || mode === "research" || mode === "template") {
         ws.send(JSON.stringify({
           type: "room:message",
@@ -3889,12 +3970,7 @@ ${text}`
 
 ${output.slice(-3e3)}`
         }));
-        ws.send(JSON.stringify({
-          type: "task:done",
-          taskId,
-          agentId: AGENT_ID,
-          result: output.slice(-5e3) || `${title} \u2014 \u65E0\u8F93\u51FA`
-        }));
+        sendTaskDone(taskId, output.slice(-5e3) || `${title} \u2014 \u65E0\u8F93\u51FA`);
         try {
           tagAndSave(title, output);
         } catch {
@@ -3902,27 +3978,23 @@ ${output.slice(-3e3)}`
       } else {
         const branchMatch = output.match(/push.*?(task\/\S+|feat\/\S+|fix\/\S+)/i);
         const branch = branchMatch ? branchMatch[1] : `worker/${AGENT_ID}`;
-        ws.send(JSON.stringify({
-          type: "task:done",
-          taskId,
-          agentId: AGENT_ID,
-          branch,
-          result: `${title} \u2014
+        sendTaskDone(taskId, `${title} \u2014
 
-${output.slice(-5e3)}`
-        }));
+${output.slice(-5e3)}`);
       }
     } else {
-      process.stderr.write(`[Worker] \u274C Claude Code \u5931\u8D25 (exit ${code})
-`);
+      log.error(`\u274C Claude Code \u5931\u8D25 (exit ${code})`);
       ws.send(JSON.stringify({
         type: "room:message",
         roomId: "#review",
         text: `\u274C ${taskId} \u6267\u884C\u5931\u8D25 (exit ${code})\u3002\u8F93\u51FA:
 ${output.slice(-1e3)}`
       }));
+      sendTaskDone(taskId, `\u274C (\u5931\u8D25 exit ${code}) ${title}
+${output.slice(-3e3)}`);
     }
     setTimeout(() => {
+      clearInterval(heartbeatTimer);
       ws.close();
       process.exit(code === 0 ? 0 : 1);
     }, 3e4);
@@ -3930,8 +4002,10 @@ ${output.slice(-1e3)}`
   child.on("error", (err) => {
     clearInterval(progressTimer);
     clearTimeout(timeoutTimer);
+    clearInterval(heartbeatTimer);
     process.stderr.write(`[Worker] Claude Code \u542F\u52A8\u5931\u8D25: ${err.message}
 `);
+    sendTaskDone(taskId, `\u274C (\u542F\u52A8\u5931\u8D25) ${title}: ${err.message}`);
     ws.send(JSON.stringify({
       type: "room:message",
       roomId: "#review",
@@ -4033,6 +4107,7 @@ connect();
 setTimeout(() => {
   if (!taskReceived) {
     process.stderr.write("[Worker] \u8D85\u65F6\u672A\u6536\u5230\u4EFB\u52A1\uFF0C\u9000\u51FA\n");
+    clearInterval(heartbeatTimer);
     ws.close();
     process.exit(1);
   }