@taole/dd-playwright 0.0.0-stage → 0.1.0

package/src/server/daemon.mjs

@@ -0,0 +1,183 @@
+import { spawn } from "node:child_process";
+import { existsSync, readFileSync, writeFileSync, unlinkSync, openSync } from "node:fs";
+import { join } from "node:path";
+import { fileURLToPath } from "node:url";
+
+/**
+ * 检查指定 PID 的进程是否在运行
+ * @param {number} pid
+ * @returns {boolean}
+ */
+export function isProcessRunning(pid) {
+  if (!pid || typeof pid !== "number") return false;
+  try {
+    // 0 信号用于检测进程是否存在且当前用户有权限发送信号
+    process.kill(pid, 0);
+    return true;
+  } catch (err) {
+    return err.code === "EPERM"; // EPERM 表示进程存在但权限不足，也说明活着
+  }
+}
+
+/**
+ * 杀死指定 PID 进程（跨平台兼容）
+ * @param {number} pid
+ */
+export async function killProcess(pid) {
+  if (!isProcessRunning(pid)) return;
+  try {
+    if (process.platform === "win32") {
+      // Windows 下通过 taskkill 递归杀掉子进程
+      const { execSync } = await import("node:child_process");
+      execSync(`taskkill /pid ${pid} /T /F`, { stdio: "ignore" });
+    } else {
+      process.kill(pid, "SIGTERM");
+      // 等待 1 秒若未退出则强制 SIGKILL
+      let count = 0;
+      while (isProcessRunning(pid) && count < 10) {
+        await new Promise((r) => setTimeout(r, 100));
+        count++;
+      }
+      if (isProcessRunning(pid)) {
+        process.kill(pid, "SIGKILL");
+      }
+    }
+  } catch {}
+}
+
+/**
+ * 读取已保存的 Web 后台进程 PID
+ * @param {string} pidFile
+ * @returns {number|null}
+ */
+export function getStoredWebPid(pidFile) {
+  if (!existsSync(pidFile)) return null;
+  try {
+    const raw = readFileSync(pidFile, "utf8").trim();
+    const pid = parseInt(raw, 10);
+    if (Number.isInteger(pid) && isProcessRunning(pid)) {
+      return pid;
+    }
+    // 已失效的 PID 文件直接清理
+    try {
+      unlinkSync(pidFile);
+    } catch {}
+    return null;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 启动后台常驻 Web 服务
+ * @param {object} options
+ * @param {number} options.port
+ * @param {string} options.dataDir
+ * @param {string} options.pidFile
+ * @param {string} options.logFile
+ */
+export async function startWebDaemon({ port, dataDir, pidFile, logFile }) {
+  const existingPid = getStoredWebPid(pidFile);
+  if (existingPid) {
+    return {
+      started: false,
+      alreadyRunning: true,
+      pid: existingPid,
+      url: `http://127.0.0.1:${port}`,
+    };
+  }
+
+  // CLI 脚本路径
+  const cliPath = fileURLToPath(new URL("../../bin/cli.mjs", import.meta.url));
+
+  // 打开日志输出文件
+  const outLog = openSync(logFile, "a");
+  const errLog = openSync(logFile, "a");
+
+  // 以 detached 模式 spawn node 运行 serve-daemon 命令
+  const child = spawn(
+    process.execPath,
+    [cliPath, "serve-daemon", "--port", String(port)],
+    {
+      detached: true,
+      stdio: ["ignore", outLog, errLog],
+      windowsHide: true,
+      env: {
+        ...process.env,
+      },
+    },
+  );
+
+  const pid = child.pid;
+  writeFileSync(pidFile, String(pid), "utf8");
+  child.unref();
+
+  // 等待最多 5 秒检测端口服务是否就绪
+  let ready = false;
+  const deadline = Date.now() + 5000;
+  while (Date.now() < deadline) {
+    try {
+      const res = await fetch(`http://127.0.0.1:${port}/api/accounts`);
+      if (res.ok) {
+        ready = true;
+        break;
+      }
+    } catch {}
+    await new Promise((r) => setTimeout(r, 200));
+  }
+
+  return {
+    started: ready,
+    pid,
+    port,
+    url: `http://127.0.0.1:${port}`,
+    logFile,
+  };
+}
+
+/**
+ * 停止后台常驻 Web 服务
+ * @param {string} pidFile
+ */
+export async function stopWebDaemon(pidFile) {
+  const pid = getStoredWebPid(pidFile);
+  if (!pid) {
+    return { stopped: false, message: "当前未发现运行中的后台 Web 服务" };
+  }
+
+  await killProcess(pid);
+
+  try {
+    if (existsSync(pidFile)) {
+      unlinkSync(pidFile);
+    }
+  } catch {}
+
+  return { stopped: true, pid };
+}
+
+/**
+ * 查询后台常驻 Web 服务运行状态
+ * @param {string} pidFile
+ * @param {number} [port]
+ */
+export async function getWebDaemonStatus(pidFile, port = 12309) {
+  const pid = getStoredWebPid(pidFile);
+  if (!pid) {
+    return { running: false };
+  }
+
+  let responding = false;
+  try {
+    const res = await fetch(`http://127.0.0.1:${port}/api/accounts`);
+    responding = res.ok;
+  } catch {}
+
+  return {
+    running: true,
+    pid,
+    port,
+    responding,
+    url: `http://127.0.0.1:${port}`,
+  };
+}