unplugin-typegpu 0.10.1 → 0.11.0

package/babel.mjs

@@ -1,159 +1,2 @@
-import { defaultOptions, embedJSON, gatherTgpuAliases, getFunctionName, isShellImplementationCall, operators, performExpressionNaming, useGpuDirective } from "./common.mjs";
-import { createFilterForId } from "./filter.mjs";
-import defu from "defu";
-import { transpileFn } from "tinyest-for-wgsl";
-import { FORMAT_VERSION } from "tinyest";
-import * as Babel from "@babel/standalone";
-
-//#region src/babel.ts
-const template = Babel.packages.template;
-const types = Babel.packages.types;
-function containsUseGpuDirective(node) {
-	var _node$body$directives, _node$body;
-	return ("directives" in node.body ? (_node$body$directives = (_node$body = node.body) === null || _node$body === void 0 ? void 0 : _node$body.directives) !== null && _node$body$directives !== void 0 ? _node$body$directives : [] : []).map((directive) => directive.value.value).includes(useGpuDirective);
-}
-function i(identifier) {
-	return types.identifier(identifier);
-}
-const fnNodeToOriginalMap = /* @__PURE__ */ new WeakMap();
-function functionToTranspiled(node, parent) {
-	var _fnNodeToOriginalMap$;
-	const { params, body, externalNames } = transpileFn((_fnNodeToOriginalMap$ = fnNodeToOriginalMap.get(node)) !== null && _fnNodeToOriginalMap$ !== void 0 ? _fnNodeToOriginalMap$ : node);
-	const maybeName = getFunctionName(node, parent);
-	const metadata = types.objectExpression([
-		types.objectProperty(i("v"), types.numericLiteral(FORMAT_VERSION)),
-		types.objectProperty(i("name"), maybeName ? types.stringLiteral(maybeName) : types.buildUndefinedNode()),
-		types.objectProperty(i("ast"), template.expression`${embedJSON({
-			params,
-			body,
-			externalNames
-		})}`()),
-		types.objectProperty(i("externals"), types.arrowFunctionExpression([], types.blockStatement([types.returnStatement(types.objectExpression(externalNames.map((name) => types.objectProperty(i(name), i(name), false, name !== "this"))))])))
-	]);
-	return types.callExpression(types.arrowFunctionExpression([i("$")], types.logicalExpression("&&", types.callExpression(types.memberExpression(types.assignmentExpression("??=", types.memberExpression(i("globalThis"), i("__TYPEGPU_META__")), types.newExpression(i("WeakMap"), [])), i("set")), [types.assignmentExpression("=", types.memberExpression(i("$"), i("f")), node), metadata]), types.memberExpression(i("$"), i("f")))), [types.objectExpression([])]);
-}
-function wrapInAutoName(node, name) {
-	return types.callExpression(template.expression("globalThis.__TYPEGPU_AUTONAME__ ?? (a => a)", { placeholderPattern: false })(), [node, types.stringLiteral(name)]);
-}
-function functionVisitor(ctx) {
-	let inUseGpuScope = false;
-	return {
-		VariableDeclarator(path) {
-			performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("init").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		AssignmentExpression(path) {
-			if (inUseGpuScope) {
-				const runtimeFn = operators[path.node.operator];
-				if (runtimeFn) path.replaceWith(types.assignmentExpression("=", path.node.left, types.callExpression(types.identifier(runtimeFn), [path.node.left, path.node.right])));
-			}
-			performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("right").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		ObjectProperty(path) {
-			performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("value").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		ClassProperty(path) {
-			performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("value").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		ImportDeclaration(path) {
-			gatherTgpuAliases(path.node, ctx);
-		},
-		BinaryExpression: { exit(path) {
-			if (!inUseGpuScope) return;
-			const runtimeFn = operators[path.node.operator];
-			if (runtimeFn) path.replaceWith(types.callExpression(types.identifier(runtimeFn), [path.node.left, path.node.right]));
-		} },
-		ArrowFunctionExpression: {
-			enter(path) {
-				if (containsUseGpuDirective(path.node)) {
-					fnNodeToOriginalMap.set(path.node, types.cloneNode(path.node, true));
-					if (inUseGpuScope) throw new Error(`Nesting 'use gpu' functions is not allowed`);
-					inUseGpuScope = true;
-				}
-			},
-			exit(path) {
-				const node = path.node;
-				if (containsUseGpuDirective(node)) {
-					inUseGpuScope = false;
-					const parent = path.parentPath.node;
-					path.replaceWith(functionToTranspiled(node, parent));
-					path.skip();
-				}
-			}
-		},
-		FunctionExpression: {
-			enter(path) {
-				if (containsUseGpuDirective(path.node)) {
-					fnNodeToOriginalMap.set(path.node, types.cloneNode(path.node, true));
-					if (inUseGpuScope) throw new Error(`Nesting 'use gpu' functions is not allowed`);
-					inUseGpuScope = true;
-				}
-			},
-			exit(path) {
-				const node = path.node;
-				if (containsUseGpuDirective(node)) {
-					inUseGpuScope = false;
-					const parent = path.parentPath.node;
-					path.replaceWith(functionToTranspiled(node, parent));
-					path.skip();
-				}
-			}
-		},
-		FunctionDeclaration: {
-			enter(path) {
-				if (containsUseGpuDirective(path.node)) {
-					fnNodeToOriginalMap.set(path.node, types.cloneNode(path.node, true));
-					if (inUseGpuScope) throw new Error(`Nesting 'use gpu' functions is not allowed`);
-					inUseGpuScope = true;
-				}
-			},
-			exit(path) {
-				var _fnNodeToOriginalMap$2;
-				const node = (_fnNodeToOriginalMap$2 = fnNodeToOriginalMap.get(path.node)) !== null && _fnNodeToOriginalMap$2 !== void 0 ? _fnNodeToOriginalMap$2 : path.node;
-				if (containsUseGpuDirective(node)) {
-					inUseGpuScope = false;
-					if (!node.id) return;
-					const parent = path.parentPath.node;
-					const expression = types.functionExpression(node.id, node.params, node.body);
-					path.replaceWith(types.variableDeclaration("const", [types.variableDeclarator(node.id, functionToTranspiled(expression, parent))]));
-					path.skip();
-				}
-			}
-		},
-		CallExpression: { exit(path) {
-			const node = path.node;
-			if (isShellImplementationCall(node, ctx)) {
-				const implementation = node.arguments[0];
-				if (implementation && (implementation.type === "FunctionExpression" || implementation.type === "ArrowFunctionExpression")) {
-					const transpiled = functionToTranspiled(implementation, null);
-					path.replaceWith(types.callExpression(node.callee, [transpiled]));
-					path.skip();
-				}
-			}
-		} }
-	};
-}
-function babel_default() {
-	return { visitor: { Program(path, state) {
-		const options = defu(state.opts, defaultOptions);
-		const id = state.filename;
-		const filter = createFilterForId(options);
-		if (id && filter && !(filter === null || filter === void 0 ? void 0 : filter(id))) return;
-		const ctx = {
-			tgpuAliases: new Set(options.forceTgpuAlias ? [options.forceTgpuAlias] : []),
-			fileId: id,
-			autoNamingEnabled: options.autoNamingEnabled
-		};
-		path.traverse(functionVisitor(ctx));
-	} } };
-}
-
-//#endregion
-export { babel_default as default };
+import { t as TypeGPUPlugin } from "./babel-Bnfeonoo.mjs";
+export { TypeGPUPlugin as default };

package/bun-9JrHWE4u.d.mts

@@ -0,0 +1,6 @@
+import { t as Options } from "./common-BvAFT5IV.mjs";
+
+//#region src/bun.d.ts
+declare const _default: (rawOptions?: Options) => Bun.BunPlugin;
+//#endregion
+export { _default as t };

package/bun-CUGitP44.cjs

@@ -0,0 +1,147 @@
+const require_common = require("./common-Ccx_dEqH.cjs");
+let defu = require("defu");
+defu = require_common.__toESM(defu);
+let magic_string_ast = require("magic-string-ast");
+let ast_kit = require("ast-kit");
+let _babel_traverse = require("@babel/traverse");
+_babel_traverse = require_common.__toESM(_babel_traverse);
+let tinyest = require("tinyest");
+require("tinyest-for-wgsl");
+let _babel_parser = require("@babel/parser");
+_babel_parser = require_common.__toESM(_babel_parser);
+let _babel_types = require("@babel/types");
+_babel_types = require_common.__toESM(_babel_types);
+//#region src/core/factory.ts
+let traverse = _babel_traverse.default;
+if (typeof traverse.default === "function") traverse = traverse.default;
+const fnWrapperTemplate = (fnCode, metadata) => `(/*#__PURE__*/($ => (globalThis.__TYPEGPU_META__ ??= new WeakMap()).set($.f = (${fnCode}), ${metadata}) && $.f)({}))`;
+function embedJSON(jsValue) {
+	return JSON.stringify(jsValue).replace(/\u2028/g, "\\u2028").replace(/\u2029/g, "\\u2029");
+}
+function assignMetadata(path, name, ast) {
+	var _path$node$start, _path$node$start2;
+	const metadata = `{
+    v: ${tinyest.FORMAT_VERSION},
+    name: ${name ? `"${name}"` : "undefined"},
+    ast: ${embedJSON(ast)},
+    externals: () => ({${ast.externalNames.map((e) => e === "this" ? "\"this\": this" : e).join(", ")}}),
+  }`;
+	const visibility = _babel_types.isFunctionDeclaration(path.node) ? require_common.getBlockScope(path) : void 0;
+	const fnCode = this.magicString.sliceNode(path.node);
+	let nodeToOverride = path.node;
+	let code = fnWrapperTemplate(fnCode, metadata);
+	let insertPos = (_path$node$start = path.node.start) !== null && _path$node$start !== void 0 ? _path$node$start : 0;
+	if (_babel_types.isFunctionDeclaration(path.node) && path.node.id) code = `const ${path.node.id.name} = ${code};\n\n`;
+	if (visibility) {
+		var _visibility$node$body, _visibility$node$body2;
+		insertPos = (_visibility$node$body = (_visibility$node$body2 = visibility.node.body[0]) === null || _visibility$node$body2 === void 0 ? void 0 : _visibility$node$body2.start) !== null && _visibility$node$body !== void 0 ? _visibility$node$body : insertPos;
+		if (_babel_types.isExportNamedDeclaration(path.parent)) {
+			nodeToOverride = path.parent;
+			code = `export ${code}`;
+		}
+	}
+	if (insertPos < ((_path$node$start2 = path.node.start) !== null && _path$node$start2 !== void 0 ? _path$node$start2 : 0)) {
+		var _nodeToOverride$leadi, _nodeToOverride$leadi2;
+		for (const comment of (_nodeToOverride$leadi = (_nodeToOverride$leadi2 = nodeToOverride.leadingComments) === null || _nodeToOverride$leadi2 === void 0 ? void 0 : _nodeToOverride$leadi2.toReversed()) !== null && _nodeToOverride$leadi !== void 0 ? _nodeToOverride$leadi : []) {
+			var _comment$start, _comment$end;
+			code = `${this.magicString.slice((_comment$start = comment.start) !== null && _comment$start !== void 0 ? _comment$start : 0, (_comment$end = comment.end) !== null && _comment$end !== void 0 ? _comment$end : 0)}\n${code}`;
+			this.magicString.removeNode(comment);
+		}
+		this.magicString.removeNode(nodeToOverride);
+		this.magicString.prependLeft(insertPos, code);
+	} else this.magicString.overwriteNode(nodeToOverride, code);
+}
+function wrapInAutoName(path, name) {
+	var _path$node$start3, _path$node$end;
+	this.magicString.appendLeft((_path$node$start3 = path.node.start) !== null && _path$node$start3 !== void 0 ? _path$node$start3 : 0, "(/*#__PURE__*/(globalThis.__TYPEGPU_AUTONAME__ ?? (a => a))(").prependRight((_path$node$end = path.node.end) !== null && _path$node$end !== void 0 ? _path$node$end : 0, `, "${name}"))`);
+}
+function replaceWithAssignmentOverload(path, runtimeFn) {
+	const lhs = this.magicString.sliceNode(path.node.left);
+	const rhs = this.magicString.sliceNode(path.node.right);
+	this.magicString.overwriteNode(path.node, `${lhs} = ${runtimeFn}(${lhs}, ${rhs})`);
+}
+function replaceWithBinaryOverload(path, runtimeFn) {
+	const lhs = this.magicString.sliceNode(path.node.left);
+	const rhs = this.magicString.sliceNode(path.node.right);
+	this.magicString.overwriteNode(path.node, `${runtimeFn}(${lhs}, ${rhs})`);
+}
+const unpluginFactory = ((rawOptions, _meta) => {
+	const options = (0, defu.default)(rawOptions, require_common.defaultOptions);
+	return {
+		name: "unplugin-typegpu",
+		enforce: options.enforce,
+		transform: {
+			filter: options.earlyPruning ? {
+				id: options,
+				code: require_common.earlyPruneRegex
+			} : { id: options },
+			handler(code, id) {
+				let ast;
+				try {
+					ast = _babel_parser.parse(code, (0, ast_kit.getBabelParserOptions)((0, ast_kit.getLang)(id), {
+						sourceType: "module",
+						allowReturnOutsideFunction: true
+					}));
+				} catch (cause) {
+					console.warn(`[unplugin-typegpu] Failed to parse ${id}. Cause: ${typeof cause === "object" && cause && "message" in cause ? cause.message : cause}`);
+					return;
+				}
+				const magicString = new magic_string_ast.MagicStringAST(code);
+				const state = {
+					filename: id,
+					magicString,
+					opts: options
+				};
+				require_common.initPluginState(state, {
+					warn: (message) => this.warn(message),
+					assignMetadata,
+					wrapInAutoName,
+					replaceWithAssignmentOverload,
+					replaceWithBinaryOverload
+				});
+				traverse(ast, require_common.functionVisitor, void 0, state);
+				return (0, magic_string_ast.generateTransform)(magicString, id);
+			}
+		}
+	};
+});
+//#endregion
+//#region src/bun.ts
+var bun_default = (rawOptions) => {
+	const options = (0, defu.default)(rawOptions, require_common.defaultOptions);
+	const include = options.include;
+	if (!(include instanceof RegExp)) throw new Error(`Unsupported 'include' options in Bun plugin. Please provide a single regular expression`);
+	if (options.exclude) throw new Error(`Unsupported 'exclude' option in Bun plugin`);
+	const rawPlugin = unpluginFactory(rawOptions, { framework: "bun" });
+	return {
+		name: "unplugin-typegpu",
+		setup(build) {
+			build.onLoad({ filter: include }, async (args) => {
+				var _result$code;
+				const codeIn = await Bun.file(args.path).text();
+				if (require_common.earlyPruneRegex.every((pattern) => !pattern.test(codeIn))) return {
+					contents: codeIn,
+					loader: args.loader
+				};
+				const result = rawPlugin.transform.handler.apply({}, [codeIn, args.path]);
+				return {
+					contents: (_result$code = result === null || result === void 0 ? void 0 : result.code) !== null && _result$code !== void 0 ? _result$code : codeIn,
+					loader: args.loader
+				};
+			});
+		}
+	};
+};
+//#endregion
+Object.defineProperty(exports, "bun_default", {
+	enumerable: true,
+	get: function() {
+		return bun_default;
+	}
+});
+Object.defineProperty(exports, "unpluginFactory", {
+	enumerable: true,
+	get: function() {
+		return unpluginFactory;
+	}
+});

package/bun-ChNeklBe.d.cts

@@ -0,0 +1,6 @@
+import { t as Options } from "./common-BKqBivuE.cjs";
+
+//#region src/bun.d.ts
+declare const _default: (rawOptions?: Options) => Bun.BunPlugin;
+//#endregion
+export { _default as t };

package/bun-DFr_PiLh.mjs

@@ -0,0 +1,132 @@
+import { a as initPluginState, i as getBlockScope, n as earlyPruneRegex, r as functionVisitor, t as defaultOptions } from "./common-e08X6w0a.mjs";
+import defu from "defu";
+import { MagicStringAST, generateTransform } from "magic-string-ast";
+import { getBabelParserOptions, getLang } from "ast-kit";
+import _traverse from "@babel/traverse";
+import { FORMAT_VERSION } from "tinyest";
+import "tinyest-for-wgsl";
+import * as parser from "@babel/parser";
+import * as t from "@babel/types";
+//#region src/core/factory.ts
+let traverse = _traverse;
+if (typeof traverse.default === "function") traverse = traverse.default;
+const fnWrapperTemplate = (fnCode, metadata) => `(/*#__PURE__*/($ => (globalThis.__TYPEGPU_META__ ??= new WeakMap()).set($.f = (${fnCode}), ${metadata}) && $.f)({}))`;
+function embedJSON(jsValue) {
+	return JSON.stringify(jsValue).replace(/\u2028/g, "\\u2028").replace(/\u2029/g, "\\u2029");
+}
+function assignMetadata(path, name, ast) {
+	var _path$node$start, _path$node$start2;
+	const metadata = `{
+    v: ${FORMAT_VERSION},
+    name: ${name ? `"${name}"` : "undefined"},
+    ast: ${embedJSON(ast)},
+    externals: () => ({${ast.externalNames.map((e) => e === "this" ? "\"this\": this" : e).join(", ")}}),
+  }`;
+	const visibility = t.isFunctionDeclaration(path.node) ? getBlockScope(path) : void 0;
+	const fnCode = this.magicString.sliceNode(path.node);
+	let nodeToOverride = path.node;
+	let code = fnWrapperTemplate(fnCode, metadata);
+	let insertPos = (_path$node$start = path.node.start) !== null && _path$node$start !== void 0 ? _path$node$start : 0;
+	if (t.isFunctionDeclaration(path.node) && path.node.id) code = `const ${path.node.id.name} = ${code};\n\n`;
+	if (visibility) {
+		var _visibility$node$body, _visibility$node$body2;
+		insertPos = (_visibility$node$body = (_visibility$node$body2 = visibility.node.body[0]) === null || _visibility$node$body2 === void 0 ? void 0 : _visibility$node$body2.start) !== null && _visibility$node$body !== void 0 ? _visibility$node$body : insertPos;
+		if (t.isExportNamedDeclaration(path.parent)) {
+			nodeToOverride = path.parent;
+			code = `export ${code}`;
+		}
+	}
+	if (insertPos < ((_path$node$start2 = path.node.start) !== null && _path$node$start2 !== void 0 ? _path$node$start2 : 0)) {
+		var _nodeToOverride$leadi, _nodeToOverride$leadi2;
+		for (const comment of (_nodeToOverride$leadi = (_nodeToOverride$leadi2 = nodeToOverride.leadingComments) === null || _nodeToOverride$leadi2 === void 0 ? void 0 : _nodeToOverride$leadi2.toReversed()) !== null && _nodeToOverride$leadi !== void 0 ? _nodeToOverride$leadi : []) {
+			var _comment$start, _comment$end;
+			code = `${this.magicString.slice((_comment$start = comment.start) !== null && _comment$start !== void 0 ? _comment$start : 0, (_comment$end = comment.end) !== null && _comment$end !== void 0 ? _comment$end : 0)}\n${code}`;
+			this.magicString.removeNode(comment);
+		}
+		this.magicString.removeNode(nodeToOverride);
+		this.magicString.prependLeft(insertPos, code);
+	} else this.magicString.overwriteNode(nodeToOverride, code);
+}
+function wrapInAutoName(path, name) {
+	var _path$node$start3, _path$node$end;
+	this.magicString.appendLeft((_path$node$start3 = path.node.start) !== null && _path$node$start3 !== void 0 ? _path$node$start3 : 0, "(/*#__PURE__*/(globalThis.__TYPEGPU_AUTONAME__ ?? (a => a))(").prependRight((_path$node$end = path.node.end) !== null && _path$node$end !== void 0 ? _path$node$end : 0, `, "${name}"))`);
+}
+function replaceWithAssignmentOverload(path, runtimeFn) {
+	const lhs = this.magicString.sliceNode(path.node.left);
+	const rhs = this.magicString.sliceNode(path.node.right);
+	this.magicString.overwriteNode(path.node, `${lhs} = ${runtimeFn}(${lhs}, ${rhs})`);
+}
+function replaceWithBinaryOverload(path, runtimeFn) {
+	const lhs = this.magicString.sliceNode(path.node.left);
+	const rhs = this.magicString.sliceNode(path.node.right);
+	this.magicString.overwriteNode(path.node, `${runtimeFn}(${lhs}, ${rhs})`);
+}
+const unpluginFactory = ((rawOptions, _meta) => {
+	const options = defu(rawOptions, defaultOptions);
+	return {
+		name: "unplugin-typegpu",
+		enforce: options.enforce,
+		transform: {
+			filter: options.earlyPruning ? {
+				id: options,
+				code: earlyPruneRegex
+			} : { id: options },
+			handler(code, id) {
+				let ast;
+				try {
+					ast = parser.parse(code, getBabelParserOptions(getLang(id), {
+						sourceType: "module",
+						allowReturnOutsideFunction: true
+					}));
+				} catch (cause) {
+					console.warn(`[unplugin-typegpu] Failed to parse ${id}. Cause: ${typeof cause === "object" && cause && "message" in cause ? cause.message : cause}`);
+					return;
+				}
+				const magicString = new MagicStringAST(code);
+				const state = {
+					filename: id,
+					magicString,
+					opts: options
+				};
+				initPluginState(state, {
+					warn: (message) => this.warn(message),
+					assignMetadata,
+					wrapInAutoName,
+					replaceWithAssignmentOverload,
+					replaceWithBinaryOverload
+				});
+				traverse(ast, functionVisitor, void 0, state);
+				return generateTransform(magicString, id);
+			}
+		}
+	};
+});
+//#endregion
+//#region src/bun.ts
+var bun_default = (rawOptions) => {
+	const options = defu(rawOptions, defaultOptions);
+	const include = options.include;
+	if (!(include instanceof RegExp)) throw new Error(`Unsupported 'include' options in Bun plugin. Please provide a single regular expression`);
+	if (options.exclude) throw new Error(`Unsupported 'exclude' option in Bun plugin`);
+	const rawPlugin = unpluginFactory(rawOptions, { framework: "bun" });
+	return {
+		name: "unplugin-typegpu",
+		setup(build) {
+			build.onLoad({ filter: include }, async (args) => {
+				var _result$code;
+				const codeIn = await Bun.file(args.path).text();
+				if (earlyPruneRegex.every((pattern) => !pattern.test(codeIn))) return {
+					contents: codeIn,
+					loader: args.loader
+				};
+				const result = rawPlugin.transform.handler.apply({}, [codeIn, args.path]);
+				return {
+					contents: (_result$code = result === null || result === void 0 ? void 0 : result.code) !== null && _result$code !== void 0 ? _result$code : codeIn,
+					loader: args.loader
+				};
+			});
+		}
+	};
+};
+//#endregion
+export { unpluginFactory as n, bun_default as t };

package/bun.cjs

@@ -1,40 +1,6 @@
-Object.defineProperties(exports, { __esModule: { value: true }, [Symbol.toStringTag]: { value: 'Module' } });
-const require_runtime = require('./_virtual/_rolldown/runtime.cjs');
-const require_common = require('./common.cjs');
-const require_babel = require('./babel.cjs');
-let defu = require("defu");
-defu = require_runtime.__toESM(defu);
-let _babel_standalone = require("@babel/standalone");
-_babel_standalone = require_runtime.__toESM(_babel_standalone);
-
-//#region src/bun.ts
-var bun_default = (rawOptions) => {
-	const options = (0, defu.default)(rawOptions, require_common.defaultOptions);
-	const include = options.include;
-	if (!(include instanceof RegExp)) throw new Error(`Unsupported 'include' options in Bun plugin. Please provide a single regular expression`);
-	if (options.exclude) throw new Error(`Unsupported 'exclude' option in Bun plugin`);
-	return {
-		name: "unplugin-typegpu",
-		setup(build) {
-			build.onLoad({ filter: include }, async (args) => {
-				const text = await Bun.file(args.path).text();
-				if (require_common.earlyPruneRegex.every((pattern) => !pattern.test(text))) return {
-					contents: text,
-					loader: args.loader
-				};
-				const result = _babel_standalone.transform(text, {
-					presets: [["typescript", { allowDeclareFields: true }]],
-					filename: args.path,
-					plugins: [require_babel.default]
-				}).code;
-				return {
-					contents: result !== null && result !== void 0 ? result : text,
-					loader: args.loader
-				};
-			});
-		}
-	};
-};
-
-//#endregion
-exports.default = bun_default;
+Object.defineProperties(exports, {
+	__esModule: { value: true },
+	[Symbol.toStringTag]: { value: "Module" }
+});
+const require_bun = require("./bun-CUGitP44.cjs");
+exports.default = require_bun.bun_default;

package/bun.d.cts

@@ -1,5 +1,2 @@
-import { Options } from "./common.cjs";
-
-//#region src/bun.d.ts
-declare const _default: (rawOptions: Options) => BunPlugin;
+import { t as _default } from "./bun-ChNeklBe.cjs";
 export = _default;

package/bun.d.mts

@@ -1,6 +1,2 @@
-import { Options } from "./common.mjs";
-
-//#region src/bun.d.ts
-declare const _default: (rawOptions: Options) => BunPlugin;
-//#endregion
+import { t as _default } from "./bun-9JrHWE4u.mjs";
 export { _default as default };

package/bun.mjs

@@ -1,36 +1,2 @@
-import { defaultOptions, earlyPruneRegex } from "./common.mjs";
-import babel_default from "./babel.mjs";
-import defu from "defu";
-import * as Babel from "@babel/standalone";
-
-//#region src/bun.ts
-var bun_default = (rawOptions) => {
-	const options = defu(rawOptions, defaultOptions);
-	const include = options.include;
-	if (!(include instanceof RegExp)) throw new Error(`Unsupported 'include' options in Bun plugin. Please provide a single regular expression`);
-	if (options.exclude) throw new Error(`Unsupported 'exclude' option in Bun plugin`);
-	return {
-		name: "unplugin-typegpu",
-		setup(build) {
-			build.onLoad({ filter: include }, async (args) => {
-				const text = await Bun.file(args.path).text();
-				if (earlyPruneRegex.every((pattern) => !pattern.test(text))) return {
-					contents: text,
-					loader: args.loader
-				};
-				const result = Babel.transform(text, {
-					presets: [["typescript", { allowDeclareFields: true }]],
-					filename: args.path,
-					plugins: [babel_default]
-				}).code;
-				return {
-					contents: result !== null && result !== void 0 ? result : text,
-					loader: args.loader
-				};
-			});
-		}
-	};
-};
-
-//#endregion
-export { bun_default as default };
+import { t as bun_default } from "./bun-DFr_PiLh.mjs";
+export { bun_default as default };

package/common-BKqBivuE.d.cts

@@ -0,0 +1,84 @@
+import * as t from "@babel/types";
+import { NodePath } from "@babel/traverse";
+import { FilterPattern } from "unplugin";
+import { transpileFn } from "tinyest-for-wgsl";
+
+//#region src/core/common.d.ts
+interface Options {
+  /** @default [/\.m?[jt]sx?$/] */
+  include?: FilterPattern;
+  /** @default undefined */
+  exclude?: FilterPattern;
+  /** @default undefined */
+  enforce?: 'post' | 'pre' | undefined;
+  /** @default undefined */
+  forceTgpuAlias?: string | undefined;
+  /** @default true */
+  autoNamingEnabled?: boolean | undefined;
+  /**
+   * Skipping files that don't contain "typegpu", "tgpu" or "use gpu".
+   * In case this early pruning hinders transformation, you
+   * can disable it.
+   *
+   * @default true
+   */
+  earlyPruning?: boolean | undefined;
+}
+type MetadatableFunction = t.FunctionDeclaration | t.FunctionExpression | t.ArrowFunctionExpression;
+interface TransformMethods {
+  warn(message: string): void;
+  /**
+   * For function statements to have metadata assigned, they first have to be manually hoisted
+   * to the top of their scope, then replaced with a variable declaration.
+   *
+   * @example
+   * ```ts
+   * const value = mod(13, 10);
+   * function mod(a: number, b: number): number {
+   *   'use gpu';
+   *   return a % b;
+   * }
+   * ```
+   *
+   * Should be transformed to:
+   *
+   * ```ts
+   * const mod = (a: number, b: number): number => {
+   *   'use gpu';
+   *   return a % b;
+   * };
+   * const value = mod(13, 10);
+   * ```
+   *
+   * Note that named function expressions aren't hoisted, so they don't pose a problem.
+   * ```ts
+   * double(2); // Uncaught ReferenceError: double is not defined
+   * console.log(function double(a) { return a * 2; });
+   * ```
+   */
+  assignMetadata(this: PluginState, path: NodePath<MetadatableFunction>, name: string | undefined, ast: ReturnType<typeof transpileFn>): void;
+  wrapInAutoName(this: PluginState, path: NodePath<t.Expression>, name: string): void;
+  replaceWithAssignmentOverload(path: NodePath<t.AssignmentExpression>, runtimeFn: string): void;
+  replaceWithBinaryOverload(path: NodePath<t.BinaryExpression>, runtimeFn: string): void;
+}
+interface PluginState extends TransformMethods {
+  /**
+   * How the `tgpu` object is used in code. Since it can be aliased, we
+   * need to catch that and act accordingly.
+   */
+  tgpuAliases: Set<string>;
+  autoNamingEnabled: boolean;
+  /**
+   * Populated by Babel
+   */
+  filename?: string | undefined;
+  /**
+   * In Babel, options are assigned to the property `opts` on the plugin state.
+   * We use this pattern everywhere for consistency.
+   */
+  opts: Options;
+  inUseGpuScope: boolean;
+  alreadyTransformed: WeakSet<t.Node>;
+}
+//#endregion
+export { PluginState as n, Options as t };