comate-ui-inspector 0.1.0 → 0.1.1

package/cli.mjs

@@ -34,7 +34,11 @@ function nmDirs() {
                   'Library/Application Support/Microsoft Edge/NativeMessagingHosts',
                   'Library/Application Support/BraveSoftware/Brave-Browser/NativeMessagingHosts',
               ]
-            : ['.config/google-chrome/NativeMessagingHosts', '.config/microsoft-edge/NativeMessagingHosts', '.config/chromium/NativeMessagingHosts'];
+            : [
+                  '.config/google-chrome/NativeMessagingHosts',
+                  '.config/microsoft-edge/NativeMessagingHosts',
+                  '.config/chromium/NativeMessagingHosts',
+              ];
     return rels.map(rel => join(home, rel));
 }
 
@@ -51,10 +55,15 @@ function out(text) {
 }
 function install() {
     chmodSync(hostPath, 0o755);
+    // 用壳脚本硬编码 node 绝对路径：Chrome 拉起 native host 时 PATH 极简（不含 nvm 等目录），
+    // 直接 `#!/usr/bin/env node` 会找不到 node 导致 host 起不来；壳脚本规避这点。
+    const wrapperPath = resolve(pkgRoot, 'extension/native-host/host-wrapper.sh');
+    writeFileSync(wrapperPath, `#!/bin/sh\nexec "${process.execPath}" "${hostPath}" "$@"\n`);
+    chmodSync(wrapperPath, 0o755);
     const manifest = {
         name: HOST_NAME,
         description: 'ui-inspector native messaging host',
-        path: hostPath,
+        path: wrapperPath,
         type: 'stdio',
         allowed_origins: [`chrome-extension://${EXT_ID}/`],
     };
@@ -97,6 +106,10 @@ function uninstall({purge}) {
             out(`removed host manifest -> ${file}`);
         }
     }
+    const wrapperPath = resolve(pkgRoot, 'extension/native-host/host-wrapper.sh');
+    if (existsSync(wrapperPath)) {
+        rmSync(wrapperPath);
+    }
     const config = readHooks();
     if (config?.hooks?.UserPromptSubmit) {
         const before = config.hooks.UserPromptSubmit.length;
@@ -151,4 +164,3 @@ if (cmd === 'install') {
 } else {
     help();
 }
-

package/extension/bridge-content.js

@@ -1,29 +1,49 @@
 /*
  * ISOLATED world 内容脚本：MAIN world 与 background 之间的中继。
  *
- * - 收到 MAIN 发来的 selection（window.postMessage）→ 转给 background 写入本地桥；
- * - 每 1.5s 问 background 要一次 /state 的 count；count 为 0（AI 调 MCP 清空后）
+ * - 收到 MAIN 发来的 selection（window.postMessage）→ 转给 background 写入本地 host；
+ * - 每 1.5s 问 background 要一次 state 的 count；count 为 0（AI 调 MCP 清空后）
  *   → 回发 cleared 给 MAIN，让它撤掉页面高亮。
+ *
+ * 健壮性：扩展被重载后旧页面里的本脚本会「上下文失效」，此时访问 chrome.runtime 会抛
+ * "Extension context invalidated"。用 alive() 判活 + try/catch 兜住，并在失效时停掉轮询。
  */
 (function () {
+    const alive = () => !!(chrome.runtime && chrome.runtime.id);
+
     window.addEventListener('message', event => {
-        if (event.source !== window) {
+        if (event.source !== window || !event.data || event.data.__uiInspector !== 'selection') {
+            return;
+        }
+        if (!alive()) {
             return;
         }
-        const data = event.data;
-        if (data && data.__uiInspector === 'selection') {
-            chrome.runtime.sendMessage({type: 'ingest', targets: data.targets || []});
+        try {
+            chrome.runtime.sendMessage(
+                {type: 'ingest', targets: event.data.targets || []},
+                () => void chrome.runtime.lastError
+            );
+        } catch {
+            // 扩展已重载、上下文失效：刷新页面即恢复
         }
     });
 
-    setInterval(() => {
-        chrome.runtime.sendMessage({type: 'state'}, resp => {
-            if (chrome.runtime.lastError) {
-                return;
-            }
-            if (resp && resp.ok && resp.count === 0) {
-                window.postMessage({__uiInspectorBridge: 'cleared'}, '*');
-            }
-        });
+    const timer = setInterval(() => {
+        if (!alive()) {
+            clearInterval(timer);
+            return;
+        }
+        try {
+            chrome.runtime.sendMessage({type: 'state'}, resp => {
+                if (chrome.runtime.lastError) {
+                    return;
+                }
+                if (resp && resp.ok && resp.count === 0) {
+                    window.postMessage({__uiInspectorBridge: 'cleared'}, '*');
+                }
+            });
+        } catch {
+            clearInterval(timer);
+        }
     }, 1500);
 })();

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "comate-ui-inspector",
-    "version": "0.1.0",
+    "version": "0.1.1",
     "description": "按住 Alt 点选页面元素，把「元素 + Vue 组件源码路径」交给 Comate 定位源码修改。浏览器扩展 + Native Messaging host + MCP + hook。",
     "type": "module",
     "license": "MIT",