mu-agent-bridge 1.0.73 → 1.0.78

package/dist/tests-ts/core-updater.test.js

@@ -4,85 +4,254 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const strict_1 = __importDefault(require("node:assert/strict"));
+const node_child_process_1 = require("node:child_process");
 const promises_1 = require("node:fs/promises");
 const node_os_1 = require("node:os");
 const node_path_1 = require("node:path");
 const node_test_1 = __importDefault(require("node:test"));
+const node_util_1 = require("node:util");
 const core_updater_1 = require("../src-ts/core-updater");
-async function fixture() {
+const execFileAsync = (0, node_util_1.promisify)(node_child_process_1.execFile);
+async function fixture(directory = "测试目录/package", version = "1.0.0") {
     const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "mab-core-update-"));
-    const packageRoot = (0, node_path_1.join)(root, "node_modules", "mu-agent-bridge");
-    const globalRoot = (0, node_path_1.join)(root, "node_modules");
+    const packageRoot = (0, node_path_1.join)(root, directory);
     await (0, promises_1.mkdir)((0, node_path_1.join)(packageRoot, "dist", "src-ts"), { recursive: true });
-    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "package.json"), JSON.stringify({ name: "mu-agent-bridge", version: "1.0.0" }));
-    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "dist", "src-ts", "cli.js"), "old");
-    return { root, packageRoot, globalRoot };
+    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "package.json"), JSON.stringify({ name: "mu-agent-bridge", version }));
+    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "dist", "src-ts", "cli.js"), `console.log(${JSON.stringify(version)});`);
+    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "dist", "obsolete.js"), "旧发布文件");
+    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "local-config.json"), "保留本地配置");
+    await (0, promises_1.mkdir)((0, node_path_1.join)(packageRoot, "node_modules", "local-tool"), { recursive: true });
+    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "node_modules", "local-tool", "index.js"), "保留本地依赖");
+    await (0, promises_1.mkdir)((0, node_path_1.join)(packageRoot, "plugins", "local-plugin"), { recursive: true });
+    await (0, promises_1.mkdir)((0, node_path_1.join)(packageRoot, "plugins", "dsh", "9.0.0"), { recursive: true });
+    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "plugins", "local-plugin", "plugin.json"), "本地插件");
+    await (0, promises_1.writeFile)((0, node_path_1.join)(packageRoot, "plugins", "dsh", "9.0.0", "plugin.json"), "高版本插件");
+    return { root, packageRoot, stateDir: (0, node_path_1.join)(root, "state") };
 }
