vite-plugin-aipanel 1.2.8 → 1.2.9

package/lib/core/mcp-proxy.cjs

@@ -45,7 +45,9 @@ var __privateWrapper = (obj, member, setter, getter) => ({
 });
 var mcp_proxy_exports = {};
 __export(mcp_proxy_exports, {
-  McpProxy: () => McpProxy
+  CORE_MCP_ARGS: () => CORE_MCP_ARGS,
+  McpProxy: () => McpProxy,
+  filterUserMcpArgs: () => filterUserMcpArgs
 });
 module.exports = __toCommonJS(mcp_proxy_exports);
 var import_node_child_process = require("node:child_process");
@@ -53,8 +55,36 @@ var import_node_readline = require("node:readline");
 var import_node_crypto = __toESM(require("node:crypto"));
 var import_node_path = __toESM(require("node:path"));
 var import_node = require("@aipanel/core/node");
-var _proc, _rl, _messageId, _internalIdBase, _pending, _args, _startPromise, _lastExit, _stderrTail, _idleTimer, _idleTimeout, _McpProxy_instances, doStart_fn, formatNotAvailable_fn, resetIdleTimer_fn;
+var _proc, _rl, _messageId, _internalIdBase, _pending, _args, _env, _startPromise, _lastExit, _stderrTail, _idleTimer, _idleTimeout, _McpProxy_instances, doStart_fn, formatNotAvailable_fn, resetIdleTimer_fn;
 const log = (0, import_node.createLogger)("McpProxy");
