@bitfab/sdk 0.38.1 → 0.38.2

package/dist/autoTrace.cjs

@@ -0,0 +1,187 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/autoTrace.ts
+var autoTrace_exports = {};
+__export(autoTrace_exports, {
+  __bitfabAutoSpan: () => __bitfabAutoSpan,
+  __bitfabAutoTraceActive: () => __bitfabAutoTraceActive,
+  __bitfabAutoWrap: () => __bitfabAutoWrap,
+  __setBitfabAutoTraceCapturePolicy: () => __setBitfabAutoTraceCapturePolicy,
+  getAutoTraceCapturePolicy: () => getAutoTraceCapturePolicy,
+  runWithAutoTraceContext: () => runWithAutoTraceContext,
+  runWithAutoTraceRootContext: () => runWithAutoTraceRootContext
+});
+module.exports = __toCommonJS(autoTrace_exports);
+
+// src/asyncStorage.ts
+var AsyncLocalStorageClass = null;
+var initDone = false;
+function registerAsyncLocalStorageClass(cls) {
+  if (!AsyncLocalStorageClass) {
+    AsyncLocalStorageClass = cls;
+  }
+  initDone = true;
+}
+var asyncStorageReady = (typeof process !== "undefined" && process.versions?.node ? (
+  // The join trick hides "node:async_hooks" from static analysis so
+  // bundlers that ban Node.js built-ins don't fail at build time.
+  // webpackIgnore tells webpack/turbopack to emit a native import()
+  // so Node.js can resolve the module at runtime.
+  import(
+    /* webpackIgnore: true */
+    ["node", "async_hooks"].join(":")
+  ).then(
+    (mod) => {
+      registerAsyncLocalStorageClass(mod.AsyncLocalStorage);
+    }
+  ).catch(() => {
+  })
+) : Promise.resolve()).then(() => {
+  initDone = true;
+});
+function createAsyncLocalStorage() {
+  return AsyncLocalStorageClass ? new AsyncLocalStorageClass() : null;
+}
+
+// src/autoTrace.ts
+var autoTraceGlobal = globalThis;
+var autoTraceState = autoTraceGlobal.__bitfabAutoTraceStateV3 ?? {
+  storage: null,
+  browserScope: void 0,
+  capturePolicies: /* @__PURE__ */ new WeakMap(),
+  activeRoots: 0
+};
+autoTraceGlobal.__bitfabAutoTraceStateV3 = autoTraceState;
+function initializeAutoTraceStorage() {
+  autoTraceState.storage ?? (autoTraceState.storage = createAsyncLocalStorage());
+}
+function runWithAutoTraceContext(context, fn, depth = 0) {
+  initializeAutoTraceStorage();
+  const scope = { context, depth };
+  if (autoTraceState.storage) {
+    return autoTraceState.storage.run(scope, fn);
+  }
+  const previous = autoTraceState.browserScope;
+  autoTraceState.browserScope = scope;
+  try {
+    return fn();
+  } finally {
+    autoTraceState.browserScope = previous;
+  }
+}
+function runWithAutoTraceRootContext(context, fn) {
+  autoTraceState.activeRoots += 1;
+  let result;
+  try {
+    result = runWithAutoTraceContext(context, fn);
+  } catch (error) {
+    autoTraceState.activeRoots -= 1;
+    throw error;
+  }
+  if (isAutoTraceAsyncGenerator(result)) {
+    autoTraceState.activeRoots -= 1;
+    return wrapAutoTraceAsyncGenerator(context, result);
+  }
+  if (result instanceof Promise) {
+    return result.finally(() => {
+      autoTraceState.activeRoots -= 1;
+    });
+  }
+  autoTraceState.activeRoots -= 1;
+  return result;
+}
+function isAutoTraceAsyncGenerator(value) {
+  if (value === null || typeof value !== "object") {
+    return false;
+  }
+  const candidate = value;
+  return typeof candidate.next === "function" && typeof candidate.return === "function" && typeof candidate.throw === "function" && typeof candidate[Symbol.asyncIterator] === "function";
+}
+function wrapAutoTraceAsyncGenerator(context, source) {
+  const step = (method, value) => runWithAutoTraceRootContext(context, () => source[method](value));
+  const wrapped = {
+    next: (value) => step("next", value),
+    return: (value) => step("return", value),
+    throw: (error) => step("throw", error),
+    [Symbol.asyncIterator]: () => wrapped
+  };
+  return wrapped;
+}
+function __bitfabAutoSpan(definition, inputs, fn) {
+  const scope = currentAutoTraceScope();
+  if (!scope) {
+    return fn();
+  }
+  return scope.context.invoke(definition, inputs, fn, scope.depth);
+}
+function __bitfabAutoWrap(definition, fn) {
+  if (fn.name === "") {
+    const nameParts = definition.name.split(".");
+    const inferredName = nameParts[nameParts.length - 1];
+    if (inferredName !== void 0) {
+      Object.defineProperty(fn, "name", {
+        configurable: true,
+        value: inferredName
+      });
+    }
+  }
+  const target = fn;
+  return new Proxy(target, {
+    apply(callTarget, thisArg, args) {
+      if (!__bitfabAutoTraceActive()) {
+        return Reflect.apply(callTarget, thisArg, args);
+      }
+      return __bitfabAutoSpan(
+        definition,
+        args,
+        () => Reflect.apply(callTarget, thisArg, args)
+      );
+    }
+  });
+}
+function __bitfabAutoTraceActive() {
+  if (autoTraceState.activeRoots === 0) {
+    return false;
+  }
+  return currentAutoTraceScope() !== void 0;
+}
+function currentAutoTraceScope() {
+  initializeAutoTraceStorage();
+  return autoTraceState.storage?.getStore() ?? autoTraceState.browserScope;
+}
+function __setBitfabAutoTraceCapturePolicy(client, traceFunctionKey, functionIds) {
+  const policies = autoTraceState.capturePolicies.get(client) ?? /* @__PURE__ */ new Map();
+  policies.set(traceFunctionKey, new Set(functionIds));
+  autoTraceState.capturePolicies.set(client, policies);
+}
+function getAutoTraceCapturePolicy(client, traceFunctionKey) {
+  return autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey) ?? /* @__PURE__ */ new Set();
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  __bitfabAutoSpan,
+  __bitfabAutoTraceActive,
+  __bitfabAutoWrap,
+  __setBitfabAutoTraceCapturePolicy,
+  getAutoTraceCapturePolicy,
+  runWithAutoTraceContext,
+  runWithAutoTraceRootContext
+});
+//# sourceMappingURL=autoTrace.cjs.map

