bitfab 0.52.5 → 0.52.6

package/dist/{chunk-EMRHO6YW.js → chunk-P26KVTI5.js}

@@ -1,11 +1,14 @@
 import {
   __bitfabAutoTraceActive,
   __setBitfabAutoTraceCapturePolicy,
+  clearAutoTraceNodeConfiguration,
+  currentAutoTraceEntries,
   getAutoTraceCapturePolicy,
-  runWithAutoTraceContext,
+  invokeThroughAutoTraceEntries,
+  runWithAutoTraceEntry,
   runWithAutoTraceNodeConfiguration,
   runWithAutoTraceRootContext
-} from "./chunk-ZUD7OFYB.js";
+} from "./chunk-E5BLYL5X.js";
 import {
   NO_MOCK_OVERRIDE,
   deserializeInputs,
@@ -16,7 +19,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-6DRFQJ75.js";
+} from "./chunk-DRTHTSHO.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -28,7 +31,7 @@ import {
   recordCallerTraceMetadata,
   retireTraceMetadata,
   warnOnce
-} from "./chunk-CAE7HE4C.js";
+} from "./chunk-NXC6NYIY.js";
 import {
   __privateAdd,
   __privateGet,
@@ -3381,11 +3384,15 @@ var Bitfab = class {
       type: options.type ?? "custom",
       surface: "opt-out"
     };
-    const tracedRoot = this.withSpan(
+    const rootName = options.name ?? name;
+    const buildTracedRoot = (spanOptions) => this.withSpan(
       traceFunctionKey,
-      rootOptions,
+      spanOptions,
       function(...args) {
-        const capturePolicy = getAutoTraceCapturePolicy(self, traceFunctionKey);
+        const capturePolicy = getAutoTraceCapturePolicy(
+          self,
+          traceFunctionKey
+        );
         self.refreshAutoTraceCapturePolicy(traceFunctionKey);
         let spansUsed = 0;
         let truncated = false;
@@ -3406,19 +3413,54 @@ var Bitfab = class {
             `"${traceFunctionKey}" hit an automatic subtree capture limit (maxDepth=${maxDepth}, maxSpans=${maxSpans}); its trace is incomplete. Raise the limits or narrow the subtree with exclude.`
           );
         };
+        const spanStackOf = (entry) => entry.spanStack;
+        const openChild = (entry, childOptions, inputs, isAsync, invokeFn) => {
+          const runInside = () => runWithAutoTraceEntry(
+            entry,
+            {
+              context: autoTraceContext,
+              depth: entry.depth + 1,
+              spanStack: getSpanStack()
+            },
+            invokeFn
+          );
+          return runWithSpanStack(spanStackOf(entry), () => {
+            if (isAsync) {
+              const tracedAsyncChild = self.withSpan(
+                traceFunctionKey,
+                childOptions,
+                async (..._inputs) => await runInside()
+              );
+              return tracedAsyncChild(...inputs);
+            }
+            const tracedChild = self.withSpan(
+              traceFunctionKey,
+              childOptions,
+              (..._inputs) => runInside()
+            );
+            return tracedChild(...inputs);
+          });
+        };
+        const overBudget = (entry) => {
+          if (entry.depth < maxDepth && spansUsed < maxSpans) {
+            return false;
+          }
+          runWithSpanStack(spanStackOf(entry), warnTruncated);
+          return true;
+        };
         const autoTraceContext = {
-          invoke(definition, inputs, invokeFn, depth, nodeConfiguration) {
+          traceFunctionKey,
+          invoke(definition, inputs, invokeFn, entry, nodeConfiguration) {
             const nameParts = definition.name.split(".");
             const simpleName = nameParts[nameParts.length - 1];
-            const invokeWithoutNode = () => nodeConfiguration === void 0 ? invokeFn() : runWithAutoTraceContext(autoTraceContext, invokeFn, depth);
+            const invokeWithoutNode = () => nodeConfiguration === void 0 ? invokeFn() : clearAutoTraceNodeConfiguration(invokeFn);
             if (excluded.has(definition.name) || simpleName !== void 0 && excluded.has(simpleName) || nodeConfiguration === void 0 && definition.wrapper === true && !includeWrappers) {
               return invokeWithoutNode();
             }
             if (nodeConfiguration?.capture === false) {
-              return runWithAutoTraceContext(autoTraceContext, invokeFn, depth);
+              return clearAutoTraceNodeConfiguration(invokeFn);
             }
-            if (depth >= maxDepth || spansUsed >= maxSpans) {
-              warnTruncated();
+            if (overBudget(entry)) {
               return invokeWithoutNode();
             }
             spansUsed += 1;
@@ -3441,34 +3483,81 @@ var Bitfab = class {
                 finalize: nodeConfiguration.finalize
               }
             };
-            const invokeWithAutoTraceContext = () => runWithAutoTraceContext(autoTraceContext, invokeFn, depth + 1);
-            if (definition.async === true) {
-              const tracedAsyncChild = self.withSpan(
-                traceFunctionKey,
-                childOptions,
-                async (..._inputs) => await invokeWithAutoTraceContext()
-              );
-              return tracedAsyncChild(...inputs);
-            }
-            const tracedChild = self.withSpan(
-              traceFunctionKey,
+            return openChild(
+              entry,
               childOptions,
-              (..._inputs) => invokeWithAutoTraceContext()
+              inputs,
+              definition.async === true,
+              invokeFn
             );
-            return tracedChild(...inputs);
+          },
+          invokeNestedRoot(nestedName, nestedFunctionName, inputs, invokeFn, entry, link) {
+            if (excluded.has(nestedName) || excluded.has(nestedFunctionName) || overBudget(entry)) {
+              return invokeFn();
+            }
+            spansUsed += 1;
+            const childOptions = {
+              name: nestedName,
+              type: "function",
+              captureWhen: "nested",
+              surface: "opt-out",
+              captureContent: true,
+              ...link !== void 0 && { nestedTrace: link }
+            };
+            return openChild(entry, childOptions, inputs, false, invokeFn);
           }
         };
         return runWithAutoTraceRootContext(
           autoTraceContext,
+          getSpanStack(),
           () => fn.apply(this, args)
         );
       }
     );
+    const tracedRoot = buildTracedRoot(rootOptions);
     const autoTraceRoot = function(...args) {
       if (!self.shouldRecord()) {
         return fn.apply(this, args);
       }
-      return tracedRoot.apply(this, args);
+      const enclosing = currentAutoTraceEntries();
+      if (enclosing.length === 0) {
+        return tracedRoot.apply(this, args);
+      }
+      const link = getReplayContext() || inSeedScope() ? void 0 : { traceFunctionKey };
+      return invokeThroughAutoTraceEntries(
+        enclosing,
+        (entry, next) => entry.context.invokeNestedRoot(
+          rootName,
+          name,
+          args,
+          next,
+          entry,
+          link
+        ),
+        () => {
+          if (link === void 0) {
+            return fn.apply(this, args);
+          }
+          const innermost = enclosing[enclosing.length - 1];
+          const innermostNow = currentAutoTraceEntries().find(
+            (entry) => entry.context === innermost.context
+          ) ?? innermost;
+          const innermostStack = innermostNow.spanStack;
+          const copy = innermostStack[innermostStack.length - 1];
+          const detachedRoot = buildTracedRoot({
+            ...rootOptions,
+            ...copy !== void 0 && {
+              enclosingTrace: {
+                traceId: copy.traceId,
+                spanId: copy.spanId,
+                traceFunctionKey: innermost.context.traceFunctionKey,
+                link
+              }
+            }
+          });
+          return runWithSpanStack([], () => detachedRoot.apply(this, args));
+        }
+      );
     };
     Object.defineProperty(autoTraceRoot, "_bitfabTraceFunctionKey", {
       value: traceFunctionKey
@@ -4057,6 +4146,10 @@ var Bitfab = class {
         const spanId = randomUuid();
         const parentSpanId = parentContext?.spanId ?? null;
         const isRootSpan = parentSpanId === null;
+        if (options.enclosingTrace !== void 0) {
+          options.enclosingTrace.link.traceId = traceId;
+          options.enclosingTrace.link.rootSpanId = spanId;
+        }
         const requestedSurface = options.surface ?? DEFAULT_SURFACE;
         const surface = resolveSurface(requestedSurface, parentContext?.surface);
         assertSurfacesCompatible(
@@ -4106,7 +4199,9 @@ var Bitfab = class {
           spanType: options.type ?? "custom",
           functionId: options.functionId,
           captureContent: options.captureContent ?? true,
-          autoTraceDefinition: options.autoTraceDefinition
+          autoTraceDefinition: options.autoTraceDefinition,
+          nestedTrace: options.nestedTrace,
+          enclosingTrace: options.enclosingTrace
         };
         const sendSpan = async (params) => {
           const replayCtx = getReplayContext();
@@ -4616,6 +4711,16 @@ var Bitfab = class {
           function_line: params.autoTraceDefinition.line,
           function_column: params.autoTraceDefinition.column
         },
+        ...params.nestedTrace?.traceId !== void 0 && {
+          nested_trace_id: params.nestedTrace.traceId,
+          nested_trace_function_key: params.nestedTrace.traceFunctionKey,
+          nested_root_span_id: params.nestedTrace.rootSpanId
+        },
+        ...params.enclosingTrace !== void 0 && {
+          enclosing_trace_id: params.enclosingTrace.traceId,
+          enclosing_trace_function_key: params.enclosingTrace.traceFunctionKey,
+          enclosing_span_id: params.enclosingTrace.spanId
+        },
         ...serializedInputs !== void 0 && {
           input: serializedInputs.json,
           ...serializedInputs.meta !== void 0 && {
@@ -4797,7 +4902,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-Q6UEKZRF.js");
+        const { flushTraces: flushTraces2 } = await import("./http-MGES3ZVX.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -4856,7 +4961,7 @@ var Bitfab = class {
         `Function is wrapped with trace function key '${wrappedKey}' but replay was called with '${traceFunctionKey}'. Pass matching keys, or pass the unwrapped function to replay it under the explicit key.`
       );
     }
-    const { replay: doReplay } = await import("./replay-UVSZFZW7.js");
+    const { replay: doReplay } = await import("./replay-5XES2LBL.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -5140,7 +5245,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-Q6UEKZRF.js");
+  const { flushTraces: flushTraces2 } = await import("./http-MGES3ZVX.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -5177,4 +5282,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-EMRHO6YW.js.map
+//# sourceMappingURL=chunk-P26KVTI5.js.map