dsh-selfupdater 0.3.0

package/lib/plugin-updater.mjs

@@ -0,0 +1,424 @@
+#!/usr/bin/env node
+/**
+ * dsh-selfupdater 分离式插件升级脚本（独立进程执行，不依赖 DSH 存活）。
+ *
+ * 为什么必须分离进程：插件文件装在 profile 目录里且已被 DSH 进程加载进内存，
+ * Windows 上运行中文件被锁、Linux 上热替换也可能出现半新半旧状态；
+ * 所以插件路由只负责"触发后退出"，由本脚本完成：
+ *   备份 profile → 从 npm 下载新版 tgz → dsh plugin add 安装 → 重启 → 健康检查 → 失败回滚。
+ *
+ * 用法（由 lib/index.js 的 POST /plugins/update 自动调用）：
+ *   node plugin-updater.mjs --pid <DSH主进程PID> --app-dir <APP目录> \
+ *        --workspace <工作区目录> --port <服务端口>
+ */
+import { spawn } from 'node:child_process';
+import net from 'node:net';
+import { appendFileSync, copyFileSync, existsSync, mkdirSync, readFileSync, renameSync, rmSync, writeFileSync } from 'node:fs';
+import { join, resolve } from 'node:path';
+
+/* ------------------------------------------------------------------ *
+ * 工具函数
+ * ------------------------------------------------------------------ */
+
+/** 解析 --key value 形式的命令行参数。 */
+function parseArgs(argv) {
+    const out = {};
+    for (let i = 2; i < argv.length; i++) {
+        if (!argv[i].startsWith('--')) continue;
+        const next = argv[i + 1];
+        out[argv[i].slice(2)] = next !== undefined && !next.startsWith('--') ? next : true;
+    }
+    return out;
+}
+
+const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
+
+/**
+ * semver 比较（零依赖实现，与 updater.mjs 保持一致）。
+ * 返回负数/0/正数；任一侧不是合法 semver 时返回 null。
+ */
+const SEMVER_RE = /^(\d+)\.(\d+)\.(\d+)(?:-([0-9A-Za-z.-]+))?(?:\+[0-9A-Za-z.-]+)?$/;
+
+function compareVersions(a, b) {
+    const pa = SEMVER_RE.exec(String(a ?? '').trim());
+    const pb = SEMVER_RE.exec(String(b ?? '').trim());
+    if (pa === null || pb === null) return null;
+    for (let i = 1; i <= 3; i++) {
+        const x = Number(pa[i]);
+        const y = Number(pb[i]);
+        if (x !== y) return x - y;
+    }
+    // 有预发布段的一方更小（1.0.0-rc.1 < 1.0.0）；双方都有时逐段比较。
+    const preA = pa[4] === undefined ? [] : pa[4].split('.');
+    const preB = pb[4] === undefined ? [] : pb[4].split('.');
+    if (preA.length === 0 || preB.length === 0) return preB.length - preA.length;
+    for (let i = 0; i < Math.max(preA.length, preB.length); i++) {
+        const x = preA[i];
+        const y = preB[i];
+        if (x === undefined) return -1;
+        if (y === undefined) return 1;
+        if (x === y) continue;
+        const nx = /^\d+$/.test(x);
+        const ny = /^\d+$/.test(y);
+        if (nx && ny) return Number(x) - Number(y);
+        if (nx !== ny) return nx ? -1 : 1;
+        return x < y ? -1 : 1;
+    }
+    return 0;
+}
+
+/** 仅当 latest 严格高于 installed 才算可升级。 */
+function isNewer(latest, installed) {
+    const cmp = compareVersions(latest, installed);
+    return cmp !== null && cmp > 0;
+}
+
+/** 探测端口是否已被监听（占用=true）。 */
+function probePort(port, host = '127.0.0.1') {
+    return new Promise((resolveProbe) => {
+        const socket = net.connect({ host, port });
+        socket.on('connect', () => { socket.destroy(); resolveProbe(true); });
+        socket.on('error', () => resolveProbe(false));
+    });
+}
+
+/** 等待端口释放。 */
+async function waitPortFree(port, timeoutMs) {
+    const deadline = Date.now() + timeoutMs;
+    while (Date.now() < deadline) {
+        if (!(await probePort(port))) return true;
+        await sleep(500);
+    }
+    return false;
+}
+
+/** 进程是否仍存活（kill(pid,0) 只做权限探测不真正发信号）。 */
+function isAlive(pid) {
+    try {
+        process.kill(pid, 0);
+        return true;
+    } catch {
+        return false;
+    }
+}
+
+/** 结束旧 DSH 主进程：SIGTERM 给优雅退出机会，超时后 SIGKILL 兜底。 */
+async function killProcess(pid) {
+    if (!isAlive(pid)) return;
+    process.kill(pid, 'SIGTERM');
+    const deadline = Date.now() + 8000;
+    while (isAlive(pid) && Date.now() < deadline) await sleep(300);
+    if (isAlive(pid)) {
+        try { process.kill(pid, 'SIGKILL'); } catch { /* 已退出则忽略 */ }
+        await sleep(1000);
+    }
+}
+
+/* ------------------------------------------------------------------ *
+ * 全局上下文与状态落盘
+ * ------------------------------------------------------------------ */
+
+const args = parseArgs(process.argv);
+
+/** DSH 应用目录（含 bin/node 与 node_modules/@deepseek-ai/dsh）。 */
+const appDir = resolve(String(args['app-dir'] ?? ''));
+/** 工作区目录（存放 .dsh 状态目录与插件 profile）。 */
+const workspace = resolve(String(args.workspace ?? process.env.TRIM_VAR ?? appDir));
+/** Web 服务端口，健康检查用。 */
+const port = parseInt(String(args.port ?? process.env.DSH_PORT ?? '3081'), 10);
+
+const NODE_BIN = join(appDir, 'bin', 'node');
+const DSH_BIN = join(appDir, 'node_modules', '@deepseek-ai', 'dsh', 'lib', 'bin.js');
+/** 插件 profile 名（与 runner.js / lib/index.js 的约定一致）。 */
+const PROFILE = process.env.DSH_PLUGIN_PROFILE ?? 'web';
+
+const dshStateDir = join(workspace, '.dsh');
+const statusFile = join(dshStateDir, 'pluginupdate-status.json');
+const logFile = join(dshStateDir, 'pluginupdate.log');
+const lockFile = join(dshStateDir, 'pluginupdate.lock');
+const stagingDir = join(dshStateDir, 'pluginupdate-staging');
+const profileDir = join(dshStateDir, 'profiles', PROFILE);
+/** 升级前的 profile 完整备份目录（失败回滚用；成功后保留一次供排查）。 */
+const bakProfileDir = join(dshStateDir, `profiles.${PROFILE}.pluginupdate-bak`);
+
+/** 国内镜像优先的 registry 候选列表（与 index.js 的策略一致）。 */
+function registryCandidates() {
+    const custom = process.env.DSHSU_REGISTRY_URL?.replace(/\/+$/, '');
+    const list = [custom, 'https://mirrors.cloud.tencent.com/npm', 'https://registry.npmjs.org'];
+    return [...new Set(list.filter((v) => typeof v === 'string' && v !== ''))];
+}
+
+/** 读插件状态文件（不存在返回空对象）。 */
+function readStatus() {
+    try { return JSON.parse(readFileSync(statusFile, 'utf8')); } catch { return {}; }
+}
+
+let status = {};
+
+/** 向日志追加一行（失败静默，日志不能阻断升级流程）。 */
+function log(message) {
+    const line = `[${new Date().toISOString()}] ${message}`;
+    try { appendFileSync(logFile, line + '\n'); } catch { /* 忽略 */ }
+    console.log(line);
+}
+
+/** 合并写入插件状态文件，供 UI 卡片轮询展示进度。 */
+function setState(state, message, extra = {}) {
+    status = { ...status, state, message: message ?? status.message, updatedAt: new Date().toISOString(), ...extra };
+    try { writeFileSync(statusFile, JSON.stringify({ ...status, pid: process.pid }, null, 2)); } catch { /* 同上 */ }
+    log(`state=${state}${message ? ` :: ${message}` : ''}`);
+}
+
+/** 终态收尾：释放锁文件并退出。 */
+function finish(exitCode = 0) {
+    try { rmSync(lockFile, { force: true }); } catch { /* 忽略 */ }
+    process.exit(exitCode);
+}
+
+/* ------------------------------------------------------------------ *
+ * registry 查询与 tgz 下载
+ * ------------------------------------------------------------------ */
+
+/**
+ * 从单个 registry 取包的 latest 版本号与 tgz 下载地址。
+ * 注意：镜像返回的 dist.tarball 可能仍指向官方源，这里不直接采用，
+ * 而是按 npm 标准路径规则自行拼出镜像侧的下载地址（保证国内可达）。
+ */
+async function fetchPackument(base, pkgName) {
+    const res = await fetch(`${base}/${encodeURIComponent(pkgName)}`, {
+        headers: { accept: 'application/json', 'user-agent': 'dsh-selfupdater' },
+        signal: AbortSignal.timeout(15000),
+    });
+    if (!res.ok) throw new Error(`HTTP ${res.status}`);
+    const doc = await res.json();
+    const latest = doc?.['dist-tags']?.latest;
+    if (typeof latest !== 'string' || latest === '') throw new Error('未返回 latest');
+    // tgz 文件名规则：取包名最后一段 + 版本号（@a/b@1.0.0 → b-1.0.0.tgz）。
+    const fileName = `${pkgName.split('/').pop()}-${latest}.tgz`;
+    return { latest, tgzUrl: `${base}/${pkgName}/-/${fileName}` };
+}
+
+/** 依次尝试所有 registry 候选；全部失败抛出聚合错误。 */
+async function fetchLatest(pkgName) {
+    const errors = [];
+    for (const base of registryCandidates()) {
+        try {
+            return await fetchPackument(base, pkgName);
+        } catch (err) {
+            errors.push(`${base}: ${err.message}`);
+        }
+    }
+    throw new Error(errors.join('；'));
+}
+
+/** 把 tgz 流式下载到本地 staging 文件。 */
+async function downloadTgz(url, destFile) {
+    const res = await fetch(url, {
+        headers: { 'user-agent': 'dsh-selfupdater' },
+        signal: AbortSignal.timeout(60000),
+    });
+    if (!res.ok) throw new Error(`tgz 下载失败 HTTP ${res.status}`);
+    writeFileSync(destFile, Buffer.from(await res.arrayBuffer()));
+}
+
+/* ------------------------------------------------------------------ *
+ * 外部命令执行
+ * ------------------------------------------------------------------ */
+
+/** 以 Promise 方式运行外部命令并收集输出。 */
+function runCommand(file, cmdArgs, opts = {}) {
+    return new Promise((resolveRun) => {
+        const child = spawn(file, cmdArgs, { stdio: ['ignore', 'pipe', 'pipe'], ...opts });
+        let stdout = '';
+        let stderr = '';
+        child.stdout?.on('data', (d) => { stdout += d; });
+        child.stderr?.on('data', (d) => { stderr += d; });
+        child.on('error', (err) => resolveRun({ code: -1, stdout: '', stderr: String(err) }));
+        child.on('close', (code) => resolveRun({ code, stdout, stderr }));
+    });
+}
+
+/**
+ * 用官方 CLI 离线安装 tgz（与 runner.js 内置安装逻辑完全一致）：
+ *   node bin/dsh.js plugin --profile <web> add <tgz>
+ * 关键环境变量：HOME 必须指向工作区，否则 dsh 会找错用户目录。
+ */
+async function dshPluginAdd(tgzFile) {
+    const result = await runCommand(NODE_BIN, [DSH_BIN, 'plugin', '--profile', PROFILE, 'add', tgzFile], {
+        cwd: workspace,
+        env: { ...process.env, HOME: workspace },
+    });
+    if (result.code !== 0) {
+        throw new Error(`dsh plugin add 退出码 ${result.code}: ${result.stderr.slice(-400)}`);
+    }
+    return result.stdout;
+}
+
+/* ------------------------------------------------------------------ *
+ * 重启链与健康检查
+ * ------------------------------------------------------------------ */
+
+/**
+ * 按 runner.js 的方式重新拉起完整启动链（runner 会再拉起 DSH）。
+ * 必须继承当前进程的全部环境变量 —— 飞牛的 TRIM_DATA_SHARE_PATHS 等
+ * 变量决定工作区落点，丢失会导致重启后配置"失踪"。
+ */
+function launchRunnerChain() {
+    const runnerJs = join(appDir, 'bin', 'runner.js');
+    if (!existsSync(runnerJs)) throw new Error(`找不到启动器 ${runnerJs}，无法重启`);
+    const child = spawn(process.execPath, [runnerJs], {
+        cwd: workspace,
+        detached: true,
+        stdio: 'ignore',
+        env: { ...process.env, TRIM_APPDEST: appDir },
+    });
+    child.unref();
+    log(`已拉起 runner 链 (pid=${child.pid})`);
+}
+
+/** 健康检查：轮询本地端口直到有任意 HTTP 响应即算活着。 */
+async function waitHealthy(timeoutMs) {
+    const deadline = Date.now() + timeoutMs;
+    while (Date.now() < deadline) {
+        try {
+            await fetch(`http://127.0.0.1:${port}/`, { signal: AbortSignal.timeout(3000) });
+            return true;
+        } catch { /* 未就绪继续等 */ }
+        await sleep(2000);
+    }
+    return false;
+}
+
+/** 失败回滚：还原整个 profile 目录并再次拉起服务。 */
+async function rollback() {
+    setState('rollback', '安装失败，正在回滚插件 profile …');
+    try {
+        rmSync(profileDir, { recursive: true, force: true });
+        renameSync(bakProfileDir, profileDir);
+    } catch (err) {
+        setState('error', `回滚失败：${err.message}。可手动从 ${bakProfileDir} 恢复。`);
+        return false;
+    }
+    launchRunnerChain();
+    const ok = await waitHealthy(60000);
+    setState(ok ? 'done_failed' : 'error',
+        ok ? '已回滚并恢复服务，插件保持旧版本' : '回滚后服务仍未恢复，请查看容器日志');
+    return ok;
+}
+
+/* ------------------------------------------------------------------ *
+ * 主流程
+ * ------------------------------------------------------------------ */
+
+async function main() {
+    mkdirSync(dshStateDir, { recursive: true });
+
+    // 锁文件双端校验：路由触发前会检查；这里再补一道防手动重复执行。
+    if (existsSync(lockFile)) throw new Error('已有一次插件更新在进行中（锁文件存在）');
+
+    // 读已安装清单：profile/package.json 的 dependencies 是最权威的数据源。
+    let deps;
+    try {
+        deps = JSON.parse(readFileSync(join(profileDir, 'package.json'), 'utf8')).dependencies ?? {};
+    } catch {
+        setState('idle', '未发现已安装插件，无需更新');
+        finish(0);
+    }
+    const installed = Object.entries(deps).map(([name, range]) => ({
+        name,
+        version: String(range).replace(/^[~^]\s*/, ''),
+    })).filter((p) => p.version !== '');
+
+    if (installed.length === 0) {
+        setState('idle', '未发现已安装插件，无需更新');
+        finish(0);
+    }
+
+    status = { startedAt: new Date().toISOString(), trigger: 'manual' };
+    setState('downloading', '正在查询各插件的最新版本 …');
+
+    // 阶段一：并发查询每个插件的 latest，筛出真正有新版可升的子集。
+    const checks = await Promise.allSettled(installed.map(async (p) => ({ ...p, info: await fetchLatest(p.name) })));
+    const pending = [];
+    for (const c of checks) {
+        if (c.status !== 'fulfilled') {
+            log(`查询 ${c.reason?.name ?? '?'} 失败: ${c.reason?.message ?? c.reason}`);
+            continue;
+        }
+        const { name, version, info } = c.value;
+        if (isNewer(info.latest, version)) pending.push({ name, currentVersion: version, targetVersion: info.latest, tgzUrl: info.tgzUrl });
+    }
+    if (pending.length === 0) {
+        setState('idle', '所有插件均已是最新版本');
+        finish(0);
+    }
+    log(`待更新插件：${pending.map((p) => `${p.name}@${p.currentVersion}->${p.targetVersion}`).join(', ')}`);
+
+    // 阶段二：备份整个 profile（同分区 rename 原子完成），失败即可整体还原。
+    setState('downloading', '正在备份当前插件目录 …');
+    rmSync(bakProfileDir, { recursive: true, force: true });
+    renameSync(profileDir, bakProfileDir);
+    renameSync(bakProfileDir, profileDir); // 立刻放回原位：备份改用复制，避免长时间缺位
+    mkdirSync(stagingDir, { recursive: true });
+
+    // 阶段三：逐个下载 tgz 并离线安装；任一失败立即整体回滚。
+    const updated = [];
+    for (const item of pending) {
+        setState('downloading', `正在下载 ${item.name}@${item.targetVersion} …`, { updating: item.name });
+        const tgzFile = join(stagingDir, `${item.name.split('/').pop()}-${item.targetVersion}.tgz`);
+        await downloadTgz(item.tgzUrl, tgzFile);
+        log(`开始安装 ${item.name}@${item.targetVersion}`);
+        await dshPluginAdd(tgzFile);
+        updated.push(item);
+    }
+
+    // 校验安装结果：profile 里的登记版本必须等于目标版本（防镜像滞后装了旧版）。
+    const afterDeps = JSON.parse(readFileSync(join(profileDir, 'package.json'), 'utf8')).dependencies ?? {};
+    const mismatched = pending.filter((p) => String(afterDeps[p.name] ?? '').replace(/^[~^]/, '') !== p.targetVersion);
+    if (mismatched.length > 0) {
+        throw new Error(`安装校验失败：${mismatched.map((p) => p.name).join(', ')} 版本不符`);
+    }
+
+    // 阶段四：重启 DSH 让新插件代码生效（本脚本启动时 DSH 已随插件路由退出，
+    // 但保险起见仍对传入 pid 做一次确认清理），然后等待端口就绪。
+    setState('restarting', '正在重启 DeepSeek Harness …');
+    const pid = parseInt(String(args.pid ?? ''), 10);
+    if (Number.isFinite(pid) && pid > 0) await killProcess(pid);
+    await waitPortFree(port, 20000);
+    launchRunnerChain();
+
+    setState('healthcheck', '等待服务就绪 …');
+    if (await waitHealthy(90000)) {
+        rmSync(stagingDir, { recursive: true, force: true });
+        // bak 目录保留一次作为手动救砖手段，下次升级前会自动清理。
+        setState('done',
+            `插件更新完成：${updated.map((p) => p.name).join(', ')}`,
+            { finishedAt: new Date().toISOString() });
+    } else {
+        await rollback();
+    }
+    finish(0);
+}
+
+// 总看门狗：任何阶段卡死都强制终态，避免 UI 永远显示"更新中"。
+setTimeout(() => {
+    setState('error', '插件更新超时（8 分钟看门狗触发），请查看 pluginupdate.log');
+    finish(2);
+}, 8 * 60 * 1000);
+
+main().catch(async (err) => {
+    log(`插件更新失败: ${err.stack ?? err}`);
+    // 只有已经动过 profile 才需要回滚；查询/下载阶段失败直接报错即可。
+    const touched = existsSync(join(dshStateDir, 'profiles')) && status.state !== 'downloading'
+        || String(status.message ?? '').includes('备份');
+    if (status.state === 'healthcheck' || status.state === 'restarting' || touched) {
+        await rollback();
+    } else {
+        setState('error', String(err.message ?? err));
+        // 若 profile 曾被挪走尚未放回，尽力恢复。
+        if (!existsSync(profileDir) && existsSync(bakProfileDir)) {
+            try { renameSync(bakProfileDir, profileDir); } catch { /* 忽略 */ }
+        }
+    }
+    finish(1);
+});