package/dist/autoTrace.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/autoTrace.ts","../src/asyncStorage.ts"],"sourcesContent":["import {\n  type AsyncLocalStorageLike,\n  createAsyncLocalStorage,\n} from \"./asyncStorage.js\"\n\nexport interface AutoTraceFunctionDefinition {\n  id: string\n  name: string\n  file: string\n  line: number\n  column: number\n  wrapper?: boolean\n}\n\nexport interface AutoTraceContext {\n  invoke<T>(\n    definition: AutoTraceFunctionDefinition,\n    inputs: unknown[],\n    fn: () => T,\n    depth: number,\n  ): T\n}\n\ninterface AutoTraceScope {\n  context: AutoTraceContext\n  depth: number\n}\n\ninterface AutoTraceState {\n  storage: AsyncLocalStorageLike<AutoTraceScope> | null\n  browserScope: AutoTraceScope | undefined\n  capturePolicies: WeakMap<object, Map<string, ReadonlySet<string>>>\n  activeRoots: number\n}\n\ninterface AutoTraceGlobal {\n  __bitfabAutoTraceStateV3?: AutoTraceState\n}\n\ninterface AutoTraceAsyncGenerator {\n  next(value?: unknown): Promise<IteratorResult<unknown, unknown>>\n  return(value?: unknown): Promise<IteratorResult<unknown, unknown>>\n  throw(error?: unknown): Promise<IteratorResult<unknown, unknown>>\n  [Symbol.asyncIterator](): AutoTraceAsyncGenerator\n}\n\nconst autoTraceGlobal = globalThis as unknown as AutoTraceGlobal\nconst autoTraceState: AutoTraceState =\n  autoTraceGlobal.__bitfabAutoTraceStateV3 ?? {\n    storage: null,\n    browserScope: undefined,\n    capturePolicies: new WeakMap(),\n    activeRoots: 0,\n  }\nautoTraceGlobal.__bitfabAutoTraceStateV3 = autoTraceState\n\nfunction initializeAutoTraceStorage(): void {\n  autoTraceState.storage ??= createAsyncLocalStorage<AutoTraceScope>()\n}\n\nexport function runWithAutoTraceContext<T>(\n  context: AutoTraceContext,\n  fn: () => T,\n  depth = 0,\n): T {\n  initializeAutoTraceStorage()\n  const scope = { context, depth }\n  if (autoTraceState.storage) {\n    return autoTraceState.storage.run(scope, fn)\n  }\n\n  const previous = autoTraceState.browserScope\n  autoTraceState.browserScope = scope\n  try {\n    return fn()\n  } finally {\n    autoTraceState.browserScope = previous\n  }\n}\n\nexport function runWithAutoTraceRootContext<T>(\n  context: AutoTraceContext,\n  fn: () => T,\n): T {\n  autoTraceState.activeRoots += 1\n  let result: T\n  try {\n    result = runWithAutoTraceContext(context, fn)\n  } catch (error) {\n    autoTraceState.activeRoots -= 1\n    throw error\n  }\n\n  if (isAutoTraceAsyncGenerator(result)) {\n    autoTraceState.activeRoots -= 1\n    return wrapAutoTraceAsyncGenerator(context, result) as T\n  }\n\n  if (result instanceof Promise) {\n    return result.finally(() => {\n      autoTraceState.activeRoots -= 1\n    }) as T\n  }\n\n  autoTraceState.activeRoots -= 1\n  return result\n}\n\nfunction isAutoTraceAsyncGenerator(\n  value: unknown,\n): value is AutoTraceAsyncGenerator {\n  if (value === null || typeof value !== \"object\") {\n    return false\n  }\n  const candidate = value as Record<PropertyKey, unknown>\n  return (\n    typeof candidate.next === \"function\" &&\n    typeof candidate.return === \"function\" &&\n    typeof candidate.throw === \"function\" &&\n    typeof candidate[Symbol.asyncIterator] === \"function\"\n  )\n}\n\nfunction wrapAutoTraceAsyncGenerator(\n  context: AutoTraceContext,\n  source: AutoTraceAsyncGenerator,\n): AutoTraceAsyncGenerator {\n  const step = (\n    method: \"next\" | \"return\" | \"throw\",\n    value?: unknown,\n  ): Promise<IteratorResult<unknown, unknown>> =>\n    runWithAutoTraceRootContext(context, () => source[method](value))\n  const wrapped: AutoTraceAsyncGenerator = {\n    next: (value) => step(\"next\", value),\n    return: (value) => step(\"return\", value),\n    throw: (error) => step(\"throw\", error),\n    [Symbol.asyncIterator]: () => wrapped,\n  }\n  return wrapped\n}\n\nexport function __bitfabAutoSpan<T>(\n  definition: AutoTraceFunctionDefinition,\n  inputs: unknown[],\n  fn: () => T,\n): T {\n  const scope = currentAutoTraceScope()\n  if (!scope) {\n    return fn()\n  }\n  return scope.context.invoke(definition, inputs, fn, scope.depth)\n}\n\n/**\n * Preserve a function's original call arguments for transform cases where\n * parameter bindings discard them, such as destructured arrow parameters.\n *\n * This helper is internal transform/runtime protocol. The proxy preserves the\n * target's callability, arity, async identity, and non-constructibility while\n * only allocating a trace closure beneath an active automatic trace root.\n *\n * @experimental The automatic tracing protocol may change.\n */\nexport function __bitfabAutoWrap<T extends (...args: never[]) => unknown>(\n  definition: AutoTraceFunctionDefinition,\n  fn: T,\n): T {\n  if (fn.name === \"\") {\n    const nameParts = definition.name.split(\".\")\n    const inferredName = nameParts[nameParts.length - 1]\n    if (inferredName !== undefined) {\n      Object.defineProperty(fn, \"name\", {\n        configurable: true,\n        value: inferredName,\n      })\n    }\n  }\n\n  const target = fn as unknown as (...args: unknown[]) => unknown\n  return new Proxy(target, {\n    apply(callTarget, thisArg, args) {\n      if (!__bitfabAutoTraceActive()) {\n        return Reflect.apply(callTarget, thisArg, args)\n      }\n      return __bitfabAutoSpan(definition, args, () =>\n        Reflect.apply(callTarget, thisArg, args),\n      )\n    },\n  }) as unknown as T\n}\n\n/**\n * Return whether the current call is inside an automatic trace root.\n *\n * Build transforms use this before allocating function metadata, captured\n * inputs, or an invocation closure. It is internal transform/runtime protocol,\n * not a supported application API.\n *\n * @experimental The automatic tracing protocol may change.\n */\nexport function __bitfabAutoTraceActive(): boolean {\n  if (autoTraceState.activeRoots === 0) {\n    return false\n  }\n  return currentAutoTraceScope() !== undefined\n}\n\nfunction currentAutoTraceScope(): AutoTraceScope | undefined {\n  initializeAutoTraceStorage()\n  return autoTraceState.storage?.getStore() ?? autoTraceState.browserScope\n}\n\nexport function __setBitfabAutoTraceCapturePolicy(\n  client: object,\n  traceFunctionKey: string,\n  functionIds: Iterable<string>,\n): void {\n  const policies = autoTraceState.capturePolicies.get(client) ?? new Map()\n  policies.set(traceFunctionKey, new Set(functionIds))\n  autoTraceState.capturePolicies.set(client, policies)\n}\n\nexport function getAutoTraceCapturePolicy(\n  client: object,\n  traceFunctionKey: string,\n): ReadonlySet<string> {\n  return (\n    autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey) ??\n    new Set()\n  )\n}\n","/**\n * Shared AsyncLocalStorage loader.\n *\n * Provides two ways to initialize AsyncLocalStorage:\n *\n * 1. **Synchronous registration** (preferred for Node.js):\n *    `asyncStorageNode.ts` calls `registerAsyncLocalStorageClass()` at module\n *    evaluation time, so the class is available immediately - no async gap.\n *    The `node.ts` entry point imports it before anything else.\n *\n * 2. **Async dynamic import** (fallback for the default entry point):\n *    Loads `node:async_hooks` via a bundler-safe dynamic import. This is used\n *    by the default `index.ts` entry point so the SDK works in browsers\n *    (where the import silently fails) and in Node.js when imported via the\n *    default entry point.\n *\n * ## Why the dynamic import looks like this\n *\n * We need to handle three environments:\n *\n * 1. **Pure Node.js** - `import(\"node:async_hooks\")` works natively.\n * 2. **Webpack/Turbopack (Next.js server)** - The bundler processes\n *    `import()` calls at build time. The `webpackIgnore` magic comment tells\n *    webpack (and turbopack) to emit a native `import()` call instead of\n *    trying to resolve it, so Node.js handles it at runtime.\n * 3. **Browsers / Edge** - The `process.versions?.node` guard prevents\n *    execution entirely. If it somehow runs, `.catch(() => {})` swallows\n *    the failure.\n */\n\nexport interface AsyncLocalStorageLike<T> {\n  getStore(): T | undefined\n  run<R>(store: T, fn: () => R): R\n}\n\nlet AsyncLocalStorageClass: (new () => AsyncLocalStorageLike<unknown>) | null =\n  null\nlet initDone = false\n\n/**\n * Register the AsyncLocalStorage class synchronously.\n *\n * Called by `asyncStorageNode.ts` at module evaluation time so the class\n * is available before any span is created - no async gap, no race condition.\n *\n * Safe to call multiple times; subsequent calls are no-ops.\n */\nexport function registerAsyncLocalStorageClass(\n  cls: new () => AsyncLocalStorageLike<unknown>,\n): void {\n  if (!AsyncLocalStorageClass) {\n    AsyncLocalStorageClass = cls\n  }\n  initDone = true\n}\n\n/**\n * Assert that AsyncLocalStorage was registered successfully.\n *\n * Called by `node.ts` after importing `asyncStorageNode.ts` to catch\n * import-order bugs at startup rather than silently degrading to the\n * browser fallback (flat spans with no nesting).\n *\n * This should ONLY be called from the Node.js entry point where we\n * know `node:async_hooks` must be available.\n */\nexport function assertAsyncStorageRegistered(): void {\n  if (!AsyncLocalStorageClass) {\n    console.warn(\n      \"Bitfab: AsyncLocalStorage not available - nested span context will not propagate.\",\n    )\n  }\n}\n\nexport const asyncStorageReady: Promise<void> = (\n  typeof process !== \"undefined\" && process.versions?.node\n    ? // The join trick hides \"node:async_hooks\" from static analysis so\n      // bundlers that ban Node.js built-ins don't fail at build time.\n      // webpackIgnore tells webpack/turbopack to emit a native import()\n      // so Node.js can resolve the module at runtime.\n      import(\n        /* webpackIgnore: true */\n        [\"node\", \"async_hooks\"].join(\":\")\n      )\n        .then(\n          (mod: {\n            AsyncLocalStorage: new () => AsyncLocalStorageLike<unknown>\n          }) => {\n            registerAsyncLocalStorageClass(mod.AsyncLocalStorage)\n          },\n        )\n        .catch(() => {})\n    : Promise.resolve()\n).then(() => {\n  initDone = true\n})\n\nexport function isAsyncStorageInitDone(): boolean {\n  return initDone\n}\n\nexport function createAsyncLocalStorage<T>(): AsyncLocalStorageLike<T> | null {\n  return AsyncLocalStorageClass\n    ? (new AsyncLocalStorageClass() as AsyncLocalStorageLike<T>)\n    : null\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACmCA,IAAI,yBACF;AACF,IAAI,WAAW;AAUR,SAAS,+BACd,KACM;AACN,MAAI,CAAC,wBAAwB;AAC3B,6BAAyB;AAAA,EAC3B;AACA,aAAW;AACb;AAoBO,IAAM,qBACX,OAAO,YAAY,eAAe,QAAQ,UAAU;AAAA;AAAA;AAAA;AAAA;AAAA,EAKhD;AAAA;AAAA,IAEE,CAAC,QAAQ,aAAa,EAAE,KAAK,GAAG;AAAA,IAE/B;AAAA,IACC,CAAC,QAEK;AACJ,qCAA+B,IAAI,iBAAiB;AAAA,IACtD;AAAA,EACF,EACC,MAAM,MAAM;AAAA,EAAC,CAAC;AAAA,IACjB,QAAQ,QAAQ,GACpB,KAAK,MAAM;AACX,aAAW;AACb,CAAC;AAMM,SAAS,0BAA8D;AAC5E,SAAO,yBACF,IAAI,uBAAuB,IAC5B;AACN;;;AD3DA,IAAM,kBAAkB;AACxB,IAAM,iBACJ,gBAAgB,4BAA4B;AAAA,EAC1C,SAAS;AAAA,EACT,cAAc;AAAA,EACd,iBAAiB,oBAAI,QAAQ;AAAA,EAC7B,aAAa;AACf;AACF,gBAAgB,2BAA2B;AAE3C,SAAS,6BAAmC;AAC1C,iBAAe,YAAf,eAAe,UAAY,wBAAwC;AACrE;AAEO,SAAS,wBACd,SACA,IACA,QAAQ,GACL;AACH,6BAA2B;AAC3B,QAAM,QAAQ,EAAE,SAAS,MAAM;AAC/B,MAAI,eAAe,SAAS;AAC1B,WAAO,eAAe,QAAQ,IAAI,OAAO,EAAE;AAAA,EAC7C;AAEA,QAAM,WAAW,eAAe;AAChC,iBAAe,eAAe;AAC9B,MAAI;AACF,WAAO,GAAG;AAAA,EACZ,UAAE;AACA,mBAAe,eAAe;AAAA,EAChC;AACF;AAEO,SAAS,4BACd,SACA,IACG;AACH,iBAAe,eAAe;AAC9B,MAAI;AACJ,MAAI;AACF,aAAS,wBAAwB,SAAS,EAAE;AAAA,EAC9C,SAAS,OAAO;AACd,mBAAe,eAAe;AAC9B,UAAM;AAAA,EACR;AAEA,MAAI,0BAA0B,MAAM,GAAG;AACrC,mBAAe,eAAe;AAC9B,WAAO,4BAA4B,SAAS,MAAM;AAAA,EACpD;AAEA,MAAI,kBAAkB,SAAS;AAC7B,WAAO,OAAO,QAAQ,MAAM;AAC1B,qBAAe,eAAe;AAAA,IAChC,CAAC;AAAA,EACH;AAEA,iBAAe,eAAe;AAC9B,SAAO;AACT;AAEA,SAAS,0BACP,OACkC;AAClC,MAAI,UAAU,QAAQ,OAAO,UAAU,UAAU;AAC/C,WAAO;AAAA,EACT;AACA,QAAM,YAAY;AAClB,SACE,OAAO,UAAU,SAAS,cAC1B,OAAO,UAAU,WAAW,cAC5B,OAAO,UAAU,UAAU,cAC3B,OAAO,UAAU,OAAO,aAAa,MAAM;AAE/C;AAEA,SAAS,4BACP,SACA,QACyB;AACzB,QAAM,OAAO,CACX,QACA,UAEA,4BAA4B,SAAS,MAAM,OAAO,MAAM,EAAE,KAAK,CAAC;AAClE,QAAM,UAAmC;AAAA,IACvC,MAAM,CAAC,UAAU,KAAK,QAAQ,KAAK;AAAA,IACnC,QAAQ,CAAC,UAAU,KAAK,UAAU,KAAK;AAAA,IACvC,OAAO,CAAC,UAAU,KAAK,SAAS,KAAK;AAAA,IACrC,CAAC,OAAO,aAAa,GAAG,MAAM;AAAA,EAChC;AACA,SAAO;AACT;AAEO,SAAS,iBACd,YACA,QACA,IACG;AACH,QAAM,QAAQ,sBAAsB;AACpC,MAAI,CAAC,OAAO;AACV,WAAO,GAAG;AAAA,EACZ;AACA,SAAO,MAAM,QAAQ,OAAO,YAAY,QAAQ,IAAI,MAAM,KAAK;AACjE;AAYO,SAAS,iBACd,YACA,IACG;AACH,MAAI,GAAG,SAAS,IAAI;AAClB,UAAM,YAAY,WAAW,KAAK,MAAM,GAAG;AAC3C,UAAM,eAAe,UAAU,UAAU,SAAS,CAAC;AACnD,QAAI,iBAAiB,QAAW;AAC9B,aAAO,eAAe,IAAI,QAAQ;AAAA,QAChC,cAAc;AAAA,QACd,OAAO;AAAA,MACT,CAAC;AAAA,IACH;AAAA,EACF;AAEA,QAAM,SAAS;AACf,SAAO,IAAI,MAAM,QAAQ;AAAA,IACvB,MAAM,YAAY,SAAS,MAAM;AAC/B,UAAI,CAAC,wBAAwB,GAAG;AAC9B,eAAO,QAAQ,MAAM,YAAY,SAAS,IAAI;AAAA,MAChD;AACA,aAAO;AAAA,QAAiB;AAAA,QAAY;AAAA,QAAM,MACxC,QAAQ,MAAM,YAAY,SAAS,IAAI;AAAA,MACzC;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAWO,SAAS,0BAAmC;AACjD,MAAI,eAAe,gBAAgB,GAAG;AACpC,WAAO;AAAA,EACT;AACA,SAAO,sBAAsB,MAAM;AACrC;AAEA,SAAS,wBAAoD;AAC3D,6BAA2B;AAC3B,SAAO,eAAe,SAAS,SAAS,KAAK,eAAe;AAC9D;AAEO,SAAS,kCACd,QACA,kBACA,aACM;AACN,QAAM,WAAW,eAAe,gBAAgB,IAAI,MAAM,KAAK,oBAAI,IAAI;AACvE,WAAS,IAAI,kBAAkB,IAAI,IAAI,WAAW,CAAC;AACnD,iBAAe,gBAAgB,IAAI,QAAQ,QAAQ;AACrD;AAEO,SAAS,0BACd,QACA,kBACqB;AACrB,SACE,eAAe,gBAAgB,IAAI,MAAM,GAAG,IAAI,gBAAgB,KAChE,oBAAI,IAAI;AAEZ;","names":[]}

