tt-help-cli-ycl 1.4.105 → 1.4.107

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tt-help-cli-ycl",
-  "version": "1.4.105",
+  "version": "1.4.107",
   "description": "TikTok user & video data scraper - extract ttSeller, verified, locationCreated from HTML source",
   "type": "module",
   "bin": {

package/src/watchdog/upgrader.js

@@ -6,7 +6,8 @@
  * 见 runUpgradeMode。旧 watchdog 收到 SIGTERM 后优雅退出，新版接管。
  */
 
-import { exec, spawn } from "child_process";
+import { exec, spawn, execFile } from "child_process";
+import { promisify } from "util";
 import { readFileSync, openSync } from "fs";
 import { fileURLToPath } from "url";
 import { dirname, join } from "path";
@@ -18,6 +19,7 @@ const PACKAGE = "tt-help-cli-ycl";
 const REGISTRY = "https://registry.npmjs.org";
 const homeDir = homedir() || process.env.HOME || process.env.USERPROFILE || "";
 const TT_HELP_CONFIG = join(homeDir, ".tt-help.json");
+const execFileP = promisify(execFile);
 
 /**
  * 升级检测源：TT_HELP_REGISTRY 环境变量 > ~/.tt-help.json 的 npmRegistry > 官方 npmjs.org
@@ -253,6 +255,140 @@ async function terminateProcess(pid, waitMs = 5000) {
   return false;
 }
 
+/** 枚举指定父进程的直接子进程 PID（Windows: PowerShell；POSIX: pgrep） */
+async function listChildPids(parentPid) {
+  if (!parentPid) return [];
+  try {
+    if (process.platform === "win32") {
+      const ps =
+        `Get-CimInstance Win32_Process -Filter "ParentProcessId=${parentPid}" ` +
+        `| Select-Object -ExpandProperty ProcessId`;
+      const { stdout } = await execFileP(
+        "powershell.exe",
+        ["-NoProfile", "-Command", ps],
+        { timeout: 15000, windowsHide: true },
+      );
+      return String(stdout)
+        .split(/\r?\n/)
+        .map((s) => parseInt(s.trim(), 10))
+        .filter((n) => n > 0);
+    }
+    const { stdout } = await execFileP("pgrep", ["-P", String(parentPid)], {
+      timeout: 15000,
+    });
+    return String(stdout)
+      .split(/\r?\n/)
+      .map((s) => parseInt(s.trim(), 10))
+      .filter((n) => n > 0);
+  } catch {
+    return [];
+  }
+}
+
+/** 强杀单个进程及其子树（Windows: taskkill /T /F；POSIX: SIGKILL） */
+async function forceKillTree(pid) {
+  try {
+    if (process.platform === "win32") {
+      await execFileP(
+        "taskkill.exe",
+        ["/PID", String(pid), "/T", "/F"],
+        { timeout: 15000, windowsHide: true },
+      );
+    } else {
+      process.kill(pid, "SIGKILL");
+    }
+  } catch {
+    /* 进程可能已退出 */
+  }
+}
+
+/**
+ * 清理旧 watchdog 残留的子进程（升级后不留孤儿占端口/资源）。
+ * 流程：枚举直接子进程（排除 excludePid）→ 逐个强杀整棵子树 → 等 1s → 复查残留再杀。
+ * 背景：Windows 上 SIGTERM=强杀，旧 watchdog 的业务子进程（webserver 等）会变孤儿；
+ *       macOS/Linux 上优雅退出未完成（SIGKILL 兜底）时同样需要清理。
+ * @returns {Promise<number[]>} 被清理的 PID 列表
+ */
+async function killChildProcesses(parentPid, excludePid = 0) {
+  const killed = [];
+  const firstRound = (await listChildPids(parentPid)).filter(
+    (p) => p !== excludePid,
+  );
+  for (const pid of firstRound) {
+    await forceKillTree(pid);
+    killed.push(pid);
+  }
+  if (killed.length > 0) {
+    console.error(
+      `[upgrade-proc] 已清理旧 watchdog 残留子进程: ${killed.join(", ")}`,
+    );
+    await new Promise((r) => setTimeout(r, 1000));
+    // 复查：仍有残留则再杀一轮
+    const remain = (await listChildPids(parentPid)).filter(
+      (p) => p !== excludePid,
+    );
+    for (const pid of remain) {
+      await forceKillTree(pid);
+      killed.push(pid);
+    }
+  }
+  return killed;
+}
+
+/** 枚举当前所有 watchdog 进程 PID（存活校验前后对比用；不含升级器 U 自身） */
+async function listWatchdogPids() {
+  try {
+    if (process.platform === "win32") {
+      const ps =
+        `Get-CimInstance Win32_Process -Filter "Name='node.exe'" ` +
+        `| Where-Object { $_.CommandLine -match 'cli\\.js watchdog' } ` +
+        `| Select-Object -ExpandProperty ProcessId`;
+      const { stdout } = await execFileP(
+        "powershell.exe",
+        ["-NoProfile", "-Command", ps],
+        { timeout: 15000, windowsHide: true },
+      );
+      return String(stdout)
+        .split(/\r?\n/)
+        .map((s) => parseInt(s.trim(), 10))
+        .filter((n) => n > 0);
+    }
+    const { stdout } = await execFileP("pgrep", ["-f", "cli.js watchdog"], {
+      timeout: 15000,
+    });
+    return String(stdout)
+      .split(/\r?\n/)
+      .map((s) => parseInt(s.trim(), 10))
+      .filter((n) => n > 0);
+  } catch {
+    return [];
+  }
+}
+
+/**
+ * 启动新版 watchdog：
+ * - Windows：`cmd.exe /c start cmd /k <node> <cli.js> watchdog ...`
+ *   start 弹出【独立交互 cmd 窗口】（与 U 完全解耦），watchdog 日志直接显示在窗口里，
+ *   /k 保持窗口打开（watchdog 退出后提示符仍在，便于排障）；中间 cmd 隐藏不闪窗。
+ * - 其他平台：直接 detached spawn node，日志落盘 logFd。
+ * @returns {import("child_process").ChildProcess}
+ */
+function spawnNewWatchdog(cliPath, args, logFd) {
+  if (process.platform === "win32") {
+    const child = spawn(
+      "cmd.exe",
+      ["/c", "start", "cmd", "/k", process.execPath, cliPath, ...args],
+      { detached: true, stdio: "ignore", windowsHide: true },
+    );
+    child.unref();
+    child.on("error", (err) => {
+      console.error(`[upgrade-proc] 新版 watchdog 启动失败: ${err.message}`);
+    });
+    return child;
+  }
+  return spawnChild([cliPath, ...args], { logFd });
+}
+
 /** 新版 watchdog 的 CLI 参数（透传 server/clientId/interval/noDaemon，保证与旧版一致） */
 function buildWatchdogArgs({ serverUrl, clientId, intervalSec, noDaemon }) {
   const args = ["watchdog"];
@@ -301,7 +437,9 @@ export async function runUpgradeMode(argv) {
   }
   console.error(`[upgrade-proc] 新版本验证通过 (${verify.version})`);
 
-  // 3) 杀旧：SIGTERM 优雅优先，5s 未退强杀兜底（U 不卡死）
+  // 3) 杀旧：先优雅发退出信号 → 等待 → 检测残留子进程 → 强杀
+  //      Windows 上 SIGTERM=强杀（无优雅），旧 watchdog 的业务子进程（webserver 等）
+  //      会变孤儿占端口，因此杀旧后必须检测并强杀残留子进程（排除 U 自己）。
   console.error(`[upgrade-proc] 通知旧 watchdog (pid=${oldPid}) 退出 ...`);
   const exited = await terminateProcess(oldPid, 5000);
   if (exited) {
@@ -310,9 +448,10 @@ export async function runUpgradeMode(argv) {
     console.error(`[upgrade-proc] 旧 watchdog 未在 5s 内退出，已强制终止`);
     await new Promise((r) => setTimeout(r, 1000)); // 给强杀一点生效时间
   }
+  await killChildProcesses(oldPid, process.pid);
 
   // 4) 旧进程已退，启新版 watchdog
-  //     新版永远 detached + windowsHide:false → 独立控制台窗口（前端可见），
+  //     Windows：显式弹独立交互 cmd 窗口（start → cmd /k），日志直接显示在窗口，
   //     与旧 cmd /k 窗口解耦（旧窗口控制台在旧 watchdog 被杀后不可复用）
   console.error(`[upgrade-proc] 启动新版 watchdog ...`);
   const cliPath = join(__dirname, "..", "..", "cli.js");
@@ -322,31 +461,40 @@ export async function runUpgradeMode(argv) {
     intervalSec,
     noDaemon,
   });
-  const logFd = openLogFile(
-    "tt-help-watchdog",
-    "upgrade-proc",
-    "新版 watchdog 日志",
-  );
+  const logFd =
+    process.platform === "win32"
+      ? undefined
+      : openLogFile("tt-help-watchdog", "upgrade-proc", "新版 watchdog 日志");
+  // 记录启动前的 watchdog 进程集合，用于存活校验前后对比（Windows 经 start 启动时
+  // 中间 cmd 秒退，不能检查其 PID）
+  const beforePids = await listWatchdogPids();
   let child;
   try {
-    // 新版永远独立控制台窗口（hide 默认 false），不再继承旧 cmd /k 窗口
-    child = spawnChild([cliPath, ...args], { logFd });
-    console.error(`[upgrade-proc] 新版 watchdog 已启动 (pid=${child.pid})`);
+    child = spawnNewWatchdog(cliPath, args, logFd);
+    if (process.platform === "win32") {
+      console.error(
+        `[upgrade-proc] 已弹独立 cmd 窗口启动新版 watchdog（窗口内可见运行日志）`,
+      );
+    } else {
+      console.error(`[upgrade-proc] 新版 watchdog 已启动 (pid=${child.pid})`);
+    }
   } catch (e) {
     console.error(`[upgrade-proc] 新版 watchdog 启动失败: ${e.message}`);
     process.exit(1);
   }
 
-  // 5) 存活校验：3s 后确认新版进程存活；未存活则明确报错（可看日志定位），不静默
+  // 5) 存活校验：3s 后确认出现新增的 watchdog 进程；未存活则明确报错，不静默
   await new Promise((r) => setTimeout(r, 3000));
-  try {
-    process.kill(child.pid, 0);
+  const newPids = (await listWatchdogPids()).filter(
+    (p) => !beforePids.includes(p),
+  );
+  if (newPids.length > 0) {
     console.error(
-      `[upgrade-proc] 新版 watchdog 存活确认 (pid=${child.pid})，升级完成`,
+      `[upgrade-proc] 新版 watchdog 存活确认 (pid=${newPids.join(",")})，升级完成`,
     );
-  } catch {
+  } else {
     console.error(
-      `[upgrade-proc] 新版 watchdog (pid=${child.pid}) 启动后未存活，请查看日志排障`,
+      `[upgrade-proc] 新版 watchdog 启动后未存活，请查看窗口/日志排障`,
     );
     process.exit(1);
   }