@hangox/mg-cli 1.1.2 → 1.1.3

package/dist/cli.js

@@ -670,8 +670,8 @@ var ConnectionManager = class {
   allConnections = /* @__PURE__ */ new Map();
   /** 心跳检查定时器 */
   heartbeatTimer = null;
-  constructor(logger) {
-    this.logger = logger;
+  constructor(logger2) {
+    this.logger = logger2;
   }
   /**
    * 启动心跳检查
@@ -898,9 +898,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 请求
@@ -1043,6 +1043,60 @@ var RequestHandler = class {
   }
 };
 
+// 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;
@@ -1074,6 +1128,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);
         resolve9(port);
       });
       this.wss.on("error", (error) => {
@@ -1133,6 +1188,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");
@@ -1210,6 +1269,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);
@@ -1355,6 +1417,15 @@ var MGServer = class {
       }
     });
   }
+  /**
+   * 处理统计事件
+   */
+  handleTrackAnalytics(message) {
+    const eventData = message.data;
+    if (eventData) {
+      trackEvent(eventData);
+    }
+  }
   /**
    * 处理节点导航请求
    */
@@ -1428,6 +1499,7 @@ var MGServer = class {
       return;
     }
     this.logger.info("\u6B63\u5728\u505C\u6B62 Server...");
+    await shutdownServerAnalytics();
     this.requestHandler.cleanupAll();
     this.connectionManager.closeAll();
     return new Promise((resolve9) => {
@@ -1488,14 +1560,14 @@ async function startServerForeground(port) {
     );
   }
   ensureConfigDir();
-  const logger = createLogger({
+  const logger2 = createLogger({
     console: true,
     file: true,
     minLevel: "INFO" /* INFO */
   });
   const server = createServer({
     port: port || DEFAULT_PORT,
-    logger
+    logger: logger2
   });
   const cleanup = async () => {
     console.log("\n\u6B63\u5728\u505C\u6B62 Server...");
@@ -1521,7 +1593,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;
   }
 }
@@ -1781,14 +1853,11 @@ var MGClient = class {
 };
 
 // src/cli/analytics.ts
-import { PostHog } from "posthog-node";
+import { WebSocket as WebSocket3 } from "ws";
 import { createHash } from "crypto";
 import { hostname, userInfo, platform, release } from "os";
-var POSTHOG_API_KEY = "phc_cSZZIxZOPhWOlsw92Wz8tmKQ5z8eQYsI3m9aM2Ujia9";
-var POSTHOG_HOST = "https://us.i.posthog.com";
-var posthogClient = null;
 var machineId = null;
-function isTelemetryDisabled() {
+function isTelemetryDisabled2() {
   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 getMachineId() {
@@ -1805,39 +1874,50 @@ function getMachineId() {
   }
   return machineId;
 }
-function getClient() {
-  if (isTelemetryDisabled()) {
-    return null;
-  }
-  if (!posthogClient) {
-    posthogClient = new PostHog(POSTHOG_API_KEY, {
-      host: POSTHOG_HOST,
-      // CLI 工具使用立即刷新模式
-      flushAt: 1,
-      flushInterval: 0
+function sendToServer(eventData) {
+  try {
+    const serverInfo = readServerInfo();
+    if (!serverInfo) {
+      return;
+    }
+    if (!isProcessRunning(serverInfo.pid)) {
+      return;
+    }
+    const ws = new WebSocket3(`ws://localhost:${serverInfo.port}`);
+    const timeout = setTimeout(() => {
+      ws.close();
+    }, 1e3);
+    ws.on("open", () => {
+      const message = {
+        type: "track_analytics" /* TRACK_ANALYTICS */,
+        data: eventData
+      };
+      ws.send(JSON.stringify(message));
+      clearTimeout(timeout);
+      ws.close();
     });
+    ws.on("error", () => {
+      clearTimeout(timeout);
+    });
+  } catch {
   }
-  return posthogClient;
 }
 function trackCommand(properties) {
-  const client = getClient();
-  if (!client) {
+  if (isTelemetryDisabled2()) {
     return;
   }
-  try {
-    client.capture({
-      distinctId: getMachineId(),
-      event: "cli_command",
-      properties: {
-        ...properties,
-        version: getVersion(),
-        os: platform(),
-        os_release: release(),
-        node_version: process.version
-      }
-    });
-  } catch {
-  }
+  const eventData = {
+    distinctId: getMachineId(),
+    event: "cli_command",
+    properties: {
+      ...properties,
+      version: getVersion(),
+      os: platform(),
+      os_release: release(),
+      node_version: process.version
+    }
+  };
+  sendToServer(eventData);
 }
 function createCommandTracker(commandName) {
   const startTime = Date.now();