fde2e-bu 1.2.0 → 1.3.1

package/src/nm/host.mjs

@@ -1,25 +1,42 @@
-// Native Messaging host（change bu-bind-http-publish A1/A2）——Chrome NM 帧协议
-// 与 bu:pair 配对消息处理。隐蔽子命令 `fde2e-bu __nm-host` 直通本模块 runNmHost。
+// Native Messaging host（change bu-bind-http-publish A1/A2；bu-online-selfupdate A2
+// 增 bu:update；bu-chrome154-downloads B1 增 bu:asset-fetch）——Chrome NM 帧协议与
+// bu:pair/bu:update/bu:asset-fetch 消息处理。隐蔽子命令
+// `fde2e-bu __nm-host` 直通本模块 runNmHost。
 //
 // 帧协议（Chrome Native Messaging 规范）：
 //   4 字节小端长度前缀 + UTF-8 JSON 载荷；单帧上限 1MB（超限回错误帧，不崩不退出
 //   ——长度前缀流无法重同步，按「4+声明长度」流式丢弃该帧字节后继续解后续帧）。
-// 消息（扩展 → CLI，唯一受支持类型）：
+// 消息（扩展 → CLI）：
 //   {type:'bu:pair', version:'<manifest version>',
 //    bindings:[{profileId, role:'master'|'cluster', name?, server?}]}
 //   （bindings 来自扩展绑定表快照；空数组合法——照常落盘并如实显示 0 条。）
+//   {type:'bu:update', feedVersion, url, sha256, size, runningVersion}
+//   （bu-online-selfupdate design C4：SW 判定有新版后请 host 定位 unpacked 安装并
+//   下载校验原子替换；url 相对形态按官方 downloads 基址解析，域名硬白名单。）
+//   {type:'bu:asset-fetch', url, name, maxBytes, allowHosts}
+//   （bu-chrome154-downloads design §A：Chrome 154 chrome.downloads 即断——SW 配图
+//   下载改 host 代取：逐跳白名单流式下载到 ~/fde2e/bu/assets/，回绝对路径＋sha256。）
 // 应答（CLI → 扩展）：
 //   成功 {type:'bu:pair:ok', cli:'fde2e-bu', version:<pkg>, saved:N}
+//        {type:'bu:update:ok', from, to, paths:[…]}（多命中全部应用）
+//        {type:'bu:asset-fetch:ok', path, bytes, finalUrl, sha256}
 //   失败 {type:'bu:pair:error', message:<原因>}（未知类型/校验不过/落盘失败）
+//        {type:'bu:update:error', code, message, paths?}
+//        （code ∈ managed|not-found|verify|bad-url|io|bad-msg——面板按码映射文案）
+//        {type:'bu:asset-fetch:error', code, message}
+//        （code ∈ bad-msg|bad-url|not-allowed|too-large|net|io——见 nm/asset.mjs）
 // 落盘：pairing.json（{version, savedAt, bindings}，atomicWrite 0600——复用
-// src/mcp/config.mjs 的原子写底座）。pairing 只含 profileId/role/name/server 等
-// 非敏感标识，绝不含 token/credential。
+//   src/mcp/config.mjs 的原子写底座）。pairing 只含 profileId/role/name/server 等
+//   非敏感标识，绝不含 token/credential。资产代取落盘＝~/fde2e/bu/assets/（nm/asset.mjs）。
 // 生命周期：stdin EOF → 退出 0；一切日志只走 stderr（stdout 只有协议帧）。
 
 import { existsSync, readFileSync } from 'node:fs';
+import { homedir } from 'node:os';
 import { join } from 'node:path';
 import { atomicWrite } from '../mcp/config.mjs';
 import { resolveConfigDir } from '../configdir.mjs';
+import { locateInstalls, applyUpdate, EXTENSION_ID, resolveNmUrl } from './update.mjs';
+import { assetFetchHost } from './asset.mjs';
 import pkg from '../../package.json' with { type: 'json' };
 
 export const NM_HOST_NAME = 'com.fde2e.bu';
