terminal-bridge-setup 2.6.1 → 2.8.0

package/bin/setup.mjs

@@ -131,16 +131,19 @@ function installProxyDeps() {
 function registerNativeHost() {
   step(3, STEPS, "注册 Native Messaging Host");
 
-  // macOS / Linux 路径不同
   const plat = platform();
+  if (plat === "win32") {
+    return registerNativeHostWindows();
+  }
+
+  // macOS / Linux 路径不同
   let destDir;
   if (plat === "darwin") {
     destDir = join(homedir(), "Library", "Application Support", "Google", "Chrome", "NativeMessagingHosts");
   } else if (plat === "linux") {
     destDir = join(homedir(), ".config", "google-chrome", "NativeMessagingHosts");
   } else {
-    console.log(c.yellow("  ⚠ Windows 平台需要手动配置 native host，跳过自动注册"));
-    console.log(c.dim("  参考：https://developer.chrome.com/docs/extensions/develop/concepts/native-messaging"));
+    console.log(c.yellow(`  ⚠ 不支持的平台 ${plat}，跳过 native host 注册`));
     return false;
   }
 
@@ -165,6 +168,55 @@ function registerNativeHost() {
   return true;
 }
 
+// Windows 注册：生成 host.bat + manifest，写注册表 HKCU
+// Chrome 在 Windows 上通过注册表找 native host：
+//   HKCU\Software\Google\Chrome\NativeMessagingHosts\com.wssniffer.host
+//   默认值 = manifest json 的绝对路径
+function registerNativeHostWindows() {
+  console.log(c.dim("  Windows 平台：写注册表方式注册"));
+
+  const nativeDir = join(INSTALL_DIR, "native");
+
+  // 1. 生成 host.bat（等价于 macOS 的 host.sh，用绝对路径调 node）
+  //    where node 拿 node 绝对路径
+  const whereRes = spawnSync("where", ["node"], { encoding: "utf8", shell: true });
+  const nodeBin = whereRes.status === 0 ? whereRes.stdout.trim().split(/\r?\n/)[0] : "node";
+  const hostBat = join(nativeDir, "host.bat");
+  // bat 里路径可能含空格（C:\Program Files\nodejs\node.exe），必须加引号
+  writeFileSync(hostBat, `@echo off\r\n"${nodeBin}" "${join(nativeDir, "host.js")}"\r\n`);
+  ok(`生成 host.bat → ${hostBat}`);
+
+  // 2. 生成 manifest json（allowed_origins 同 macOS/Linux）
+  const manifest = {
+    name: "com.wssniffer.host",
+    description: "Terminal bridge native messaging host",
+    path: hostBat,
+    type: "stdio",
+    allowed_origins: [`chrome-extension://${EXTENSION_ID}/`],
+  };
+  const manifestPath = join(nativeDir, "com.wssniffer.host.json");
+  writeFileSync(manifestPath, JSON.stringify(manifest, null, 2));
+  ok(`生成 manifest → ${manifestPath}`);
+
+  // 3. 写注册表 HKCU（用户级，不需要管理员权限）
+  const regKey = "HKCU\\Software\\Google\\Chrome\\NativeMessagingHosts\\com.wssniffer.host";
+  const regRes = spawnSync("reg", ["add", regKey, "/ve", "/t", "REG_SZ", "/d", manifestPath, "/f"],
+    { stdio: "pipe", encoding: "utf8", shell: true });
+  if (regRes.status !== 0) {
+    fail(`注册表写入失败：${regRes.stderr || regRes.stdout}`);
+  }
+  ok(`已写注册表 ${regKey}`);
+  console.log(c.dim(`  → ${manifestPath}`));
+
+  // 4. 校验注册表
+  const verifyRes = spawnSync("reg", ["query", regKey, "/ve"], { encoding: "utf8", shell: true });
+  if (verifyRes.status !== 0 || !verifyRes.stdout.includes(manifestPath)) {
+    fail("注册表校验失败");
+  }
+  ok("注册表校验通过");
+  return true;
+}
+
 // ===================== 步骤 4：安装 skill =====================
 // 把 skill 释放到 ~/.agents/skills/jumpserver-term-bridge/
 // 这样 Agent（如 ZCode）能自动发现并触发，知道怎么调用桥接、怎么处理各种错误。
@@ -211,6 +263,7 @@ function guideLoadExtension() {
 
   // ① 打开 chrome://extensions
   //    macOS: open -a 指定 Chrome；open URL scheme 更可靠
+  //    Windows: start 命令走默认关联程序（chrome:// 会由默认浏览器处理）
   //    Linux: 直接给 google-chrome 传 URL
   let chromeOpened = false;
   try {
@@ -221,6 +274,9 @@ function guideLoadExtension() {
         spawnSync("open", ["chrome://extensions"], { stdio: "ignore" });
       }
       chromeOpened = true;
+    } else if (plat === "win32") {
+      spawnSync("cmd", ["/c", "start", "", "chrome://extensions"], { stdio: "ignore", shell: true });
+      chromeOpened = true;
     } else if (plat === "linux") {
       // 试几个常见的 Chrome 命令名
       for (const bin of ["google-chrome", "google-chrome-stable", "chromium", "chromium-browser"]) {
@@ -239,6 +295,9 @@ function guideLoadExtension() {
       // open <dir> 会用 Finder 打开，并选中该文件夹
       spawnSync("open", [extDir], { stdio: "ignore" });
       ok(`已在 Finder 打开 ${extDir}`);
+    } else if (plat === "win32") {
+      spawnSync("explorer", [extDir], { stdio: "ignore" });
+      ok(`已在资源管理器打开 ${extDir}`);
     } else if (plat === "linux") {
       spawnSync("xdg-open", [extDir], { stdio: "ignore" });
       ok(`已在文件管理器打开 ${extDir}`);
@@ -301,7 +360,10 @@ async function main() {
   console.log(`  ${c.cyan("•")} 两个绿灯亮，即可通过 Agent 发命令`);
   console.log("");
   console.log(c.dim(`文件位置：${INSTALL_DIR}`));
-  console.log(c.dim(`卸载：rm -rf ${INSTALL_DIR} && rm ~/Library/Application\\ Support/Google/Chrome/NativeMessagingHosts/com.wssniffer.host.json`));
+  const uninstallHint = platform() === "win32"
+    ? `卸载：rd /s /q ${INSTALL_DIR} && reg delete HKCU\\Software\\Google\\Chrome\\NativeMessagingHosts\\com.wssniffer.host /f`
+    : `卸载：rm -rf ${INSTALL_DIR} && rm ~/Library/Application\\ Support/Google/Chrome/NativeMessagingHosts/com.wssniffer.host.json`;
+  console.log(c.dim(uninstallHint));
   console.log("");
 }
 

package/files/native/host.js

@@ -59,15 +59,13 @@ function isProxyListening() {
 
 function readPid() {
   try {
-    const pid = parseInt(fs.readFileSync(PID_FILE, "utf8").trim(), 10);
-    return isNaN(pid) ? null : pid;
+    const raw = fs.readFileSync(PID_FILE, "utf8").trim();
+    if (!/^\d{1,10}$/.test(raw)) return null;  // 纯数字白名单
+    const pid = Number(raw);
+    return pid > 0 ? pid : null;
   } catch { return null; }
 }
 
-function writePid(pid) {
-  try { fs.writeFileSync(PID_FILE, String(pid)); } catch {}
-}
-
 function clearPid() {
   try { fs.unlinkSync(PID_FILE); } catch {}
 }
@@ -79,21 +77,6 @@ function isPidAlive(pid) {
   } catch { return false; }
 }
 
-// 通过端口找占用进程的 PID（PID 文件缺失时的回退方案）
-// 用 lsof -ti :port 拿监听该端口的进程 PID
-function findPidByPort(port) {
-  return new Promise((resolve) => {
-    const child = spawn("lsof", ["-ti", `-i:${port}`, "-sTCP:LISTEN"], { stdio: ["ignore", "pipe", "ignore"] });
-    let out = "";
-    child.stdout.on("data", (d) => { out += d.toString(); });
-    child.on("error", () => resolve(null));      // lsof 不存在或没权限
-    child.on("close", () => {
-      const pid = parseInt(out.trim().split("\n")[0], 10);
-      resolve(isNaN(pid) ? null : pid);
-    });
-  });
-}
-
 // ============== 命令处理 ==============
 async function startProxy() {
   // 已在运行？
@@ -117,7 +100,8 @@ async function startProxy() {
     });
     // unref 让父进程（host）可以不等子进程就退出
     child.unref();
-    writePid(child.pid);
+    // 先落一份 spawn 返回的数字 PID（非外部输入）；代理 listening 后会自写权威值
+    try { fs.writeFileSync(PID_FILE, String(child.pid)); } catch {}
     fs.writeFileSync(logFd, `\n[host] proxy started PID=${child.pid} at ${new Date().toISOString()}\n`);
 
     // 等一小段时间确认端口起来
@@ -135,40 +119,32 @@ async function startProxy() {
 
 function stopProxy() {
   return new Promise(async (resolve) => {
-    let pid = readPid();
-    let pidSource = pid ? "pid file" : null;
-
-    // PID 文件缺失 → 回退用 lsof 按端口找
+    const pid = readPid();
     if (!pid) {
+      // 没有PID 文件。代理启动时会自己写 PID（server.js listening 后落盘），
+      // 走到这里说明代理未通过正常途径启动——用端口状态兜底判断。
       const up = await isProxyListening();
-      if (!up) {
-        resolve({ ok: true, status: "stopped", msg: "proxy was not running" });
-        return;
-      }
-      pid = await findPidByPort(PROXY_PORT);
-      pidSource = pid ? "port lookup" : null;
-      if (!pid) {
-        resolve({ ok: false, status: "running", msg: "无法停止：代理在监听但找不到 PID（lsof 失败），请手动 kill" });
-        return;
-      }
+      resolve(up
+        ? { ok: false, status: "running", msg: "无法停止：PID 文件缺失，请手动停止占用 8787 端口的进程" }
+        : { ok: true, status: "stopped", msg: "proxy was not running" });
+      return;
     }
 
     try {
       process.kill(pid, "SIGTERM");
-      // 等 2s 确认退出
+      // 等 2s 确认退出（Windows 上 SIGTERM 即强制终止）
       setTimeout(async () => {
         if (isPidAlive(pid)) {
           try { process.kill(pid, "SIGKILL"); } catch {}
         }
         clearPid();
-        // 再确认端口确实释放了
         const stillUp = await isProxyListening();
         resolve({
           ok: !stillUp,
           status: stillUp ? "running" : "stopped",
           msg: stillUp
-            ? `SIGTERM+SIGKILL 已发（PID ${pid} via ${pidSource}）但端口仍被占用，可能有其他进程`
-            : `proxy stopped (PID ${pid} via ${pidSource})`,
+            ? "已发停止信号但端口仍被占用，可能有其他进程"
+            : "proxy stopped",
         });
       }, 2000);
     } catch (err) {

package/files/proxy/server.js

@@ -13,6 +13,11 @@
 
 import { WebSocketServer } from "ws";
 import { randomBytes } from "node:crypto";
+import { writeFileSync, unlinkSync } from "node:fs";
+import { fileURLToPath } from "node:url";
+
+// PID 文件：native host「停止代理」时读取。代理自己在 listening 后写入。
+const PID_FILE = fileURLToPath(new URL("./.proxy.pid", import.meta.url));
 import {
   auditArthasCommand,
   isArthasCommand,
@@ -138,6 +143,7 @@ function maybeRunNext() {
     resolve: (result) => {
       if (pending.has(job.reqId)) {
         clearTimeout(entry.timer);
+        if (entry.weakTimer != null) { clearTimeout(entry.weakTimer); entry.weakTimer = null; }
         pending.delete(job.reqId);
         try {
           job.ws.send(JSON.stringify({ type: "result", reqId: job.reqId, ...result }));
@@ -198,7 +204,7 @@ let terminalType = "unknown";  // "unknown" | "jumpserver" | "arthas"
 
 // 各类型的 prompt 正则
 const PROMPT_RE_BY_TYPE = {
-  // JumpServer shell prompt：[user@host /dir]# 或 ]$
+  // JumpServer shell prompt（强匹配）：[user@host /dir]# 或 ]$
   // 注意：不匹配裸 >，避免输出内容里的 > 行误触发
   jumpserver: /\]\s*[#$]\s*$/,
   // Arthas prompt：arthas@pid>（带 arthas@ 前缀，不匹配裸 >）
@@ -207,16 +213,38 @@ const PROMPT_RE_BY_TYPE = {
   unknown: /\]\s*[#$]\s*$|>\s*$/,
 };
 
+// 弱 prompt（无方括号的 bash 默认 PS1）：user@host:~/path$ 或 root@host:~#
+// Ubuntu/Debian 系资产的 PS1 没有方括号，强正则永远不匹配 → 全部超时。
+// 弱匹配特征：行尾是 $ 或 #，且行内含 user@host: 或 ~/ 特征。
+// 防误判（输出行恰好以 #/$ 结尾）：不立即判定，等 WEAK_QUIET_MS 无新数据才确认。
+const WEAK_PROMPT_LINE_RE = /[$#]$/;
+const WEAK_PROMPT_CONTEXT_RE = /@[\w.-]+:|~\//;
+const WEAK_QUIET_MS = 350;
+
 // 旧名保留（cleanOutput 等处仍引用），指向宽松兜底，仅用于"删 prompt 行"的清理逻辑
 const PROMPT_RE = PROMPT_RE_BY_TYPE.unknown;
 
+// ANSI 清理（CSI + OSC），prompt 匹配统一在清理后的文本上做
+function stripAnsi(s) {
+  return s
+    .replace(/\x1b\[[0-9;?]*[ -/]*[@-~]/g, "")
+    .replace(/\x1b\][^\x07\x1b]*(\x07|\x1b\\)/g, "");
+}
+
+// 取 ANSI 清理后文本的最后一个非空行（去尾部空白），prompt 判定用
+function lastNonEmptyLine(s) {
+  const lines = stripAnsi(s).split("\n").map(l => l.replace(/\s+$/, ""));
+  for (let i = lines.length - 1; i >= 0; i--) {
+    if (lines[i].length > 0) return lines[i];
+  }
+  return "";
+}
+
 // 探测终端类型：扫描 ANSI 清理后的文本，按 prompt 特征判定
 // 支持切换：如果已锁定类型 A，但检测到明确的类型 B 特征，则切换到 B
 // （用户在 popup 切 tab 从 Arthas 切到 JumpServer 时，代理靠这条路径纠正）
 function detectTerminalType(text) {
-  const clean = text
-    .replace(/\x1b\[[0-9;?]*[ -/]*[@-~]/g, "")
-    .replace(/\x1b\][^\x07\x1b]*(\x07|\x1b\\)/g, "");
+  const clean = stripAnsi(text);
 
   // Arthas prompt 特征：arthas@<pid>> （出现在任意行尾）
   if (/arthas@\S+>\s*$/m.test(clean)) {
@@ -227,7 +255,7 @@ function detectTerminalType(text) {
     return;
   }
 
-  // JumpServer shell prompt 特征：[user@host /dir]# 或 ]$
+  // JumpServer shell prompt 特征（红帽系）：[user@host /dir]# 或 ]$
   if (/\[[^\]\n]+@[^\]\n]+\][^\n]*[#$]\s*$/m.test(clean)) {
     if (terminalType !== "jumpserver") {
       terminalType = "jumpserver";
@@ -235,6 +263,16 @@ function detectTerminalType(text) {
     }
     return;
   }
+
+  // JumpServer shell prompt 特征（Ubuntu/Debian 默认 PS1，无方括号）：
+  //   user@host:~/path$  或  root@host:~#
+  if (/[\w.-]+@[\w.-]+:\S*[$#]\s*$/m.test(clean)) {
+    if (terminalType !== "jumpserver") {
+      terminalType = "jumpserver";
+      console.log(TAG, "[probe] 终端类型切换 → jumpserver（检测到 user@host:path$ prompt，无方括号）");
+    }
+    return;
+  }
   // 未命中任一特征，保持当前类型不变
 }
 
@@ -345,6 +383,37 @@ function handleWsRecv(payload) {
           elapsedMs: Date.now() - entry.sentAt
         });
       }
+    } else if (entry.phase === 0 && (terminalType === "jumpserver" || terminalType === "unknown")) {
+      // ====== 弱 prompt 兜底（Ubuntu/Debian 默认 PS1 无方括号）======
+      // 强正则要求 ] 前缀；user@host:~/path$ 这种 prompt 永远匹配不上 → 全部超时。
+      // 弱匹配：最后一个非空行以 $/# 结尾且行内含 @ 或 : 特征。
+      // 防误判：不立即判定，等 WEAK_QUIET_MS 无新数据再确认
+      // （输出行恰好以 #/$ 结尾时，后续输出到达会取消定时器）。
+      const lastLine = lastNonEmptyLine(entry.buffer);
+      const weakHit = WEAK_PROMPT_LINE_RE.test(lastLine) && WEAK_PROMPT_CONTEXT_RE.test(lastLine);
+      if (weakHit) {
+        if (entry.weakTimer == null) {
+          const snapLen = entry.buffer.length;
+          entry.weakTimer = setTimeout(() => {
+            entry.weakTimer = null;
+            if (!pending.has(reqId) || entry.buffer.length !== snapLen) return;
+            const lineNow = lastNonEmptyLine(entry.buffer);
+            if (WEAK_PROMPT_LINE_RE.test(lineNow) && WEAK_PROMPT_CONTEXT_RE.test(lineNow)) {
+              console.log(TAG, `[${reqId}] 弱 prompt 确认完成（${WEAK_QUIET_MS}ms 静默）: ${lineNow.slice(-60)}`);
+              entry.lastScanLen = entry.buffer.length;
+              entry.resolve({
+                ok: true,
+                output: cleanOutput(entry.buffer, entry.cmd),
+                elapsedMs: Date.now() - entry.sentAt
+              });
+            }
+          }, WEAK_QUIET_MS);
+        }
+      } else if (entry.weakTimer != null) {
+        // 新数据不再是弱 prompt 形态，取消待确认的判定
+        clearTimeout(entry.weakTimer);
+        entry.weakTimer = null;
+      }
     }
   }
 }
@@ -393,11 +462,14 @@ function cleanOutput(text, cmd) {
     .replace(/\r\n?/g, "\n");
 
   // 删除 prompt 行：
-  //   - shell 风格：行里出现 [user@host /cwd]# 或 ]$ → 整行删
+  //   - shell 风格（红帽系）：行里出现 [user@host /cwd]# 或 ]$ → 整行删
   //     （删 "[root@host ~]# echo hello" 这种命令回显行，含粘在前面的碎片）
+  //   - shell 风格（Ubuntu/Debian，无方括号）：user@host:~/path$ 结尾的行
   //   - Arthas 风格：行尾是 arthas@xxx> 或单独的 > → 整行删
   out = out.split("\n").filter(line => {
     if (/\[[^\]\n]+@[^\]\n]+\][^\n]*[#$]/.test(line)) return false;
+    // Ubuntu prompt 行（含粘在命令回显前的情况）
+    if (/[\w.-]+@[\w.-]+:\S*[$#]\s*$/.test(line)) return false;
     // Arthas / 通用 REPL prompt：行尾 > （允许前面有 arthas@xxx 等前缀）
     if (/>\s*$/.test(line) && line.trim().length <= 60) return false;
     return true;
@@ -515,10 +587,19 @@ wss.on("listening", () => {
   const { address, port } = wss.address();
   console.log(TAG, `监听 ws://${address}:${port}/ssh`);
   console.log(TAG, "等待插件连接（role: extension）和 Agent 连接（发 run 请求）");
+
+  // 写 PID 文件：native host 的「停止代理」靠它找到本进程。
+  // 由代理自己写（而不是 host 代写）保证准确性——手动 node server.js
+  // 启动时也有 PID 文件可停。进程自己写自己的 process.pid，无注入面。
+  try {
+    writeFileSync(PID_FILE, String(process.pid));
+    console.log(TAG, `PID ${process.pid} → ${PID_FILE}`);
+  } catch {}
 });
 
 function shutdown() {
   console.error(TAG, "shutting down");
+  try { unlinkSync(PID_FILE); } catch {}
   wss.clients.forEach((c) => c.close(1001, "shutting down"));
   for (const [, entry] of pending) {
     clearTimeout(entry.timer);

package/files/skill/SKILL.md

@@ -78,8 +78,8 @@ Arthas 的能力不止于读，以下命令/用法一律禁止通过桥接执行
 | 页面形态 | koko connect **iframe**（嵌在 luna 父页里） | **顶层文档**（无 iframe） |
 | WebSocket URL | `wss://.../koko/ws/...` | `wss://.../ws?method=connectArthas...` |
 | WS 帧格式 | **二进制帧 opcode=2**，payload 是 base64，解码后是 SSH PTY 明文 | **文本帧 opcode=1**，payload 直接是明文 ANSI |
-| prompt 样式 | shell 风格 `[root@host /dir]#` | Arthas 风格 `arthas@pid>` 或 `[arthas@...]` |
-| prompt 锚点正则 | `/\]\s*[#$]\s*$/` | `/>/（行尾）` |
+| prompt 样式 | 红帽系 `[root@host /dir]#`；Ubuntu 资产 `user@host:~/path$`（无方括号） | Arthas 风格 `arthas@pid>` 或 `[arthas@...]` |
+| prompt 锚点 | 强匹配 `]\s*[#$]\s*$`；无方括号 prompt 走弱兜底（行尾 `$`/`#` + `user@host:` 特征，350ms 静默确认） | `arthas@\S+>\s*$` |
 | sudo 检测 | 需要（cat/df 等可能被 alias 成 sudo） | 不适用（Java 诊断工具无 sudo 概念） |
 
 **桥接层已自动处理这些差异**：content script 按是否有 `.xterm` 元素识别终端 frame（不依赖 URL），proxy 按 opcode 自动决定是否 base64 解码，prompt 锚点同时匹配两种风格。Agent 侧调用方式完全一致——都是发 `run` 帧、收 `result` 帧。

package/files/skill/references/protocol.md

@@ -129,8 +129,11 @@ const PROMPT_RE = /\]\s*[#$]\s*$|>\s*$/;
 
 | 终端 | prompt 样式 | 匹配部分 |
 |------|------------|---------|
-| JumpServer (shell) | `[root@host /path]#` 或 `]$` | `]\s*[#$]\s*$` |
-| Arthas | `arthas@pid>` 或 `[arthas@...]` | `>\s*$` |
+| JumpServer (shell, 红帽系) | `[root@host /path]#` 或 `]$` | 强匹配 `]\s*[#$]\s*$`（命中即完成） |
+| JumpServer (shell, Ubuntu/Debian) | `user@host:~/path$`（无方括号） | 弱匹配兜底：行尾 `$`/`#` + `user@host:`/`~/` 特征，需 350ms 静默确认 |
+| Arthas | `arthas@pid>` 或 `[arthas@...]` | `arthas@\S+>\s*$` |
+
+说明：Ubuntu/Debian 默认 PS1 没有方括号，强正则永远不命中会导致所有命令超时（曾出现在 Windows 同事的 Ubuntu 资产上）。弱兜底以 350ms 无新数据为确认条件，避免输出行恰好以 `#`/`$` 结尾时误判提前完成。
 
 注意：单独的 `>` 较宽（命令输出里 `>` 偶尔出现），但配合"行尾 + ANSI 清理后 + 注入命令后才出现"三个条件，误判率可接受。
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "terminal-bridge-setup",
-  "version": "2.6.1",
+  "version": "2.8.0",
   "description": "一次性安装器：释放终端桥接（JumpServer / Arthas）的本地代理 + Chrome 插件，并注册 native messaging host。让 Agent 能通过浏览器 xterm 终端执行命令并拿回输出。",
   "license": "MIT",
   "author": "encorearon",