-(0, node_test_1.default)("mab start 在核心有新版本时安装、启动并等待健康检查", async () => {
+async function stagePackage(args, version = "1.1.0") {
+    const prefixIndex = args.indexOf("--prefix");
+    strict_1.default.notEqual(prefixIndex, -1, "安装应在暂存目录执行");
+    strict_1.default.ok(args.includes("--install-strategy=nested"), "依赖需要随核心一起迁移");
+    const candidate = (0, node_path_1.join)(args[prefixIndex + 1], "node_modules", "mu-agent-bridge");
+    await (0, promises_1.mkdir)((0, node_path_1.join)(candidate, "dist", "src-ts"), { recursive: true });
+    await (0, promises_1.mkdir)((0, node_path_1.join)(candidate, "node_modules", "fixture-dependency"), { recursive: true });
+    await (0, promises_1.writeFile)((0, node_path_1.join)(candidate, "package.json"), JSON.stringify({ name: "mu-agent-bridge", version }));
+    await (0, promises_1.writeFile)((0, node_path_1.join)(candidate, "dist", "src-ts", "cli.js"), 'console.log(require("../../package.json").version + ":" + require("fixture-dependency"));');
+    await (0, promises_1.writeFile)((0, node_path_1.join)(candidate, "node_modules", "fixture-dependency", "index.js"), 'module.exports="依赖可用";');
+    if (process.platform !== "win32") {
+        await (0, promises_1.mkdir)((0, node_path_1.join)(candidate, "node_modules", ".bin"));
+        await (0, promises_1.symlink)("../fixture-dependency/index.js", (0, node_path_1.join)(candidate, "node_modules", ".bin", "fixture"));
+    }
+    await (0, promises_1.writeFile)((0, node_path_1.join)(candidate, "LICENSE"), "新发布文件");
+    await (0, promises_1.mkdir)((0, node_path_1.join)(candidate, "plugins", "dsh"), { recursive: true });
+    await (0, promises_1.writeFile)((0, node_path_1.join)(candidate, "plugins", "dsh", "plugin.json"), "新版内置插件");
+    return "";
+}
+for (const directory of ["node_modules/mu-agent-bridge", "测试目录/package"]) {
+    (0, node_test_1.default)(`${directory} 的低版本升级后仍由原路径运行新核心及依赖`, async () => {
+        const value = await fixture(directory);
+        const events = [];
+        try {
+            const updated = await (0, core_updater_1.updateCoreAtStartup)({
+                ...value,
+                npm: async (args) => {
+                    events.push(args[0]);
+                    if (args[0] === "root")
+                        return (0, node_path_1.join)(value.root, "node_modules");
+                    if (args[0] === "view")
+                        return '"1.1.0"';
+                    return stagePackage(args);
+                },
+                stopDaemon: async () => { events.push("stop"); },
+                startDaemon: async () => {
+                    events.push("start");
+                    const { stdout } = await execFileAsync(process.execPath, [(0, node_path_1.join)(value.packageRoot, "dist", "src-ts", "cli.js")]);
+                    strict_1.default.equal(stdout.trim(), "1.1.0:依赖可用");
+                },
+                waitForHealth: async () => true,
+            });
+            strict_1.default.equal(updated, true);
+            strict_1.default.deepEqual(events, ["view", "install", "stop", "start"]);
+            strict_1.default.equal(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "local-config.json"), "utf8"), "保留本地配置");
+            strict_1.default.equal(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "node_modules", "local-tool", "index.js"), "utf8"), "保留本地依赖");
+            strict_1.default.equal(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "plugins", "local-plugin", "plugin.json"), "utf8"), "本地插件");
+            strict_1.default.equal(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "plugins", "dsh", "9.0.0", "plugin.json"), "utf8"), "高版本插件");
+            strict_1.default.equal(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "plugins", "dsh", "plugin.json"), "utf8"), "新版内置插件");
+            if (process.platform !== "win32")
+                strict_1.default.equal(await (0, promises_1.readlink)((0, node_path_1.join)(value.packageRoot, "node_modules", ".bin", "fixture")), "../fixture-dependency/index.js");
+            await strict_1.default.rejects((0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "dist", "obsolete.js")), { code: "ENOENT" });
+            strict_1.default.deepEqual(await (0, promises_1.readdir)(value.stateDir), []);
+        }
+        finally {
+            await (0, promises_1.rm)(value.root, { recursive: true, force: true });
+        }
+    });
+    for (const currentVersion of ["1.1.0", "9.0.0"]) {
+        (0, node_test_1.default)(`${directory} 的当前版本 ${currentVersion} 不被 npmjs 的相同或较低版本覆盖`, async () => {
+            const value = await fixture(directory, currentVersion);
+            const calls = [];
+            try {
+                const updated = await (0, core_updater_1.updateCoreAtStartup)({
+                    ...value,
+                    npm: async (args) => { calls.push(args[0]); return args[0] === "root" ? (0, node_path_1.join)(value.root, "node_modules") : '"1.1.0"'; },
+                    stopDaemon: async () => { strict_1.default.fail("不应停止当前后台"); },
+                    startDaemon: async () => { strict_1.default.fail("不应启动其他版本"); },
+                    waitForHealth: async () => true,
+                });
+                strict_1.default.equal(updated, false);
+                strict_1.default.deepEqual(calls, ["view"]);
+                strict_1.default.equal(JSON.parse(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "package.json"), "utf8")).version, currentVersion);
+            }
+            finally {
+                await (0, promises_1.rm)(value.root, { recursive: true, force: true });
+            }
+        });
+    }
+}
+(0, node_test_1.default)("新 daemon 健康检查失败时恢复原路径的旧核心及文件", async () => {
     const value = await fixture();
     const events = [];
+    let healthChecks = 0;
     try {
         const updated = await (0, core_updater_1.updateCoreAtStartup)({
-            packageRoot: value.packageRoot,
-            globalRoot: value.globalRoot,
-            stateDir: (0, node_path_1.join)(value.root, "state"),
-            npm: async (args) => {
-                events.push(args.join(" "));
-                if (args[0] === "view")
-                    return '"1.1.0"';
-                await (0, promises_1.writeFile)((0, node_path_1.join)(value.packageRoot, "package.json"), JSON.stringify({ name: "mu-agent-bridge", version: "1.1.0" }));
-                await (0, promises_1.writeFile)((0, node_path_1.join)(value.packageRoot, "dist", "src-ts", "cli.js"), "new");
-                return "";
-            },
+            ...value,
+            npm: async (args) => args[0] === "root" ? (0, node_path_1.join)(value.root, "node_modules") : args[0] === "view" ? '"1.1.0"' : stagePackage(args),
             stopDaemon: async () => { events.push("stop"); },
-            startDaemon: async () => { events.push("start"); },
-            waitForHealth: async () => true,
+            startDaemon: async () => {
+                const { stdout } = await execFileAsync(process.execPath, [(0, node_path_1.join)(value.packageRoot, "dist", "src-ts", "cli.js")]);
+                events.push(stdout.trim());
+            },
+            waitForHealth: async () => ++healthChecks > 1,
         });
-        strict_1.default.equal(updated, true);
-        strict_1.default.equal(JSON.parse(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "package.json"), "utf8")).version, "1.1.0");
-        strict_1.default.deepEqual(events, ["view mu-agent-bridge version --json", "stop", "install -g mu-agent-bridge@1.1.0 --no-fund --no-audit", "start"]);
+        strict_1.default.equal(updated, false);
+        strict_1.default.deepEqual(events, ["stop", "1.1.0:依赖可用", "stop", "1.0.0"]);
+        strict_1.default.equal(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "dist", "obsolete.js"), "utf8"), "旧发布文件");
+        strict_1.default.equal(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "local-config.json"), "utf8"), "保留本地配置");
+        await strict_1.default.rejects((0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "LICENSE")), { code: "ENOENT" });
     }
     finally {
         await (0, promises_1.rm)(value.root, { recursive: true, force: true });
     }
 });
