unplugin-typegpu 0.10.2 → 0.11.1

package/README.md

@@ -2,7 +2,6 @@
 
 # unplugin-typegpu
 
-🚧 **Under Construction** 🚧 -
 [GitHub](https://github.com/software-mansion/TypeGPU/tree/main/packages/unplugin-typegpu)
 
 Read more about the plugin in the
@@ -10,10 +9,10 @@ Read more about the plugin in the
 
 </div>
 
-Build plugins for [TypeGPU](https://typegpu.com) that enable:
+A set of bundler plugins that enhance [TypeGPU](https://typegpu.com) with:
 
-- Seamless JavaScript -> WGSL transpilation
-- [🚧 TODO] Improved debugging with automatic naming of resources
+- JavaScript/TypeScript shader support ('use gpu' directive)
+- Improved debugging with automatic naming of resources
 
 ## Getting Started
 
@@ -48,6 +47,17 @@ export default defineConfig({
 });
 ```
 
+- bun
+
+```ts
+// preload.ts
+
+import { plugin } from 'bun';
+import typegpu from 'unplugin-typegpu/bun';
+
+void plugin(typegpu());
+```
+
 ## TypeGPU is created by Software Mansion
 
 [![swm](https://logo.swmansion.com/logo?color=white&variant=desktop&width=150&tag=typegpu-github 'Software Mansion')](https://swmansion.com)

package/babel-C8UzthFd.d.cts

@@ -0,0 +1,13 @@
+import { n as PluginState, r as t } from "./common-DZ2cLtbe.cjs";
+import { NodePath } from "@babel/traverse";
+
+//#region src/babel.d.ts
+declare function TypeGPUPlugin(): {
+  name: string;
+  pre(this: PluginState): void;
+  visitor: {
+    Program(this: PluginState, path: NodePath<t.Program>, state: PluginState): void;
+  };
+};
+//#endregion
+export { TypeGPUPlugin as t };

package/babel-CbManzjz.js

@@ -0,0 +1,128 @@
+import { a as initPluginState, i as getBlockScope, o as t, r as functionVisitor, t as defaultOptions } from "./common-CElmMmuq.js";
+import defu from "defu";
+import { transpileFn } from "tinyest-for-wgsl";
+import { FORMAT_VERSION } from "tinyest";
+import { resolve } from "pathe";
+import picomatch from "picomatch";
+
+//#region src/core/filter.ts
+function toArray(array) {
+	const result = array || [];
+	if (Array.isArray(result)) return result;
+	return [result];
+}
+const BACKSLASH_REGEX = /\\/g;
+function normalize(path) {
+	return path.replace(BACKSLASH_REGEX, "/");
+}
+const ABSOLUTE_PATH_REGEX = /^(?:\/|(?:[A-Z]:)?[/\\|])/i;
+function isAbsolute(path) {
+	return ABSOLUTE_PATH_REGEX.test(path);
+}
+function getMatcherString(glob, cwd) {
+	if (glob.startsWith("**") || isAbsolute(glob)) return normalize(glob);
+	return normalize(resolve(cwd, glob));
+}
+function patternToIdFilter(pattern) {
+	if (pattern instanceof RegExp) return (id) => {
+		const normalizedId = normalize(id);
+		const result = pattern.test(normalizedId);
+		pattern.lastIndex = 0;
+		return result;
+	};
+	const matcher = picomatch(getMatcherString(pattern, process.cwd()), { dot: true });
+	return (id) => {
+		return matcher(normalize(id));
+	};
+}
+function createFilter(exclude, include) {
+	if (!exclude && !include) return;
+	return (input) => {
+		if (exclude === null || exclude === void 0 ? void 0 : exclude.some((filter) => filter(input))) return false;
+		if (include === null || include === void 0 ? void 0 : include.some((filter) => filter(input))) return true;
+		return !(include && include.length > 0);
+	};
+}
+function normalizeFilter(filter) {
+	if (typeof filter === "string" || filter instanceof RegExp) return { include: [filter] };
+	if (Array.isArray(filter)) return { include: filter };
+	return {
+		exclude: filter.exclude ? toArray(filter.exclude) : void 0,
+		include: filter.include ? toArray(filter.include) : void 0
+	};
+}
+function createIdFilter(filter) {
+	if (!filter) return;
+	const { exclude, include } = normalizeFilter(filter);
+	return createFilter(exclude === null || exclude === void 0 ? void 0 : exclude.map(patternToIdFilter), include === null || include === void 0 ? void 0 : include.map(patternToIdFilter));
+}
+function createFilterForId(filter) {
+	const filterFunction = createIdFilter(filter);
+	return filterFunction ? (id) => !!filterFunction(id) : void 0;
+}
+
+//#endregion
+//#region src/babel.ts
+function i(identifier) {
+	return t.identifier(identifier);
+}
+function assignMetadata(path, name, ast) {
+	const metadata = t.objectExpression([
+		t.objectProperty(i("v"), t.numericLiteral(FORMAT_VERSION)),
+		t.objectProperty(i("name"), t.valueToNode(name)),
+		t.objectProperty(i("ast"), t.valueToNode(ast)),
+		t.objectProperty(i("externals"), t.arrowFunctionExpression([], t.blockStatement([t.returnStatement(t.objectExpression(ast.externalNames.map((name$1) => name$1 === "this" ? t.objectProperty(i("this"), t.thisExpression()) : t.objectProperty(i(name$1), i(name$1), false, name$1 !== "this"))))])))
+	]);
+	let expression;
+	const visibility = t.isFunctionDeclaration(path.node) ? getBlockScope(path) : void 0;
+	if (t.isFunctionDeclaration(path.node)) expression = t.functionExpression(path.node.id, path.node.params, path.node.body);
+	else expression = path.node;
+	const callExpr = t.callExpression(t.arrowFunctionExpression([i("$")], t.logicalExpression("&&", t.callExpression(t.memberExpression(t.assignmentExpression("??=", t.memberExpression(i("globalThis"), i("__TYPEGPU_META__")), t.newExpression(i("WeakMap"), [])), i("set")), [t.assignmentExpression("=", t.memberExpression(i("$"), i("f")), expression), metadata]), t.memberExpression(i("$"), i("f")))), [t.objectExpression([])]);
+	t.addComment(callExpr, "leading", "#__PURE__");
+	let replacement = callExpr;
+	if (t.isFunctionDeclaration(path.node) && path.node.id) {
+		const declaration = t.variableDeclaration("const", [t.variableDeclarator(path.node.id, callExpr)]);
+		t.inheritLeadingComments(declaration, path.node);
+		replacement = declaration;
+	}
+	if (visibility) {
+		visibility.unshiftContainer("body", replacement);
+		this.alreadyTransformed.add(expression);
+		path.remove();
+	} else path.replaceWith(replacement);
+	path.skip();
+}
+function wrapInAutoName(path, name) {
+	const callExpr = t.callExpression(t.logicalExpression("??", t.memberExpression(i("globalThis"), i("__TYPEGPU_AUTONAME__")), t.arrowFunctionExpression([i("a")], i("a"))), [path.node, t.stringLiteral(name)]);
+	t.addComment(callExpr, "leading", "#__PURE__");
+	path.replaceWith(callExpr);
+}
+function replaceWithAssignmentOverload(path, runtimeFn) {
+	path.replaceWith(t.assignmentExpression("=", path.node.left, t.callExpression(i(runtimeFn), [path.node.left, path.node.right])));
+}
+function replaceWithBinaryOverload(path, runtimeFn) {
+	path.replaceWith(t.callExpression(i(runtimeFn), [path.node.left, path.node.right]));
+}
+function TypeGPUPlugin() {
+	return {
+		name: "typegpu",
+		pre() {
+			this.opts = defu(this.opts, defaultOptions);
+			initPluginState(this, {
+				warn: (message) => console.warn(message),
+				assignMetadata,
+				wrapInAutoName,
+				replaceWithAssignmentOverload,
+				replaceWithBinaryOverload
+			});
+		},
+		visitor: { Program(path, state) {
+			const filter = createFilterForId(state.opts);
+			if (state.filename && filter && !(filter === null || filter === void 0 ? void 0 : filter(state.filename))) return;
+			path.traverse(functionVisitor, state);
+		} }
+	};
+}
+
+//#endregion
+export { TypeGPUPlugin as t };

package/babel-CmCnq-43.d.ts

@@ -0,0 +1,13 @@
+import { n as PluginState, r as t } from "./common-BlvLUwEz.js";
+import { NodePath } from "@babel/traverse";
+
+//#region src/babel.d.ts
+declare function TypeGPUPlugin(): {
+  name: string;
+  pre(this: PluginState): void;
+  visitor: {
+    Program(this: PluginState, path: NodePath<t.Program>, state: PluginState): void;
+  };
+};
+//#endregion
+export { TypeGPUPlugin as t };

package/babel-Dis-Pwfu.cjs

@@ -0,0 +1,140 @@
+const require_common = require('./common-BoeQmwpt.cjs');
+let defu = require("defu");
+defu = require_common.__toESM(defu);
+let tinyest_for_wgsl = require("tinyest-for-wgsl");
+tinyest_for_wgsl = require_common.__toESM(tinyest_for_wgsl);
+let tinyest = require("tinyest");
+tinyest = require_common.__toESM(tinyest);
+let pathe = require("pathe");
+pathe = require_common.__toESM(pathe);
+let picomatch = require("picomatch");
+picomatch = require_common.__toESM(picomatch);
+let __babel_types = require("@babel/types");
+__babel_types = require_common.__toESM(__babel_types);
+
+//#region src/core/filter.ts
+function toArray(array) {
+	const result = array || [];
+	if (Array.isArray(result)) return result;
+	return [result];
+}
+const BACKSLASH_REGEX = /\\/g;
+function normalize(path) {
+	return path.replace(BACKSLASH_REGEX, "/");
+}
+const ABSOLUTE_PATH_REGEX = /^(?:\/|(?:[A-Z]:)?[/\\|])/i;
+function isAbsolute(path) {
+	return ABSOLUTE_PATH_REGEX.test(path);
+}
+function getMatcherString(glob, cwd) {
+	if (glob.startsWith("**") || isAbsolute(glob)) return normalize(glob);
+	return normalize((0, pathe.resolve)(cwd, glob));
+}
+function patternToIdFilter(pattern) {
+	if (pattern instanceof RegExp) return (id) => {
+		const normalizedId = normalize(id);
+		const result = pattern.test(normalizedId);
+		pattern.lastIndex = 0;
+		return result;
+	};
+	const matcher = (0, picomatch.default)(getMatcherString(pattern, process.cwd()), { dot: true });
+	return (id) => {
+		return matcher(normalize(id));
+	};
+}
+function createFilter(exclude, include) {
+	if (!exclude && !include) return;
+	return (input) => {
+		if (exclude === null || exclude === void 0 ? void 0 : exclude.some((filter) => filter(input))) return false;
+		if (include === null || include === void 0 ? void 0 : include.some((filter) => filter(input))) return true;
+		return !(include && include.length > 0);
+	};
+}
+function normalizeFilter(filter) {
+	if (typeof filter === "string" || filter instanceof RegExp) return { include: [filter] };
+	if (Array.isArray(filter)) return { include: filter };
+	return {
+		exclude: filter.exclude ? toArray(filter.exclude) : void 0,
+		include: filter.include ? toArray(filter.include) : void 0
+	};
+}
+function createIdFilter(filter) {
+	if (!filter) return;
+	const { exclude, include } = normalizeFilter(filter);
+	return createFilter(exclude === null || exclude === void 0 ? void 0 : exclude.map(patternToIdFilter), include === null || include === void 0 ? void 0 : include.map(patternToIdFilter));
+}
+function createFilterForId(filter) {
+	const filterFunction = createIdFilter(filter);
+	return filterFunction ? (id) => !!filterFunction(id) : void 0;
+}
+
+//#endregion
+//#region src/babel.ts
+function i(identifier) {
+	return __babel_types.identifier(identifier);
+}
+function assignMetadata(path, name, ast) {
+	const metadata = __babel_types.objectExpression([
+		__babel_types.objectProperty(i("v"), __babel_types.numericLiteral(tinyest.FORMAT_VERSION)),
+		__babel_types.objectProperty(i("name"), __babel_types.valueToNode(name)),
+		__babel_types.objectProperty(i("ast"), __babel_types.valueToNode(ast)),
+		__babel_types.objectProperty(i("externals"), __babel_types.arrowFunctionExpression([], __babel_types.blockStatement([__babel_types.returnStatement(__babel_types.objectExpression(ast.externalNames.map((name$1) => name$1 === "this" ? __babel_types.objectProperty(i("this"), __babel_types.thisExpression()) : __babel_types.objectProperty(i(name$1), i(name$1), false, name$1 !== "this"))))])))
+	]);
+	let expression;
+	const visibility = __babel_types.isFunctionDeclaration(path.node) ? require_common.getBlockScope(path) : void 0;
+	if (__babel_types.isFunctionDeclaration(path.node)) expression = __babel_types.functionExpression(path.node.id, path.node.params, path.node.body);
+	else expression = path.node;
+	const callExpr = __babel_types.callExpression(__babel_types.arrowFunctionExpression([i("$")], __babel_types.logicalExpression("&&", __babel_types.callExpression(__babel_types.memberExpression(__babel_types.assignmentExpression("??=", __babel_types.memberExpression(i("globalThis"), i("__TYPEGPU_META__")), __babel_types.newExpression(i("WeakMap"), [])), i("set")), [__babel_types.assignmentExpression("=", __babel_types.memberExpression(i("$"), i("f")), expression), metadata]), __babel_types.memberExpression(i("$"), i("f")))), [__babel_types.objectExpression([])]);
+	__babel_types.addComment(callExpr, "leading", "#__PURE__");
+	let replacement = callExpr;
+	if (__babel_types.isFunctionDeclaration(path.node) && path.node.id) {
+		const declaration = __babel_types.variableDeclaration("const", [__babel_types.variableDeclarator(path.node.id, callExpr)]);
+		__babel_types.inheritLeadingComments(declaration, path.node);
+		replacement = declaration;
+	}
+	if (visibility) {
+		visibility.unshiftContainer("body", replacement);
+		this.alreadyTransformed.add(expression);
+		path.remove();
+	} else path.replaceWith(replacement);
+	path.skip();
+}
+function wrapInAutoName(path, name) {
+	const callExpr = __babel_types.callExpression(__babel_types.logicalExpression("??", __babel_types.memberExpression(i("globalThis"), i("__TYPEGPU_AUTONAME__")), __babel_types.arrowFunctionExpression([i("a")], i("a"))), [path.node, __babel_types.stringLiteral(name)]);
+	__babel_types.addComment(callExpr, "leading", "#__PURE__");
+	path.replaceWith(callExpr);
+}
+function replaceWithAssignmentOverload(path, runtimeFn) {
+	path.replaceWith(__babel_types.assignmentExpression("=", path.node.left, __babel_types.callExpression(i(runtimeFn), [path.node.left, path.node.right])));
+}
+function replaceWithBinaryOverload(path, runtimeFn) {
+	path.replaceWith(__babel_types.callExpression(i(runtimeFn), [path.node.left, path.node.right]));
+}
+function TypeGPUPlugin() {
+	return {
+		name: "typegpu",
+		pre() {
+			this.opts = (0, defu.default)(this.opts, require_common.defaultOptions);
+			require_common.initPluginState(this, {
+				warn: (message) => console.warn(message),
+				assignMetadata,
+				wrapInAutoName,
+				replaceWithAssignmentOverload,
+				replaceWithBinaryOverload
+			});
+		},
+		visitor: { Program(path, state) {
+			const filter = createFilterForId(state.opts);
+			if (state.filename && filter && !(filter === null || filter === void 0 ? void 0 : filter(state.filename))) return;
+			path.traverse(require_common.functionVisitor, state);
+		} }
+	};
+}
+
+//#endregion
+Object.defineProperty(exports, 'TypeGPUPlugin', {
+  enumerable: true,
+  get: function () {
+    return TypeGPUPlugin;
+  }
+});

package/babel.cjs

@@ -1,161 +1,5 @@
-Object.defineProperties(exports, { __esModule: { value: true }, [Symbol.toStringTag]: { value: 'Module' } });
-const require_filter = require('./filter-CKmD4HRk.cjs');
-let defu = require("defu");
-defu = require_filter.__toESM(defu);
-let tinyest_for_wgsl = require("tinyest-for-wgsl");
-let tinyest = require("tinyest");
-let _babel_standalone = require("@babel/standalone");
-_babel_standalone = require_filter.__toESM(_babel_standalone);
+Object.defineProperty(exports, '__esModule', { value: true });
+require('./common-BoeQmwpt.cjs');
+const require_babel = require('./babel-Dis-Pwfu.cjs');
 
-//#region src/babel.ts
-const template = _babel_standalone.packages.template;
-const types = _babel_standalone.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(require_filter.useGpuDirective);
-}
-function i(identifier) {
-	return types.identifier(identifier);
-}
-const fnNodeToOriginalMap = /* @__PURE__ */ new WeakMap();
-function functionToTranspiled(node, parent) {
-	var _fnNodeToOriginalMap$;
-	const { params, body, externalNames } = (0, tinyest_for_wgsl.transpileFn)((_fnNodeToOriginalMap$ = fnNodeToOriginalMap.get(node)) !== null && _fnNodeToOriginalMap$ !== void 0 ? _fnNodeToOriginalMap$ : node);
-	const maybeName = require_filter.getFunctionName(node, parent);
-	const metadata = types.objectExpression([
-		types.objectProperty(i("v"), types.numericLiteral(tinyest.FORMAT_VERSION)),
-		types.objectProperty(i("name"), maybeName ? types.stringLiteral(maybeName) : types.buildUndefinedNode()),
-		types.objectProperty(i("ast"), template.expression`${require_filter.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) {
-			require_filter.performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("init").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		AssignmentExpression(path) {
-			if (inUseGpuScope) {
-				const runtimeFn = require_filter.operators[path.node.operator];
-				if (runtimeFn) path.replaceWith(types.assignmentExpression("=", path.node.left, types.callExpression(types.identifier(runtimeFn), [path.node.left, path.node.right])));
-			}
-			require_filter.performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("right").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		ObjectProperty(path) {
-			require_filter.performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("value").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		ClassProperty(path) {
-			require_filter.performExpressionNaming(ctx, path.node, (node, name) => {
-				path.get("value").replaceWith(wrapInAutoName(node, name));
-			});
-		},
-		ImportDeclaration(path) {
-			require_filter.gatherTgpuAliases(path.node, ctx);
-		},
-		BinaryExpression: { exit(path) {
-			if (!inUseGpuScope) return;
-			const runtimeFn = require_filter.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 (require_filter.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 = (0, defu.default)(state.opts, require_filter.defaultOptions);
-		const id = state.filename;
-		const filter = require_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
-exports.default = babel_default;
+exports.default = require_babel.TypeGPUPlugin;

package/babel.d.cts

@@ -1,10 +1,3 @@
-import { n as Program, t as Node } from "./index-legacy-xKi5C2ga.cjs";
-import * as _babel_traverse0 from "@babel/traverse";
-
-//#region src/babel.d.ts
-declare function export_default(): {
-  visitor: {
-    Program(this: Node, path: _babel_traverse0.NodePath<Program>, state: Node): void;
-  };
-};
-export = export_default;
+import "./common-DZ2cLtbe.cjs";
+import { t as TypeGPUPlugin } from "./babel-C8UzthFd.cjs";
+export = TypeGPUPlugin;

package/babel.d.ts

@@ -0,0 +1,3 @@
+import "./common-BlvLUwEz.js";
+import { t as TypeGPUPlugin } from "./babel-CmCnq-43.js";
+export { TypeGPUPlugin as default };

package/babel.js

@@ -0,0 +1,4 @@
+import "./common-CElmMmuq.js";
+import { t as TypeGPUPlugin } from "./babel-CbManzjz.js";
+
+export { TypeGPUPlugin as default };

package/bun-CDECNeIU.js

@@ -0,0 +1,134 @@
+import { a as initPluginState, i as getBlockScope, n as earlyPruneRegex, o as t, r as functionVisitor, t as defaultOptions } from "./common-CElmMmuq.js";
+import defu from "defu";
+import { transpileFn } from "tinyest-for-wgsl";
+import { FORMAT_VERSION } from "tinyest";
+import { MagicStringAST, generateTransform } from "magic-string-ast";
+import { getBabelParserOptions, getLang } from "ast-kit";
+import _traverse from "@babel/traverse";
+import * as parser from "@babel/parser";
+
+//#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 };