package/dist/autoTrace.d.cts

@@ -0,0 +1,39 @@
+interface AutoTraceFunctionDefinition {
+    id: string;
+    name: string;
+    file: string;
+    line: number;
+    column: number;
+    wrapper?: boolean;
+}
+interface AutoTraceContext {
+    invoke<T>(definition: AutoTraceFunctionDefinition, inputs: unknown[], fn: () => T, depth: number): T;
+}
+declare function runWithAutoTraceContext<T>(context: AutoTraceContext, fn: () => T, depth?: number): T;
+declare function runWithAutoTraceRootContext<T>(context: AutoTraceContext, fn: () => T): T;
+declare function __bitfabAutoSpan<T>(definition: AutoTraceFunctionDefinition, inputs: unknown[], fn: () => T): T;
+/**
+ * Preserve a function's original call arguments for transform cases where
+ * parameter bindings discard them, such as destructured arrow parameters.
+ *
+ * This helper is internal transform/runtime protocol. The proxy preserves the
+ * target's callability, arity, async identity, and non-constructibility while
+ * only allocating a trace closure beneath an active automatic trace root.
+ *
+ * @experimental The automatic tracing protocol may change.
+ */
+declare function __bitfabAutoWrap<T extends (...args: never[]) => unknown>(definition: AutoTraceFunctionDefinition, fn: T): T;
+/**
+ * Return whether the current call is inside an automatic trace root.
+ *
+ * Build transforms use this before allocating function metadata, captured
+ * inputs, or an invocation closure. It is internal transform/runtime protocol,
+ * not a supported application API.
+ *
+ * @experimental The automatic tracing protocol may change.
+ */
+declare function __bitfabAutoTraceActive(): boolean;
+declare function __setBitfabAutoTraceCapturePolicy(client: object, traceFunctionKey: string, functionIds: Iterable<string>): void;
+declare function getAutoTraceCapturePolicy(client: object, traceFunctionKey: string): ReadonlySet<string>;
+
+export { type AutoTraceContext, type AutoTraceFunctionDefinition, __bitfabAutoSpan, __bitfabAutoTraceActive, __bitfabAutoWrap, __setBitfabAutoTraceCapturePolicy, getAutoTraceCapturePolicy, runWithAutoTraceContext, runWithAutoTraceRootContext };