-(0, node_test_1.default)("新 daemon 健康检查失败时恢复旧核心包并继续旧进程", async () => {
+(0, node_test_1.default)("并发检查遇到已有进程的升级锁时中止，避免启动入口继续覆盖后台", async () => {
     const value = await fixture();
-    const events = [];
+    const lock = (0, node_path_1.join)(value.stateDir, "core-update.lock");
+    await (0, promises_1.mkdir)(lock, { recursive: true });
+    await (0, promises_1.writeFile)((0, node_path_1.join)(lock, String(process.pid)), "");
+    try {
+        await strict_1.default.rejects((0, core_updater_1.updateCoreAtStartup)({
+            ...value,
+            npm: async () => strict_1.default.fail("锁冲突时不能调用 npm"),
+            stopDaemon: async () => strict_1.default.fail("锁冲突时不能停止后台"),
+            startDaemon: async () => strict_1.default.fail("锁冲突时不能启动后台"),
+            waitForHealth: async () => false,
+        }), /已有核心升级检查正在运行/);
+        strict_1.default.equal(JSON.parse(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "package.json"), "utf8")).version, "1.0.0");
+        strict_1.default.deepEqual(await (0, promises_1.readdir)(lock), [String(process.pid)]);
+    }
+    finally {
+        await (0, promises_1.rm)(value.root, { recursive: true, force: true });
+    }
+});
+(0, node_test_1.default)("回滚后的旧后台仍不健康时清理本次进程并保留备份", async () => {
+    const value = await fixture();
+    const stops = [];
+    let launches = 0;
+    try {
+        await strict_1.default.rejects((0, core_updater_1.updateCoreAtStartup)({
+            ...value,
+            npm: async (args) => args[0] === "view" ? '"1.1.0"' : stagePackage(args),
+            stopDaemon: async (pid) => { stops.push(pid); },
+            startDaemon: async () => ({ pid: 1000 + ++launches }),
+            waitForHealth: async () => false,
+        }), /旧 daemon 恢复后的健康检查失败/);
+        strict_1.default.deepEqual(stops, [undefined, 1001, 1002]);
+        strict_1.default.equal(JSON.parse(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "package.json"), "utf8")).version, "1.0.0");
+        const remaining = await (0, promises_1.readdir)(value.stateDir);
+        strict_1.default.equal(remaining.length, 1);
+        strict_1.default.match(remaining[0], /^mab-core-update-/);
+        strict_1.default.equal(JSON.parse(await (0, promises_1.readFile)((0, node_path_1.join)(value.stateDir, remaining[0], "backup", "package.json"), "utf8")).version, "1.0.0");
+    }
+    finally {
+        await (0, promises_1.rm)(value.root, { recursive: true, force: true });
+    }
+});
+(0, node_test_1.default)("退出进程遗留的升级锁由下一次检查清理", async () => {
+    const value = await fixture();
+    const lock = (0, node_path_1.join)(value.stateDir, "core-update.lock");
+    await (0, promises_1.mkdir)(lock, { recursive: true });
+    await (0, promises_1.writeFile)((0, node_path_1.join)(lock, "2147483647"), "");
+    let views = 0;
     try {
         const updated = await (0, core_updater_1.updateCoreAtStartup)({
-            packageRoot: value.packageRoot,
-            globalRoot: value.globalRoot,
-            stateDir: (0, node_path_1.join)(value.root, "state"),
-            npm: async (args) => {
-                if (args[0] === "view")
-                    return '"1.1.0"';
-                await (0, promises_1.writeFile)((0, node_path_1.join)(value.packageRoot, "package.json"), JSON.stringify({ name: "mu-agent-bridge", version: "1.1.0" }));
-                return "";
-            },
-            stopDaemon: async () => { events.push("stop"); },
-            startDaemon: async () => { events.push("start"); },
-            waitForHealth: async () => { events.push("health"); return false; },
+            ...value,
+            npm: async () => { views++; return '"1.0.0"'; },
+            stopDaemon: async () => strict_1.default.fail("版本相同时不能停止后台"),
+            startDaemon: async () => strict_1.default.fail("版本相同时不能启动后台"),
+            waitForHealth: async () => false,
         });
         strict_1.default.equal(updated, false);
-        strict_1.default.equal(JSON.parse(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "package.json"), "utf8")).version, "1.0.0");
-        strict_1.default.deepEqual(events, ["stop", "start", "health", "stop", "start"]);
+        strict_1.default.equal(views, 1);
+        strict_1.default.deepEqual(await (0, promises_1.readdir)(value.stateDir), []);
     }
     finally {
         await (0, promises_1.rm)(value.root, { recursive: true, force: true });
     }
 });
