@hangox/mg-cli 1.1.2 → 1.1.3

package/dist/index.js

@@ -45,6 +45,7 @@ var MessageType = /* @__PURE__ */ ((MessageType2) => {
   MessageType2["OPEN_PAGE"] = "open_page";
   MessageType2["NAVIGATE_TO_NODE"] = "navigate_to_node";
   MessageType2["GET_SERVER_STATUS"] = "get_server_status";
+  MessageType2["TRACK_ANALYTICS"] = "track_analytics";
   MessageType2["RESPONSE"] = "response";
   MessageType2["ERROR"] = "error";
   return MessageType2;
@@ -665,8 +666,8 @@ var ConnectionManager = class {
   allConnections = /* @__PURE__ */ new Map();
   /** 心跳检查定时器 */
   heartbeatTimer = null;
-  constructor(logger) {
-    this.logger = logger;
+  constructor(logger2) {
+    this.logger = logger2;
   }
   /**
    * 启动心跳检查
@@ -893,9 +894,9 @@ var RequestHandler = class {
   connectionManager;
   /** 待处理的请求 */
   pendingRequests = /* @__PURE__ */ new Map();
-  constructor(connectionManager, logger) {
+  constructor(connectionManager, logger2) {
     this.connectionManager = connectionManager;
-    this.logger = logger;
+    this.logger = logger2;
   }
   /**
    * 处理 Consumer 请求
@@ -1080,6 +1081,60 @@ function isVersionMatch(version1, version2) {
   return version1 === version2;
 }
 
+// src/server/analytics.ts
+import { PostHog } from "posthog-node";
+var POSTHOG_API_KEY = "phc_cSZZIxZOPhWOlsw92Wz8tmKQ5z8eQYsI3m9aM2Ujia9";
+var POSTHOG_HOST = "https://us.i.posthog.com";
+var posthogClient = null;
+var logger = null;
+function isTelemetryDisabled() {
+  return process.env.MG_TELEMETRY_DISABLED === "1" || process.env.MG_TELEMETRY_DISABLED === "true" || process.env.DO_NOT_TRACK === "1" || process.env.DO_NOT_TRACK === "true";
+}
+function initServerAnalytics(serverLogger) {
+  logger = serverLogger;
+  if (isTelemetryDisabled()) {
+    logger.info("\u9065\u6D4B\u5DF2\u7981\u7528 (MG_TELEMETRY_DISABLED \u6216 DO_NOT_TRACK)");
+    return;
+  }
+  if (!posthogClient) {
+    posthogClient = new PostHog(POSTHOG_API_KEY, {
+      host: POSTHOG_HOST,
+      // Server 端使用批量模式，减少网络请求
+      flushAt: 10,
+      flushInterval: 3e4
+      // 30 秒
+    });
+    logger.info("PostHog Analytics \u5DF2\u521D\u59CB\u5316");
+  }
+}
+function trackEvent(eventData) {
+  if (!posthogClient) {
+    return;
+  }
+  try {
+    posthogClient.capture({
+      distinctId: eventData.distinctId,
+      event: eventData.event,
+      properties: eventData.properties
+    });
+    logger?.info(`\u7EDF\u8BA1\u4E8B\u4EF6\u5DF2\u8BB0\u5F55: ${eventData.event}`, eventData.properties);
+  } catch (error) {
+    logger?.error("\u7EDF\u8BA1\u4E8B\u4EF6\u8BB0\u5F55\u5931\u8D25:", error);
+  }
+}
+async function shutdownServerAnalytics() {
+  if (posthogClient) {
+    try {
+      logger?.info("\u6B63\u5728\u53D1\u9001\u5269\u4F59\u7EDF\u8BA1\u4E8B\u4EF6...");
+      await posthogClient.shutdown();
+      logger?.info("PostHog Analytics \u5DF2\u5173\u95ED");
+    } catch (error) {
+      logger?.error("\u5173\u95ED PostHog \u5931\u8D25:", error);
+    }
+    posthogClient = null;
+  }
+}
+
 // src/server/websocket-server.ts
 var MGServer = class {
   wss = null;
@@ -1111,6 +1166,7 @@ var MGServer = class {
         this.startedAt = /* @__PURE__ */ new Date();
         this.logger.info(`Server \u542F\u52A8\u6210\u529F\uFF0C\u76D1\u542C\u7AEF\u53E3: ${port}`);
         this.connectionManager.startHeartbeatCheck(HEARTBEAT_INTERVAL);
+        initServerAnalytics(this.logger);
         resolve2(port);
       });
       this.wss.on("error", (error) => {
@@ -1170,6 +1226,10 @@ var MGServer = class {
           this.handleRegister(ws, message);
           return;
         }
+        if (message.type === "track_analytics" /* TRACK_ANALYTICS */) {
+          this.handleTrackAnalytics(message);
+          return;
+        }
         const managedWs = ws;
         if (!managedWs.connectionId) {
           this.logger.warn("\u672A\u6CE8\u518C\u7684\u8FDE\u63A5\u53D1\u9001\u6D88\u606F\uFF0C\u5FFD\u7565");
@@ -1247,6 +1307,9 @@ var MGServer = class {
       case "navigate_to_node" /* NAVIGATE_TO_NODE */:
         this.handleNavigateToNode(ws, message);
         break;
+      case "track_analytics" /* TRACK_ANALYTICS */:
+        this.handleTrackAnalytics(message);
+        break;
       case "response" /* RESPONSE */:
       case "error" /* ERROR */:
         this.requestHandler.handleResponse(message);
@@ -1392,6 +1455,15 @@ var MGServer = class {
       }
     });
   }
+  /**
+   * 处理统计事件
+   */
+  handleTrackAnalytics(message) {
+    const eventData = message.data;
+    if (eventData) {
+      trackEvent(eventData);
+    }
+  }
   /**
    * 处理节点导航请求
    */
@@ -1465,6 +1537,7 @@ var MGServer = class {
       return;
     }
     this.logger.info("\u6B63\u5728\u505C\u6B62 Server...");
+    await shutdownServerAnalytics();
     this.requestHandler.cleanupAll();
     this.connectionManager.closeAll();
     return new Promise((resolve2) => {