package/dist/autoTrace.d.ts

@@ -0,0 +1,39 @@
+interface AutoTraceFunctionDefinition {
+    id: string;
+    name: string;
+    file: string;
+    line: number;
+    column: number;
+    wrapper?: boolean;
+}
+interface AutoTraceContext {
+    invoke<T>(definition: AutoTraceFunctionDefinition, inputs: unknown[], fn: () => T, depth: number): T;
+}
+declare function runWithAutoTraceContext<T>(context: AutoTraceContext, fn: () => T, depth?: number): T;
+declare function runWithAutoTraceRootContext<T>(context: AutoTraceContext, fn: () => T): T;
+declare function __bitfabAutoSpan<T>(definition: AutoTraceFunctionDefinition, inputs: unknown[], fn: () => T): T;
+/**
+ * Preserve a function's original call arguments for transform cases where
+ * parameter bindings discard them, such as destructured arrow parameters.
+ *
+ * This helper is internal transform/runtime protocol. The proxy preserves the
+ * target's callability, arity, async identity, and non-constructibility while
+ * only allocating a trace closure beneath an active automatic trace root.
+ *
+ * @experimental The automatic tracing protocol may change.
+ */
+declare function __bitfabAutoWrap<T extends (...args: never[]) => unknown>(definition: AutoTraceFunctionDefinition, fn: T): T;
+/**
+ * Return whether the current call is inside an automatic trace root.
+ *
+ * Build transforms use this before allocating function metadata, captured
+ * inputs, or an invocation closure. It is internal transform/runtime protocol,
+ * not a supported application API.
+ *
+ * @experimental The automatic tracing protocol may change.
+ */
+declare function __bitfabAutoTraceActive(): boolean;
+declare function __setBitfabAutoTraceCapturePolicy(client: object, traceFunctionKey: string, functionIds: Iterable<string>): void;
+declare function getAutoTraceCapturePolicy(client: object, traceFunctionKey: string): ReadonlySet<string>;
+
+export { type AutoTraceContext, type AutoTraceFunctionDefinition, __bitfabAutoSpan, __bitfabAutoTraceActive, __bitfabAutoWrap, __setBitfabAutoTraceCapturePolicy, getAutoTraceCapturePolicy, runWithAutoTraceContext, runWithAutoTraceRootContext };

