@hangox/mg-cli 1.0.8 → 1.0.9

package/dist/server.js

@@ -41,6 +41,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 = {
@@ -61,6 +62,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 {
@@ -210,6 +212,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();
   /** 心跳检查定时器 */
@@ -304,6 +312,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;
+        }
+      }
     }
   }
   /**
@@ -357,6 +375,7 @@ var ConnectionManager = class {
     return {
       providers: providerCount,
       consumers: this.consumers.size,
+      extensions: this.chromeExtensions.size,
       total: this.allConnections.size
     };
   }
@@ -366,6 +385,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;
+  }
   /**
    * 关闭所有连接
    */
@@ -377,6 +438,8 @@ var ConnectionManager = class {
     }
     this.providers.clear();
     this.consumers.clear();
+    this.chromeExtensions.clear();
+    this.extensionIndexMap.clear();
     this.allConnections.clear();
   }
 };
@@ -494,6 +557,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();
+  }
   /**
    * 清理特定连接的所有待处理请求
    */
@@ -666,12 +747,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 */,
@@ -679,7 +770,8 @@ var MGServer = class {
       data: {
         connectionId: managedWs.connectionId,
         pageUrl,
-        serverVersion: getVersion()
+        serverVersion: getVersion(),
+        ...extensionIndex !== void 0 && { extensionIndex }
       }
     };
     ws.send(JSON.stringify(ack));
@@ -696,6 +788,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);
@@ -728,6 +829,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,
@@ -737,7 +845,8 @@ var MGServer = class {
       uptime: formatDuration(uptimeMs),
       version: getVersion(),
       stats,
-      connectedPages
+      connectedPages,
+      connectedExtensions
     };
     const response = {
       id: message.id || "",
@@ -748,6 +857,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));
+      }
+    });
+  }
   /**
    * 停止服务器
    */