+const CORE_MCP_ARGS = [
+  "--auto-connect",
+  "--no-usage-statistics",
+  "--no-performance-crux",
+  // chrome-devtools-mcp >=1.8.0 默认开启 pageIdRouting（要求每个页面级工具都传 pageId）。
+  // 代理层已自行校验 pageId 并用 select_page 选中目标页面后再转发，故显式关闭，
+  // 避免底层工具 schema 强制必填 pageId 导致转发时的参数校验失败。
+  "--no-page-id-routing"
+];
+const PROTECTED_FLAGS = /* @__PURE__ */ new Set([
+  "auto-connect",
+  "usage-statistics",
+  "performance-crux",
+  "page-id-routing"
+]);
+function canonicalFlagName(arg) {
+  let flag = arg.trim();
+  if (!flag.startsWith("-")) return null;
+  flag = flag.replace(/^--?/, "").split("=")[0] ?? "";
+  if (flag.startsWith("no-")) flag = flag.slice(3);
+  return flag || null;
+}
+function filterUserMcpArgs(userArgs) {
+  return userArgs.filter((arg) => {
+    const name = canonicalFlagName(arg);
+    return name === null || !PROTECTED_FLAGS.has(name);
+  });
+}
 function resolveChromeDevToolsMcpBin() {
   const pluginDir = (0, import_node.resolvePackageDir)("vite-plugin-aipanel");
   const require2 = (0, import_node.createPackageRequire)(pluginDir);
@@ -74,6 +104,7 @@ class McpProxy {
     __privateAdd(this, _internalIdBase, 1e6);
     __privateAdd(this, _pending, /* @__PURE__ */ new Map());
     __privateAdd(this, _args);
+    __privateAdd(this, _env);
     __privateAdd(this, _startPromise, null);
     /** 最近一次进程退出信息，用于生成精确错误（无退出记录时为 null） */
     __privateAdd(this, _lastExit, null);
@@ -82,15 +113,14 @@ class McpProxy {
     __privateAdd(this, _idleTimer, null);
     __privateAdd(this, _idleTimeout);
     __publicField(this, "sessionId");
-    __privateSet(this, _args, options.args ?? [
-      "--auto-connect",
-      "--no-usage-statistics",
-      "--no-performance-crux",
-      // chrome-devtools-mcp >=1.8.0 默认开启 pageIdRouting（要求每个页面级工具都传 pageId）。
-      // 代理层已自行校验 pageId 并用 select_page 选中目标页面后再转发，故显式关闭，
-      // 避免底层工具 schema 强制必填 pageId 导致转发时的参数校验失败。
-      "--no-page-id-routing"
-    ]);
+    const userArgs = options.userArgs ?? [];
+    const filtered = filterUserMcpArgs(userArgs);
+    if (filtered.length !== userArgs.length) {
+      const dropped = userArgs.filter((a) => !filtered.includes(a));
+      log.warn("chrome MCP \u7528\u6237\u53C2\u6570\u4E0E\u6838\u5FC3\u53D7\u4FDD\u62A4\u9879\u51B2\u7A81\uFF0C\u5DF2\u5FFD\u7565", { dropped });
+    }
+    __privateSet(this, _args, [...filtered, ...CORE_MCP_ARGS]);
+    __privateSet(this, _env, options.env);
     __privateSet(this, _idleTimeout, options.idleTimeout ?? 0);
     this.sessionId = import_node_crypto.default.randomUUID();
   }
@@ -184,6 +214,7 @@ _messageId = new WeakMap();
 _internalIdBase = new WeakMap();
 _pending = new WeakMap();
 _args = new WeakMap();
+_env = new WeakMap();
 _startPromise = new WeakMap();
 _lastExit = new WeakMap();
 _stderrTail = new WeakMap();
@@ -197,7 +228,8 @@ doStart_fn = async function() {
     __privateSet(this, _lastExit, null);
     __privateSet(this, _stderrTail, []);
     __privateSet(this, _proc, (0, import_node_child_process.spawn)(process.execPath, [binPath, ...__privateGet(this, _args)], {
-      stdio: ["pipe", "pipe", "pipe"]
+      stdio: ["pipe", "pipe", "pipe"],
+      env: { ...process.env, ...__privateGet(this, _env) }
     }));
     log.debug("Using local chrome-devtools-mcp");
   } catch {
@@ -287,5 +319,7 @@ resetIdleTimer_fn = function() {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  McpProxy
+  CORE_MCP_ARGS,
+  McpProxy,
+  filterUserMcpArgs
 });

package/lib/core/mcp-proxy.d.ts

@@ -1,5 +1,12 @@
+/** chrome-devtools-mcp 的核心受保护参数（用户不可覆盖；始终最后注入确保生效） */
+export declare const CORE_MCP_ARGS: readonly ["--auto-connect", "--no-usage-statistics", "--no-performance-crux", "--no-page-id-routing"];
+/** 过滤用户透传参数：剔除与核心受保护 flag 冲突的项，其余原样保留 */
+export declare function filterUserMcpArgs(userArgs: readonly string[]): string[];
 export interface McpProxyOptions {
-    args?: string[];
+    /** 用户透传的额外 CLI 参数（追加；与核心受保护项冲突的会被剔除） */
+    userArgs?: string[];
+    /** 用户透传的额外环境变量（合并到 process.env 之上） */
+    env?: Record<string, string>;
     idleTimeout?: number;
 }
 export declare class McpProxy {

package/lib/core/proxy-server.cjs

@@ -32,7 +32,8 @@ __export(proxy_server_exports, {
 module.exports = __toCommonJS(proxy_server_exports);
 var import_http = __toESM(require("http"));
 var import_core = require("@aipanel/core");
-const log = (0, import_core.createLogger)("ProxyServer");
+var import_node = require("@aipanel/core/node");
+const log = (0, import_node.createLogger)("ProxyServer");
 function startProxyServer(targetUrl, port, options = {}) {
   return new Promise((resolve, reject) => {
     const target = new URL(targetUrl);
@@ -85,18 +86,20 @@ function startProxyServer(targetUrl, port, options = {}) {
           });
           proxyRes.on("end", () => {
             let body = Buffer.concat(chunks).toString("utf-8");
-            if (body.match(/<\/head>/i)) {
-              body = body.replace(
-                /<\/head>/i,
-                `<script src="${import_core.BRIDGE_SCRIPT_PATH}"></script></head>`
-              );
-            } else if (body.match(/<\/body>/i)) {
-              body = body.replace(
-                /<\/body>/i,
-                `<script src="${import_core.BRIDGE_SCRIPT_PATH}"></script></body>`
-              );
-            } else {
-              body += `<script src="${import_core.BRIDGE_SCRIPT_PATH}"></script>`;
+            if (bridgeScript.trim().length > 0) {
+              if (body.match(/<\/head>/i)) {
+                body = body.replace(
+                  /<\/head>/i,
+                  `<script src="${import_core.BRIDGE_SCRIPT_PATH}"></script></head>`
+                );
+              } else if (body.match(/<\/body>/i)) {
+                body = body.replace(
+                  /<\/body>/i,
+                  `<script src="${import_core.BRIDGE_SCRIPT_PATH}"></script></body>`
+                );
+              } else {
+                body += `<script src="${import_core.BRIDGE_SCRIPT_PATH}"></script>`;
+              }
             }
             const headers = {};
             for (const [key, value] of Object.entries(proxyRes.headers)) {

package/lib/index.cjs

@@ -57,7 +57,7 @@ const SILENT_CONTEXT_SCRIPT = `<script>
     if (!force && key === last) return;
     last = key;
     var sessionId = "";
-    try { sessionId = sessionStorage.getItem("_aipanel_pk") || ""; } catch (e) {}
+    try { sessionId = sessionStorage.getItem("${import_core.SESSION_ID_KEY}") || ""; } catch (e) {}
     fetch(API, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
@@ -113,7 +113,12 @@ function createAIPanelPlugin(options = {}) {
   let activeTabId = DEFAULT_TAB;
   const serviceInstanceId = import_crypto.default.randomUUID();
   const sseClients = /* @__PURE__ */ new Set();
-  const mcpProxy = new import_mcp_proxy.McpProxy({ idleTimeout: 5 * 60 * 1e3 });
+  const mcpProxy = new import_mcp_proxy.McpProxy({
+    idleTimeout: 5 * 60 * 1e3,
+    // 用户 chrome MCP 透传：追加 args/env（核心受保护参数不可覆盖，由 McpProxy 过滤）
+    userArgs: config.chromeMcp?.args,
+    env: config.chromeMcp?.env
+  });
   let provider = null;
   const service = new import_service.AIPanelService(
     config,
@@ -329,7 +334,7 @@ function createAIPanelPlugin(options = {}) {
       ];
       const titleInject = `<script>
         (function () {
-          var KEY = "_aipanel_pk";
+          var KEY = "${import_core.SESSION_ID_KEY}";
           if (!sessionStorage.getItem(KEY)) {
             sessionStorage.setItem(KEY, "[" + Math.random().toString(36).slice(2, 10) + "]");
           }

package/lib/utils/format-bridge.cjs

@@ -24,20 +24,21 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var import_fs = __toESM(require("fs"));
 var import_http = __toESM(require("http"));
 var import_path = __toESM(require("path"));
+var import_core = require("@aipanel/core");
 const filePath = process.argv[2];
 if (!filePath) {
   console.error("Usage: node format-bridge.cjs <filePath>");
   process.exit(1);
 }
 const absPath = import_path.default.resolve(filePath);
-const PORT = parseInt(process.env.AIPANEL_VSCODE_PORT || "", 10) || 51939;
+const PORT = parseInt(process.env[import_core.ENV_VSCODE_PORT] || "", 10) || import_core.VSCODE_EXTENSION_PORT;
 function touchFile() {
   const now = /* @__PURE__ */ new Date();
   import_fs.default.utimesSync(absPath, now, now);
 }
 function probeHealth() {
   return new Promise((resolve) => {
-    const req = import_http.default.get(`http://127.0.0.1:${PORT}/health`, (res) => {
+    const req = import_http.default.get(`http://${import_core.DEFAULT_HOSTNAME}:${PORT}${import_core.VSCODE_ROUTE_HEALTH}`, (res) => {
       res.resume();
       resolve(res.statusCode === 200);
     });
@@ -53,9 +54,9 @@ function formatViaExtension() {
     const body = JSON.stringify({ filePath: absPath });
     const req = import_http.default.request(
       {
-        hostname: "127.0.0.1",
+        hostname: import_core.DEFAULT_HOSTNAME,
         port: PORT,
-        path: "/format",
+        path: import_core.VSCODE_ROUTE_FORMAT,
         method: "POST",
         headers: { "Content-Type": "application/json" },
         timeout: 5e3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-aipanel",
-  "version": "1.2.8",
+  "version": "1.2.9",
   "description": "Embed OpenCode Web UI in your Vite dev server for real-time code modification and preview",
   "type": "module",
   "main": "lib/index.cjs",
@@ -41,12 +41,12 @@
     "execa": "^9.6.1",
     "typescript-language-server": "^5.3.0",
     "unplugin-vue-inspector": "^3.0.0",
-    "@aipanel/core": "1.2.8",
-    "@aipanel/provider-opencode": "1.2.8"
+    "@aipanel/core": "1.2.9",
+    "@aipanel/provider-opencode": "1.2.9"
   },
   "peerDependencies": {
     "vite": ">=5.0.0",
-    "@aipanel/provider-deepseek": "1.2.8"
+    "@aipanel/provider-deepseek": "1.2.9"
   },
   "peerDependenciesMeta": {
     "@aipanel/provider-deepseek": {
@@ -55,7 +55,7 @@
   },
   "devDependencies": {
     "sharp": "^0.34.5",
-    "@aipanel/client": "1.2.8"
+    "@aipanel/client": "1.2.9"
   },
   "scripts": {
     "build": "pagoda-cli build && vite build -c vite.client.config.ts",