@bitfab/sdk 0.76.2 → 0.76.3

package/dist/{babelPlugin-BpEeVvk4.cjs → babelPlugin-C4ERQ8DW.cjs}

@@ -1,3 +1,5 @@
+let fs = require("fs");
+let path = require("path");
 //#region ../bitfab-typescript-transform/src/transformOptions.ts
 const AUTO_TRACE_FUNCTION_ID_VERSION = "ts-auto-v1";
 const AUTO_TRACE_TRANSFORM_MARKER = "__bitfab_auto_trace_transformed_v1__";
@@ -28,43 +30,43 @@ function propertyName(node, types) {
 	if (types.isStringLiteral(node)) return node.value;
 	return null;
 }
-function functionName(path, types) {
-	const node = path.node;
+function functionName(path$1, types) {
+	const node = path$1.node;
 	if (types.isFunctionDeclaration(node) || types.isFunctionExpression(node)) {
 		if (node.id) return node.id.name;
 	}
 	if ((types.isClassMethod(node) || types.isObjectMethod(node)) && !node.computed && types.isIdentifier(node.key)) return node.key.name;
-	const parent = path.parentPath?.node;
+	const parent = path$1.parentPath?.node;
 	if (types.isVariableDeclarator(parent)) return propertyName(parent.id, types) ?? "anonymous";
 	if (types.isObjectProperty(parent) && !parent.computed) return propertyName(parent.key, types) ?? "anonymous";
 	if (types.isClassProperty(parent) && !parent.computed) {
 		const name = propertyName(parent.key, types);
 		if (name) return name;
 	}
-	const { line, column } = sourcePosition(path);
+	const { line, column } = sourcePosition(path$1);
 	return `anonymous-${line}-${column}`;
 }
 const JEV_METHOD = "systemOne";
-function isJevMethodDefinition(path, types) {
-	const parent = path.parentPath?.node;
-	return (types.isClassMethod(path.node) || types.isObjectMethod(path.node) || types.isObjectProperty(parent) || types.isClassProperty(parent)) && functionName(path, types) === JEV_METHOD;
+function isJevMethodDefinition(path$2, types) {
+	const parent = path$2.parentPath?.node;
+	return (types.isClassMethod(path$2.node) || types.isObjectMethod(path$2.node) || types.isObjectProperty(parent) || types.isClassProperty(parent)) && functionName(path$2, types) === JEV_METHOD;
 }
 function isInstrumentableFunction(node, types) {
 	if (node.generator || types.isClassPrivateMethod(node)) return false;
 	return !(types.isClassMethod(node) || types.isObjectMethod(node)) || node.kind === "method";
 }
-function sourcePosition(path) {
-	const start = ((path.parentPath?.isExportDeclaration() ? path.parentPath.node : void 0) ?? path.node).loc?.start;
+function sourcePosition(path$3) {
+	const start = ((path$3.parentPath?.isExportDeclaration() ? path$3.parentPath.node : void 0) ?? path$3.node).loc?.start;
 	return {
 		line: start?.line ?? 0,
 		column: start?.column ?? 0
 	};
 }
-function qualifiedFunctionName(path, types) {
-	const ownName = functionName(path, types);
+function qualifiedFunctionName(path$4, types) {
+	const ownName = functionName(path$4, types);
 	if (ownName.startsWith("anonymous-")) return ownName;
 	const scopes = [];
-	let ancestor = path.parentPath;
+	let ancestor = path$4.parentPath;
 	while (ancestor) {
 		if (ancestor.isFunction()) {
 			if (!isInstrumentableFunction(ancestor.node, types)) {
@@ -88,9 +90,9 @@ function qualifiedFunctionName(path, types) {
 	}
 	return [...scopes, ownName].join(".");
 }
-function isTraceDecoratorRoot(path, types) {
-	const node = path.node;
-	const parent = path.parentPath?.node;
+function isTraceDecoratorRoot(path$5, types) {
+	const node = path$5.node;
+	const parent = path$5.parentPath?.node;
 	const decorated = types.isClassMethod(node) || types.isClassPrivateMethod(node) ? node : types.isClassProperty(parent) && parent.value === node ? parent : null;
 	if (!decorated) return false;
 	return decorated.decorators?.some((decorator) => {
@@ -100,11 +102,11 @@ function isTraceDecoratorRoot(path, types) {
 		return types.isMemberExpression(callee) && !callee.computed && types.isIdentifier(callee.property, { name: "trace" });
 	}) ?? false;
 }
-function withNodeFunctionName(path, types) {
-	if (path.node.arguments.length > 2) return null;
-	const callee = path.node.callee;
+function withNodeFunctionName(path$6, types) {
+	if (path$6.node.arguments.length > 2) return null;
+	const callee = path$6.node.callee;
 	if (!types.isMemberExpression(callee) || callee.computed || !types.isIdentifier(callee.property, { name: "withNode" })) return null;
-	const fn = path.node.arguments[path.node.arguments.length - 1];
+	const fn = path$6.node.arguments[path$6.node.arguments.length - 1];
 	if (types.isIdentifier(fn)) return fn.name;
 	if (types.isMemberExpression(fn) && !fn.computed) return propertyName(fn.property, types);
 	if (types.isFunctionExpression(fn) && fn.id) return fn.id.name;
@@ -113,13 +115,55 @@ function withNodeFunctionName(path, types) {
 const BAML_GENERATED_HEADER = "This file was generated by BAML";
 const BAML_CLIENT_CLASSES = /* @__PURE__ */ new Set(["BamlAsyncClient", "BamlSyncClient"]);
 const BAML_OPTIONS_PARAMETER = "__baml_options__";
+const BAML_PACKAGE = "@boundaryml/baml";
+const DEFAULT_BAML_REGISTRATION_MODULE = "@bitfab/sdk/baml";
+function importsBamlAtRuntime(program, types) {
+	return program.body.some((statement) => types.isImportDeclaration(statement) && statement.source.value === BAML_PACKAGE && statement.importKind !== "type" && statement.specifiers.some((specifier) => !types.isImportSpecifier(specifier) || specifier.importKind !== "type"));
+}
+const SDK_PACKAGE_PATH = [
+	"node_modules",
+	"@bitfab",
+	"sdk",
+	"package.json"
+];
+const BAML_REGISTRATION_EXPORT = "./baml";
+const sdkExportsBamlRegistrationByDirectory = /* @__PURE__ */ new Map();
+function readPackageJson(path$7) {
+	try {
+		return JSON.parse((0, fs.readFileSync)(path$7, "utf8"));
+	} catch {
+		return;
+	}
+}
+function exportsBamlRegistration(packageJson) {
+	if (typeof packageJson !== "object" || packageJson === null) return false;
+	const exports = packageJson.exports;
+	return typeof exports === "object" && exports !== null && BAML_REGISTRATION_EXPORT in exports;
+}
+function nearestSdkExportsBamlRegistration(directory) {
+	const cached = sdkExportsBamlRegistrationByDirectory.get(directory);
+	if (cached !== void 0) return cached;
+	const packageJson = readPackageJson((0, path.join)(directory, ...SDK_PACKAGE_PATH));
+	const parent = (0, path.dirname)(directory);
+	const exported = packageJson !== void 0 ? exportsBamlRegistration(packageJson) : parent !== directory && nearestSdkExportsBamlRegistration(parent);
+	sdkExportsBamlRegistrationByDirectory.set(directory, exported);
+	return exported;
+}
+function defaultRegistrationResolves(filename) {
+	return (0, path.isAbsolute)(filename) && nearestSdkExportsBamlRegistration((0, path.dirname)(filename));
+}
+function bamlRegistrationModuleFor(filename, configured) {
+	if (configured === false) return null;
+	if (configured !== void 0) return configured;
+	return defaultRegistrationResolves(filename) ? DEFAULT_BAML_REGISTRATION_MODULE : null;
+}
 function isBamlGeneratedSource(code) {
 	return code.slice(0, 4096).includes(BAML_GENERATED_HEADER);
 }
-function isBamlClientMethod(path, types) {
-	const node = path.node;
+function isBamlClientMethod(path$8, types) {
+	const node = path$8.node;
 	if (!types.isClassMethod(node) || node.kind !== "method" || node.static) return false;
-	const classPath = path.parentPath?.parentPath;
+	const classPath = path$8.parentPath?.parentPath;
 	if (!classPath || !(classPath.isClassDeclaration() || classPath.isClassExpression()) || !classPath.node.id || !BAML_CLIENT_CLASSES.has(classPath.node.id.name)) return false;
 	const parameters = runtimeParameters(node, types);
 	const last = parameters[parameters.length - 1];
@@ -131,10 +175,10 @@ function bamlOptionsDeclaration(callerOptions, withCollector, types) {
 function runtimeParameters(node, types) {
 	return node.params.filter((parameter) => !types.isIdentifier(parameter, { name: "this" }));
 }
-function parameterInputs(path, types) {
+function parameterInputs(path$9, types) {
 	const elements = [];
-	for (const [index, parameter] of runtimeParameters(path.node, types).entries()) {
-		if (!types.isArrowFunctionExpression(path.node)) {
+	for (const [index, parameter] of runtimeParameters(path$9.node, types).entries()) {
+		if (!types.isArrowFunctionExpression(path$9.node)) {
 			if (types.isRestElement(parameter)) elements.push(types.spreadElement(types.callExpression(types.memberExpression(types.arrayExpression([types.spreadElement(types.identifier("arguments"))]), types.identifier("slice")), [types.numericLiteral(index)])));
 			else elements.push(types.memberExpression(types.identifier("arguments"), types.numericLiteral(index), true));
 			continue;
@@ -144,11 +188,11 @@ function parameterInputs(path, types) {
 	}
 	return types.arrayExpression(elements);
 }
-function needsArgumentWrapper(path, types) {
-	return types.isArrowFunctionExpression(path.node) && path.node.params.some((parameter) => !types.isIdentifier(parameter) && !(types.isRestElement(parameter) && types.isIdentifier(parameter.argument)));
+function needsArgumentWrapper(path$10, types) {
+	return types.isArrowFunctionExpression(path$10.node) && path$10.node.params.some((parameter) => !types.isIdentifier(parameter) && !(types.isRestElement(parameter) && types.isIdentifier(parameter.argument)));
 }
-function isRestWrapper(path, types) {
-	const parameters = runtimeParameters(path.node, types);
+function isRestWrapper(path$11, types) {
+	const parameters = runtimeParameters(path$11.node, types);
 	return parameters.length === 1 && types.isRestElement(parameters[0]);
 }
 function isNamedMember(node, name, types) {
@@ -250,13 +294,13 @@ function recordedCallTarget(node, types) {
 function isOptionalChain(node, types) {
 	return types.isOptionalMemberExpression(node) || types.isOptionalCallExpression(node);
 }
-function continuesOptionalChain(path, types) {
-	const parent = path.parent;
-	return types.isOptionalMemberExpression(parent) && parent.object === path.node || types.isOptionalCallExpression(parent) && parent.callee === path.node;
+function continuesOptionalChain(path$12, types) {
+	const parent = path$12.parent;
+	return types.isOptionalMemberExpression(parent) && parent.object === path$12.node || types.isOptionalCallExpression(parent) && parent.callee === path$12.node;
 }
-function rewriteRecordedClientCall(path, state, types) {
+function rewriteRecordedClientCall(path$13, state, types) {
 	if (state.skipFile || state.bamlGenerated) return;
-	const node = path.node;
+	const node = path$13.node;
 	const target = recordedCallTarget(node, types);
 	const identifier = target === null ? void 0 : state.recordedCallIdentifiers?.get(target.call.runtimeExport);
 	if (target === null || identifier === void 0) return;
@@ -264,10 +308,10 @@ function rewriteRecordedClientCall(path, state, types) {
 	const optionalReceiver = types.isOptionalMemberExpression(node.callee) && node.callee.optional;
 	const optionalMethod = types.isOptionalCallExpression(node) && node.optional;
 	if (types.isOptionalCallExpression(node)) {
-		if (continuesOptionalChain(path, types) || !optionalReceiver && isOptionalChain(receiver, types)) return;
+		if (continuesOptionalChain(path$13, types) || !optionalReceiver && isOptionalChain(receiver, types)) return;
 	}
 	const guarded = optionalReceiver || optionalMethod;
-	const receiverReference = guarded && !path.scope.isStatic(receiver) ? path.scope.generateDeclaredUidIdentifier(recordedCall.receiverHint) : null;
+	const receiverReference = guarded && !path$13.scope.isStatic(receiver) ? path$13.scope.generateDeclaredUidIdentifier(recordedCall.receiverHint) : null;
 	const receiverValue = () => receiverReference === null ? types.cloneNode(receiver) : types.cloneNode(receiverReference);
 	const call = types.callExpression(types.cloneNode(identifier), recordedCall.endpoint === void 0 ? [receiverValue(), ...node.arguments] : [
 		types.stringLiteral(recordedCall.endpoint),
@@ -278,14 +322,14 @@ function rewriteRecordedClientCall(path, state, types) {
 	state.usedRuntime = true;
 	state.usedRecordedCalls?.add(recordedCall.runtimeExport);
 	if (!guarded) {
-		path.replaceWith(call);
+		path$13.replaceWith(call);
 		return;
 	}
 	const firstReceiver = receiverReference === null ? types.cloneNode(receiver) : types.assignmentExpression("=", types.cloneNode(receiverReference), receiver);
 	const missing = (value) => types.binaryExpression("==", value, types.nullLiteral());
 	const methodOf = (value) => types.memberExpression(value, types.identifier(method));
 	const test = optionalReceiver ? optionalMethod ? types.logicalExpression("||", missing(firstReceiver), missing(methodOf(receiverValue()))) : missing(firstReceiver) : missing(methodOf(firstReceiver));
-	path.replaceWith(types.conditionalExpression(test, types.unaryExpression("void", types.numericLiteral(0)), call));
+	path$13.replaceWith(types.conditionalExpression(test, types.unaryExpression("void", types.numericLiteral(0)), call));
 }
 function bitfabBabelPlugin(api) {
 	const { types } = api;
@@ -293,22 +337,27 @@ function bitfabBabelPlugin(api) {
 		name: "bitfab-auto-trace",
 		visitor: {
 			Program: {
-				enter(path, state) {
-					state.skipFile = !shouldTransformBitfabFile(state.file.opts.filename ?? "unknown", state.opts) || path.scope.hasBinding("__bitfab_auto_trace_transformed_v1__") || path.node.directives.some((directive) => directive.value.value === "use client");
-					state.runtimeIdentifier = path.scope.generateUidIdentifier("bitfabAutoSpan");
-					state.activeIdentifier = path.scope.generateUidIdentifier("bitfabAutoTraceActive");
-					state.wrapperIdentifier = path.scope.generateUidIdentifier("bitfabAutoWrap");
-					state.recordedCallIdentifiers = new Map(RECORDED_CALL_RUNTIMES.map((call) => [call.runtimeExport, path.scope.generateUidIdentifier(call.identifierHint)]));
+				enter(path$14, state) {
+					state.skipFile = !shouldTransformBitfabFile(state.file.opts.filename ?? "unknown", state.opts) || path$14.scope.hasBinding("__bitfab_auto_trace_transformed_v1__") || path$14.node.directives.some((directive) => directive.value.value === "use client");
+					state.runtimeIdentifier = path$14.scope.generateUidIdentifier("bitfabAutoSpan");
+					state.activeIdentifier = path$14.scope.generateUidIdentifier("bitfabAutoTraceActive");
+					state.wrapperIdentifier = path$14.scope.generateUidIdentifier("bitfabAutoWrap");
+					state.recordedCallIdentifiers = new Map(RECORDED_CALL_RUNTIMES.map((call) => [call.runtimeExport, path$14.scope.generateUidIdentifier(call.identifierHint)]));
 					state.bamlGenerated = isBamlGeneratedSource(state.file.code ?? "");
 					state.definitions = [];
 					state.usedRuntime = false;
 					state.usedWrapper = false;
 					state.usedRecordedCalls = /* @__PURE__ */ new Set();
 				},
-				exit(path, state) {
+				exit(path$15, state) {
+					const bamlRegistrationModule = !state.skipFile && state.bamlGenerated && importsBamlAtRuntime(path$15.node, types) ? bamlRegistrationModuleFor(sourceFilename(state.file.opts.filename ?? ""), state.opts.bamlRegistrationModule) : null;
+					if (bamlRegistrationModule !== null) {
+						path$15.unshiftContainer("body", types.importDeclaration([], types.stringLiteral(bamlRegistrationModule)));
+						state.file.metadata[AUTO_TRACE_TRANSFORM_MARKER] = true;
+					}
 					if (state.skipFile || !state.usedRuntime || !state.runtimeIdentifier || !state.activeIdentifier || !state.wrapperIdentifier) return;
 					state.file.metadata[AUTO_TRACE_TRANSFORM_MARKER] = true;
-					path.unshiftContainer("body", [
+					path$15.unshiftContainer("body", [
 						types.importDeclaration([
 							types.importSpecifier(state.runtimeIdentifier, types.identifier("__bitfabAutoSpan")),
 							types.importSpecifier(state.activeIdentifier, types.identifier("__bitfabAutoTraceActive")),
@@ -323,30 +372,30 @@ function bitfabBabelPlugin(api) {
 					]);
 				}
 			},
-			CallExpression: { exit(path, state) {
+			CallExpression: { exit(path$16, state) {
 				if (state.skipFile) return;
-				if (recordedCallTarget(path.node, types) !== null) {
-					rewriteRecordedClientCall(path, state, types);
+				if (recordedCallTarget(path$16.node, types) !== null) {
+					rewriteRecordedClientCall(path$16, state, types);
 					return;
 				}
-				const configuredName = withNodeFunctionName(path, types);
+				const configuredName = withNodeFunctionName(path$16, types);
 				if (configuredName === null) return;
-				if (path.node.arguments.length === 1) path.node.arguments.push(types.identifier("undefined"));
-				path.node.arguments.push(types.stringLiteral(configuredName));
+				if (path$16.node.arguments.length === 1) path$16.node.arguments.push(types.identifier("undefined"));
+				path$16.node.arguments.push(types.stringLiteral(configuredName));
 			} },
-			OptionalCallExpression: { exit(path, state) {
-				rewriteRecordedClientCall(path, state, types);
+			OptionalCallExpression: { exit(path$17, state) {
+				rewriteRecordedClientCall(path$17, state, types);
 			} },
-			Function: { exit(path, state) {
-				const node = path.node;
-				if (state.skipFile || node._bitfabInstrumented || !isInstrumentableFunction(node, types) || isTraceDecoratorRoot(path, types) || !state.runtimeIdentifier || !state.activeIdentifier || !state.wrapperIdentifier) return;
-				const bamlMethod = state.bamlGenerated === true && isBamlClientMethod(path, types);
+			Function: { exit(path$18, state) {
+				const node = path$18.node;
+				if (state.skipFile || node._bitfabInstrumented || !isInstrumentableFunction(node, types) || isTraceDecoratorRoot(path$18, types) || !state.runtimeIdentifier || !state.activeIdentifier || !state.wrapperIdentifier) return;
+				const bamlMethod = state.bamlGenerated === true && isBamlClientMethod(path$18, types);
 				if (state.bamlGenerated && !bamlMethod) return;
-				if (isJevMethodDefinition(path, types)) return;
-				const name = qualifiedFunctionName(path, types);
+				if (isJevMethodDefinition(path$18, types)) return;
+				const name = qualifiedFunctionName(path$18, types);
 				if (name.startsWith("anonymous-")) return;
 				const normalizedFilename = autoTraceFilename(state.file.opts.filename ?? "unknown", state.opts.cwd ?? state.file.opts.cwd ?? process.cwd());
-				const { line, column } = sourcePosition(path);
+				const { line, column } = sourcePosition(path$18);
 				const definition = types.objectExpression([
 					types.objectProperty(types.identifier("id"), types.stringLiteral(autoTraceFunctionId(normalizedFilename, name))),
 					types.objectProperty(types.identifier("name"), types.stringLiteral(name)),
@@ -355,13 +404,13 @@ function bitfabBabelPlugin(api) {
 					types.objectProperty(types.identifier("column"), types.numericLiteral(column)),
 					...node.async ? [types.objectProperty(types.identifier("async"), types.booleanLiteral(true))] : [],
 					...bamlMethod ? [types.objectProperty(types.identifier("baml"), types.booleanLiteral(true))] : [],
-					...isRestWrapper(path, types) ? [types.objectProperty(types.identifier("wrapper"), types.booleanLiteral(true))] : []
+					...isRestWrapper(path$18, types) ? [types.objectProperty(types.identifier("wrapper"), types.booleanLiteral(true))] : []
 				]);
-				const definitionIdentifier = path.scope.getProgramParent().generateUidIdentifier("bitfabAutoDefinition");
+				const definitionIdentifier = path$18.scope.getProgramParent().generateUidIdentifier("bitfabAutoDefinition");
 				state.definitions?.push(types.variableDeclaration("const", [types.variableDeclarator(definitionIdentifier, definition)]));
-				if (needsArgumentWrapper(path, types)) {
+				if (needsArgumentWrapper(path$18, types)) {
 					node._bitfabInstrumented = true;
-					path.replaceWith(types.callExpression(types.cloneNode(state.wrapperIdentifier), [types.cloneNode(definitionIdentifier), node]));
+					path$18.replaceWith(types.callExpression(types.cloneNode(state.wrapperIdentifier), [types.cloneNode(definitionIdentifier), node]));
 					state.usedRuntime = true;
 					state.usedWrapper = true;
 					return;
@@ -369,12 +418,12 @@ function bitfabBabelPlugin(api) {
 				const originalBody = types.isBlockStatement(node.body) ? node.body : types.blockStatement([types.returnStatement(node.body)]);
 				const directives = originalBody.directives;
 				originalBody.directives = [];
-				const callerOptions = bamlMethod ? path.scope.generateUidIdentifier("bitfabBamlOptions") : void 0;
-				const withCollector = bamlMethod ? path.scope.generateUidIdentifier("bitfabBamlWithCollector") : void 0;
+				const callerOptions = bamlMethod ? path$18.scope.generateUidIdentifier("bitfabBamlOptions") : void 0;
+				const withCollector = bamlMethod ? path$18.scope.generateUidIdentifier("bitfabBamlWithCollector") : void 0;
 				const tracedBody = callerOptions && withCollector ? types.blockStatement([bamlOptionsDeclaration(callerOptions, withCollector, types), ...types.cloneNode(originalBody, true).body], []) : types.cloneNode(originalBody, true);
 				node.body = types.blockStatement([types.ifStatement(types.callExpression(types.cloneNode(state.activeIdentifier), []), types.blockStatement([...callerOptions ? [types.variableDeclaration("const", [types.variableDeclarator(callerOptions, types.identifier(BAML_OPTIONS_PARAMETER))])] : [], types.returnStatement(types.callExpression(types.cloneNode(state.runtimeIdentifier), [
 					types.cloneNode(definitionIdentifier),
-					parameterInputs(path, types),
+					parameterInputs(path$18, types),
 					types.arrowFunctionExpression(withCollector ? [types.cloneNode(withCollector)] : [], tracedBody, node.async ?? false)
 				]))])), ...originalBody.body], directives);
 				node._bitfabInstrumented = true;
@@ -409,4 +458,4 @@ Object.defineProperty(exports, "sourceFilename", {
 	}
 });
 
-//# sourceMappingURL=babelPlugin-BpEeVvk4.cjs.map
+//# sourceMappingURL=babelPlugin-C4ERQ8DW.cjs.map

package/dist/babelPlugin-C4ERQ8DW.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"babelPlugin-C4ERQ8DW.cjs","names":["path","readFileSync","join","dirname","isAbsolute"],"sources":["../../bitfab-typescript-transform/src/transformOptions.ts","../../bitfab-typescript-transform/src/babelPlugin.ts"],"sourcesContent":["export interface BitfabTransformOptions {\n  /** Module that exports the automatic span runtime. */\n  runtimeModule?: string\n  /** Module imported at the top of BAML's generated client files so the SDK can use the `@boundaryml/baml` the bundler loads. Pass `false` to import nothing. Defaults to `\"@bitfab/sdk/baml\"`, added only when it resolves from the file's folder. */\n  bamlRegistrationModule?: string | false\n  /** Working directory removed from generated function IDs. */\n  cwd?: string\n  /** Return false to leave a file unchanged. */\n  include?: (filename: string) => boolean\n}\n\nconst AUTO_TRACE_FUNCTION_ID_VERSION = \"ts-auto-v1\"\nexport const AUTO_TRACE_TRANSFORM_MARKER =\n  \"__bitfab_auto_trace_transformed_v1__\"\n\nexport function autoTraceFunctionId(file: string, name: string): string {\n  return `${AUTO_TRACE_FUNCTION_ID_VERSION}:${file}#${name}`\n}\n\nexport function sourceFilename(filename: string): string {\n  return filename.replace(/\\\\/g, \"/\").split(/[?#]/, 1)[0] ?? \"\"\n}\n\nexport function autoTraceFilename(filename: string, cwd: string): string {\n  const normalized = sourceFilename(filename)\n  const normalizedCwd = cwd.replace(/\\\\/g, \"/\").replace(/\\/$/, \"\")\n  return normalized.startsWith(`${normalizedCwd}/`)\n    ? normalized.slice(normalizedCwd.length + 1)\n    : normalized\n}\n\nexport function shouldTransformBitfabFile(\n  filename: string,\n  options: BitfabTransformOptions,\n): boolean {\n  if (!filename) {\n    return false\n  }\n  if (options.include && !options.include(filename)) {\n    return false\n  }\n\n  const normalized = sourceFilename(filename)\n  const normalizedCwd = options.cwd?.replace(/\\\\/g, \"/\").replace(/\\/$/, \"\")\n  const isAbsolute =\n    normalized.startsWith(\"/\") || /^[A-Za-z]:\\//.test(normalized)\n  if (\n    normalizedCwd &&\n    isAbsolute &&\n    normalized !== normalizedCwd &&\n    !normalized.startsWith(`${normalizedCwd}/`)\n  ) {\n    return false\n  }\n  return (\n    !normalized.includes(\"/node_modules/\") &&\n    !normalized.endsWith(\".d.ts\") &&\n    /\\.[cm]?[jt]sx?$/.test(normalized)\n  )\n}\n","import { readFileSync } from \"node:fs\"\nimport { dirname, isAbsolute, join } from \"node:path\"\nimport type { NodePath, PluginObject, PluginPass } from \"@babel/core\"\nimport type * as BabelTypes from \"@babel/types\"\nimport type { BitfabTransformOptions } from \"./transformOptions.js\"\nimport {\n  AUTO_TRACE_TRANSFORM_MARKER,\n  autoTraceFilename,\n  autoTraceFunctionId,\n  shouldTransformBitfabFile,\n  sourceFilename,\n} from \"./transformOptions.js\"\n\ntype BabelTypesApi = typeof BabelTypes\n\ntype BitfabPluginOptions = BitfabTransformOptions\n\ninterface BitfabPluginState extends PluginPass {\n  opts: BitfabPluginOptions\n  runtimeIdentifier?: ReturnType<BabelTypesApi[\"identifier\"]>\n  activeIdentifier?: ReturnType<BabelTypesApi[\"identifier\"]>\n  wrapperIdentifier?: ReturnType<BabelTypesApi[\"identifier\"]>\n  recordedCallIdentifiers?: Map<string, ReturnType<BabelTypesApi[\"identifier\"]>>\n  definitions?: Array<ReturnType<BabelTypesApi[\"variableDeclaration\"]>>\n  skipFile?: boolean\n  bamlGenerated?: boolean\n  usedRuntime?: boolean\n  usedWrapper?: boolean\n  usedRecordedCalls?: Set<string>\n}\n\ninterface InstrumentedFunction {\n  _bitfabInstrumented?: boolean\n}\n\nfunction propertyName(\n  node: BabelTypes.Node | null | undefined,\n  types: BabelTypesApi,\n): string | null {\n  if (types.isIdentifier(node)) {\n    return node.name\n  }\n  if (types.isStringLiteral(node)) {\n    return node.value\n  }\n  return null\n}\n\nfunction functionName(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): string {\n  const node = path.node\n  if (types.isFunctionDeclaration(node) || types.isFunctionExpression(node)) {\n    if (node.id) {\n      return node.id.name\n    }\n  }\n\n  if (\n    (types.isClassMethod(node) || types.isObjectMethod(node)) &&\n    !node.computed &&\n    types.isIdentifier(node.key)\n  ) {\n    return node.key.name\n  }\n\n  const parent = path.parentPath?.node\n  if (types.isVariableDeclarator(parent)) {\n    return propertyName(parent.id, types) ?? \"anonymous\"\n  }\n  if (types.isObjectProperty(parent) && !parent.computed) {\n    return propertyName(parent.key, types) ?? \"anonymous\"\n  }\n  if (types.isClassProperty(parent) && !parent.computed) {\n    const name = propertyName(parent.key, types)\n    if (name) {\n      return name\n    }\n  }\n\n  const { line, column } = sourcePosition(path)\n  return `anonymous-${line}-${column}`\n}\n\nconst JEV_METHOD = \"systemOne\"\n\nfunction isJevMethodDefinition(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): boolean {\n  const parent = path.parentPath?.node\n  const isMethod =\n    types.isClassMethod(path.node) ||\n    types.isObjectMethod(path.node) ||\n    types.isObjectProperty(parent) ||\n    types.isClassProperty(parent)\n  return isMethod && functionName(path, types) === JEV_METHOD\n}\n\nfunction isInstrumentableFunction(\n  node: BabelTypes.Function,\n  types: BabelTypesApi,\n): boolean {\n  if (node.generator || types.isClassPrivateMethod(node)) {\n    return false\n  }\n  return (\n    !(types.isClassMethod(node) || types.isObjectMethod(node)) ||\n    node.kind === \"method\"\n  )\n}\n\nfunction sourcePosition(path: NodePath<BabelTypes.Function>): {\n  line: number\n  column: number\n} {\n  const exportDeclaration = path.parentPath?.isExportDeclaration()\n    ? path.parentPath.node\n    : undefined\n  const start = (exportDeclaration ?? path.node).loc?.start\n  return { line: start?.line ?? 0, column: start?.column ?? 0 }\n}\n\nfunction qualifiedFunctionName(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): string {\n  const ownName = functionName(path, types)\n  if (ownName.startsWith(\"anonymous-\")) {\n    return ownName\n  }\n  const scopes: string[] = []\n  let ancestor: NodePath | null = path.parentPath\n  while (ancestor) {\n    if (ancestor.isFunction()) {\n      if (!isInstrumentableFunction(ancestor.node, types)) {\n        ancestor = ancestor.parentPath\n        continue\n      }\n      const name = functionName(\n        ancestor as NodePath<BabelTypes.Function>,\n        types,\n      )\n      if (!name.startsWith(\"anonymous-\")) {\n        scopes.unshift(name)\n      }\n    } else if (\n      (ancestor.isClassDeclaration() || ancestor.isClassExpression()) &&\n      ancestor.node.id\n    ) {\n      scopes.unshift(ancestor.node.id.name)\n    } else if (ancestor.isObjectExpression()) {\n      const parent = ancestor.parentPath?.node\n      if (types.isVariableDeclarator(parent)) {\n        const name = propertyName(parent.id, types)\n        if (name) {\n          scopes.unshift(name)\n        }\n      } else if (types.isObjectProperty(parent) && !parent.computed) {\n        const name = propertyName(parent.key, types)\n        if (name) {\n          scopes.unshift(name)\n        }\n      }\n    }\n    ancestor = ancestor.parentPath\n  }\n  return [...scopes, ownName].join(\".\")\n}\n\nfunction isTraceDecoratorRoot(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): boolean {\n  const node = path.node\n  const parent = path.parentPath?.node\n  const decorated =\n    types.isClassMethod(node) || types.isClassPrivateMethod(node)\n      ? node\n      : types.isClassProperty(parent) && parent.value === node\n        ? parent\n        : null\n  if (!decorated) {\n    return false\n  }\n\n  return (\n    decorated.decorators?.some((decorator) => {\n      const expression = decorator.expression\n      if (!types.isCallExpression(expression)) {\n        return false\n      }\n      const callee = expression.callee\n      return (\n        types.isMemberExpression(callee) &&\n        !callee.computed &&\n        types.isIdentifier(callee.property, { name: \"trace\" })\n      )\n    }) ?? false\n  )\n}\n\nfunction withNodeFunctionName(\n  path: NodePath<BabelTypes.CallExpression>,\n  types: BabelTypesApi,\n): string | null {\n  if (path.node.arguments.length > 2) {\n    return null\n  }\n  const callee = path.node.callee\n  if (\n    !types.isMemberExpression(callee) ||\n    callee.computed ||\n    !types.isIdentifier(callee.property, { name: \"withNode\" })\n  ) {\n    return null\n  }\n  const fn = path.node.arguments[path.node.arguments.length - 1]\n  if (types.isIdentifier(fn)) {\n    return fn.name\n  }\n  if (types.isMemberExpression(fn) && !fn.computed) {\n    return propertyName(fn.property, types)\n  }\n  if (types.isFunctionExpression(fn) && fn.id) {\n    return fn.id.name\n  }\n  return null\n}\n\nconst BAML_GENERATED_HEADER = \"This file was generated by BAML\"\nconst BAML_CLIENT_CLASSES = new Set([\"BamlAsyncClient\", \"BamlSyncClient\"])\nconst BAML_OPTIONS_PARAMETER = \"__baml_options__\"\n\nconst BAML_PACKAGE = \"@boundaryml/baml\"\nconst DEFAULT_BAML_REGISTRATION_MODULE = \"@bitfab/sdk/baml\"\n\nfunction importsBamlAtRuntime(\n  program: BabelTypes.Program,\n  types: BabelTypesApi,\n): boolean {\n  return program.body.some(\n    (statement) =>\n      types.isImportDeclaration(statement) &&\n      statement.source.value === BAML_PACKAGE &&\n      statement.importKind !== \"type\" &&\n      statement.specifiers.some(\n        (specifier) =>\n          !types.isImportSpecifier(specifier) ||\n          specifier.importKind !== \"type\",\n      ),\n  )\n}\n\nconst SDK_PACKAGE_PATH = [\"node_modules\", \"@bitfab\", \"sdk\", \"package.json\"]\nconst BAML_REGISTRATION_EXPORT = \"./baml\"\nconst sdkExportsBamlRegistrationByDirectory = new Map<string, boolean>()\n\nfunction readPackageJson(path: string): unknown {\n  try {\n    return JSON.parse(readFileSync(path, \"utf8\"))\n  } catch {\n    return undefined\n  }\n}\n\nfunction exportsBamlRegistration(packageJson: unknown): boolean {\n  if (typeof packageJson !== \"object\" || packageJson === null) {\n    return false\n  }\n  const exports = (packageJson as { exports?: unknown }).exports\n  return (\n    typeof exports === \"object\" &&\n    exports !== null &&\n    BAML_REGISTRATION_EXPORT in exports\n  )\n}\n\nfunction nearestSdkExportsBamlRegistration(directory: string): boolean {\n  const cached = sdkExportsBamlRegistrationByDirectory.get(directory)\n  if (cached !== undefined) {\n    return cached\n  }\n  const packageJson = readPackageJson(join(directory, ...SDK_PACKAGE_PATH))\n  const parent = dirname(directory)\n  const exported =\n    packageJson !== undefined\n      ? exportsBamlRegistration(packageJson)\n      : parent !== directory && nearestSdkExportsBamlRegistration(parent)\n  sdkExportsBamlRegistrationByDirectory.set(directory, exported)\n  return exported\n}\n\nfunction defaultRegistrationResolves(filename: string): boolean {\n  return (\n    isAbsolute(filename) && nearestSdkExportsBamlRegistration(dirname(filename))\n  )\n}\n\nfunction bamlRegistrationModuleFor(\n  filename: string,\n  configured: string | false | undefined,\n): string | null {\n  if (configured === false) {\n    return null\n  }\n  if (configured !== undefined) {\n    return configured\n  }\n  return defaultRegistrationResolves(filename)\n    ? DEFAULT_BAML_REGISTRATION_MODULE\n    : null\n}\n\nfunction isBamlGeneratedSource(code: string): boolean {\n  return code.slice(0, 4096).includes(BAML_GENERATED_HEADER)\n}\n\nfunction isBamlClientMethod(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): boolean {\n  const node = path.node\n  if (!types.isClassMethod(node) || node.kind !== \"method\" || node.static) {\n    return false\n  }\n  const classPath = path.parentPath?.parentPath\n  if (\n    !classPath ||\n    !(classPath.isClassDeclaration() || classPath.isClassExpression()) ||\n    !classPath.node.id ||\n    !BAML_CLIENT_CLASSES.has(classPath.node.id.name)\n  ) {\n    return false\n  }\n  const parameters = runtimeParameters(node, types)\n  const last = parameters[parameters.length - 1]\n  return types.isIdentifier(last, { name: BAML_OPTIONS_PARAMETER })\n}\n\nfunction bamlOptionsDeclaration(\n  callerOptions: BabelTypes.Identifier,\n  withCollector: BabelTypes.Identifier,\n  types: BabelTypesApi,\n): BabelTypes.VariableDeclaration {\n  return types.variableDeclaration(\"const\", [\n    types.variableDeclarator(\n      types.identifier(BAML_OPTIONS_PARAMETER),\n      types.conditionalExpression(\n        types.binaryExpression(\n          \"===\",\n          types.cloneNode(withCollector),\n          types.identifier(\"undefined\"),\n        ),\n        types.cloneNode(callerOptions),\n        types.callExpression(types.cloneNode(withCollector), [\n          types.cloneNode(callerOptions),\n          types.memberExpression(\n            types.thisExpression(),\n            types.identifier(\"bamlOptions\"),\n          ),\n        ]),\n      ),\n    ),\n  ])\n}\n\nfunction runtimeParameters(\n  node: BabelTypes.Function,\n  types: BabelTypesApi,\n): BabelTypes.Function[\"params\"] {\n  return node.params.filter(\n    (parameter) => !types.isIdentifier(parameter, { name: \"this\" }),\n  )\n}\n\nfunction parameterInputs(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): ReturnType<BabelTypesApi[\"arrayExpression\"]> {\n  const elements: Array<BabelTypes.Expression | BabelTypes.SpreadElement> = []\n\n  for (const [index, parameter] of runtimeParameters(\n    path.node,\n    types,\n  ).entries()) {\n    if (!types.isArrowFunctionExpression(path.node)) {\n      if (types.isRestElement(parameter)) {\n        elements.push(\n          types.spreadElement(\n            types.callExpression(\n              types.memberExpression(\n                types.arrayExpression([\n                  types.spreadElement(types.identifier(\"arguments\")),\n                ]),\n                types.identifier(\"slice\"),\n              ),\n              [types.numericLiteral(index)],\n            ),\n          ),\n        )\n      } else {\n        elements.push(\n          types.memberExpression(\n            types.identifier(\"arguments\"),\n            types.numericLiteral(index),\n            true,\n          ),\n        )\n      }\n      continue\n    }\n\n    if (types.isIdentifier(parameter)) {\n      elements.push(types.identifier(parameter.name))\n    } else if (\n      types.isRestElement(parameter) &&\n      types.isIdentifier(parameter.argument)\n    ) {\n      elements.push(\n        types.spreadElement(types.identifier(parameter.argument.name)),\n      )\n    }\n  }\n\n  return types.arrayExpression(elements)\n}\n\nfunction needsArgumentWrapper(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): boolean {\n  return (\n    types.isArrowFunctionExpression(path.node) &&\n    path.node.params.some(\n      (parameter) =>\n        !types.isIdentifier(parameter) &&\n        !(\n          types.isRestElement(parameter) &&\n          types.isIdentifier(parameter.argument)\n        ),\n    )\n  )\n}\n\nfunction isRestWrapper(\n  path: NodePath<BabelTypes.Function>,\n  types: BabelTypesApi,\n): boolean {\n  const parameters = runtimeParameters(path.node, types)\n  return parameters.length === 1 && types.isRestElement(parameters[0])\n}\n\ntype RecordedCallNode =\n  | BabelTypes.CallExpression\n  | BabelTypes.OptionalCallExpression\n\ninterface RecordedClientCall {\n  methods: readonly string[]\n  endpoint?: string\n  runtimeExport: string\n  identifierHint: string\n  receiverHint: string\n  matchesReceiver: (\n    receiver: BabelTypes.Expression,\n    types: BabelTypesApi,\n  ) => boolean\n}\n\nfunction isNamedMember(\n  node: BabelTypes.Node,\n  name: string,\n  types: BabelTypesApi,\n): node is BabelTypes.MemberExpression | BabelTypes.OptionalMemberExpression {\n  return (\n    (types.isMemberExpression(node) ||\n      types.isOptionalMemberExpression(node)) &&\n    !node.computed &&\n    types.isIdentifier(node.property, { name })\n  )\n}\n\nfunction isChatCompletionsReceiver(\n  receiver: BabelTypes.Expression,\n  types: BabelTypesApi,\n): boolean {\n  return (\n    isNamedMember(receiver, \"completions\", types) &&\n    isNamedMember(receiver.object, \"chat\", types)\n  )\n}\n\nfunction isMessagesReceiver(\n  receiver: BabelTypes.Expression,\n  types: BabelTypesApi,\n): boolean {\n  return (\n    isNamedMember(receiver, \"messages\", types) &&\n    !isNamedMember(receiver.object, \"beta\", types)\n  )\n}\n\nfunction isResponsesReceiver(\n  receiver: BabelTypes.Expression,\n  types: BabelTypesApi,\n): boolean {\n  return isNamedMember(receiver, \"responses\", types)\n}\n\nfunction isEmbeddingsReceiver(\n  receiver: BabelTypes.Expression,\n  types: BabelTypesApi,\n): boolean {\n  return isNamedMember(receiver, \"embeddings\", types)\n}\n\nfunction isTranscriptionsReceiver(\n  receiver: BabelTypes.Expression,\n  types: BabelTypesApi,\n): boolean {\n  return (\n    isNamedMember(receiver, \"transcriptions\", types) &&\n    isNamedMember(receiver.object, \"audio\", types)\n  )\n}\n\nfunction isModelsReceiver(\n  receiver: BabelTypes.Expression,\n  types: BabelTypesApi,\n): boolean {\n  return isNamedMember(receiver, \"models\", types)\n}\n\nconst CLIENT_CALL_RUNTIME = {\n  runtimeExport: \"__bitfabAutoClientCall\",\n  identifierHint: \"bitfabAutoClientCall\",\n  receiverHint: \"bitfabClientCallReceiver\",\n}\n\nconst RECORDED_CLIENT_CALLS: readonly RecordedClientCall[] = [\n  {\n    methods: [JEV_METHOD],\n    runtimeExport: \"__bitfabAutoJev\",\n    identifierHint: \"bitfabAutoJev\",\n    receiverHint: \"bitfabJevReceiver\",\n    matchesReceiver: () => true,\n  },\n  {\n    ...CLIENT_CALL_RUNTIME,\n    endpoint: \"chat.completions\",\n    methods: [\"create\", \"parse\", \"stream\"],\n    matchesReceiver: isChatCompletionsReceiver,\n  },\n  {\n    ...CLIENT_CALL_RUNTIME,\n    endpoint: \"embeddings\",\n    methods: [\"create\"],\n    matchesReceiver: isEmbeddingsReceiver,\n  },\n  {\n    ...CLIENT_CALL_RUNTIME,\n    endpoint: \"audio.transcriptions\",\n    methods: [\"create\"],\n    matchesReceiver: isTranscriptionsReceiver,\n  },\n  {\n    ...CLIENT_CALL_RUNTIME,\n    endpoint: \"messages\",\n    methods: [\"create\", \"parse\", \"stream\"],\n    matchesReceiver: isMessagesReceiver,\n  },\n  {\n    ...CLIENT_CALL_RUNTIME,\n    endpoint: \"responses\",\n    methods: [\"create\", \"parse\", \"stream\"],\n    matchesReceiver: isResponsesReceiver,\n  },\n  {\n    ...CLIENT_CALL_RUNTIME,\n    endpoint: \"models\",\n    methods: [\"generateContent\", \"generateContentStream\"],\n    matchesReceiver: isModelsReceiver,\n  },\n]\n\nconst RECORDED_CALL_RUNTIMES = [\n  ...new Map(\n    RECORDED_CLIENT_CALLS.map((call) => [call.runtimeExport, call]),\n  ).values(),\n]\n\nfunction recordedCallTarget(\n  node: RecordedCallNode,\n  types: BabelTypesApi,\n): {\n  call: RecordedClientCall\n  receiver: BabelTypes.Expression\n  method: string\n} | null {\n  const callee = node.callee\n  if (\n    !(\n      types.isMemberExpression(callee) ||\n      types.isOptionalMemberExpression(callee)\n    ) ||\n    callee.computed ||\n    !types.isIdentifier(callee.property) ||\n    types.isSuper(callee.object)\n  ) {\n    return null\n  }\n  const method = callee.property.name\n  const receiver = callee.object\n  const call = RECORDED_CLIENT_CALLS.find(\n    (candidate) =>\n      candidate.methods.includes(method) &&\n      candidate.matchesReceiver(receiver, types),\n  )\n  return call === undefined ? null : { call, receiver, method }\n}\n\nfunction isOptionalChain(node: BabelTypes.Node, types: BabelTypesApi): boolean {\n  return (\n    types.isOptionalMemberExpression(node) ||\n    types.isOptionalCallExpression(node)\n  )\n}\n\nfunction continuesOptionalChain(\n  path: NodePath<BabelTypes.OptionalCallExpression>,\n  types: BabelTypesApi,\n): boolean {\n  const parent = path.parent\n  return (\n    (types.isOptionalMemberExpression(parent) && parent.object === path.node) ||\n    (types.isOptionalCallExpression(parent) && parent.callee === path.node)\n  )\n}\n\nfunction rewriteRecordedClientCall(\n  path: NodePath<RecordedCallNode>,\n  state: BitfabPluginState,\n  types: BabelTypesApi,\n): void {\n  if (state.skipFile || state.bamlGenerated) {\n    return\n  }\n  const node = path.node\n  const target = recordedCallTarget(node, types)\n  const identifier =\n    target === null\n      ? undefined\n      : state.recordedCallIdentifiers?.get(target.call.runtimeExport)\n  if (target === null || identifier === undefined) {\n    return\n  }\n  const { call: recordedCall, receiver, method } = target\n  const optionalReceiver =\n    types.isOptionalMemberExpression(node.callee) && node.callee.optional\n  const optionalMethod = types.isOptionalCallExpression(node) && node.optional\n  if (types.isOptionalCallExpression(node)) {\n    if (\n      continuesOptionalChain(\n        path as NodePath<BabelTypes.OptionalCallExpression>,\n        types,\n      ) ||\n      (!optionalReceiver && isOptionalChain(receiver, types))\n    ) {\n      return\n    }\n  }\n  const guarded = optionalReceiver || optionalMethod\n  const receiverReference =\n    guarded && !path.scope.isStatic(receiver)\n      ? path.scope.generateDeclaredUidIdentifier(recordedCall.receiverHint)\n      : null\n  const receiverValue = (): BabelTypes.Expression =>\n    receiverReference === null\n      ? types.cloneNode(receiver)\n      : types.cloneNode(receiverReference)\n  const call = types.callExpression(\n    types.cloneNode(identifier),\n    recordedCall.endpoint === undefined\n      ? [receiverValue(), ...node.arguments]\n      : [\n          types.stringLiteral(recordedCall.endpoint),\n          receiverValue(),\n          types.stringLiteral(method),\n          ...node.arguments,\n        ],\n  )\n  state.usedRuntime = true\n  state.usedRecordedCalls?.add(recordedCall.runtimeExport)\n  if (!guarded) {\n    path.replaceWith(call)\n    return\n  }\n  const firstReceiver =\n    receiverReference === null\n      ? types.cloneNode(receiver)\n      : types.assignmentExpression(\n          \"=\",\n          types.cloneNode(receiverReference),\n          receiver,\n        )\n  const missing = (value: BabelTypes.Expression): BabelTypes.BinaryExpression =>\n    types.binaryExpression(\"==\", value, types.nullLiteral())\n  const methodOf = (value: BabelTypes.Expression): BabelTypes.Expression =>\n    types.memberExpression(value, types.identifier(method))\n  const test = optionalReceiver\n    ? optionalMethod\n      ? types.logicalExpression(\n          \"||\",\n          missing(firstReceiver),\n          missing(methodOf(receiverValue())),\n        )\n      : missing(firstReceiver)\n    : missing(methodOf(firstReceiver))\n  path.replaceWith(\n    types.conditionalExpression(\n      test,\n      types.unaryExpression(\"void\", types.numericLiteral(0)),\n      call,\n    ),\n  )\n}\n\n/**\n * Babel adapter for Bitfab's experimental automatic call-span runtime.\n *\n * @experimental The transform protocol and generated function identities may\n * change before the feature is stable.\n */\nexport default function bitfabBabelPlugin(api: {\n  types: BabelTypesApi\n}): PluginObject<BitfabPluginState> {\n  const { types } = api\n\n  return {\n    name: \"bitfab-auto-trace\",\n    visitor: {\n      Program: {\n        enter(path: NodePath<BabelTypes.Program>, state: BitfabPluginState) {\n          const filename = state.file.opts.filename ?? \"unknown\"\n          state.skipFile =\n            !shouldTransformBitfabFile(filename, state.opts) ||\n            path.scope.hasBinding(AUTO_TRACE_TRANSFORM_MARKER) ||\n            path.node.directives.some(\n              (directive) => directive.value.value === \"use client\",\n            )\n          state.runtimeIdentifier =\n            path.scope.generateUidIdentifier(\"bitfabAutoSpan\")\n          state.activeIdentifier = path.scope.generateUidIdentifier(\n            \"bitfabAutoTraceActive\",\n          )\n          state.wrapperIdentifier =\n            path.scope.generateUidIdentifier(\"bitfabAutoWrap\")\n          state.recordedCallIdentifiers = new Map(\n            RECORDED_CALL_RUNTIMES.map((call) => [\n              call.runtimeExport,\n              path.scope.generateUidIdentifier(call.identifierHint),\n            ]),\n          )\n          state.bamlGenerated = isBamlGeneratedSource(state.file.code ?? \"\")\n          state.definitions = []\n          state.usedRuntime = false\n          state.usedWrapper = false\n          state.usedRecordedCalls = new Set()\n        },\n        exit(path: NodePath<BabelTypes.Program>, state: BitfabPluginState) {\n          const bamlRegistrationModule =\n            !state.skipFile &&\n            state.bamlGenerated &&\n            importsBamlAtRuntime(path.node, types)\n              ? bamlRegistrationModuleFor(\n                  sourceFilename(state.file.opts.filename ?? \"\"),\n                  state.opts.bamlRegistrationModule,\n                )\n              : null\n          if (bamlRegistrationModule !== null) {\n            path.unshiftContainer(\n              \"body\",\n              types.importDeclaration(\n                [],\n                types.stringLiteral(bamlRegistrationModule),\n              ),\n            )\n            state.file.metadata[AUTO_TRACE_TRANSFORM_MARKER] = true\n          }\n          if (\n            state.skipFile ||\n            !state.usedRuntime ||\n            !state.runtimeIdentifier ||\n            !state.activeIdentifier ||\n            !state.wrapperIdentifier\n          ) {\n            return\n          }\n\n          state.file.metadata[AUTO_TRACE_TRANSFORM_MARKER] = true\n          path.unshiftContainer(\"body\", [\n            types.importDeclaration(\n              [\n                types.importSpecifier(\n                  state.runtimeIdentifier,\n                  types.identifier(\"__bitfabAutoSpan\"),\n                ),\n                types.importSpecifier(\n                  state.activeIdentifier,\n                  types.identifier(\"__bitfabAutoTraceActive\"),\n                ),\n                ...(state.usedWrapper\n                  ? [\n                      types.importSpecifier(\n                        state.wrapperIdentifier,\n                        types.identifier(\"__bitfabAutoWrap\"),\n                      ),\n                    ]\n                  : []),\n                ...RECORDED_CALL_RUNTIMES.flatMap((call) => {\n                  const identifier = state.recordedCallIdentifiers?.get(\n                    call.runtimeExport,\n                  )\n                  return state.usedRecordedCalls?.has(call.runtimeExport) &&\n                    identifier\n                    ? [\n                        types.importSpecifier(\n                          identifier,\n                          types.identifier(call.runtimeExport),\n                        ),\n                      ]\n                    : []\n                }),\n              ],\n              types.stringLiteral(\n                state.opts.runtimeModule ?? \"@bitfab/sdk/auto\",\n              ),\n            ),\n            types.variableDeclaration(\"const\", [\n              types.variableDeclarator(\n                types.identifier(AUTO_TRACE_TRANSFORM_MARKER),\n                types.assignmentExpression(\n                  \"=\",\n                  types.memberExpression(\n                    types.identifier(\"globalThis\"),\n                    types.identifier(AUTO_TRACE_TRANSFORM_MARKER),\n                  ),\n                  types.booleanLiteral(true),\n                ),\n              ),\n            ]),\n            ...(state.definitions ?? []),\n          ])\n        },\n      },\n      CallExpression: {\n        exit(\n          path: NodePath<BabelTypes.CallExpression>,\n          state: BitfabPluginState,\n        ) {\n          if (state.skipFile) {\n            return\n          }\n          if (recordedCallTarget(path.node, types) !== null) {\n            rewriteRecordedClientCall(path, state, types)\n            return\n          }\n          const configuredName = withNodeFunctionName(path, types)\n          if (configuredName === null) {\n            return\n          }\n          if (path.node.arguments.length === 1) {\n            path.node.arguments.push(types.identifier(\"undefined\"))\n          }\n          path.node.arguments.push(types.stringLiteral(configuredName))\n        },\n      },\n      OptionalCallExpression: {\n        exit(\n          path: NodePath<BabelTypes.OptionalCallExpression>,\n          state: BitfabPluginState,\n        ) {\n          rewriteRecordedClientCall(path, state, types)\n        },\n      },\n      Function: {\n        exit(path: NodePath<BabelTypes.Function>, state: BitfabPluginState) {\n          const node = path.node as typeof path.node & InstrumentedFunction\n          if (\n            state.skipFile ||\n            node._bitfabInstrumented ||\n            !isInstrumentableFunction(node, types) ||\n            isTraceDecoratorRoot(path, types) ||\n            !state.runtimeIdentifier ||\n            !state.activeIdentifier ||\n            !state.wrapperIdentifier\n          ) {\n            return\n          }\n          const bamlMethod =\n            state.bamlGenerated === true && isBamlClientMethod(path, types)\n          if (state.bamlGenerated && !bamlMethod) {\n            return\n          }\n          if (isJevMethodDefinition(path, types)) {\n            return\n          }\n          const name = qualifiedFunctionName(path, types)\n          if (name.startsWith(\"anonymous-\")) {\n            return\n          }\n          const normalizedFilename = autoTraceFilename(\n            state.file.opts.filename ?? \"unknown\",\n            state.opts.cwd ?? state.file.opts.cwd ?? process.cwd(),\n          )\n          const { line, column } = sourcePosition(path)\n          const definition = types.objectExpression([\n            types.objectProperty(\n              types.identifier(\"id\"),\n              types.stringLiteral(\n                autoTraceFunctionId(normalizedFilename, name),\n              ),\n            ),\n            types.objectProperty(\n              types.identifier(\"name\"),\n              types.stringLiteral(name),\n            ),\n            types.objectProperty(\n              types.identifier(\"file\"),\n              types.stringLiteral(normalizedFilename),\n            ),\n            types.objectProperty(\n              types.identifier(\"line\"),\n              types.numericLiteral(line),\n            ),\n            types.objectProperty(\n              types.identifier(\"column\"),\n              types.numericLiteral(column),\n            ),\n            ...(node.async\n              ? [\n                  types.objectProperty(\n                    types.identifier(\"async\"),\n                    types.booleanLiteral(true),\n                  ),\n                ]\n              : []),\n            ...(bamlMethod\n              ? [\n                  types.objectProperty(\n                    types.identifier(\"baml\"),\n                    types.booleanLiteral(true),\n                  ),\n                ]\n              : []),\n            ...(isRestWrapper(path, types)\n              ? [\n                  types.objectProperty(\n                    types.identifier(\"wrapper\"),\n                    types.booleanLiteral(true),\n                  ),\n                ]\n              : []),\n          ])\n          const definitionIdentifier = path.scope\n            .getProgramParent()\n            .generateUidIdentifier(\"bitfabAutoDefinition\")\n          state.definitions?.push(\n            types.variableDeclaration(\"const\", [\n              types.variableDeclarator(definitionIdentifier, definition),\n            ]),\n          )\n\n          if (needsArgumentWrapper(path, types)) {\n            node._bitfabInstrumented = true\n            path.replaceWith(\n              types.callExpression(types.cloneNode(state.wrapperIdentifier), [\n                types.cloneNode(definitionIdentifier),\n                node as BabelTypes.ArrowFunctionExpression,\n              ]),\n            )\n            state.usedRuntime = true\n            state.usedWrapper = true\n            return\n          }\n\n          const originalBody = types.isBlockStatement(node.body)\n            ? node.body\n            : types.blockStatement([types.returnStatement(node.body)])\n          const directives = originalBody.directives\n          originalBody.directives = []\n\n          const callerOptions = bamlMethod\n            ? path.scope.generateUidIdentifier(\"bitfabBamlOptions\")\n            : undefined\n          const withCollector = bamlMethod\n            ? path.scope.generateUidIdentifier(\"bitfabBamlWithCollector\")\n            : undefined\n          const tracedBody =\n            callerOptions && withCollector\n              ? types.blockStatement(\n                  [\n                    bamlOptionsDeclaration(callerOptions, withCollector, types),\n                    ...types.cloneNode(originalBody, true).body,\n                  ],\n                  [],\n                )\n              : types.cloneNode(originalBody, true)\n          const body = types.blockStatement(\n            [\n              types.ifStatement(\n                types.callExpression(\n                  types.cloneNode(state.activeIdentifier),\n                  [],\n                ),\n                types.blockStatement([\n                  ...(callerOptions\n                    ? [\n                        types.variableDeclaration(\"const\", [\n                          types.variableDeclarator(\n                            callerOptions,\n                            types.identifier(BAML_OPTIONS_PARAMETER),\n                          ),\n                        ]),\n                      ]\n                    : []),\n                  types.returnStatement(\n                    types.callExpression(\n                      types.cloneNode(state.runtimeIdentifier),\n                      [\n                        types.cloneNode(definitionIdentifier),\n                        parameterInputs(path, types),\n                        types.arrowFunctionExpression(\n                          withCollector ? [types.cloneNode(withCollector)] : [],\n                          tracedBody,\n                          node.async ?? false,\n                        ),\n                      ],\n                    ),\n                  ),\n                ]),\n              ),\n              ...originalBody.body,\n            ],\n            directives,\n          )\n          node.body = body\n          node._bitfabInstrumented = true\n          state.usedRuntime = true\n        },\n      },\n    },\n  }\n}\n"],"mappings":";;;AAWA,MAAM,iCAAiC;AACvC,MAAa,8BACX;AAEF,SAAgB,oBAAoB,MAAc,MAAsB;CACtE,OAAO,GAAG,+BAA+B,GAAG,KAAK,GAAG;AACtD;AAEA,SAAgB,eAAe,UAA0B;CACvD,OAAO,SAAS,QAAQ,OAAO,GAAG,CAAC,CAAC,MAAM,QAAQ,CAAC,CAAC,CAAC,MAAM;AAC7D;AAEA,SAAgB,kBAAkB,UAAkB,KAAqB;CACvE,MAAM,aAAa,eAAe,QAAQ;CAC1C,MAAM,gBAAgB,IAAI,QAAQ,OAAO,GAAG,CAAC,CAAC,QAAQ,OAAO,EAAE;CAC/D,OAAO,WAAW,WAAW,GAAG,cAAc,EAAE,IAC5C,WAAW,MAAM,cAAc,SAAS,CAAC,IACzC;AACN;AAEA,SAAgB,0BACd,UACA,SACS;CACT,IAAI,CAAC,UACH,OAAO;CAET,IAAI,QAAQ,WAAW,CAAC,QAAQ,QAAQ,QAAQ,GAC9C,OAAO;CAGT,MAAM,aAAa,eAAe,QAAQ;CAC1C,MAAM,gBAAgB,QAAQ,KAAK,QAAQ,OAAO,GAAG,CAAC,CAAC,QAAQ,OAAO,EAAE;CACxE,MAAM,aACJ,WAAW,WAAW,GAAG,KAAK,eAAe,KAAK,UAAU;CAC9D,IACE,iBACA,cACA,eAAe,iBACf,CAAC,WAAW,WAAW,GAAG,cAAc,EAAE,GAE1C,OAAO;CAET,OACE,CAAC,WAAW,SAAS,gBAAgB,KACrC,CAAC,WAAW,SAAS,OAAO,KAC5B,kBAAkB,KAAK,UAAU;AAErC;;;ACxBA,SAAS,aACP,MACA,OACe;CACf,IAAI,MAAM,aAAa,IAAI,GACzB,OAAO,KAAK;CAEd,IAAI,MAAM,gBAAgB,IAAI,GAC5B,OAAO,KAAK;CAEd,OAAO;AACT;AAEA,SAAS,aACP,QACA,OACQ;CACR,MAAM,OAAOA,OAAK;CAClB,IAAI,MAAM,sBAAsB,IAAI,KAAK,MAAM,qBAAqB,IAAI,GAClE;MAAA,KAAK,IACP,OAAO,KAAK,GAAG;CAAA;CAInB,KACG,MAAM,cAAc,IAAI,KAAK,MAAM,eAAe,IAAI,MACvD,CAAC,KAAK,YACN,MAAM,aAAa,KAAK,GAAG,GAE3B,OAAO,KAAK,IAAI;CAGlB,MAAM,SAASA,OAAK,YAAY;CAChC,IAAI,MAAM,qBAAqB,MAAM,GACnC,OAAO,aAAa,OAAO,IAAI,KAAK,KAAK;CAE3C,IAAI,MAAM,iBAAiB,MAAM,KAAK,CAAC,OAAO,UAC5C,OAAO,aAAa,OAAO,KAAK,KAAK,KAAK;CAE5C,IAAI,MAAM,gBAAgB,MAAM,KAAK,CAAC,OAAO,UAAU;EACrD,MAAM,OAAO,aAAa,OAAO,KAAK,KAAK;EAC3C,IAAI,MACF,OAAO;CAEX;CAEA,MAAM,EAAE,MAAM,WAAW,eAAeA,MAAI;CAC5C,OAAO,aAAa,KAAK,GAAG;AAC9B;AAEA,MAAM,aAAa;AAEnB,SAAS,sBACP,QACA,OACS;CACT,MAAM,SAASA,OAAK,YAAY;CAMhC,QAJE,MAAM,cAAcA,OAAK,IAAI,KAC7B,MAAM,eAAeA,OAAK,IAAI,KAC9B,MAAM,iBAAiB,MAAM,KAC7B,MAAM,gBAAgB,MAAM,MACX,aAAaA,QAAM,KAAK,MAAM;AACnD;AAEA,SAAS,yBACP,MACA,OACS;CACT,IAAI,KAAK,aAAa,MAAM,qBAAqB,IAAI,GACnD,OAAO;CAET,OACE,EAAE,MAAM,cAAc,IAAI,KAAK,MAAM,eAAe,IAAI,MACxD,KAAK,SAAS;AAElB;AAEA,SAAS,eAAe,QAGtB;CAIA,MAAM,UAHoBA,OAAK,YAAY,oBAAoB,IAC3DA,OAAK,WAAW,OAChB,KAAA,MACgCA,OAAK,KAAA,CAAM,KAAK;CACpD,OAAO;EAAE,MAAM,OAAO,QAAQ;EAAG,QAAQ,OAAO,UAAU;CAAE;AAC9D;AAEA,SAAS,sBACP,QACA,OACQ;CACR,MAAM,UAAU,aAAaA,QAAM,KAAK;CACxC,IAAI,QAAQ,WAAW,YAAY,GACjC,OAAO;CAET,MAAM,SAAmB,CAAC;CAC1B,IAAI,WAA4BA,OAAK;CACrC,OAAO,UAAU;EACf,IAAI,SAAS,WAAW,GAAG;GACzB,IAAI,CAAC,yBAAyB,SAAS,MAAM,KAAK,GAAG;IACnD,WAAW,SAAS;IACpB;GACF;GACA,MAAM,OAAO,aACX,UACA,KACF;GACA,IAAI,CAAC,KAAK,WAAW,YAAY,GAC/B,OAAO,QAAQ,IAAI;EAEvB,OAAO,KACJ,SAAS,mBAAmB,KAAK,SAAS,kBAAkB,MAC7D,SAAS,KAAK,IAEd,OAAO,QAAQ,SAAS,KAAK,GAAG,IAAI;OAC/B,IAAI,SAAS,mBAAmB,GAAG;GACxC,MAAM,SAAS,SAAS,YAAY;GACpC,IAAI,MAAM,qBAAqB,MAAM,GAAG;IACtC,MAAM,OAAO,aAAa,OAAO,IAAI,KAAK;IAC1C,IAAI,MACF,OAAO,QAAQ,IAAI;GAEvB,OAAO,IAAI,MAAM,iBAAiB,MAAM,KAAK,CAAC,OAAO,UAAU;IAC7D,MAAM,OAAO,aAAa,OAAO,KAAK,KAAK;IAC3C,IAAI,MACF,OAAO,QAAQ,IAAI;GAEvB;EACF;EACA,WAAW,SAAS;CACtB;CACA,OAAO,CAAC,GAAG,QAAQ,OAAO,CAAC,CAAC,KAAK,GAAG;AACtC;AAEA,SAAS,qBACP,QACA,OACS;CACT,MAAM,OAAOA,OAAK;CAClB,MAAM,SAASA,OAAK,YAAY;CAChC,MAAM,YACJ,MAAM,cAAc,IAAI,KAAK,MAAM,qBAAqB,IAAI,IACxD,OACA,MAAM,gBAAgB,MAAM,KAAK,OAAO,UAAU,OAChD,SACA;CACR,IAAI,CAAC,WACH,OAAO;CAGT,OACE,UAAU,YAAY,MAAM,cAAc;EACxC,MAAM,aAAa,UAAU;EAC7B,IAAI,CAAC,MAAM,iBAAiB,UAAU,GACpC,OAAO;EAET,MAAM,SAAS,WAAW;EAC1B,OACE,MAAM,mBAAmB,MAAM,KAC/B,CAAC,OAAO,YACR,MAAM,aAAa,OAAO,UAAU,EAAE,MAAM,QAAQ,CAAC;CAEzD,CAAC,KAAK;AAEV;AAEA,SAAS,qBACP,QACA,OACe;CACf,IAAIA,OAAK,KAAK,UAAU,SAAS,GAC/B,OAAO;CAET,MAAM,SAASA,OAAK,KAAK;CACzB,IACE,CAAC,MAAM,mBAAmB,MAAM,KAChC,OAAO,YACP,CAAC,MAAM,aAAa,OAAO,UAAU,EAAE,MAAM,WAAW,CAAC,GAEzD,OAAO;CAET,MAAM,KAAKA,OAAK,KAAK,UAAUA,OAAK,KAAK,UAAU,SAAS;CAC5D,IAAI,MAAM,aAAa,EAAE,GACvB,OAAO,GAAG;CAEZ,IAAI,MAAM,mBAAmB,EAAE,KAAK,CAAC,GAAG,UACtC,OAAO,aAAa,GAAG,UAAU,KAAK;CAExC,IAAI,MAAM,qBAAqB,EAAE,KAAK,GAAG,IACvC,OAAO,GAAG,GAAG;CAEf,OAAO;AACT;AAEA,MAAM,wBAAwB;AAC9B,MAAM,sCAAsB,IAAI,IAAI,CAAC,mBAAmB,gBAAgB,CAAC;AACzE,MAAM,yBAAyB;AAE/B,MAAM,eAAe;AACrB,MAAM,mCAAmC;AAEzC,SAAS,qBACP,SACA,OACS;CACT,OAAO,QAAQ,KAAK,MACjB,cACC,MAAM,oBAAoB,SAAS,KACnC,UAAU,OAAO,UAAU,gBAC3B,UAAU,eAAe,UACzB,UAAU,WAAW,MAClB,cACC,CAAC,MAAM,kBAAkB,SAAS,KAClC,UAAU,eAAe,MAC7B,CACJ;AACF;AAEA,MAAM,mBAAmB;CAAC;CAAgB;CAAW;CAAO;AAAc;AAC1E,MAAM,2BAA2B;AACjC,MAAM,wDAAwC,IAAI,IAAqB;AAEvE,SAAS,gBAAgB,QAAuB;CAC9C,IAAI;EACF,OAAO,KAAK,OAAA,GAAMC,GAAAA,aAAAA,CAAaD,QAAM,MAAM,CAAC;CAC9C,QAAQ;EACN;CACF;AACF;AAEA,SAAS,wBAAwB,aAA+B;CAC9D,IAAI,OAAO,gBAAgB,YAAY,gBAAgB,MACrD,OAAO;CAET,MAAM,UAAW,YAAsC;CACvD,OACE,OAAO,YAAY,YACnB,YAAY,QACZ,4BAA4B;AAEhC;AAEA,SAAS,kCAAkC,WAA4B;CACrE,MAAM,SAAS,sCAAsC,IAAI,SAAS;CAClE,IAAI,WAAW,KAAA,GACb,OAAO;CAET,MAAM,cAAc,iBAAA,GAAgBE,KAAAA,KAAAA,CAAK,WAAW,GAAG,gBAAgB,CAAC;CACxE,MAAM,UAAA,GAASC,KAAAA,QAAAA,CAAQ,SAAS;CAChC,MAAM,WACJ,gBAAgB,KAAA,IACZ,wBAAwB,WAAW,IACnC,WAAW,aAAa,kCAAkC,MAAM;CACtE,sCAAsC,IAAI,WAAW,QAAQ;CAC7D,OAAO;AACT;AAEA,SAAS,4BAA4B,UAA2B;CAC9D,QAAA,GACEC,KAAAA,WAAAA,CAAW,QAAQ,KAAK,mCAAA,GAAkCD,KAAAA,QAAAA,CAAQ,QAAQ,CAAC;AAE/E;AAEA,SAAS,0BACP,UACA,YACe;CACf,IAAI,eAAe,OACjB,OAAO;CAET,IAAI,eAAe,KAAA,GACjB,OAAO;CAET,OAAO,4BAA4B,QAAQ,IACvC,mCACA;AACN;AAEA,SAAS,sBAAsB,MAAuB;CACpD,OAAO,KAAK,MAAM,GAAG,IAAI,CAAC,CAAC,SAAS,qBAAqB;AAC3D;AAEA,SAAS,mBACP,QACA,OACS;CACT,MAAM,OAAOH,OAAK;CAClB,IAAI,CAAC,MAAM,cAAc,IAAI,KAAK,KAAK,SAAS,YAAY,KAAK,QAC/D,OAAO;CAET,MAAM,YAAYA,OAAK,YAAY;CACnC,IACE,CAAC,aACD,EAAE,UAAU,mBAAmB,KAAK,UAAU,kBAAkB,MAChE,CAAC,UAAU,KAAK,MAChB,CAAC,oBAAoB,IAAI,UAAU,KAAK,GAAG,IAAI,GAE/C,OAAO;CAET,MAAM,aAAa,kBAAkB,MAAM,KAAK;CAChD,MAAM,OAAO,WAAW,WAAW,SAAS;CAC5C,OAAO,MAAM,aAAa,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAClE;AAEA,SAAS,uBACP,eACA,eACA,OACgC;CAChC,OAAO,MAAM,oBAAoB,SAAS,CACxC,MAAM,mBACJ,MAAM,WAAW,sBAAsB,GACvC,MAAM,sBACJ,MAAM,iBACJ,OACA,MAAM,UAAU,aAAa,GAC7B,MAAM,WAAW,WAAW,CAC9B,GACA,MAAM,UAAU,aAAa,GAC7B,MAAM,eAAe,MAAM,UAAU,aAAa,GAAG,CACnD,MAAM,UAAU,aAAa,GAC7B,MAAM,iBACJ,MAAM,eAAe,GACrB,MAAM,WAAW,aAAa,CAChC,CACF,CAAC,CACH,CACF,CACF,CAAC;AACH;AAEA,SAAS,kBACP,MACA,OAC+B;CAC/B,OAAO,KAAK,OAAO,QAChB,cAAc,CAAC,MAAM,aAAa,WAAW,EAAE,MAAM,OAAO,CAAC,CAChE;AACF;AAEA,SAAS,gBACP,QACA,OAC8C;CAC9C,MAAM,WAAoE,CAAC;CAE3E,KAAK,MAAM,CAAC,OAAO,cAAc,kBAC/BA,OAAK,MACL,KACF,CAAC,CAAC,QAAQ,GAAG;EACX,IAAI,CAAC,MAAM,0BAA0BA,OAAK,IAAI,GAAG;GAC/C,IAAI,MAAM,cAAc,SAAS,GAC/B,SAAS,KACP,MAAM,cACJ,MAAM,eACJ,MAAM,iBACJ,MAAM,gBAAgB,CACpB,MAAM,cAAc,MAAM,WAAW,WAAW,CAAC,CACnD,CAAC,GACD,MAAM,WAAW,OAAO,CAC1B,GACA,CAAC,MAAM,eAAe,KAAK,CAAC,CAC9B,CACF,CACF;QAEA,SAAS,KACP,MAAM,iBACJ,MAAM,WAAW,WAAW,GAC5B,MAAM,eAAe,KAAK,GAC1B,IACF,CACF;GAEF;EACF;EAEA,IAAI,MAAM,aAAa,SAAS,GAC9B,SAAS,KAAK,MAAM,WAAW,UAAU,IAAI,CAAC;OACzC,IACL,MAAM,cAAc,SAAS,KAC7B,MAAM,aAAa,UAAU,QAAQ,GAErC,SAAS,KACP,MAAM,cAAc,MAAM,WAAW,UAAU,SAAS,IAAI,CAAC,CAC/D;CAEJ;CAEA,OAAO,MAAM,gBAAgB,QAAQ;AACvC;AAEA,SAAS,qBACP,SACA,OACS;CACT,OACE,MAAM,0BAA0BA,QAAK,IAAI,KACzCA,QAAK,KAAK,OAAO,MACd,cACC,CAAC,MAAM,aAAa,SAAS,KAC7B,EACE,MAAM,cAAc,SAAS,KAC7B,MAAM,aAAa,UAAU,QAAQ,EAE3C;AAEJ;AAEA,SAAS,cACP,SACA,OACS;CACT,MAAM,aAAa,kBAAkBA,QAAK,MAAM,KAAK;CACrD,OAAO,WAAW,WAAW,KAAK,MAAM,cAAc,WAAW,EAAE;AACrE;AAkBA,SAAS,cACP,MACA,MACA,OAC2E;CAC3E,QACG,MAAM,mBAAmB,IAAI,KAC5B,MAAM,2BAA2B,IAAI,MACvC,CAAC,KAAK,YACN,MAAM,aAAa,KAAK,UAAU,EAAE,KAAK,CAAC;AAE9C;AAEA,SAAS,0BACP,UACA,OACS;CACT,OACE,cAAc,UAAU,eAAe,KAAK,KAC5C,cAAc,SAAS,QAAQ,QAAQ,KAAK;AAEhD;AAEA,SAAS,mBACP,UACA,OACS;CACT,OACE,cAAc,UAAU,YAAY,KAAK,KACzC,CAAC,cAAc,SAAS,QAAQ,QAAQ,KAAK;AAEjD;AAEA,SAAS,oBACP,UACA,OACS;CACT,OAAO,cAAc,UAAU,aAAa,KAAK;AACnD;AAEA,SAAS,qBACP,UACA,OACS;CACT,OAAO,cAAc,UAAU,cAAc,KAAK;AACpD;AAEA,SAAS,yBACP,UACA,OACS;CACT,OACE,cAAc,UAAU,kBAAkB,KAAK,KAC/C,cAAc,SAAS,QAAQ,SAAS,KAAK;AAEjD;AAEA,SAAS,iBACP,UACA,OACS;CACT,OAAO,cAAc,UAAU,UAAU,KAAK;AAChD;AAEA,MAAM,sBAAsB;CAC1B,eAAe;CACf,gBAAgB;CAChB,cAAc;AAChB;AAEA,MAAM,wBAAuD;CAC3D;EACE,SAAS,CAAC,UAAU;EACpB,eAAe;EACf,gBAAgB;EAChB,cAAc;EACd,uBAAuB;CACzB;CACA;EACE,GAAG;EACH,UAAU;EACV,SAAS;GAAC;GAAU;GAAS;EAAQ;EACrC,iBAAiB;CACnB;CACA;EACE,GAAG;EACH,UAAU;EACV,SAAS,CAAC,QAAQ;EAClB,iBAAiB;CACnB;CACA;EACE,GAAG;EACH,UAAU;EACV,SAAS,CAAC,QAAQ;EAClB,iBAAiB;CACnB;CACA;EACE,GAAG;EACH,UAAU;EACV,SAAS;GAAC;GAAU;GAAS;EAAQ;EACrC,iBAAiB;CACnB;CACA;EACE,GAAG;EACH,UAAU;EACV,SAAS;GAAC;GAAU;GAAS;EAAQ;EACrC,iBAAiB;CACnB;CACA;EACE,GAAG;EACH,UAAU;EACV,SAAS,CAAC,mBAAmB,uBAAuB;EACpD,iBAAiB;CACnB;AACF;AAEA,MAAM,yBAAyB,CAC7B,GAAG,IAAI,IACL,sBAAsB,KAAK,SAAS,CAAC,KAAK,eAAe,IAAI,CAAC,CAChE,CAAC,CAAC,OAAO,CACX;AAEA,SAAS,mBACP,MACA,OAKO;CACP,MAAM,SAAS,KAAK;CACpB,IACE,EACE,MAAM,mBAAmB,MAAM,KAC/B,MAAM,2BAA2B,MAAM,MAEzC,OAAO,YACP,CAAC,MAAM,aAAa,OAAO,QAAQ,KACnC,MAAM,QAAQ,OAAO,MAAM,GAE3B,OAAO;CAET,MAAM,SAAS,OAAO,SAAS;CAC/B,MAAM,WAAW,OAAO;CACxB,MAAM,OAAO,sBAAsB,MAChC,cACC,UAAU,QAAQ,SAAS,MAAM,KACjC,UAAU,gBAAgB,UAAU,KAAK,CAC7C;CACA,OAAO,SAAS,KAAA,IAAY,OAAO;EAAE;EAAM;EAAU;CAAO;AAC9D;AAEA,SAAS,gBAAgB,MAAuB,OAA+B;CAC7E,OACE,MAAM,2BAA2B,IAAI,KACrC,MAAM,yBAAyB,IAAI;AAEvC;AAEA,SAAS,uBACP,SACA,OACS;CACT,MAAM,SAASA,QAAK;CACpB,OACG,MAAM,2BAA2B,MAAM,KAAK,OAAO,WAAWA,QAAK,QACnE,MAAM,yBAAyB,MAAM,KAAK,OAAO,WAAWA,QAAK;AAEtE;AAEA,SAAS,0BACP,SACA,OACA,OACM;CACN,IAAI,MAAM,YAAY,MAAM,eAC1B;CAEF,MAAM,OAAOA,QAAK;CAClB,MAAM,SAAS,mBAAmB,MAAM,KAAK;CAC7C,MAAM,aACJ,WAAW,OACP,KAAA,IACA,MAAM,yBAAyB,IAAI,OAAO,KAAK,aAAa;CAClE,IAAI,WAAW,QAAQ,eAAe,KAAA,GACpC;CAEF,MAAM,EAAE,MAAM,cAAc,UAAU,WAAW;CACjD,MAAM,mBACJ,MAAM,2BAA2B,KAAK,MAAM,KAAK,KAAK,OAAO;CAC/D,MAAM,iBAAiB,MAAM,yBAAyB,IAAI,KAAK,KAAK;CACpE,IAAI,MAAM,yBAAyB,IAAI,GAEnC;MAAA,uBACEA,SACA,KACF,KACC,CAAC,oBAAoB,gBAAgB,UAAU,KAAK,GAErD;CAAA;CAGJ,MAAM,UAAU,oBAAoB;CACpC,MAAM,oBACJ,WAAW,CAACA,QAAK,MAAM,SAAS,QAAQ,IACpCA,QAAK,MAAM,8BAA8B,aAAa,YAAY,IAClE;CACN,MAAM,sBACJ,sBAAsB,OAClB,MAAM,UAAU,QAAQ,IACxB,MAAM,UAAU,iBAAiB;CACvC,MAAM,OAAO,MAAM,eACjB,MAAM,UAAU,UAAU,GAC1B,aAAa,aAAa,KAAA,IACtB,CAAC,cAAc,GAAG,GAAG,KAAK,SAAS,IACnC;EACE,MAAM,cAAc,aAAa,QAAQ;EACzC,cAAc;EACd,MAAM,cAAc,MAAM;EAC1B,GAAG,KAAK;CACV,CACN;CACA,MAAM,cAAc;CACpB,MAAM,mBAAmB,IAAI,aAAa,aAAa;CACvD,IAAI,CAAC,SAAS;EACZ,QAAK,YAAY,IAAI;EACrB;CACF;CACA,MAAM,gBACJ,sBAAsB,OAClB,MAAM,UAAU,QAAQ,IACxB,MAAM,qBACJ,KACA,MAAM,UAAU,iBAAiB,GACjC,QACF;CACN,MAAM,WAAW,UACf,MAAM,iBAAiB,MAAM,OAAO,MAAM,YAAY,CAAC;CACzD,MAAM,YAAY,UAChB,MAAM,iBAAiB,OAAO,MAAM,WAAW,MAAM,CAAC;CACxD,MAAM,OAAO,mBACT,iBACE,MAAM,kBACJ,MACA,QAAQ,aAAa,GACrB,QAAQ,SAAS,cAAc,CAAC,CAAC,CACnC,IACA,QAAQ,aAAa,IACvB,QAAQ,SAAS,aAAa,CAAC;CACnC,QAAK,YACH,MAAM,sBACJ,MACA,MAAM,gBAAgB,QAAQ,MAAM,eAAe,CAAC,CAAC,GACrD,IACF,CACF;AACF;AAQA,SAAwB,kBAAkB,KAEN;CAClC,MAAM,EAAE,UAAU;CAElB,OAAO;EACL,MAAM;EACN,SAAS;GACP,SAAS;IACP,MAAM,SAAoC,OAA0B;KAElE,MAAM,WACJ,CAAC,0BAFc,MAAM,KAAK,KAAK,YAAY,WAEN,MAAM,IAAI,KAC/CA,QAAK,MAAM,WAAA,sCAAsC,KACjDA,QAAK,KAAK,WAAW,MAClB,cAAc,UAAU,MAAM,UAAU,YAC3C;KACF,MAAM,oBACJA,QAAK,MAAM,sBAAsB,gBAAgB;KACnD,MAAM,mBAAmBA,QAAK,MAAM,sBAClC,uBACF;KACA,MAAM,oBACJA,QAAK,MAAM,sBAAsB,gBAAgB;KACnD,MAAM,0BAA0B,IAAI,IAClC,uBAAuB,KAAK,SAAS,CACnC,KAAK,eACLA,QAAK,MAAM,sBAAsB,KAAK,cAAc,CACtD,CAAC,CACH;KACA,MAAM,gBAAgB,sBAAsB,MAAM,KAAK,QAAQ,EAAE;KACjE,MAAM,cAAc,CAAC;KACrB,MAAM,cAAc;KACpB,MAAM,cAAc;KACpB,MAAM,oCAAoB,IAAI,IAAI;IACpC;IACA,KAAK,SAAoC,OAA0B;KACjE,MAAM,yBACJ,CAAC,MAAM,YACP,MAAM,iBACN,qBAAqBA,QAAK,MAAM,KAAK,IACjC,0BACE,eAAe,MAAM,KAAK,KAAK,YAAY,EAAE,GAC7C,MAAM,KAAK,sBACb,IACA;KACN,IAAI,2BAA2B,MAAM;MACnC,QAAK,iBACH,QACA,MAAM,kBACJ,CAAC,GACD,MAAM,cAAc,sBAAsB,CAC5C,CACF;MACA,MAAM,KAAK,SAAS,+BAA+B;KACrD;KACA,IACE,MAAM,YACN,CAAC,MAAM,eACP,CAAC,MAAM,qBACP,CAAC,MAAM,oBACP,CAAC,MAAM,mBAEP;KAGF,MAAM,KAAK,SAAS,+BAA+B;KACnD,QAAK,iBAAiB,QAAQ;MAC5B,MAAM,kBACJ;OACE,MAAM,gBACJ,MAAM,mBACN,MAAM,WAAW,kBAAkB,CACrC;OACA,MAAM,gBACJ,MAAM,kBACN,MAAM,WAAW,yBAAyB,CAC5C;OACA,GAAI,MAAM,cACN,CACE,MAAM,gBACJ,MAAM,mBACN,MAAM,WAAW,kBAAkB,CACrC,CACF,IACA,CAAC;OACL,GAAG,uBAAuB,SAAS,SAAS;QAC1C,MAAM,aAAa,MAAM,yBAAyB,IAChD,KAAK,aACP;QACA,OAAO,MAAM,mBAAmB,IAAI,KAAK,aAAa,KACpD,aACE,CACE,MAAM,gBACJ,YACA,MAAM,WAAW,KAAK,aAAa,CACrC,CACF,IACA,CAAC;OACP,CAAC;MACH,GACA,MAAM,cACJ,MAAM,KAAK,iBAAiB,kBAC9B,CACF;MACA,MAAM,oBAAoB,SAAS,CACjC,MAAM,mBACJ,MAAM,WAAW,2BAA2B,GAC5C,MAAM,qBACJ,KACA,MAAM,iBACJ,MAAM,WAAW,YAAY,GAC7B,MAAM,WAAW,2BAA2B,CAC9C,GACA,MAAM,eAAe,IAAI,CAC3B,CACF,CACF,CAAC;MACD,GAAI,MAAM,eAAe,CAAC;KAC5B,CAAC;IACH;GACF;GACA,gBAAgB,EACd,KACE,SACA,OACA;IACA,IAAI,MAAM,UACR;IAEF,IAAI,mBAAmBA,QAAK,MAAM,KAAK,MAAM,MAAM;KACjD,0BAA0BA,SAAM,OAAO,KAAK;KAC5C;IACF;IACA,MAAM,iBAAiB,qBAAqBA,SAAM,KAAK;IACvD,IAAI,mBAAmB,MACrB;IAEF,IAAIA,QAAK,KAAK,UAAU,WAAW,GACjC,QAAK,KAAK,UAAU,KAAK,MAAM,WAAW,WAAW,CAAC;IAExD,QAAK,KAAK,UAAU,KAAK,MAAM,cAAc,cAAc,CAAC;GAC9D,EACF;GACA,wBAAwB,EACtB,KACE,SACA,OACA;IACA,0BAA0BA,SAAM,OAAO,KAAK;GAC9C,EACF;GACA,UAAU,EACR,KAAK,SAAqC,OAA0B;IAClE,MAAM,OAAOA,QAAK;IAClB,IACE,MAAM,YACN,KAAK,uBACL,CAAC,yBAAyB,MAAM,KAAK,KACrC,qBAAqBA,SAAM,KAAK,KAChC,CAAC,MAAM,qBACP,CAAC,MAAM,oBACP,CAAC,MAAM,mBAEP;IAEF,MAAM,aACJ,MAAM,kBAAkB,QAAQ,mBAAmBA,SAAM,KAAK;IAChE,IAAI,MAAM,iBAAiB,CAAC,YAC1B;IAEF,IAAI,sBAAsBA,SAAM,KAAK,GACnC;IAEF,MAAM,OAAO,sBAAsBA,SAAM,KAAK;IAC9C,IAAI,KAAK,WAAW,YAAY,GAC9B;IAEF,MAAM,qBAAqB,kBACzB,MAAM,KAAK,KAAK,YAAY,WAC5B,MAAM,KAAK,OAAO,MAAM,KAAK,KAAK,OAAO,QAAQ,IAAI,CACvD;IACA,MAAM,EAAE,MAAM,WAAW,eAAeA,OAAI;IAC5C,MAAM,aAAa,MAAM,iBAAiB;KACxC,MAAM,eACJ,MAAM,WAAW,IAAI,GACrB,MAAM,cACJ,oBAAoB,oBAAoB,IAAI,CAC9C,CACF;KACA,MAAM,eACJ,MAAM,WAAW,MAAM,GACvB,MAAM,cAAc,IAAI,CAC1B;KACA,MAAM,eACJ,MAAM,WAAW,MAAM,GACvB,MAAM,cAAc,kBAAkB,CACxC;KACA,MAAM,eACJ,MAAM,WAAW,MAAM,GACvB,MAAM,eAAe,IAAI,CAC3B;KACA,MAAM,eACJ,MAAM,WAAW,QAAQ,GACzB,MAAM,eAAe,MAAM,CAC7B;KACA,GAAI,KAAK,QACL,CACE,MAAM,eACJ,MAAM,WAAW,OAAO,GACxB,MAAM,eAAe,IAAI,CAC3B,CACF,IACA,CAAC;KACL,GAAI,aACA,CACE,MAAM,eACJ,MAAM,WAAW,MAAM,GACvB,MAAM,eAAe,IAAI,CAC3B,CACF,IACA,CAAC;KACL,GAAI,cAAcA,SAAM,KAAK,IACzB,CACE,MAAM,eACJ,MAAM,WAAW,SAAS,GAC1B,MAAM,eAAe,IAAI,CAC3B,CACF,IACA,CAAC;IACP,CAAC;IACD,MAAM,uBAAuBA,QAAK,MAC/B,iBAAiB,CAAC,CAClB,sBAAsB,sBAAsB;IAC/C,MAAM,aAAa,KACjB,MAAM,oBAAoB,SAAS,CACjC,MAAM,mBAAmB,sBAAsB,UAAU,CAC3D,CAAC,CACH;IAEA,IAAI,qBAAqBA,SAAM,KAAK,GAAG;KACrC,KAAK,sBAAsB;KAC3B,QAAK,YACH,MAAM,eAAe,MAAM,UAAU,MAAM,iBAAiB,GAAG,CAC7D,MAAM,UAAU,oBAAoB,GACpC,IACF,CAAC,CACH;KACA,MAAM,cAAc;KACpB,MAAM,cAAc;KACpB;IACF;IAEA,MAAM,eAAe,MAAM,iBAAiB,KAAK,IAAI,IACjD,KAAK,OACL,MAAM,eAAe,CAAC,MAAM,gBAAgB,KAAK,IAAI,CAAC,CAAC;IAC3D,MAAM,aAAa,aAAa;IAChC,aAAa,aAAa,CAAC;IAE3B,MAAM,gBAAgB,aAClBA,QAAK,MAAM,sBAAsB,mBAAmB,IACpD,KAAA;IACJ,MAAM,gBAAgB,aAClBA,QAAK,MAAM,sBAAsB,yBAAyB,IAC1D,KAAA;IACJ,MAAM,aACJ,iBAAiB,gBACb,MAAM,eACJ,CACE,uBAAuB,eAAe,eAAe,KAAK,GAC1D,GAAG,MAAM,UAAU,cAAc,IAAI,CAAC,CAAC,IACzC,GACA,CAAC,CACH,IACA,MAAM,UAAU,cAAc,IAAI;IAuCxC,KAAK,OAtCQ,MAAM,eACjB,CACE,MAAM,YACJ,MAAM,eACJ,MAAM,UAAU,MAAM,gBAAgB,GACtC,CAAC,CACH,GACA,MAAM,eAAe,CACnB,GAAI,gBACA,CACE,MAAM,oBAAoB,SAAS,CACjC,MAAM,mBACJ,eACA,MAAM,WAAW,sBAAsB,CACzC,CACF,CAAC,CACH,IACA,CAAC,GACL,MAAM,gBACJ,MAAM,eACJ,MAAM,UAAU,MAAM,iBAAiB,GACvC;KACE,MAAM,UAAU,oBAAoB;KACpC,gBAAgBA,SAAM,KAAK;KAC3B,MAAM,wBACJ,gBAAgB,CAAC,MAAM,UAAU,aAAa,CAAC,IAAI,CAAC,GACpD,YACA,KAAK,SAAS,KAChB;IACF,CACF,CACF,CACF,CAAC,CACH,GACA,GAAG,aAAa,IAClB,GACA,UAEa;IACf,KAAK,sBAAsB;IAC3B,MAAM,cAAc;GACtB,EACF;EACF;CACF;AACF"}

package/dist/{babelPlugin-PgjdBuQe.js → babelPlugin-tupc_WZS.js}

@@ -1,3 +1,5 @@
+import { readFileSync } from "fs";
+import { dirname, isAbsolute, join } from "path";
 //#region ../bitfab-typescript-transform/src/transformOptions.ts
 const AUTO_TRACE_FUNCTION_ID_VERSION = "ts-auto-v1";
 const AUTO_TRACE_TRANSFORM_MARKER = "__bitfab_auto_trace_transformed_v1__";
@@ -113,6 +115,48 @@ function withNodeFunctionName(path, types) {
 const BAML_GENERATED_HEADER = "This file was generated by BAML";
 const BAML_CLIENT_CLASSES = /* @__PURE__ */ new Set(["BamlAsyncClient", "BamlSyncClient"]);
 const BAML_OPTIONS_PARAMETER = "__baml_options__";
+const BAML_PACKAGE = "@boundaryml/baml";
+const DEFAULT_BAML_REGISTRATION_MODULE = "@bitfab/sdk/baml";
+function importsBamlAtRuntime(program, types) {
+	return program.body.some((statement) => types.isImportDeclaration(statement) && statement.source.value === BAML_PACKAGE && statement.importKind !== "type" && statement.specifiers.some((specifier) => !types.isImportSpecifier(specifier) || specifier.importKind !== "type"));
+}
+const SDK_PACKAGE_PATH = [
+	"node_modules",
+	"@bitfab",
+	"sdk",
+	"package.json"
+];
+const BAML_REGISTRATION_EXPORT = "./baml";
+const sdkExportsBamlRegistrationByDirectory = /* @__PURE__ */ new Map();
+function readPackageJson(path) {
+	try {
+		return JSON.parse(readFileSync(path, "utf8"));
+	} catch {
+		return;
+	}
+}
+function exportsBamlRegistration(packageJson) {
+	if (typeof packageJson !== "object" || packageJson === null) return false;
+	const exports = packageJson.exports;
+	return typeof exports === "object" && exports !== null && BAML_REGISTRATION_EXPORT in exports;
+}
+function nearestSdkExportsBamlRegistration(directory) {
+	const cached = sdkExportsBamlRegistrationByDirectory.get(directory);
+	if (cached !== void 0) return cached;
+	const packageJson = readPackageJson(join(directory, ...SDK_PACKAGE_PATH));
+	const parent = dirname(directory);
+	const exported = packageJson !== void 0 ? exportsBamlRegistration(packageJson) : parent !== directory && nearestSdkExportsBamlRegistration(parent);
+	sdkExportsBamlRegistrationByDirectory.set(directory, exported);
+	return exported;
+}
+function defaultRegistrationResolves(filename) {
+	return isAbsolute(filename) && nearestSdkExportsBamlRegistration(dirname(filename));
+}
+function bamlRegistrationModuleFor(filename, configured) {
+	if (configured === false) return null;
+	if (configured !== void 0) return configured;
+	return defaultRegistrationResolves(filename) ? DEFAULT_BAML_REGISTRATION_MODULE : null;
+}
 function isBamlGeneratedSource(code) {
 	return code.slice(0, 4096).includes(BAML_GENERATED_HEADER);
 }
@@ -306,6 +350,11 @@ function bitfabBabelPlugin(api) {
 					state.usedRecordedCalls = /* @__PURE__ */ new Set();
 				},
 				exit(path, state) {
+					const bamlRegistrationModule = !state.skipFile && state.bamlGenerated && importsBamlAtRuntime(path.node, types) ? bamlRegistrationModuleFor(sourceFilename(state.file.opts.filename ?? ""), state.opts.bamlRegistrationModule) : null;
+					if (bamlRegistrationModule !== null) {
+						path.unshiftContainer("body", types.importDeclaration([], types.stringLiteral(bamlRegistrationModule)));
+						state.file.metadata[AUTO_TRACE_TRANSFORM_MARKER] = true;
+					}
 					if (state.skipFile || !state.usedRuntime || !state.runtimeIdentifier || !state.activeIdentifier || !state.wrapperIdentifier) return;
 					state.file.metadata[AUTO_TRACE_TRANSFORM_MARKER] = true;
 					path.unshiftContainer("body", [
@@ -386,4 +435,4 @@ function bitfabBabelPlugin(api) {
 //#endregion
 export { sourceFilename as i, AUTO_TRACE_TRANSFORM_MARKER as n, shouldTransformBitfabFile as r, bitfabBabelPlugin as t };
 
-//# sourceMappingURL=babelPlugin-PgjdBuQe.js.map
+//# sourceMappingURL=babelPlugin-tupc_WZS.js.map