-(0, node_test_1.default)("核心已是最新或不是 npm 全局安装时不停止 daemon", async () => {
+(0, node_test_1.default)("两个检查并发时只有锁持有者可以调用 npm", async () => {
     const value = await fixture();
-    let stops = 0;
+    let release;
+    let entered;
+    const pending = new Promise(done => { release = done; });
+    const started = new Promise(done => { entered = done; });
+    let checks = 0;
+    const options = {
+        ...value,
+        npm: async () => { checks++; entered(); await pending; return '"1.0.0"'; },
+        stopDaemon: async () => strict_1.default.fail("版本相同时不能停止后台"),
+        startDaemon: async () => strict_1.default.fail("版本相同时不能启动后台"),
+        waitForHealth: async () => false,
+    };
+    const first = (0, core_updater_1.updateCoreAtStartup)(options);
     try {
-        const current = await (0, core_updater_1.updateCoreAtStartup)({ packageRoot: value.packageRoot, globalRoot: value.globalRoot, stateDir: (0, node_path_1.join)(value.root, "state"), npm: async () => '"1.0.0"', stopDaemon: async () => { stops++; }, startDaemon: async () => undefined, waitForHealth: async () => true });
-        const local = await (0, core_updater_1.updateCoreAtStartup)({ packageRoot: value.packageRoot, globalRoot: value.root, stateDir: (0, node_path_1.join)(value.root, "state"), npm: async () => '"2.0.0"', stopDaemon: async () => { stops++; }, startDaemon: async () => undefined, waitForHealth: async () => true });
-        strict_1.default.equal(current, false);
-        strict_1.default.equal(local, false);
-        strict_1.default.equal(stops, 0);
+        await started;
+        await strict_1.default.rejects((0, core_updater_1.updateCoreAtStartup)(options), /已有核心升级检查正在运行/);
+        strict_1.default.equal(checks, 1);
     }
     finally {
+        release();
+        await first;
         await (0, promises_1.rm)(value.root, { recursive: true, force: true });
     }
 });
+for (const failure of ["下载失败", "包版本不匹配"]) {
+    (0, node_test_1.default)(`${failure} 时保留原后台和安装内容`, async () => {
+        const value = await fixture();
+        const logs = [];
+        try {
+            const updated = await (0, core_updater_1.updateCoreAtStartup)({
+                ...value,
+                npm: async (args) => {
+                    if (args[0] === "root")
+                        return (0, node_path_1.join)(value.root, "node_modules");
+                    if (args[0] === "view")
+                        return '"1.1.0"';
+                    if (failure === "下载失败")
+                        throw new Error(failure);
+                    return stagePackage(args, "1.0.9");
+                },
+                stopDaemon: async () => { strict_1.default.fail("下载和验证完成前不能停止后台"); },
+                startDaemon: async () => { strict_1.default.fail("不应启动无效包"); },
+                waitForHealth: async () => true,
+                log: (_level, message, fields) => { logs.push(message + JSON.stringify(fields)); },
+            });
+            strict_1.default.equal(updated, false);
+            strict_1.default.equal(JSON.parse(await (0, promises_1.readFile)((0, node_path_1.join)(value.packageRoot, "package.json"), "utf8")).version, "1.0.0");
+            strict_1.default.ok(logs.some(line => /下载失败|版本不匹配/.test(line)), JSON.stringify(logs));
+            strict_1.default.deepEqual(await (0, promises_1.readdir)(value.stateDir), []);
+        }
+        finally {
+            await (0, promises_1.rm)(value.root, { recursive: true, force: true });
+        }
+    });
+}

package/dist/tests-ts/daemon-process.test.js

@@ -43,12 +43,15 @@ const promises_1 = require("node:fs/promises");
 const node_os_1 = require("node:os");
 const node_path_1 = require("node:path");
 const node_test_1 = __importDefault(require("node:test"));
+const node_util_1 = require("node:util");
 const daemon_process_1 = require("../src-ts/daemon-process");
 const cli = "C:\\Users\\molian\\AppData\\Roaming\\npm\\node_modules\\mu-agent-bridge\\dist\\src-ts\\cli.js";
 (0, node_test_1.default)("识别同一安装的全部旧 daemon，兼容 Windows 路径与引号", () => {
     strict_1.default.equal((0, daemon_process_1.isDaemonCommand)(`"C:\\Program Files\\nodejs\\node.exe" ${cli.replaceAll("\\", "/")} start`, cli), true);
     strict_1.default.equal((0, daemon_process_1.isDaemonCommand)(`node.exe "${cli}" --serve`, cli), true);
     strict_1.default.equal((0, daemon_process_1.isDaemonCommand)(`node.exe "${cli.replaceAll("\\", "\\\\")}" start`, cli), true);
+    const spacedCli = cli.replace("molian", "测试 目录");
+    strict_1.default.equal((0, daemon_process_1.isDaemonCommand)(["node", spacedCli, "start"], spacedCli), true);
 });
 (0, node_test_1.default)("清理进程时保留 CLI 操作、其他安装及普通 Node 程序", () => {
     for (const args of ["restart", "stop", "update", "status"])
@@ -68,7 +71,7 @@ const cli = "C:\\Users\\molian\\AppData\\Roaming\\npm\\node_modules\\mu-agent-br
         await (0, promises_1.mkdir)(plugin, { recursive: true });
         await (0, promises_1.writeFile)((0, node_path_1.join)(plugin, "plugin.json"), JSON.stringify({ id: "aun", version: "1.0.0", type: "channel", entry: "index.cjs", apiVersion: "1", capabilities: [] }));
         await (0, promises_1.writeFile)((0, node_path_1.join)(plugin, "index.cjs"), "module.exports={activate:async()=>{},deactivate:async()=>{},health:async()=>({status:'ready'})};");
-        await (0, promises_1.writeFile)(preload, `require(${JSON.stringify((0, node_path_1.resolve)(__dirname, "../src-ts/plugin-manager.js"))}).PluginManager.prototype.checkPackageUpdates=async()=>{console.log('UPDATE_BLOCKED');return new Promise(()=>{});};`);
+        await (0, promises_1.writeFile)(preload, `require(${JSON.stringify((0, node_path_1.resolve)(__dirname, "../src-ts/core-updater.js"))}).updateCoreAtStartup=async()=>false;require(${JSON.stringify((0, node_path_1.resolve)(__dirname, "../src-ts/plugin-manager.js"))}).PluginManager.prototype.checkPackageUpdates=async()=>{console.log('UPDATE_BLOCKED');return new Promise(()=>{});};`);
         child = (0, node_child_process_1.spawn)(process.execPath, ["--require", preload, (0, node_path_1.resolve)(__dirname, "../src-ts/cli.js"), "start"], {
             env: { ...process.env, AGENT_BRIDGE_STATE_DIR: state, AGENT_BRIDGE_PLUGINS_DIR: (0, node_path_1.join)(root, "plugins"), AGENT_BRIDGE_HOST: "127.0.0.1", AGENT_BRIDGE_PORT: "0", AGENT_BRIDGE_TOKEN: "" },
             stdio: ["ignore", "pipe", "pipe"],
@@ -118,3 +121,125 @@ const cli = "C:\\Users\\molian\\AppData\\Roaming\\npm\\node_modules\\mu-agent-br
         await (0, promises_1.rm)(root, { recursive: true, force: true });
     }
 });
+(0, node_test_1.default)("核心升级等待旧后台慢停，且只认可新进程的健康状态", { skip: process.platform === "win32", timeout: 15_000 }, async () => {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "mab-update-restart-"));
+    const state = (0, node_path_1.join)(root, "state");
+    const addressFile = (0, node_path_1.join)(state, "daemon-address.json");
+    const stoppedFile = (0, node_path_1.join)(root, "old-stopped");
+    const preload = (0, node_path_1.join)(root, "update.cjs");
+    let old;
+    let newPid;
+    try {
+        await (0, promises_1.mkdir)(state);
+        const plugin = (0, node_path_1.join)(root, "slow-plugins", "aun");
+        await (0, promises_1.mkdir)(plugin, { recursive: true });
+        await (0, promises_1.writeFile)((0, node_path_1.join)(plugin, "plugin.json"), JSON.stringify({ id: "aun", version: "1.0.0", type: "channel", entry: "index.cjs", apiVersion: "1", capabilities: [] }));
+        await (0, promises_1.writeFile)((0, node_path_1.join)(plugin, "index.cjs"), `module.exports={activate:async()=>{},deactivate:async()=>{await new Promise(done=>setTimeout(done,1200));require('node:fs').writeFileSync(${JSON.stringify(stoppedFile)},'stopped');},health:async()=>({status:'ready'})};`);
+        const oldPreload = (0, node_path_1.join)(root, "old.cjs");
+        await (0, promises_1.writeFile)(oldPreload, `require(${JSON.stringify((0, node_path_1.resolve)(__dirname, "../src-ts/plugin-manager.js"))}).PluginManager.prototype.checkPackageUpdates=async()=>{console.log('READY');return [];};`);
+        old = (0, node_child_process_1.spawn)(process.execPath, [(0, node_path_1.resolve)(__dirname, "../src-ts/cli.js"), "start"], {
+            env: { ...process.env, NODE_OPTIONS: `--require=${oldPreload}`, AGENT_BRIDGE_CORE_UPDATE_CHILD: "1", AGENT_BRIDGE_STATE_DIR: state, AGENT_BRIDGE_PLUGINS_DIR: (0, node_path_1.join)(root, "slow-plugins"), AGENT_BRIDGE_HOST: "127.0.0.1", AGENT_BRIDGE_PORT: "0", AGENT_BRIDGE_TOKEN: "" },
+            stdio: ["ignore", "pipe", "pipe"],
+        });
+        let output = "";
+        await new Promise(done => old.stdout.on("data", chunk => { output += chunk; if (output.includes("READY"))
+            done(); }));
+        const previous = JSON.parse(await (0, promises_1.readFile)(addressFile, "utf8"));
+        await (0, promises_1.writeFile)(preload, `
+      const assert=require('node:assert/strict');
+      const fs=require('node:fs/promises');
+      const readFile=fs.readFile;
+      let maskPid=false;
+      fs.readFile=async function(path,...args){
+        const data=await readFile(path,...args);
+        return maskPid && String(path)===${JSON.stringify(addressFile)} ? JSON.stringify({...JSON.parse(data),pid:process.pid}) : data;
+      };
+      require(${JSON.stringify((0, node_path_1.resolve)(__dirname, "../src-ts/plugin-manager.js"))}).PluginManager.prototype.checkPackageUpdates=async()=>[];
+      if(process.env.MAB_TEST_HOLD_DISCOVERY==='1'){
+        require(${JSON.stringify((0, node_path_1.resolve)(__dirname, "../src-ts/plugin-manager.js"))}).PluginManager.prototype.discover=async()=>{
+          await fs.writeFile(${JSON.stringify((0, node_path_1.join)(root, "hanging-pid"))},String(process.pid));
+          setInterval(()=>{},1000);return new Promise(()=>{});
+        };
+      }
+      require(${JSON.stringify((0, node_path_1.resolve)(__dirname, "../src-ts/core-updater.js"))}).updateCoreAtStartup=async options=>{
+        await options.stopDaemon();
+        assert.equal(await readFile(${JSON.stringify(stoppedFile)},'utf8'),'stopped','停止操作必须等旧后台释放端口');
+        await options.startDaemon();
+        for(let attempt=0;attempt<100;attempt++){
+          try { await readFile(${JSON.stringify(addressFile)},'utf8');break; } catch { await new Promise(done=>setTimeout(done,20)); }
+        }
+        maskPid=true;
+        let settled=false;
+        const health=options.waitForHealth().then(value=>{settled=true;return value;});
+        await new Promise(done=>setTimeout(done,200));
+        try { assert.equal(settled,false,'其他进程的健康响应不能判定升级成功'); }
+        finally { maskPid=false; }
+        assert.equal(await health,true);
+        await options.stopDaemon();
+        process.env.MAB_TEST_HOLD_DISCOVERY='1';
+        const hanging=await options.startDaemon();
+        for(let attempt=0;attempt<100;attempt++){
+          try { await readFile(${JSON.stringify((0, node_path_1.join)(root, "hanging-pid"))},'utf8');break; } catch { await new Promise(done=>setTimeout(done,20)); }
+        }
+        await assert.rejects(readFile(${JSON.stringify(addressFile)},'utf8'),{code:'ENOENT'});
+        await options.stopDaemon(hanging.pid);
+        assert.throws(()=>process.kill(hanging.pid,0),{code:'ESRCH'},'未登记的新进程也必须在回滚前停止');
+        delete process.env.MAB_TEST_HOLD_DISCOVERY;
+        await options.startDaemon();
+        assert.equal(await options.waitForHealth(),true);
+        console.log('UPDATER_HEALTH=true');
+        return true;
+      };
+    `);
+        const { stdout } = await (0, node_util_1.promisify)(node_child_process_1.execFile)(process.execPath, [(0, node_path_1.resolve)(__dirname, "../src-ts/cli.js"), "update"], {
+            env: { ...process.env, NODE_OPTIONS: `--require=${preload}`, AGENT_BRIDGE_STATE_DIR: state, AGENT_BRIDGE_PLUGINS_DIR: (0, node_path_1.join)(root, "plugins"), AGENT_BRIDGE_HOST: "127.0.0.1", AGENT_BRIDGE_PORT: String(previous.port), AGENT_BRIDGE_TOKEN: "" },
+            timeout: 10_000,
+        });
+        strict_1.default.match(stdout, /UPDATER_HEALTH=true/);
+        const address = JSON.parse(await (0, promises_1.readFile)(addressFile, "utf8"));
+        newPid = address.pid;
+        strict_1.default.notEqual(newPid, old.pid);
+        strict_1.default.equal(address.port, previous.port);
+        strict_1.default.equal((await fetch(`http://127.0.0.1:${address.port}/api/health`, { signal: AbortSignal.timeout(2000) })).status, 200);
+    }
+    finally {
+        try {
+            newPid ??= JSON.parse(await (0, promises_1.readFile)(addressFile, "utf8")).pid;
+        }
+        catch { /* 新后台可能未启动 */ }
+        if (newPid && newPid !== old?.pid) {
+            try {
+                process.kill(newPid, "SIGKILL");
+            }
+            catch { /* 已退出 */ }
+        }
+        if (old?.pid && old.exitCode === null && old.signalCode === null) {
+            const exited = (0, node_events_1.once)(old, "exit");
+            old.kill("SIGKILL");
+            await exited;
+        }
+        await (0, promises_1.rm)(root, { recursive: true, force: true });
+    }
+});
+(0, node_test_1.default)("旧地址文件的 PID 指向无关进程时只清理地址，不停止该进程", { skip: process.platform === "win32", timeout: 10_000 }, async () => {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "mab-stale-pid-"));
+    const unrelated = (0, node_child_process_1.spawn)(process.execPath, ["-e", "setInterval(()=>{},1000);console.log('READY');"], { stdio: ["ignore", "pipe", "pipe"] });
+    try {
+        await (0, node_events_1.once)(unrelated.stdout, "data");
+        const addressFile = (0, node_path_1.join)(root, "daemon-address.json");
+        await (0, promises_1.writeFile)(addressFile, JSON.stringify({ pid: unrelated.pid, port: 26107 }));
+        await (0, node_util_1.promisify)(node_child_process_1.execFile)(process.execPath, [(0, node_path_1.resolve)(__dirname, "../src-ts/cli.js"), "stop"], {
+            env: { ...process.env, AGENT_BRIDGE_STATE_DIR: root }, timeout: 5000,
+        });
+        strict_1.default.doesNotThrow(() => process.kill(unrelated.pid, 0), "无关进程应继续运行");
+        await strict_1.default.rejects((0, promises_1.readFile)(addressFile), { code: "ENOENT" });
+    }
+    finally {
+        if (unrelated.exitCode === null && unrelated.signalCode === null) {
+            const exited = (0, node_events_1.once)(unrelated, "exit");
+            unrelated.kill("SIGKILL");
+            await exited;
+        }
+        await (0, promises_1.rm)(root, { recursive: true, force: true });
+    }
+});

