@hangox/mg-cli 1.0.8 → 1.0.9

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { S as ServerInfo, N as NodeInfo, a as SpaceNodeInfo, M as MessageType } from './index-DmySkKst.js';
-export { W as AllPagesInfo, B as BaseMessage, C as CONFIG_DIR, T as CliOptions, X as ConnectedPageInfo, F as ConnectionInfo, a0 as ConnectionManager, q as ConnectionType, f as DEFAULT_PORT, D as DEV_DEFAULT_PORT, e as DEV_PORT_RANGE_END, d as DEV_PORT_RANGE_START, E as ErrorCode, x as ErrorInfo, s as ErrorMessages, r as ErrorNames, O as ExportImageParams, K as GetAllNodesParams, G as GetNodeParams, J as GetPageParams, H as HEARTBEAT_INTERVAL, m as HEARTBEAT_TIMEOUT, I as IS_DEV_MODE, L as LOG_DIR, a5 as LogLevel, a3 as Logger, a6 as LoggerOptions, i as MAX_PORT_ATTEMPTS, p as MAX_RETRY_COUNT, t as MGError, Z as MGServer, a1 as ManagedWebSocket, V as MgPageInfo, Q as OutputFormatter, h as PORT_RANGE_END, g as PORT_RANGE_START, j as PORT_SCAN_TIMEOUT, P as PROD_DEFAULT_PORT, c as PROD_PORT_RANGE_END, b as PROD_PORT_RANGE_START, z as PingMessage, A as PongMessage, R as REQUEST_TIMEOUT, o as RETRY_INTERVALS, U as RGBA, y as RegisterMessage, a2 as RequestHandler, v as RequestMessage, w as ResponseMessage, k as SERVER_INFO_FILE, l as SERVER_LOG_FILE, n as SERVER_START_TIMEOUT, $ as ServerOptions, Y as ServerStatusResponse, u as createError, a4 as createLogger, _ as createServer } from './index-DmySkKst.js';
+import { S as ServerInfo, N as NodeInfo, a as SpaceNodeInfo, M as MessageType } from './index-cEKly9mt.js';
+export { W as AllPagesInfo, B as BaseMessage, C as CONFIG_DIR, T as CliOptions, _ as ConnectedPageInfo, F as ConnectionInfo, a3 as ConnectionManager, q as ConnectionType, f as DEFAULT_PORT, D as DEV_DEFAULT_PORT, e as DEV_PORT_RANGE_END, d as DEV_PORT_RANGE_START, E as ErrorCode, x as ErrorInfo, s as ErrorMessages, r as ErrorNames, O as ExportImageParams, X as ExtensionInfo, K as GetAllNodesParams, G as GetNodeParams, J as GetPageParams, H as HEARTBEAT_INTERVAL, m as HEARTBEAT_TIMEOUT, I as IS_DEV_MODE, L as LOG_DIR, a8 as LogLevel, a6 as Logger, a9 as LoggerOptions, i as MAX_PORT_ATTEMPTS, p as MAX_RETRY_COUNT, t as MGError, a0 as MGServer, a4 as ManagedWebSocket, V as MgPageInfo, Z as NavigateToNodeParams, Y as OpenPageParams, Q as OutputFormatter, h as PORT_RANGE_END, g as PORT_RANGE_START, j as PORT_SCAN_TIMEOUT, P as PROD_DEFAULT_PORT, c as PROD_PORT_RANGE_END, b as PROD_PORT_RANGE_START, z as PingMessage, A as PongMessage, R as REQUEST_TIMEOUT, o as RETRY_INTERVALS, U as RGBA, y as RegisterMessage, a5 as RequestHandler, v as RequestMessage, w as ResponseMessage, k as SERVER_INFO_FILE, l as SERVER_LOG_FILE, n as SERVER_START_TIMEOUT, a2 as ServerOptions, $ as ServerStatusResponse, u as createError, a7 as createLogger, a1 as createServer } from './index-cEKly9mt.js';
 import 'ws';
 
 /**
@@ -155,7 +155,7 @@ declare const NODE_DEFAULTS: Record<string, unknown>;
 declare function trimNodeDefaults<T extends Record<string, unknown>>(node: T): T;
 /**
  * 提取节点的空间信息
- * 只保留 id、name、x、y、width、height 和 children 字段
+ * 只保留 id、name、x、y、width、height、absoluteX、absoluteY 和 children 字段
  * 用于让 AI 以最少的字段理解节点的空间布局关系
  *
  * @param node 原始节点数据

package/dist/index.js

@@ -26,6 +26,7 @@ var MAX_RETRY_COUNT = 3;
 var ConnectionType = /* @__PURE__ */ ((ConnectionType2) => {
   ConnectionType2["CONSUMER"] = "consumer";
   ConnectionType2["PROVIDER"] = "provider";
+  ConnectionType2["CHROME_EXTENSION"] = "chrome_extension";
   return ConnectionType2;
 })(ConnectionType || {});
 var MessageType = /* @__PURE__ */ ((MessageType2) => {
@@ -40,6 +41,9 @@ var MessageType = /* @__PURE__ */ ((MessageType2) => {
   MessageType2["EXPORT_IMAGE"] = "export_image";
   MessageType2["EXECUTE_CODE"] = "execute_code";
   MessageType2["GET_ALL_PAGES"] = "get_all_pages";
+  MessageType2["LIST_EXTENSIONS"] = "list_extensions";
+  MessageType2["OPEN_PAGE"] = "open_page";
+  MessageType2["NAVIGATE_TO_NODE"] = "navigate_to_node";
   MessageType2["GET_SERVER_STATUS"] = "get_server_status";
   MessageType2["RESPONSE"] = "response";
   MessageType2["ERROR"] = "error";
@@ -65,6 +69,7 @@ var ErrorCode = /* @__PURE__ */ ((ErrorCode2) => {
   ErrorCode2["SERVER_START_FAILED"] = "E015";
   ErrorCode2["SERVER_ALREADY_RUNNING"] = "E016";
   ErrorCode2["CONNECTION_LOST"] = "E017";
+  ErrorCode2["NO_EXTENSION_CONNECTED"] = "E018";
   ErrorCode2["UNKNOWN_ERROR"] = "E099";
   return ErrorCode2;
 })(ErrorCode || {});
@@ -86,6 +91,7 @@ var ErrorNames = {
   ["E015" /* SERVER_START_FAILED */]: "SERVER_START_FAILED",
   ["E016" /* SERVER_ALREADY_RUNNING */]: "SERVER_ALREADY_RUNNING",
   ["E017" /* CONNECTION_LOST */]: "CONNECTION_LOST",
+  ["E018" /* NO_EXTENSION_CONNECTED */]: "NO_EXTENSION_CONNECTED",
   ["E099" /* UNKNOWN_ERROR */]: "UNKNOWN_ERROR"
 };
 var ErrorMessages = {
@@ -106,6 +112,7 @@ var ErrorMessages = {
   ["E015" /* SERVER_START_FAILED */]: "\u81EA\u52A8\u542F\u52A8 Server \u5931\u8D25",
   ["E016" /* SERVER_ALREADY_RUNNING */]: "Server \u5DF2\u5728\u8FD0\u884C\u4E2D",
   ["E017" /* CONNECTION_LOST */]: "\u8FDE\u63A5\u65AD\u5F00",
+  ["E018" /* NO_EXTENSION_CONNECTED */]: "\u6CA1\u6709 Chrome \u6269\u5C55\u8FDE\u63A5\u5230 Server\u3002\u8BF7\u786E\u4FDD\u6D4F\u89C8\u5668\u5DF2\u6253\u5F00\u5E76\u5B89\u88C5\u4E86 MG Plugin",
   ["E099" /* UNKNOWN_ERROR */]: "\u672A\u77E5\u9519\u8BEF"
 };
 var MGError = class extends Error {
@@ -489,6 +496,12 @@ function extractSpaceInfo(node) {
     width: roundToOneDecimal(typeof node.width === "number" ? node.width : 0),
     height: roundToOneDecimal(typeof node.height === "number" ? node.height : 0)
   };
+  if (typeof node.absoluteX === "number") {
+    result.absoluteX = roundToOneDecimal(node.absoluteX);
+  }
+  if (typeof node.absoluteY === "number") {
+    result.absoluteY = roundToOneDecimal(node.absoluteY);
+  }
   if (node.children && Array.isArray(node.children) && node.children.length > 0) {
     result.children = node.children.map((child) => extractSpaceInfo(child));
   }
@@ -584,6 +597,12 @@ var ConnectionManager = class {
   providers = /* @__PURE__ */ new Map();
   /** Consumer 连接 */
   consumers = /* @__PURE__ */ new Map();
+  /** Chrome 扩展连接（按扩展 ID 索引） */
+  chromeExtensions = /* @__PURE__ */ new Map();
+  /** 扩展序号映射（扩展 ID → 序号） */
+  extensionIndexMap = /* @__PURE__ */ new Map();
+  /** 下一个可用的扩展序号 */
+  nextExtensionIndex = 1;
   /** 所有连接（按 ID 索引） */
   allConnections = /* @__PURE__ */ new Map();
   /** 心跳检查定时器 */
@@ -678,6 +697,16 @@ var ConnectionManager = class {
     } else if (connectionInfo.type === "consumer" /* CONSUMER */) {
       this.consumers.delete(connectionId);
       this.logger.info(`Consumer \u65AD\u5F00: ${connectionId}`);
+    } else if (connectionInfo.type === "chrome_extension" /* CHROME_EXTENSION */) {
+      for (const [extensionId, extWs] of this.chromeExtensions) {
+        if (extWs.connectionId === connectionId) {
+          const index = this.extensionIndexMap.get(extensionId);
+          this.chromeExtensions.delete(extensionId);
+          this.extensionIndexMap.delete(extensionId);
+          this.logger.info(`Chrome \u6269\u5C55\u65AD\u5F00: ${extensionId} (\u5E8F\u53F7: #${index})`);
+          break;
+        }
+      }
     }
   }
   /**
@@ -731,6 +760,7 @@ var ConnectionManager = class {
     return {
       providers: providerCount,
       consumers: this.consumers.size,
+      extensions: this.chromeExtensions.size,
       total: this.allConnections.size
     };
   }
@@ -740,6 +770,48 @@ var ConnectionManager = class {
   getConnectedPageUrls() {
     return Array.from(this.providers.keys());
   }
+  /**
+   * 添加 Chrome 扩展连接
+   */
+  addChromeExtension(ws, extensionId) {
+    const managedWs = this.addConnection(ws, "chrome_extension" /* CHROME_EXTENSION */);
+    const index = this.nextExtensionIndex++;
+    this.extensionIndexMap.set(extensionId, index);
+    this.chromeExtensions.set(extensionId, managedWs);
+    this.logger.info(`Chrome \u6269\u5C55\u8FDE\u63A5: ${extensionId} (\u5206\u914D\u5E8F\u53F7: #${index})`);
+    return managedWs;
+  }
+  /**
+   * 根据序号查找扩展
+   */
+  findExtensionByIndex(index) {
+    for (const [extensionId, extIndex] of this.extensionIndexMap) {
+      if (extIndex === index) {
+        return this.chromeExtensions.get(extensionId);
+      }
+    }
+    return void 0;
+  }
+  /**
+   * 获取第一个扩展
+   */
+  getFirstExtension() {
+    return this.chromeExtensions.values().next().value;
+  }
+  /**
+   * 获取所有扩展信息
+   */
+  getAllExtensions() {
+    const result = [];
+    for (const [extensionId, ws] of this.chromeExtensions) {
+      const index = this.extensionIndexMap.get(extensionId);
+      if (index !== void 0) {
+        result.push({ index, extensionId, ws });
+      }
+    }
+    result.sort((a, b) => a.index - b.index);
+    return result;
+  }
   /**
    * 关闭所有连接
    */
@@ -751,6 +823,8 @@ var ConnectionManager = class {
     }
     this.providers.clear();
     this.consumers.clear();
+    this.chromeExtensions.clear();
+    this.extensionIndexMap.clear();
     this.allConnections.clear();
   }
 };
@@ -868,6 +942,24 @@ var RequestHandler = class {
       this.logger.error(`\u53D1\u9001\u9519\u8BEF\u54CD\u5E94\u5931\u8D25: ${requestId}`, err);
     }
   }
+  /**
+   * 注册待处理请求（供外部使用，如 OPEN_PAGE）
+   * @param requestId 请求 ID
+   * @param consumer 发起请求的 Consumer
+   * @param sendCallback 发送请求的回调函数
+   */
+  registerPendingRequest(requestId, consumer, sendCallback) {
+    const timer = setTimeout(() => {
+      this.handleTimeout(requestId);
+    }, REQUEST_TIMEOUT);
+    this.pendingRequests.set(requestId, {
+      id: requestId,
+      consumer,
+      timer,
+      timestamp: Date.now()
+    });
+    sendCallback();
+  }
   /**
    * 清理特定连接的所有待处理请求
    */
@@ -1047,12 +1139,22 @@ var MGServer = class {
    */
   handleRegister(ws, message) {
     const { connectionType, pageUrl, pageId } = message.data;
-    const managedWs = this.connectionManager.addConnection(
-      ws,
-      connectionType,
-      pageUrl,
-      pageId
-    );
+    const extensionId = message.data.extensionId;
+    let managedWs;
+    let extensionIndex;
+    if (connectionType === "chrome_extension" /* CHROME_EXTENSION */ && extensionId) {
+      managedWs = this.connectionManager.addChromeExtension(ws, extensionId);
+      const exts = this.connectionManager.getAllExtensions();
+      const ext = exts.find((e) => e.extensionId === extensionId);
+      extensionIndex = ext?.index;
+    } else {
+      managedWs = this.connectionManager.addConnection(
+        ws,
+        connectionType,
+        pageUrl,
+        pageId
+      );
+    }
     const ack = {
       id: message.id || "",
       type: "register_ack" /* REGISTER_ACK */,
@@ -1060,7 +1162,8 @@ var MGServer = class {
       data: {
         connectionId: managedWs.connectionId,
         pageUrl,
-        serverVersion: getVersion()
+        serverVersion: getVersion(),
+        ...extensionIndex !== void 0 && { extensionIndex }
       }
     };
     ws.send(JSON.stringify(ack));
@@ -1077,6 +1180,15 @@ var MGServer = class {
       case "get_server_status" /* GET_SERVER_STATUS */:
         this.handleGetServerStatus(ws, message);
         break;
+      case "list_extensions" /* LIST_EXTENSIONS */:
+        this.handleListExtensions(ws, message);
+        break;
+      case "open_page" /* OPEN_PAGE */:
+        this.handleOpenPage(ws, message);
+        break;
+      case "navigate_to_node" /* NAVIGATE_TO_NODE */:
+        this.handleNavigateToNode(ws, message);
+        break;
       case "response" /* RESPONSE */:
       case "error" /* ERROR */:
         this.requestHandler.handleResponse(message);
@@ -1109,6 +1221,13 @@ var MGServer = class {
       connectedAt: info.connectedAt.toISOString(),
       lastActiveAt: info.lastActiveAt.toISOString()
     }));
+    const extensions = this.connectionManager.getAllExtensions();
+    const connectedExtensions = extensions.map((ext) => ({
+      index: ext.index,
+      extensionId: ext.extensionId,
+      connectedAt: ext.ws.connectionInfo.connectedAt.toISOString(),
+      lastActiveAt: ext.ws.connectionInfo.lastActiveAt.toISOString()
+    }));
     const uptimeMs = this.startedAt ? Date.now() - this.startedAt.getTime() : 0;
     const statusData = {
       running: this.isRunning,
@@ -1118,7 +1237,8 @@ var MGServer = class {
       uptime: formatDuration(uptimeMs),
       version: getVersion(),
       stats,
-      connectedPages
+      connectedPages,
+      connectedExtensions
     };
     const response = {
       id: message.id || "",
@@ -1129,6 +1249,156 @@ var MGServer = class {
     ws.send(JSON.stringify(response));
     this.logger.info("\u8FD4\u56DE Server \u72B6\u6001\u4FE1\u606F");
   }
+  /**
+   * 处理扩展列表查询
+   */
+  handleListExtensions(ws, message) {
+    const extensions = this.connectionManager.getAllExtensions();
+    const extensionList = extensions.map((ext) => ({
+      index: ext.index,
+      extensionId: ext.extensionId,
+      connectedAt: ext.ws.connectionInfo.connectedAt.toISOString(),
+      lastActiveAt: ext.ws.connectionInfo.lastActiveAt.toISOString()
+    }));
+    const response = {
+      id: message.id || "",
+      type: "response" /* RESPONSE */,
+      success: true,
+      data: {
+        extensions: extensionList,
+        totalCount: extensionList.length
+      }
+    };
+    ws.send(JSON.stringify(response));
+    this.logger.info(`\u8FD4\u56DE\u6269\u5C55\u5217\u8868\uFF0C\u5171 ${extensionList.length} \u4E2A\u6269\u5C55`);
+  }
+  /**
+   * 处理打开页面请求
+   */
+  handleOpenPage(ws, message) {
+    const params = message.params;
+    const requestId = message.id || "";
+    if (!params?.url) {
+      const errorResponse = {
+        id: requestId,
+        type: "error" /* ERROR */,
+        success: false,
+        error: {
+          code: "E011" /* INVALID_PARAMS */,
+          name: "INVALID_PARAMS",
+          message: "\u7F3A\u5C11 url \u53C2\u6570"
+        }
+      };
+      ws.send(JSON.stringify(errorResponse));
+      return;
+    }
+    const targetIndex = params.extensionIndex || 1;
+    const targetExtension = this.connectionManager.findExtensionByIndex(targetIndex);
+    if (!targetExtension) {
+      const errorResponse = {
+        id: requestId,
+        type: "error" /* ERROR */,
+        success: false,
+        error: {
+          code: "E018" /* NO_EXTENSION_CONNECTED */,
+          name: "NO_EXTENSION_CONNECTED",
+          message: targetIndex === 1 ? "\u6CA1\u6709 Chrome \u6269\u5C55\u8FDE\u63A5\u5230 Server\u3002\u8BF7\u786E\u4FDD\u6D4F\u89C8\u5668\u5DF2\u6253\u5F00\u5E76\u5B89\u88C5\u4E86 MG Plugin" : `\u672A\u627E\u5230\u5E8F\u53F7\u4E3A #${targetIndex} \u7684\u6269\u5C55\u5B9E\u4F8B`
+        }
+      };
+      ws.send(JSON.stringify(errorResponse));
+      this.logger.warn(`\u672A\u627E\u5230\u6269\u5C55\u5B9E\u4F8B #${targetIndex}`);
+      return;
+    }
+    const forwardMessage = {
+      id: requestId,
+      type: "open_page" /* OPEN_PAGE */,
+      params: { url: params.url }
+    };
+    this.requestHandler.registerPendingRequest(requestId, ws, () => {
+      try {
+        targetExtension.send(JSON.stringify(forwardMessage));
+        this.logger.info(`\u8F6C\u53D1 OPEN_PAGE \u8BF7\u6C42\u5230\u6269\u5C55 #${targetIndex}: ${params.url}`);
+      } catch (error) {
+        this.logger.error(`\u8F6C\u53D1\u8BF7\u6C42\u5931\u8D25: ${error}`);
+        const errorResponse = {
+          id: requestId,
+          type: "error" /* ERROR */,
+          success: false,
+          error: {
+            code: "E001" /* CONNECTION_FAILED */,
+            name: "CONNECTION_FAILED",
+            message: "\u8F6C\u53D1\u8BF7\u6C42\u5931\u8D25"
+          }
+        };
+        ws.send(JSON.stringify(errorResponse));
+      }
+    });
+  }
+  /**
+   * 处理节点导航请求
+   */
+  handleNavigateToNode(ws, message) {
+    const params = message.params;
+    const requestId = message.id || "";
+    if (!params?.pageUrl || !params?.nodeId) {
+      const errorResponse = {
+        id: requestId,
+        type: "error" /* ERROR */,
+        success: false,
+        error: {
+          code: "E011" /* INVALID_PARAMS */,
+          name: "INVALID_PARAMS",
+          message: "\u7F3A\u5C11 pageUrl \u6216 nodeId \u53C2\u6570"
+        }
+      };
+      ws.send(JSON.stringify(errorResponse));
+      return;
+    }
+    const targetIndex = params.extensionIndex || 1;
+    const targetExtension = this.connectionManager.findExtensionByIndex(targetIndex);
+    if (!targetExtension) {
+      const errorResponse = {
+        id: requestId,
+        type: "error" /* ERROR */,
+        success: false,
+        error: {
+          code: "E018" /* NO_EXTENSION_CONNECTED */,
+          name: "NO_EXTENSION_CONNECTED",
+          message: targetIndex === 1 ? "\u6CA1\u6709 Chrome \u6269\u5C55\u8FDE\u63A5\u5230 Server\u3002\u8BF7\u786E\u4FDD\u6D4F\u89C8\u5668\u5DF2\u6253\u5F00\u5E76\u5B89\u88C5\u4E86 MG Plugin" : `\u672A\u627E\u5230\u5E8F\u53F7\u4E3A #${targetIndex} \u7684\u6269\u5C55\u5B9E\u4F8B`
+        }
+      };
+      ws.send(JSON.stringify(errorResponse));
+      this.logger.warn(`\u672A\u627E\u5230\u6269\u5C55\u5B9E\u4F8B #${targetIndex}`);
+      return;
+    }
+    const forwardMessage = {
+      id: requestId,
+      type: "navigate_to_node" /* NAVIGATE_TO_NODE */,
+      params: {
+        pageUrl: params.pageUrl,
+        nodeId: params.nodeId
+      }
+    };
+    this.requestHandler.registerPendingRequest(requestId, ws, () => {
+      try {
+        targetExtension.send(JSON.stringify(forwardMessage));
+        this.logger.info(`\u8F6C\u53D1 NAVIGATE_TO_NODE \u8BF7\u6C42\u5230\u6269\u5C55 #${targetIndex}: ${params.pageUrl} -> ${params.nodeId}`);
+      } catch (error) {
+        this.logger.error(`\u8F6C\u53D1\u8BF7\u6C42\u5931\u8D25: ${error}`);
+        const errorResponse = {
+          id: requestId,
+          type: "error" /* ERROR */,
+          success: false,
+          error: {
+            code: "E001" /* CONNECTION_FAILED */,
+            name: "CONNECTION_FAILED",
+            message: "\u8F6C\u53D1\u8BF7\u6C42\u5931\u8D25"
+          }
+        };
+        ws.send(JSON.stringify(errorResponse));
+      }
+    });
+  }
   /**
    * 停止服务器
    */