bitfab 0.52.4 → 0.52.6

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

@@ -1,21 +1,25 @@
 import {
   __bitfabAutoTraceActive,
   __setBitfabAutoTraceCapturePolicy,
+  clearAutoTraceNodeConfiguration,
+  currentAutoTraceEntries,
   getAutoTraceCapturePolicy,
-  runWithAutoTraceContext,
+  invokeThroughAutoTraceEntries,
+  runWithAutoTraceEntry,
   runWithAutoTraceNodeConfiguration,
   runWithAutoTraceRootContext
-} from "./chunk-ZUD7OFYB.js";
+} from "./chunk-E5BLYL5X.js";
 import {
   NO_MOCK_OVERRIDE,
   deserializeInputs,
   deserializeValue,
+  gitRunner,
   randomUuid,
   resolveMockValue,
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-U2D7ACC4.js";
+} from "./chunk-DRTHTSHO.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -23,10 +27,11 @@ import {
   MixedTracingError,
   callerTraceMetadata,
   getReplayContext,
+  readEnv,
   recordCallerTraceMetadata,
   retireTraceMetadata,
   warnOnce
-} from "./chunk-34B7PVAM.js";
+} from "./chunk-NXC6NYIY.js";
 import {
   __privateAdd,
   __privateGet,
@@ -1017,24 +1022,237 @@ function mixedTracingError(enteredApi, entered, enclosing, traceFunctionKey) {
     `opt-in and opt-out tracing can't be mixed in one call stack: ${enteredApi} (${entered})${subject} was entered inside a ${SURFACE_API[enclosing]} call (${enclosing}). ${MIXED_SURFACE_REMEDY[entered]}`
   );
 }
