unplugin-typegpu 0.10.1 → 0.11.0

package/rollup-impl.cjs

@@ -1,127 +0,0 @@
-const require_runtime = require('./_virtual/_rolldown/runtime.cjs');
-const require_common = require('./common.cjs');
-let defu = require("defu");
-defu = require_runtime.__toESM(defu);
-let magic_string_ast = require("magic-string-ast");
-let estree_walker = require("estree-walker");
-let tinyest_for_wgsl = require("tinyest-for-wgsl");
-let tinyest = require("tinyest");
-
-//#region src/rollup-impl.ts
-function containsUseGpuDirective(node) {
-	if (node.body.type === "BlockStatement") {
-		for (const statement of node.body.body) if (statement.type === "ExpressionStatement" && statement.directive === require_common.useGpuDirective) return true;
-	}
-	return false;
-}
-function removeUseGpuDirective(node) {
-	const cloned = structuredClone(node);
-	if (cloned.body.type === "BlockStatement") cloned.body.body = cloned.body.body.filter((statement) => !(statement.type === "ExpressionStatement" && statement.directive === require_common.useGpuDirective));
-	return cloned;
-}
-function assignMetadata(magicString, node, metadata) {
-	magicString.prependLeft(node.start, "(($ => (globalThis.__TYPEGPU_META__ ??= new WeakMap()).set($.f = (").appendRight(node.end, `), ${metadata}) && $.f)({}))`);
-}
-function wrapInAutoName(magicString, node, name) {
-	magicString.prependLeft(node.start, "((globalThis.__TYPEGPU_AUTONAME__ ?? (a => a))(").appendRight(node.end, `, "${name}"))`);
-}
-const rollUpImpl = (rawOptions) => {
-	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) {
-				const ctx = {
-					tgpuAliases: new Set(options.forceTgpuAlias ? [options.forceTgpuAlias] : []),
-					fileId: id,
-					autoNamingEnabled: options.autoNamingEnabled
-				};
-				let ast;
-				try {
-					ast = this.parse(code, {
-						lang: "ts",
-						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 tgslFunctionDefs = [];
-				const magicString = new magic_string_ast.MagicStringAST(code);
-				(0, estree_walker.walk)(ast, { enter(_node, _parent) {
-					const node = _node;
-					const parent = _parent;
-					require_common.performExpressionNaming(ctx, node, (node, name) => {
-						wrapInAutoName(magicString, node, name);
-					});
-					if (node.type === "ImportDeclaration") require_common.gatherTgpuAliases(node, ctx);
-					if (node.type === "CallExpression") {
-						if (require_common.isShellImplementationCall(node, ctx)) {
-							const implementation = node.arguments[0];
-							if (implementation && (implementation.type === "FunctionExpression" || implementation.type === "ArrowFunctionExpression")) {
-								tgslFunctionDefs.push({ def: removeUseGpuDirective(implementation) });
-								this.skip();
-							}
-						}
-					}
-					if (node.type === "ArrowFunctionExpression" || node.type === "FunctionExpression" || node.type === "FunctionDeclaration") {
-						if (containsUseGpuDirective(node)) {
-							tgslFunctionDefs.push({
-								def: removeUseGpuDirective(node),
-								name: require_common.getFunctionName(node, parent)
-							});
-							this.skip();
-						}
-					}
-				} });
-				for (const { def, name } of tgslFunctionDefs) {
-					const { params, body, externalNames } = (0, tinyest_for_wgsl.transpileFn)(def);
-					const isFunctionStatement = def.type === "FunctionDeclaration";
-					(0, estree_walker.walk)(def, { leave(_node) {
-						const node = _node;
-						if (node.type === "AssignmentExpression") {
-							const runtimeFn = require_common.operators[node.operator];
-							if (runtimeFn) {
-								const left = node.left;
-								const right = node.right;
-								const lhs = magicString.sliceNode(left);
-								const rhs = magicString.sliceNode(right);
-								magicString.overwriteNode(node, `${lhs} = ${runtimeFn}(${lhs}, ${rhs})`);
-							}
-						} else if (node.type === "BinaryExpression") {
-							const runtimeFn = require_common.operators[node.operator];
-							if (runtimeFn) {
-								const left = node.left;
-								const right = node.right;
-								const lhs = magicString.sliceNode(left);
-								const rhs = magicString.sliceNode(right);
-								magicString.overwriteNode(node, `${runtimeFn}(${lhs}, ${rhs})`);
-							}
-						}
-					} });
-					if (isFunctionStatement && name && code.slice(0, def.start).search(new RegExp(`(?<![\\w_.])${name}(?![\\w_])`)) !== -1) console.warn(`File ${id}: function "${name}" might have been referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);
-					assignMetadata(magicString, def, `{
-              v: ${tinyest.FORMAT_VERSION},
-              name: ${name ? `"${name}"` : "undefined"},
-              ast: ${require_common.embedJSON({
-						params,
-						body,
-						externalNames
-					})},
-              externals: () => ({${externalNames.map((e) => e === "this" ? "\"this\": this" : e).join(", ")}}),
-            }`);
-					if (isFunctionStatement && name) magicString.prependLeft(def.start, `const ${name} = `);
-				}
-				return (0, magic_string_ast.generateTransform)(magicString, id);
-			}
-		}
-	};
-};
-
-//#endregion
-exports.rollUpImpl = rollUpImpl;

package/rollup-impl.d.cts

@@ -1,38 +0,0 @@
-import { Options } from "./common.cjs";
-import * as unplugin from "unplugin";
-import { UnpluginBuildContext, UnpluginContext } from "unplugin";
-import * as magic_string_ast0 from "magic-string-ast";
-
-//#region src/rollup-impl.d.ts
-declare const rollUpImpl: (rawOptions: Options) => {
-  name: "unplugin-typegpu";
-  enforce: "post" | "pre" | undefined;
-  transform: {
-    filter: {
-      id: Omit<Options, "include" | "autoNamingEnabled" | "earlyPruning"> & Omit<{
-        include: RegExp;
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      }, "include" | "autoNamingEnabled" | "earlyPruning"> & {
-        include: string | RegExp | unplugin.StringOrRegExp[];
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      };
-      code: RegExp[];
-    } | {
-      id: Omit<Options, "include" | "autoNamingEnabled" | "earlyPruning"> & Omit<{
-        include: RegExp;
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      }, "include" | "autoNamingEnabled" | "earlyPruning"> & {
-        include: string | RegExp | unplugin.StringOrRegExp[];
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      };
-      code?: never;
-    };
-    handler(this: UnpluginBuildContext & UnpluginContext, code: string, id: string): magic_string_ast0.CodeTransform | undefined;
-  };
-};
-//#endregion
-export { rollUpImpl };

package/rollup-impl.d.mts

@@ -1,39 +0,0 @@
-import { Options } from "./common.mjs";
-import * as unplugin from "unplugin";
-import { UnpluginBuildContext, UnpluginContext } from "unplugin";
-import * as magic_string_ast0 from "magic-string-ast";
-import { MagicStringAST } from "magic-string-ast";
-
-//#region src/rollup-impl.d.ts
-declare const rollUpImpl: (rawOptions: Options) => {
-  name: "unplugin-typegpu";
-  enforce: "post" | "pre" | undefined;
-  transform: {
-    filter: {
-      id: Omit<Options, "include" | "autoNamingEnabled" | "earlyPruning"> & Omit<{
-        include: RegExp;
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      }, "include" | "autoNamingEnabled" | "earlyPruning"> & {
-        include: string | RegExp | unplugin.StringOrRegExp[];
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      };
-      code: RegExp[];
-    } | {
-      id: Omit<Options, "include" | "autoNamingEnabled" | "earlyPruning"> & Omit<{
-        include: RegExp;
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      }, "include" | "autoNamingEnabled" | "earlyPruning"> & {
-        include: string | RegExp | unplugin.StringOrRegExp[];
-        autoNamingEnabled: boolean;
-        earlyPruning: boolean;
-      };
-      code?: never;
-    };
-    handler(this: UnpluginBuildContext & UnpluginContext, code: string, id: string): magic_string_ast0.CodeTransform | undefined;
-  };
-};
-//#endregion
-export { rollUpImpl };

package/rollup-impl.mjs

@@ -1,125 +0,0 @@
-import { defaultOptions, earlyPruneRegex, embedJSON, gatherTgpuAliases, getFunctionName, isShellImplementationCall, operators, performExpressionNaming, useGpuDirective } from "./common.mjs";
-import defu from "defu";
-import { MagicStringAST, generateTransform } from "magic-string-ast";
-import { walk } from "estree-walker";
-import { transpileFn } from "tinyest-for-wgsl";
-import { FORMAT_VERSION } from "tinyest";
-
-//#region src/rollup-impl.ts
-function containsUseGpuDirective(node) {
-	if (node.body.type === "BlockStatement") {
-		for (const statement of node.body.body) if (statement.type === "ExpressionStatement" && statement.directive === useGpuDirective) return true;
-	}
-	return false;
-}
-function removeUseGpuDirective(node) {
-	const cloned = structuredClone(node);
-	if (cloned.body.type === "BlockStatement") cloned.body.body = cloned.body.body.filter((statement) => !(statement.type === "ExpressionStatement" && statement.directive === useGpuDirective));
-	return cloned;
-}
-function assignMetadata(magicString, node, metadata) {
-	magicString.prependLeft(node.start, "(($ => (globalThis.__TYPEGPU_META__ ??= new WeakMap()).set($.f = (").appendRight(node.end, `), ${metadata}) && $.f)({}))`);
-}
-function wrapInAutoName(magicString, node, name) {
-	magicString.prependLeft(node.start, "((globalThis.__TYPEGPU_AUTONAME__ ?? (a => a))(").appendRight(node.end, `, "${name}"))`);
-}
-const rollUpImpl = (rawOptions) => {
-	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) {
-				const ctx = {
-					tgpuAliases: new Set(options.forceTgpuAlias ? [options.forceTgpuAlias] : []),
-					fileId: id,
-					autoNamingEnabled: options.autoNamingEnabled
-				};
-				let ast;
-				try {
-					ast = this.parse(code, {
-						lang: "ts",
-						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 tgslFunctionDefs = [];
-				const magicString = new MagicStringAST(code);
-				walk(ast, { enter(_node, _parent) {
-					const node = _node;
-					const parent = _parent;
-					performExpressionNaming(ctx, node, (node, name) => {
-						wrapInAutoName(magicString, node, name);
-					});
-					if (node.type === "ImportDeclaration") gatherTgpuAliases(node, ctx);
-					if (node.type === "CallExpression") {
-						if (isShellImplementationCall(node, ctx)) {
-							const implementation = node.arguments[0];
-							if (implementation && (implementation.type === "FunctionExpression" || implementation.type === "ArrowFunctionExpression")) {
-								tgslFunctionDefs.push({ def: removeUseGpuDirective(implementation) });
-								this.skip();
-							}
-						}
-					}
-					if (node.type === "ArrowFunctionExpression" || node.type === "FunctionExpression" || node.type === "FunctionDeclaration") {
-						if (containsUseGpuDirective(node)) {
-							tgslFunctionDefs.push({
-								def: removeUseGpuDirective(node),
-								name: getFunctionName(node, parent)
-							});
-							this.skip();
-						}
-					}
-				} });
-				for (const { def, name } of tgslFunctionDefs) {
-					const { params, body, externalNames } = transpileFn(def);
-					const isFunctionStatement = def.type === "FunctionDeclaration";
-					walk(def, { leave(_node) {
-						const node = _node;
-						if (node.type === "AssignmentExpression") {
-							const runtimeFn = operators[node.operator];
-							if (runtimeFn) {
-								const left = node.left;
-								const right = node.right;
-								const lhs = magicString.sliceNode(left);
-								const rhs = magicString.sliceNode(right);
-								magicString.overwriteNode(node, `${lhs} = ${runtimeFn}(${lhs}, ${rhs})`);
-							}
-						} else if (node.type === "BinaryExpression") {
-							const runtimeFn = operators[node.operator];
-							if (runtimeFn) {
-								const left = node.left;
-								const right = node.right;
-								const lhs = magicString.sliceNode(left);
-								const rhs = magicString.sliceNode(right);
-								magicString.overwriteNode(node, `${runtimeFn}(${lhs}, ${rhs})`);
-							}
-						}
-					} });
-					if (isFunctionStatement && name && code.slice(0, def.start).search(new RegExp(`(?<![\\w_.])${name}(?![\\w_])`)) !== -1) console.warn(`File ${id}: function "${name}" might have been referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);
-					assignMetadata(magicString, def, `{
-              v: ${FORMAT_VERSION},
-              name: ${name ? `"${name}"` : "undefined"},
-              ast: ${embedJSON({
-						params,
-						body,
-						externalNames
-					})},
-              externals: () => ({${externalNames.map((e) => e === "this" ? "\"this\": this" : e).join(", ")}}),
-            }`);
-					if (isFunctionStatement && name) magicString.prependLeft(def.start, `const ${name} = `);
-				}
-				return generateTransform(magicString, id);
-			}
-		}
-	};
-};
-
-//#endregion
-export { rollUpImpl };