taist 0.1.14 → 0.1.15

package/lib/trace-context.js

@@ -5,22 +5,35 @@
  * across function calls without explicit parameter passing.
  *
  * This is the same approach used by OpenTelemetry, Datadog, and other APM tools.
+ *
+ * IMPORTANT: Uses globalThis to share context across bundled modules.
+ * Without this, each bundle would have its own AsyncLocalStorage instance,
+ * breaking context propagation across bundle boundaries.
  */
 
 import { AsyncLocalStorage } from 'async_hooks';
 
-// Global storage for trace context
-export const traceContext = new AsyncLocalStorage();
+// Use globalThis to share context across all bundles/modules
+// This ensures trace context propagates even when code is bundled separately
+const TAIST_CONTEXT_KEY = '__taist_trace_context__';
+const TAIST_COUNTER_KEY = '__taist_id_counter__';
+
+if (!globalThis[TAIST_CONTEXT_KEY]) {
+  globalThis[TAIST_CONTEXT_KEY] = new AsyncLocalStorage();
+}
+if (globalThis[TAIST_COUNTER_KEY] === undefined) {
+  globalThis[TAIST_COUNTER_KEY] = 0;
+}
 
-// Counter for generating unique IDs
-let idCounter = 0;
+// Global storage for trace context (shared via globalThis)
+export const traceContext = globalThis[TAIST_CONTEXT_KEY];
 
 /**
  * Generate a unique trace/span ID
  * @returns {string} Unique identifier
  */
 export function generateId() {
-  return `__${++idCounter}_${Date.now()}`;
+  return `__${++globalThis[TAIST_COUNTER_KEY]}_${Date.now()}`;
 }
 
 /**
@@ -76,5 +89,5 @@ export function startTrace(fn) {
  * Reset the ID counter (for testing)
  */
 export function resetIdCounter() {
-  idCounter = 0;
+  globalThis[TAIST_COUNTER_KEY] = 0;
 }

package/package.json

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