analyticsconfig

Author: RomneyDa

core/config/profile/doLoadConfig.ts

@@ -124,12 +124,14 @@ export default async function doLoadConfig(
 
   if (newConfig.analytics) {
     await TeamAnalytics.setup(
-      newConfig.analytics as any, // TODO: Need to get rid of index.d.ts once and for all
+      newConfig.analytics,
       uniqueId,
       ideInfo.extensionVersion,
       controlPlaneClient,
       controlPlaneProxyInfo,
     );
+  } else {
+    await TeamAnalytics.setup(undefined, "NOT_UNIQUE");
   }
 
   newConfig = await injectControlPlaneProxyInfo(

core/control-plane/TeamAnalytics.ts

@@ -10,9 +10,10 @@ import {
 import LogStashAnalyticsProvider from "./analytics/LogStashAnalyticsProvider.js";
 import PostHogAnalyticsProvider from "./analytics/PostHogAnalyticsProvider.js";
 import { ControlPlaneClient } from "./client.js";
+import { AnalyticsConfig } from "../index.js";
 
 function createAnalyticsProvider(
-  config: Analytics,
+  config: AnalyticsConfig,
 ): IAnalyticsProvider | undefined {
   // @ts-ignore
   switch (config.provider) {
@@ -42,7 +43,7 @@ export class TeamAnalytics {
   }
 
   static async setup(
-    config: Analytics,
+    config: AnalyticsConfig,
     uniqueId: string,
     extensionVersion: string,
     controlPlaneClient: ControlPlaneClient,
@@ -52,22 +53,27 @@ export class TeamAnalytics {
     TeamAnalytics.os = os.platform();
     TeamAnalytics.extensionVersion = extensionVersion;
 
-    if (!config) {
-      await TeamAnalytics.provider?.shutdown();
-      TeamAnalytics.provider = undefined;
-    } else {
-      TeamAnalytics.provider = createAnalyticsProvider(config);
-      await TeamAnalytics.provider?.setup(
-        config,
-        uniqueId,
-        controlPlaneProxyInfo,
-      );
+    TeamAnalytics.provider = createAnalyticsProvider(config);
+    await TeamAnalytics.provider?.setup(
+      config,
+      uniqueId,
+      controlPlaneProxyInfo,
+    );
 
-      if (config.provider === "continue-proxy") {
-        (
-          TeamAnalytics.provider as ContinueProxyAnalyticsProvider
-        ).controlPlaneClient = controlPlaneClient;
-      }
+    if (config.provider === "continue-proxy") {
+      (
+        TeamAnalytics.provider as ContinueProxyAnalyticsProvider
+      ).controlPlaneClient = controlPlaneClient;
+    }
+  }
+
+  static async shutdown() {
+    if (TeamAnalytics.provider) {
+      await TeamAnalytics.provider.shutdown();
+      TeamAnalytics.provider = undefined;
+      TeamAnalytics.os = undefined;
+      TeamAnalytics.extensionVersion = undefined;
+      TeamAnalytics.uniqueId = "NOT_UNIQUE";
     }
   }
 }