@@ -166,6 +183,112 @@ export function validatePairMessage(msg) {
   return { ok: true, bindings };
 }
 
+// ---------- bu:update 校验与处理（bu-online-selfupdate A2，design C4） ----------
+
+/**
+ * 校验 bu:update 消息形状：{type, feedVersion, url, sha256, size, runningVersion}
+ * 六字段齐备（url/sha256/size 后续在下载层再深校验——此处先拒结构性畸形）。
+ * @returns {{ ok:true } | { ok:false, message:string }}
+ */
+export function validateUpdateMessage(msg) {
+  if (msg === null || typeof msg !== 'object' || Array.isArray(msg)) {
+    return { ok: false, message: '消息须为 JSON 对象' };
+  }
+  if (msg.type !== 'bu:update') {
+    return { ok: false, message: `未知消息类型：${JSON.stringify(msg.type ?? null)}` };
+  }
+  for (const k of ['feedVersion', 'url', 'runningVersion']) {
+    if (typeof msg[k] !== 'string' || msg[k] === '') {
+      return { ok: false, message: `bu:update.${k} 须为非空字符串（收到 ${JSON.stringify(msg[k] ?? null)}）` };
+    }
+  }
+  if (typeof msg.sha256 !== 'string' || !/^[0-9a-fA-F]{64}$/.test(msg.sha256)) {
+    return { ok: false, message: 'bu:update.sha256 须为 64 位 hex 字符串' };
+  }
+  if (!Number.isSafeInteger(msg.size) || msg.size <= 0) {
+    return { ok: false, message: `bu:update.size 须为正整数（收到 ${JSON.stringify(msg.size ?? null)}）` };
+  }
+  return { ok: true };
+}
+
+function updateErrorReply(code, message, paths) {
+  return paths === undefined
+    ? { type: 'bu:update:error', code, message }
+    : { type: 'bu:update:error', code, message, paths };
+}
+
+/**
+ * 处理 bu:update（async——下载/替换是异步动作）：locateInstalls 定位 → 受管拒更 →
+ * 多命中全部应用（同一 zip 逐一 applyUpdate；任一失败聚合 error，已成功的不回滚、
+ * paths 如实全列）。依赖全注入可测（home/platform/fetch/sleep/env）。
+ * @param {{ home?: string, platform?: string, extId?: string,
+ *           fetchImpl?: Function, sleepImpl?: Function, env?: object }} [deps]
+ * @returns {Promise<{type:'bu:update:ok', from, to, paths}
+ *                 |{type:'bu:update:error', code, message, paths?}>}
+ */
+export async function handleUpdateMessage(msg, deps = {}) {
+  const check = validateUpdateMessage(msg);
+  if (!check.ok) return updateErrorReply('bad-msg', check.message);
+  try {
+    resolveNmUrl(msg.url, deps.env ?? process.env); // 白名单前置：坏 url 先拒（省一次定位）
+  } catch (err) {
+    return updateErrorReply('bad-url', err?.message ?? 'url 非法');
+  }
+  let located;
+  try {
+    located = locateInstalls({
+      home: deps.home ?? homedir(),
+      platform: deps.platform,
+      extId: deps.extId ?? EXTENSION_ID,
+      runningVersion: msg.runningVersion,
+    });
+  } catch (err) {
+    return updateErrorReply('io', `安装定位失败：${err?.message ?? err}`);
+  }
+  if (located.matches.length === 0) {
+    if (located.managed.length > 0) {
+      // 全部命中都是受管安装 → 让路 omni App（proposal 硬约束 1）
+      return updateErrorReply('managed', '安装由 FDE2E Omni 桌面版管理，插件随 App 更新', located.managed.map((m) => m.path));
+    }
+    const clue = located.rejected.length > 0
+      ? `（发现 ${located.rejected.length} 处版本不符的旧拷贝：${located.rejected.map((r) => `${r.path}（v${r.found}≠v${r.expected}）`).join('；')}）`
+      : '';
+    return updateErrorReply('not-found', `未发现运行 v${msg.runningVersion} 的 unpacked 安装${clue}`);
+  }
+  const results = [];
+  const failures = [];
+  for (const m of located.matches) {
+    try {
+      results.push(await applyUpdate({
+        url: msg.url, sha256: msg.sha256, size: msg.size, destDir: m.path,
+        ...(deps.fetchImpl ? { fetchImpl: deps.fetchImpl } : {}),
+        ...(deps.sleepImpl ? { sleepImpl: deps.sleepImpl } : {}),
+        ...(deps.env ? { env: deps.env } : {}),
+      }));
+    } catch (err) {
+      failures.push({ path: m.path, code: ['verify', 'bad-url', 'io'].includes(err?.code) ? err.code : 'io', message: err?.message ?? String(err) });
+    }
+  }
+  if (failures.length > 0) {
+    // 聚合报错但已成功的不回滚：paths 如实列出全部（成功在前、失败在后）
+    return updateErrorReply(
+      failures[0].code,
+      `已应用 ${results.length}/${located.matches.length}，失败：${failures.map((f) => `${f.path}（${f.code}：${f.message}）`).join('；')}`,
+      [...results.map((r) => r.path), ...failures.map((f) => f.path)],
+    );
+  }
+  return {
+    type: 'bu:update:ok',
+    from: results[0]?.from ?? msg.runningVersion,
+    to: results[0]?.to ?? msg.feedVersion,
+    paths: results.map((r) => r.path),
+    // 部分受管时如实让面板知情（对抗审查 P2.4——静默跳过会让「全成功」失真）
+    ...(located.managed.length > 0
+      ? { skippedManaged: located.managed.map((m) => m.path) }
+      : {}),
+  };
+}
+
 // ---------- 消息处理（依赖注入，可测） ----------
 
 function errorReply(message) {
@@ -174,17 +297,36 @@ function errorReply(message) {
 
 /**
  * 单条 NM 消息 → 应答对象（不碰流；落盘经 deps 注入，纯逻辑可测）。
+ * bu:update 走异步 handleUpdateMessage、bu:asset-fetch 走异步 assetFetchHost
+ * （返回 Promise——runNmHost 的写队列统一序列化应答帧）；其余（bu:pair/未知类型）
+ * 同步应答，行为不变。
  * @param {unknown} msg 解析后的 JSON 值
- * @param {{ version?: string, now?: () => string, save?: (pairing:object) => void }} [deps]
+ * @param {{ version?: string, now?: () => string, save?: (pairing:object) => void,
+ *           home?: string, platform?: string, extId?: string,
+ *           fetchImpl?: Function, sleepImpl?: Function, env?: object,
+ *           fs?: object, sweepTtlMs?: number }} [deps]
  *   save 注入落盘动作；缺省＝写 resolveConfigDir() 下 pairing.json（调用时才解析
- *   env/家目录——测试可用 BU_CONFIG_DIR 重定向）。
+ *   env/家目录——测试可用 BU_CONFIG_DIR 重定向）。home/platform/fetchImpl/
+ *   sleepImpl/env 属 bu:update 通道（定位与下载注入）；home/fs/fetchImpl/now/
+ *   sweepTtlMs 属 bu:asset-fetch 通道（资产目录与下载注入）。
  * @returns {{type:'bu:pair:ok', cli:string, version:string, saved:number,
  *            received:number, truncated:boolean}
- *          |{type:'bu:pair:error', message:string}}
+ *          |{type:'bu:pair:error', message:string}
+ *          |Promise<{type:'bu:update:ok'|'bu:update:error'|'bu:asset-fetch:ok'
+ *                   |'bu:asset-fetch:error', …}>}
  *   P1-2：received＝收到的原始条数、truncated＝扩展侧是否声明截断——截断要如实
  *   回执给 UI 显示「K 条未同步」，不再只报 saved 装作全量。
  */
 export function handleNmMessage(msg, deps = {}) {
+  // bu:update 分流（异步通道）：形状校验/定位/下载全在 handleUpdateMessage 内
+  if (msg !== null && typeof msg === 'object' && !Array.isArray(msg) && msg.type === 'bu:update') {
+    return handleUpdateMessage(msg, deps);
+  }
+  // bu:asset-fetch 分流（异步通道，bu-chrome154-downloads B1）：校验/逐跳白名单
+  // 下载/原子落盘全在 assetFetchHost 内；bu:pair/bu:update 行为零变更
+  if (msg !== null && typeof msg === 'object' && !Array.isArray(msg) && msg.type === 'bu:asset-fetch') {
+    return assetFetchHost(msg, deps);
+  }
   const version = deps.version ?? pkg.version;
   const check = validatePairMessage(msg);
   if (!check.ok) return errorReply(check.message);
@@ -235,6 +377,21 @@ export function runNmHost({ stdin = process.stdin, stdout = process.stdout, stde
         stderr.write(`nm-host: 应答帧写出失败：${err?.message ?? err}\n`);
       }
     };
+    // A2（bu-online-selfupdate）：应答帧统一经写队列串行化——bu:update 的应答是
+    // 异步产物（下载/替换），与后续消息的同步应答必须保序；同步应答只多等一个
+    // microtask，对端无感。处理异常（含 Promise 拒绝）兜底回 bu:pair:error 帧。
+    let writeChain = Promise.resolve();
+    const enqueueReply = (produce) => {
+      writeChain = writeChain.then(
+        () => Promise.resolve().then(produce).then(writeFrame, (err) => {
+          stderr.write(`nm-host: 消息处理异常：${err?.message ?? err}\n`);
+          writeFrame(errorReply(`处理异常：${err?.message ?? err}`));
+        }),
+        (err) => {
+          stderr.write(`nm-host: 应答写出失败：${err?.message ?? err}\n`);
+        },
+      );
+    };
     let buf = Buffer.alloc(0);
     // P2-1：超限帧流式丢弃——不清空整个缓冲（那会把同批到达的后续合法帧一起杀
     // 掉），改为按「4+声明长度」记账，随到达字节扣减；扣完即恢复常规解帧。
@@ -258,7 +415,7 @@ export function runNmHost({ stdin = process.stdin, stdout = process.stdout, stde
         const r = decodeFrame(buf);
         if (r.error === 'too-large') {
           stderr.write(`nm-host: 帧超限（声明 ${r.declared}B > ${MAX_FRAME_BYTES}B）——流式丢弃 ${4 + r.declared}B\n`);
-          writeFrame(errorReply(`帧超限：声明 ${r.declared}B > 上限 ${MAX_FRAME_BYTES}B`));
+          enqueueReply(() => errorReply(`帧超限：声明 ${r.declared}B > 上限 ${MAX_FRAME_BYTES}B`));
           const total = 4 + r.declared;
           const take = Math.min(buf.length, total);
           discard = total - take;
@@ -273,15 +430,10 @@ export function runNmHost({ stdin = process.stdin, stdout = process.stdout, stde
           msg = JSON.parse(r.payload);
         } catch {
           stderr.write('nm-host: 帧载荷不是合法 JSON\n');
-          writeFrame(errorReply('帧载荷不是合法 JSON'));
+          enqueueReply(() => errorReply('帧载荷不是合法 JSON'));
           continue;
         }
-        try {
-          writeFrame(handleNmMessage(msg, deps));
-        } catch (err) {
-          stderr.write(`nm-host: 消息处理异常：${err?.message ?? err}\n`);
-          writeFrame(errorReply(`处理异常：${err?.message ?? err}`));
-        }
+        enqueueReply(() => handleNmMessage(msg, deps));
       }
     });
     stdin.on('end', () => finish(0)); // 客户端断开（Chrome 端 port.disconnect）→ 干净退出 0