package/dist/tests-ts/desktop-launch.test.js

@@ -59,6 +59,31 @@ const desktop_launch_1 = require("../src-ts/plugins/desktop-launch");
         await (0, promises_1.rm)(root, { recursive: true, force: true });
     }
 });
+(0, node_test_1.default)("豆包自动发现支持 LOCALAPPDATA 下的用户安装目录", async () => {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "desktop-localappdata-"));
+    const platform = Object.getOwnPropertyDescriptor(process, "platform");
+    const previousLocalAppData = process.env.LOCALAPPDATA;
+    try {
+        const application = (0, node_path_1.join)(root, "Doubao", "Application");
+        const executable = (0, node_path_1.join)(application, "app", "Doubao.exe");
+        await (0, promises_1.mkdir)((0, node_path_1.join)(application, "app"), { recursive: true });
+        await (0, promises_1.writeFile)((0, node_path_1.join)(application, "Doubao.exe"), "");
+        await (0, promises_1.writeFile)(executable, "");
+        process.env.LOCALAPPDATA = root;
+        Object.defineProperty(process, "platform", { value: "win32" });
+        const unregistered = async () => ({ stdout: "" });
+        strict_1.default.equal(await (0, desktop_launch_1.findWindowsExecutable)("Doubao.exe", undefined, unregistered), executable);
+        strict_1.default.equal(await (0, desktop_launch_1.findWindowsExecutable)("Doubao.exe", (0, node_path_1.join)(root, "missing", "Doubao.exe"), unregistered), executable);
+    }
+    finally {
+        Object.defineProperty(process, "platform", platform);
+        if (previousLocalAppData === undefined)
+            delete process.env.LOCALAPPDATA;
+        else
+            process.env.LOCALAPPDATA = previousLocalAppData;
+        await (0, promises_1.rm)(root, { recursive: true, force: true });
+    }
+});
 (0, node_test_1.default)("桌面程序发现优先选择较新的版本目录", async () => {
     const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "desktop-launch-versions-"));
     try {
@@ -131,6 +156,11 @@ const desktop_launch_1 = require("../src-ts/plugins/desktop-launch");
         await (0, promises_1.rm)(root, { recursive: true, force: true });
     }
 });
