taist 0.2.3 → 0.2.5

package/lib/trace-reporter.js

@@ -247,6 +247,15 @@ export class TraceReporter extends EventEmitter {
   report(trace) {
     if (this.closed) return;
 
+    // Update socketPath from env if not set (env may be set after construction)
+    // This handles the case where vitest-reporter sets TAIST_COLLECTOR_SOCKET
+    // after the reporter was created (e.g., in bundled code loaded early)
+    if (!this.socketPath && process.env.TAIST_COLLECTOR_SOCKET) {
+      this.socketPath = process.env.TAIST_COLLECTOR_SOCKET;
+      logger.debug("[reporter] Late socketPath update from env:", this.socketPath);
+      this._setupExitHandlers();
+    }
+
     logger.debug("[reporter] report() called:", trace.name, trace.type);
 
     this.buffer.push(trace);
@@ -412,17 +421,20 @@ export class TraceReporter extends EventEmitter {
   }
 }
 
-// Global reporter instance for easy access from instrumented code
-let globalReporter = null;
+// Global reporter instance key for cross-module singleton
+// Using globalThis ensures the same instance is shared even when the module
+// is bundled separately (e.g., in Rollup/Vite builds)
+const TAIST_REPORTER_KEY = '__taist_global_reporter__';
 
 /**
- * Get or create the global reporter instance
+ * Get or create the global reporter instance.
+ * Uses globalThis to ensure singleton works across bundled modules.
  */
 export function getGlobalReporter(options = {}) {
-  if (!globalReporter) {
-    globalReporter = new TraceReporter(options);
+  if (!globalThis[TAIST_REPORTER_KEY]) {
+    globalThis[TAIST_REPORTER_KEY] = new TraceReporter(options);
   }
-  return globalReporter;
+  return globalThis[TAIST_REPORTER_KEY];
 }
 
 /**

package/lib/transform.js

@@ -150,6 +150,15 @@ const __taist_debug = process.env.TAIST_DEBUG === 'true';
 if (process.env.TAIST_COLLECTOR_SOCKET && !__taist_reporter.isConnected()) {
   __taist_reporter.connectEager();
 }
+// Helper to find correlationId from GraphQL context argument (Apollo: parent, args, context, info)
+const __taist_findCorrelationIdInArgs = (args) => {
+  for (const arg of args) {
+    if (arg && typeof arg === 'object' && arg.taistCorrelationId) {
+      return arg.taistCorrelationId;
+    }
+  }
+  return null;
+};
 const __taist_wrap = (fn, name) => {
   if (typeof fn !== 'function') return fn;
   if (__taist_debug) console.log('[taist] wrapping:', name);
@@ -158,8 +167,10 @@ const __taist_wrap = (fn, name) => {
     const parentCtx = __taist_getContext();
     const id = __taist_generateId();
     const depth = parentCtx.depth;
-    // Get correlationId from parent context or fallback global
-    const correlationId = parentCtx.correlationId || __taist_getCorrelationId();
+    // Get correlationId: 1) parent context, 2) GraphQL context arg, 3) fallback global
+    const argsCorrelationId = __taist_findCorrelationIdInArgs(args);
+    const correlationId = parentCtx.correlationId || argsCorrelationId || __taist_getCorrelationId();
+    if (__taist_debug && argsCorrelationId) console.log('[taist] Found correlationId in args:', argsCorrelationId);
     const newCtx = {
       depth: depth + 1,
       traceId: parentCtx.traceId || id,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "taist",
-  "version": "0.2.3",
+  "version": "0.2.5",
   "description": "Token-Optimized Testing Framework for AI-Assisted Development",
   "main": "index.js",
   "type": "module",