unplugin-typegpu 0.1.0-alpha.4 → 0.1.0-alpha.6

package/babel.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunk4LM3AL5Scjs = require('./chunk-4LM3AL5S.cjs');exports.default = _chunk4LM3AL5Scjs.e;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkKPVZYMM6cjs = require('./chunk-KPVZYMM6.cjs');exports.default = _chunkKPVZYMM6cjs.e;
 //# sourceMappingURL=babel.cjs.map

package/babel.cjs.map

@@ -1 +1 @@
-{"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"}
+{"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"}

package/babel.js

@@ -1,2 +1,2 @@
-import{e as a}from"./chunk-ULMT6YNR.js";export{a as default};
+import{e as a}from"./chunk-EBLN5KE7.js";export{a as default};
 //# sourceMappingURL=babel.js.map

package/chunk-7CDU3IXW.js

@@ -0,0 +1,2 @@
+import{a as f,b as d,c as y,d as x,e as P}from"./chunk-EBLN5KE7.js";import{walk as h}from"estree-walker";import w from"magic-string";import{transpileFn as A}from"tinyest-for-wgsl";import{createUnplugin as T}from"unplugin";var N=r=>({name:"unplugin-typegpu",transform(l,s){if(d(r,s,l))return;let a={tgpuAliases:new Set(r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:s},b=this.parse(l,{allowReturnOutsideFunction:!0}),u=[];h(b,{enter(e,m,c,p){let o=e;if(o.type==="ImportDeclaration"&&y(o,a),o.type==="CallExpression"&&x(o,a)){let i=o.arguments[0];i&&i.type!=="TemplateLiteral"&&i.type!=="Literal"&&u.push({varDecl:o,implementation:i})}}});let n=new w(l),g=a.tgpuAliases.values().next().value;if(g===void 0&&u.length>0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${s}`);for(let e of u){let{argNames:m,body:c,externalNames:p}=A(e.implementation);n.appendLeft(e.implementation.start,`${g}.__assignAst(`),n.appendRight(e.implementation.end,`, ${f({argNames:m,body:c,externalNames:p})}`),p.length>0?n.appendRight(e.implementation.end,`, {${p.join(", ")}})`):n.appendRight(e.implementation.end,")")}return{code:n.toString(),map:n.generateMap()}}}),t=T(N),S=t,E=t.vite,R=t.rollup,$=t.rolldown,L=t.webpack,U=t.rspack,_=t.esbuild,I=t.farm,M=P;export{S as a,E as b,R as c,$ as d,L as e,U as f,_ as g,I as h,M as i};
+//# sourceMappingURL=chunk-7CDU3IXW.js.map

package/chunk-7CDU3IXW.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import type { AnyNode, CallExpression } from 'acorn';\nimport { type Node, walk } from 'estree-walker';\nimport MagicString from 'magic-string';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { type UnpluginFactory, createUnplugin } from 'unplugin';\nimport babel from './babel';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\n\ntype TgslFunctionDef = {\n  varDecl: CallExpression;\n  implementation: AnyNode;\n};\n\nconst typegpu: UnpluginFactory<TypegpuPluginOptions> = (\n  options: TypegpuPluginOptions,\n) => ({\n  name: 'unplugin-typegpu' as const,\n  transform(code, id) {\n    if (shouldSkipFile(options, id, code)) {\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    const ast = this.parse(code, {\n      allowReturnOutsideFunction: true,\n    }) as Node;\n\n    const tgslFunctionDefs: TgslFunctionDef[] = [];\n\n    walk(ast, {\n      enter(_node, _parent, prop, index) {\n        const node = _node as AnyNode;\n\n        if (node.type === 'ImportDeclaration') {\n          gatherTgpuAliases(node, ctx);\n        }\n\n        if (node.type === 'CallExpression') {\n          if (isDoesCall(node, ctx)) {\n            const implementation = node.arguments[0];\n\n            if (\n              implementation &&\n              !(implementation.type === 'TemplateLiteral') &&\n              !(implementation.type === 'Literal')\n            ) {\n              tgslFunctionDefs.push({\n                varDecl: node,\n                implementation,\n              });\n            }\n          }\n        }\n      },\n    });\n\n    const magicString = new MagicString(code);\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 expr of tgslFunctionDefs) {\n      const { argNames, body, externalNames } = transpileFn(\n        expr.implementation,\n      );\n\n      // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n      magicString.appendLeft(\n        expr.implementation.start,\n        `${tgpuAlias}.__assignAst(`,\n      );\n      magicString.appendRight(\n        expr.implementation.end,\n        `, ${embedJSON({ argNames, body, externalNames })}`,\n      );\n\n      if (externalNames.length > 0) {\n        magicString.appendRight(\n          expr.implementation.end,\n          `, {${externalNames.join(', ')}})`,\n        );\n      } else {\n        magicString.appendRight(expr.implementation.end, ')');\n      }\n    }\n\n    return {\n      code: magicString.toString(),\n      map: magicString.generateMap(),\n    };\n  },\n});\n\nconst unplugin = createUnplugin(typegpu);\n\nexport type { TypegpuPluginOptions } from './common';\n\nexport default unplugin;\n\nexport const vitePlugin = unplugin.vite;\nexport const rollupPlugin = unplugin.rollup;\nexport const rolldownPlugin = unplugin.rolldown;\nexport const webpackPlugin = unplugin.webpack;\nexport const rspackPlugin = unplugin.rspack;\nexport const esbuildPlugin = unplugin.esbuild;\nexport const farmPlugin = unplugin.farm;\nexport const babelPlugin = babel;\n"],"mappings":"oEACA,OAAoB,QAAAA,MAAY,gBAChC,OAAOC,MAAiB,eACxB,OAAS,eAAAC,MAAmB,mBAC5B,OAA+B,kBAAAC,MAAsB,WAgBrD,IAAMC,EACJC,IACI,CACJ,KAAM,mBACN,UAAUC,EAAMC,EAAI,CAClB,GAAIC,EAAeH,EAASE,EAAID,CAAI,EAClC,OAGF,IAAMG,EAAe,CACnB,YAAa,IAAI,IACfJ,EAAQ,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACvD,EACA,OAAQE,CACV,EAEMG,EAAM,KAAK,MAAMJ,EAAM,CAC3B,2BAA4B,EAC9B,CAAC,EAEKK,EAAsC,CAAC,EAE7CC,EAAKF,EAAK,CACR,MAAMG,EAAOC,EAASC,EAAMC,EAAO,CACjC,IAAMC,EAAOJ,EAMb,GAJII,EAAK,OAAS,qBAChBC,EAAkBD,EAAMR,CAAG,EAGzBQ,EAAK,OAAS,kBACZE,EAAWF,EAAMR,CAAG,EAAG,CACzB,IAAMW,EAAiBH,EAAK,UAAU,CAAC,EAGrCG,GACEA,EAAe,OAAS,mBACxBA,EAAe,OAAS,WAE1BT,EAAiB,KAAK,CACpB,QAASM,EACT,eAAAG,CACF,CAAC,CAEL,CAEJ,CACF,CAAC,EAED,IAAMC,EAAc,IAAIC,EAAYhB,CAAI,EAClCiB,EAAYd,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAElD,GAAIc,IAAc,QAAaZ,EAAiB,OAAS,EACvD,MAAM,IAAI,MACR,gEAAgEJ,CAAE,EACpE,EAGF,QAAWiB,KAAQb,EAAkB,CACnC,GAAM,CAAE,SAAAc,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EACxCJ,EAAK,cACP,EAGAH,EAAY,WACVG,EAAK,eAAe,MACpB,GAAGD,CAAS,eACd,EACAF,EAAY,YACVG,EAAK,eAAe,IACpB,KAAKK,EAAU,CAAE,SAAAJ,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,EACnD,EAEIA,EAAc,OAAS,EACzBN,EAAY,YACVG,EAAK,eAAe,IACpB,MAAMG,EAAc,KAAK,IAAI,CAAC,IAChC,EAEAN,EAAY,YAAYG,EAAK,eAAe,IAAK,GAAG,CAExD,CAEA,MAAO,CACL,KAAMH,EAAY,SAAS,EAC3B,IAAKA,EAAY,YAAY,CAC/B,CACF,CACF,GAEMS,EAAWC,EAAe3B,CAAO,EAIhC4B,EAAQF,EAEFG,EAAaH,EAAS,KACtBI,EAAeJ,EAAS,OACxBK,EAAiBL,EAAS,SAC1BM,EAAgBN,EAAS,QACzBO,EAAeP,EAAS,OACxBQ,EAAgBR,EAAS,QACzBS,EAAaT,EAAS,KACtBU,EAAcC","names":["walk","MagicString","transpileFn","createUnplugin","typegpu","options","code","id","shouldSkipFile","ctx","ast","tgslFunctionDefs","walk","_node","_parent","prop","index","node","gatherTgpuAliases","isDoesCall","implementation","magicString","MagicString","tgpuAlias","expr","argNames","body","externalNames","transpileFn","embedJSON","unplugin","createUnplugin","index_default","vitePlugin","rollupPlugin","rolldownPlugin","webpackPlugin","rspackPlugin","esbuildPlugin","farmPlugin","babelPlugin","babel_default"]}

package/chunk-EBLN5KE7.js

@@ -0,0 +1,2 @@
+import*as o from"@babel/standalone";import{transpileFn as T}from"tinyest-for-wgsl";function f(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function b(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 x=/import.*from\s*['"]typegpu.*['"]/,A=/import\s*\(\s*['"]\s*typegpu.*['"]/,I=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/;function m(e,n,t){if(t&&!(e!=null&&e.include)){if(!x.test(t)&&!I.test(t)&&!A.test(t))return!0}else if(e!=null&&e.include&&e.include!=="all"&&(!n||!e.include.some(r=>r.test(n))))return!0;return!1}function g(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`)}function y(e,n){return e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&(e.callee.property.name==="procedure"&&b(n,e.callee.object)||e.callee.property.name==="does")}var d=o.packages.template,p=o.packages.types;function k(e){return{ImportDeclaration(n){g(n.node,e)},CallExpression(n){var r;let t=n.node;if(y(t,e)){let i=t.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let{argNames:a,body:s,externalNames:l}=T(i),u=e.tgpuAliases.values().next().value;if(u===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(r=e.fileId)!=null?r:""}`);n.replaceWith(p.callExpression(t.callee,[p.callExpression(d.expression(`${u}.__assignAst`)(),[i,d.expression`${f({argNames:a,body:s,externalNames:l})}`(),p.objectExpression(l.map(c=>p.objectProperty(p.identifier(c),p.identifier(c))))])])),n.skip()}}}}}function S(){return{visitor:{Program(e,n){var s;let t=(s=n.file)==null?void 0:s.code,r=n.opts,i=n.filename;if(m(r,i,t))return;let a={tgpuAliases:new Set(r!=null&&r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:i};e.traverse(k(a))}}}}export{f as a,m as b,g as c,y as d,S as e};
+//# sourceMappingURL=chunk-EBLN5KE7.js.map

package/chunk-EBLN5KE7.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/babel.ts","../src/common.ts"],"sourcesContent":["import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as t from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\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 t } }).packages\n  .types;\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isDoesCall(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          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(\n                template.expression(`${tgpuAlias}.__assignAst`)(),\n                [\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        // @ts-ignore\n        const code: string | undefined = state.file?.code;\n        // @ts-ignore\n        const options: TypegpuPluginOptions | undefined = state.opts;\n        // @ts-ignore\n        const id: string | undefined = state.filename;\n\n        if (shouldSkipFile(options, id, code)) {\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';\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 TypegpuPluginOptions {\n  include?: 'all' | RegExp[];\n  forceTgpuAlias?: string;\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\nexport function shouldSkipFile(\n  options: TypegpuPluginOptions | undefined,\n  id: string | undefined,\n  code: string | undefined,\n) {\n  if (code && !options?.include) {\n    if (\n      !typegpuImportRegex.test(code) &&\n      !typegpuRequireRegex.test(code) &&\n      !typegpuDynamicImportRegex.test(code)\n    ) {\n      // No imports to `typegpu` or its sub modules, exiting early.\n      return true;\n    }\n  } else if (\n    options?.include &&\n    options.include !== 'all' &&\n    (!id || !options.include.some((pattern) => pattern.test(id)))\n  ) {\n    return true;\n  }\n\n  return false;\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\nexport function isDoesCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    node.callee.type === 'MemberExpression' &&\n    node.arguments.length === 1 &&\n    node.callee.property.type === 'Identifier' &&\n    ((node.callee.property.name === 'procedure' &&\n      isTgpu(ctx, node.callee.object)) ||\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"],"mappings":"AAAA,UAAYA,MAAW,oBAIvB,OAAS,eAAAC,MAAmB,mBCarB,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,2CAErB,SAASC,EACdC,EACAC,EACAC,EACA,CACA,GAAIA,GAAQ,EAACF,GAAA,MAAAA,EAAS,UACpB,GACE,CAACJ,EAAmB,KAAKM,CAAI,GAC7B,CAACJ,EAAoB,KAAKI,CAAI,GAC9B,CAACL,EAA0B,KAAKK,CAAI,EAGpC,MAAO,WAGTF,GAAA,MAAAA,EAAS,SACTA,EAAQ,UAAY,QACnB,CAACC,GAAM,CAACD,EAAQ,QAAQ,KAAMG,GAAYA,EAAQ,KAAKF,CAAE,CAAC,GAE3D,MAAO,GAGT,MAAO,EACT,CAEO,SAASG,EACdX,EACAD,EACA,CACA,GAAIC,EAAK,OAAO,QAAU,UACxB,QAAWY,KAAQZ,EAAK,WAGpBY,EAAK,OAAS,0BAEbA,EAAK,OAAS,mBACbA,EAAK,SAAS,OAAS,cACvBA,EAAK,SAAS,OAAS,OAEzBb,EAAI,YAAY,IAAIa,EAAK,MAAM,IAAI,EAC1BA,EAAK,OAAS,4BAEvBb,EAAI,YAAY,IAAI,GAAGa,EAAK,MAAM,IAAI,OAAO,CAIrD,CAEO,SAASC,EACdb,EACAD,EACA,CACA,OACEC,EAAK,OAAO,OAAS,oBACrBA,EAAK,UAAU,SAAW,GAC1BA,EAAK,OAAO,SAAS,OAAS,eAC5BA,EAAK,OAAO,SAAS,OAAS,aAC9BF,EAAOC,EAAKC,EAAK,OAAO,MAAM,GAI9BA,EAAK,OAAO,SAAS,OAAS,OAEpC,CDtGA,IAAMc,EAEJ,WAAS,SACLC,EAAgE,WACnE,MAEH,SAASC,EAAgBC,EAA+B,CACtD,MAAO,CACL,kBAAkBC,EAAM,CACtBC,EAAkBD,EAAK,KAAMD,CAAG,CAClC,EAEA,eAAeC,EAAM,CA5BzB,IAAAE,EA6BM,IAAMC,EAAOH,EAAK,KAElB,GAAII,EAAWD,EAAMJ,CAAG,EAAG,CACzB,IAAMM,EAAiBF,EAAK,UAAU,CAAC,EAEvC,GACEE,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,GAAM,CAAE,SAAAC,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYJ,CAAc,EAC9DK,EAAYX,EAAI,YAAY,OAAO,EAAE,KAAK,EAAE,MAClD,GAAIW,IAAc,OAChB,MAAM,IAAI,MACR,iEAAgER,EAAAH,EAAI,SAAJ,KAAAG,EAAc,EAAE,EAClF,EAGFF,EAAK,YACHH,EAAM,eAAeM,EAAK,OAAQ,CAChCN,EAAM,eACJD,EAAS,WAAW,GAAGc,CAAS,cAAc,EAAE,EAChD,CACEL,EACAT,EAAS,aAAae,EAAU,CAAE,SAAAL,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,GAAG,EACrEX,EAAM,iBACJW,EAAc,IAAKI,GACjBf,EAAM,eACJA,EAAM,WAAWe,CAAI,EACrBf,EAAM,WAAWe,CAAI,CACvB,CACF,CACF,CACF,CACF,CACF,CAAC,CACH,EAEAZ,EAAK,KAAK,CACZ,CACF,CACF,CACF,CACF,CAEe,SAARa,GAAoB,CACzB,MAAO,CACL,QAAS,CACP,QAAQb,EAAMc,EAAO,CA7E3B,IAAAZ,EA+EQ,IAAMa,GAA2Bb,EAAAY,EAAM,OAAN,YAAAZ,EAAY,KAEvCc,EAA4CF,EAAM,KAElDG,EAAyBH,EAAM,SAErC,GAAII,EAAeF,EAASC,EAAIF,CAAI,EAClC,OAGF,IAAMhB,EAAe,CACnB,YAAa,IAAI,IACfiB,GAAA,MAAAA,EAAS,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACxD,EACA,OAAQC,CACV,EAEAjB,EAAK,SAASF,EAAgBC,CAAG,CAAC,CACpC,CACF,CACF,CACF","names":["Babel","transpileFn","embedJSON","jsValue","isTgpu","ctx","node","path","tail","typegpuImportRegex","typegpuDynamicImportRegex","typegpuRequireRegex","shouldSkipFile","options","id","code","pattern","gatherTgpuAliases","spec","isDoesCall","template","types","functionVisitor","ctx","path","gatherTgpuAliases","_a","node","isDoesCall","implementation","argNames","body","externalNames","transpileFn","tgpuAlias","embedJSON","name","babel_default","state","code","options","id","shouldSkipFile"]}

package/chunk-KPVZYMM6.cjs

@@ -0,0 +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; } }var _standalone = require('@babel/standalone'); var o = _interopRequireWildcard(_standalone);var _tinyestforwgsl = require('tinyest-for-wgsl');function f(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function b(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 x=/import.*from\s*['"]typegpu.*['"]/,A=/import\s*\(\s*['"]\s*typegpu.*['"]/,I=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/;function m(e,n,t){if(t&&!(e!=null&&e.include)){if(!x.test(t)&&!I.test(t)&&!A.test(t))return!0}else if(e!=null&&e.include&&e.include!=="all"&&(!n||!e.include.some(r=>r.test(n))))return!0;return!1}function g(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`)}function y(e,n){return e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&(e.callee.property.name==="procedure"&&b(n,e.callee.object)||e.callee.property.name==="does")}var d=o.packages.template,p=o.packages.types;function k(e){return{ImportDeclaration(n){g(n.node,e)},CallExpression(n){var r;let t=n.node;if(y(t,e)){let i=t.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let{argNames:a,body:s,externalNames:l}=_tinyestforwgsl.transpileFn.call(void 0, i),u=e.tgpuAliases.values().next().value;if(u===void 0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${(r=e.fileId)!=null?r:""}`);n.replaceWith(p.callExpression(t.callee,[p.callExpression(d.expression(`${u}.__assignAst`)(),[i,d.expression`${f({argNames:a,body:s,externalNames:l})}`(),p.objectExpression(l.map(c=>p.objectProperty(p.identifier(c),p.identifier(c))))])])),n.skip()}}}}}function S(){return{visitor:{Program(e,n){var s;let t=(s=n.file)==null?void 0:s.code,r=n.opts,i=n.filename;if(m(r,i,t))return;let a={tgpuAliases:new Set(r!=null&&r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:i};e.traverse(k(a))}}}}exports.a = f; exports.b = m; exports.c = g; exports.d = y; exports.e = S;
+//# sourceMappingURL=chunk-KPVZYMM6.cjs.map

package/chunk-KPVZYMM6.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/babel.ts","../src/common.ts"],"names":["embedJSON","jsValue","isTgpu","ctx","node","path","tail"],"mappings":"AAAA,wbAAuB,kDAIK,SCaZA,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-KPVZYMM6.cjs","sourcesContent":["import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as t from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\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 t } }).packages\n  .types;\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isDoesCall(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          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(\n                template.expression(`${tgpuAlias}.__assignAst`)(),\n                [\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        // @ts-ignore\n        const code: string | undefined = state.file?.code;\n        // @ts-ignore\n        const options: TypegpuPluginOptions | undefined = state.opts;\n        // @ts-ignore\n        const id: string | undefined = state.filename;\n\n        if (shouldSkipFile(options, id, code)) {\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';\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 TypegpuPluginOptions {\n  include?: 'all' | RegExp[];\n  forceTgpuAlias?: string;\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\nexport function shouldSkipFile(\n  options: TypegpuPluginOptions | undefined,\n  id: string | undefined,\n  code: string | undefined,\n) {\n  if (code && !options?.include) {\n    if (\n      !typegpuImportRegex.test(code) &&\n      !typegpuRequireRegex.test(code) &&\n      !typegpuDynamicImportRegex.test(code)\n    ) {\n      // No imports to `typegpu` or its sub modules, exiting early.\n      return true;\n    }\n  } else if (\n    options?.include &&\n    options.include !== 'all' &&\n    (!id || !options.include.some((pattern) => pattern.test(id)))\n  ) {\n    return true;\n  }\n\n  return false;\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\nexport function isDoesCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    node.callee.type === 'MemberExpression' &&\n    node.arguments.length === 1 &&\n    node.callee.property.type === 'Identifier' &&\n    ((node.callee.property.name === 'procedure' &&\n      isTgpu(ctx, node.callee.object)) ||\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"]}

package/chunk-QVBCNCKX.cjs

@@ -0,0 +1,2 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var _chunkKPVZYMM6cjs = require('./chunk-KPVZYMM6.cjs');var _estreewalker = require('estree-walker');var _magicstring = require('magic-string'); var _magicstring2 = _interopRequireDefault(_magicstring);var _tinyestforwgsl = require('tinyest-for-wgsl');var _unplugin = require('unplugin');var N=r=>({name:"unplugin-typegpu",transform(l,s){if(_chunkKPVZYMM6cjs.b.call(void 0, r,s,l))return;let a={tgpuAliases:new Set(r.forceTgpuAlias?[r.forceTgpuAlias]:[]),fileId:s},b=this.parse(l,{allowReturnOutsideFunction:!0}),u=[];_estreewalker.walk.call(void 0, b,{enter(e,m,c,p){let o=e;if(o.type==="ImportDeclaration"&&_chunkKPVZYMM6cjs.c.call(void 0, o,a),o.type==="CallExpression"&&_chunkKPVZYMM6cjs.d.call(void 0, o,a)){let i=o.arguments[0];i&&i.type!=="TemplateLiteral"&&i.type!=="Literal"&&u.push({varDecl:o,implementation:i})}}});let n=new (0, _magicstring2.default)(l),g=a.tgpuAliases.values().next().value;if(g===void 0&&u.length>0)throw new Error(`No tgpu import found, cannot assign ast to function in file: ${s}`);for(let e of u){let{argNames:m,body:c,externalNames:p}=_tinyestforwgsl.transpileFn.call(void 0, e.implementation);n.appendLeft(e.implementation.start,`${g}.__assignAst(`),n.appendRight(e.implementation.end,`, ${_chunkKPVZYMM6cjs.a.call(void 0, {argNames:m,body:c,externalNames:p})}`),p.length>0?n.appendRight(e.implementation.end,`, {${p.join(", ")}})`):n.appendRight(e.implementation.end,")")}return{code:n.toString(),map:n.generateMap()}}}),t=_unplugin.createUnplugin.call(void 0, N),S= exports.a =t,E= exports.b =t.vite,R= exports.c =t.rollup,$= exports.d =t.rolldown,L= exports.e =t.webpack,U= exports.f =t.rspack,_= exports.g =t.esbuild,I= exports.h =t.farm,M= exports.i =_chunkKPVZYMM6cjs.e;exports.a = S; exports.b = E; exports.c = R; exports.d = $; exports.e = L; exports.f = U; exports.g = _; exports.h = I; exports.i = M;
+//# sourceMappingURL=chunk-QVBCNCKX.cjs.map

package/chunk-QVBCNCKX.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/chunk-QVBCNCKX.cjs","../src/index.ts"],"names":["typegpu","options","code","id","shouldSkipFile","ctx","ast","tgslFunctionDefs","walk","_node","_parent","prop","index","node","gatherTgpuAliases","isDoesCall","implementation","magicString","MagicString","tgpuAlias"],"mappings":"AAAA,iOAAoE,6CCCpC,qGACR,kDACI,oCACyB,IAgB/CA,CAAAA,CACJC,CAAAA,EAAAA,CACI,CACJ,IAAA,CAAM,kBAAA,CACN,SAAA,CAAUC,CAAAA,CAAMC,CAAAA,CAAI,CAClB,EAAA,CAAIC,iCAAAA,CAAeH,CAASE,CAAAA,CAAID,CAAI,CAAA,CAClC,MAAA,CAGF,IAAMG,CAAAA,CAAe,CACnB,WAAA,CAAa,IAAI,GAAA,CACfJ,CAAAA,CAAQ,cAAA,CAAiB,CAACA,CAAAA,CAAQ,cAAc,CAAA,CAAI,CAAC,CACvD,CAAA,CACA,MAAA,CAAQE,CACV,CAAA,CAEMG,CAAAA,CAAM,IAAA,CAAK,KAAA,CAAMJ,CAAAA,CAAM,CAC3B,0BAAA,CAA4B,CAAA,CAC9B,CAAC,CAAA,CAEKK,CAAAA,CAAsC,CAAC,CAAA,CAE7CC,gCAAAA,CAAKF,CAAK,CACR,KAAA,CAAMG,CAAAA,CAAOC,CAAAA,CAASC,CAAAA,CAAMC,CAAAA,CAAO,CACjC,IAAMC,CAAAA,CAAOJ,CAAAA,CAMb,EAAA,CAJII,CAAAA,CAAK,IAAA,GAAS,mBAAA,EAChBC,iCAAAA,CAAkBD,CAAMR,CAAG,CAAA,CAGzBQ,CAAAA,CAAK,IAAA,GAAS,gBAAA,EACZE,iCAAAA,CAAWF,CAAMR,CAAG,CAAA,CAAG,CACzB,IAAMW,CAAAA,CAAiBH,CAAAA,CAAK,SAAA,CAAU,CAAC,CAAA,CAGrCG,CAAAA,EACEA,CAAAA,CAAe,IAAA,GAAS,iBAAA,EACxBA,CAAAA,CAAe,IAAA,GAAS,SAAA,EAE1BT,CAAAA,CAAiB,IAAA,CAAK,CACpB,OAAA,CAASM,CAAAA,CACT,cAAA,CAAAG,CACF,CAAC,CAEL,CAEJ,CACF,CAAC,CAAA,CAED,IAAMC,CAAAA,CAAc,IAAIC,0BAAAA,CAAYhB,CAAI,CAAA,CAClCiB,CAAAA,CAAYd,CAAAA,CAAI,WAAA,CAAY,MAAA,CAAO,CAAA,CAAE,IAAA,CAAK,CAAA,CAAE,KAAA,CAElD,EAAA,CAAIc,CAAAA,GAAc,KAAA,CAAA,EAAaZ,CAAAA,CAAiB,MAAA,CAAS,CAAA,CACvD,MAAM,IAAI,KAAA,CACR,CAAA,6DAAA,EAAgEJ,CAAE,CAAA,CAAA","file":"/Users/marcinhawryluk/Documents/wigsill/packages/unplugin-typegpu/dist/chunk-QVBCNCKX.cjs","sourcesContent":[null,"import type { AnyNode, CallExpression } from 'acorn';\nimport { type Node, walk } from 'estree-walker';\nimport MagicString from 'magic-string';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { type UnpluginFactory, createUnplugin } from 'unplugin';\nimport babel from './babel';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\n\ntype TgslFunctionDef = {\n  varDecl: CallExpression;\n  implementation: AnyNode;\n};\n\nconst typegpu: UnpluginFactory<TypegpuPluginOptions> = (\n  options: TypegpuPluginOptions,\n) => ({\n  name: 'unplugin-typegpu' as const,\n  transform(code, id) {\n    if (shouldSkipFile(options, id, code)) {\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    const ast = this.parse(code, {\n      allowReturnOutsideFunction: true,\n    }) as Node;\n\n    const tgslFunctionDefs: TgslFunctionDef[] = [];\n\n    walk(ast, {\n      enter(_node, _parent, prop, index) {\n        const node = _node as AnyNode;\n\n        if (node.type === 'ImportDeclaration') {\n          gatherTgpuAliases(node, ctx);\n        }\n\n        if (node.type === 'CallExpression') {\n          if (isDoesCall(node, ctx)) {\n            const implementation = node.arguments[0];\n\n            if (\n              implementation &&\n              !(implementation.type === 'TemplateLiteral') &&\n              !(implementation.type === 'Literal')\n            ) {\n              tgslFunctionDefs.push({\n                varDecl: node,\n                implementation,\n              });\n            }\n          }\n        }\n      },\n    });\n\n    const magicString = new MagicString(code);\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 expr of tgslFunctionDefs) {\n      const { argNames, body, externalNames } = transpileFn(\n        expr.implementation,\n      );\n\n      // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n      magicString.appendLeft(\n        expr.implementation.start,\n        `${tgpuAlias}.__assignAst(`,\n      );\n      magicString.appendRight(\n        expr.implementation.end,\n        `, ${embedJSON({ argNames, body, externalNames })}`,\n      );\n\n      if (externalNames.length > 0) {\n        magicString.appendRight(\n          expr.implementation.end,\n          `, {${externalNames.join(', ')}})`,\n        );\n      } else {\n        magicString.appendRight(expr.implementation.end, ')');\n      }\n    }\n\n    return {\n      code: magicString.toString(),\n      map: magicString.generateMap(),\n    };\n  },\n});\n\nconst unplugin = createUnplugin(typegpu);\n\nexport type { TypegpuPluginOptions } from './common';\n\nexport default unplugin;\n\nexport const vitePlugin = unplugin.vite;\nexport const rollupPlugin = unplugin.rollup;\nexport const rolldownPlugin = unplugin.rolldown;\nexport const webpackPlugin = unplugin.webpack;\nexport const rspackPlugin = unplugin.rspack;\nexport const esbuildPlugin = unplugin.esbuild;\nexport const farmPlugin = unplugin.farm;\nexport const babelPlugin = babel;\n"]}

package/esbuild.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');var l=_chunkEDH7U5CGcjs.g;exports.default = l;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');var l=_chunkQVBCNCKXcjs.g;exports.default = l;
 //# sourceMappingURL=esbuild.cjs.map

package/esbuild.cjs.map

@@ -1 +1 @@
-{"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';\n\nexport default esbuildPlugin;\n"]}
+{"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';\n\nexport default esbuildPlugin;\n"]}

package/esbuild.js

@@ -1,2 +1,2 @@
-import{g as e}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.js";var l=e;export{l as default};
+import{g as e}from"./chunk-7CDU3IXW.js";import"./chunk-EBLN5KE7.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 _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');var m=_chunkEDH7U5CGcjs.h;exports.default = m;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');var m=_chunkQVBCNCKXcjs.h;exports.default = m;
 //# sourceMappingURL=farm.cjs.map

package/farm.cjs.map

@@ -1 +1 @@
-{"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';\n\nexport default farmPlugin;\n"]}
+{"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';\n\nexport default farmPlugin;\n"]}

package/farm.js

@@ -1,2 +1,2 @@
-import{h as r}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.js";var m=r;export{m as default};
+import{h as r}from"./chunk-7CDU3IXW.js";import"./chunk-EBLN5KE7.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 _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');exports.babelPlugin = _chunkEDH7U5CGcjs.i; exports.default = _chunkEDH7U5CGcjs.a; exports.esbuildPlugin = _chunkEDH7U5CGcjs.g; exports.farmPlugin = _chunkEDH7U5CGcjs.h; exports.rolldownPlugin = _chunkEDH7U5CGcjs.d; exports.rollupPlugin = _chunkEDH7U5CGcjs.c; exports.rspackPlugin = _chunkEDH7U5CGcjs.f; exports.vitePlugin = _chunkEDH7U5CGcjs.b; exports.webpackPlugin = _chunkEDH7U5CGcjs.e;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');exports.babelPlugin = _chunkQVBCNCKXcjs.i; exports.default = _chunkQVBCNCKXcjs.a; exports.esbuildPlugin = _chunkQVBCNCKXcjs.g; exports.farmPlugin = _chunkQVBCNCKXcjs.h; exports.rolldownPlugin = _chunkQVBCNCKXcjs.d; exports.rollupPlugin = _chunkQVBCNCKXcjs.c; exports.rspackPlugin = _chunkQVBCNCKXcjs.f; exports.vitePlugin = _chunkQVBCNCKXcjs.b; exports.webpackPlugin = _chunkQVBCNCKXcjs.e;
 //# sourceMappingURL=index.cjs.map

package/index.cjs.map

@@ -1 +1 @@
-{"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"}
+{"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"}

package/index.d.cts

@@ -6,6 +6,7 @@ import '@babel/types';
 
 interface TypegpuPluginOptions {
     include?: 'all' | RegExp[];
+    forceTgpuAlias?: string;
 }
 
 declare const unplugin: _unplugin.UnpluginInstance<TypegpuPluginOptions, boolean>;

package/index.d.ts

@@ -6,6 +6,7 @@ import '@babel/types';
 
 interface TypegpuPluginOptions {
     include?: 'all' | RegExp[];
+    forceTgpuAlias?: string;
 }
 
 declare const unplugin: _unplugin.UnpluginInstance<TypegpuPluginOptions, boolean>;

package/index.js

@@ -1,2 +1,2 @@
-import{a,b,c,d,e,f,g,h,i}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.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-7CDU3IXW.js";import"./chunk-EBLN5KE7.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-alpha.4",
+  "version": "0.1.0-alpha.6",
   "description": "Build plugins for TypeGPU, enabling seamless JavaScript -> WGSL transpilation and improved debugging.",
   "keywords": [
     "rollup-plugin",
@@ -84,9 +84,9 @@
   "homepage": "https://typegpu.com",
   "dependencies": {
     "@babel/standalone": "^7.26.6",
-    "tinyest-for-wgsl": "~0.1.0-alpha.4",
     "estree-walker": "^3.0.3",
     "magic-string": "^0.30.11",
+    "tinyest-for-wgsl": "~0.1.0-alpha.7",
     "unplugin": "^2.2.0"
   },
   "main": "./index.js",

package/rolldown.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');var r=_chunkEDH7U5CGcjs.d;exports.default = r;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');var r=_chunkQVBCNCKXcjs.d;exports.default = r;
 //# sourceMappingURL=rolldown.cjs.map

package/rolldown.cjs.map

@@ -1 +1 @@
-{"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';\n\nexport default rolldownPlugin;\n"]}
+{"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';\n\nexport default rolldownPlugin;\n"]}

package/rolldown.js

@@ -1,2 +1,2 @@
-import{d as o}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.js";var r=o;export{r as default};
+import{d as o}from"./chunk-7CDU3IXW.js";import"./chunk-EBLN5KE7.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 _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');var r=_chunkEDH7U5CGcjs.c;exports.default = r;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');var r=_chunkQVBCNCKXcjs.c;exports.default = r;
 //# sourceMappingURL=rollup.cjs.map

package/rollup.cjs.map

@@ -1 +1 @@
-{"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';\n\nexport default rollupPlugin;\n"]}
+{"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';\n\nexport default rollupPlugin;\n"]}

package/rollup.js

@@ -1,2 +1,2 @@
-import{c as l}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.js";var r=l;export{r as default};
+import{c as l}from"./chunk-7CDU3IXW.js";import"./chunk-EBLN5KE7.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 _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');var p=_chunkEDH7U5CGcjs.f;exports.default = p;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');var p=_chunkQVBCNCKXcjs.f;exports.default = p;
 //# sourceMappingURL=rspack.cjs.map

package/rspack.cjs.map

@@ -1 +1 @@
-{"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';\n\nexport default rspackPlugin;\n"]}
+{"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';\n\nexport default rspackPlugin;\n"]}

package/rspack.js

@@ -1,2 +1,2 @@
-import{f as r}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.js";var p=r;export{p as default};
+import{f as r}from"./chunk-7CDU3IXW.js";import"./chunk-EBLN5KE7.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 _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');var i=_chunkEDH7U5CGcjs.b;exports.default = i;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');var i=_chunkQVBCNCKXcjs.b;exports.default = i;
 //# sourceMappingURL=vite.cjs.map

package/vite.cjs.map

@@ -1 +1 @@
-{"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';\n\nexport default vitePlugin;\n"]}
+{"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';\n\nexport default vitePlugin;\n"]}

package/vite.js

@@ -1,2 +1,2 @@
-import{b as t}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.js";var i=t;export{i as default};
+import{b as t}from"./chunk-7CDU3IXW.js";import"./chunk-EBLN5KE7.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 _chunkEDH7U5CGcjs = require('./chunk-EDH7U5CG.cjs');require('./chunk-4LM3AL5S.cjs');var p=_chunkEDH7U5CGcjs.e;exports.default = p;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkQVBCNCKXcjs = require('./chunk-QVBCNCKX.cjs');require('./chunk-KPVZYMM6.cjs');var p=_chunkQVBCNCKXcjs.e;exports.default = p;
 //# sourceMappingURL=webpack.cjs.map

package/webpack.cjs.map

@@ -1 +1 @@
-{"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';\n\nexport default webpackPlugin;\n"]}
+{"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';\n\nexport default webpackPlugin;\n"]}

package/webpack.js

@@ -1,2 +1,2 @@
-import{e}from"./chunk-AS55AP7V.js";import"./chunk-ULMT6YNR.js";var p=e;export{p as default};
+import{e}from"./chunk-7CDU3IXW.js";import"./chunk-EBLN5KE7.js";var p=e;export{p as default};
 //# sourceMappingURL=webpack.js.map

package/chunk-4LM3AL5S.cjs

@@ -1,2 +0,0 @@
-"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; } }var _standalone = require('@babel/standalone'); var o = _interopRequireWildcard(_standalone);var _tinyestforwgsl = require('tinyest-for-wgsl');function u(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function y(e,r){let t="",p=r;for(;;)if(p.type==="MemberExpression"){if(p.property.type!=="Identifier")break;t=t?`${p.property.name}.${t}`:p.property.name,p=p.object}else if(p.type==="Identifier"){t=t?`${p.name}.${t}`:p.name;break}else break;return e.tgpuAliases.has(t)}var d=/import.*from\s*['"]typegpu.*['"]/g,b=/import\s*\(\s*['"]\s*typegpu.*['"]/g,x=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/g;function c(e,r,t){if(t&&!(e!=null&&e.include)){if(!d.test(t)&&!x.test(t)&&!b.test(t))return!0}else if(e!=null&&e.include&&e.include!=="all"&&(!r||!e.include.some(p=>p.test(r))))return!0;return!1}function m(e,r){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"?r.tgpuAliases.add(t.local.name):t.type==="ImportNamespaceSpecifier"&&r.tgpuAliases.add(`${t.local.name}.tgpu`)}function f(e,r){return e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&(e.callee.property.name==="procedure"&&y(r,e.callee.object)||e.callee.property.name==="does")}var g=o.packages.template,n=o.packages.types;function C(e){return{ImportDeclaration(r){m(r.node,e)},CallExpression(r){let t=r.node;if(f(t,e)){let p=t.arguments[0];if(p&&(p.type==="FunctionExpression"||p.type==="ArrowFunctionExpression")){let{argNames:s,body:a,externalNames:i}=_tinyestforwgsl.transpileFn.call(void 0, p);r.replaceWith(n.callExpression(t.callee,[n.callExpression(g.expression("tgpu.__assignAst")(),[p,g.expression`${u({argNames:s,body:a,externalNames:i})}`(),n.objectExpression(i.map(l=>n.objectProperty(n.identifier(l),n.identifier(l))))])])),r.skip()}}}}}function A(){return{visitor:{Program(e,r){var i;let t=(i=r.file)==null?void 0:i.code,p=r.opts,s=r.filename;if(c(p,s,t))return;let a={tgpuAliases:new Set(["tgpu"])};e.traverse(C(a))}}}}exports.a = u; exports.b = c; exports.c = m; exports.d = f; exports.e = A;
-//# sourceMappingURL=chunk-4LM3AL5S.cjs.map

package/chunk-4LM3AL5S.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/babel.ts","../src/common.ts"],"names":["embedJSON","jsValue","isTgpu","ctx","node","path","tail"],"mappings":"AAAA,wbAAuB,kDAIK,SCWZA,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-4LM3AL5S.cjs","sourcesContent":["import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as t from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\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 t } }).packages\n  .types;\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isDoesCall(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\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(template.expression('tgpu.__assignAst')(), [\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          path.skip();\n        }\n      }\n    },\n  };\n}\n\nexport default function () {\n  return {\n    visitor: {\n      Program(path, state) {\n        // @ts-ignore\n        const code: string | undefined = state.file?.code;\n        // @ts-ignore\n        const options: TypegpuPluginOptions | undefined = state.opts;\n        // @ts-ignore\n        const id: string | undefined = state.filename;\n\n        if (shouldSkipFile(options, id, code)) {\n          return;\n        }\n\n        const ctx: Context = {\n          tgpuAliases: new Set(['tgpu']),\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';\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};\n\nexport interface TypegpuPluginOptions {\n  include?: 'all' | RegExp[];\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.*['\"]/g;\nconst typegpuDynamicImportRegex = /import\\s*\\(\\s*['\"]\\s*typegpu.*['\"]/g;\nconst typegpuRequireRegex = /require\\s*\\(\\s*['\"]\\s*typegpu.*['\"]\\s*\\)/g;\n\nexport function shouldSkipFile(\n  options: TypegpuPluginOptions | undefined,\n  id: string | undefined,\n  code: string | undefined,\n) {\n  if (code && !options?.include) {\n    if (\n      !typegpuImportRegex.test(code) &&\n      !typegpuRequireRegex.test(code) &&\n      !typegpuDynamicImportRegex.test(code)\n    ) {\n      // No imports to `typegpu` or its sub modules, exiting early.\n      return true;\n    }\n  } else if (\n    options?.include &&\n    options.include !== 'all' &&\n    (!id || !options.include.some((pattern) => pattern.test(id)))\n  ) {\n    return true;\n  }\n\n  return false;\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\nexport function isDoesCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    node.callee.type === 'MemberExpression' &&\n    node.arguments.length === 1 &&\n    node.callee.property.type === 'Identifier' &&\n    ((node.callee.property.name === 'procedure' &&\n      isTgpu(ctx, node.callee.object)) ||\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"]}

package/chunk-AS55AP7V.js

@@ -1,2 +0,0 @@
-import{a as g,b as m,c,d as f,e as d}from"./chunk-ULMT6YNR.js";import{walk as b}from"estree-walker";import h from"magic-string";import{transpileFn as w}from"tinyest-for-wgsl";import{createUnplugin as F}from"unplugin";var N=y=>({name:"unplugin-typegpu",transform(r,x){if(m(y,x,r))return;let l={tgpuAliases:new Set(["tgpu"])},P=this.parse(r,{allowReturnOutsideFunction:!0}),s=[];b(P,{enter(e,a,u,p){let o=e;if(o.type==="ImportDeclaration"&&c(o,l),o.type==="CallExpression"&&f(o,l)){let i=o.arguments[0];i&&i.type!=="TemplateLiteral"&&i.type!=="Literal"&&s.push({varDecl:o,implementation:i})}}});let n=new h(r);for(let e of s){let{argNames:a,body:u,externalNames:p}=w(e.implementation);n.appendLeft(e.implementation.start,"tgpu.__assignAst("),n.appendRight(e.implementation.end,`, ${g({argNames:a,body:u,externalNames:p})}`),p.length>0?n.appendRight(e.implementation.end,`, {${p.join(", ")}})`):n.appendRight(e.implementation.end,", undefined)")}return{code:n.toString(),map:n.generateMap()}}}),t=F(N),S=t,v=t.vite,R=t.rollup,E=t.rolldown,L=t.webpack,U=t.rspack,_=t.esbuild,M=t.farm,$=d;export{S as a,v as b,R as c,E as d,L as e,U as f,_ as g,M as h,$ as i};
-//# sourceMappingURL=chunk-AS55AP7V.js.map

package/chunk-AS55AP7V.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import type { AnyNode, CallExpression } from 'acorn';\nimport { type Node, walk } from 'estree-walker';\nimport MagicString from 'magic-string';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { type UnpluginFactory, createUnplugin } from 'unplugin';\nimport babel from './babel';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\n\ntype TgslFunctionDef = {\n  varDecl: CallExpression;\n  implementation: AnyNode;\n};\n\nconst typegpu: UnpluginFactory<TypegpuPluginOptions> = (\n  options: TypegpuPluginOptions,\n) => ({\n  name: 'unplugin-typegpu' as const,\n  transform(code, id) {\n    if (shouldSkipFile(options, id, code)) {\n      return;\n    }\n\n    const ctx: Context = {\n      tgpuAliases: new Set(['tgpu']),\n    };\n\n    const ast = this.parse(code, {\n      allowReturnOutsideFunction: true,\n    }) as Node;\n\n    const tgslFunctionDefs: TgslFunctionDef[] = [];\n\n    walk(ast, {\n      enter(_node, _parent, prop, index) {\n        const node = _node as AnyNode;\n\n        if (node.type === 'ImportDeclaration') {\n          gatherTgpuAliases(node, ctx);\n        }\n\n        if (node.type === 'CallExpression') {\n          if (isDoesCall(node, ctx)) {\n            const implementation = node.arguments[0];\n\n            if (\n              implementation &&\n              !(implementation.type === 'TemplateLiteral') &&\n              !(implementation.type === 'Literal')\n            ) {\n              tgslFunctionDefs.push({\n                varDecl: node,\n                implementation,\n              });\n            }\n          }\n        }\n      },\n    });\n\n    const magicString = new MagicString(code);\n\n    for (const expr of tgslFunctionDefs) {\n      const { argNames, body, externalNames } = transpileFn(\n        expr.implementation,\n      );\n\n      // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n      magicString.appendLeft(expr.implementation.start, 'tgpu.__assignAst(');\n      magicString.appendRight(\n        expr.implementation.end,\n        `, ${embedJSON({ argNames, body, externalNames })}`,\n      );\n\n      if (externalNames.length > 0) {\n        magicString.appendRight(\n          expr.implementation.end,\n          `, {${externalNames.join(', ')}})`,\n        );\n      } else {\n        magicString.appendRight(expr.implementation.end, ', undefined)');\n      }\n    }\n\n    return {\n      code: magicString.toString(),\n      map: magicString.generateMap(),\n    };\n  },\n});\n\nconst unplugin = createUnplugin(typegpu);\n\nexport type { TypegpuPluginOptions } from './common';\n\nexport default unplugin;\n\nexport const vitePlugin = unplugin.vite;\nexport const rollupPlugin = unplugin.rollup;\nexport const rolldownPlugin = unplugin.rolldown;\nexport const webpackPlugin = unplugin.webpack;\nexport const rspackPlugin = unplugin.rspack;\nexport const esbuildPlugin = unplugin.esbuild;\nexport const farmPlugin = unplugin.farm;\nexport const babelPlugin = babel;\n"],"mappings":"+DACA,OAAoB,QAAAA,MAAY,gBAChC,OAAOC,MAAiB,eACxB,OAAS,eAAAC,MAAmB,mBAC5B,OAA+B,kBAAAC,MAAsB,WAgBrD,IAAMC,EACJC,IACI,CACJ,KAAM,mBACN,UAAUC,EAAMC,EAAI,CAClB,GAAIC,EAAeH,EAASE,EAAID,CAAI,EAClC,OAGF,IAAMG,EAAe,CACnB,YAAa,IAAI,IAAI,CAAC,MAAM,CAAC,CAC/B,EAEMC,EAAM,KAAK,MAAMJ,EAAM,CAC3B,2BAA4B,EAC9B,CAAC,EAEKK,EAAsC,CAAC,EAE7CC,EAAKF,EAAK,CACR,MAAMG,EAAOC,EAASC,EAAMC,EAAO,CACjC,IAAMC,EAAOJ,EAMb,GAJII,EAAK,OAAS,qBAChBC,EAAkBD,EAAMR,CAAG,EAGzBQ,EAAK,OAAS,kBACZE,EAAWF,EAAMR,CAAG,EAAG,CACzB,IAAMW,EAAiBH,EAAK,UAAU,CAAC,EAGrCG,GACEA,EAAe,OAAS,mBACxBA,EAAe,OAAS,WAE1BT,EAAiB,KAAK,CACpB,QAASM,EACT,eAAAG,CACF,CAAC,CAEL,CAEJ,CACF,CAAC,EAED,IAAMC,EAAc,IAAIC,EAAYhB,CAAI,EAExC,QAAWiB,KAAQZ,EAAkB,CACnC,GAAM,CAAE,SAAAa,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EACxCJ,EAAK,cACP,EAGAF,EAAY,WAAWE,EAAK,eAAe,MAAO,mBAAmB,EACrEF,EAAY,YACVE,EAAK,eAAe,IACpB,KAAKK,EAAU,CAAE,SAAAJ,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,EACnD,EAEIA,EAAc,OAAS,EACzBL,EAAY,YACVE,EAAK,eAAe,IACpB,MAAMG,EAAc,KAAK,IAAI,CAAC,IAChC,EAEAL,EAAY,YAAYE,EAAK,eAAe,IAAK,cAAc,CAEnE,CAEA,MAAO,CACL,KAAMF,EAAY,SAAS,EAC3B,IAAKA,EAAY,YAAY,CAC/B,CACF,CACF,GAEMQ,EAAWC,EAAe1B,CAAO,EAIhC2B,EAAQF,EAEFG,EAAaH,EAAS,KACtBI,EAAeJ,EAAS,OACxBK,EAAiBL,EAAS,SAC1BM,EAAgBN,EAAS,QACzBO,EAAeP,EAAS,OACxBQ,EAAgBR,EAAS,QACzBS,EAAaT,EAAS,KACtBU,EAAcC","names":["walk","MagicString","transpileFn","createUnplugin","typegpu","options","code","id","shouldSkipFile","ctx","ast","tgslFunctionDefs","walk","_node","_parent","prop","index","node","gatherTgpuAliases","isDoesCall","implementation","magicString","MagicString","expr","argNames","body","externalNames","transpileFn","embedJSON","unplugin","createUnplugin","index_default","vitePlugin","rollupPlugin","rolldownPlugin","webpackPlugin","rspackPlugin","esbuildPlugin","farmPlugin","babelPlugin","babel_default"]}

package/chunk-EDH7U5CG.cjs

@@ -1,2 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var _chunk4LM3AL5Scjs = require('./chunk-4LM3AL5S.cjs');var _estreewalker = require('estree-walker');var _magicstring = require('magic-string'); var _magicstring2 = _interopRequireDefault(_magicstring);var _tinyestforwgsl = require('tinyest-for-wgsl');var _unplugin = require('unplugin');var N=y=>({name:"unplugin-typegpu",transform(r,x){if(_chunk4LM3AL5Scjs.b.call(void 0, y,x,r))return;let l={tgpuAliases:new Set(["tgpu"])},P=this.parse(r,{allowReturnOutsideFunction:!0}),s=[];_estreewalker.walk.call(void 0, P,{enter(e,a,u,p){let o=e;if(o.type==="ImportDeclaration"&&_chunk4LM3AL5Scjs.c.call(void 0, o,l),o.type==="CallExpression"&&_chunk4LM3AL5Scjs.d.call(void 0, o,l)){let i=o.arguments[0];i&&i.type!=="TemplateLiteral"&&i.type!=="Literal"&&s.push({varDecl:o,implementation:i})}}});let n=new (0, _magicstring2.default)(r);for(let e of s){let{argNames:a,body:u,externalNames:p}=_tinyestforwgsl.transpileFn.call(void 0, e.implementation);n.appendLeft(e.implementation.start,"tgpu.__assignAst("),n.appendRight(e.implementation.end,`, ${_chunk4LM3AL5Scjs.a.call(void 0, {argNames:a,body:u,externalNames:p})}`),p.length>0?n.appendRight(e.implementation.end,`, {${p.join(", ")}})`):n.appendRight(e.implementation.end,", undefined)")}return{code:n.toString(),map:n.generateMap()}}}),t=_unplugin.createUnplugin.call(void 0, N),S= exports.a =t,v= exports.b =t.vite,R= exports.c =t.rollup,E= exports.d =t.rolldown,L= exports.e =t.webpack,U= exports.f =t.rspack,_= exports.g =t.esbuild,M= exports.h =t.farm,$= exports.i =_chunk4LM3AL5Scjs.e;exports.a = S; exports.b = v; exports.c = R; exports.d = E; exports.e = L; exports.f = U; exports.g = _; exports.h = M; exports.i = $;
-//# sourceMappingURL=chunk-EDH7U5CG.cjs.map

package/chunk-EDH7U5CG.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-EDH7U5CG.cjs","../src/index.ts"],"names":["typegpu","options","code","id","shouldSkipFile","ctx","ast","tgslFunctionDefs","walk","_node","_parent","prop","index","node","gatherTgpuAliases","isDoesCall","implementation","magicString","MagicString","expr","argNames","body","externalNames","transpileFn","embedJSON"],"mappings":"AAAA,iOAA+D,6CCC/B,qGACR,kDACI,oCACyB,IAgB/CA,CAAAA,CACJC,CAAAA,EAAAA,CACI,CACJ,IAAA,CAAM,kBAAA,CACN,SAAA,CAAUC,CAAAA,CAAMC,CAAAA,CAAI,CAClB,EAAA,CAAIC,iCAAAA,CAAeH,CAASE,CAAAA,CAAID,CAAI,CAAA,CAClC,MAAA,CAGF,IAAMG,CAAAA,CAAe,CACnB,WAAA,CAAa,IAAI,GAAA,CAAI,CAAC,MAAM,CAAC,CAC/B,CAAA,CAEMC,CAAAA,CAAM,IAAA,CAAK,KAAA,CAAMJ,CAAAA,CAAM,CAC3B,0BAAA,CAA4B,CAAA,CAC9B,CAAC,CAAA,CAEKK,CAAAA,CAAsC,CAAC,CAAA,CAE7CC,gCAAAA,CAAKF,CAAK,CACR,KAAA,CAAMG,CAAAA,CAAOC,CAAAA,CAASC,CAAAA,CAAMC,CAAAA,CAAO,CACjC,IAAMC,CAAAA,CAAOJ,CAAAA,CAMb,EAAA,CAJII,CAAAA,CAAK,IAAA,GAAS,mBAAA,EAChBC,iCAAAA,CAAkBD,CAAMR,CAAG,CAAA,CAGzBQ,CAAAA,CAAK,IAAA,GAAS,gBAAA,EACZE,iCAAAA,CAAWF,CAAMR,CAAG,CAAA,CAAG,CACzB,IAAMW,CAAAA,CAAiBH,CAAAA,CAAK,SAAA,CAAU,CAAC,CAAA,CAGrCG,CAAAA,EACEA,CAAAA,CAAe,IAAA,GAAS,iBAAA,EACxBA,CAAAA,CAAe,IAAA,GAAS,SAAA,EAE1BT,CAAAA,CAAiB,IAAA,CAAK,CACpB,OAAA,CAASM,CAAAA,CACT,cAAA,CAAAG,CACF,CAAC,CAEL,CAEJ,CACF,CAAC,CAAA,CAED,IAAMC,CAAAA,CAAc,IAAIC,0BAAAA,CAAYhB,CAAI,CAAA,CAExC,GAAA,CAAA,IAAWiB,EAAAA,GAAQZ,CAAAA,CAAkB,CACnC,GAAM,CAAE,QAAA,CAAAa,CAAAA,CAAU,IAAA,CAAAC,CAAAA,CAAM,aAAA,CAAAC,CAAc,CAAA,CAAIC,yCAAAA,CACxCJ,CAAK,cACP,CAAA,CAGAF,CAAAA,CAAY,UAAA,CAAWE,CAAAA,CAAK,cAAA,CAAe,KAAA,CAAO,mBAAmB,CAAA,CACrEF,CAAAA,CAAY,WAAA,CACVE,CAAAA,CAAK,cAAA,CAAe,GAAA,CACpB,CAAA,EAAA,EAAKK,iCAAAA,CAAY,QAAA,CAAAJ,CAAAA,CAAU,IAAA,CAAAC,CAAAA,CAAM,aAAA,CAAAC,CAAc,CAAC,CAAC,CAAA,CAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-EDH7U5CG.cjs","sourcesContent":[null,"import type { AnyNode, CallExpression } from 'acorn';\nimport { type Node, walk } from 'estree-walker';\nimport MagicString from 'magic-string';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { type UnpluginFactory, createUnplugin } from 'unplugin';\nimport babel from './babel';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\n\ntype TgslFunctionDef = {\n  varDecl: CallExpression;\n  implementation: AnyNode;\n};\n\nconst typegpu: UnpluginFactory<TypegpuPluginOptions> = (\n  options: TypegpuPluginOptions,\n) => ({\n  name: 'unplugin-typegpu' as const,\n  transform(code, id) {\n    if (shouldSkipFile(options, id, code)) {\n      return;\n    }\n\n    const ctx: Context = {\n      tgpuAliases: new Set(['tgpu']),\n    };\n\n    const ast = this.parse(code, {\n      allowReturnOutsideFunction: true,\n    }) as Node;\n\n    const tgslFunctionDefs: TgslFunctionDef[] = [];\n\n    walk(ast, {\n      enter(_node, _parent, prop, index) {\n        const node = _node as AnyNode;\n\n        if (node.type === 'ImportDeclaration') {\n          gatherTgpuAliases(node, ctx);\n        }\n\n        if (node.type === 'CallExpression') {\n          if (isDoesCall(node, ctx)) {\n            const implementation = node.arguments[0];\n\n            if (\n              implementation &&\n              !(implementation.type === 'TemplateLiteral') &&\n              !(implementation.type === 'Literal')\n            ) {\n              tgslFunctionDefs.push({\n                varDecl: node,\n                implementation,\n              });\n            }\n          }\n        }\n      },\n    });\n\n    const magicString = new MagicString(code);\n\n    for (const expr of tgslFunctionDefs) {\n      const { argNames, body, externalNames } = transpileFn(\n        expr.implementation,\n      );\n\n      // Wrap the implementation in a call to `tgpu.__assignAst` to associate the AST with the implementation.\n      magicString.appendLeft(expr.implementation.start, 'tgpu.__assignAst(');\n      magicString.appendRight(\n        expr.implementation.end,\n        `, ${embedJSON({ argNames, body, externalNames })}`,\n      );\n\n      if (externalNames.length > 0) {\n        magicString.appendRight(\n          expr.implementation.end,\n          `, {${externalNames.join(', ')}})`,\n        );\n      } else {\n        magicString.appendRight(expr.implementation.end, ', undefined)');\n      }\n    }\n\n    return {\n      code: magicString.toString(),\n      map: magicString.generateMap(),\n    };\n  },\n});\n\nconst unplugin = createUnplugin(typegpu);\n\nexport type { TypegpuPluginOptions } from './common';\n\nexport default unplugin;\n\nexport const vitePlugin = unplugin.vite;\nexport const rollupPlugin = unplugin.rollup;\nexport const rolldownPlugin = unplugin.rolldown;\nexport const webpackPlugin = unplugin.webpack;\nexport const rspackPlugin = unplugin.rspack;\nexport const esbuildPlugin = unplugin.esbuild;\nexport const farmPlugin = unplugin.farm;\nexport const babelPlugin = babel;\n"]}

package/chunk-ULMT6YNR.js

@@ -1,2 +0,0 @@
-import*as o from"@babel/standalone";import{transpileFn as k}from"tinyest-for-wgsl";function u(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function y(e,r){let t="",p=r;for(;;)if(p.type==="MemberExpression"){if(p.property.type!=="Identifier")break;t=t?`${p.property.name}.${t}`:p.property.name,p=p.object}else if(p.type==="Identifier"){t=t?`${p.name}.${t}`:p.name;break}else break;return e.tgpuAliases.has(t)}var d=/import.*from\s*['"]typegpu.*['"]/g,b=/import\s*\(\s*['"]\s*typegpu.*['"]/g,x=/require\s*\(\s*['"]\s*typegpu.*['"]\s*\)/g;function c(e,r,t){if(t&&!(e!=null&&e.include)){if(!d.test(t)&&!x.test(t)&&!b.test(t))return!0}else if(e!=null&&e.include&&e.include!=="all"&&(!r||!e.include.some(p=>p.test(r))))return!0;return!1}function m(e,r){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"?r.tgpuAliases.add(t.local.name):t.type==="ImportNamespaceSpecifier"&&r.tgpuAliases.add(`${t.local.name}.tgpu`)}function f(e,r){return e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&(e.callee.property.name==="procedure"&&y(r,e.callee.object)||e.callee.property.name==="does")}var g=o.packages.template,n=o.packages.types;function C(e){return{ImportDeclaration(r){m(r.node,e)},CallExpression(r){let t=r.node;if(f(t,e)){let p=t.arguments[0];if(p&&(p.type==="FunctionExpression"||p.type==="ArrowFunctionExpression")){let{argNames:s,body:a,externalNames:i}=k(p);r.replaceWith(n.callExpression(t.callee,[n.callExpression(g.expression("tgpu.__assignAst")(),[p,g.expression`${u({argNames:s,body:a,externalNames:i})}`(),n.objectExpression(i.map(l=>n.objectProperty(n.identifier(l),n.identifier(l))))])])),r.skip()}}}}}function A(){return{visitor:{Program(e,r){var i;let t=(i=r.file)==null?void 0:i.code,p=r.opts,s=r.filename;if(c(p,s,t))return;let a={tgpuAliases:new Set(["tgpu"])};e.traverse(C(a))}}}}export{u as a,c as b,m as c,f as d,A as e};
-//# sourceMappingURL=chunk-ULMT6YNR.js.map

package/chunk-ULMT6YNR.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/babel.ts","../src/common.ts"],"sourcesContent":["import * as Babel from '@babel/standalone';\nimport type TemplateGenerator from '@babel/template';\nimport type { TraverseOptions } from '@babel/traverse';\nimport type * as t from '@babel/types';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  type TypegpuPluginOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  isDoesCall,\n  shouldSkipFile,\n} from './common';\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 t } }).packages\n  .types;\n\nfunction functionVisitor(ctx: Context): TraverseOptions {\n  return {\n    ImportDeclaration(path) {\n      gatherTgpuAliases(path.node, ctx);\n    },\n\n    CallExpression(path) {\n      const node = path.node;\n\n      if (isDoesCall(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\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              types.callExpression(template.expression('tgpu.__assignAst')(), [\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          path.skip();\n        }\n      }\n    },\n  };\n}\n\nexport default function () {\n  return {\n    visitor: {\n      Program(path, state) {\n        // @ts-ignore\n        const code: string | undefined = state.file?.code;\n        // @ts-ignore\n        const options: TypegpuPluginOptions | undefined = state.opts;\n        // @ts-ignore\n        const id: string | undefined = state.filename;\n\n        if (shouldSkipFile(options, id, code)) {\n          return;\n        }\n\n        const ctx: Context = {\n          tgpuAliases: new Set(['tgpu']),\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';\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};\n\nexport interface TypegpuPluginOptions {\n  include?: 'all' | RegExp[];\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.*['\"]/g;\nconst typegpuDynamicImportRegex = /import\\s*\\(\\s*['\"]\\s*typegpu.*['\"]/g;\nconst typegpuRequireRegex = /require\\s*\\(\\s*['\"]\\s*typegpu.*['\"]\\s*\\)/g;\n\nexport function shouldSkipFile(\n  options: TypegpuPluginOptions | undefined,\n  id: string | undefined,\n  code: string | undefined,\n) {\n  if (code && !options?.include) {\n    if (\n      !typegpuImportRegex.test(code) &&\n      !typegpuRequireRegex.test(code) &&\n      !typegpuDynamicImportRegex.test(code)\n    ) {\n      // No imports to `typegpu` or its sub modules, exiting early.\n      return true;\n    }\n  } else if (\n    options?.include &&\n    options.include !== 'all' &&\n    (!id || !options.include.some((pattern) => pattern.test(id)))\n  ) {\n    return true;\n  }\n\n  return false;\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\nexport function isDoesCall(\n  node: acorn.CallExpression | babel.CallExpression,\n  ctx: Context,\n) {\n  return (\n    node.callee.type === 'MemberExpression' &&\n    node.arguments.length === 1 &&\n    node.callee.property.type === 'Identifier' &&\n    ((node.callee.property.name === 'procedure' &&\n      isTgpu(ctx, node.callee.object)) ||\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"],"mappings":"AAAA,UAAYA,MAAW,oBAIvB,OAAS,eAAAC,MAAmB,mBCWrB,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,oCACrBC,EAA4B,sCAC5BC,EAAsB,4CAErB,SAASC,EACdC,EACAC,EACAC,EACA,CACA,GAAIA,GAAQ,EAACF,GAAA,MAAAA,EAAS,UACpB,GACE,CAACJ,EAAmB,KAAKM,CAAI,GAC7B,CAACJ,EAAoB,KAAKI,CAAI,GAC9B,CAACL,EAA0B,KAAKK,CAAI,EAGpC,MAAO,WAGTF,GAAA,MAAAA,EAAS,SACTA,EAAQ,UAAY,QACnB,CAACC,GAAM,CAACD,EAAQ,QAAQ,KAAMG,GAAYA,EAAQ,KAAKF,CAAE,CAAC,GAE3D,MAAO,GAGT,MAAO,EACT,CAEO,SAASG,EACdX,EACAD,EACA,CACA,GAAIC,EAAK,OAAO,QAAU,UACxB,QAAWY,KAAQZ,EAAK,WAGpBY,EAAK,OAAS,0BAEbA,EAAK,OAAS,mBACbA,EAAK,SAAS,OAAS,cACvBA,EAAK,SAAS,OAAS,OAEzBb,EAAI,YAAY,IAAIa,EAAK,MAAM,IAAI,EAC1BA,EAAK,OAAS,4BAEvBb,EAAI,YAAY,IAAI,GAAGa,EAAK,MAAM,IAAI,OAAO,CAIrD,CAEO,SAASC,EACdb,EACAD,EACA,CACA,OACEC,EAAK,OAAO,OAAS,oBACrBA,EAAK,UAAU,SAAW,GAC1BA,EAAK,OAAO,SAAS,OAAS,eAC5BA,EAAK,OAAO,SAAS,OAAS,aAC9BF,EAAOC,EAAKC,EAAK,OAAO,MAAM,GAI9BA,EAAK,OAAO,SAAS,OAAS,OAEpC,CDpGA,IAAMc,EAEJ,WAAS,SACLC,EAAgE,WACnE,MAEH,SAASC,EAAgBC,EAA+B,CACtD,MAAO,CACL,kBAAkBC,EAAM,CACtBC,EAAkBD,EAAK,KAAMD,CAAG,CAClC,EAEA,eAAeC,EAAM,CACnB,IAAME,EAAOF,EAAK,KAElB,GAAIG,EAAWD,EAAMH,CAAG,EAAG,CACzB,IAAMK,EAAiBF,EAAK,UAAU,CAAC,EAEvC,GACEE,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,GAAM,CAAE,SAAAC,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYJ,CAAc,EAEpEJ,EAAK,YACHH,EAAM,eAAeK,EAAK,OAAQ,CAChCL,EAAM,eAAeD,EAAS,WAAW,kBAAkB,EAAE,EAAG,CAC9DQ,EACAR,EAAS,aAAaa,EAAU,CAAE,SAAAJ,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC,GAAG,EACrEV,EAAM,iBACJU,EAAc,IAAKG,GACjBb,EAAM,eACJA,EAAM,WAAWa,CAAI,EACrBb,EAAM,WAAWa,CAAI,CACvB,CACF,CACF,CACF,CAAC,CACH,CAAC,CACH,EAEAV,EAAK,KAAK,CACZ,CACF,CACF,CACF,CACF,CAEe,SAARW,GAAoB,CACzB,MAAO,CACL,QAAS,CACP,QAAQX,EAAMY,EAAO,CApE3B,IAAAC,EAsEQ,IAAMC,GAA2BD,EAAAD,EAAM,OAAN,YAAAC,EAAY,KAEvCE,EAA4CH,EAAM,KAElDI,EAAyBJ,EAAM,SAErC,GAAIK,EAAeF,EAASC,EAAIF,CAAI,EAClC,OAGF,IAAMf,EAAe,CACnB,YAAa,IAAI,IAAI,CAAC,MAAM,CAAC,CAC/B,EAEAC,EAAK,SAASF,EAAgBC,CAAG,CAAC,CACpC,CACF,CACF,CACF","names":["Babel","transpileFn","embedJSON","jsValue","isTgpu","ctx","node","path","tail","typegpuImportRegex","typegpuDynamicImportRegex","typegpuRequireRegex","shouldSkipFile","options","id","code","pattern","gatherTgpuAliases","spec","isDoesCall","template","types","functionVisitor","ctx","path","gatherTgpuAliases","node","isDoesCall","implementation","argNames","body","externalNames","transpileFn","embedJSON","name","babel_default","state","_a","code","options","id","shouldSkipFile"]}