vite-plugin-aipanel 1.2.0-beta.0

package/es/core/mcp-proxy.mjs

@@ -0,0 +1,212 @@
+var __defProp = Object.defineProperty;
+var __typeError = (msg) => {
+  throw TypeError(msg);
+};
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
+var __accessCheck = (obj, member, msg) => member.has(obj) || __typeError("Cannot " + msg);
+var __privateGet = (obj, member, getter) => (__accessCheck(obj, member, "read from private field"), getter ? getter.call(obj) : member.get(obj));
+var __privateAdd = (obj, member, value) => member.has(obj) ? __typeError("Cannot add the same private member more than once") : member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
+var __privateSet = (obj, member, value, setter) => (__accessCheck(obj, member, "write to private field"), setter ? setter.call(obj, value) : member.set(obj, value), value);
+var __privateMethod = (obj, member, method) => (__accessCheck(obj, member, "access private method"), method);
+var __privateWrapper = (obj, member, setter, getter) => ({
+  set _(value) {
+    __privateSet(obj, member, value, setter);
+  },
+  get _() {
+    return __privateGet(obj, member, getter);
+  }
+});
+var _proc, _rl, _messageId, _internalIdBase, _pending, _args, _startPromise, _idleTimer, _idleTimeout, _McpProxy_instances, doStart_fn, resetIdleTimer_fn;
+import { spawn } from "node:child_process";
+import { createInterface } from "node:readline";
+import crypto from "node:crypto";
+import path from "node:path";
+import { createLogger, createPackageRequire, resolvePackageDir } from "@aipanel/core/node";
+const log = createLogger("McpProxy");
+function resolveChromeDevToolsMcpBin() {
+  const pluginDir = resolvePackageDir("vite-plugin-aipanel");
+  const require2 = createPackageRequire(pluginDir);
+  const pkgJsonPath = require2.resolve("chrome-devtools-mcp/package.json");
+  const pkgDir = path.dirname(pkgJsonPath);
+  const { bin } = require2(pkgJsonPath);
+  const binEntry = typeof bin === "string" ? bin : Object.values(bin)[0];
+  return path.resolve(pkgDir, binEntry);
+}
+class McpProxy {
+  constructor(options = {}) {
+    __privateAdd(this, _McpProxy_instances);
+    __privateAdd(this, _proc, null);
+    __privateAdd(this, _rl, null);
+    __privateAdd(this, _messageId, 0);
+    /** 内部调用使用的高位 ID 起始值，避免与客户端 ID 冲突 */
+    __privateAdd(this, _internalIdBase, 1e6);
+    __privateAdd(this, _pending, /* @__PURE__ */ new Map());
+    __privateAdd(this, _args);
+    __privateAdd(this, _startPromise, null);
+    __privateAdd(this, _idleTimer, null);
+    __privateAdd(this, _idleTimeout);
+    __publicField(this, "sessionId");
+    __privateSet(this, _args, options.args ?? [
+      "--auto-connect",
+      "--no-usage-statistics",
+      "--no-performance-crux"
+    ]);
+    __privateSet(this, _idleTimeout, options.idleTimeout ?? 0);
+    this.sessionId = crypto.randomUUID();
+  }
+  get isRunning() {
+    return __privateGet(this, _proc) !== null && !__privateGet(this, _proc).killed;
+  }
+  async start() {
+    if (this.isRunning) return;
+    if (__privateGet(this, _startPromise)) return __privateGet(this, _startPromise);
+    __privateSet(this, _startPromise, __privateMethod(this, _McpProxy_instances, doStart_fn).call(this));
+    return __privateGet(this, _startPromise);
+  }
+  /** 转发原始 JSON-RPC 请求，保留客户端 ID */
+  async forward(rawRequest) {
+    await this.start();
+    if (!__privateGet(this, _proc) || !__privateGet(this, _proc).stdin) {
+      throw new Error("MCP process not available");
+    }
+    __privateMethod(this, _McpProxy_instances, resetIdleTimer_fn).call(this);
+    let msg;
+    try {
+      msg = JSON.parse(rawRequest);
+    } catch {
+      throw new Error("Invalid JSON-RPC request");
+    }
+    return new Promise((resolve) => {
+      const id = msg.id;
+      if (id !== void 0) {
+        __privateGet(this, _pending).set(id, (raw) => resolve(JSON.stringify(raw)));
+      } else {
+        __privateGet(this, _proc).stdin.write(rawRequest + "\n");
+        resolve("");
+        return;
+      }
+      __privateGet(this, _proc).stdin.write(rawRequest + "\n");
+    });
+  }
+  /** 调用 MCP 工具 */
+  async call(method, params = {}) {
+    await this.start();
+    if (!__privateGet(this, _proc) || !__privateGet(this, _proc).stdin) {
+      throw new Error("MCP process not available");
+    }
+    __privateMethod(this, _McpProxy_instances, resetIdleTimer_fn).call(this);
+    const id = __privateGet(this, _internalIdBase) + ++__privateWrapper(this, _messageId)._;
+    return new Promise((resolve) => {
+      __privateGet(this, _pending).set(id, resolve);
+      const request = JSON.stringify({ jsonrpc: "2.0", id, method, params });
+      __privateGet(this, _proc).stdin.write(request + "\n");
+    });
+  }
+  /** 直接调用 chrome-devtools-mcp 底层工具（内部使用） */
+  async callChromeDevTool(name, args = {}) {
+    return this.call("tools/call", { name, arguments: args });
+  }
+  /** 验证 MCP 是否可用 + 预热 CDP 连接 */
+  async verify() {
+    try {
+      await this.callChromeDevTool("list_pages", {});
+      return { ok: true };
+    } catch (e) {
+      return { ok: false, error: e instanceof Error ? e.message : String(e) };
+    }
+  }
+  stop() {
+    if (__privateGet(this, _idleTimer)) clearTimeout(__privateGet(this, _idleTimer));
+    const err = new Error("MCP server shutting down");
+    for (const resolve of __privateGet(this, _pending).values()) {
+      resolve({ error: { code: -32e3, message: err.message } });
+    }
+    __privateGet(this, _pending).clear();
+    if (__privateGet(this, _rl)) {
+      __privateGet(this, _rl).close();
+      __privateSet(this, _rl, null);
+    }
+    if (__privateGet(this, _proc) && !__privateGet(this, _proc).killed) {
+      __privateGet(this, _proc).kill();
+      __privateSet(this, _proc, null);
+    }
+    __privateSet(this, _startPromise, null);
+  }
+}
+_proc = new WeakMap();
+_rl = new WeakMap();
+_messageId = new WeakMap();
+_internalIdBase = new WeakMap();
+_pending = new WeakMap();
+_args = new WeakMap();
+_startPromise = new WeakMap();
+_idleTimer = new WeakMap();
+_idleTimeout = new WeakMap();
+_McpProxy_instances = new WeakSet();
+doStart_fn = async function() {
+  log.debug("Starting MCP process", { args: __privateGet(this, _args) });
+  try {
+    const binPath = resolveChromeDevToolsMcpBin();
+    __privateSet(this, _proc, spawn(process.execPath, [binPath, ...__privateGet(this, _args)], {
+      stdio: ["pipe", "pipe", "pipe"]
+    }));
+    log.debug("Using local chrome-devtools-mcp");
+  } catch {
+    throw new Error(
+      "Cannot find chrome-devtools-mcp. Please ensure it is installed: npm install chrome-devtools-mcp"
+    );
+  }
+  __privateSet(this, _rl, createInterface({ input: __privateGet(this, _proc).stdout }));
+  __privateGet(this, _rl).on("line", (line) => {
+    try {
+      const msg = JSON.parse(line);
+      log.debug("MCP stdout", {
+        id: msg.id,
+        method: msg.method,
+        hasResult: !!msg.result,
+        hasError: !!msg.error
+      });
+      if (msg.id !== void 0 && __privateGet(this, _pending).has(msg.id)) {
+        const resolve = __privateGet(this, _pending).get(msg.id);
+        __privateGet(this, _pending).delete(msg.id);
+        log.debug("MCP pending resolved", { id: msg.id });
+        resolve(msg);
+      }
+    } catch {
+    }
+  });
+  __privateGet(this, _proc).stderr?.on("data", (d) => {
+    const text = d.toString().trim();
+    if (!text || /No handler registered for issue code \w+/i.test(text)) return;
+    log.debug("[MCP stderr]", { text: text.substring(0, 200) });
+  });
+  __privateGet(this, _proc).on("close", (code) => {
+    log.debug("MCP process closed", { code });
+    for (const resolve of __privateGet(this, _pending).values()) {
+      resolve({ error: { code: -32e3, message: `MCP process exited with code ${code}` } });
+    }
+    __privateGet(this, _pending).clear();
+    __privateSet(this, _proc, null);
+    __privateSet(this, _rl, null);
+    __privateSet(this, _startPromise, null);
+  });
+  await this.call("initialize", {
+    protocolVersion: "2024-11-05",
+    capabilities: {},
+    clientInfo: { name: "vite-plugin-aipanel", version: "1.0.0" }
+  });
+  log.debug("MCP proxy ready");
+  __privateSet(this, _startPromise, null);
+};
+resetIdleTimer_fn = function() {
+  if (__privateGet(this, _idleTimeout) <= 0) return;
+  if (__privateGet(this, _idleTimer)) clearTimeout(__privateGet(this, _idleTimer));
+  __privateSet(this, _idleTimer, setTimeout(() => {
+    log.debug("MCP process idle timeout, stopping");
+    this.stop();
+  }, __privateGet(this, _idleTimeout)));
+};
+export {
+  McpProxy
+};

package/es/core/mcp-tools.d.ts

@@ -0,0 +1,15 @@
+/**
+ * 自定义 DevTools 工具定义
+ * 所有工具（除 chrome-devtools_list_pages）必须传入 pageId 参数，
+ * 代理层校验 pageId 是否为项目页面后方可调用 chrome-devtools-mcp。
+ */
+export interface CustomTool {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: "object";
+        properties: Record<string, unknown>;
+        required?: string[];
+    };
+}
+export declare const CUSTOM_TOOLS: CustomTool[];