evergreen-sdk 1.0.0 → 1.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "evergreen-sdk",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Evergreen Trace Backend SDK",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -17,4 +17,4 @@
     "@types/ws": "^8.5.5",
     "typescript": "^5.0.0"
   }
-}
+}

package/src/index.ts

@@ -23,6 +23,19 @@ let ws: WebSocket | null = null;
 let isConnected = false;
 const pendingTelemetry: any[] = [];
 
+function getTelemetryWsUrl(): string {
+  const explicit = process.env.EVERGREEN_TRACE_WS_URL;
+  if (explicit && explicit.trim()) return explicit.trim();
+  return "ws://localhost:4321";
+}
+
+function shouldDebugTelemetryConnection(): boolean {
+  return (
+    process.env.EVERGREEN_TRACE_DEBUG === "1" ||
+    process.env.EVERGREEN_TRACE_DEBUG === "true"
+  );
+}
+
 // Connect to the VS Code Extension Hub
 function connect() {
   if (
@@ -31,11 +44,23 @@ function connect() {
   )
     return;
 
+  const url = getTelemetryWsUrl();
+  const debug = shouldDebugTelemetryConnection();
+
   try {
-    ws = new WebSocket("ws://localhost:4321");
+    if (debug) {
+      // eslint-disable-next-line no-console
+      console.log(`🌲 Evergreen Trace: connecting to ${url}`);
+    }
+
+    ws = new WebSocket(url);
 
     ws.on("open", () => {
       isConnected = true;
+      if (debug) {
+        // eslint-disable-next-line no-console
+        console.log(`🌲 Evergreen Trace: connected to ${url}`);
+      }
       // Flush any telemetry that was emitted before the socket became ready.
       while (pendingTelemetry.length > 0) {
         const queued = pendingTelemetry.shift();