terminal-bridge-setup 2.3.0 → 2.5.0

package/files/extension/background.js

@@ -288,7 +288,9 @@ function connectBridge() {
 
 // ============== 手动扫描 xterm（popup 触发）==============
 // 遍历所有 tab 的所有 frame，发 term-ping 探测有没有 xterm。
-// 找到就登记到 termReadyTabs 并 attach debugger，免去刷新页面的麻烦。
+// 关键：reload 插件后已打开的页面里没有 content script，ping 无人响应。
+// 所以 ping 失败时用 chrome.scripting 主动注入 content.js，再 ping 一次。
+// 这样用户 reload 插件后不用刷新页面，点「捕捉终端」就能自动恢复。
 async function scanAllTabsForXterm() {
   const tabs = await chrome.tabs.query({});
   let found = 0;
@@ -306,26 +308,25 @@ async function scanAllTabsForXterm() {
     } catch { continue; }
     if (!frames) continue;
 
-    // 对每个 frame 发 term-ping（必须指定 frameId 才能到 iframe）
     for (const frame of frames) {
-      const res = await new Promise((resolve) => {
-        chrome.tabs.sendMessage(
-          tab.id,
-          { type: "term-ping" },
-          { frameId: frame.frameId },
-          (r) => {
-            if (chrome.runtime.lastError) { resolve(null); return; }
-            resolve(r);
-          }
-        );
-      });
+      // 第一次 ping（content script 可能已经在跑）
+      let res = await pingFrame(tab.id, frame.frameId);
+
+      // ping 失败：content script 没注入（reload 插件后常见）→ 主动注入再 ping
+      if (!res) {
+        console.log("[bg] ping 失败，尝试注入 content.js 到 tab", tab.id, "frame", frame.frameId);
+        await injectContentScript(tab.id, frame.frameId);
+        // 注入后等一下让 IIFE 执行 + term-ping 监听器注册
+        await new Promise(r => setTimeout(r, 200));
+        res = await pingFrame(tab.id, frame.frameId);
+      }
+
       if (res && res.ready) {
         found++;
         termReadyTabs.add(tab.id);
-        termReadyFrames.set(tab.id, frame.frameId);  // 记下 frame，注入时直接用
+        termReadyFrames.set(tab.id, frame.frameId);
         foundTabs.push({ tabId: tab.id, href: res.href });
         console.log("[bg] 扫描发现 xterm: tab", tab.id, "frame", frame.frameId, res.href);
-        // 自动 attach
         attachDebugger(tab.id);
         break;  // 一个 tab 找到一个就够
       }
@@ -341,6 +342,39 @@ async function scanAllTabsForXterm() {
   };
 }
 
+// 向指定 frame 发 term-ping
+function pingFrame(tabId, frameId) {
+  return new Promise((resolve) => {
+    try {
+      chrome.tabs.sendMessage(
+        tabId,
+        { type: "term-ping" },
+        { frameId },
+        (r) => {
+          if (chrome.runtime.lastError) { resolve(null); return; }
+          resolve(r);
+        }
+      );
+    } catch { resolve(null); }
+  });
+}
+
+// 用 chrome.scripting 主动注入 content.js 到指定 frame
+// 解决 reload 插件后已打开页面没有 content script 的问题
+async function injectContentScript(tabId, frameId) {
+  try {
+    await chrome.scripting.executeScript({
+      target: { tabId, frameIds: [frameId] },
+      files: ["content.js"],
+    });
+    return true;
+  } catch (err) {
+    // 有些 frame（如 about:blank）注入会失败，忽略
+    console.log("[bg] 注入 content.js 失败 tab", tabId, "frame", frameId, ":", err.message);
+    return false;
+  }
+}
+
 function sendToBridge(obj) {
   if (bridgeWs && bridgeWs.readyState === WebSocket.OPEN) {
     try { bridgeWs.send(JSON.stringify(obj)); } catch {}
@@ -401,15 +435,35 @@ async function injectToXterm(text, reqId) {
       return;
     }
 
+    // 第一轮：ping 现有 content script
     for (const frame of frames) {
       const r = await sendToFrame(tabId, frame.frameId, text, reqId, true /*pingOnly*/);
       if (r) {
         targetFrameId = frame.frameId;
-        termReadyFrames.set(tabId, targetFrameId); // 记下来，下次直接用
+        termReadyFrames.set(tabId, targetFrameId);
         ok = await sendToFrame(tabId, targetFrameId, text, reqId);
         break;
       }
     }
+
+    // 第二轮：第一轮全失败说明 content script 没注入（reload 插件后常见）
+    // 主动注入 content.js 到每个 frame，再 ping + 注入
+    if (!ok) {
+      console.log("[bg] 降级探测全失败，尝试主动注入 content.js");
+      for (const frame of frames) {
+        await injectContentScript(tabId, frame.frameId);
+      }
+      await new Promise(r => setTimeout(r, 200));  // 等 IIFE 执行
+      for (const frame of frames) {
+        const r = await sendToFrame(tabId, frame.frameId, text, reqId, true /*pingOnly*/);
+        if (r) {
+          targetFrameId = frame.frameId;
+          termReadyFrames.set(tabId, targetFrameId);
+          ok = await sendToFrame(tabId, targetFrameId, text, reqId);
+          break;
+        }
+      }
+    }
   }
 
   if (ok) {
@@ -502,4 +556,4 @@ chrome.alarms.onAlarm.addListener(() => {
 
 // ============== 启动 ==============
 connectBridge();
-console.log('[WS Sniffer] background 已启动');
+console.log('[Terminal Bridge] background 已启动');

package/files/extension/content.js

@@ -1,4 +1,4 @@
-// WS Sniffer - content script (ISOLATED world).
+// Terminal Bridge - content script (ISOLATED world).
 //
 // 注：manifest 用 matches: <all_urls> + all_frames:true + run_at:document_start
 // 注入，是为了确保能进终端 frame（koko connect iframe / Arthas console 顶层文档等）。
@@ -13,10 +13,10 @@
 // observer 只监听 documentElement 的 childList，开销可忽略。
 
 (function () {
-  const TAG = "[ws-sniffer-cs]";
+  const TAG = "[terminal-bridge-cs]";
 
-  if (window.__wsSnifferContentLoaded) return;
-  window.__wsSnifferContentLoaded = true;
+  if (window.__terminalBridgeContentLoaded) return;
+  window.__terminalBridgeContentLoaded = true;
 
   // ---------- xterm 定位 ----------
   function findXtermTextarea() {

package/files/extension/manifest.json

@@ -4,7 +4,7 @@
   "version": "2.0.0",
   "description": "桥接 JumpServer Web 终端，让 Agent 能远程执行命令并获取输出",
   "key": "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAtsnqR6PcFUueZwYria79tVbstvjk+tM7PpvIXILm5xbd6bAdjDIhzg3lsnKioVfvxjfvT+s6vJsiOYa9ojVZyJMFc5m/05TYqr770ovYwQmz0e88fmiy6dUoSulbtKvBCSLbN6OOL7u+ul8ixLZ/HautxSmou/eNgAFPmhE+4UueE7wfCqcgMYvjLvEzlqTVumMW+5LKw9YsRk6WhHPghY1a3MVUn3eQOWXBtQTEUy3wBM3v4wHxLwDeinVOR4f/P87IlUNo84C5DeimoFit0qCj3K04hS8MIYCLCYZc3v9ftRJDJBkAoah6Eaqj7JajbS3KvLR1ctOYfDhubgmURQIDAQAB",
-  "permissions": ["debugger", "tabs", "alarms", "webNavigation", "nativeMessaging"],
+  "permissions": ["debugger", "tabs", "alarms", "webNavigation", "nativeMessaging", "scripting"],
   "background": {
     "service_worker": "background.js"
   },

package/files/native/host.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-// WS Sniffer Native Messaging Host
+// Terminal Bridge Native Messaging Host
 //
 // Chrome 通过 native messaging 和这个脚本通信（stdin/stdout，4 字节小端长度前缀）。
 // 职责：接收插件命令，启动/停止/查询 proxy/server.js。

package/files/proxy/package.json

@@ -1,9 +1,9 @@
 {
-  "name": "ws-sniffer-proxy",
+  "name": "terminal-bridge-proxy",
   "version": "1.0.0",
   "private": true,
   "type": "module",
-  "description": "JumpServer xterm 桥接代理：Agent <-> Chrome 插件，请求-响应配对",
+  "description": "Terminal Bridge 代理：Agent <-> Chrome 插件，请求-响应配对",
   "main": "server.js",
   "scripts": {
     "start": "node server.js",

package/files/proxy/server.js

@@ -120,7 +120,7 @@ function maybeRunNext() {
   // kitty 终端逐字符注入会重绘，把任何标记字符串（BEGIN/END/哨兵）打散，
   // 标记方案不可靠。改用 shell prompt 作为命令完成的锚点（expect/pexpect 经典做法）。
   //
-  // 你的 PS1=[\u@\h \w]\$ 渲染成 [root@k8s-master-test /home/operation]#
+  // 典型 shell prompt: [root@host /path]# 或 [user@host ~]$
   // 关键：prompt 是服务端 shell 在命令完成后输出的，不受 kitty 输入重绘影响。
   //
   // 流程：

package/files/skill/SKILL.md

@@ -91,17 +91,17 @@ Arthas 的能力不止于读，以下命令/用法一律禁止通过桥接执行
 lsof -i:8787 | grep LISTEN
 
 # 2. 没跑就启动（也可从插件 popup 的"启动代理"按钮启动）
-cd ~/Code/testwork/ws-sniffer/proxy && node server.js &
+cd ~/.terminal-bridge/proxy && node server.js &
 ```
 
 浏览器侧（任选其一或都开）：
-- **JumpServer**：WS Sniffer 插件已加载 + JumpServer 终端页面已打开 + 已连上一个资产（终端可见、能敲字）
-- **Arthas**：WS Sniffer 插件已加载 + Arthas Console 页面已打开 + 已 Connect 上目标 JVM
+- **JumpServer**：Terminal Bridge 插件已加载 + JumpServer 终端页面已打开 + 已连上一个资产（终端可见、能敲字）
+- **Arthas**：Terminal Bridge 插件已加载 + Arthas Console 页面已打开 + 已 Connect 上目标 JVM
 
 多终端场景：如果同时开了 JumpServer 和 Arthas 两个 tab，代理命令只会发给 popup 里"当前选中"的那个 tab。切换用 popup 的 tab 选择器，或让用户在 popup 点选。
 
 如果用户说"命令没反应"或"inject-failed"：
-1. `chrome://extensions/` 刷新 WS Sniffer 插件 ↻
+1. `chrome://extensions/` 刷新 Terminal Bridge 插件 ↻
 2. 让用户在终端页面按 F5 刷新（让 content script 重新识别 xterm）
 3. 让用户在 popup 点"捕捉 xterm"按钮（手动扫描，免刷新）
 4. 确认终端 tab 顶部有黄色调试条（说明 CDP 已 attach）
@@ -113,7 +113,7 @@ cd ~/Code/testwork/ws-sniffer/proxy && node server.js &
 最简方式（用示例客户端）：
 
 ```bash
-cd ~/Code/testwork/ws-sniffer/proxy
+cd ~/.terminal-bridge/proxy
 
 # JumpServer 场景
 node client-example.mjs "uname -a"

package/files/skill/references/protocol.md

@@ -129,7 +129,7 @@ const PROMPT_RE = /\]\s*[#$]\s*$|>\s*$/;
 
 | 终端 | prompt 样式 | 匹配部分 |
 |------|------------|---------|
-| JumpServer (shell) | `[root@k8s-master /home/op]#` 或 `]$` | `]\s*[#$]\s*$` |
+| JumpServer (shell) | `[root@host /path]#` 或 `]$` | `]\s*[#$]\s*$` |
 | Arthas | `arthas@pid>` 或 `[arthas@...]` | `>\s*$` |
 
 注意：单独的 `>` 较宽（命令输出里 `>` 偶尔出现），但配合"行尾 + ANSI 清理后 + 注入命令后才出现"三个条件，误判率可接受。

package/package.json

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