terminal-bridge-setup 2.2.0 → 2.4.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) {

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

@@ -79,6 +79,21 @@ 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() {
   // 已在运行？
@@ -120,24 +135,41 @@ async function startProxy() {
 
 function stopProxy() {
   return new Promise(async (resolve) => {
-    const pid = readPid();
+    let pid = readPid();
+    let pidSource = pid ? "pid file" : null;
+
+    // PID 文件缺失 → 回退用 lsof 按端口找
     if (!pid) {
-      // 没有 PID 文件，但端口可能在监听（手动启动的），尝试用 lsof 找
       const up = await isProxyListening();
-      resolve(up
-        ? { ok: false, status: "running", msg: "无法停止：PID 文件缺失（可能是手动启动的代理），请手动 kill" }
-        : { ok: true, status: "stopped", msg: "proxy was not running" });
-      return;
+      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;
+      }
     }
+
     try {
       process.kill(pid, "SIGTERM");
       // 等 2s 确认退出
-      setTimeout(() => {
+      setTimeout(async () => {
         if (isPidAlive(pid)) {
           try { process.kill(pid, "SIGKILL"); } catch {}
         }
         clearPid();
-        resolve({ ok: true, status: "stopped", msg: "proxy stopped" });
+        // 再确认端口确实释放了
+        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})`,
+        });
       }, 2000);
     } catch (err) {
       clearPid();

package/package.json

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