-function assertSurfacesCompatible(requested, resolved, parentSurface, traceFunctionKey) {
+function assertSurfacesCompatible(requested, resolved2, parentSurface, traceFunctionKey) {
   if (requested === "inherit" || requested === "neutral") {
     return;
   }
-  if (resolved === void 0 || parentSurface === void 0) {
+  if (resolved2 === void 0 || parentSurface === void 0) {
     return;
   }
-  if (parentSurface === resolved) {
+  if (parentSurface === resolved2) {
     return;
   }
   throw mixedTracingError(
-    SURFACE_API[resolved],
-    resolved,
+    SURFACE_API[resolved2],
+    resolved2,
     parentSurface,
     traceFunctionKey
   );
 }
 
+// src/commitRef.ts
+var EXPLICIT_SHA_ENV = "BITFAB_COMMIT_SHA";
+var DISABLE_ENV = "BITFAB_DISABLE_COMMIT_REF";
+var GIT_TIMEOUT_MS = 2e3;
+var GIT_MAX_BUFFER = 1024 * 1024;
+var VERCEL_PROVIDER_HOSTS = {
+  github: "github.com",
+  gitlab: "gitlab.com",
+  bitbucket: "bitbucket.org"
+};
+function read(env, name) {
+  if (name === null) {
+    return null;
+  }
+  const value = env(name)?.trim();
+  return value ? value : null;
+}
+function remoteFrom(...names) {
+  return (env) => {
+    const parts = names.map((name) => read(env, name));
+    if (parts.some((part) => part === null)) {
+      return null;
+    }
+    return normalizeRemote(parts.join("/"));
+  };
+}
+function vercelRemote(env) {
+  const provider = read(env, "VERCEL_GIT_PROVIDER")?.toLowerCase() ?? "";
+  const host = VERCEL_PROVIDER_HOSTS[provider];
+  const owner = read(env, "VERCEL_GIT_REPO_OWNER");
+  const slug = read(env, "VERCEL_GIT_REPO_SLUG");
+  if (!host || !owner || !slug) {
+    return null;
+  }
+  return `${host}/${owner}/${slug}`;
+}
+function noRemote() {
+  return null;
+}
+var PLATFORM_ENVS = [
+  ["VERCEL_GIT_COMMIT_SHA", "VERCEL_GIT_COMMIT_REF", vercelRemote],
+  [
+    "GITHUB_SHA",
+    "GITHUB_REF_NAME",
+    remoteFrom("GITHUB_SERVER_URL", "GITHUB_REPOSITORY")
+  ],
+  ["RAILWAY_GIT_COMMIT_SHA", "RAILWAY_GIT_BRANCH", noRemote],
+  ["RENDER_GIT_COMMIT", "RENDER_GIT_BRANCH", noRemote],
+  ["SOURCE_VERSION", null, noRemote],
+  ["CF_PAGES_COMMIT_SHA", "CF_PAGES_BRANCH", noRemote],
+  ["CI_COMMIT_SHA", "CI_COMMIT_REF_NAME", remoteFrom("CI_REPOSITORY_URL")],
+  [
+    "BUILD_SOURCEVERSION",
+    "BUILD_SOURCEBRANCHNAME",
+    remoteFrom("BUILD_REPOSITORY_URI")
+  ],
+  ["CIRCLE_SHA1", "CIRCLE_BRANCH", remoteFrom("CIRCLE_REPOSITORY_URL")]
+];
+var COMMIT_REF_ENV_NAMES = [
+  EXPLICIT_SHA_ENV,
+  DISABLE_ENV,
+  ...PLATFORM_ENVS.flatMap(
+    ([sha, branch]) => branch === null ? [sha] : [sha, branch]
+  ),
+  "GITHUB_SERVER_URL",
+  "GITHUB_REPOSITORY",
+  "CI_REPOSITORY_URL",
+  "BUILD_REPOSITORY_URI",
+  "CIRCLE_REPOSITORY_URL",
+  "VERCEL_GIT_PROVIDER",
+  "VERCEL_GIT_REPO_OWNER",
+  "VERCEL_GIT_REPO_SLUG"
+];
+function normalizeRemote(raw) {
+  const value = raw?.trim();
+  if (!value) {
+    return null;
+  }
+  let host = "";
+  let path = value;
+  if (value.includes("://")) {
+    try {
+      const parsed = new URL(value);
+      host = parsed.hostname;
+      path = parsed.pathname;
+    } catch {
+      return null;
+    }
+  } else {
+    const separator = value.indexOf(":");
+    if (separator !== -1) {
+      const head = value.slice(0, separator);
+      host = head.slice(head.lastIndexOf("@") + 1);
+      path = value.slice(separator + 1);
+    }
+  }
+  path = path.replace(/^\/+|\/+$/g, "");
+  if (path.endsWith(".git")) {
+    path = path.slice(0, -".git".length).replace(/\/+$/, "");
+  }
+  if (!host) {
+    return path || null;
+  }
+  return path ? `${host}/${path}` : host;
+}
+function resolveCommitRefFromEnv(env) {
+  const explicit = read(env, EXPLICIT_SHA_ENV);
+  const platform = PLATFORM_ENVS.find(([sha]) => read(env, sha) !== null);
+  if (!platform) {
+    if (explicit === null) {
+      return null;
+    }
+    return {
+      sha: explicit,
+      branch: null,
+      dirty: null,
+      remote: null,
+      rootSha: null
+    };
+  }
+  const [shaEnv, branchEnv, readRemote] = platform;
+  return {
+    sha: explicit ?? read(env, shaEnv),
+    branch: read(env, branchEnv),
+    dirty: null,
+    remote: readRemote(env),
+    rootSha: null
+  };
+}
+async function resolveCommitRefFromGit(cwd) {
+  const git = await gitRunner({
+    timeoutMs: GIT_TIMEOUT_MS,
+    maxBuffer: GIT_MAX_BUFFER,
+    keepProcessAlive: false
+  });
+  if (!git) {
+    return null;
+  }
+  const run = async (args) => (await git(cwd, args))?.trim() ?? null;
+  const sha = await run(["rev-parse", "HEAD"]);
+  if (!sha) {
+    return null;
+  }
+  const [branch, status, remote, roots] = await Promise.all([
+    run(["symbolic-ref", "--short", "-q", "HEAD"]),
+    run(["status", "--porcelain"]),
+    run(["remote", "get-url", "origin"]),
+    run(["rev-list", "--max-parents=0", "HEAD"])
+  ]);
+  return {
+    sha,
+    branch: branch || null,
+    dirty: status === null ? null : status.length > 0,
+    remote: normalizeRemote(remote),
+    rootSha: roots ? roots.split(/\s+/).sort()[0] ?? null : null
+  };
+}
+function commitRefWire(ref2) {
+  return {
+    sha: ref2.sha,
+    branch: ref2.branch,
+    dirty: ref2.dirty,
+    remote: ref2.remote,
+    root_sha: ref2.rootSha
+  };
+}
+var resolved = false;
+var ref = null;
+var gitStarted = false;
+var settled = Promise.resolve();
+function startCommitRefResolution() {
+  if (resolved || gitStarted) {
+    return;
+  }
+  if (read(readEnv, DISABLE_ENV) !== null) {
+    resolved = true;
+    return;
+  }
+  const fromEnv = resolveCommitRefFromEnv(readEnv);
+  if (fromEnv) {
+    ref = fromEnv;
+    resolved = true;
+    return;
+  }
+  gitStarted = true;
+  if (typeof process === "undefined") {
+    resolved = true;
+    return;
+  }
+  let cwd;
+  try {
+    cwd = process.cwd?.() ?? ".";
+  } catch {
+    resolved = true;
+    return;
+  }
+  settled = resolveCommitRefFromGit(cwd).then(
+    (fromGit) => {
+      if (!resolved) {
+        ref = fromGit;
+        resolved = true;
+      }
+    },
+    () => {
+      resolved = true;
+    }
+  );
+}
+function currentCommitRef() {
+  startCommitRefResolution();
+  return ref;
+}
+
 // src/datasets.ts
 var DEFAULT_RERUN_TIMEOUT_MS = 9e4;
 var DEFAULT_RERUN_POLL_INTERVAL_MS = 1e3;
@@ -1512,11 +1730,11 @@ var BitfabLangGraphCallbackHandler = class {
         this.invocations.set(invocation.rootRunId, invocation);
       }
       if (!willHide) {
-        let resolved = parentSpan;
-        while (resolved?.hidden === true) {
-          resolved = resolved.parentId ? this.runToSpan.get(resolved.parentId) : void 0;
+        let resolved2 = parentSpan;
+        while (resolved2?.hidden === true) {
+          resolved2 = resolved2.parentId ? this.runToSpan.get(resolved2.parentId) : void 0;
         }
-        effectiveParentId = resolved ? resolved.spanId : invocation.activeContext?.spanId ?? null;
+        effectiveParentId = resolved2 ? resolved2.spanId : invocation.activeContext?.spanId ?? null;
       } else {
         effectiveParentId = parentRunId ?? null;
       }
@@ -2948,7 +3166,7 @@ function getCurrentTrace() {
     }
   };
 }
-function readEnv(name) {
+function readEnv2(name) {
   if (typeof process !== "undefined" && process.env) {
     return process.env[name];
   }
@@ -2995,6 +3213,7 @@ var Bitfab = class {
       validateDbSnapshotConfig(config.dbSnapshot);
     }
     this.dbSnapshot = config.dbSnapshot;
+    startCommitRefResolution();
     this.httpClient = new HttpClient({
       apiKey: () => this.resolveApiKey(),
       serviceUrl: this.serviceUrl,
@@ -3165,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;
@@ -3190,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;
@@ -3225,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
@@ -3330,7 +3635,7 @@ var Bitfab = class {
       return this.resolvedApiKey;
     }
     const fromConfig = typeof this.apiKeyConfig === "function" ? this.apiKeyConfig() : this.apiKeyConfig;
-    const candidate = fromConfig && fromConfig.trim() !== "" ? fromConfig : readEnv("BITFAB_API_KEY");
+    const candidate = fromConfig && fromConfig.trim() !== "" ? fromConfig : readEnv2("BITFAB_API_KEY");
     const key = candidate && candidate.trim() !== "" ? candidate : void 0;
     if (key) {
       this.resolvedApiKey = key;
@@ -3841,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(
@@ -3890,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();
@@ -4088,15 +4399,15 @@ var Bitfab = class {
                 );
               }
               return runWithSpanStack(newStack, async () => {
-                const resolved = await resolution;
-                if (resolved.matched) {
+                const resolved2 = await resolution;
+                if (resolved2.matched) {
                   void sendSpan({
-                    result: resolved.output,
+                    result: resolved2.output,
                     mocked: true,
                     mockTarget: "output",
                     mockSource: "override"
                   });
-                  return resolved.output;
+                  return resolved2.output;
                 }
                 if (shouldMockWithBaseStrategy && !mockSpan) {
                   throw new BitfabError(
@@ -4332,6 +4643,10 @@ var Bitfab = class {
     if (params.dbSnapshotRef) {
       rawTrace.db_snapshot_ref = params.dbSnapshotRef;
     }
+    const commitRef = currentCommitRef();
+    if (commitRef) {
+      rawTrace.commit_ref = commitRefWire(commitRef);
+    }
     if (params.ingestionType) {
       rawTrace.ingestion_type = params.ingestionType;
     }
@@ -4396,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 && {
@@ -4577,7 +4902,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-RFNQ7LSU.js");
+        const { flushTraces: flushTraces2 } = await import("./http-MGES3ZVX.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -4636,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-ZOTYMBMD.js");
+    const { replay: doReplay } = await import("./replay-5XES2LBL.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -4920,7 +5245,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-RFNQ7LSU.js");
+  const { flushTraces: flushTraces2 } = await import("./http-MGES3ZVX.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -4957,4 +5282,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-AUROMI67.js.map
+//# sourceMappingURL=chunk-P26KVTI5.js.map