package/dist/autoTrace.js

@@ -0,0 +1,20 @@
+import {
+  __bitfabAutoSpan,
+  __bitfabAutoTraceActive,
+  __bitfabAutoWrap,
+  __setBitfabAutoTraceCapturePolicy,
+  getAutoTraceCapturePolicy,
+  runWithAutoTraceContext,
+  runWithAutoTraceRootContext
+} from "./chunk-GFBQ2AMO.js";
+import "./chunk-H6LZRFMN.js";
+export {
+  __bitfabAutoSpan,
+  __bitfabAutoTraceActive,
+  __bitfabAutoWrap,
+  __setBitfabAutoTraceCapturePolicy,
+  getAutoTraceCapturePolicy,
+  runWithAutoTraceContext,
+  runWithAutoTraceRootContext
+};
+//# sourceMappingURL=autoTrace.js.map

package/dist/{chunk-WD4AO3BK.js → chunk-CWXCVY75.js}

@@ -1,22 +1,30 @@
+import {
+  __setBitfabAutoTraceCapturePolicy,
+  getAutoTraceCapturePolicy,
+  runWithAutoTraceContext,
+  runWithAutoTraceRootContext
+} from "./chunk-GFBQ2AMO.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
   HttpClient,
-  __privateAdd,
-  __privateGet,
-  __privateSet,
-  asyncStorageReady,
-  createAsyncLocalStorage,
   deserializeValue,
   getReplayContext,