+(0, node_test_1.default)("管理式启动把 spawn 异步错误转成可捕获异常并清理归属", async () => {
+    let owned;
+    await strict_1.default.rejects(() => Promise.resolve().then(() => __importStar(require("../src-ts/plugins/desktop-launch"))).then(({ launchManagedAndWait }) => launchManagedAndWait((0, node_path_1.join)((0, node_os_1.tmpdir)(), `missing-desktop-${Date.now()}.exe`), [], undefined, async () => { throw new Error("仍在连接"); }, 400, process => { owned = process?.pid; })), error => error.code === "ENOENT");
+    strict_1.default.equal(owned, undefined);
+});
 (0, node_test_1.default)("使用客户端原有用户目录时，管理式启动不覆盖用户数据目录", async () => {
     const { launchManagedAndWait } = await Promise.resolve().then(() => __importStar(require("../src-ts/plugins/desktop-launch")));
     let args = [];
@@ -230,3 +260,46 @@ const desktop_launch_1 = require("../src-ts/plugins/desktop-launch");
         await (0, promises_1.rm)(root, { recursive: true, force: true });
     }
 });
+(0, node_test_1.default)("豆包优先使用注册表或开始菜单路径，并从启动器定位实际客户端", async () => {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "doubao-system-registration-"));
+    const platform = Object.getOwnPropertyDescriptor(process, "platform");
+    try {
+        const launcher = (0, node_path_1.join)(root, "Application", "Doubao.exe");
+        const client = (0, node_path_1.join)(root, "Application", "app", "Doubao.exe");
+        await (0, promises_1.mkdir)((0, node_path_1.join)(root, "Application", "app"), { recursive: true });
+        await (0, promises_1.writeFile)(launcher, "");
+        await (0, promises_1.writeFile)(client, "");
+        Object.defineProperty(process, "platform", { value: "win32" });
+        const run = async (_file, _args) => ({ stdout: `${launcher}\r\n` });
+        strict_1.default.equal(await (0, desktop_launch_1.findWindowsExecutable)("Doubao.exe", undefined, run), client);
+    }
+    finally {
+        Object.defineProperty(process, "platform", platform);
+        await (0, promises_1.rm)(root, { recursive: true, force: true });
+    }
+});
+(0, node_test_1.default)("ZCode 从安装注册信息或开始菜单发现自定义路径，失效时查当前用户目录", async () => {
+    const root = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "zcode-discovery-"));
+    const platform = Object.getOwnPropertyDescriptor(process, "platform");
+    const previous = process.env.LOCALAPPDATA;
+    try {
+        const registered = (0, node_path_1.join)(root, "自定义应用", "ZCode.exe");
+        const fallback = (0, node_path_1.join)(root, "Programs", "ZCode", "ZCode.exe");
+        await (0, promises_1.mkdir)((0, node_path_1.join)(root, "自定义应用"), { recursive: true });
+        await (0, promises_1.mkdir)((0, node_path_1.join)(root, "Programs", "ZCode"), { recursive: true });
+        await (0, promises_1.writeFile)(registered, "");
+        await (0, promises_1.writeFile)(fallback, "");
+        Object.defineProperty(process, "platform", { value: "win32" });
+        process.env.LOCALAPPDATA = root;
+        strict_1.default.equal(await (0, desktop_launch_1.findWindowsExecutable)("ZCode.exe", undefined, async () => ({ stdout: registered })), registered);
+        strict_1.default.equal(await (0, desktop_launch_1.findWindowsExecutable)("ZCode.exe", undefined, async () => ({ stdout: (0, node_path_1.join)(root, "已卸载", "ZCode.exe") })), fallback);
+    }
+    finally {
+        Object.defineProperty(process, "platform", platform);
+        if (previous === undefined)
+            delete process.env.LOCALAPPDATA;
+        else
+            process.env.LOCALAPPDATA = previous;
+        await (0, promises_1.rm)(root, { recursive: true, force: true });
+    }
+});