unplugin-typegpu 0.1.0 → 0.1.2

package/LICENSE.md

@@ -2,20 +2,19 @@ MIT License
 
 Copyright (c) 2025 Software Mansion <swmansion.com>
 
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+the Software, and to permit persons to whom the Software is furnished to do so,
+subject to the following conditions:
 
 The above copyright notice and this permission notice shall be included in all
 copies or substantial portions of the Software.
 
 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

package/README.md

@@ -2,16 +2,27 @@
 
 # unplugin-typegpu
 
-🚧 **Under Construction** 🚧 - [GitHub](https://github.com/software-mansion/TypeGPU/tree/main/packages/unplugin-typegpu)
+🚧 **Under Construction** 🚧 -
+[GitHub](https://github.com/software-mansion/TypeGPU/tree/main/packages/unplugin-typegpu)
+
+Read more about the plugin in the
+[TypeGPU documentation](https://docs.swmansion.com/TypeGPU/tooling/unplugin-typegpu/).
 
 </div>
 
 Build plugins for [TypeGPU](https://typegpu.com) that enable:
+
 - Seamless JavaScript -> WGSL transpilation
 - [🚧 TODO] Improved debugging with automatic naming of resources
 
-Currently supported:
-* babel
+## Getting Started
+
+```sh
+npm install unplugin-typegpu
+```
+
+- babel
+
 ```js
 // babel.config.js (React Native with Expo)
 
@@ -19,34 +30,30 @@ module.exports = function (api) {
   api.cache(true);
   return {
     presets: ['babel-preset-expo'],
-    plugins: ['unplugin-typegpu/babel']
+    plugins: ['unplugin-typegpu/babel'],
   };
 };
-
 ```
 
-* rollup
+- vite
 
 ```ts
 // vite.config.js
 
 import { defineConfig } from 'vite';
-import typegpu from 'unplugin-typegpu/rollup';
+import typegpu from 'unplugin-typegpu/vite';
 
 export default defineConfig({
   plugins: [typegpu()],
 });
-
-```
-
-## Getting Started
-
-```sh
-npm install unplugin-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)
 
-Since 2012 [Software Mansion](https://swmansion.com) is a software agency with experience in building web and mobile apps. We are Core React Native Contributors and experts in dealing with all kinds of React Native issues. We can help you build your next dream product – [Hire us](https://swmansion.com/contact/projects?utm_source=typegpu&utm_medium=readme).
+Since 2012 [Software Mansion](https://swmansion.com) is a software agency with
+experience in building web and mobile apps. We are Core React Native
+Contributors and experts in dealing with all kinds of React Native issues. We
+can help you build your next dream product –
+[Hire us](https://swmansion.com/contact/projects?utm_source=typegpu&utm_medium=readme).

package/babel.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHQFSG5ZMcjs = require('./chunk-HQFSG5ZM.cjs');exports.default = _chunkHQFSG5ZMcjs.h;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkBCOJLF4Tcjs = require('./chunk-BCOJLF4T.cjs');exports.default = _chunkBCOJLF4Tcjs.h;
 //# sourceMappingURL=babel.cjs.map

package/babel.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/babel.cjs"],"names":[],"mappings":"AAAA,iIAAwC,sCAAqB","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/babel.cjs"}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/babel.cjs"],"names":[],"mappings":"AAAA,iIAAwC,sCAAqB","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/babel.cjs"}

package/babel.js

@@ -1,2 +1,2 @@
-import{h as a}from"./chunk-Y4WARDXR.js";export{a as default};
+import{h as a}from"./chunk-4ONRHQSA.js";export{a as default};
 //# sourceMappingURL=babel.js.map

package/{chunk-Y4WARDXR.js → chunk-4ONRHQSA.js}

@@ -1,2 +1,2 @@
-var D=Object.defineProperty;var b=Object.getOwnPropertySymbols;var N=Object.prototype.hasOwnProperty,O=Object.prototype.propertyIsEnumerable;var x=(e,n,t)=>n in e?D(e,n,{enumerable:!0,configurable:!0,writable:!0,value:t}):e[n]=t,V=(e,n)=>{for(var t in n||(n={}))N.call(n,t)&&x(e,t,n[t]);if(b)for(var t of b(n))O.call(n,t)&&x(e,t,n[t]);return e};import*as g from"@babel/standalone";import{transpileFn as P}from"tinyest-for-wgsl";var L={include:[/\.m?[jt]sx?$/]};function d(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function F(e,n){let t="",r=n;for(;;)if(r.type==="MemberExpression"){if(r.property.type!=="Identifier")break;t=t?`${r.property.name}.${t}`:r.property.name,r=r.object}else if(r.type==="Identifier"){t=t?`${r.name}.${t}`:r.name;break}else break;return e.tgpuAliases.has(t)}var R=/import.*from\s*['"]typegpu.*['"]/,$=/import\s*\(\s*['"]\s*typegpu.*['"]/,w=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/,E=[R,$,w];function A(e,n){if(e.source.value==="typegpu")for(let t of e.specifiers)t.type==="ImportDefaultSpecifier"||t.type==="ImportSpecifier"&&t.imported.type==="Identifier"&&t.imported.name==="tgpu"?n.tgpuAliases.add(t.local.name):t.type==="ImportNamespaceSpecifier"&&n.tgpuAliases.add(`${t.local.name}.tgpu`)}var j=["fn","vertexFn","fragmentFn","computeFn"];function v(e,n){return e.callee.type==="CallExpression"&&e.callee.callee.type==="MemberExpression"&&e.callee.callee.property.type==="Identifier"&&j.includes(e.callee.callee.property.name)&&e.arguments.length===1&&(e.callee.callee.object.type==="MemberExpression"?F(n,e.callee.callee.object.object):F(n,e.callee.callee.object))||e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&e.callee.property.name==="does"}var I=["kernel","kernel & js"];import{resolve as h}from"node:path";import _ from"picomatch";function S(e){let n=e||[];return Array.isArray(n)?n:[n]}var z=/\\/g;function f(e){return e.replace(z,"/")}var B=/^(?:\/|(?:[A-Z]:)?[/\\|])/i;function J(e){return B.test(e)}function K(e,n){if(e.startsWith("**")||J(e))return f(e);let t=h(n,e);return f(t)}function k(e){if(e instanceof RegExp)return i=>{let o=f(i),l=e.test(o);return e.lastIndex=0,l};let n=process.cwd(),t=K(e,n),r=_(t,{dot:!0});return i=>{let o=f(i);return r(o)}}function M(e,n){if(!(!e&&!n))return t=>e!=null&&e.some(r=>r(t))?!1:n!=null&&n.some(r=>r(t))?!0:!(n&&n.length>0)}function W(e){return typeof e=="string"||e instanceof RegExp?{include:[e]}:Array.isArray(e)?{include:e}:{exclude:e.exclude?S(e.exclude):void 0,include:e.include?S(e.include):void 0}}function G(e){if(!e)return;let{exclude:n,include:t}=W(e),r=n==null?void 0:n.map(k),i=t==null?void 0:t.map(k);return M(r,i)}function T(e){let n=G(e);return n?t=>!!n(t):void 0}var c=g.packages.template,s=g.packages.types;function C(e){var t,r;let n=("directives"in e.body?(r=(t=e.body)==null?void 0:t.directives)!=null?r:[]:[]).map(i=>i.value.value);for(let i of I)if(n.includes(i))return i}function m(e,n,t){var p;let r=C(e);if(!r)return null;let{argNames:i,body:o,externalNames:l}=P(e),a=n.tgpuAliases.values().next().value;if(a===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(p=n.fileId)!=null?p:""}`);return s.callExpression(c.expression(`${a}.__assignAst`)(),[r==="kernel & js"?e:c.expression`${a}.__removedJsImpl(${t?`"${t}"`:""})`(),c.expression`${d({argNames:i,body:o,externalNames:l})}`(),s.objectExpression(l.map(u=>s.objectProperty(s.identifier(u),s.identifier(u))))])}function H(e){return{ImportDeclaration(n){A(n.node,e)},ArrowFunctionExpression(n){let t=m(n.node,e,n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionExpression(n){var r;let t=m(n.node,e,(r=n.node.id)!=null&&r.name?n.node.id.name:n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionDeclaration(n){var o;let t=n.node,r=s.functionExpression(t.id,t.params,t.body),i=m(r,e,(o=t.id)==null?void 0:o.name);i&&t.id&&(n.replaceWith(s.variableDeclaration("const",[s.variableDeclarator(t.id,i)])),n.skip())},CallExpression(n){var r;let t=n.node;if(v(t,e)){let i=t.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let{argNames:o,body:l,externalNames:a}=P(i),p=e.tgpuAliases.values().next().value;if(p===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(r=e.fileId)!=null?r:""}`);let u=C(i);n.replaceWith(s.callExpression(t.callee,[s.callExpression(c.expression(`${p}.__assignAst`)(),[u!=="kernel & js"?c.expression`${p}.__removedJsImpl()`():i,c.expression`${d({argNames:o,body:l,externalNames:a})}`(),s.objectExpression(a.map(y=>s.objectProperty(s.identifier(y),s.identifier(y))))])])),n.skip()}}}}}function te(){return{visitor:{Program(e,n){var a;let t=(a=n.file)==null?void 0:a.code,r=n.opts,i=n.filename,o=T(r);if(i&&o&&!(o!=null&&o(i))||!(r!=null&&r.forceTgpuAlias)&&t&&!E.some(p=>p.test(t)))return;let l={tgpuAliases:new Set(r!=null&&r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:i};e.traverse(H(l))}}}}export{V as a,L as b,d as c,E as d,A as e,v as f,I as g,te as h};
-//# sourceMappingURL=chunk-Y4WARDXR.js.map
+var D=Object.defineProperty;var b=Object.getOwnPropertySymbols;var O=Object.prototype.hasOwnProperty,R=Object.prototype.propertyIsEnumerable;var x=(e,n,t)=>n in e?D(e,n,{enumerable:!0,configurable:!0,writable:!0,value:t}):e[n]=t,V=(e,n)=>{for(var t in n||(n={}))O.call(n,t)&&x(e,t,n[t]);if(b)for(var t of b(n))R.call(n,t)&&x(e,t,n[t]);return e};import*as g from"@babel/standalone";import{transpileFn as P}from"tinyest-for-wgsl";var L={include:[/\.m?[jt]sx?$/]};function d(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function F(e,n){let t="",r=n;for(;;)if(r.type==="MemberExpression"){if(r.property.type!=="Identifier")break;t=t?`${r.property.name}.${t}`:r.property.name,r=r.object}else if(r.type==="Identifier"){t=t?`${r.name}.${t}`:r.name;break}else break;return e.tgpuAliases.has(t)}var $=/import.*from\s*['"]typegpu.*['"]/,w=/import\s*\(\s*['"]\s*typegpu.*['"]/,N=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/,E=[$,w,N];function A(e,n){if(e.source.value==="typegpu")for(let t of e.specifiers)t.type==="ImportDefaultSpecifier"||t.type==="ImportSpecifier"&&t.imported.type==="Identifier"&&t.imported.name==="tgpu"?n.tgpuAliases.add(t.local.name):t.type==="ImportNamespaceSpecifier"&&n.tgpuAliases.add(`${t.local.name}.tgpu`)}var j=["fn","vertexFn","fragmentFn","computeFn"];function v(e,n){return e.callee.type==="CallExpression"&&e.callee.callee.type==="MemberExpression"&&e.callee.callee.property.type==="Identifier"&&j.includes(e.callee.callee.property.name)&&e.arguments.length===1&&(e.callee.callee.object.type==="MemberExpression"?F(n,e.callee.callee.object.object):F(n,e.callee.callee.object))||e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&e.callee.property.name==="does"}var I=["kernel","kernel & js"];import{resolve as h}from"node:path";import _ from"picomatch";function S(e){let n=e||[];return Array.isArray(n)?n:[n]}var z=/\\/g;function f(e){return e.replace(z,"/")}var B=/^(?:\/|(?:[A-Z]:)?[/\\|])/i;function J(e){return B.test(e)}function K(e,n){if(e.startsWith("**")||J(e))return f(e);let t=h(n,e);return f(t)}function k(e){if(e instanceof RegExp)return i=>{let o=f(i),l=e.test(o);return e.lastIndex=0,l};let n=process.cwd(),t=K(e,n),r=_(t,{dot:!0});return i=>{let o=f(i);return r(o)}}function M(e,n){if(!(!e&&!n))return t=>e!=null&&e.some(r=>r(t))?!1:n!=null&&n.some(r=>r(t))?!0:!(n&&n.length>0)}function W(e){return typeof e=="string"||e instanceof RegExp?{include:[e]}:Array.isArray(e)?{include:e}:{exclude:e.exclude?S(e.exclude):void 0,include:e.include?S(e.include):void 0}}function G(e){if(!e)return;let{exclude:n,include:t}=W(e),r=n==null?void 0:n.map(k),i=t==null?void 0:t.map(k);return M(r,i)}function T(e){let n=G(e);return n?t=>!!n(t):void 0}var c=g.packages.template,s=g.packages.types;function C(e){var t,r;let n=("directives"in e.body?(r=(t=e.body)==null?void 0:t.directives)!=null?r:[]:[]).map(i=>i.value.value);for(let i of I)if(n.includes(i))return i}function m(e,n,t){var p;let r=C(e);if(!r)return null;let{params:i,body:o,externalNames:l}=P(e),a=n.tgpuAliases.values().next().value;if(a===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(p=n.fileId)!=null?p:""}`);return s.callExpression(c.expression(`${a}.__assignAst`)(),[r==="kernel & js"?e:c.expression`${a}.__removedJsImpl(${t?`"${t}"`:""})`(),c.expression`${d({params:i,body:o,externalNames:l})}`(),s.objectExpression(l.map(u=>s.objectProperty(s.identifier(u),s.identifier(u))))])}function H(e){return{ImportDeclaration(n){A(n.node,e)},ArrowFunctionExpression(n){let t=m(n.node,e,n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionExpression(n){var r;let t=m(n.node,e,(r=n.node.id)!=null&&r.name?n.node.id.name:n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionDeclaration(n){var o;let t=n.node,r=s.functionExpression(t.id,t.params,t.body),i=m(r,e,(o=t.id)==null?void 0:o.name);i&&t.id&&(n.replaceWith(s.variableDeclaration("const",[s.variableDeclarator(t.id,i)])),n.skip())},CallExpression(n){var r;let t=n.node;if(v(t,e)){let i=t.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let{params:o,body:l,externalNames:a}=P(i),p=e.tgpuAliases.values().next().value;if(p===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(r=e.fileId)!=null?r:""}`);let u=C(i);n.replaceWith(s.callExpression(t.callee,[s.callExpression(c.expression(`${p}.__assignAst`)(),[u!=="kernel & js"?c.expression`${p}.__removedJsImpl()`():i,c.expression`${d({params:o,body:l,externalNames:a})}`(),s.objectExpression(a.map(y=>s.objectProperty(s.identifier(y),s.identifier(y))))])])),n.skip()}}}}}function te(){return{visitor:{Program(e,n){var a;let t=(a=n.file)==null?void 0:a.code,r=n.opts,i=n.filename,o=T(r);if(i&&o&&!(o!=null&&o(i))||!(r!=null&&r.forceTgpuAlias)&&t&&!E.some(p=>p.test(t)))return;let l={tgpuAliases:new Set(r!=null&&r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:i};e.traverse(H(l))}}}}export{V as a,L as b,d as c,E as d,A as e,v as f,I as g,te as h};
+//# sourceMappingURL=chunk-4ONRHQSA.js.map

package/chunk-4ONRHQSA.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/babel.ts","../src/common.ts","../src/filter.ts"],"sourcesContent":["import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as babel from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  codeFilterRegexes,\n  type Context,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  type KernelDirective,\n  kernelDirectives,\n  type Options,\n} from './common.ts';\nimport { createFilterForId } from './filter.ts';\n\n// NOTE: @babel/standalone does expose internal packages, as specified in the docs, but the\n// typing for @babel/standalone does not expose them.\nconst template = (\n  Babel as unknown as { packages: { template: typeof TemplateGenerator } }\n).packages.template;\nconst types = (Babel as unknown as { packages: { types: typeof babel } })\n  .packages.types;\n\nfunction getKernelDirective(\n  node:\n    | babel.FunctionDeclaration\n    | babel.FunctionExpression\n    | babel.ArrowFunctionExpression,\n): KernelDirective | undefined {\n  const directives = (\n    'directives' in node.body ? (node.body?.directives ?? []) : []\n  ).map((directive) => directive.value.value);\n\n  for (const directive of kernelDirectives) {\n    if (directives.includes(directive)) {\n      return directive;\n    }\n  }\n}\n\nfunction functionToTranspiled(\n  node: babel.ArrowFunctionExpression | babel.FunctionExpression,\n  ctx: Context,\n  name?: string | undefined,\n): babel.CallExpression | null {\n  const directive = getKernelDirective(node);\n  if (!directive) {\n    return null;\n  }\n\n  const { params, body, externalNames } = transpileFn(node);\n  const tgpuAlias = ctx.tgpuAliases.values().next().value;\n  if (tgpuAlias === undefined) {\n    throw new Error(\n      `No tgpu import found, cannot assign ast to function in file: ${\n        ctx.fileId ?? ''\n      }`,\n    );\n  }\n\n  return types.callExpression(\n    template.expression(`${tgpuAlias}.__assignAst`)(),\n    [\n      directive === 'kernel & js'\n        ? node\n        : template.expression`${tgpuAlias}.__removedJsImpl(${\n          name ? `\"${name}\"` : ''\n        })`(),\n      template.expression`${embedJSON({ params, body, externalNames })}`(),\n      types.objectExpression(\n        externalNames.map((name) =>\n          types.objectProperty(types.identifier(name), types.identifier(name))\n        ),\n      ),\n    ],\n  );\n}\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    ArrowFunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.parentPath.node.type === 'VariableDeclarator'\n          ? path.parentPath.node.id.type === 'Identifier'\n            ? path.parentPath.node.id.name\n            : undefined\n          : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.node.id?.name\n          ? path.node.id.name\n          : path.parentPath.node.type === 'VariableDeclarator'\n          ? path.parentPath.node.id.type === 'Identifier'\n            ? path.parentPath.node.id.name\n            : undefined\n          : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionDeclaration(path) {\n      const node = path.node;\n      const expression = types.functionExpression(\n        node.id,\n        node.params,\n        node.body,\n      );\n      const transpiled = functionToTranspiled(expression, ctx, node.id?.name);\n      if (transpiled && node.id) {\n        path.replaceWith(\n          types.variableDeclaration('const', [\n            types.variableDeclarator(node.id, transpiled),\n          ]),\n        );\n        path.skip();\n      }\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isShellImplementationCall(node, ctx)) {\n        const implementation = node.arguments[0];\n\n        if (\n          implementation &&\n          (implementation.type === 'FunctionExpression' ||\n            implementation.type === 'ArrowFunctionExpression')\n        ) {\n          const { params, body, externalNames } = transpileFn(implementation);\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n          if (tgpuAlias === undefined) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${\n                ctx.fileId ?? ''\n              }`,\n            );\n          }\n\n          const directive = getKernelDirective(implementation);\n\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(\n                template.expression(`${tgpuAlias}.__assignAst`)(),\n                [\n                  directive !== 'kernel & js'\n                    ? template.expression`${tgpuAlias}.__removedJsImpl()`()\n                    : implementation,\n                  template.expression`${\n                    embedJSON({ params, body, externalNames })\n                  }`(),\n                  types.objectExpression(\n                    externalNames.map((name) =>\n                      types.objectProperty(\n                        types.identifier(name),\n                        types.identifier(name),\n                      )\n                    ),\n                  ),\n                ],\n              ),\n            ]),\n          );\n\n          path.skip();\n        }\n      }\n    },\n  };\n}\n\nexport default function () {\n  return {\n    visitor: {\n      Program(path, state) {\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const code: string | undefined = (state as any).file?.code;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const options: Options | undefined = (state as any).opts;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const id: string | undefined = (state as any).filename;\n\n        const filter = createFilterForId(options);\n        if (id && filter && !filter?.(id)) {\n          return;\n        }\n\n        if (\n          !options?.forceTgpuAlias &&\n          code &&\n          !codeFilterRegexes.some((reg) => reg.test(code))\n        ) {\n          return;\n        }\n\n        const ctx: Context = {\n          tgpuAliases: new Set<string>(\n            options?.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n          ),\n          fileId: id,\n        };\n\n        path.traverse(functionVisitor(ctx));\n      },\n    } satisfies TraverseOptions,\n  };\n}\n","import type * as babel from '@babel/types';\nimport type * as acorn from 'acorn';\nimport type { FilterPattern } from 'unplugin';\n\nexport type Context = {\n  /**\n   * How the `tgpu` object is used in code. Since it can be aliased, we\n   * need to catch that and act accordingly.\n   */\n  tgpuAliases: Set<string>;\n  fileId?: string | undefined;\n};\n\nexport interface Options {\n  include?: FilterPattern;\n  exclude?: FilterPattern;\n  enforce?: 'post' | 'pre' | undefined;\n  forceTgpuAlias?: string;\n}\n\nexport const defaultOptions = {\n  include: [/\\.m?[jt]sx?$/],\n};\n\nexport function embedJSON(jsValue: unknown) {\n  return JSON.stringify(jsValue)\n    .replace(/\\u2028/g, '\\\\u2028')\n    .replace(/\\u2029/g, '\\\\u2029');\n}\n\n/**\n * Checks if `node` is an alias for the 'tgpu' object, traditionally\n * available via `import tgpu from 'typegpu'`.\n */\nfunction isTgpu(ctx: Context, node: babel.Node | acorn.AnyNode): boolean {\n  let path = '';\n\n  let tail = node;\n  while (true) {\n    if (tail.type === 'MemberExpression') {\n      if (tail.property.type !== 'Identifier') {\n        // Not handling computed expressions.\n        break;\n      }\n\n      path = path ? `${tail.property.name}.${path}` : tail.property.name;\n      tail = tail.object;\n    } else if (tail.type === 'Identifier') {\n      path = path ? `${tail.name}.${path}` : tail.name;\n      break;\n    } else {\n      break;\n    }\n  }\n\n  return ctx.tgpuAliases.has(path);\n}\n\nconst typegpuImportRegex = /import.*from\\s*['\"]typegpu.*['\"]/;\nconst typegpuDynamicImportRegex = /import\\s*\\(\\s*['\"]\\s*typegpu.*['\"]/;\nconst typegpuRequireRegex = /require\\s*\\(\\s*['\"]\\s*typegpu.*['\"]\\s*\\)/;\n\n/**\n * Regexes used to efficiently determine if a file is\n * meant to be processed by our plugin. We assume every file\n * that should be processed imports `typegpu` in some way.\n */\nexport const codeFilterRegexes = [\n  typegpuImportRegex,\n  typegpuDynamicImportRegex,\n  typegpuRequireRegex,\n];\n\nexport function gatherTgpuAliases(\n  node: acorn.ImportDeclaration | babel.ImportDeclaration,\n  ctx: Context,\n) {\n  if (node.source.value === 'typegpu') {\n    for (const spec of node.specifiers) {\n      if (\n        // The default export of 'typegpu' is the `tgpu` object.\n        spec.type === 'ImportDefaultSpecifier' ||\n        // Aliasing 'tgpu' while importing, e.g. import { tgpu as t } from 'typegpu';\n        (spec.type === 'ImportSpecifier' &&\n          spec.imported.type === 'Identifier' &&\n          spec.imported.name === 'tgpu')\n      ) {\n        ctx.tgpuAliases.add(spec.local.name);\n      } else if (spec.type === 'ImportNamespaceSpecifier') {\n        // Importing everything, e.g. import * as t from 'typegpu';\n        ctx.tgpuAliases.add(`${spec.local.name}.tgpu`);\n      }\n    }\n  }\n}\n\nconst fnShellFunctionNames = ['fn', 'vertexFn', 'fragmentFn', 'computeFn'];\n\nexport function isShellImplementationCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    (node.callee.type === 'CallExpression' &&\n      node.callee.callee.type === 'MemberExpression' &&\n      node.callee.callee.property.type === 'Identifier' &&\n      fnShellFunctionNames.includes(node.callee.callee.property.name) &&\n      node.arguments.length === 1 &&\n      (node.callee.callee.object.type === 'MemberExpression'\n        ? isTgpu(ctx, node.callee.callee.object.object)\n        : isTgpu(ctx, node.callee.callee.object))) || // TODO: remove along with the deprecated 'does' method\n    (node.callee.type === 'MemberExpression' &&\n      node.arguments.length === 1 &&\n      node.callee.property.type === 'Identifier' &&\n      // Assuming that every call to `.does` is related to TypeGPU\n      // because shells can be created separately from calls to `tgpu`,\n      // making it hard to detect.\n      node.callee.property.name === 'does')\n  );\n}\n\nexport const kernelDirectives = ['kernel', 'kernel & js'] as const;\nexport type KernelDirective = (typeof kernelDirectives)[number];\n","// Copied from https://github.com/unjs/unplugin/blob/f514bf8e2d751b48b3a5acb1077eb1292d9711b3/src/utils/filter.ts#L2\n// Used only in the babel version of the plugin, since others can rely on native unplugin functionality.\n\nimport { resolve } from 'node:path';\nimport picomatch from 'picomatch';\nimport type {\n  Arrayable,\n  Nullable,\n  StringFilter,\n  StringOrRegExp,\n} from 'unplugin';\n\nfunction toArray<T>(array?: Nullable<Arrayable<T>>): Array<T> {\n  const result = array || [];\n  if (Array.isArray(result)) return result;\n  return [result];\n}\n\nconst BACKSLASH_REGEX = /\\\\/g;\nfunction normalize(path: string): string {\n  return path.replace(BACKSLASH_REGEX, '/');\n}\n\nconst ABSOLUTE_PATH_REGEX = /^(?:\\/|(?:[A-Z]:)?[/\\\\|])/i;\nfunction isAbsolute(path: string): boolean {\n  return ABSOLUTE_PATH_REGEX.test(path);\n}\n\nexport type PluginFilter = (input: string) => boolean;\nexport type TransformHookFilter = (id: string, code: string) => boolean;\n\ninterface NormalizedStringFilter {\n  include?: StringOrRegExp[] | undefined;\n  exclude?: StringOrRegExp[] | undefined;\n}\n\nfunction getMatcherString(glob: string, cwd: string) {\n  if (glob.startsWith('**') || isAbsolute(glob)) {\n    return normalize(glob);\n  }\n\n  const resolved = resolve(cwd, glob);\n  return normalize(resolved);\n}\n\nfunction patternToIdFilter(pattern: StringOrRegExp): PluginFilter {\n  if (pattern instanceof RegExp) {\n    return (id: string) => {\n      const normalizedId = normalize(id);\n      const result = pattern.test(normalizedId);\n      pattern.lastIndex = 0;\n      return result;\n    };\n  }\n  const cwd = process.cwd();\n  const glob = getMatcherString(pattern, cwd);\n  const matcher = picomatch(glob, { dot: true });\n  return (id: string) => {\n    const normalizedId = normalize(id);\n    return matcher(normalizedId);\n  };\n}\n\nfunction patternToCodeFilter(pattern: StringOrRegExp): PluginFilter {\n  if (pattern instanceof RegExp) {\n    return (code: string) => {\n      const result = pattern.test(code);\n      pattern.lastIndex = 0;\n      return result;\n    };\n  }\n  return (code: string) => code.includes(pattern);\n}\n\nfunction createFilter(\n  exclude: PluginFilter[] | undefined,\n  include: PluginFilter[] | undefined,\n): PluginFilter | undefined {\n  if (!exclude && !include) {\n    return;\n  }\n\n  return (input) => {\n    if (exclude?.some((filter) => filter(input))) {\n      return false;\n    }\n    if (include?.some((filter) => filter(input))) {\n      return true;\n    }\n    return !(include && include.length > 0);\n  };\n}\n\nfunction normalizeFilter(filter: StringFilter): NormalizedStringFilter {\n  if (typeof filter === 'string' || filter instanceof RegExp) {\n    return {\n      include: [filter],\n    };\n  }\n  if (Array.isArray(filter)) {\n    return {\n      include: filter,\n    };\n  }\n  return {\n    exclude: filter.exclude ? toArray(filter.exclude) : undefined,\n    include: filter.include ? toArray(filter.include) : undefined,\n  };\n}\n\nfunction createIdFilter(\n  filter: StringFilter | undefined,\n): PluginFilter | undefined {\n  if (!filter) return;\n  const { exclude, include } = normalizeFilter(filter);\n  const excludeFilter = exclude?.map(patternToIdFilter);\n  const includeFilter = include?.map(patternToIdFilter);\n  return createFilter(excludeFilter, includeFilter);\n}\n\nfunction createCodeFilter(\n  filter: StringFilter | undefined,\n): PluginFilter | undefined {\n  if (!filter) return;\n  const { exclude, include } = normalizeFilter(filter);\n  const excludeFilter = exclude?.map(patternToCodeFilter);\n  const includeFilter = include?.map(patternToCodeFilter);\n  return createFilter(excludeFilter, includeFilter);\n}\n\nexport function createFilterForId(\n  filter: StringFilter | undefined,\n): PluginFilter | undefined {\n  const filterFunction = createIdFilter(filter);\n  return filterFunction ? (id) => !!filterFunction(id) : undefined;\n}\n\nfunction createFilterForTransform(\n  idFilter: StringFilter | undefined,\n  codeFilter: StringFilter | undefined,\n): TransformHookFilter | undefined {\n  if (!idFilter && !codeFilter) return;\n  const idFilterFunction = createIdFilter(idFilter);\n  const codeFilterFunction = createCodeFilter(codeFilter);\n  return (id, code) => {\n    let fallback = true;\n    if (idFilterFunction) {\n      fallback &&= idFilterFunction(id);\n    }\n    if (!fallback) {\n      return false;\n    }\n\n    if (codeFilterFunction) {\n      fallback &&= codeFilterFunction(code);\n    }\n    return fallback;\n  };\n}\n"],"mappings":"yVAAA,UAAYA,MAAW,oBAIvB,OAAS,eAAAC,MAAmB,mBCgBrB,IAAMC,EAAiB,CAC5B,QAAS,CAAC,cAAc,CAC1B,EAEO,SAASC,EAAUC,EAAkB,CAC1C,OAAO,KAAK,UAAUA,CAAO,EAC1B,QAAQ,UAAW,SAAS,EAC5B,QAAQ,UAAW,SAAS,CACjC,CAMA,SAASC,EAAOC,EAAcC,EAA2C,CACvE,IAAIC,EAAO,GAEPC,EAAOF,EACX,OACE,GAAIE,EAAK,OAAS,mBAAoB,CACpC,GAAIA,EAAK,SAAS,OAAS,aAEzB,MAGFD,EAAOA,EAAO,GAAGC,EAAK,SAAS,IAAI,IAAID,CAAI,GAAKC,EAAK,SAAS,KAC9DA,EAAOA,EAAK,MACd,SAAWA,EAAK,OAAS,aAAc,CACrCD,EAAOA,EAAO,GAAGC,EAAK,IAAI,IAAID,CAAI,GAAKC,EAAK,KAC5C,KACF,KACE,OAIJ,OAAOH,EAAI,YAAY,IAAIE,CAAI,CACjC,CAEA,IAAME,EAAqB,mCACrBC,EAA4B,qCAC5BC,EAAsB,2CAOfC,EAAoB,CAC/BH,EACAC,EACAC,CACF,EAEO,SAASE,EACdP,EACAD,EACA,CACA,GAAIC,EAAK,OAAO,QAAU,UACxB,QAAWQ,KAAQR,EAAK,WAGpBQ,EAAK,OAAS,0BAEbA,EAAK,OAAS,mBACbA,EAAK,SAAS,OAAS,cACvBA,EAAK,SAAS,OAAS,OAEzBT,EAAI,YAAY,IAAIS,EAAK,MAAM,IAAI,EAC1BA,EAAK,OAAS,4BAEvBT,EAAI,YAAY,IAAI,GAAGS,EAAK,MAAM,IAAI,OAAO,CAIrD,CAEA,IAAMC,EAAuB,CAAC,KAAM,WAAY,aAAc,WAAW,EAElE,SAASC,EACdV,EACAD,EACA,CACA,OACGC,EAAK,OAAO,OAAS,kBACpBA,EAAK,OAAO,OAAO,OAAS,oBAC5BA,EAAK,OAAO,OAAO,SAAS,OAAS,cACrCS,EAAqB,SAAST,EAAK,OAAO,OAAO,SAAS,IAAI,GAC9DA,EAAK,UAAU,SAAW,IACzBA,EAAK,OAAO,OAAO,OAAO,OAAS,mBAChCF,EAAOC,EAAKC,EAAK,OAAO,OAAO,OAAO,MAAM,EAC5CF,EAAOC,EAAKC,EAAK,OAAO,OAAO,MAAM,IAC1CA,EAAK,OAAO,OAAS,oBACpBA,EAAK,UAAU,SAAW,GAC1BA,EAAK,OAAO,SAAS,OAAS,cAI9BA,EAAK,OAAO,SAAS,OAAS,MAEpC,CAEO,IAAMW,EAAmB,CAAC,SAAU,aAAa,ECtHxD,OAAS,WAAAC,MAAe,YACxB,OAAOC,MAAe,YAQtB,SAASC,EAAWC,EAA0C,CAC5D,IAAMC,EAASD,GAAS,CAAC,EACzB,OAAI,MAAM,QAAQC,CAAM,EAAUA,EAC3B,CAACA,CAAM,CAChB,CAEA,IAAMC,EAAkB,MACxB,SAASC,EAAUC,EAAsB,CACvC,OAAOA,EAAK,QAAQF,EAAiB,GAAG,CAC1C,CAEA,IAAMG,EAAsB,6BAC5B,SAASC,EAAWF,EAAuB,CACzC,OAAOC,EAAoB,KAAKD,CAAI,CACtC,CAUA,SAASG,EAAiBC,EAAcC,EAAa,CACnD,GAAID,EAAK,WAAW,IAAI,GAAKF,EAAWE,CAAI,EAC1C,OAAOL,EAAUK,CAAI,EAGvB,IAAME,EAAWb,EAAQY,EAAKD,CAAI,EAClC,OAAOL,EAAUO,CAAQ,CAC3B,CAEA,SAASC,EAAkBC,EAAuC,CAChE,GAAIA,aAAmB,OACrB,OAAQC,GAAe,CACrB,IAAMC,EAAeX,EAAUU,CAAE,EAC3BZ,EAASW,EAAQ,KAAKE,CAAY,EACxC,OAAAF,EAAQ,UAAY,EACbX,CACT,EAEF,IAAMQ,EAAM,QAAQ,IAAI,EAClBD,EAAOD,EAAiBK,EAASH,CAAG,EACpCM,EAAUjB,EAAUU,EAAM,CAAE,IAAK,EAAK,CAAC,EAC7C,OAAQK,GAAe,CACrB,IAAMC,EAAeX,EAAUU,CAAE,EACjC,OAAOE,EAAQD,CAAY,CAC7B,CACF,CAaA,SAASE,EACPC,EACAC,EAC0B,CAC1B,GAAI,GAACD,GAAW,CAACC,GAIjB,OAAQC,GACFF,GAAA,MAAAA,EAAS,KAAMG,GAAWA,EAAOD,CAAK,GACjC,GAELD,GAAA,MAAAA,EAAS,KAAME,GAAWA,EAAOD,CAAK,GACjC,GAEF,EAAED,GAAWA,EAAQ,OAAS,EAEzC,CAEA,SAASG,EAAgBD,EAA8C,CACrE,OAAI,OAAOA,GAAW,UAAYA,aAAkB,OAC3C,CACL,QAAS,CAACA,CAAM,CAClB,EAEE,MAAM,QAAQA,CAAM,EACf,CACL,QAASA,CACX,EAEK,CACL,QAASA,EAAO,QAAUE,EAAQF,EAAO,OAAO,EAAI,OACpD,QAASA,EAAO,QAAUE,EAAQF,EAAO,OAAO,EAAI,MACtD,CACF,CAEA,SAASG,EACPH,EAC0B,CAC1B,GAAI,CAACA,EAAQ,OACb,GAAM,CAAE,QAAAH,EAAS,QAAAC,CAAQ,EAAIG,EAAgBD,CAAM,EAC7CI,EAAgBP,GAAA,YAAAA,EAAS,IAAIQ,GAC7BC,EAAgBR,GAAA,YAAAA,EAAS,IAAIO,GACnC,OAAOT,EAAaQ,EAAeE,CAAa,CAClD,CAYO,SAASC,EACdC,EAC0B,CAC1B,IAAMC,EAAiBC,EAAeF,CAAM,EAC5C,OAAOC,EAAkBE,GAAO,CAAC,CAACF,EAAeE,CAAE,EAAI,MACzD,CFpHA,IAAMC,EAEJ,WAAS,SACLC,EACH,WAAS,MAEZ,SAASC,EACPC,EAI6B,CA9B/B,IAAAC,EAAAC,EA+BE,IAAMC,GACJ,eAAgBH,EAAK,MAAQE,GAAAD,EAAAD,EAAK,OAAL,YAAAC,EAAW,aAAX,KAAAC,EAAyB,CAAC,EAAK,CAAC,GAC7D,IAAKE,GAAcA,EAAU,MAAM,KAAK,EAE1C,QAAWA,KAAaC,EACtB,GAAIF,EAAW,SAASC,CAAS,EAC/B,OAAOA,CAGb,CAEA,SAASE,EACPN,EACAO,EACAC,EAC6B,CA9C/B,IAAAP,EA+CE,IAAMG,EAAYL,EAAmBC,CAAI,EACzC,GAAI,CAACI,EACH,OAAO,KAGT,GAAM,CAAE,OAAAK,EAAQ,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYZ,CAAI,EAClDa,EAAYN,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAClD,GAAIM,IAAc,OAChB,MAAM,IAAI,MACR,iEACEZ,EAAAM,EAAI,SAAJ,KAAAN,EAAc,EAChB,EACF,EAGF,OAAOH,EAAM,eACXD,EAAS,WAAW,GAAGgB,CAAS,cAAc,EAAE,EAChD,CACET,IAAc,cACVJ,EACAH,EAAS,aAAagB,CAAS,oBAC/BL,EAAO,IAAIA,CAAI,IAAM,EACvB,IAAI,EACNX,EAAS,aAAaiB,EAAU,CAAE,OAAAL,EAAQ,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,GAAG,EACnEb,EAAM,iBACJa,EAAc,IAAKH,GACjBV,EAAM,eAAeA,EAAM,WAAWU,CAAI,EAAGV,EAAM,WAAWU,CAAI,CAAC,CACrE,CACF,CACF,CACF,CACF,CAEA,SAASO,EAAgBR,EAA+B,CACtD,MAAO,CACL,kBAAkBS,EAAM,CACtBC,EAAkBD,EAAK,KAAMT,CAAG,CAClC,EAEA,wBAAwBS,EAAM,CAC5B,IAAME,EAAaZ,EACjBU,EAAK,KACLT,EACAS,EAAK,WAAW,KAAK,OAAS,sBAC1BA,EAAK,WAAW,KAAK,GAAG,OAAS,aAC/BA,EAAK,WAAW,KAAK,GAAG,KAE1B,MACN,EACIE,IACFF,EAAK,YAAYE,CAAU,EAC3BF,EAAK,KAAK,EAEd,EAEA,mBAAmBA,EAAM,CAtG7B,IAAAf,EAuGM,IAAMiB,EAAaZ,EACjBU,EAAK,KACLT,GACAN,EAAAe,EAAK,KAAK,KAAV,MAAAf,EAAc,KACVe,EAAK,KAAK,GAAG,KACbA,EAAK,WAAW,KAAK,OAAS,sBAC9BA,EAAK,WAAW,KAAK,GAAG,OAAS,aAC/BA,EAAK,WAAW,KAAK,GAAG,KAE1B,MACN,EACIE,IACFF,EAAK,YAAYE,CAAU,EAC3BF,EAAK,KAAK,EAEd,EAEA,oBAAoBA,EAAM,CAxH9B,IAAAf,EAyHM,IAAMD,EAAOgB,EAAK,KACZG,EAAarB,EAAM,mBACvBE,EAAK,GACLA,EAAK,OACLA,EAAK,IACP,EACMkB,EAAaZ,EAAqBa,EAAYZ,GAAKN,EAAAD,EAAK,KAAL,YAAAC,EAAS,IAAI,EAClEiB,GAAclB,EAAK,KACrBgB,EAAK,YACHlB,EAAM,oBAAoB,QAAS,CACjCA,EAAM,mBAAmBE,EAAK,GAAIkB,CAAU,CAC9C,CAAC,CACH,EACAF,EAAK,KAAK,EAEd,EAEA,eAAeA,EAAM,CA1IzB,IAAAf,EA2IM,IAAMD,EAAOgB,EAAK,KAElB,GAAII,EAA0BpB,EAAMO,CAAG,EAAG,CACxC,IAAMc,EAAiBrB,EAAK,UAAU,CAAC,EAEvC,GACEqB,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,GAAM,CAAE,OAAAZ,EAAQ,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYS,CAAc,EAC5DR,EAAYN,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAClD,GAAIM,IAAc,OAChB,MAAM,IAAI,MACR,iEACEZ,EAAAM,EAAI,SAAJ,KAAAN,EAAc,EAChB,EACF,EAGF,IAAMG,EAAYL,EAAmBsB,CAAc,EAEnDL,EAAK,YACHlB,EAAM,eAAeE,EAAK,OAAQ,CAChCF,EAAM,eACJD,EAAS,WAAW,GAAGgB,CAAS,cAAc,EAAE,EAChD,CACET,IAAc,cACVP,EAAS,aAAagB,CAAS,qBAAqB,EACpDQ,EACJxB,EAAS,aACPiB,EAAU,CAAE,OAAAL,EAAQ,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAC3C,GAAG,EACHb,EAAM,iBACJa,EAAc,IAAKH,GACjBV,EAAM,eACJA,EAAM,WAAWU,CAAI,EACrBV,EAAM,WAAWU,CAAI,CACvB,CACF,CACF,CACF,CACF,CACF,CAAC,CACH,EAEAQ,EAAK,KAAK,CACZ,CACF,CACF,CACF,CACF,CAEe,SAARM,IAAoB,CACzB,MAAO,CACL,QAAS,CACP,QAAQN,EAAMO,EAAO,CAnM3B,IAAAtB,EAqMQ,IAAMuB,GAA4BvB,EAAAsB,EAAc,OAAd,YAAAtB,EAAoB,KAEhDwB,EAAgCF,EAAc,KAE9CG,EAA0BH,EAAc,SAExCI,EAASC,EAAkBH,CAAO,EAKxC,GAJIC,GAAMC,GAAU,EAACA,GAAA,MAAAA,EAASD,KAK5B,EAACD,GAAA,MAAAA,EAAS,iBACVD,GACA,CAACK,EAAkB,KAAMC,GAAQA,EAAI,KAAKN,CAAI,CAAC,EAE/C,OAGF,IAAMjB,EAAe,CACnB,YAAa,IAAI,IACfkB,GAAA,MAAAA,EAAS,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACxD,EACA,OAAQC,CACV,EAEAV,EAAK,SAASD,EAAgBR,CAAG,CAAC,CACpC,CACF,CACF,CACF","names":["Babel","transpileFn","defaultOptions","embedJSON","jsValue","isTgpu","ctx","node","path","tail","typegpuImportRegex","typegpuDynamicImportRegex","typegpuRequireRegex","codeFilterRegexes","gatherTgpuAliases","spec","fnShellFunctionNames","isShellImplementationCall","kernelDirectives","resolve","picomatch","toArray","array","result","BACKSLASH_REGEX","normalize","path","ABSOLUTE_PATH_REGEX","isAbsolute","getMatcherString","glob","cwd","resolved","patternToIdFilter","pattern","id","normalizedId","matcher","createFilter","exclude","include","input","filter","normalizeFilter","toArray","createIdFilter","excludeFilter","patternToIdFilter","includeFilter","createFilterForId","filter","filterFunction","createIdFilter","id","template","types","getKernelDirective","node","_a","_b","directives","directive","kernelDirectives","functionToTranspiled","ctx","name","params","body","externalNames","transpileFn","tgpuAlias","embedJSON","functionVisitor","path","gatherTgpuAliases","transpiled","expression","isShellImplementationCall","implementation","babel_default","state","code","options","id","filter","createFilterForId","codeFilterRegexes","reg"]}

package/{chunk-HQFSG5ZM.cjs → chunk-BCOJLF4T.cjs}

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var D=Object.defineProperty;var b=Object.getOwnPropertySymbols;var N=Object.prototype.hasOwnProperty,O=Object.prototype.propertyIsEnumerable;var x=(e,n,t)=>n in e?D(e,n,{enumerable:!0,configurable:!0,writable:!0,value:t}):e[n]=t,V= exports.a =(e,n)=>{for(var t in n||(n={}))N.call(n,t)&&x(e,t,n[t]);if(b)for(var t of b(n))O.call(n,t)&&x(e,t,n[t]);return e};var _standalone = require('@babel/standalone'); var g = _interopRequireWildcard(_standalone);var _tinyestforwgsl = require('tinyest-for-wgsl');var L={include:[/\.m?[jt]sx?$/]};function d(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function F(e,n){let t="",r=n;for(;;)if(r.type==="MemberExpression"){if(r.property.type!=="Identifier")break;t=t?`${r.property.name}.${t}`:r.property.name,r=r.object}else if(r.type==="Identifier"){t=t?`${r.name}.${t}`:r.name;break}else break;return e.tgpuAliases.has(t)}var R=/import.*from\s*['"]typegpu.*['"]/,$=/import\s*\(\s*['"]\s*typegpu.*['"]/,w=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/,E= exports.d =[R,$,w];function A(e,n){if(e.source.value==="typegpu")for(let t of e.specifiers)t.type==="ImportDefaultSpecifier"||t.type==="ImportSpecifier"&&t.imported.type==="Identifier"&&t.imported.name==="tgpu"?n.tgpuAliases.add(t.local.name):t.type==="ImportNamespaceSpecifier"&&n.tgpuAliases.add(`${t.local.name}.tgpu`)}var j=["fn","vertexFn","fragmentFn","computeFn"];function v(e,n){return e.callee.type==="CallExpression"&&e.callee.callee.type==="MemberExpression"&&e.callee.callee.property.type==="Identifier"&&j.includes(e.callee.callee.property.name)&&e.arguments.length===1&&(e.callee.callee.object.type==="MemberExpression"?F(n,e.callee.callee.object.object):F(n,e.callee.callee.object))||e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&e.callee.property.name==="does"}var I=["kernel","kernel & js"];var _path = require('path');var _picomatch = require('picomatch'); var _picomatch2 = _interopRequireDefault(_picomatch);function S(e){let n=e||[];return Array.isArray(n)?n:[n]}var z=/\\/g;function f(e){return e.replace(z,"/")}var B=/^(?:\/|(?:[A-Z]:)?[/\\|])/i;function J(e){return B.test(e)}function K(e,n){if(e.startsWith("**")||J(e))return f(e);let t=_path.resolve.call(void 0, n,e);return f(t)}function k(e){if(e instanceof RegExp)return i=>{let o=f(i),l=e.test(o);return e.lastIndex=0,l};let n=process.cwd(),t=K(e,n),r=_picomatch2.default.call(void 0, t,{dot:!0});return i=>{let o=f(i);return r(o)}}function M(e,n){if(!(!e&&!n))return t=>e!=null&&e.some(r=>r(t))?!1:n!=null&&n.some(r=>r(t))?!0:!(n&&n.length>0)}function W(e){return typeof e=="string"||e instanceof RegExp?{include:[e]}:Array.isArray(e)?{include:e}:{exclude:e.exclude?S(e.exclude):void 0,include:e.include?S(e.include):void 0}}function G(e){if(!e)return;let{exclude:n,include:t}=W(e),r=n==null?void 0:n.map(k),i=t==null?void 0:t.map(k);return M(r,i)}function T(e){let n=G(e);return n?t=>!!n(t):void 0}var c=g.packages.template,s=g.packages.types;function C(e){var t,r;let n=("directives"in e.body?(r=(t=e.body)==null?void 0:t.directives)!=null?r:[]:[]).map(i=>i.value.value);for(let i of I)if(n.includes(i))return i}function m(e,n,t){var p;let r=C(e);if(!r)return null;let{argNames:i,body:o,externalNames:l}=_tinyestforwgsl.transpileFn.call(void 0, e),a=n.tgpuAliases.values().next().value;if(a===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(p=n.fileId)!=null?p:""}`);return s.callExpression(c.expression(`${a}.__assignAst`)(),[r==="kernel & js"?e:c.expression`${a}.__removedJsImpl(${t?`"${t}"`:""})`(),c.expression`${d({argNames:i,body:o,externalNames:l})}`(),s.objectExpression(l.map(u=>s.objectProperty(s.identifier(u),s.identifier(u))))])}function H(e){return{ImportDeclaration(n){A(n.node,e)},ArrowFunctionExpression(n){let t=m(n.node,e,n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionExpression(n){var r;let t=m(n.node,e,(r=n.node.id)!=null&&r.name?n.node.id.name:n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionDeclaration(n){var o;let t=n.node,r=s.functionExpression(t.id,t.params,t.body),i=m(r,e,(o=t.id)==null?void 0:o.name);i&&t.id&&(n.replaceWith(s.variableDeclaration("const",[s.variableDeclarator(t.id,i)])),n.skip())},CallExpression(n){var r;let t=n.node;if(v(t,e)){let i=t.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let{argNames:o,body:l,externalNames:a}=_tinyestforwgsl.transpileFn.call(void 0, i),p=e.tgpuAliases.values().next().value;if(p===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(r=e.fileId)!=null?r:""}`);let u=C(i);n.replaceWith(s.callExpression(t.callee,[s.callExpression(c.expression(`${p}.__assignAst`)(),[u!=="kernel & js"?c.expression`${p}.__removedJsImpl()`():i,c.expression`${d({argNames:o,body:l,externalNames:a})}`(),s.objectExpression(a.map(y=>s.objectProperty(s.identifier(y),s.identifier(y))))])])),n.skip()}}}}}function te(){return{visitor:{Program(e,n){var a;let t=(a=n.file)==null?void 0:a.code,r=n.opts,i=n.filename,o=T(r);if(i&&o&&!(o!=null&&o(i))||!(r!=null&&r.forceTgpuAlias)&&t&&!E.some(p=>p.test(t)))return;let l={tgpuAliases:new Set(r!=null&&r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:i};e.traverse(H(l))}}}}exports.a = V; exports.b = L; exports.c = d; exports.d = E; exports.e = A; exports.f = v; exports.g = I; exports.h = te;
-//# sourceMappingURL=chunk-HQFSG5ZM.cjs.map
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var D=Object.defineProperty;var b=Object.getOwnPropertySymbols;var O=Object.prototype.hasOwnProperty,R=Object.prototype.propertyIsEnumerable;var x=(e,n,t)=>n in e?D(e,n,{enumerable:!0,configurable:!0,writable:!0,value:t}):e[n]=t,V= exports.a =(e,n)=>{for(var t in n||(n={}))O.call(n,t)&&x(e,t,n[t]);if(b)for(var t of b(n))R.call(n,t)&&x(e,t,n[t]);return e};var _standalone = require('@babel/standalone'); var g = _interopRequireWildcard(_standalone);var _tinyestforwgsl = require('tinyest-for-wgsl');var L={include:[/\.m?[jt]sx?$/]};function d(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function F(e,n){let t="",r=n;for(;;)if(r.type==="MemberExpression"){if(r.property.type!=="Identifier")break;t=t?`${r.property.name}.${t}`:r.property.name,r=r.object}else if(r.type==="Identifier"){t=t?`${r.name}.${t}`:r.name;break}else break;return e.tgpuAliases.has(t)}var $=/import.*from\s*['"]typegpu.*['"]/,w=/import\s*\(\s*['"]\s*typegpu.*['"]/,N=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/,E= exports.d =[$,w,N];function A(e,n){if(e.source.value==="typegpu")for(let t of e.specifiers)t.type==="ImportDefaultSpecifier"||t.type==="ImportSpecifier"&&t.imported.type==="Identifier"&&t.imported.name==="tgpu"?n.tgpuAliases.add(t.local.name):t.type==="ImportNamespaceSpecifier"&&n.tgpuAliases.add(`${t.local.name}.tgpu`)}var j=["fn","vertexFn","fragmentFn","computeFn"];function v(e,n){return e.callee.type==="CallExpression"&&e.callee.callee.type==="MemberExpression"&&e.callee.callee.property.type==="Identifier"&&j.includes(e.callee.callee.property.name)&&e.arguments.length===1&&(e.callee.callee.object.type==="MemberExpression"?F(n,e.callee.callee.object.object):F(n,e.callee.callee.object))||e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&e.callee.property.name==="does"}var I=["kernel","kernel & js"];var _path = require('path');var _picomatch = require('picomatch'); var _picomatch2 = _interopRequireDefault(_picomatch);function S(e){let n=e||[];return Array.isArray(n)?n:[n]}var z=/\\/g;function f(e){return e.replace(z,"/")}var B=/^(?:\/|(?:[A-Z]:)?[/\\|])/i;function J(e){return B.test(e)}function K(e,n){if(e.startsWith("**")||J(e))return f(e);let t=_path.resolve.call(void 0, n,e);return f(t)}function k(e){if(e instanceof RegExp)return i=>{let o=f(i),l=e.test(o);return e.lastIndex=0,l};let n=process.cwd(),t=K(e,n),r=_picomatch2.default.call(void 0, t,{dot:!0});return i=>{let o=f(i);return r(o)}}function M(e,n){if(!(!e&&!n))return t=>e!=null&&e.some(r=>r(t))?!1:n!=null&&n.some(r=>r(t))?!0:!(n&&n.length>0)}function W(e){return typeof e=="string"||e instanceof RegExp?{include:[e]}:Array.isArray(e)?{include:e}:{exclude:e.exclude?S(e.exclude):void 0,include:e.include?S(e.include):void 0}}function G(e){if(!e)return;let{exclude:n,include:t}=W(e),r=n==null?void 0:n.map(k),i=t==null?void 0:t.map(k);return M(r,i)}function T(e){let n=G(e);return n?t=>!!n(t):void 0}var c=g.packages.template,s=g.packages.types;function C(e){var t,r;let n=("directives"in e.body?(r=(t=e.body)==null?void 0:t.directives)!=null?r:[]:[]).map(i=>i.value.value);for(let i of I)if(n.includes(i))return i}function m(e,n,t){var p;let r=C(e);if(!r)return null;let{params:i,body:o,externalNames:l}=_tinyestforwgsl.transpileFn.call(void 0, e),a=n.tgpuAliases.values().next().value;if(a===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(p=n.fileId)!=null?p:""}`);return s.callExpression(c.expression(`${a}.__assignAst`)(),[r==="kernel & js"?e:c.expression`${a}.__removedJsImpl(${t?`"${t}"`:""})`(),c.expression`${d({params:i,body:o,externalNames:l})}`(),s.objectExpression(l.map(u=>s.objectProperty(s.identifier(u),s.identifier(u))))])}function H(e){return{ImportDeclaration(n){A(n.node,e)},ArrowFunctionExpression(n){let t=m(n.node,e,n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionExpression(n){var r;let t=m(n.node,e,(r=n.node.id)!=null&&r.name?n.node.id.name:n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);t&&(n.replaceWith(t),n.skip())},FunctionDeclaration(n){var o;let t=n.node,r=s.functionExpression(t.id,t.params,t.body),i=m(r,e,(o=t.id)==null?void 0:o.name);i&&t.id&&(n.replaceWith(s.variableDeclaration("const",[s.variableDeclarator(t.id,i)])),n.skip())},CallExpression(n){var r;let t=n.node;if(v(t,e)){let i=t.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let{params:o,body:l,externalNames:a}=_tinyestforwgsl.transpileFn.call(void 0, i),p=e.tgpuAliases.values().next().value;if(p===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(r=e.fileId)!=null?r:""}`);let u=C(i);n.replaceWith(s.callExpression(t.callee,[s.callExpression(c.expression(`${p}.__assignAst`)(),[u!=="kernel & js"?c.expression`${p}.__removedJsImpl()`():i,c.expression`${d({params:o,body:l,externalNames:a})}`(),s.objectExpression(a.map(y=>s.objectProperty(s.identifier(y),s.identifier(y))))])])),n.skip()}}}}}function te(){return{visitor:{Program(e,n){var a;let t=(a=n.file)==null?void 0:a.code,r=n.opts,i=n.filename,o=T(r);if(i&&o&&!(o!=null&&o(i))||!(r!=null&&r.forceTgpuAlias)&&t&&!E.some(p=>p.test(t)))return;let l={tgpuAliases:new Set(r!=null&&r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:i};e.traverse(H(l))}}}}exports.a = V; exports.b = L; exports.c = d; exports.d = E; exports.e = A; exports.f = v; exports.g = I; exports.h = te;
+//# sourceMappingURL=chunk-BCOJLF4T.cjs.map

package/chunk-BCOJLF4T.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-BCOJLF4T.cjs","../src/babel.ts","../src/common.ts"],"names":["defaultOptions","embedJSON","jsValue","isTgpu","ctx","node","path","tail"],"mappings":"AAAA,+bAAI,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CCAxV,6FAAuB,kDAIK,ICgBfA,CAAAA,CAAiB,CAC5B,OAAA,CAAS,CAAC,cAAc,CAC1B,CAAA,CAEO,SAASC,CAAAA,CAAUC,CAAAA,CAAkB,CAC1C,OAAO,IAAA,CAAK,SAAA,CAAUA,CAAO,CAAA,CAC1B,OAAA,CAAQ,SAAA,CAAW,SAAS,CAAA,CAC5B,OAAA,CAAQ,SAAA,CAAW,SAAS,CACjC,CAMA,SAASC,CAAAA,CAAOC,CAAAA,CAAcC,CAAAA,CAA2C,CACvE,IAAIC,CAAAA,CAAO,EAAA,CAEPC,CAAAA,CAAOF,CAAAA,CACX,GAAA,CAAA,CAAA,CAAA,CACE,EAAA,CAAIE,CAAAA,CAAK,IAAA,GAAS,kBAAA,CAAoB,CACpC,EAAA,CAAIA,CAAAA,CAAK,QAAA,CAAS,IAAA,GAAS,YAAA,CAEzB,KAAA,CAGFD,CAAAA,CAAOA,CAAAA,CAAO,CAAA,EAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-BCOJLF4T.cjs","sourcesContent":[null,"import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as babel from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  codeFilterRegexes,\n  type Context,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  type KernelDirective,\n  kernelDirectives,\n  type Options,\n} from './common.ts';\nimport { createFilterForId } from './filter.ts';\n\n// NOTE: @babel/standalone does expose internal packages, as specified in the docs, but the\n// typing for @babel/standalone does not expose them.\nconst template = (\n  Babel as unknown as { packages: { template: typeof TemplateGenerator } }\n).packages.template;\nconst types = (Babel as unknown as { packages: { types: typeof babel } })\n  .packages.types;\n\nfunction getKernelDirective(\n  node:\n    | babel.FunctionDeclaration\n    | babel.FunctionExpression\n    | babel.ArrowFunctionExpression,\n): KernelDirective | undefined {\n  const directives = (\n    'directives' in node.body ? (node.body?.directives ?? []) : []\n  ).map((directive) => directive.value.value);\n\n  for (const directive of kernelDirectives) {\n    if (directives.includes(directive)) {\n      return directive;\n    }\n  }\n}\n\nfunction functionToTranspiled(\n  node: babel.ArrowFunctionExpression | babel.FunctionExpression,\n  ctx: Context,\n  name?: string | undefined,\n): babel.CallExpression | null {\n  const directive = getKernelDirective(node);\n  if (!directive) {\n    return null;\n  }\n\n  const { params, body, externalNames } = transpileFn(node);\n  const tgpuAlias = ctx.tgpuAliases.values().next().value;\n  if (tgpuAlias === undefined) {\n    throw new Error(\n      `No tgpu import found, cannot assign ast to function in file: ${\n        ctx.fileId ?? ''\n      }`,\n    );\n  }\n\n  return types.callExpression(\n    template.expression(`${tgpuAlias}.__assignAst`)(),\n    [\n      directive === 'kernel & js'\n        ? node\n        : template.expression`${tgpuAlias}.__removedJsImpl(${\n          name ? `\"${name}\"` : ''\n        })`(),\n      template.expression`${embedJSON({ params, body, externalNames })}`(),\n      types.objectExpression(\n        externalNames.map((name) =>\n          types.objectProperty(types.identifier(name), types.identifier(name))\n        ),\n      ),\n    ],\n  );\n}\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    ArrowFunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.parentPath.node.type === 'VariableDeclarator'\n          ? path.parentPath.node.id.type === 'Identifier'\n            ? path.parentPath.node.id.name\n            : undefined\n          : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.node.id?.name\n          ? path.node.id.name\n          : path.parentPath.node.type === 'VariableDeclarator'\n          ? path.parentPath.node.id.type === 'Identifier'\n            ? path.parentPath.node.id.name\n            : undefined\n          : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionDeclaration(path) {\n      const node = path.node;\n      const expression = types.functionExpression(\n        node.id,\n        node.params,\n        node.body,\n      );\n      const transpiled = functionToTranspiled(expression, ctx, node.id?.name);\n      if (transpiled && node.id) {\n        path.replaceWith(\n          types.variableDeclaration('const', [\n            types.variableDeclarator(node.id, transpiled),\n          ]),\n        );\n        path.skip();\n      }\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isShellImplementationCall(node, ctx)) {\n        const implementation = node.arguments[0];\n\n        if (\n          implementation &&\n          (implementation.type === 'FunctionExpression' ||\n            implementation.type === 'ArrowFunctionExpression')\n        ) {\n          const { params, body, externalNames } = transpileFn(implementation);\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n          if (tgpuAlias === undefined) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${\n                ctx.fileId ?? ''\n              }`,\n            );\n          }\n\n          const directive = getKernelDirective(implementation);\n\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(\n                template.expression(`${tgpuAlias}.__assignAst`)(),\n                [\n                  directive !== 'kernel & js'\n                    ? template.expression`${tgpuAlias}.__removedJsImpl()`()\n                    : implementation,\n                  template.expression`${\n                    embedJSON({ params, body, externalNames })\n                  }`(),\n                  types.objectExpression(\n                    externalNames.map((name) =>\n                      types.objectProperty(\n                        types.identifier(name),\n                        types.identifier(name),\n                      )\n                    ),\n                  ),\n                ],\n              ),\n            ]),\n          );\n\n          path.skip();\n        }\n      }\n    },\n  };\n}\n\nexport default function () {\n  return {\n    visitor: {\n      Program(path, state) {\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const code: string | undefined = (state as any).file?.code;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const options: Options | undefined = (state as any).opts;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const id: string | undefined = (state as any).filename;\n\n        const filter = createFilterForId(options);\n        if (id && filter && !filter?.(id)) {\n          return;\n        }\n\n        if (\n          !options?.forceTgpuAlias &&\n          code &&\n          !codeFilterRegexes.some((reg) => reg.test(code))\n        ) {\n          return;\n        }\n\n        const ctx: Context = {\n          tgpuAliases: new Set<string>(\n            options?.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n          ),\n          fileId: id,\n        };\n\n        path.traverse(functionVisitor(ctx));\n      },\n    } satisfies TraverseOptions,\n  };\n}\n","import type * as babel from '@babel/types';\nimport type * as acorn from 'acorn';\nimport type { FilterPattern } from 'unplugin';\n\nexport type Context = {\n  /**\n   * How the `tgpu` object is used in code. Since it can be aliased, we\n   * need to catch that and act accordingly.\n   */\n  tgpuAliases: Set<string>;\n  fileId?: string | undefined;\n};\n\nexport interface Options {\n  include?: FilterPattern;\n  exclude?: FilterPattern;\n  enforce?: 'post' | 'pre' | undefined;\n  forceTgpuAlias?: string;\n}\n\nexport const defaultOptions = {\n  include: [/\\.m?[jt]sx?$/],\n};\n\nexport function embedJSON(jsValue: unknown) {\n  return JSON.stringify(jsValue)\n    .replace(/\\u2028/g, '\\\\u2028')\n    .replace(/\\u2029/g, '\\\\u2029');\n}\n\n/**\n * Checks if `node` is an alias for the 'tgpu' object, traditionally\n * available via `import tgpu from 'typegpu'`.\n */\nfunction isTgpu(ctx: Context, node: babel.Node | acorn.AnyNode): boolean {\n  let path = '';\n\n  let tail = node;\n  while (true) {\n    if (tail.type === 'MemberExpression') {\n      if (tail.property.type !== 'Identifier') {\n        // Not handling computed expressions.\n        break;\n      }\n\n      path = path ? `${tail.property.name}.${path}` : tail.property.name;\n      tail = tail.object;\n    } else if (tail.type === 'Identifier') {\n      path = path ? `${tail.name}.${path}` : tail.name;\n      break;\n    } else {\n      break;\n    }\n  }\n\n  return ctx.tgpuAliases.has(path);\n}\n\nconst typegpuImportRegex = /import.*from\\s*['\"]typegpu.*['\"]/;\nconst typegpuDynamicImportRegex = /import\\s*\\(\\s*['\"]\\s*typegpu.*['\"]/;\nconst typegpuRequireRegex = /require\\s*\\(\\s*['\"]\\s*typegpu.*['\"]\\s*\\)/;\n\n/**\n * Regexes used to efficiently determine if a file is\n * meant to be processed by our plugin. We assume every file\n * that should be processed imports `typegpu` in some way.\n */\nexport const codeFilterRegexes = [\n  typegpuImportRegex,\n  typegpuDynamicImportRegex,\n  typegpuRequireRegex,\n];\n\nexport function gatherTgpuAliases(\n  node: acorn.ImportDeclaration | babel.ImportDeclaration,\n  ctx: Context,\n) {\n  if (node.source.value === 'typegpu') {\n    for (const spec of node.specifiers) {\n      if (\n        // The default export of 'typegpu' is the `tgpu` object.\n        spec.type === 'ImportDefaultSpecifier' ||\n        // Aliasing 'tgpu' while importing, e.g. import { tgpu as t } from 'typegpu';\n        (spec.type === 'ImportSpecifier' &&\n          spec.imported.type === 'Identifier' &&\n          spec.imported.name === 'tgpu')\n      ) {\n        ctx.tgpuAliases.add(spec.local.name);\n      } else if (spec.type === 'ImportNamespaceSpecifier') {\n        // Importing everything, e.g. import * as t from 'typegpu';\n        ctx.tgpuAliases.add(`${spec.local.name}.tgpu`);\n      }\n    }\n  }\n}\n\nconst fnShellFunctionNames = ['fn', 'vertexFn', 'fragmentFn', 'computeFn'];\n\nexport function isShellImplementationCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    (node.callee.type === 'CallExpression' &&\n      node.callee.callee.type === 'MemberExpression' &&\n      node.callee.callee.property.type === 'Identifier' &&\n      fnShellFunctionNames.includes(node.callee.callee.property.name) &&\n      node.arguments.length === 1 &&\n      (node.callee.callee.object.type === 'MemberExpression'\n        ? isTgpu(ctx, node.callee.callee.object.object)\n        : isTgpu(ctx, node.callee.callee.object))) || // TODO: remove along with the deprecated 'does' method\n    (node.callee.type === 'MemberExpression' &&\n      node.arguments.length === 1 &&\n      node.callee.property.type === 'Identifier' &&\n      // Assuming that every call to `.does` is related to TypeGPU\n      // because shells can be created separately from calls to `tgpu`,\n      // making it hard to detect.\n      node.callee.property.name === 'does')\n  );\n}\n\nexport const kernelDirectives = ['kernel', 'kernel & js'] as const;\nexport type KernelDirective = (typeof kernelDirectives)[number];\n"]}

package/{chunk-EX5XCHMD.js → chunk-CTUSYGR2.js}

@@ -1,2 +1,2 @@
-import{a as x,b as v,c as b,d as F,e as w,f as h,g,h as D}from"./chunk-Y4WARDXR.js";import N from"defu";import{walk as I}from"estree-walker";import{MagicStringAST as S,generateTransform as P}from"magic-string-ast";import{transpileFn as K}from"tinyest-for-wgsl";import{createUnplugin as J}from"unplugin";function k(c){if(c.body.type==="BlockStatement"){for(let n of c.body.body)if(n.type==="ExpressionStatement"&&g.includes(n.directive))return n.directive}}function A(c){let n=structuredClone(c);return n.body.type==="BlockStatement"&&(n.body.body=n.body.body.filter(s=>!(s.type==="ExpressionStatement"&&s.directive&&g.includes(s.directive)))),n}var r=J(c=>{let n=N(c,v);return{name:"unplugin-typegpu",enforce:n.enforce,transform:{filter:x({id:n},n.forceTgpuAlias?{}:{code:F}),handler(s,l){let d={tgpuAliases:new Set(n.forceTgpuAlias?[n.forceTgpuAlias]:[]),fileId:l},$=this.parse(s,{allowReturnOutsideFunction:!0}),u=[],a=new S(s);I($,{enter(i,t,m,y){var p;let e=i;if(e.type==="ImportDeclaration"&&w(e,d),e.type==="CallExpression"&&h(e,d)){let o=e.arguments[0];if(o&&(o.type==="FunctionExpression"||o.type==="ArrowFunctionExpression")){let E=k(o);u.push({def:A(o),removeJsImplementation:E!=="kernel & js"}),this.skip()}}if(e.type==="ArrowFunctionExpression"||e.type==="FunctionExpression"||e.type==="FunctionDeclaration"){let o=k(e);o&&(u.push({def:A(e),name:e.type==="FunctionDeclaration"||e.type==="FunctionExpression"?(p=e.id)==null?void 0:p.name:(t==null?void 0:t.type)==="VariableDeclarator"&&t.id.type==="Identifier"?t.id.name:void 0,removeJsImplementation:o!=="kernel & js"}),this.skip())}}});let f=d.tgpuAliases.values().next().value;if(f===void 0&&u.length>0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${l}`);for(let{def:i,name:t,removeJsImplementation:m}of u){let{argNames:y,body:e,externalNames:p}=K(i),o=i.type==="FunctionDeclaration";if(o&&t&&s.slice(0,i.start).search(new RegExp(`(?<![\\w_.])${t}(?![\\w_])`))!==-1)throw new Error(`File ${l}: function "${t}", containing ${m?"kernel":"kernel & js"} directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);a.appendLeft(i.start,`${o&&t?`const ${t} = `:""}${f}.__assignAst(`),a.appendRight(i.end,`, ${b({argNames:y,body:e,externalNames:p})}`),p.length>0?a.appendRight(i.end,`, {${p.join(", ")}})`):a.appendRight(i.end,`)${o&&t?";":""}`),m&&a.overwriteNode(i,`${f}.__removedJsImpl(${t?`"${t}"`:""})`)}return P(a,l)}}}}),L=r,M=r.vite,V=r.rollup,q=r.rolldown,z=r.webpack,G=r.rspack,H=r.esbuild,Q=r.farm,W=D;export{L as a,M as b,V as c,q as d,z as e,G as f,H as g,Q as h,W as i};
-//# sourceMappingURL=chunk-EX5XCHMD.js.map
+import{a as x,b as v,c as b,d as F,e as w,f as h,g,h as D}from"./chunk-4ONRHQSA.js";import I from"defu";import{walk as N}from"estree-walker";import{generateTransform as S,MagicStringAST as P}from"magic-string-ast";import{transpileFn as K}from"tinyest-for-wgsl";import{createUnplugin as J}from"unplugin";function k(c){if(c.body.type==="BlockStatement"){for(let n of c.body.body)if(n.type==="ExpressionStatement"&&g.includes(n.directive))return n.directive}}function A(c){let n=structuredClone(c);return n.body.type==="BlockStatement"&&(n.body.body=n.body.body.filter(s=>!(s.type==="ExpressionStatement"&&s.directive&&g.includes(s.directive)))),n}var r=J(c=>{let n=I(c,v);return{name:"unplugin-typegpu",enforce:n.enforce,transform:{filter:x({id:n},n.forceTgpuAlias?{}:{code:F}),handler(s,l){let d={tgpuAliases:new Set(n.forceTgpuAlias?[n.forceTgpuAlias]:[]),fileId:l},$=this.parse(s,{allowReturnOutsideFunction:!0}),u=[],p=new P(s);N($,{enter(i,t,m,y){var a;let e=i;if(e.type==="ImportDeclaration"&&w(e,d),e.type==="CallExpression"&&h(e,d)){let o=e.arguments[0];if(o&&(o.type==="FunctionExpression"||o.type==="ArrowFunctionExpression")){let E=k(o);u.push({def:A(o),removeJsImplementation:E!=="kernel & js"}),this.skip()}}if(e.type==="ArrowFunctionExpression"||e.type==="FunctionExpression"||e.type==="FunctionDeclaration"){let o=k(e);o&&(u.push({def:A(e),name:e.type==="FunctionDeclaration"||e.type==="FunctionExpression"?(a=e.id)==null?void 0:a.name:(t==null?void 0:t.type)==="VariableDeclarator"&&t.id.type==="Identifier"?t.id.name:void 0,removeJsImplementation:o!=="kernel & js"}),this.skip())}}});let f=d.tgpuAliases.values().next().value;if(f===void 0&&u.length>0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${l}`);for(let{def:i,name:t,removeJsImplementation:m}of u){let{params:y,body:e,externalNames:a}=K(i),o=i.type==="FunctionDeclaration";if(o&&t&&s.slice(0,i.start).search(new RegExp(`(?<![\\w_.])${t}(?![\\w_])`))!==-1)throw new Error(`File ${l}: function "${t}", containing ${m?"kernel":"kernel & js"} directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);p.appendLeft(i.start,`${o&&t?`const ${t} = `:""}${f}.__assignAst(`),p.appendRight(i.end,`, ${b({params:y,body:e,externalNames:a})}`),a.length>0?p.appendRight(i.end,`, {${a.join(", ")}})`):p.appendRight(i.end,`)${o&&t?";":""}`),m&&p.overwriteNode(i,`${f}.__removedJsImpl(${t?`"${t}"`:""})`)}return S(p,l)}}}}),L=r,M=r.vite,V=r.rollup,q=r.rolldown,z=r.webpack,G=r.rspack,H=r.esbuild,Q=r.farm,W=D;export{L as a,M as b,V as c,q as d,z as e,G as f,H as g,Q as h,W as i};
+//# sourceMappingURL=chunk-CTUSYGR2.js.map

package/chunk-CTUSYGR2.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import type * as acorn from 'acorn';\nimport defu from 'defu';\nimport { type Node, walk } from 'estree-walker';\nimport { generateTransform, MagicStringAST } from 'magic-string-ast';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { createUnplugin, type UnpluginInstance } from 'unplugin';\nimport babel from './babel.ts';\nimport {\n  codeFilterRegexes,\n  type Context,\n  defaultOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  type KernelDirective,\n  kernelDirectives,\n  type Options,\n} from './common.ts';\n\ntype FunctionNode =\n  | acorn.FunctionDeclaration\n  | acorn.AnonymousFunctionDeclaration\n  | acorn.FunctionExpression\n  | acorn.ArrowFunctionExpression;\n\nfunction getKernelDirective(node: FunctionNode): KernelDirective | undefined {\n  if (node.body.type === 'BlockStatement') {\n    for (const statement of node.body.body) {\n      if (statement.type === 'ExpressionStatement') {\n        if (kernelDirectives.includes(statement.directive as KernelDirective)) {\n          return statement.directive as KernelDirective;\n        }\n      }\n    }\n  }\n}\n\nfunction removeKernelDirective(node: FunctionNode) {\n  const cloned = structuredClone(node);\n\n  if (cloned.body.type === 'BlockStatement') {\n    cloned.body.body = cloned.body.body.filter(\n      (statement) =>\n        !(\n          statement.type === 'ExpressionStatement' &&\n          statement.directive &&\n          kernelDirectives.includes(statement.directive as KernelDirective)\n        ),\n    );\n  }\n\n  return cloned;\n}\n\nconst typegpu: UnpluginInstance<Options, false> = createUnplugin(\n  (rawOptions) => {\n    const options = defu(rawOptions, defaultOptions);\n\n    return {\n      name: 'unplugin-typegpu' as const,\n      enforce: options.enforce,\n      transform: {\n        filter: {\n          id: options,\n          ...(options.forceTgpuAlias ? {} : { code: codeFilterRegexes }),\n        },\n        handler(code, id) {\n          const ctx: Context = {\n            tgpuAliases: new Set<string>(\n              options.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n            ),\n            fileId: id,\n          };\n\n          const ast = this.parse(code, {\n            allowReturnOutsideFunction: true,\n          }) as Node;\n\n          const tgslFunctionDefs: {\n            def: FunctionNode;\n            name?: string | undefined;\n            removeJsImplementation: boolean;\n          }[] = [];\n\n          const magicString = new MagicStringAST(code);\n\n          walk(ast, {\n            enter(_node, _parent, prop, index) {\n              const node = _node as acorn.AnyNode;\n\n              if (node.type === 'ImportDeclaration') {\n                gatherTgpuAliases(node, ctx);\n              }\n\n              if (node.type === 'CallExpression') {\n                if (isShellImplementationCall(node, ctx)) {\n                  const implementation = node.arguments[0];\n\n                  if (\n                    implementation &&\n                    (implementation.type === 'FunctionExpression' ||\n                      implementation.type === 'ArrowFunctionExpression')\n                  ) {\n                    const directive = getKernelDirective(implementation);\n                    tgslFunctionDefs.push({\n                      def: removeKernelDirective(implementation),\n                      removeJsImplementation: directive !== 'kernel & js',\n                    });\n                    this.skip();\n                  }\n                }\n              }\n\n              if (\n                node.type === 'ArrowFunctionExpression' ||\n                node.type === 'FunctionExpression' ||\n                node.type === 'FunctionDeclaration'\n              ) {\n                const directive = getKernelDirective(node);\n                if (directive) {\n                  tgslFunctionDefs.push({\n                    def: removeKernelDirective(node),\n                    name: node.type === 'FunctionDeclaration' ||\n                        node.type === 'FunctionExpression'\n                      ? node.id?.name\n                      : _parent?.type === 'VariableDeclarator'\n                      ? _parent.id.type === 'Identifier'\n                        ? _parent.id.name\n                        : undefined\n                      : undefined,\n                    removeJsImplementation: directive !== 'kernel & js',\n                  });\n                  this.skip();\n                }\n              }\n            },\n          });\n\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n\n          if (tgpuAlias === undefined && tgslFunctionDefs.length > 0) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${id}`,\n            );\n          }\n\n          for (\n            const {\n              def,\n              name,\n              removeJsImplementation,\n            } of tgslFunctionDefs\n          ) {\n            const { params, body, externalNames } = transpileFn(def);\n            const isFunctionStatement = def.type === 'FunctionDeclaration';\n\n            if (\n              isFunctionStatement &&\n              name &&\n              code\n                  .slice(0, def.start)\n                  .search(new RegExp(`(?<![\\\\w_.])${name}(?![\\\\w_])`)) !== -1\n            ) {\n              throw new Error(\n                `File ${id}: function \"${name}\", containing ${\n                  removeJsImplementation ? 'kernel' : 'kernel & js'\n                } directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`,\n              );\n            }\n\n            // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n            magicString.appendLeft(\n              def.start,\n              `${\n                isFunctionStatement && name ? `const ${name} = ` : ''\n              }${tgpuAlias}.__assignAst(`,\n            );\n            magicString.appendRight(\n              def.end,\n              `, ${embedJSON({ params, body, externalNames })}`,\n            );\n\n            if (externalNames.length > 0) {\n              magicString.appendRight(\n                def.end,\n                `, {${externalNames.join(', ')}})`,\n              );\n            } else {\n              magicString.appendRight(\n                def.end,\n                `)${isFunctionStatement && name ? ';' : ''}`,\n              );\n            }\n\n            if (removeJsImplementation) {\n              magicString.overwriteNode(\n                def,\n                `${tgpuAlias}.__removedJsImpl(${name ? `\"${name}\"` : ''})`,\n              );\n            }\n          }\n\n          return generateTransform(magicString, id);\n        },\n      },\n    };\n  },\n);\n\nexport type { Options } from './common.ts';\n\nexport default typegpu;\n\nexport const vitePlugin = typegpu.vite;\nexport const rollupPlugin = typegpu.rollup;\nexport const rolldownPlugin = typegpu.rolldown;\nexport const webpackPlugin = typegpu.webpack;\nexport const rspackPlugin = typegpu.rspack;\nexport const esbuildPlugin = typegpu.esbuild;\nexport const farmPlugin = typegpu.farm;\nexport const babelPlugin = babel;\n"],"mappings":"oFACA,OAAOA,MAAU,OACjB,OAAoB,QAAAC,MAAY,gBAChC,OAAS,qBAAAC,EAAmB,kBAAAC,MAAsB,mBAClD,OAAS,eAAAC,MAAmB,mBAC5B,OAAS,kBAAAC,MAA6C,WAoBtD,SAASC,EAAmBC,EAAiD,CAC3E,GAAIA,EAAK,KAAK,OAAS,kBACrB,QAAWC,KAAaD,EAAK,KAAK,KAChC,GAAIC,EAAU,OAAS,uBACjBC,EAAiB,SAASD,EAAU,SAA4B,EAClE,OAAOA,EAAU,UAK3B,CAEA,SAASE,EAAsBH,EAAoB,CACjD,IAAMI,EAAS,gBAAgBJ,CAAI,EAEnC,OAAII,EAAO,KAAK,OAAS,mBACvBA,EAAO,KAAK,KAAOA,EAAO,KAAK,KAAK,OACjCH,GACC,EACEA,EAAU,OAAS,uBACnBA,EAAU,WACVC,EAAiB,SAASD,EAAU,SAA4B,EAEtE,GAGKG,CACT,CAEA,IAAMC,EAA4CC,EAC/CC,GAAe,CACd,IAAMC,EAAUC,EAAKF,EAAYG,CAAc,EAE/C,MAAO,CACL,KAAM,mBACN,QAASF,EAAQ,QACjB,UAAW,CACT,OAAQG,EAAA,CACN,GAAIH,GACAA,EAAQ,eAAiB,CAAC,EAAI,CAAE,KAAMI,CAAkB,GAE9D,QAAQC,EAAMC,EAAI,CAChB,IAAMC,EAAe,CACnB,YAAa,IAAI,IACfP,EAAQ,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACvD,EACA,OAAQM,CACV,EAEME,EAAM,KAAK,MAAMH,EAAM,CAC3B,2BAA4B,EAC9B,CAAC,EAEKI,EAIA,CAAC,EAEDC,EAAc,IAAIC,EAAeN,CAAI,EAE3CO,EAAKJ,EAAK,CACR,MAAMK,EAAOC,EAASC,EAAMC,EAAO,CAvF/C,IAAAC,EAwFc,IAAMzB,EAAOqB,EAMb,GAJIrB,EAAK,OAAS,qBAChB0B,EAAkB1B,EAAMe,CAAG,EAGzBf,EAAK,OAAS,kBACZ2B,EAA0B3B,EAAMe,CAAG,EAAG,CACxC,IAAMa,EAAiB5B,EAAK,UAAU,CAAC,EAEvC,GACE4B,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,IAAMC,EAAY9B,EAAmB6B,CAAc,EACnDX,EAAiB,KAAK,CACpB,IAAKd,EAAsByB,CAAc,EACzC,uBAAwBC,IAAc,aACxC,CAAC,EACD,KAAK,KAAK,CACZ,CACF,CAGF,GACE7B,EAAK,OAAS,2BACdA,EAAK,OAAS,sBACdA,EAAK,OAAS,sBACd,CACA,IAAM6B,EAAY9B,EAAmBC,CAAI,EACrC6B,IACFZ,EAAiB,KAAK,CACpB,IAAKd,EAAsBH,CAAI,EAC/B,KAAMA,EAAK,OAAS,uBAChBA,EAAK,OAAS,sBACdyB,EAAAzB,EAAK,KAAL,YAAAyB,EAAS,MACTH,GAAA,YAAAA,EAAS,QAAS,sBAClBA,EAAQ,GAAG,OAAS,aAClBA,EAAQ,GAAG,KAEb,OACJ,uBAAwBO,IAAc,aACxC,CAAC,EACD,KAAK,KAAK,EAEd,CACF,CACF,CAAC,EAED,IAAMC,EAAYf,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAElD,GAAIe,IAAc,QAAab,EAAiB,OAAS,EACvD,MAAM,IAAI,MACR,gEAAgEH,CAAE,EACpE,EAGF,OACQ,CACJ,IAAAiB,EACA,KAAAC,EACA,uBAAAC,CACF,IAAKhB,EACL,CACA,GAAM,CAAE,OAAAiB,EAAQ,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYN,CAAG,EACjDO,EAAsBP,EAAI,OAAS,sBAEzC,GACEO,GACAN,GACAnB,EACK,MAAM,EAAGkB,EAAI,KAAK,EAClB,OAAO,IAAI,OAAO,eAAeC,CAAI,YAAY,CAAC,IAAM,GAE7D,MAAM,IAAI,MACR,QAAQlB,CAAE,eAAekB,CAAI,iBAC3BC,EAAyB,SAAW,aACtC,8HACF,EAIFf,EAAY,WACVa,EAAI,MACJ,GACEO,GAAuBN,EAAO,SAASA,CAAI,MAAQ,EACrD,GAAGF,CAAS,eACd,EACAZ,EAAY,YACVa,EAAI,IACJ,KAAKQ,EAAU,CAAE,OAAAL,EAAQ,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,EACjD,EAEIA,EAAc,OAAS,EACzBlB,EAAY,YACVa,EAAI,IACJ,MAAMK,EAAc,KAAK,IAAI,CAAC,IAChC,EAEAlB,EAAY,YACVa,EAAI,IACJ,IAAIO,GAAuBN,EAAO,IAAM,EAAE,EAC5C,EAGEC,GACFf,EAAY,cACVa,EACA,GAAGD,CAAS,oBAAoBE,EAAO,IAAIA,CAAI,IAAM,EAAE,GACzD,CAEJ,CAEA,OAAOQ,EAAkBtB,EAAaJ,CAAE,CAC1C,CACF,CACF,CACF,CACF,EAIO2B,EAAQpC,EAEFqC,EAAarC,EAAQ,KACrBsC,EAAetC,EAAQ,OACvBuC,EAAiBvC,EAAQ,SACzBwC,EAAgBxC,EAAQ,QACxByC,EAAezC,EAAQ,OACvB0C,EAAgB1C,EAAQ,QACxB2C,EAAa3C,EAAQ,KACrB4C,EAAcC","names":["defu","walk","generateTransform","MagicStringAST","transpileFn","createUnplugin","getKernelDirective","node","statement","kernelDirectives","removeKernelDirective","cloned","typegpu","createUnplugin","rawOptions","options","defu","defaultOptions","__spreadValues","codeFilterRegexes","code","id","ctx","ast","tgslFunctionDefs","magicString","MagicStringAST","walk","_node","_parent","prop","index","_a","gatherTgpuAliases","isShellImplementationCall","implementation","directive","tgpuAlias","def","name","removeJsImplementation","params","body","externalNames","transpileFn","isFunctionStatement","embedJSON","generateTransform","index_default","vitePlugin","rollupPlugin","rolldownPlugin","webpackPlugin","rspackPlugin","esbuildPlugin","farmPlugin","babelPlugin","babel_default"]}

package/chunk-EPZPFIYG.cjs

@@ -0,0 +1,2 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var _chunkBCOJLF4Tcjs = require('./chunk-BCOJLF4T.cjs');var _defu = require('defu'); var _defu2 = _interopRequireDefault(_defu);var _estreewalker = require('estree-walker');var _magicstringast = require('magic-string-ast');var _tinyestforwgsl = require('tinyest-for-wgsl');var _unplugin = require('unplugin');function k(c){if(c.body.type==="BlockStatement"){for(let n of c.body.body)if(n.type==="ExpressionStatement"&&_chunkBCOJLF4Tcjs.g.includes(n.directive))return n.directive}}function A(c){let n=structuredClone(c);return n.body.type==="BlockStatement"&&(n.body.body=n.body.body.filter(s=>!(s.type==="ExpressionStatement"&&s.directive&&_chunkBCOJLF4Tcjs.g.includes(s.directive)))),n}var r=_unplugin.createUnplugin.call(void 0, c=>{let n=_defu2.default.call(void 0, c,_chunkBCOJLF4Tcjs.b);return{name:"unplugin-typegpu",enforce:n.enforce,transform:{filter:_chunkBCOJLF4Tcjs.a.call(void 0, {id:n},n.forceTgpuAlias?{}:{code:_chunkBCOJLF4Tcjs.d}),handler(s,l){let d={tgpuAliases:new Set(n.forceTgpuAlias?[n.forceTgpuAlias]:[]),fileId:l},$=this.parse(s,{allowReturnOutsideFunction:!0}),u=[],p=new (0, _magicstringast.MagicStringAST)(s);_estreewalker.walk.call(void 0, $,{enter(i,t,m,y){var a;let e=i;if(e.type==="ImportDeclaration"&&_chunkBCOJLF4Tcjs.e.call(void 0, e,d),e.type==="CallExpression"&&_chunkBCOJLF4Tcjs.f.call(void 0, e,d)){let o=e.arguments[0];if(o&&(o.type==="FunctionExpression"||o.type==="ArrowFunctionExpression")){let E=k(o);u.push({def:A(o),removeJsImplementation:E!=="kernel & js"}),this.skip()}}if(e.type==="ArrowFunctionExpression"||e.type==="FunctionExpression"||e.type==="FunctionDeclaration"){let o=k(e);o&&(u.push({def:A(e),name:e.type==="FunctionDeclaration"||e.type==="FunctionExpression"?(a=e.id)==null?void 0:a.name:(t==null?void 0:t.type)==="VariableDeclarator"&&t.id.type==="Identifier"?t.id.name:void 0,removeJsImplementation:o!=="kernel & js"}),this.skip())}}});let f=d.tgpuAliases.values().next().value;if(f===void 0&&u.length>0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${l}`);for(let{def:i,name:t,removeJsImplementation:m}of u){let{params:y,body:e,externalNames:a}=_tinyestforwgsl.transpileFn.call(void 0, i),o=i.type==="FunctionDeclaration";if(o&&t&&s.slice(0,i.start).search(new RegExp(`(?<![\\w_.])${t}(?![\\w_])`))!==-1)throw new Error(`File ${l}: function "${t}", containing ${m?"kernel":"kernel & js"} directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);p.appendLeft(i.start,`${o&&t?`const ${t} = `:""}${f}.__assignAst(`),p.appendRight(i.end,`, ${_chunkBCOJLF4Tcjs.c.call(void 0, {params:y,body:e,externalNames:a})}`),a.length>0?p.appendRight(i.end,`, {${a.join(", ")}})`):p.appendRight(i.end,`)${o&&t?";":""}`),m&&p.overwriteNode(i,`${f}.__removedJsImpl(${t?`"${t}"`:""})`)}return _magicstringast.generateTransform.call(void 0, p,l)}}}}),L= exports.a =r,M= exports.b =r.vite,V= exports.c =r.rollup,q= exports.d =r.rolldown,z= exports.e =r.webpack,G= exports.f =r.rspack,H= exports.g =r.esbuild,Q= exports.h =r.farm,W= exports.i =_chunkBCOJLF4Tcjs.h;exports.a = L; exports.b = M; exports.c = V; exports.d = q; exports.e = z; exports.f = G; exports.g = H; exports.h = Q; exports.i = W;
+//# sourceMappingURL=chunk-EPZPFIYG.cjs.map

package/chunk-EPZPFIYG.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-EPZPFIYG.cjs","../src/index.ts"],"names":["getKernelDirective","node","statement","kernelDirectives","removeKernelDirective","cloned","typegpu","createUnplugin","rawOptions","options","defu","defaultOptions","__spreadValues","codeFilterRegexes","code","id","ctx","ast","tgslFunctionDefs","magicString","MagicStringAST","walk","_node","_parent","prop","index","_a","gatherTgpuAliases","isShellImplementationCall","implementation","directive","tgpuAlias"],"mappings":"AAAA,iOAAoF,wECCnE,6CACe,kDACkB,kDACtB,oCAC0B,SAoB7CA,CAAAA,CAAmBC,CAAAA,CAAiD,CAC3E,EAAA,CAAIA,CAAAA,CAAK,IAAA,CAAK,IAAA,GAAS,gBAAA,CAAA,CACrB,GAAA,CAAA,IAAWC,EAAAA,GAAaD,CAAAA,CAAK,IAAA,CAAK,IAAA,CAChC,EAAA,CAAIC,CAAAA,CAAU,IAAA,GAAS,qBAAA,EACjBC,mBAAAA,CAAiB,QAAA,CAASD,CAAAA,CAAU,SAA4B,CAAA,CAClE,OAAOA,CAAAA,CAAU,SAAA,CAK3B,CAEA,SAASE,CAAAA,CAAsBH,CAAAA,CAAoB,CACjD,IAAMI,CAAAA,CAAS,eAAA,CAAgBJ,CAAI,CAAA,CAEnC,OAAII,CAAAA,CAAO,IAAA,CAAK,IAAA,GAAS,gBAAA,EAAA,CACvBA,CAAAA,CAAO,IAAA,CAAK,IAAA,CAAOA,CAAAA,CAAO,IAAA,CAAK,IAAA,CAAK,MAAA,CACjCH,CAAAA,EACC,CAAA,CACEA,CAAAA,CAAU,IAAA,GAAS,qBAAA,EACnBA,CAAAA,CAAU,SAAA,EACVC,mBAAAA,CAAiB,QAAA,CAASD,CAAAA,CAAU,SAA4B,CAAA,CAEtE,CAAA,CAAA,CAGKG,CACT,CAEA,IAAMC,CAAAA,CAA4CC,sCAAAA,CAC/CC,EAAe,CACd,IAAMC,CAAAA,CAAUC,4BAAAA,CAAKF,CAAYG,mBAAc,CAAA,CAE/C,MAAO,CACL,IAAA,CAAM,kBAAA,CACN,OAAA,CAASF,CAAAA,CAAQ,OAAA,CACjB,SAAA,CAAW,CACT,MAAA,CAAQG,iCAAAA,CACN,EAAA,CAAIH,CAAAA,CAAAA,CACAA,CAAAA,CAAQ,cAAA,CAAiB,CAAC,CAAA,CAAI,CAAE,IAAA,CAAMI,mBAAkB,CAAA,CAAA,CAE9D,OAAA,CAAQC,CAAAA,CAAMC,CAAAA,CAAI,CAChB,IAAMC,CAAAA,CAAe,CACnB,WAAA,CAAa,IAAI,GAAA,CACfP,CAAAA,CAAQ,cAAA,CAAiB,CAACA,CAAAA,CAAQ,cAAc,CAAA,CAAI,CAAC,CACvD,CAAA,CACA,MAAA,CAAQM,CACV,CAAA,CAEME,CAAAA,CAAM,IAAA,CAAK,KAAA,CAAMH,CAAAA,CAAM,CAC3B,0BAAA,CAA4B,CAAA,CAC9B,CAAC,CAAA,CAEKI,CAAAA,CAIA,CAAC,CAAA,CAEDC,CAAAA,CAAc,IAAIC,mCAAAA,CAAeN,CAAI,CAAA,CAE3CO,gCAAAA,CAAKJ,CAAK,CACR,KAAA,CAAMK,CAAAA,CAAOC,CAAAA,CAASC,CAAAA,CAAMC,CAAAA,CAAO,CAvF/C,IAAAC,CAAAA,CAwFc,IAAMzB,CAAAA,CAAOqB,CAAAA,CAMb,EAAA,CAJIrB,CAAAA,CAAK,IAAA,GAAS,mBAAA,EAChB0B,iCAAAA,CAAkB1B,CAAMe,CAAG,CAAA,CAGzBf,CAAAA,CAAK,IAAA,GAAS,gBAAA,EACZ2B,iCAAAA,CAA0B3B,CAAMe,CAAG,CAAA,CAAG,CACxC,IAAMa,CAAAA,CAAiB5B,CAAAA,CAAK,SAAA,CAAU,CAAC,CAAA,CAEvC,EAAA,CACE4B,CAAAA,EAAAA,CACCA,CAAAA,CAAe,IAAA,GAAS,oBAAA,EACvBA,CAAAA,CAAe,IAAA,GAAS,yBAAA,CAAA,CAC1B,CACA,IAAMC,CAAAA,CAAY9B,CAAAA,CAAmB6B,CAAc,CAAA,CACnDX,CAAAA,CAAiB,IAAA,CAAK,CACpB,GAAA,CAAKd,CAAAA,CAAsByB,CAAc,CAAA,CACzC,sBAAA,CAAwBC,CAAAA,GAAc,aACxC,CAAC,CAAA,CACD,IAAA,CAAK,IAAA,CAAK,CACZ,CACF,CAGF,EAAA,CACE7B,CAAAA,CAAK,IAAA,GAAS,yBAAA,EACdA,CAAAA,CAAK,IAAA,GAAS,oBAAA,EACdA,CAAAA,CAAK,IAAA,GAAS,qBAAA,CACd,CACA,IAAM6B,CAAAA,CAAY9B,CAAAA,CAAmBC,CAAI,CAAA,CACrC6B,CAAAA,EAAAA,CACFZ,CAAAA,CAAiB,IAAA,CAAK,CACpB,GAAA,CAAKd,CAAAA,CAAsBH,CAAI,CAAA,CAC/B,IAAA,CAAMA,CAAAA,CAAK,IAAA,GAAS,qBAAA,EAChBA,CAAAA,CAAK,IAAA,GAAS,oBAAA,CAAA,CACdyB,CAAAA,CAAAzB,CAAAA,CAAK,EAAA,CAAA,EAAL,IAAA,CAAA,KAAA,CAAA,CAAAyB,CAAAA,CAAS,IAAA,CAAA,CACTH,CAAAA,EAAA,IAAA,CAAA,KAAA,CAAA,CAAAA,CAAAA,CAAS,IAAA,CAAA,GAAS,oBAAA,EAClBA,CAAAA,CAAQ,EAAA,CAAG,IAAA,GAAS,YAAA,CAClBA,CAAAA,CAAQ,EAAA,CAAG,IAAA,CAEb,KAAA,CAAA,CACJ,sBAAA,CAAwBO,CAAAA,GAAc,aACxC,CAAC,CAAA,CACD,IAAA,CAAK,IAAA,CAAK,CAAA,CAEd,CACF,CACF,CAAC,CAAA,CAED,IAAMC,CAAAA,CAAYf,CAAAA,CAAI,WAAA,CAAY,MAAA,CAAO,CAAA,CAAE,IAAA,CAAK,CAAA,CAAE,KAAA,CAElD,EAAA,CAAIe,CAAAA,GAAc,KAAA,CAAA,EAAab,CAAAA,CAAiB,MAAA,CAAS,CAAA,CACvD,MAAM,IAAI,KAAA,CACR,CAAA,6DAAA,EAAgEH,CAAE,CAAA,CAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-EPZPFIYG.cjs","sourcesContent":[null,"import type * as acorn from 'acorn';\nimport defu from 'defu';\nimport { type Node, walk } from 'estree-walker';\nimport { generateTransform, MagicStringAST } from 'magic-string-ast';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { createUnplugin, type UnpluginInstance } from 'unplugin';\nimport babel from './babel.ts';\nimport {\n  codeFilterRegexes,\n  type Context,\n  defaultOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  type KernelDirective,\n  kernelDirectives,\n  type Options,\n} from './common.ts';\n\ntype FunctionNode =\n  | acorn.FunctionDeclaration\n  | acorn.AnonymousFunctionDeclaration\n  | acorn.FunctionExpression\n  | acorn.ArrowFunctionExpression;\n\nfunction getKernelDirective(node: FunctionNode): KernelDirective | undefined {\n  if (node.body.type === 'BlockStatement') {\n    for (const statement of node.body.body) {\n      if (statement.type === 'ExpressionStatement') {\n        if (kernelDirectives.includes(statement.directive as KernelDirective)) {\n          return statement.directive as KernelDirective;\n        }\n      }\n    }\n  }\n}\n\nfunction removeKernelDirective(node: FunctionNode) {\n  const cloned = structuredClone(node);\n\n  if (cloned.body.type === 'BlockStatement') {\n    cloned.body.body = cloned.body.body.filter(\n      (statement) =>\n        !(\n          statement.type === 'ExpressionStatement' &&\n          statement.directive &&\n          kernelDirectives.includes(statement.directive as KernelDirective)\n        ),\n    );\n  }\n\n  return cloned;\n}\n\nconst typegpu: UnpluginInstance<Options, false> = createUnplugin(\n  (rawOptions) => {\n    const options = defu(rawOptions, defaultOptions);\n\n    return {\n      name: 'unplugin-typegpu' as const,\n      enforce: options.enforce,\n      transform: {\n        filter: {\n          id: options,\n          ...(options.forceTgpuAlias ? {} : { code: codeFilterRegexes }),\n        },\n        handler(code, id) {\n          const ctx: Context = {\n            tgpuAliases: new Set<string>(\n              options.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n            ),\n            fileId: id,\n          };\n\n          const ast = this.parse(code, {\n            allowReturnOutsideFunction: true,\n          }) as Node;\n\n          const tgslFunctionDefs: {\n            def: FunctionNode;\n            name?: string | undefined;\n            removeJsImplementation: boolean;\n          }[] = [];\n\n          const magicString = new MagicStringAST(code);\n\n          walk(ast, {\n            enter(_node, _parent, prop, index) {\n              const node = _node as acorn.AnyNode;\n\n              if (node.type === 'ImportDeclaration') {\n                gatherTgpuAliases(node, ctx);\n              }\n\n              if (node.type === 'CallExpression') {\n                if (isShellImplementationCall(node, ctx)) {\n                  const implementation = node.arguments[0];\n\n                  if (\n                    implementation &&\n                    (implementation.type === 'FunctionExpression' ||\n                      implementation.type === 'ArrowFunctionExpression')\n                  ) {\n                    const directive = getKernelDirective(implementation);\n                    tgslFunctionDefs.push({\n                      def: removeKernelDirective(implementation),\n                      removeJsImplementation: directive !== 'kernel & js',\n                    });\n                    this.skip();\n                  }\n                }\n              }\n\n              if (\n                node.type === 'ArrowFunctionExpression' ||\n                node.type === 'FunctionExpression' ||\n                node.type === 'FunctionDeclaration'\n              ) {\n                const directive = getKernelDirective(node);\n                if (directive) {\n                  tgslFunctionDefs.push({\n                    def: removeKernelDirective(node),\n                    name: node.type === 'FunctionDeclaration' ||\n                        node.type === 'FunctionExpression'\n                      ? node.id?.name\n                      : _parent?.type === 'VariableDeclarator'\n                      ? _parent.id.type === 'Identifier'\n                        ? _parent.id.name\n                        : undefined\n                      : undefined,\n                    removeJsImplementation: directive !== 'kernel & js',\n                  });\n                  this.skip();\n                }\n              }\n            },\n          });\n\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n\n          if (tgpuAlias === undefined && tgslFunctionDefs.length > 0) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${id}`,\n            );\n          }\n\n          for (\n            const {\n              def,\n              name,\n              removeJsImplementation,\n            } of tgslFunctionDefs\n          ) {\n            const { params, body, externalNames } = transpileFn(def);\n            const isFunctionStatement = def.type === 'FunctionDeclaration';\n\n            if (\n              isFunctionStatement &&\n              name &&\n              code\n                  .slice(0, def.start)\n                  .search(new RegExp(`(?<![\\\\w_.])${name}(?![\\\\w_])`)) !== -1\n            ) {\n              throw new Error(\n                `File ${id}: function \"${name}\", containing ${\n                  removeJsImplementation ? 'kernel' : 'kernel & js'\n                } directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`,\n              );\n            }\n\n            // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n            magicString.appendLeft(\n              def.start,\n              `${\n                isFunctionStatement && name ? `const ${name} = ` : ''\n              }${tgpuAlias}.__assignAst(`,\n            );\n            magicString.appendRight(\n              def.end,\n              `, ${embedJSON({ params, body, externalNames })}`,\n            );\n\n            if (externalNames.length > 0) {\n              magicString.appendRight(\n                def.end,\n                `, {${externalNames.join(', ')}})`,\n              );\n            } else {\n              magicString.appendRight(\n                def.end,\n                `)${isFunctionStatement && name ? ';' : ''}`,\n              );\n            }\n\n            if (removeJsImplementation) {\n              magicString.overwriteNode(\n                def,\n                `${tgpuAlias}.__removedJsImpl(${name ? `\"${name}\"` : ''})`,\n              );\n            }\n          }\n\n          return generateTransform(magicString, id);\n        },\n      },\n    };\n  },\n);\n\nexport type { Options } from './common.ts';\n\nexport default typegpu;\n\nexport const vitePlugin = typegpu.vite;\nexport const rollupPlugin = typegpu.rollup;\nexport const rolldownPlugin = typegpu.rolldown;\nexport const webpackPlugin = typegpu.webpack;\nexport const rspackPlugin = typegpu.rspack;\nexport const esbuildPlugin = typegpu.esbuild;\nexport const farmPlugin = typegpu.farm;\nexport const babelPlugin = babel;\n"]}

package/esbuild.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');var l=_chunkHPSWWSWVcjs.g;exports.default = l;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');var l=_chunkEPZPFIYGcjs.g;exports.default = l;
 //# sourceMappingURL=esbuild.cjs.map

package/esbuild.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/esbuild.cjs","../src/esbuild.ts"],"names":["esbuild_default","esbuildPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/esbuild.cjs","sourcesContent":[null,"import { esbuildPlugin } from './index.ts';\n\nexport default esbuildPlugin;\n"]}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/esbuild.cjs","../src/esbuild.ts"],"names":["esbuild_default","esbuildPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/esbuild.cjs","sourcesContent":[null,"import { esbuildPlugin } from './index.ts';\n\nexport default esbuildPlugin;\n"]}

package/esbuild.js

@@ -1,2 +1,2 @@
-import{g as e}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";var l=e;export{l as default};
+import{g as e}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";var l=e;export{l as default};
 //# sourceMappingURL=esbuild.js.map

package/farm.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');var m=_chunkHPSWWSWVcjs.h;exports.default = m;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');var m=_chunkEPZPFIYGcjs.h;exports.default = m;
 //# sourceMappingURL=farm.cjs.map

package/farm.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/farm.cjs","../src/farm.ts"],"names":["farm_default","farmPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/farm.cjs","sourcesContent":[null,"import { farmPlugin } from './index.ts';\n\nexport default farmPlugin;\n"]}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/farm.cjs","../src/farm.ts"],"names":["farm_default","farmPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/farm.cjs","sourcesContent":[null,"import { farmPlugin } from './index.ts';\n\nexport default farmPlugin;\n"]}

package/farm.js

@@ -1,2 +1,2 @@
-import{h as r}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";var m=r;export{m as default};
+import{h as r}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";var m=r;export{m as default};
 //# sourceMappingURL=farm.js.map

package/index.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');exports.babelPlugin = _chunkHPSWWSWVcjs.i; exports.default = _chunkHPSWWSWVcjs.a; exports.esbuildPlugin = _chunkHPSWWSWVcjs.g; exports.farmPlugin = _chunkHPSWWSWVcjs.h; exports.rolldownPlugin = _chunkHPSWWSWVcjs.d; exports.rollupPlugin = _chunkHPSWWSWVcjs.c; exports.rspackPlugin = _chunkHPSWWSWVcjs.f; exports.vitePlugin = _chunkHPSWWSWVcjs.b; exports.webpackPlugin = _chunkHPSWWSWVcjs.e;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');exports.babelPlugin = _chunkEPZPFIYGcjs.i; exports.default = _chunkEPZPFIYGcjs.a; exports.esbuildPlugin = _chunkEPZPFIYGcjs.g; exports.farmPlugin = _chunkEPZPFIYGcjs.h; exports.rolldownPlugin = _chunkEPZPFIYGcjs.d; exports.rollupPlugin = _chunkEPZPFIYGcjs.c; exports.rspackPlugin = _chunkEPZPFIYGcjs.f; exports.vitePlugin = _chunkEPZPFIYGcjs.b; exports.webpackPlugin = _chunkEPZPFIYGcjs.e;
 //# sourceMappingURL=index.cjs.map

package/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/index.cjs"],"names":[],"mappings":"AAAA,iIAAmD,gCAA6B,qYAAoK","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/index.cjs"}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/index.cjs"],"names":[],"mappings":"AAAA,iIAAmD,gCAA6B,qYAAoK","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/index.cjs"}

package/index.js

@@ -1,2 +1,2 @@
-import{a,b,c,d,e,f,g,h,i}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";export{i as babelPlugin,a as default,g as esbuildPlugin,h as farmPlugin,d as rolldownPlugin,c as rollupPlugin,f as rspackPlugin,b as vitePlugin,e as webpackPlugin};
+import{a,b,c,d,e,f,g,h,i}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";export{i as babelPlugin,a as default,g as esbuildPlugin,h as farmPlugin,d as rolldownPlugin,c as rollupPlugin,f as rspackPlugin,b as vitePlugin,e as webpackPlugin};
 //# sourceMappingURL=index.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "unplugin-typegpu",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Build plugins for TypeGPU, enabling seamless JavaScript -> WGSL transpilation and improved debugging.",
   "keywords": [
     "rollup-plugin",
@@ -18,6 +18,7 @@
   "license": "MIT",
   "type": "module",
   "exports": {
+    "./package.json": "./package.json",
     ".": {
       "types": "./index.d.ts",
       "module": "./index.js",
@@ -88,11 +89,11 @@
     "magic-string-ast": "^0.9.1",
     "picomatch": "^4.0.2",
     "defu": "^6.1.4",
-    "tinyest-for-wgsl": "~0.1.0",
+    "tinyest-for-wgsl": "~0.1.2",
     "unplugin": "^2.3.1"
   },
   "peerDependencies": {
-    "typegpu": "^0.5.6"
+    "typegpu": "^0.5.8"
   },
   "main": "./index.js",
   "types": "./index.d.ts",

package/rolldown.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');var r=_chunkHPSWWSWVcjs.d;exports.default = r;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');var r=_chunkEPZPFIYGcjs.d;exports.default = r;
 //# sourceMappingURL=rolldown.cjs.map

package/rolldown.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/rolldown.cjs","../src/rolldown.ts"],"names":["rolldown_default","rolldownPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/rolldown.cjs","sourcesContent":[null,"import { rolldownPlugin } from './index.ts';\n\nexport default rolldownPlugin;\n"]}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/rolldown.cjs","../src/rolldown.ts"],"names":["rolldown_default","rolldownPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/rolldown.cjs","sourcesContent":[null,"import { rolldownPlugin } from './index.ts';\n\nexport default rolldownPlugin;\n"]}

package/rolldown.js

@@ -1,2 +1,2 @@
-import{d as o}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";var r=o;export{r as default};
+import{d as o}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";var r=o;export{r as default};
 //# sourceMappingURL=rolldown.js.map

package/rollup.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');var r=_chunkHPSWWSWVcjs.c;exports.default = r;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');var r=_chunkEPZPFIYGcjs.c;exports.default = r;
 //# sourceMappingURL=rollup.cjs.map

package/rollup.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/rollup.cjs","../src/rollup.ts"],"names":["rollup_default","rollupPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/rollup.cjs","sourcesContent":[null,"import { rollupPlugin } from './index.ts';\n\nexport default rollupPlugin;\n"]}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/rollup.cjs","../src/rollup.ts"],"names":["rollup_default","rollupPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/rollup.cjs","sourcesContent":[null,"import { rollupPlugin } from './index.ts';\n\nexport default rollupPlugin;\n"]}

package/rollup.js

@@ -1,2 +1,2 @@
-import{c as l}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";var r=l;export{r as default};
+import{c as l}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";var r=l;export{r as default};
 //# sourceMappingURL=rollup.js.map

package/rspack.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');var p=_chunkHPSWWSWVcjs.f;exports.default = p;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');var p=_chunkEPZPFIYGcjs.f;exports.default = p;
 //# sourceMappingURL=rspack.cjs.map

package/rspack.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/rspack.cjs","../src/rspack.ts"],"names":["rspack_default","rspackPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/rspack.cjs","sourcesContent":[null,"import { rspackPlugin } from './index.ts';\n\nexport default rspackPlugin;\n"]}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/rspack.cjs","../src/rspack.ts"],"names":["rspack_default","rspackPlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/rspack.cjs","sourcesContent":[null,"import { rspackPlugin } from './index.ts';\n\nexport default rspackPlugin;\n"]}

package/rspack.js

@@ -1,2 +1,2 @@
-import{f as r}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";var p=r;export{p as default};
+import{f as r}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";var p=r;export{p as default};
 //# sourceMappingURL=rspack.js.map

package/vite.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');var i=_chunkHPSWWSWVcjs.b;exports.default = i;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');var i=_chunkEPZPFIYGcjs.b;exports.default = i;
 //# sourceMappingURL=vite.cjs.map

package/vite.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/vite.cjs","../src/vite.ts"],"names":["vite_default","vitePlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/vite.cjs","sourcesContent":[null,"import { vitePlugin } from './index.ts';\n\nexport default vitePlugin;\n"]}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/vite.cjs","../src/vite.ts"],"names":["vite_default","vitePlugin"],"mappings":"AAAA,iIAAwC,gCAA6B,ICE9DA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/vite.cjs","sourcesContent":[null,"import { vitePlugin } from './index.ts';\n\nexport default vitePlugin;\n"]}

package/vite.js

@@ -1,2 +1,2 @@
-import{b as t}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";var i=t;export{i as default};
+import{b as t}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";var i=t;export{i as default};
 //# sourceMappingURL=vite.js.map

package/webpack.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkHPSWWSWVcjs = require('./chunk-HPSWWSWV.cjs');require('./chunk-HQFSG5ZM.cjs');var p=_chunkHPSWWSWVcjs.e;exports.default = p;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEPZPFIYGcjs = require('./chunk-EPZPFIYG.cjs');require('./chunk-BCOJLF4T.cjs');var p=_chunkEPZPFIYGcjs.e;exports.default = p;
 //# sourceMappingURL=webpack.cjs.map

package/webpack.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/webpack.cjs","../src/webpack.ts"],"names":["webpack_default","webpackPlugin"],"mappings":"AAAA,iIAAmC,gCAA6B,ICEzDA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/webpack.cjs","sourcesContent":[null,"import { webpackPlugin } from './index.ts';\n\nexport default webpackPlugin;\n"]}
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/webpack.cjs","../src/webpack.ts"],"names":["webpack_default","webpackPlugin"],"mappings":"AAAA,iIAAmC,gCAA6B,ICEzDA,CAAAA,CAAQC,mBAAAA,CAAAA,oBAAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/webpack.cjs","sourcesContent":[null,"import { webpackPlugin } from './index.ts';\n\nexport default webpackPlugin;\n"]}

package/webpack.js

@@ -1,2 +1,2 @@
-import{e}from"./chunk-EX5XCHMD.js";import"./chunk-Y4WARDXR.js";var p=e;export{p as default};
+import{e}from"./chunk-CTUSYGR2.js";import"./chunk-4ONRHQSA.js";var p=e;export{p as default};
 //# sourceMappingURL=webpack.js.map

package/chunk-EX5XCHMD.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import type * as acorn from 'acorn';\nimport defu from 'defu';\nimport { type Node, walk } from 'estree-walker';\nimport { MagicStringAST, generateTransform } from 'magic-string-ast';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { type UnpluginInstance, createUnplugin } from 'unplugin';\nimport babel from './babel.ts';\nimport {\n  type Context,\n  type KernelDirective,\n  type Options,\n  codeFilterRegexes,\n  defaultOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  kernelDirectives,\n} from './common.ts';\n\ntype FunctionNode =\n  | acorn.FunctionDeclaration\n  | acorn.AnonymousFunctionDeclaration\n  | acorn.FunctionExpression\n  | acorn.ArrowFunctionExpression;\n\nfunction getKernelDirective(node: FunctionNode): KernelDirective | undefined {\n  if (node.body.type === 'BlockStatement') {\n    for (const statement of node.body.body) {\n      if (statement.type === 'ExpressionStatement') {\n        if (kernelDirectives.includes(statement.directive as KernelDirective)) {\n          return statement.directive as KernelDirective;\n        }\n      }\n    }\n  }\n}\n\nfunction removeKernelDirective(node: FunctionNode) {\n  const cloned = structuredClone(node);\n\n  if (cloned.body.type === 'BlockStatement') {\n    cloned.body.body = cloned.body.body.filter(\n      (statement) =>\n        !(\n          statement.type === 'ExpressionStatement' &&\n          statement.directive &&\n          kernelDirectives.includes(statement.directive as KernelDirective)\n        ),\n    );\n  }\n\n  return cloned;\n}\n\nconst typegpu: UnpluginInstance<Options, false> = createUnplugin(\n  (rawOptions) => {\n    const options = defu(rawOptions, defaultOptions);\n\n    return {\n      name: 'unplugin-typegpu' as const,\n      enforce: options.enforce,\n      transform: {\n        filter: {\n          id: options,\n          ...(options.forceTgpuAlias ? {} : { code: codeFilterRegexes }),\n        },\n        handler(code, id) {\n          const ctx: Context = {\n            tgpuAliases: new Set<string>(\n              options.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n            ),\n            fileId: id,\n          };\n\n          const ast = this.parse(code, {\n            allowReturnOutsideFunction: true,\n          }) as Node;\n\n          const tgslFunctionDefs: {\n            def: FunctionNode;\n            name?: string | undefined;\n            removeJsImplementation: boolean;\n          }[] = [];\n\n          const magicString = new MagicStringAST(code);\n\n          walk(ast, {\n            enter(_node, _parent, prop, index) {\n              const node = _node as acorn.AnyNode;\n\n              if (node.type === 'ImportDeclaration') {\n                gatherTgpuAliases(node, ctx);\n              }\n\n              if (node.type === 'CallExpression') {\n                if (isShellImplementationCall(node, ctx)) {\n                  const implementation = node.arguments[0];\n\n                  if (\n                    implementation &&\n                    (implementation.type === 'FunctionExpression' ||\n                      implementation.type === 'ArrowFunctionExpression')\n                  ) {\n                    const directive = getKernelDirective(implementation);\n                    tgslFunctionDefs.push({\n                      def: removeKernelDirective(implementation),\n                      removeJsImplementation: directive !== 'kernel & js',\n                    });\n                    this.skip();\n                  }\n                }\n              }\n\n              if (\n                node.type === 'ArrowFunctionExpression' ||\n                node.type === 'FunctionExpression' ||\n                node.type === 'FunctionDeclaration'\n              ) {\n                const directive = getKernelDirective(node);\n                if (directive) {\n                  tgslFunctionDefs.push({\n                    def: removeKernelDirective(node),\n                    name:\n                      node.type === 'FunctionDeclaration' ||\n                      node.type === 'FunctionExpression'\n                        ? node.id?.name\n                        : _parent?.type === 'VariableDeclarator'\n                          ? _parent.id.type === 'Identifier'\n                            ? _parent.id.name\n                            : undefined\n                          : undefined,\n                    removeJsImplementation: directive !== 'kernel & js',\n                  });\n                  this.skip();\n                }\n              }\n            },\n          });\n\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n\n          if (tgpuAlias === undefined && tgslFunctionDefs.length > 0) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${id}`,\n            );\n          }\n\n          for (const {\n            def,\n            name,\n            removeJsImplementation,\n          } of tgslFunctionDefs) {\n            const { argNames, body, externalNames } = transpileFn(def);\n            const isFunctionStatement = def.type === 'FunctionDeclaration';\n\n            if (\n              isFunctionStatement &&\n              name &&\n              code\n                .slice(0, def.start)\n                .search(new RegExp(`(?<![\\\\w_.])${name}(?![\\\\w_])`)) !== -1\n            ) {\n              throw new Error(\n                `File ${id}: function \"${name}\", containing ${removeJsImplementation ? 'kernel' : 'kernel & js'} directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`,\n              );\n            }\n\n            // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n            magicString.appendLeft(\n              def.start,\n              `${isFunctionStatement && name ? `const ${name} = ` : ''}${tgpuAlias}.__assignAst(`,\n            );\n            magicString.appendRight(\n              def.end,\n              `, ${embedJSON({ argNames, body, externalNames })}`,\n            );\n\n            if (externalNames.length > 0) {\n              magicString.appendRight(\n                def.end,\n                `, {${externalNames.join(', ')}})`,\n              );\n            } else {\n              magicString.appendRight(\n                def.end,\n                `)${isFunctionStatement && name ? ';' : ''}`,\n              );\n            }\n\n            if (removeJsImplementation) {\n              magicString.overwriteNode(\n                def,\n                `${tgpuAlias}.__removedJsImpl(${name ? `\"${name}\"` : ''})`,\n              );\n            }\n          }\n\n          return generateTransform(magicString, id);\n        },\n      },\n    };\n  },\n);\n\nexport type { Options } from './common.ts';\n\nexport default typegpu;\n\nexport const vitePlugin = typegpu.vite;\nexport const rollupPlugin = typegpu.rollup;\nexport const rolldownPlugin = typegpu.rolldown;\nexport const webpackPlugin = typegpu.webpack;\nexport const rspackPlugin = typegpu.rspack;\nexport const esbuildPlugin = typegpu.esbuild;\nexport const farmPlugin = typegpu.farm;\nexport const babelPlugin = babel;\n"],"mappings":"oFACA,OAAOA,MAAU,OACjB,OAAoB,QAAAC,MAAY,gBAChC,OAAS,kBAAAC,EAAgB,qBAAAC,MAAyB,mBAClD,OAAS,eAAAC,MAAmB,mBAC5B,OAAgC,kBAAAC,MAAsB,WAoBtD,SAASC,EAAmBC,EAAiD,CAC3E,GAAIA,EAAK,KAAK,OAAS,kBACrB,QAAWC,KAAaD,EAAK,KAAK,KAChC,GAAIC,EAAU,OAAS,uBACjBC,EAAiB,SAASD,EAAU,SAA4B,EAClE,OAAOA,EAAU,UAK3B,CAEA,SAASE,EAAsBH,EAAoB,CACjD,IAAMI,EAAS,gBAAgBJ,CAAI,EAEnC,OAAII,EAAO,KAAK,OAAS,mBACvBA,EAAO,KAAK,KAAOA,EAAO,KAAK,KAAK,OACjCH,GACC,EACEA,EAAU,OAAS,uBACnBA,EAAU,WACVC,EAAiB,SAASD,EAAU,SAA4B,EAEtE,GAGKG,CACT,CAEA,IAAMC,EAA4CC,EAC/CC,GAAe,CACd,IAAMC,EAAUC,EAAKF,EAAYG,CAAc,EAE/C,MAAO,CACL,KAAM,mBACN,QAASF,EAAQ,QACjB,UAAW,CACT,OAAQG,EAAA,CACN,GAAIH,GACAA,EAAQ,eAAiB,CAAC,EAAI,CAAE,KAAMI,CAAkB,GAE9D,QAAQC,EAAMC,EAAI,CAChB,IAAMC,EAAe,CACnB,YAAa,IAAI,IACfP,EAAQ,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACvD,EACA,OAAQM,CACV,EAEME,EAAM,KAAK,MAAMH,EAAM,CAC3B,2BAA4B,EAC9B,CAAC,EAEKI,EAIA,CAAC,EAEDC,EAAc,IAAIC,EAAeN,CAAI,EAE3CO,EAAKJ,EAAK,CACR,MAAMK,EAAOC,EAASC,EAAMC,EAAO,CAvF/C,IAAAC,EAwFc,IAAMzB,EAAOqB,EAMb,GAJIrB,EAAK,OAAS,qBAChB0B,EAAkB1B,EAAMe,CAAG,EAGzBf,EAAK,OAAS,kBACZ2B,EAA0B3B,EAAMe,CAAG,EAAG,CACxC,IAAMa,EAAiB5B,EAAK,UAAU,CAAC,EAEvC,GACE4B,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,IAAMC,EAAY9B,EAAmB6B,CAAc,EACnDX,EAAiB,KAAK,CACpB,IAAKd,EAAsByB,CAAc,EACzC,uBAAwBC,IAAc,aACxC,CAAC,EACD,KAAK,KAAK,CACZ,CACF,CAGF,GACE7B,EAAK,OAAS,2BACdA,EAAK,OAAS,sBACdA,EAAK,OAAS,sBACd,CACA,IAAM6B,EAAY9B,EAAmBC,CAAI,EACrC6B,IACFZ,EAAiB,KAAK,CACpB,IAAKd,EAAsBH,CAAI,EAC/B,KACEA,EAAK,OAAS,uBACdA,EAAK,OAAS,sBACVyB,EAAAzB,EAAK,KAAL,YAAAyB,EAAS,MACTH,GAAA,YAAAA,EAAS,QAAS,sBAChBA,EAAQ,GAAG,OAAS,aAClBA,EAAQ,GAAG,KAEb,OACR,uBAAwBO,IAAc,aACxC,CAAC,EACD,KAAK,KAAK,EAEd,CACF,CACF,CAAC,EAED,IAAMC,EAAYf,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAElD,GAAIe,IAAc,QAAab,EAAiB,OAAS,EACvD,MAAM,IAAI,MACR,gEAAgEH,CAAE,EACpE,EAGF,OAAW,CACT,IAAAiB,EACA,KAAAC,EACA,uBAAAC,CACF,IAAKhB,EAAkB,CACrB,GAAM,CAAE,SAAAiB,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYN,CAAG,EACnDO,EAAsBP,EAAI,OAAS,sBAEzC,GACEO,GACAN,GACAnB,EACG,MAAM,EAAGkB,EAAI,KAAK,EAClB,OAAO,IAAI,OAAO,eAAeC,CAAI,YAAY,CAAC,IAAM,GAE3D,MAAM,IAAI,MACR,QAAQlB,CAAE,eAAekB,CAAI,iBAAiBC,EAAyB,SAAW,aAAa,8HACjG,EAIFf,EAAY,WACVa,EAAI,MACJ,GAAGO,GAAuBN,EAAO,SAASA,CAAI,MAAQ,EAAE,GAAGF,CAAS,eACtE,EACAZ,EAAY,YACVa,EAAI,IACJ,KAAKQ,EAAU,CAAE,SAAAL,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,EACnD,EAEIA,EAAc,OAAS,EACzBlB,EAAY,YACVa,EAAI,IACJ,MAAMK,EAAc,KAAK,IAAI,CAAC,IAChC,EAEAlB,EAAY,YACVa,EAAI,IACJ,IAAIO,GAAuBN,EAAO,IAAM,EAAE,EAC5C,EAGEC,GACFf,EAAY,cACVa,EACA,GAAGD,CAAS,oBAAoBE,EAAO,IAAIA,CAAI,IAAM,EAAE,GACzD,CAEJ,CAEA,OAAOQ,EAAkBtB,EAAaJ,CAAE,CAC1C,CACF,CACF,CACF,CACF,EAIO2B,EAAQpC,EAEFqC,EAAarC,EAAQ,KACrBsC,EAAetC,EAAQ,OACvBuC,EAAiBvC,EAAQ,SACzBwC,EAAgBxC,EAAQ,QACxByC,EAAezC,EAAQ,OACvB0C,EAAgB1C,EAAQ,QACxB2C,EAAa3C,EAAQ,KACrB4C,EAAcC","names":["defu","walk","MagicStringAST","generateTransform","transpileFn","createUnplugin","getKernelDirective","node","statement","kernelDirectives","removeKernelDirective","cloned","typegpu","createUnplugin","rawOptions","options","defu","defaultOptions","__spreadValues","codeFilterRegexes","code","id","ctx","ast","tgslFunctionDefs","magicString","MagicStringAST","walk","_node","_parent","prop","index","_a","gatherTgpuAliases","isShellImplementationCall","implementation","directive","tgpuAlias","def","name","removeJsImplementation","argNames","body","externalNames","transpileFn","isFunctionStatement","embedJSON","generateTransform","index_default","vitePlugin","rollupPlugin","rolldownPlugin","webpackPlugin","rspackPlugin","esbuildPlugin","farmPlugin","babelPlugin","babel_default"]}

package/chunk-HPSWWSWV.cjs

@@ -1,2 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var _chunkHQFSG5ZMcjs = require('./chunk-HQFSG5ZM.cjs');var _defu = require('defu'); var _defu2 = _interopRequireDefault(_defu);var _estreewalker = require('estree-walker');var _magicstringast = require('magic-string-ast');var _tinyestforwgsl = require('tinyest-for-wgsl');var _unplugin = require('unplugin');function k(c){if(c.body.type==="BlockStatement"){for(let n of c.body.body)if(n.type==="ExpressionStatement"&&_chunkHQFSG5ZMcjs.g.includes(n.directive))return n.directive}}function A(c){let n=structuredClone(c);return n.body.type==="BlockStatement"&&(n.body.body=n.body.body.filter(s=>!(s.type==="ExpressionStatement"&&s.directive&&_chunkHQFSG5ZMcjs.g.includes(s.directive)))),n}var r=_unplugin.createUnplugin.call(void 0, c=>{let n=_defu2.default.call(void 0, c,_chunkHQFSG5ZMcjs.b);return{name:"unplugin-typegpu",enforce:n.enforce,transform:{filter:_chunkHQFSG5ZMcjs.a.call(void 0, {id:n},n.forceTgpuAlias?{}:{code:_chunkHQFSG5ZMcjs.d}),handler(s,l){let d={tgpuAliases:new Set(n.forceTgpuAlias?[n.forceTgpuAlias]:[]),fileId:l},$=this.parse(s,{allowReturnOutsideFunction:!0}),u=[],a=new (0, _magicstringast.MagicStringAST)(s);_estreewalker.walk.call(void 0, $,{enter(i,t,m,y){var p;let e=i;if(e.type==="ImportDeclaration"&&_chunkHQFSG5ZMcjs.e.call(void 0, e,d),e.type==="CallExpression"&&_chunkHQFSG5ZMcjs.f.call(void 0, e,d)){let o=e.arguments[0];if(o&&(o.type==="FunctionExpression"||o.type==="ArrowFunctionExpression")){let E=k(o);u.push({def:A(o),removeJsImplementation:E!=="kernel & js"}),this.skip()}}if(e.type==="ArrowFunctionExpression"||e.type==="FunctionExpression"||e.type==="FunctionDeclaration"){let o=k(e);o&&(u.push({def:A(e),name:e.type==="FunctionDeclaration"||e.type==="FunctionExpression"?(p=e.id)==null?void 0:p.name:(t==null?void 0:t.type)==="VariableDeclarator"&&t.id.type==="Identifier"?t.id.name:void 0,removeJsImplementation:o!=="kernel & js"}),this.skip())}}});let f=d.tgpuAliases.values().next().value;if(f===void 0&&u.length>0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${l}`);for(let{def:i,name:t,removeJsImplementation:m}of u){let{argNames:y,body:e,externalNames:p}=_tinyestforwgsl.transpileFn.call(void 0, i),o=i.type==="FunctionDeclaration";if(o&&t&&s.slice(0,i.start).search(new RegExp(`(?<![\\w_.])${t}(?![\\w_])`))!==-1)throw new Error(`File ${l}: function "${t}", containing ${m?"kernel":"kernel & js"} directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);a.appendLeft(i.start,`${o&&t?`const ${t} = `:""}${f}.__assignAst(`),a.appendRight(i.end,`, ${_chunkHQFSG5ZMcjs.c.call(void 0, {argNames:y,body:e,externalNames:p})}`),p.length>0?a.appendRight(i.end,`, {${p.join(", ")}})`):a.appendRight(i.end,`)${o&&t?";":""}`),m&&a.overwriteNode(i,`${f}.__removedJsImpl(${t?`"${t}"`:""})`)}return _magicstringast.generateTransform.call(void 0, a,l)}}}}),L= exports.a =r,M= exports.b =r.vite,V= exports.c =r.rollup,q= exports.d =r.rolldown,z= exports.e =r.webpack,G= exports.f =r.rspack,H= exports.g =r.esbuild,Q= exports.h =r.farm,W= exports.i =_chunkHQFSG5ZMcjs.h;exports.a = L; exports.b = M; exports.c = V; exports.d = q; exports.e = z; exports.f = G; exports.g = H; exports.h = Q; exports.i = W;
-//# sourceMappingURL=chunk-HPSWWSWV.cjs.map

package/chunk-HPSWWSWV.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/chunk-HPSWWSWV.cjs","../src/index.ts"],"names":["getKernelDirective","node","statement","kernelDirectives","removeKernelDirective","cloned","typegpu","createUnplugin","rawOptions","options","defu","defaultOptions","__spreadValues","codeFilterRegexes","code","id","ctx","ast","tgslFunctionDefs","magicString","MagicStringAST","walk","_node","_parent","prop","index","_a","gatherTgpuAliases","isShellImplementationCall","implementation","directive","tgpuAlias"],"mappings":"AAAA,iOAAoF,wECCnE,6CACe,kDACkB,kDACtB,oCAC0B,SAoB7CA,CAAAA,CAAmBC,CAAAA,CAAiD,CAC3E,EAAA,CAAIA,CAAAA,CAAK,IAAA,CAAK,IAAA,GAAS,gBAAA,CAAA,CACrB,GAAA,CAAA,IAAWC,EAAAA,GAAaD,CAAAA,CAAK,IAAA,CAAK,IAAA,CAChC,EAAA,CAAIC,CAAAA,CAAU,IAAA,GAAS,qBAAA,EACjBC,mBAAAA,CAAiB,QAAA,CAASD,CAAAA,CAAU,SAA4B,CAAA,CAClE,OAAOA,CAAAA,CAAU,SAAA,CAK3B,CAEA,SAASE,CAAAA,CAAsBH,CAAAA,CAAoB,CACjD,IAAMI,CAAAA,CAAS,eAAA,CAAgBJ,CAAI,CAAA,CAEnC,OAAII,CAAAA,CAAO,IAAA,CAAK,IAAA,GAAS,gBAAA,EAAA,CACvBA,CAAAA,CAAO,IAAA,CAAK,IAAA,CAAOA,CAAAA,CAAO,IAAA,CAAK,IAAA,CAAK,MAAA,CACjCH,CAAAA,EACC,CAAA,CACEA,CAAAA,CAAU,IAAA,GAAS,qBAAA,EACnBA,CAAAA,CAAU,SAAA,EACVC,mBAAAA,CAAiB,QAAA,CAASD,CAAAA,CAAU,SAA4B,CAAA,CAEtE,CAAA,CAAA,CAGKG,CACT,CAEA,IAAMC,CAAAA,CAA4CC,sCAAAA,CAC/CC,EAAe,CACd,IAAMC,CAAAA,CAAUC,4BAAAA,CAAKF,CAAYG,mBAAc,CAAA,CAE/C,MAAO,CACL,IAAA,CAAM,kBAAA,CACN,OAAA,CAASF,CAAAA,CAAQ,OAAA,CACjB,SAAA,CAAW,CACT,MAAA,CAAQG,iCAAAA,CACN,EAAA,CAAIH,CAAAA,CAAAA,CACAA,CAAAA,CAAQ,cAAA,CAAiB,CAAC,CAAA,CAAI,CAAE,IAAA,CAAMI,mBAAkB,CAAA,CAAA,CAE9D,OAAA,CAAQC,CAAAA,CAAMC,CAAAA,CAAI,CAChB,IAAMC,CAAAA,CAAe,CACnB,WAAA,CAAa,IAAI,GAAA,CACfP,CAAAA,CAAQ,cAAA,CAAiB,CAACA,CAAAA,CAAQ,cAAc,CAAA,CAAI,CAAC,CACvD,CAAA,CACA,MAAA,CAAQM,CACV,CAAA,CAEME,CAAAA,CAAM,IAAA,CAAK,KAAA,CAAMH,CAAAA,CAAM,CAC3B,0BAAA,CAA4B,CAAA,CAC9B,CAAC,CAAA,CAEKI,CAAAA,CAIA,CAAC,CAAA,CAEDC,CAAAA,CAAc,IAAIC,mCAAAA,CAAeN,CAAI,CAAA,CAE3CO,gCAAAA,CAAKJ,CAAK,CACR,KAAA,CAAMK,CAAAA,CAAOC,CAAAA,CAASC,CAAAA,CAAMC,CAAAA,CAAO,CAvF/C,IAAAC,CAAAA,CAwFc,IAAMzB,CAAAA,CAAOqB,CAAAA,CAMb,EAAA,CAJIrB,CAAAA,CAAK,IAAA,GAAS,mBAAA,EAChB0B,iCAAAA,CAAkB1B,CAAMe,CAAG,CAAA,CAGzBf,CAAAA,CAAK,IAAA,GAAS,gBAAA,EACZ2B,iCAAAA,CAA0B3B,CAAMe,CAAG,CAAA,CAAG,CACxC,IAAMa,CAAAA,CAAiB5B,CAAAA,CAAK,SAAA,CAAU,CAAC,CAAA,CAEvC,EAAA,CACE4B,CAAAA,EAAAA,CACCA,CAAAA,CAAe,IAAA,GAAS,oBAAA,EACvBA,CAAAA,CAAe,IAAA,GAAS,yBAAA,CAAA,CAC1B,CACA,IAAMC,CAAAA,CAAY9B,CAAAA,CAAmB6B,CAAc,CAAA,CACnDX,CAAAA,CAAiB,IAAA,CAAK,CACpB,GAAA,CAAKd,CAAAA,CAAsByB,CAAc,CAAA,CACzC,sBAAA,CAAwBC,CAAAA,GAAc,aACxC,CAAC,CAAA,CACD,IAAA,CAAK,IAAA,CAAK,CACZ,CACF,CAGF,EAAA,CACE7B,CAAAA,CAAK,IAAA,GAAS,yBAAA,EACdA,CAAAA,CAAK,IAAA,GAAS,oBAAA,EACdA,CAAAA,CAAK,IAAA,GAAS,qBAAA,CACd,CACA,IAAM6B,CAAAA,CAAY9B,CAAAA,CAAmBC,CAAI,CAAA,CACrC6B,CAAAA,EAAAA,CACFZ,CAAAA,CAAiB,IAAA,CAAK,CACpB,GAAA,CAAKd,CAAAA,CAAsBH,CAAI,CAAA,CAC/B,IAAA,CACEA,CAAAA,CAAK,IAAA,GAAS,qBAAA,EACdA,CAAAA,CAAK,IAAA,GAAS,oBAAA,CAAA,CACVyB,CAAAA,CAAAzB,CAAAA,CAAK,EAAA,CAAA,EAAL,IAAA,CAAA,KAAA,CAAA,CAAAyB,CAAAA,CAAS,IAAA,CAAA,CACTH,CAAAA,EAAA,IAAA,CAAA,KAAA,CAAA,CAAAA,CAAAA,CAAS,IAAA,CAAA,GAAS,oBAAA,EAChBA,CAAAA,CAAQ,EAAA,CAAG,IAAA,GAAS,YAAA,CAClBA,CAAAA,CAAQ,EAAA,CAAG,IAAA,CAEb,KAAA,CAAA,CACR,sBAAA,CAAwBO,CAAAA,GAAc,aACxC,CAAC,CAAA,CACD,IAAA,CAAK,IAAA,CAAK,CAAA,CAEd,CACF,CACF,CAAC,CAAA,CAED,IAAMC,CAAAA,CAAYf,CAAAA,CAAI,WAAA,CAAY,MAAA,CAAO,CAAA,CAAE,IAAA,CAAK,CAAA,CAAE,KAAA,CAElD,EAAA,CAAIe,CAAAA,GAAc,KAAA,CAAA,EAAab,CAAAA,CAAiB,MAAA,CAAS,CAAA,CACvD,MAAM,IAAI,KAAA,CACR,CAAA,6DAAA,EAAgEH,CAAE,CAAA,CAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/chunk-HPSWWSWV.cjs","sourcesContent":[null,"import type * as acorn from 'acorn';\nimport defu from 'defu';\nimport { type Node, walk } from 'estree-walker';\nimport { MagicStringAST, generateTransform } from 'magic-string-ast';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { type UnpluginInstance, createUnplugin } from 'unplugin';\nimport babel from './babel.ts';\nimport {\n  type Context,\n  type KernelDirective,\n  type Options,\n  codeFilterRegexes,\n  defaultOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  kernelDirectives,\n} from './common.ts';\n\ntype FunctionNode =\n  | acorn.FunctionDeclaration\n  | acorn.AnonymousFunctionDeclaration\n  | acorn.FunctionExpression\n  | acorn.ArrowFunctionExpression;\n\nfunction getKernelDirective(node: FunctionNode): KernelDirective | undefined {\n  if (node.body.type === 'BlockStatement') {\n    for (const statement of node.body.body) {\n      if (statement.type === 'ExpressionStatement') {\n        if (kernelDirectives.includes(statement.directive as KernelDirective)) {\n          return statement.directive as KernelDirective;\n        }\n      }\n    }\n  }\n}\n\nfunction removeKernelDirective(node: FunctionNode) {\n  const cloned = structuredClone(node);\n\n  if (cloned.body.type === 'BlockStatement') {\n    cloned.body.body = cloned.body.body.filter(\n      (statement) =>\n        !(\n          statement.type === 'ExpressionStatement' &&\n          statement.directive &&\n          kernelDirectives.includes(statement.directive as KernelDirective)\n        ),\n    );\n  }\n\n  return cloned;\n}\n\nconst typegpu: UnpluginInstance<Options, false> = createUnplugin(\n  (rawOptions) => {\n    const options = defu(rawOptions, defaultOptions);\n\n    return {\n      name: 'unplugin-typegpu' as const,\n      enforce: options.enforce,\n      transform: {\n        filter: {\n          id: options,\n          ...(options.forceTgpuAlias ? {} : { code: codeFilterRegexes }),\n        },\n        handler(code, id) {\n          const ctx: Context = {\n            tgpuAliases: new Set<string>(\n              options.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n            ),\n            fileId: id,\n          };\n\n          const ast = this.parse(code, {\n            allowReturnOutsideFunction: true,\n          }) as Node;\n\n          const tgslFunctionDefs: {\n            def: FunctionNode;\n            name?: string | undefined;\n            removeJsImplementation: boolean;\n          }[] = [];\n\n          const magicString = new MagicStringAST(code);\n\n          walk(ast, {\n            enter(_node, _parent, prop, index) {\n              const node = _node as acorn.AnyNode;\n\n              if (node.type === 'ImportDeclaration') {\n                gatherTgpuAliases(node, ctx);\n              }\n\n              if (node.type === 'CallExpression') {\n                if (isShellImplementationCall(node, ctx)) {\n                  const implementation = node.arguments[0];\n\n                  if (\n                    implementation &&\n                    (implementation.type === 'FunctionExpression' ||\n                      implementation.type === 'ArrowFunctionExpression')\n                  ) {\n                    const directive = getKernelDirective(implementation);\n                    tgslFunctionDefs.push({\n                      def: removeKernelDirective(implementation),\n                      removeJsImplementation: directive !== 'kernel & js',\n                    });\n                    this.skip();\n                  }\n                }\n              }\n\n              if (\n                node.type === 'ArrowFunctionExpression' ||\n                node.type === 'FunctionExpression' ||\n                node.type === 'FunctionDeclaration'\n              ) {\n                const directive = getKernelDirective(node);\n                if (directive) {\n                  tgslFunctionDefs.push({\n                    def: removeKernelDirective(node),\n                    name:\n                      node.type === 'FunctionDeclaration' ||\n                      node.type === 'FunctionExpression'\n                        ? node.id?.name\n                        : _parent?.type === 'VariableDeclarator'\n                          ? _parent.id.type === 'Identifier'\n                            ? _parent.id.name\n                            : undefined\n                          : undefined,\n                    removeJsImplementation: directive !== 'kernel & js',\n                  });\n                  this.skip();\n                }\n              }\n            },\n          });\n\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n\n          if (tgpuAlias === undefined && tgslFunctionDefs.length > 0) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${id}`,\n            );\n          }\n\n          for (const {\n            def,\n            name,\n            removeJsImplementation,\n          } of tgslFunctionDefs) {\n            const { argNames, body, externalNames } = transpileFn(def);\n            const isFunctionStatement = def.type === 'FunctionDeclaration';\n\n            if (\n              isFunctionStatement &&\n              name &&\n              code\n                .slice(0, def.start)\n                .search(new RegExp(`(?<![\\\\w_.])${name}(?![\\\\w_])`)) !== -1\n            ) {\n              throw new Error(\n                `File ${id}: function \"${name}\", containing ${removeJsImplementation ? 'kernel' : 'kernel & js'} directive, is referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`,\n              );\n            }\n\n            // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n            magicString.appendLeft(\n              def.start,\n              `${isFunctionStatement && name ? `const ${name} = ` : ''}${tgpuAlias}.__assignAst(`,\n            );\n            magicString.appendRight(\n              def.end,\n              `, ${embedJSON({ argNames, body, externalNames })}`,\n            );\n\n            if (externalNames.length > 0) {\n              magicString.appendRight(\n                def.end,\n                `, {${externalNames.join(', ')}})`,\n              );\n            } else {\n              magicString.appendRight(\n                def.end,\n                `)${isFunctionStatement && name ? ';' : ''}`,\n              );\n            }\n\n            if (removeJsImplementation) {\n              magicString.overwriteNode(\n                def,\n                `${tgpuAlias}.__removedJsImpl(${name ? `\"${name}\"` : ''})`,\n              );\n            }\n          }\n\n          return generateTransform(magicString, id);\n        },\n      },\n    };\n  },\n);\n\nexport type { Options } from './common.ts';\n\nexport default typegpu;\n\nexport const vitePlugin = typegpu.vite;\nexport const rollupPlugin = typegpu.rollup;\nexport const rolldownPlugin = typegpu.rolldown;\nexport const webpackPlugin = typegpu.webpack;\nexport const rspackPlugin = typegpu.rspack;\nexport const esbuildPlugin = typegpu.esbuild;\nexport const farmPlugin = typegpu.farm;\nexport const babelPlugin = babel;\n"]}

package/chunk-HQFSG5ZM.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/chunk-HQFSG5ZM.cjs","../src/babel.ts","../src/common.ts"],"names":["defaultOptions","embedJSON","jsValue","isTgpu","ctx","node","path","tail"],"mappings":"AAAA,+bAAI,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CCAxV,6FAAuB,kDAIK,ICgBfA,CAAAA,CAAiB,CAC5B,OAAA,CAAS,CAAC,cAAc,CAC1B,CAAA,CAEO,SAASC,CAAAA,CAAUC,CAAAA,CAAkB,CAC1C,OAAO,IAAA,CAAK,SAAA,CAAUA,CAAO,CAAA,CAC1B,OAAA,CAAQ,SAAA,CAAW,SAAS,CAAA,CAC5B,OAAA,CAAQ,SAAA,CAAW,SAAS,CACjC,CAMA,SAASC,CAAAA,CAAOC,CAAAA,CAAcC,CAAAA,CAA2C,CACvE,IAAIC,CAAAA,CAAO,EAAA,CAEPC,CAAAA,CAAOF,CAAAA,CACX,GAAA,CAAA,CAAA,CAAA,CACE,EAAA,CAAIE,CAAAA,CAAK,IAAA,GAAS,kBAAA,CAAoB,CACpC,EAAA,CAAIA,CAAAA,CAAK,QAAA,CAAS,IAAA,GAAS,YAAA,CAEzB,KAAA,CAGFD,CAAAA,CAAOA,CAAAA,CAAO,CAAA,EAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/chunk-HQFSG5ZM.cjs","sourcesContent":[null,"import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as babel from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  type KernelDirective,\n  type Options,\n  codeFilterRegexes,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  kernelDirectives,\n} from './common.ts';\nimport { createFilterForId } from './filter.ts';\n\n// NOTE: @babel/standalone does expose internal packages, as specified in the docs, but the\n// typing for @babel/standalone does not expose them.\nconst template = (\n  Babel as unknown as { packages: { template: typeof TemplateGenerator } }\n).packages.template;\nconst types = (Babel as unknown as { packages: { types: typeof babel } })\n  .packages.types;\n\nfunction getKernelDirective(\n  node:\n    | babel.FunctionDeclaration\n    | babel.FunctionExpression\n    | babel.ArrowFunctionExpression,\n): KernelDirective | undefined {\n  const directives = (\n    'directives' in node.body ? (node.body?.directives ?? []) : []\n  ).map((directive) => directive.value.value);\n\n  for (const directive of kernelDirectives) {\n    if (directives.includes(directive)) {\n      return directive;\n    }\n  }\n}\n\nfunction functionToTranspiled(\n  node: babel.ArrowFunctionExpression | babel.FunctionExpression,\n  ctx: Context,\n  name?: string | undefined,\n): babel.CallExpression | null {\n  const directive = getKernelDirective(node);\n  if (!directive) {\n    return null;\n  }\n\n  const { argNames, body, externalNames } = transpileFn(node);\n  const tgpuAlias = ctx.tgpuAliases.values().next().value;\n  if (tgpuAlias === undefined) {\n    throw new Error(\n      `No tgpu import found, cannot assign ast to function in file: ${ctx.fileId ?? ''}`,\n    );\n  }\n\n  return types.callExpression(\n    template.expression(`${tgpuAlias}.__assignAst`)(),\n    [\n      directive === 'kernel & js'\n        ? node\n        : template.expression`${tgpuAlias}.__removedJsImpl(${name ? `\"${name}\"` : ''})`(),\n      template.expression`${embedJSON({ argNames, body, externalNames })}`(),\n      types.objectExpression(\n        externalNames.map((name) =>\n          types.objectProperty(types.identifier(name), types.identifier(name)),\n        ),\n      ),\n    ],\n  );\n}\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    ArrowFunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.parentPath.node.type === 'VariableDeclarator'\n          ? path.parentPath.node.id.type === 'Identifier'\n            ? path.parentPath.node.id.name\n            : undefined\n          : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.node.id?.name\n          ? path.node.id.name\n          : path.parentPath.node.type === 'VariableDeclarator'\n            ? path.parentPath.node.id.type === 'Identifier'\n              ? path.parentPath.node.id.name\n              : undefined\n            : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionDeclaration(path) {\n      const node = path.node;\n      const expression = types.functionExpression(\n        node.id,\n        node.params,\n        node.body,\n      );\n      const transpiled = functionToTranspiled(expression, ctx, node.id?.name);\n      if (transpiled && node.id) {\n        path.replaceWith(\n          types.variableDeclaration('const', [\n            types.variableDeclarator(node.id, transpiled),\n          ]),\n        );\n        path.skip();\n      }\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isShellImplementationCall(node, ctx)) {\n        const implementation = node.arguments[0];\n\n        if (\n          implementation &&\n          (implementation.type === 'FunctionExpression' ||\n            implementation.type === 'ArrowFunctionExpression')\n        ) {\n          const { argNames, body, externalNames } = transpileFn(implementation);\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n          if (tgpuAlias === undefined) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${ctx.fileId ?? ''}`,\n            );\n          }\n\n          const directive = getKernelDirective(implementation);\n\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(\n                template.expression(`${tgpuAlias}.__assignAst`)(),\n                [\n                  directive !== 'kernel & js'\n                    ? template.expression`${tgpuAlias}.__removedJsImpl()`()\n                    : implementation,\n                  template.expression`${embedJSON({ argNames, body, externalNames })}`(),\n                  types.objectExpression(\n                    externalNames.map((name) =>\n                      types.objectProperty(\n                        types.identifier(name),\n                        types.identifier(name),\n                      ),\n                    ),\n                  ),\n                ],\n              ),\n            ]),\n          );\n\n          path.skip();\n        }\n      }\n    },\n  };\n}\n\nexport default function () {\n  return {\n    visitor: {\n      Program(path, state) {\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const code: string | undefined = (state as any).file?.code;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const options: Options | undefined = (state as any).opts;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const id: string | undefined = (state as any).filename;\n\n        const filter = createFilterForId(options);\n        if (id && filter && !filter?.(id)) {\n          return;\n        }\n\n        if (\n          !options?.forceTgpuAlias &&\n          code &&\n          !codeFilterRegexes.some((reg) => reg.test(code))\n        ) {\n          return;\n        }\n\n        const ctx: Context = {\n          tgpuAliases: new Set<string>(\n            options?.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n          ),\n          fileId: id,\n        };\n\n        path.traverse(functionVisitor(ctx));\n      },\n    } satisfies TraverseOptions,\n  };\n}\n","import type * as babel from '@babel/types';\nimport type * as acorn from 'acorn';\nimport type { FilterPattern } from 'unplugin';\n\nexport type Context = {\n  /**\n   * How the `tgpu` object is used in code. Since it can be aliased, we\n   * need to catch that and act accordingly.\n   */\n  tgpuAliases: Set<string>;\n  fileId?: string | undefined;\n};\n\nexport interface Options {\n  include?: FilterPattern;\n  exclude?: FilterPattern;\n  enforce?: 'post' | 'pre' | undefined;\n  forceTgpuAlias?: string;\n}\n\nexport const defaultOptions = {\n  include: [/\\.m?[jt]sx?$/],\n};\n\nexport function embedJSON(jsValue: unknown) {\n  return JSON.stringify(jsValue)\n    .replace(/\\u2028/g, '\\\\u2028')\n    .replace(/\\u2029/g, '\\\\u2029');\n}\n\n/**\n * Checks if `node` is an alias for the 'tgpu' object, traditionally\n * available via `import tgpu from 'typegpu'`.\n */\nfunction isTgpu(ctx: Context, node: babel.Node | acorn.AnyNode): boolean {\n  let path = '';\n\n  let tail = node;\n  while (true) {\n    if (tail.type === 'MemberExpression') {\n      if (tail.property.type !== 'Identifier') {\n        // Not handling computed expressions.\n        break;\n      }\n\n      path = path ? `${tail.property.name}.${path}` : tail.property.name;\n      tail = tail.object;\n    } else if (tail.type === 'Identifier') {\n      path = path ? `${tail.name}.${path}` : tail.name;\n      break;\n    } else {\n      break;\n    }\n  }\n\n  return ctx.tgpuAliases.has(path);\n}\n\nconst typegpuImportRegex = /import.*from\\s*['\"]typegpu.*['\"]/;\nconst typegpuDynamicImportRegex = /import\\s*\\(\\s*['\"]\\s*typegpu.*['\"]/;\nconst typegpuRequireRegex = /require\\s*\\(\\s*['\"]\\s*typegpu.*['\"]\\s*\\)/;\n\n/**\n * Regexes used to efficiently determine if a file is\n * meant to be processed by our plugin. We assume every file\n * that should be processed imports `typegpu` in some way.\n */\nexport const codeFilterRegexes = [\n  typegpuImportRegex,\n  typegpuDynamicImportRegex,\n  typegpuRequireRegex,\n];\n\nexport function gatherTgpuAliases(\n  node: acorn.ImportDeclaration | babel.ImportDeclaration,\n  ctx: Context,\n) {\n  if (node.source.value === 'typegpu') {\n    for (const spec of node.specifiers) {\n      if (\n        // The default export of 'typegpu' is the `tgpu` object.\n        spec.type === 'ImportDefaultSpecifier' ||\n        // Aliasing 'tgpu' while importing, e.g. import { tgpu as t } from 'typegpu';\n        (spec.type === 'ImportSpecifier' &&\n          spec.imported.type === 'Identifier' &&\n          spec.imported.name === 'tgpu')\n      ) {\n        ctx.tgpuAliases.add(spec.local.name);\n      } else if (spec.type === 'ImportNamespaceSpecifier') {\n        // Importing everything, e.g. import * as t from 'typegpu';\n        ctx.tgpuAliases.add(`${spec.local.name}.tgpu`);\n      }\n    }\n  }\n}\n\nconst fnShellFunctionNames = ['fn', 'vertexFn', 'fragmentFn', 'computeFn'];\n\nexport function isShellImplementationCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    (node.callee.type === 'CallExpression' &&\n      node.callee.callee.type === 'MemberExpression' &&\n      node.callee.callee.property.type === 'Identifier' &&\n      fnShellFunctionNames.includes(node.callee.callee.property.name) &&\n      node.arguments.length === 1 &&\n      (node.callee.callee.object.type === 'MemberExpression'\n        ? isTgpu(ctx, node.callee.callee.object.object)\n        : isTgpu(ctx, node.callee.callee.object))) || // TODO: remove along with the deprecated 'does' method\n    (node.callee.type === 'MemberExpression' &&\n      node.arguments.length === 1 &&\n      node.callee.property.type === 'Identifier' &&\n      // Assuming that every call to `.does` is related to TypeGPU\n      // because shells can be created separately from calls to `tgpu`,\n      // making it hard to detect.\n      node.callee.property.name === 'does')\n  );\n}\n\nexport const kernelDirectives = ['kernel', 'kernel & js'] as const;\nexport type KernelDirective = (typeof kernelDirectives)[number];\n"]}

package/chunk-Y4WARDXR.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/babel.ts","../src/common.ts","../src/filter.ts"],"sourcesContent":["import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as babel from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  type KernelDirective,\n  type Options,\n  codeFilterRegexes,\n  embedJSON,\n  gatherTgpuAliases,\n  isShellImplementationCall,\n  kernelDirectives,\n} from './common.ts';\nimport { createFilterForId } from './filter.ts';\n\n// NOTE: @babel/standalone does expose internal packages, as specified in the docs, but the\n// typing for @babel/standalone does not expose them.\nconst template = (\n  Babel as unknown as { packages: { template: typeof TemplateGenerator } }\n).packages.template;\nconst types = (Babel as unknown as { packages: { types: typeof babel } })\n  .packages.types;\n\nfunction getKernelDirective(\n  node:\n    | babel.FunctionDeclaration\n    | babel.FunctionExpression\n    | babel.ArrowFunctionExpression,\n): KernelDirective | undefined {\n  const directives = (\n    'directives' in node.body ? (node.body?.directives ?? []) : []\n  ).map((directive) => directive.value.value);\n\n  for (const directive of kernelDirectives) {\n    if (directives.includes(directive)) {\n      return directive;\n    }\n  }\n}\n\nfunction functionToTranspiled(\n  node: babel.ArrowFunctionExpression | babel.FunctionExpression,\n  ctx: Context,\n  name?: string | undefined,\n): babel.CallExpression | null {\n  const directive = getKernelDirective(node);\n  if (!directive) {\n    return null;\n  }\n\n  const { argNames, body, externalNames } = transpileFn(node);\n  const tgpuAlias = ctx.tgpuAliases.values().next().value;\n  if (tgpuAlias === undefined) {\n    throw new Error(\n      `No tgpu import found, cannot assign ast to function in file: ${ctx.fileId ?? ''}`,\n    );\n  }\n\n  return types.callExpression(\n    template.expression(`${tgpuAlias}.__assignAst`)(),\n    [\n      directive === 'kernel & js'\n        ? node\n        : template.expression`${tgpuAlias}.__removedJsImpl(${name ? `\"${name}\"` : ''})`(),\n      template.expression`${embedJSON({ argNames, body, externalNames })}`(),\n      types.objectExpression(\n        externalNames.map((name) =>\n          types.objectProperty(types.identifier(name), types.identifier(name)),\n        ),\n      ),\n    ],\n  );\n}\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    ArrowFunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.parentPath.node.type === 'VariableDeclarator'\n          ? path.parentPath.node.id.type === 'Identifier'\n            ? path.parentPath.node.id.name\n            : undefined\n          : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionExpression(path) {\n      const transpiled = functionToTranspiled(\n        path.node,\n        ctx,\n        path.node.id?.name\n          ? path.node.id.name\n          : path.parentPath.node.type === 'VariableDeclarator'\n            ? path.parentPath.node.id.type === 'Identifier'\n              ? path.parentPath.node.id.name\n              : undefined\n            : undefined,\n      );\n      if (transpiled) {\n        path.replaceWith(transpiled);\n        path.skip();\n      }\n    },\n\n    FunctionDeclaration(path) {\n      const node = path.node;\n      const expression = types.functionExpression(\n        node.id,\n        node.params,\n        node.body,\n      );\n      const transpiled = functionToTranspiled(expression, ctx, node.id?.name);\n      if (transpiled && node.id) {\n        path.replaceWith(\n          types.variableDeclaration('const', [\n            types.variableDeclarator(node.id, transpiled),\n          ]),\n        );\n        path.skip();\n      }\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isShellImplementationCall(node, ctx)) {\n        const implementation = node.arguments[0];\n\n        if (\n          implementation &&\n          (implementation.type === 'FunctionExpression' ||\n            implementation.type === 'ArrowFunctionExpression')\n        ) {\n          const { argNames, body, externalNames } = transpileFn(implementation);\n          const tgpuAlias = ctx.tgpuAliases.values().next().value;\n          if (tgpuAlias === undefined) {\n            throw new Error(\n              `No tgpu import found, cannot assign ast to function in file: ${ctx.fileId ?? ''}`,\n            );\n          }\n\n          const directive = getKernelDirective(implementation);\n\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(\n                template.expression(`${tgpuAlias}.__assignAst`)(),\n                [\n                  directive !== 'kernel & js'\n                    ? template.expression`${tgpuAlias}.__removedJsImpl()`()\n                    : implementation,\n                  template.expression`${embedJSON({ argNames, body, externalNames })}`(),\n                  types.objectExpression(\n                    externalNames.map((name) =>\n                      types.objectProperty(\n                        types.identifier(name),\n                        types.identifier(name),\n                      ),\n                    ),\n                  ),\n                ],\n              ),\n            ]),\n          );\n\n          path.skip();\n        }\n      }\n    },\n  };\n}\n\nexport default function () {\n  return {\n    visitor: {\n      Program(path, state) {\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const code: string | undefined = (state as any).file?.code;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const options: Options | undefined = (state as any).opts;\n        // biome-ignore lint/suspicious/noExplicitAny: <oh babel babel...>\n        const id: string | undefined = (state as any).filename;\n\n        const filter = createFilterForId(options);\n        if (id && filter && !filter?.(id)) {\n          return;\n        }\n\n        if (\n          !options?.forceTgpuAlias &&\n          code &&\n          !codeFilterRegexes.some((reg) => reg.test(code))\n        ) {\n          return;\n        }\n\n        const ctx: Context = {\n          tgpuAliases: new Set<string>(\n            options?.forceTgpuAlias ? [options.forceTgpuAlias] : [],\n          ),\n          fileId: id,\n        };\n\n        path.traverse(functionVisitor(ctx));\n      },\n    } satisfies TraverseOptions,\n  };\n}\n","import type * as babel from '@babel/types';\nimport type * as acorn from 'acorn';\nimport type { FilterPattern } from 'unplugin';\n\nexport type Context = {\n  /**\n   * How the `tgpu` object is used in code. Since it can be aliased, we\n   * need to catch that and act accordingly.\n   */\n  tgpuAliases: Set<string>;\n  fileId?: string | undefined;\n};\n\nexport interface Options {\n  include?: FilterPattern;\n  exclude?: FilterPattern;\n  enforce?: 'post' | 'pre' | undefined;\n  forceTgpuAlias?: string;\n}\n\nexport const defaultOptions = {\n  include: [/\\.m?[jt]sx?$/],\n};\n\nexport function embedJSON(jsValue: unknown) {\n  return JSON.stringify(jsValue)\n    .replace(/\\u2028/g, '\\\\u2028')\n    .replace(/\\u2029/g, '\\\\u2029');\n}\n\n/**\n * Checks if `node` is an alias for the 'tgpu' object, traditionally\n * available via `import tgpu from 'typegpu'`.\n */\nfunction isTgpu(ctx: Context, node: babel.Node | acorn.AnyNode): boolean {\n  let path = '';\n\n  let tail = node;\n  while (true) {\n    if (tail.type === 'MemberExpression') {\n      if (tail.property.type !== 'Identifier') {\n        // Not handling computed expressions.\n        break;\n      }\n\n      path = path ? `${tail.property.name}.${path}` : tail.property.name;\n      tail = tail.object;\n    } else if (tail.type === 'Identifier') {\n      path = path ? `${tail.name}.${path}` : tail.name;\n      break;\n    } else {\n      break;\n    }\n  }\n\n  return ctx.tgpuAliases.has(path);\n}\n\nconst typegpuImportRegex = /import.*from\\s*['\"]typegpu.*['\"]/;\nconst typegpuDynamicImportRegex = /import\\s*\\(\\s*['\"]\\s*typegpu.*['\"]/;\nconst typegpuRequireRegex = /require\\s*\\(\\s*['\"]\\s*typegpu.*['\"]\\s*\\)/;\n\n/**\n * Regexes used to efficiently determine if a file is\n * meant to be processed by our plugin. We assume every file\n * that should be processed imports `typegpu` in some way.\n */\nexport const codeFilterRegexes = [\n  typegpuImportRegex,\n  typegpuDynamicImportRegex,\n  typegpuRequireRegex,\n];\n\nexport function gatherTgpuAliases(\n  node: acorn.ImportDeclaration | babel.ImportDeclaration,\n  ctx: Context,\n) {\n  if (node.source.value === 'typegpu') {\n    for (const spec of node.specifiers) {\n      if (\n        // The default export of 'typegpu' is the `tgpu` object.\n        spec.type === 'ImportDefaultSpecifier' ||\n        // Aliasing 'tgpu' while importing, e.g. import { tgpu as t } from 'typegpu';\n        (spec.type === 'ImportSpecifier' &&\n          spec.imported.type === 'Identifier' &&\n          spec.imported.name === 'tgpu')\n      ) {\n        ctx.tgpuAliases.add(spec.local.name);\n      } else if (spec.type === 'ImportNamespaceSpecifier') {\n        // Importing everything, e.g. import * as t from 'typegpu';\n        ctx.tgpuAliases.add(`${spec.local.name}.tgpu`);\n      }\n    }\n  }\n}\n\nconst fnShellFunctionNames = ['fn', 'vertexFn', 'fragmentFn', 'computeFn'];\n\nexport function isShellImplementationCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    (node.callee.type === 'CallExpression' &&\n      node.callee.callee.type === 'MemberExpression' &&\n      node.callee.callee.property.type === 'Identifier' &&\n      fnShellFunctionNames.includes(node.callee.callee.property.name) &&\n      node.arguments.length === 1 &&\n      (node.callee.callee.object.type === 'MemberExpression'\n        ? isTgpu(ctx, node.callee.callee.object.object)\n        : isTgpu(ctx, node.callee.callee.object))) || // TODO: remove along with the deprecated 'does' method\n    (node.callee.type === 'MemberExpression' &&\n      node.arguments.length === 1 &&\n      node.callee.property.type === 'Identifier' &&\n      // Assuming that every call to `.does` is related to TypeGPU\n      // because shells can be created separately from calls to `tgpu`,\n      // making it hard to detect.\n      node.callee.property.name === 'does')\n  );\n}\n\nexport const kernelDirectives = ['kernel', 'kernel & js'] as const;\nexport type KernelDirective = (typeof kernelDirectives)[number];\n","// Copied from https://github.com/unjs/unplugin/blob/f514bf8e2d751b48b3a5acb1077eb1292d9711b3/src/utils/filter.ts#L2\n// Used only in the babel version of the plugin, since others can rely on native unplugin functionality.\n\nimport { resolve } from 'node:path';\nimport picomatch from 'picomatch';\nimport type {\n  Arrayable,\n  Nullable,\n  StringFilter,\n  StringOrRegExp,\n} from 'unplugin';\n\nfunction toArray<T>(array?: Nullable<Arrayable<T>>): Array<T> {\n  const result = array || [];\n  if (Array.isArray(result)) return result;\n  return [result];\n}\n\nconst BACKSLASH_REGEX = /\\\\/g;\nfunction normalize(path: string): string {\n  return path.replace(BACKSLASH_REGEX, '/');\n}\n\nconst ABSOLUTE_PATH_REGEX = /^(?:\\/|(?:[A-Z]:)?[/\\\\|])/i;\nfunction isAbsolute(path: string): boolean {\n  return ABSOLUTE_PATH_REGEX.test(path);\n}\n\nexport type PluginFilter = (input: string) => boolean;\nexport type TransformHookFilter = (id: string, code: string) => boolean;\n\ninterface NormalizedStringFilter {\n  include?: StringOrRegExp[] | undefined;\n  exclude?: StringOrRegExp[] | undefined;\n}\n\nfunction getMatcherString(glob: string, cwd: string) {\n  if (glob.startsWith('**') || isAbsolute(glob)) {\n    return normalize(glob);\n  }\n\n  const resolved = resolve(cwd, glob);\n  return normalize(resolved);\n}\n\nfunction patternToIdFilter(pattern: StringOrRegExp): PluginFilter {\n  if (pattern instanceof RegExp) {\n    return (id: string) => {\n      const normalizedId = normalize(id);\n      const result = pattern.test(normalizedId);\n      pattern.lastIndex = 0;\n      return result;\n    };\n  }\n  const cwd = process.cwd();\n  const glob = getMatcherString(pattern, cwd);\n  const matcher = picomatch(glob, { dot: true });\n  return (id: string) => {\n    const normalizedId = normalize(id);\n    return matcher(normalizedId);\n  };\n}\n\nfunction patternToCodeFilter(pattern: StringOrRegExp): PluginFilter {\n  if (pattern instanceof RegExp) {\n    return (code: string) => {\n      const result = pattern.test(code);\n      pattern.lastIndex = 0;\n      return result;\n    };\n  }\n  return (code: string) => code.includes(pattern);\n}\n\nfunction createFilter(\n  exclude: PluginFilter[] | undefined,\n  include: PluginFilter[] | undefined,\n): PluginFilter | undefined {\n  if (!exclude && !include) {\n    return;\n  }\n\n  return (input) => {\n    if (exclude?.some((filter) => filter(input))) {\n      return false;\n    }\n    if (include?.some((filter) => filter(input))) {\n      return true;\n    }\n    return !(include && include.length > 0);\n  };\n}\n\nfunction normalizeFilter(filter: StringFilter): NormalizedStringFilter {\n  if (typeof filter === 'string' || filter instanceof RegExp) {\n    return {\n      include: [filter],\n    };\n  }\n  if (Array.isArray(filter)) {\n    return {\n      include: filter,\n    };\n  }\n  return {\n    exclude: filter.exclude ? toArray(filter.exclude) : undefined,\n    include: filter.include ? toArray(filter.include) : undefined,\n  };\n}\n\nfunction createIdFilter(\n  filter: StringFilter | undefined,\n): PluginFilter | undefined {\n  if (!filter) return;\n  const { exclude, include } = normalizeFilter(filter);\n  const excludeFilter = exclude?.map(patternToIdFilter);\n  const includeFilter = include?.map(patternToIdFilter);\n  return createFilter(excludeFilter, includeFilter);\n}\n\nfunction createCodeFilter(\n  filter: StringFilter | undefined,\n): PluginFilter | undefined {\n  if (!filter) return;\n  const { exclude, include } = normalizeFilter(filter);\n  const excludeFilter = exclude?.map(patternToCodeFilter);\n  const includeFilter = include?.map(patternToCodeFilter);\n  return createFilter(excludeFilter, includeFilter);\n}\n\nexport function createFilterForId(\n  filter: StringFilter | undefined,\n): PluginFilter | undefined {\n  const filterFunction = createIdFilter(filter);\n  return filterFunction ? (id) => !!filterFunction(id) : undefined;\n}\n\nfunction createFilterForTransform(\n  idFilter: StringFilter | undefined,\n  codeFilter: StringFilter | undefined,\n): TransformHookFilter | undefined {\n  if (!idFilter && !codeFilter) return;\n  const idFilterFunction = createIdFilter(idFilter);\n  const codeFilterFunction = createCodeFilter(codeFilter);\n  return (id, code) => {\n    let fallback = true;\n    if (idFilterFunction) {\n      fallback &&= idFilterFunction(id);\n    }\n    if (!fallback) {\n      return false;\n    }\n\n    if (codeFilterFunction) {\n      fallback &&= codeFilterFunction(code);\n    }\n    return fallback;\n  };\n}\n"],"mappings":"yVAAA,UAAYA,MAAW,oBAIvB,OAAS,eAAAC,MAAmB,mBCgBrB,IAAMC,EAAiB,CAC5B,QAAS,CAAC,cAAc,CAC1B,EAEO,SAASC,EAAUC,EAAkB,CAC1C,OAAO,KAAK,UAAUA,CAAO,EAC1B,QAAQ,UAAW,SAAS,EAC5B,QAAQ,UAAW,SAAS,CACjC,CAMA,SAASC,EAAOC,EAAcC,EAA2C,CACvE,IAAIC,EAAO,GAEPC,EAAOF,EACX,OACE,GAAIE,EAAK,OAAS,mBAAoB,CACpC,GAAIA,EAAK,SAAS,OAAS,aAEzB,MAGFD,EAAOA,EAAO,GAAGC,EAAK,SAAS,IAAI,IAAID,CAAI,GAAKC,EAAK,SAAS,KAC9DA,EAAOA,EAAK,MACd,SAAWA,EAAK,OAAS,aAAc,CACrCD,EAAOA,EAAO,GAAGC,EAAK,IAAI,IAAID,CAAI,GAAKC,EAAK,KAC5C,KACF,KACE,OAIJ,OAAOH,EAAI,YAAY,IAAIE,CAAI,CACjC,CAEA,IAAME,EAAqB,mCACrBC,EAA4B,qCAC5BC,EAAsB,2CAOfC,EAAoB,CAC/BH,EACAC,EACAC,CACF,EAEO,SAASE,EACdP,EACAD,EACA,CACA,GAAIC,EAAK,OAAO,QAAU,UACxB,QAAWQ,KAAQR,EAAK,WAGpBQ,EAAK,OAAS,0BAEbA,EAAK,OAAS,mBACbA,EAAK,SAAS,OAAS,cACvBA,EAAK,SAAS,OAAS,OAEzBT,EAAI,YAAY,IAAIS,EAAK,MAAM,IAAI,EAC1BA,EAAK,OAAS,4BAEvBT,EAAI,YAAY,IAAI,GAAGS,EAAK,MAAM,IAAI,OAAO,CAIrD,CAEA,IAAMC,EAAuB,CAAC,KAAM,WAAY,aAAc,WAAW,EAElE,SAASC,EACdV,EACAD,EACA,CACA,OACGC,EAAK,OAAO,OAAS,kBACpBA,EAAK,OAAO,OAAO,OAAS,oBAC5BA,EAAK,OAAO,OAAO,SAAS,OAAS,cACrCS,EAAqB,SAAST,EAAK,OAAO,OAAO,SAAS,IAAI,GAC9DA,EAAK,UAAU,SAAW,IACzBA,EAAK,OAAO,OAAO,OAAO,OAAS,mBAChCF,EAAOC,EAAKC,EAAK,OAAO,OAAO,OAAO,MAAM,EAC5CF,EAAOC,EAAKC,EAAK,OAAO,OAAO,MAAM,IAC1CA,EAAK,OAAO,OAAS,oBACpBA,EAAK,UAAU,SAAW,GAC1BA,EAAK,OAAO,SAAS,OAAS,cAI9BA,EAAK,OAAO,SAAS,OAAS,MAEpC,CAEO,IAAMW,EAAmB,CAAC,SAAU,aAAa,ECtHxD,OAAS,WAAAC,MAAe,YACxB,OAAOC,MAAe,YAQtB,SAASC,EAAWC,EAA0C,CAC5D,IAAMC,EAASD,GAAS,CAAC,EACzB,OAAI,MAAM,QAAQC,CAAM,EAAUA,EAC3B,CAACA,CAAM,CAChB,CAEA,IAAMC,EAAkB,MACxB,SAASC,EAAUC,EAAsB,CACvC,OAAOA,EAAK,QAAQF,EAAiB,GAAG,CAC1C,CAEA,IAAMG,EAAsB,6BAC5B,SAASC,EAAWF,EAAuB,CACzC,OAAOC,EAAoB,KAAKD,CAAI,CACtC,CAUA,SAASG,EAAiBC,EAAcC,EAAa,CACnD,GAAID,EAAK,WAAW,IAAI,GAAKF,EAAWE,CAAI,EAC1C,OAAOL,EAAUK,CAAI,EAGvB,IAAME,EAAWb,EAAQY,EAAKD,CAAI,EAClC,OAAOL,EAAUO,CAAQ,CAC3B,CAEA,SAASC,EAAkBC,EAAuC,CAChE,GAAIA,aAAmB,OACrB,OAAQC,GAAe,CACrB,IAAMC,EAAeX,EAAUU,CAAE,EAC3BZ,EAASW,EAAQ,KAAKE,CAAY,EACxC,OAAAF,EAAQ,UAAY,EACbX,CACT,EAEF,IAAMQ,EAAM,QAAQ,IAAI,EAClBD,EAAOD,EAAiBK,EAASH,CAAG,EACpCM,EAAUjB,EAAUU,EAAM,CAAE,IAAK,EAAK,CAAC,EAC7C,OAAQK,GAAe,CACrB,IAAMC,EAAeX,EAAUU,CAAE,EACjC,OAAOE,EAAQD,CAAY,CAC7B,CACF,CAaA,SAASE,EACPC,EACAC,EAC0B,CAC1B,GAAI,GAACD,GAAW,CAACC,GAIjB,OAAQC,GACFF,GAAA,MAAAA,EAAS,KAAMG,GAAWA,EAAOD,CAAK,GACjC,GAELD,GAAA,MAAAA,EAAS,KAAME,GAAWA,EAAOD,CAAK,GACjC,GAEF,EAAED,GAAWA,EAAQ,OAAS,EAEzC,CAEA,SAASG,EAAgBD,EAA8C,CACrE,OAAI,OAAOA,GAAW,UAAYA,aAAkB,OAC3C,CACL,QAAS,CAACA,CAAM,CAClB,EAEE,MAAM,QAAQA,CAAM,EACf,CACL,QAASA,CACX,EAEK,CACL,QAASA,EAAO,QAAUE,EAAQF,EAAO,OAAO,EAAI,OACpD,QAASA,EAAO,QAAUE,EAAQF,EAAO,OAAO,EAAI,MACtD,CACF,CAEA,SAASG,EACPH,EAC0B,CAC1B,GAAI,CAACA,EAAQ,OACb,GAAM,CAAE,QAAAH,EAAS,QAAAC,CAAQ,EAAIG,EAAgBD,CAAM,EAC7CI,EAAgBP,GAAA,YAAAA,EAAS,IAAIQ,GAC7BC,EAAgBR,GAAA,YAAAA,EAAS,IAAIO,GACnC,OAAOT,EAAaQ,EAAeE,CAAa,CAClD,CAYO,SAASC,EACdC,EAC0B,CAC1B,IAAMC,EAAiBC,EAAeF,CAAM,EAC5C,OAAOC,EAAkBE,GAAO,CAAC,CAACF,EAAeE,CAAE,EAAI,MACzD,CFpHA,IAAMC,EAEJ,WAAS,SACLC,EACH,WAAS,MAEZ,SAASC,EACPC,EAI6B,CA9B/B,IAAAC,EAAAC,EA+BE,IAAMC,GACJ,eAAgBH,EAAK,MAAQE,GAAAD,EAAAD,EAAK,OAAL,YAAAC,EAAW,aAAX,KAAAC,EAAyB,CAAC,EAAK,CAAC,GAC7D,IAAKE,GAAcA,EAAU,MAAM,KAAK,EAE1C,QAAWA,KAAaC,EACtB,GAAIF,EAAW,SAASC,CAAS,EAC/B,OAAOA,CAGb,CAEA,SAASE,EACPN,EACAO,EACAC,EAC6B,CA9C/B,IAAAP,EA+CE,IAAMG,EAAYL,EAAmBC,CAAI,EACzC,GAAI,CAACI,EACH,OAAO,KAGT,GAAM,CAAE,SAAAK,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYZ,CAAI,EACpDa,EAAYN,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAClD,GAAIM,IAAc,OAChB,MAAM,IAAI,MACR,iEAAgEZ,EAAAM,EAAI,SAAJ,KAAAN,EAAc,EAAE,EAClF,EAGF,OAAOH,EAAM,eACXD,EAAS,WAAW,GAAGgB,CAAS,cAAc,EAAE,EAChD,CACET,IAAc,cACVJ,EACAH,EAAS,aAAagB,CAAS,oBAAoBL,EAAO,IAAIA,CAAI,IAAM,EAAE,IAAI,EAClFX,EAAS,aAAaiB,EAAU,CAAE,SAAAL,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,GAAG,EACrEb,EAAM,iBACJa,EAAc,IAAKH,GACjBV,EAAM,eAAeA,EAAM,WAAWU,CAAI,EAAGV,EAAM,WAAWU,CAAI,CAAC,CACrE,CACF,CACF,CACF,CACF,CAEA,SAASO,EAAgBR,EAA+B,CACtD,MAAO,CACL,kBAAkBS,EAAM,CACtBC,EAAkBD,EAAK,KAAMT,CAAG,CAClC,EAEA,wBAAwBS,EAAM,CAC5B,IAAME,EAAaZ,EACjBU,EAAK,KACLT,EACAS,EAAK,WAAW,KAAK,OAAS,sBAC1BA,EAAK,WAAW,KAAK,GAAG,OAAS,aAC/BA,EAAK,WAAW,KAAK,GAAG,KAE1B,MACN,EACIE,IACFF,EAAK,YAAYE,CAAU,EAC3BF,EAAK,KAAK,EAEd,EAEA,mBAAmBA,EAAM,CAlG7B,IAAAf,EAmGM,IAAMiB,EAAaZ,EACjBU,EAAK,KACLT,GACAN,EAAAe,EAAK,KAAK,KAAV,MAAAf,EAAc,KACVe,EAAK,KAAK,GAAG,KACbA,EAAK,WAAW,KAAK,OAAS,sBAC5BA,EAAK,WAAW,KAAK,GAAG,OAAS,aAC/BA,EAAK,WAAW,KAAK,GAAG,KAE1B,MACR,EACIE,IACFF,EAAK,YAAYE,CAAU,EAC3BF,EAAK,KAAK,EAEd,EAEA,oBAAoBA,EAAM,CApH9B,IAAAf,EAqHM,IAAMD,EAAOgB,EAAK,KACZG,EAAarB,EAAM,mBACvBE,EAAK,GACLA,EAAK,OACLA,EAAK,IACP,EACMkB,EAAaZ,EAAqBa,EAAYZ,GAAKN,EAAAD,EAAK,KAAL,YAAAC,EAAS,IAAI,EAClEiB,GAAclB,EAAK,KACrBgB,EAAK,YACHlB,EAAM,oBAAoB,QAAS,CACjCA,EAAM,mBAAmBE,EAAK,GAAIkB,CAAU,CAC9C,CAAC,CACH,EACAF,EAAK,KAAK,EAEd,EAEA,eAAeA,EAAM,CAtIzB,IAAAf,EAuIM,IAAMD,EAAOgB,EAAK,KAElB,GAAII,EAA0BpB,EAAMO,CAAG,EAAG,CACxC,IAAMc,EAAiBrB,EAAK,UAAU,CAAC,EAEvC,GACEqB,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,GAAM,CAAE,SAAAZ,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYS,CAAc,EAC9DR,EAAYN,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAClD,GAAIM,IAAc,OAChB,MAAM,IAAI,MACR,iEAAgEZ,EAAAM,EAAI,SAAJ,KAAAN,EAAc,EAAE,EAClF,EAGF,IAAMG,EAAYL,EAAmBsB,CAAc,EAEnDL,EAAK,YACHlB,EAAM,eAAeE,EAAK,OAAQ,CAChCF,EAAM,eACJD,EAAS,WAAW,GAAGgB,CAAS,cAAc,EAAE,EAChD,CACET,IAAc,cACVP,EAAS,aAAagB,CAAS,qBAAqB,EACpDQ,EACJxB,EAAS,aAAaiB,EAAU,CAAE,SAAAL,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,GAAG,EACrEb,EAAM,iBACJa,EAAc,IAAKH,GACjBV,EAAM,eACJA,EAAM,WAAWU,CAAI,EACrBV,EAAM,WAAWU,CAAI,CACvB,CACF,CACF,CACF,CACF,CACF,CAAC,CACH,EAEAQ,EAAK,KAAK,CACZ,CACF,CACF,CACF,CACF,CAEe,SAARM,IAAoB,CACzB,MAAO,CACL,QAAS,CACP,QAAQN,EAAMO,EAAO,CA3L3B,IAAAtB,EA6LQ,IAAMuB,GAA4BvB,EAAAsB,EAAc,OAAd,YAAAtB,EAAoB,KAEhDwB,EAAgCF,EAAc,KAE9CG,EAA0BH,EAAc,SAExCI,EAASC,EAAkBH,CAAO,EAKxC,GAJIC,GAAMC,GAAU,EAACA,GAAA,MAAAA,EAASD,KAK5B,EAACD,GAAA,MAAAA,EAAS,iBACVD,GACA,CAACK,EAAkB,KAAMC,GAAQA,EAAI,KAAKN,CAAI,CAAC,EAE/C,OAGF,IAAMjB,EAAe,CACnB,YAAa,IAAI,IACfkB,GAAA,MAAAA,EAAS,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACxD,EACA,OAAQC,CACV,EAEAV,EAAK,SAASD,EAAgBR,CAAG,CAAC,CACpC,CACF,CACF,CACF","names":["Babel","transpileFn","defaultOptions","embedJSON","jsValue","isTgpu","ctx","node","path","tail","typegpuImportRegex","typegpuDynamicImportRegex","typegpuRequireRegex","codeFilterRegexes","gatherTgpuAliases","spec","fnShellFunctionNames","isShellImplementationCall","kernelDirectives","resolve","picomatch","toArray","array","result","BACKSLASH_REGEX","normalize","path","ABSOLUTE_PATH_REGEX","isAbsolute","getMatcherString","glob","cwd","resolved","patternToIdFilter","pattern","id","normalizedId","matcher","createFilter","exclude","include","input","filter","normalizeFilter","toArray","createIdFilter","excludeFilter","patternToIdFilter","includeFilter","createFilterForId","filter","filterFunction","createIdFilter","id","template","types","getKernelDirective","node","_a","_b","directives","directive","kernelDirectives","functionToTranspiled","ctx","name","argNames","body","externalNames","transpileFn","tgpuAlias","embedJSON","functionVisitor","path","gatherTgpuAliases","transpiled","expression","isShellImplementationCall","implementation","babel_default","state","code","options","id","filter","createFilterForId","codeFilterRegexes","reg"]}