@hangox/mg-cli 1.1.2 → 1.1.3

package/dist/daemon-runner.js

@@ -267,8 +267,8 @@ var ConnectionManager = class {
   allConnections = /* @__PURE__ */ new Map();
   /** 心跳检查定时器 */
   heartbeatTimer = null;
-  constructor(logger) {
-    this.logger = logger;
+  constructor(logger2) {
+    this.logger = logger2;
   }
   /**
    * 启动心跳检查
@@ -495,9 +495,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 请求
@@ -675,6 +675,60 @@ function getVersion() {
   }
 }
 
+// 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;
@@ -706,6 +760,7 @@ var MGServer = class {
         this.startedAt = /* @__PURE__ */ new Date();
         this.logger.info(`Server \u542F\u52A8\u6210\u529F\uFF0C\u76D1\u542C\u7AEF\u53E3: ${port2}`);
         this.connectionManager.startHeartbeatCheck(HEARTBEAT_INTERVAL);
+        initServerAnalytics(this.logger);
         resolve2(port2);
       });
       this.wss.on("error", (error) => {
@@ -765,6 +820,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");
@@ -842,6 +901,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);
@@ -987,6 +1049,15 @@ var MGServer = class {
       }
     });
   }
+  /**
+   * 处理统计事件
+   */
+  handleTrackAnalytics(message) {
+    const eventData = message.data;
+    if (eventData) {
+      trackEvent(eventData);
+    }
+  }
   /**
    * 处理节点导航请求
    */
@@ -1060,6 +1131,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) => {
@@ -1120,14 +1192,14 @@ async function startServerForeground(port2) {
     );
   }
   ensureConfigDir();
-  const logger = createLogger({
+  const logger2 = createLogger({
     console: true,
     file: true,
     minLevel: "INFO" /* INFO */
   });
   const server = createServer({
     port: port2 || DEFAULT_PORT,
-    logger
+    logger: logger2
   });
   const cleanup = async () => {
     console.log("\n\u6B63\u5728\u505C\u6B62 Server...");
@@ -1153,7 +1225,7 @@ MG Server \u542F\u52A8\u6210\u529F`);
     console.log(`
 \u6309 Ctrl+C \u505C\u6B62...`);
   } catch (error) {
-    logger.error("Server \u542F\u52A8\u5931\u8D25:", error);
+    logger2.error("Server \u542F\u52A8\u5931\u8D25:", error);
     throw error;
   }
 }