pi-proactive 0.2.0-beta.5 → 0.2.0-beta.7

package/tools/daemon-common.mjs

@@ -0,0 +1,289 @@
+/**
+ * pi-proactive 常驻助理（**跨平台用户侧入口**）的共享底座。
+ *
+ * 为什么要这个文件：`tools/daemon-install.mjs`、`tools/daemon-uninstall.mjs`、
+ * `tools/verify-windows.mjs` 三者共用「平台判定 / 路径解析 / 模板渲染 / 命令执行」——
+ * **同一口径只写一份**（避免安装器与卸载器对「端点路径」「单元位置」的理解漂移）。
+ *
+ * 硬约束：
+ *   · 零第三方依赖（只用 Node 内置模块）；
+ *   · 纯 ESM（`.mjs`）—— **不是 TypeScript**：`.mjs` 里不写类型注解（类型信息走 JSDoc），
+ *     否则 Node 直接语法报错；包内零 .ts 同样是发布纪律 W6；
+ *   · Windows / macOS / Linux 同一份代码。
+ */
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import { spawnSync } from "node:child_process";
+import { fileURLToPath, pathToFileURL } from "node:url";
+
+/** unix socket 端点路径上限（与 `src/socket.ts` 的 MAX_SOCKET_PATH_BYTES 逐字一致）。 */
+export const MAX_SOCKET_PATH_BYTES = 100;
+
+/**
+ * 平台判定。
+ *
+ * @param {string} [override] 显式平台（`win32` / `darwin` / `linux`）—— 供计划打印与测试用；
+ *                             缺省读环境变量 `PI_PROACTIVE_PLATFORM`，再缺省用真实平台。
+ * @returns {string}
+ */
+export function detectPlatform(override = "") {
+	const raw = String(override || process.env.PI_PROACTIVE_PLATFORM || "").toLowerCase();
+	if (raw === "win32" || raw === "windows") return "win32";
+	if (raw === "darwin" || raw === "macos") return "darwin";
+	if (raw === "linux") return "linux";
+	if (process.platform === "win32") return "win32";
+	if (process.platform === "darwin") return "darwin";
+	if (process.platform === "linux") return "linux";
+	return process.platform;
+}
+
+/**
+ * 包根目录（tools/ 的上一级）。
+ * @param {string} importMetaUrl
+ * @returns {string}
+ */
+export function pkgRootFrom(importMetaUrl) {
+	return path.resolve(path.dirname(fileURLToPath(importMetaUrl)), "..");
+}
+
+/**
+ * agent 根目录（默认 `~/.pi/agent`；可用 `PI_CODING_AGENT_DIR` 覆盖）。
+ * @returns {string}
+ */
+export function defaultAgentDir() {
+	return process.env.PI_CODING_AGENT_DIR || path.join(os.homedir(), ".pi", "agent");
+}
+
+/**
+ * 稳定身份（跨重启连续；缺省按主机名派生）。
+ * @returns {string}
+ */
+export function defaultAgentId() {
+	return `employee#${os.hostname()}#pi-proactive-daemon`;
+}
+
+/**
+ * 常驻入口（值班进程本体）。
+ *
+ * 开发态（仓库）：`src/daemon.ts` 是真源 → 优先；
+ * 产物态（npm 包）：包里**没有** src/ → 落到 `build/src/daemon.js`。
+ * 候选清单与错误信息同源 → 报错时说的就是真的试过的路径。
+ *
+ * @param {string} pkgRoot
+ * @returns {{ entry: string; needsStripTypes: boolean } | null}
+ */
+export function resolveDaemonEntry(pkgRoot) {
+	for (const c of ["src/daemon.ts", "build/src/daemon.js"]) {
+		const p = path.join(pkgRoot, c);
+		if (fs.existsSync(p)) return { entry: p, needsStripTypes: p.endsWith(".ts") };
+	}
+	return null;
+}
+
+/**
+ * 端点（socket）路径 —— **复用 `src/socket.ts` 的实现**（不复制口径）。
+ *
+ * 用子进程 import 源码模块：`.ts` 走 Node 的类型擦除，`.js` 直接跑；
+ * 产物态与开发态都指向同一份逻辑 → 不会算错。
+ *
+ * @param {string} pkgRoot
+ * @param {string} agentId
+ * @param {string} agentDir
+ * @returns {{ path: string; bytes: number } | null}
+ */
+export function resolveSocketPath(pkgRoot, agentId, agentDir) {
+	const mod = ["src/socket.ts", "build/src/socket.js"]
+		.map((c) => path.join(pkgRoot, c))
+		.find((p) => fs.existsSync(p));
+	if (!mod) return null;
+	const code =
+		`import(${JSON.stringify(pathToFileURL(mod).href)})` +
+		`.then((m)=>process.stdout.write(m.socketPathFor(process.argv[1], process.argv[2])))` +
+		`.catch(()=>process.exit(3))`;
+	const res = spawnSync(process.execPath, ["--experimental-strip-types", "-e", code, agentId, agentDir], {
+		encoding: "utf8",
+		timeout: 20_000,
+	});
+	if (res.status !== 0) return null;
+	const p = (res.stdout ?? "").trim();
+	if (!p) return null;
+	return { path: p, bytes: Buffer.byteLength(p, "utf8") };
+}
+
+/**
+ * 同步跑一条命令（不继承 stdin；带超时守卫）。
+ *
+ * @param {string} cmd
+ * @param {string[]} args
+ * @param {{ cwd?: string; timeout?: number }} [opts]
+ */
+export function run(cmd, args, opts = {}) {
+	const res = spawnSync(cmd, args, {
+		encoding: "utf8",
+		timeout: opts.timeout ?? 60_000,
+		cwd: opts.cwd,
+		stdio: ["ignore", "pipe", "pipe"],
+	});
+	return {
+		code: res.status ?? (res.error ? 1 : 0),
+		ok: res.status === 0 && !res.error,
+		stdout: res.stdout ?? "",
+		stderr: res.stderr ?? "",
+		error: res.error ? String(res.error.message) : "",
+	};
+}
+
+/**
+ * 进程是否还活着（pid 无效 → false）。
+ * @param {number | undefined | null} pid
+ * @returns {boolean}
+ */
+export function isAlive(pid) {
+	if (!pid || pid <= 0) return false;
+	try {
+		process.kill(pid, 0);
+		return true;
+	} catch {
+		return false;
+	}
+}
+
+/**
+ * 读 JSON（坏数据不抛 → 返回 null；巡检脚本绝不能被脏输入搞崩）。
+ * @param {string} file
+ */
+export function readJsonSafe(file) {
+	try {
+		return JSON.parse(fs.readFileSync(file, "utf8"));
+	} catch {
+		return null;
+	}
+}
+
+/**
+ * Windows 命令行参数引号化（含空格/引号 → 必须加引号）。
+ * @param {string} value
+ * @returns {string}
+ */
+export function quoteWindows(value) {
+	return `"${String(value).replace(/"/g, '\\"')}"`;
+}
+
+/**
+ * Windows 计划任务的可执行命令串（`schtasks /TR` 的值）。
+ *
+ * 关键：**路径含空格必须引号化**（`C:\Program Files\nodejs\node.exe`）。
+ *
+ * @param {{ node: string; runner: string; extraArgs?: string[] }} o
+ * @returns {string}
+ */
+export function buildWindowsTaskCommand(o) {
+	const parts = [quoteWindows(o.node), quoteWindows(o.runner), ...(o.extraArgs ?? []).map(quoteWindows)];
+	return parts.join(" ");
+}
+
+/**
+ * systemd 的参数引号化（路径含空格 → 加引号）。
+ * @param {string} value
+ * @returns {string}
+ */
+export function quoteSystemd(value) {
+	const s = String(value);
+	return /[\s"']/.test(s) ? `"${s.replace(/"/g, '\\"')}"` : s;
+}
+
+/**
+ * XML 文本转义（plist 里路径可能带 & < >）。
+ * @param {string} value
+ * @returns {string}
+ */
+export function xmlEscape(value) {
+	return String(value)
+		.replace(/&/g, "&amp;")
+		.replace(/</g, "&lt;")
+		.replace(/>/g, "&gt;");
+}
+
+/**
+ * 渲染 launchd 单元（**复用 `daemon/launchd.plist` 模板，不改模板本身**）。
+ *
+ * 只做两件事：
+ *   ① 占位符替换（label / pkgRoot / agentId / logDir / runner）；
+ *   ② 把 `ProgramArguments` 换成**直接调用 node**（模板里是 `/bin/bash` →
+ *      包内不得依赖 bash；unix 侧 bash 仍在，但统一走 node 才能三平台同源）。
+ *
+ * @param {string} template
+ * @param {{ label: string; pkgRoot: string; agentId: string; logDir: string; runner: string; programArgs: string[] }} o
+ * @returns {string}
+ */
+export function renderLaunchdPlist(template, o) {
+	let out = template
+		.replace(/__LABEL__/g, o.label)
+		.replace(/__PKG_ROOT__/g, o.pkgRoot)
+		.replace(/__AGENT_ID__/g, o.agentId)
+		.replace(/__LOG_DIR__/g, o.logDir)
+		.replace(/__RUNNER__/g, o.runner);
+	const entries = o.programArgs.map((a) => `\t\t<string>${xmlEscape(a)}</string>`).join("\n");
+	out = out.replace(
+		/(<key>ProgramArguments<\/key>\s*<array>)[\s\S]*?(<\/array>)/,
+		(_m, head, tail) => `${head}\n${entries}\n\t${tail}`,
+	);
+	return out;
+}
+
+/**
+ * 渲染 systemd --user 单元（**复用 `daemon/pi-proactive-daemon.service` 模板**）。
+ *
+ * 与 plist 同理：`ExecStart` 换成直接调用 node（不再经 bash）。
+ *
+ * @param {string} template
+ * @param {{ pkgRoot: string; agentId: string; workDir: string; execStart: string }} o
+ * @returns {string}
+ */
+export function renderSystemdUnit(template, o) {
+	return template
+		.split("\n")
+		.map((line) => {
+			if (line.startsWith("ExecStart=")) return `ExecStart=${o.execStart}`;
+			if (line.startsWith("WorkingDirectory=")) return `WorkingDirectory=${o.workDir || o.pkgRoot}`;
+			if (line.startsWith("Environment=PI_PROACTIVE_AGENT_ID=")) {
+				return `Environment=PI_PROACTIVE_AGENT_ID=${o.agentId}`;
+			}
+			return line;
+		})
+		.join("\n");
+}
+
+/**
+ * 选一个**够短**的 agent 根（socket 端点路径 < 100B 的硬约束）。
+ *
+ * @param {string[]} candidates 候选根（按优先级）
+ * @param {(dir: string) => number | null} measure 量一条候选的端点路径字节数（null = 量不出来）
+ * @returns {{ dir: string; bytes: number } | null}
+ */
+export function chooseShortAgentDir(candidates, measure) {
+	for (const dir of candidates) {
+		const bytes = measure(dir);
+		if (bytes !== null && bytes <= MAX_SOCKET_PATH_BYTES) return { dir, bytes };
+	}
+	return null;
+}
+
+/**
+ * 心跳文件路径（与 `src/daemon-config.ts` 的 heartbeatPath 同口径）。
+ * @param {string} agentDir
+ * @returns {string}
+ */
+export function heartbeatPathFor(agentDir) {
+	return path.join(agentDir, "proactive", "daemon", "heartbeat.json");
+}
+
+/**
+ * 值班锁目录（与 `daemon/run-daemon.mjs` 同口径）。
+ * @param {string} agentDir
+ * @returns {string}
+ */
+export function lockPathFor(agentDir) {
+	return path.join(agentDir, "proactive", "daemon", "daemon.lock");
+}