-  isAsyncStorageInitDone,
   randomUuid,
   resolveMockValue,
   serializeValue,
   toJsonSafe,
   toJsonSafeReport,
   warnOnce
-} from "./chunk-MGA7ROIK.js";
+} from "./chunk-SU7EAKOV.js";
+import {
+  __privateAdd,
+  __privateGet,
+  __privateSet,
+  asyncStorageReady,
+  createAsyncLocalStorage,
+  isAsyncStorageInitDone
+} from "./chunk-H6LZRFMN.js";
 
 // src/processorPayload.ts
 var SERIALIZATION_DEGRADED_STEP = "serialization_degraded";
@@ -2337,6 +2345,14 @@ function readEnv(name) {
   }
   return void 0;
 }
+var DEFAULT_AUTO_TRACE_MAX_DEPTH = 30;
+var DEFAULT_AUTO_TRACE_MAX_SPANS = 500;
+var AUTO_TRACE_PROTOCOL = "ts-auto-v1";
+var AUTO_TRACE_POLICY_REFRESH_MS = 6e4;
+var AUTO_TRACE_POLICY_RETRY_MS = 1e4;
+function autoTraceLimit(value, fallback) {
+  return value !== void 0 && Number.isFinite(value) && value >= 0 ? Math.floor(value) : fallback;
+}
 var Bitfab = class {
   /**
    * Initialize the Bitfab client.
@@ -2346,6 +2362,7 @@ var Bitfab = class {
   constructor(config) {
     /** Gate the empty-key warning to fire at most once. */
     this.apiKeyWarned = false;
+    this.autoTracePolicyRefreshes = /* @__PURE__ */ new Map();
     /**
      * Mock overrides registered via {@link Bitfab.registerMockOverride}, applied
      * to every `replay` on this client (after any per-call `mockOverride`). In
@@ -2369,6 +2386,178 @@ var Bitfab = class {
       timeout: this.timeout
     });
   }
+  /**
+   * Decorate a class method as an automatically expanded trace root.
+   *
+   * Build instrumentation turns repository functions called beneath this
+   * method into nested spans. Every generated span preserves structure; only
+   * function IDs selected by the capture policy include inputs and output.
+   * Without a compatible build transform, this still records the decorated
+   * method as a normal rich root span but cannot discover child calls.
+   *
+   * @param traceFunctionKey - Groups traces and their capture policy.
+   * @param options - Root presentation, subtree bounds, and exclusions.
+   * @experimental Automatic child-call instrumentation is experimental.
+   */
+  trace(traceFunctionKey, options = {}) {
+    const decorator = (...args) => {
+      if (args.length === 3) {
+        const propertyKey = args[1];
+        const descriptor = args[2];
+        if (!descriptor || typeof descriptor.value !== "function") {
+          throw new BitfabError("@bitfab.trace can only decorate methods");
+        }
+        if (!this.explicitlyEnabled) {
+          return;
+        }
+        descriptor.value = this.createAutoTraceRoot(
+          traceFunctionKey,
+          String(propertyKey),
+          options,
+          descriptor.value
+        );
+        return;
+      }
+      const method = args[0];
+      const context = args[1];
+      if (typeof method !== "function" || context?.kind !== "method" || context.name === void 0) {
+        throw new BitfabError("@bitfab.trace can only decorate methods");
+      }
+      if (!this.explicitlyEnabled) {
+        return method;
+      }
+      return this.createAutoTraceRoot(
+        traceFunctionKey,
+        String(context.name),
+        options,
+        method
+      );
+    };
+    return decorator;
+  }
+  withTrace(traceFunctionKey, optionsOrFn, maybeFn) {
+    const options = typeof optionsOrFn === "function" ? {} : optionsOrFn;
+    const fn = typeof optionsOrFn === "function" ? optionsOrFn : maybeFn;
+    if (!fn) {
+      throw new BitfabError("bitfab.withTrace requires a function");
+    }
+    if (!this.explicitlyEnabled) {
+      return fn;
+    }
+    const name = fn.name !== "" ? fn.name : traceFunctionKey;
+    return this.createAutoTraceRoot(traceFunctionKey, name, options, fn);
+  }
+  createAutoTraceRoot(traceFunctionKey, name, options, fn) {
+    const self = this;
+    const maxDepth = autoTraceLimit(
+      options.maxDepth,
+      DEFAULT_AUTO_TRACE_MAX_DEPTH
+    );
+    const maxSpans = autoTraceLimit(
+      options.maxSpans,
+      DEFAULT_AUTO_TRACE_MAX_SPANS
+    );
+    const excluded = new Set(options.exclude ?? []);
+    const includeWrappers = options.includeWrappers ?? false;
+    const tracedRoot = this.withSpan(
+      traceFunctionKey,
+      { name: options.name ?? name, type: options.type ?? "custom" },
+      function(...args) {
+        const capturePolicy = getAutoTraceCapturePolicy(self, traceFunctionKey);
+        self.refreshAutoTraceCapturePolicy(traceFunctionKey);
+        let spansUsed = 0;
+        let truncated = false;
+        const warnTruncated = () => {
+          if (!truncated) {
+            truncated = true;
+            getCurrentTrace().setMetadata({
+              bitfabAutoTrace: {
+                protocol: AUTO_TRACE_PROTOCOL,
+                truncated: true,
+                maxDepth,
+                maxSpans
+              }
+            });
+          }
+          warnOnce(
+            `auto-trace-truncated:${traceFunctionKey}`,
+            `"${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 autoTraceContext = {
+          invoke(definition, inputs, invokeFn, depth) {
+            const nameParts = definition.name.split(".");
+            const simpleName = nameParts[nameParts.length - 1];
+            if (excluded.has(definition.name) || simpleName !== void 0 && excluded.has(simpleName) || definition.wrapper === true && !includeWrappers) {
+              return invokeFn();
+            }
+            if (depth >= maxDepth || spansUsed >= maxSpans) {
+              warnTruncated();
+              return invokeFn();
+            }
+            spansUsed += 1;
+            const childOptions = {
+              name: definition.name,
+              type: "function",
+              captureWhen: "nested",
+              functionId: definition.id,
+              captureContent: capturePolicy.has(definition.id),
+              autoTraceDefinition: definition
+            };
+            const tracedChild = self.withSpan(
+              traceFunctionKey,
+              childOptions,
+              (..._inputs) => runWithAutoTraceContext(autoTraceContext, invokeFn, depth + 1)
+            );
+            return tracedChild(...inputs);
+          }
+        };
+        return runWithAutoTraceRootContext(
+          autoTraceContext,
+          () => fn.apply(this, args)
+        );
+      }
+    );
+    const autoTraceRoot = function(...args) {
+      if (!self.isTracingEnabled()) {
+        return fn.apply(this, args);
+      }
+      return tracedRoot.apply(this, args);
+    };
+    Object.defineProperty(autoTraceRoot, "_bitfabTraceFunctionKey", {
+      value: traceFunctionKey
+    });
+    return autoTraceRoot;
+  }
+  refreshAutoTraceCapturePolicy(traceFunctionKey) {
+    const now = Date.now();
+    const state = this.autoTracePolicyRefreshes.get(traceFunctionKey) ?? {
+      refreshAfter: 0
+    };
+    if (state.inFlight || now < state.refreshAfter) {
+      return;
+    }
+    const request = this.httpClient.getAutoTracePolicy(
+      traceFunctionKey,
+      AUTO_TRACE_PROTOCOL
+    ).then((policy) => {
+      if (policy.protocol !== AUTO_TRACE_PROTOCOL) {
+        state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_RETRY_MS;
+        return;
+      }
+      const functionIds = Array.isArray(policy.functionIds) ? policy.functionIds.filter(
+        (id) => typeof id === "string" && id.startsWith(`${AUTO_TRACE_PROTOCOL}:`)
+      ).slice(0, DEFAULT_AUTO_TRACE_MAX_SPANS) : [];
+      __setBitfabAutoTraceCapturePolicy(this, traceFunctionKey, functionIds);
+      state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_REFRESH_MS;
+    }).catch(() => {
+      state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_RETRY_MS;
+    }).finally(() => {
+      state.inFlight = void 0;
+    });
+    state.inFlight = request;
+    this.autoTracePolicyRefreshes.set(traceFunctionKey, state);
+  }
   /**
    * Flush and permanently close this client's tracing resources: its pending
    * requests and the single span-transport worker shared by its decorators and
@@ -2915,7 +3104,10 @@ var Bitfab = class {
           parentSpanId,
           inputs,
           startedAt,
-          spanType: options.type ?? "custom"
+          spanType: options.type ?? "custom",
+          functionId: options.functionId,
+          captureContent: options.captureContent ?? true,
+          autoTraceDefinition: options.autoTraceDefinition
         };
         const sendSpan = async (params) => {
           const replayCtx = getReplayContext();
@@ -3080,7 +3272,16 @@ var Bitfab = class {
           }
         };
         executeWithContext = () => {
-          const result = fn.apply(this, args);
+          let result;
+          try {
+            result = fn.apply(this, args);
+          } catch (error) {
+            void sendSpan({
+              result: void 0,
+              error: error instanceof Error ? error.message : String(error)
+            });
+            throw error;
+          }
           if (result instanceof Promise) {
             return result.then((resolvedResult) => {
               recordSpan(resolvedResult);
@@ -3321,8 +3522,8 @@ var Bitfab = class {
    * Queued on the client's span transport; delivery is the transport's job.
    */
   sendWrapperSpan(params) {
-    const serializedInputs = serializeValue(params.inputs);
-    const serializedResult = serializeValue(params.result);
+    const serializedInputs = params.captureContent ? serializeValue(params.inputs) : void 0;
+    const serializedResult = params.captureContent ? serializeValue(params.result) : void 0;
     const externalSpan = {
       id: params.spanId,
       trace_id: params.traceId,
@@ -3331,26 +3532,38 @@ var Bitfab = class {
       span_data: {
         name: params.spanName,
         type: params.spanType,
-        input: serializedInputs.json,
-        output: serializedResult.json,
-        // Include superjson meta for type preservation
-        ...serializedInputs.meta !== void 0 && {
-          input_meta: serializedInputs.meta
+        ...params.functionId !== void 0 && {
+          function_id: params.functionId,
+          content_captured: params.captureContent
         },
-        ...serializedResult.meta !== void 0 && {
-          output_meta: serializedResult.meta
+        ...params.autoTraceDefinition !== void 0 && {
+          function_file: params.autoTraceDefinition.file,
+          function_line: params.autoTraceDefinition.line,
+          function_column: params.autoTraceDefinition.column
+        },
+        ...serializedInputs !== void 0 && {
+          input: serializedInputs.json,
+          ...serializedInputs.meta !== void 0 && {
+            input_meta: serializedInputs.meta
+          }
+        },
+        ...serializedResult !== void 0 && {
+          output: serializedResult.json,
+          ...serializedResult.meta !== void 0 && {
+            output_meta: serializedResult.meta
+          }
         },
         ...params.functionName !== void 0 && {
           function_name: params.functionName
         },
-        ...params.error !== void 0 && {
+        ...params.captureContent && params.error !== void 0 && {
           error: params.error,
           error_source: "code"
         },
-        ...params.contexts && params.contexts.length > 0 && {
+        ...params.captureContent && params.contexts && params.contexts.length > 0 && {
           contexts: params.contexts
         },
-        ...params.prompt !== void 0 && { prompt: params.prompt }
+        ...params.captureContent && params.prompt !== void 0 && { prompt: params.prompt }
       }
     };
     if (params.parentSpanId) {
@@ -3395,7 +3608,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-WSFDZ2UV.js");
+    const { replay: doReplay } = await import("./replay-A5UVFNZ3.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -3633,4 +3846,4 @@ export {
   BitfabFunction,
   finalizers
 };
-//# sourceMappingURL=chunk-WD4AO3BK.js.map
+//# sourceMappingURL=chunk-CWXCVY75.js.map