unplugin-typegpu 0.1.1 → 0.1.3

package/babel.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkYA33RT2Lcjs = require('./chunk-YA33RT2L.cjs');exports.default = _chunkYA33RT2Lcjs.h;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkG6NIIU55cjs = require('./chunk-G6NIIU55.cjs');exports.default = _chunkG6NIIU55cjs.g;
 //# sourceMappingURL=babel.cjs.map

package/babel.js

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

package/chunk-DO6SOWHC.js

@@ -0,0 +1,6 @@
+import*as f from"@babel/standalone";import{FORMAT_VERSION as h}from"tinyest";import{transpileFn as N}from"tinyest-for-wgsl";var M={include:[/\.m?[jt]sx?$/]};function g(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function m(e,n){let r="",t=n;for(;;)if(t.type==="MemberExpression"){if(t.property.type!=="Identifier")break;r=r?`${t.property.name}.${r}`:t.property.name,t=t.object}else if(t.type==="Identifier"){r=r?`${t.name}.${r}`:t.name;break}else break;return e.tgpuAliases.has(r)}function b(e,n){if(e.source.value==="typegpu")for(let r of e.specifiers)r.type==="ImportDefaultSpecifier"||r.type==="ImportSpecifier"&&r.imported.type==="Identifier"&&r.imported.name==="tgpu"?n.tgpuAliases.add(r.local.name):r.type==="ImportNamespaceSpecifier"&&n.tgpuAliases.add(`${r.local.name}.tgpu`)}var k=["fn","vertexFn","fragmentFn","computeFn"];function y(e,n){return e.callee.type==="CallExpression"&&e.callee.callee.type==="MemberExpression"&&e.callee.callee.property.type==="Identifier"&&k.includes(e.callee.callee.property.name)&&e.arguments.length===1&&(e.callee.callee.object.type==="MemberExpression"?m(n,e.callee.callee.object.object):m(n,e.callee.callee.object))||e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&e.callee.property.name==="does"}var x=["kernel","kernel & js"];function F(e){return`The function "${e!=null?e:"<unnamed>"}" is invokable only on the GPU. If you want to use it on the CPU, mark it with the "kernel & js" directive.`}import{resolve as I}from"path";import v from"picomatch";function E(e){let n=e||[];return Array.isArray(n)?n:[n]}var T=/\\/g;function c(e){return e.replace(T,"/")}var P=/^(?:\/|(?:[A-Z]:)?[/\\|])/i;function C(e){return P.test(e)}function w(e,n){if(e.startsWith("**")||C(e))return c(e);let r=I(n,e);return c(r)}function S(e){if(e instanceof RegExp)return i=>{let s=c(i),l=e.test(s);return e.lastIndex=0,l};let n=process.cwd(),r=w(e,n),t=v(r,{dot:!0});return i=>{let s=c(i);return t(s)}}function D(e,n){if(!(!e&&!n))return r=>e!=null&&e.some(t=>t(r))?!1:n!=null&&n.some(t=>t(r))?!0:!(n&&n.length>0)}function O(e){return typeof e=="string"||e instanceof RegExp?{include:[e]}:Array.isArray(e)?{include:e}:{exclude:e.exclude?E(e.exclude):void 0,include:e.include?E(e.include):void 0}}function $(e){if(!e)return;let{exclude:n,include:r}=O(e),t=n==null?void 0:n.map(S),i=r==null?void 0:r.map(S);return D(t,i)}function A(e){let n=$(e);return n?r=>!!n(r):void 0}var j=f.packages.template,o=f.packages.types;function u(e){var r,t;let n=("directives"in e.body?(t=(r=e.body)==null?void 0:r.directives)!=null?t:[]:[]).map(i=>i.value.value);for(let i of x)if(n.includes(i))return i}function a(e){return o.identifier(e)}function d(e,n,r){if(!n)return null;let{params:t,body:i,externalNames:s}=N(e),l=`{
+    v: ${h},
+    ast: ${g({params:t,body:i,externalNames:s})},
+    externals: {${s.join(", ")}},
+  }`,p=n==="kernel & js"?e:o.arrowFunctionExpression([],o.blockStatement([o.throwStatement(o.newExpression(a("Error"),[o.stringLiteral(F(r))]))]));return o.callExpression(o.arrowFunctionExpression([a("$")],o.logicalExpression("&&",o.callExpression(o.memberExpression(o.assignmentExpression("??=",o.memberExpression(a("globalThis"),a("__TYPEGPU_META__")),o.newExpression(a("WeakMap"),[])),a("set")),[o.assignmentExpression("=",o.memberExpression(a("$"),a("f")),p),j.expression`${l}`()]),o.memberExpression(a("$"),a("f")))),[o.objectExpression([])])}function R(e){return{ImportDeclaration(n){b(n.node,e)},ArrowFunctionExpression(n){let r=d(n.node,u(n.node),n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);r&&(n.replaceWith(r),n.skip())},FunctionExpression(n){var t;let r=d(n.node,u(n.node),(t=n.node.id)!=null&&t.name?n.node.id.name:n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);r&&(n.replaceWith(r),n.skip())},FunctionDeclaration(n){var s;let r=n.node,t=o.functionExpression(r.id,r.params,r.body),i=d(t,u(n.node),(s=r.id)==null?void 0:s.name);i&&r.id&&(n.replaceWith(o.variableDeclaration("const",[o.variableDeclarator(r.id,i)])),n.skip())},CallExpression(n){var t;let r=n.node;if(y(r,e)){let i=r.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let s=d(i,(t=u(i))!=null?t:"kernel");n.replaceWith(o.callExpression(r.callee,[s])),n.skip()}}}}}function V(){return{visitor:{Program(e,n){var p;let r=(p=n.file)==null?void 0:p.code,t=n.opts,i=n.filename,s=A(t);if(i&&s&&!(s!=null&&s(i)))return;let l={tgpuAliases:new Set(t!=null&&t.forceTgpuAlias?[t.forceTgpuAlias]:[]),fileId:i};e.traverse(R(l))}}}}export{M as a,g as b,b as c,y as d,x as e,F as f,V as g};
+//# sourceMappingURL=chunk-DO6SOWHC.js.map

package/chunk-DO6SOWHC.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 { FORMAT_VERSION } from 'tinyest';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  embedJSON,\n  gatherTgpuAliases,\n  getErrorMessage,\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 i(identifier: string): babel.Identifier {\n  return types.identifier(identifier);\n}\n\nfunction functionToTranspiled(\n  node: babel.ArrowFunctionExpression | babel.FunctionExpression,\n  directive: KernelDirective | undefined,\n  name?: string | undefined,\n): babel.CallExpression | null {\n  if (!directive) {\n    return null;\n  }\n\n  const { params, body, externalNames } = transpileFn(node);\n\n  const metadata = `{\n    v: ${FORMAT_VERSION},\n    ast: ${embedJSON({ params, body, externalNames })},\n    externals: {${externalNames.join(', ')}},\n  }`;\n\n  const jsImpl = directive === 'kernel & js'\n    ? node\n    : types.arrowFunctionExpression(\n      [],\n      types.blockStatement(\n        [types.throwStatement(\n          types.newExpression(i('Error'), [\n            types.stringLiteral(getErrorMessage(name)),\n          ]),\n        )],\n      ),\n    );\n\n  return types.callExpression(\n    types.arrowFunctionExpression(\n      [i('$')],\n      types.logicalExpression(\n        '&&',\n        types.callExpression(\n          types.memberExpression(\n            types.assignmentExpression(\n              '??=',\n              types.memberExpression(i('globalThis'), i('__TYPEGPU_META__')),\n              types.newExpression(i('WeakMap'), []),\n            ),\n            i('set'),\n          ),\n          [\n            types.assignmentExpression(\n              '=',\n              types.memberExpression(i('$'), i('f')),\n              jsImpl,\n            ),\n            template.expression`${metadata}`(),\n          ],\n        ),\n        types.memberExpression(i('$'), i('f')),\n      ),\n    ),\n    [types.objectExpression([])],\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        getKernelDirective(path.node),\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        getKernelDirective(path.node),\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(\n        expression,\n        getKernelDirective(path.node),\n        node.id?.name,\n      );\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 transpiled = functionToTranspiled(\n            implementation,\n            getKernelDirective(implementation) ?? 'kernel',\n          ) as babel.CallExpression;\n\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              transpiled,\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        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\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\nexport function getErrorMessage(name: string | undefined) {\n  return `The function \"${\n    name ?? '<unnamed>'\n  }\" is invokable only on the GPU. If you want to use it on the CPU, mark it with the \"kernel & js\" directive.`;\n}\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":"AAAA,UAAYA,MAAW,oBAIvB,OAAS,kBAAAC,MAAsB,UAC/B,OAAS,eAAAC,MAAmB,mBCerB,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,CAEO,SAASE,EACdH,EACAD,EACA,CACA,GAAIC,EAAK,OAAO,QAAU,UACxB,QAAWI,KAAQJ,EAAK,WAGpBI,EAAK,OAAS,0BAEbA,EAAK,OAAS,mBACbA,EAAK,SAAS,OAAS,cACvBA,EAAK,SAAS,OAAS,OAEzBL,EAAI,YAAY,IAAIK,EAAK,MAAM,IAAI,EAC1BA,EAAK,OAAS,4BAEvBL,EAAI,YAAY,IAAI,GAAGK,EAAK,MAAM,IAAI,OAAO,CAIrD,CAEA,IAAMC,EAAuB,CAAC,KAAM,WAAY,aAAc,WAAW,EAElE,SAASC,EACdN,EACAD,EACA,CACA,OACGC,EAAK,OAAO,OAAS,kBACpBA,EAAK,OAAO,OAAO,OAAS,oBAC5BA,EAAK,OAAO,OAAO,SAAS,OAAS,cACrCK,EAAqB,SAASL,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,IAAMO,EAAmB,CAAC,SAAU,aAAa,EAGjD,SAASC,EAAgBC,EAA0B,CACxD,MAAO,iBACLA,GAAA,KAAAA,EAAQ,WACV,6GACF,CC9GA,OAAS,WAAAC,MAAe,OACxB,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,CFnHA,IAAMC,EAEJ,WAAS,SACLC,EACH,WAAS,MAEZ,SAASC,EACPC,EAI6B,CA/B/B,IAAAC,EAAAC,EAgCE,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,EAAEC,EAAsC,CAC/C,OAAOT,EAAM,WAAWS,CAAU,CACpC,CAEA,SAASC,EACPR,EACAI,EACAK,EAC6B,CAC7B,GAAI,CAACL,EACH,OAAO,KAGT,GAAM,CAAE,OAAAM,EAAQ,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYb,CAAI,EAElDc,EAAW;AAAA,SACVC,CAAc;AAAA,WACZC,EAAU,CAAE,OAAAN,EAAQ,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC;AAAA,kBACnCA,EAAc,KAAK,IAAI,CAAC;AAAA,KAGlCK,EAASb,IAAc,cACzBJ,EACAF,EAAM,wBACN,CAAC,EACDA,EAAM,eACJ,CAACA,EAAM,eACLA,EAAM,cAAcQ,EAAE,OAAO,EAAG,CAC9BR,EAAM,cAAcoB,EAAgBT,CAAI,CAAC,CAC3C,CAAC,CACH,CAAC,CACH,CACF,EAEF,OAAOX,EAAM,eACXA,EAAM,wBACJ,CAACQ,EAAE,GAAG,CAAC,EACPR,EAAM,kBACJ,KACAA,EAAM,eACJA,EAAM,iBACJA,EAAM,qBACJ,MACAA,EAAM,iBAAiBQ,EAAE,YAAY,EAAGA,EAAE,kBAAkB,CAAC,EAC7DR,EAAM,cAAcQ,EAAE,SAAS,EAAG,CAAC,CAAC,CACtC,EACAA,EAAE,KAAK,CACT,EACA,CACER,EAAM,qBACJ,IACAA,EAAM,iBAAiBQ,EAAE,GAAG,EAAGA,EAAE,GAAG,CAAC,EACrCW,CACF,EACApB,EAAS,aAAaiB,CAAQ,GAAG,CACnC,CACF,EACAhB,EAAM,iBAAiBQ,EAAE,GAAG,EAAGA,EAAE,GAAG,CAAC,CACvC,CACF,EACA,CAACR,EAAM,iBAAiB,CAAC,CAAC,CAAC,CAC7B,CACF,CAEA,SAASqB,EAAgBC,EAA+B,CACtD,MAAO,CACL,kBAAkBC,EAAM,CACtBC,EAAkBD,EAAK,KAAMD,CAAG,CAClC,EAEA,wBAAwBC,EAAM,CAC5B,IAAME,EAAaf,EACjBa,EAAK,KACLtB,EAAmBsB,EAAK,IAAI,EAC5BA,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,CAjI7B,IAAApB,EAkIM,IAAMsB,EAAaf,EACjBa,EAAK,KACLtB,EAAmBsB,EAAK,IAAI,GAC5BpB,EAAAoB,EAAK,KAAK,KAAV,MAAApB,EAAc,KACVoB,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,CAnJ9B,IAAApB,EAoJM,IAAMD,EAAOqB,EAAK,KACZG,EAAa1B,EAAM,mBACvBE,EAAK,GACLA,EAAK,OACLA,EAAK,IACP,EACMuB,EAAaf,EACjBgB,EACAzB,EAAmBsB,EAAK,IAAI,GAC5BpB,EAAAD,EAAK,KAAL,YAAAC,EAAS,IACX,EACIsB,GAAcvB,EAAK,KACrBqB,EAAK,YACHvB,EAAM,oBAAoB,QAAS,CACjCA,EAAM,mBAAmBE,EAAK,GAAIuB,CAAU,CAC9C,CAAC,CACH,EACAF,EAAK,KAAK,EAEd,EAEA,eAAeA,EAAM,CAzKzB,IAAApB,EA0KM,IAAMD,EAAOqB,EAAK,KAElB,GAAII,EAA0BzB,EAAMoB,CAAG,EAAG,CACxC,IAAMM,EAAiB1B,EAAK,UAAU,CAAC,EAEvC,GACE0B,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,IAAMH,EAAaf,EACjBkB,GACAzB,EAAAF,EAAmB2B,CAAc,IAAjC,KAAAzB,EAAsC,QACxC,EAEAoB,EAAK,YACHvB,EAAM,eAAeE,EAAK,OAAQ,CAChCuB,CACF,CAAC,CACH,EAEAF,EAAK,KAAK,CACZ,CACF,CACF,CACF,CACF,CAEe,SAARM,GAAoB,CACzB,MAAO,CACL,QAAS,CACP,QAAQN,EAAMO,EAAO,CAzM3B,IAAA3B,EA2MQ,IAAM4B,GAA4B5B,EAAA2B,EAAc,OAAd,YAAA3B,EAAoB,KAEhD6B,EAAgCF,EAAc,KAE9CG,EAA0BH,EAAc,SAExCI,EAASC,EAAkBH,CAAO,EACxC,GAAIC,GAAMC,GAAU,EAACA,GAAA,MAAAA,EAASD,IAC5B,OAGF,IAAMX,EAAe,CACnB,YAAa,IAAI,IACfU,GAAA,MAAAA,EAAS,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACxD,EACA,OAAQC,CACV,EAEAV,EAAK,SAASF,EAAgBC,CAAG,CAAC,CACpC,CACF,CACF,CACF","names":["Babel","FORMAT_VERSION","transpileFn","defaultOptions","embedJSON","jsValue","isTgpu","ctx","node","path","tail","gatherTgpuAliases","spec","fnShellFunctionNames","isShellImplementationCall","kernelDirectives","getErrorMessage","name","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","i","identifier","functionToTranspiled","name","params","body","externalNames","transpileFn","metadata","FORMAT_VERSION","embedJSON","jsImpl","getErrorMessage","functionVisitor","ctx","path","gatherTgpuAliases","transpiled","expression","isShellImplementationCall","implementation","babel_default","state","code","options","id","filter","createFilterForId"]}

package/chunk-G6NIIU55.cjs

@@ -0,0 +1,6 @@
+"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 _standalone = require('@babel/standalone'); var f = _interopRequireWildcard(_standalone);var _tinyest = require('tinyest');var _tinyestforwgsl = require('tinyest-for-wgsl');var M={include:[/\.m?[jt]sx?$/]};function g(e){return JSON.stringify(e).replace(/\u2028/g,"\\u2028").replace(/\u2029/g,"\\u2029")}function m(e,n){let r="",t=n;for(;;)if(t.type==="MemberExpression"){if(t.property.type!=="Identifier")break;r=r?`${t.property.name}.${r}`:t.property.name,t=t.object}else if(t.type==="Identifier"){r=r?`${t.name}.${r}`:t.name;break}else break;return e.tgpuAliases.has(r)}function b(e,n){if(e.source.value==="typegpu")for(let r of e.specifiers)r.type==="ImportDefaultSpecifier"||r.type==="ImportSpecifier"&&r.imported.type==="Identifier"&&r.imported.name==="tgpu"?n.tgpuAliases.add(r.local.name):r.type==="ImportNamespaceSpecifier"&&n.tgpuAliases.add(`${r.local.name}.tgpu`)}var k=["fn","vertexFn","fragmentFn","computeFn"];function y(e,n){return e.callee.type==="CallExpression"&&e.callee.callee.type==="MemberExpression"&&e.callee.callee.property.type==="Identifier"&&k.includes(e.callee.callee.property.name)&&e.arguments.length===1&&(e.callee.callee.object.type==="MemberExpression"?m(n,e.callee.callee.object.object):m(n,e.callee.callee.object))||e.callee.type==="MemberExpression"&&e.arguments.length===1&&e.callee.property.type==="Identifier"&&e.callee.property.name==="does"}var x=["kernel","kernel & js"];function F(e){return`The function "${e!=null?e:"<unnamed>"}" is invokable only on the GPU. If you want to use it on the CPU, mark it with the "kernel & js" directive.`}var _path = require('path');var _picomatch = require('picomatch'); var _picomatch2 = _interopRequireDefault(_picomatch);function E(e){let n=e||[];return Array.isArray(n)?n:[n]}var T=/\\/g;function c(e){return e.replace(T,"/")}var P=/^(?:\/|(?:[A-Z]:)?[/\\|])/i;function C(e){return P.test(e)}function w(e,n){if(e.startsWith("**")||C(e))return c(e);let r=_path.resolve.call(void 0, n,e);return c(r)}function S(e){if(e instanceof RegExp)return i=>{let s=c(i),l=e.test(s);return e.lastIndex=0,l};let n=process.cwd(),r=w(e,n),t=_picomatch2.default.call(void 0, r,{dot:!0});return i=>{let s=c(i);return t(s)}}function D(e,n){if(!(!e&&!n))return r=>e!=null&&e.some(t=>t(r))?!1:n!=null&&n.some(t=>t(r))?!0:!(n&&n.length>0)}function O(e){return typeof e=="string"||e instanceof RegExp?{include:[e]}:Array.isArray(e)?{include:e}:{exclude:e.exclude?E(e.exclude):void 0,include:e.include?E(e.include):void 0}}function $(e){if(!e)return;let{exclude:n,include:r}=O(e),t=n==null?void 0:n.map(S),i=r==null?void 0:r.map(S);return D(t,i)}function A(e){let n=$(e);return n?r=>!!n(r):void 0}var j=f.packages.template,o=f.packages.types;function u(e){var r,t;let n=("directives"in e.body?(t=(r=e.body)==null?void 0:r.directives)!=null?t:[]:[]).map(i=>i.value.value);for(let i of x)if(n.includes(i))return i}function a(e){return o.identifier(e)}function d(e,n,r){if(!n)return null;let{params:t,body:i,externalNames:s}=_tinyestforwgsl.transpileFn.call(void 0, e),l=`{
+    v: ${_tinyest.FORMAT_VERSION},
+    ast: ${g({params:t,body:i,externalNames:s})},
+    externals: {${s.join(", ")}},
+  }`,p=n==="kernel & js"?e:o.arrowFunctionExpression([],o.blockStatement([o.throwStatement(o.newExpression(a("Error"),[o.stringLiteral(F(r))]))]));return o.callExpression(o.arrowFunctionExpression([a("$")],o.logicalExpression("&&",o.callExpression(o.memberExpression(o.assignmentExpression("??=",o.memberExpression(a("globalThis"),a("__TYPEGPU_META__")),o.newExpression(a("WeakMap"),[])),a("set")),[o.assignmentExpression("=",o.memberExpression(a("$"),a("f")),p),j.expression`${l}`()]),o.memberExpression(a("$"),a("f")))),[o.objectExpression([])])}function R(e){return{ImportDeclaration(n){b(n.node,e)},ArrowFunctionExpression(n){let r=d(n.node,u(n.node),n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);r&&(n.replaceWith(r),n.skip())},FunctionExpression(n){var t;let r=d(n.node,u(n.node),(t=n.node.id)!=null&&t.name?n.node.id.name:n.parentPath.node.type==="VariableDeclarator"&&n.parentPath.node.id.type==="Identifier"?n.parentPath.node.id.name:void 0);r&&(n.replaceWith(r),n.skip())},FunctionDeclaration(n){var s;let r=n.node,t=o.functionExpression(r.id,r.params,r.body),i=d(t,u(n.node),(s=r.id)==null?void 0:s.name);i&&r.id&&(n.replaceWith(o.variableDeclaration("const",[o.variableDeclarator(r.id,i)])),n.skip())},CallExpression(n){var t;let r=n.node;if(y(r,e)){let i=r.arguments[0];if(i&&(i.type==="FunctionExpression"||i.type==="ArrowFunctionExpression")){let s=d(i,(t=u(i))!=null?t:"kernel");n.replaceWith(o.callExpression(r.callee,[s])),n.skip()}}}}}function V(){return{visitor:{Program(e,n){var p;let r=(p=n.file)==null?void 0:p.code,t=n.opts,i=n.filename,s=A(t);if(i&&s&&!(s!=null&&s(i)))return;let l={tgpuAliases:new Set(t!=null&&t.forceTgpuAlias?[t.forceTgpuAlias]:[]),fileId:i};e.traverse(R(l))}}}}exports.a = M; exports.b = g; exports.c = b; exports.d = y; exports.e = x; exports.f = F; exports.g = V;
+//# sourceMappingURL=chunk-G6NIIU55.cjs.map

package/chunk-G6NIIU55.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/babel.ts","../src/common.ts"],"names":["defaultOptions","embedJSON","jsValue","isTgpu","ctx","node","path","tail"],"mappings":"AAAA,whBAAuB,kCAIQ,kDACH,ICefA,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;ADcG,OAAA;AAC8B,SAAA;AACX,gBAAA;AA0BtB,GAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-G6NIIU55.cjs","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 { FORMAT_VERSION } from 'tinyest';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport {\n  type Context,\n  embedJSON,\n  gatherTgpuAliases,\n  getErrorMessage,\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 i(identifier: string): babel.Identifier {\n  return types.identifier(identifier);\n}\n\nfunction functionToTranspiled(\n  node: babel.ArrowFunctionExpression | babel.FunctionExpression,\n  directive: KernelDirective | undefined,\n  name?: string | undefined,\n): babel.CallExpression | null {\n  if (!directive) {\n    return null;\n  }\n\n  const { params, body, externalNames } = transpileFn(node);\n\n  const metadata = `{\n    v: ${FORMAT_VERSION},\n    ast: ${embedJSON({ params, body, externalNames })},\n    externals: {${externalNames.join(', ')}},\n  }`;\n\n  const jsImpl = directive === 'kernel & js'\n    ? node\n    : types.arrowFunctionExpression(\n      [],\n      types.blockStatement(\n        [types.throwStatement(\n          types.newExpression(i('Error'), [\n            types.stringLiteral(getErrorMessage(name)),\n          ]),\n        )],\n      ),\n    );\n\n  return types.callExpression(\n    types.arrowFunctionExpression(\n      [i('$')],\n      types.logicalExpression(\n        '&&',\n        types.callExpression(\n          types.memberExpression(\n            types.assignmentExpression(\n              '??=',\n              types.memberExpression(i('globalThis'), i('__TYPEGPU_META__')),\n              types.newExpression(i('WeakMap'), []),\n            ),\n            i('set'),\n          ),\n          [\n            types.assignmentExpression(\n              '=',\n              types.memberExpression(i('$'), i('f')),\n              jsImpl,\n            ),\n            template.expression`${metadata}`(),\n          ],\n        ),\n        types.memberExpression(i('$'), i('f')),\n      ),\n    ),\n    [types.objectExpression([])],\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        getKernelDirective(path.node),\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        getKernelDirective(path.node),\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(\n        expression,\n        getKernelDirective(path.node),\n        node.id?.name,\n      );\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 transpiled = functionToTranspiled(\n            implementation,\n            getKernelDirective(implementation) ?? 'kernel',\n          ) as babel.CallExpression;\n\n          path.replaceWith(\n            types.callExpression(node.callee, [\n              transpiled,\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        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\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\nexport function getErrorMessage(name: string | undefined) {\n  return `The function \"${\n    name ?? '<unnamed>'\n  }\" is invokable only on the GPU. If you want to use it on the CPU, mark it with the \"kernel & js\" directive.`;\n}\n"]}

package/chunk-LI2BURW4.cjs

@@ -0,0 +1,10 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var _chunkG6NIIU55cjs = require('./chunk-G6NIIU55.cjs');var _defu = require('defu'); var _defu2 = _interopRequireDefault(_defu);var _estreewalker = require('estree-walker');var _magicstringast = require('magic-string-ast');var _tinyest = require('tinyest');var _tinyestforwgsl = require('tinyest-for-wgsl');var _unplugin = require('unplugin');function D(c){if(c.body.type==="BlockStatement"){for(let n of c.body.body)if(n.type==="ExpressionStatement"&&_chunkG6NIIU55cjs.e.includes(n.directive))return n.directive}}function E(c){let n=structuredClone(c);return n.body.type==="BlockStatement"&&(n.body.body=n.body.body.filter(s=>!(s.type==="ExpressionStatement"&&s.directive&&_chunkG6NIIU55cjs.e.includes(s.directive)))),n}var r=_unplugin.createUnplugin.call(void 0, c=>{let n=_defu2.default.call(void 0, c,_chunkG6NIIU55cjs.a);return{name:"unplugin-typegpu",enforce:n.enforce,transform:{filter:{id:n},handler(s,p){let g={tgpuAliases:new Set(n.forceTgpuAlias?[n.forceTgpuAlias]:[]),fileId:p},k=this.parse(s,{allowReturnOutsideFunction:!0}),l=[],u=new (0, _magicstringast.MagicStringAST)(s);_estreewalker.walk.call(void 0, k,{enter(i,t,m,y){var a;let e=i;if(e.type==="ImportDeclaration"&&_chunkG6NIIU55cjs.c.call(void 0, e,g),e.type==="CallExpression"&&_chunkG6NIIU55cjs.d.call(void 0, e,g)){let o=e.arguments[0];if(o&&(o.type==="FunctionExpression"||o.type==="ArrowFunctionExpression")){let f=D(o);l.push({def:E(o),removeJsImplementation:f!=="kernel & js"}),this.skip()}}if(e.type==="ArrowFunctionExpression"||e.type==="FunctionExpression"||e.type==="FunctionDeclaration"){let o=D(e);o&&(l.push({def:E(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())}}});for(let{def:i,name:t,removeJsImplementation:m}of l){let{params:y,body:e,externalNames:a}=_tinyestforwgsl.transpileFn.call(void 0, i),o=i.type==="FunctionDeclaration";o&&t&&s.slice(0,i.start).search(new RegExp(`(?<![\\w_.])${t}(?![\\w_])`))!==-1&&console.warn(`File ${p}: function "${t}", containing ${m?"kernel":"kernel & js"} directive, might have been referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);let f=`{
+              v: ${_tinyest.FORMAT_VERSION},
+              ast: ${_chunkG6NIIU55cjs.b.call(void 0, {params:y,body:e,externalNames:a})},
+              externals: {${a.join(", ")}},
+            }`;u.appendLeft(i.start,`${o&&t?`const ${t} = `:""}
+              (($) => ((globalThis.__TYPEGPU_META__ ??= new WeakMap()).set(
+                $.f = (`).appendRight(i.end,`) , ${f}) && $.f))({})`),m&&u.overwriteNode(i,`() => {
+                  throw new Error(\`${_chunkG6NIIU55cjs.f.call(void 0, t)}\`);
+                }`)}return _magicstringast.generateTransform.call(void 0, u,p)}}}}),B= exports.a =r,V= exports.b =r.vite,_= exports.c =r.rollup,G= exports.d =r.rolldown,L= exports.e =r.webpack,W= exports.f =r.rspack,Y= exports.g =r.esbuild,q= exports.h =r.farm,z= exports.i =_chunkG6NIIU55cjs.g;exports.a = B; exports.b = V; exports.c = _; exports.d = G; exports.e = L; exports.f = W; exports.g = Y; exports.h = q; exports.i = z;
+//# sourceMappingURL=chunk-LI2BURW4.cjs.map

package/chunk-LI2BURW4.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-LI2BURW4.cjs","../src/index.ts"],"names":["getKernelDirective","node","statement","kernelDirectives","removeKernelDirective","cloned","typegpu","createUnplugin","rawOptions","options","defu","defaultOptions","code","id","ctx","ast","tgslFunctionDefs","magicString","MagicStringAST","walk","_node","_parent","prop","index","_a","gatherTgpuAliases","isShellImplementationCall","implementation","directive","def","name","removeJsImplementation","params","body","externalNames","transpileFn","isFunctionStatement","metadata","FORMAT_VERSION","embedJSON"],"mappings":"AAAA,iOAA6E,wECC5D,6CACe,kDACkB,kCACnB,kDACH,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,CAAQ,CACN,EAAA,CAAIA,CACN,CAAA,CACA,OAAA,CAAQG,CAAAA,CAAMC,CAAAA,CAAI,CAChB,IAAMC,CAAAA,CAAe,CACnB,WAAA,CAAa,IAAI,GAAA,CACfL,CAAAA,CAAQ,cAAA,CAAiB,CAACA,CAAAA,CAAQ,cAAc,CAAA,CAAI,CAAC,CACvD,CAAA,CACA,MAAA,CAAQI,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,IAAMvB,CAAAA,CAAOmB,CAAAA,CAMb,EAAA,CAJInB,CAAAA,CAAK,IAAA,GAAS,mBAAA,EAChBwB,iCAAAA,CAAkBxB,CAAMa,CAAG,CAAA,CAGzBb,CAAAA,CAAK,IAAA,GAAS,gBAAA,EACZyB,iCAAAA,CAA0BzB,CAAMa,CAAG,CAAA,CAAG,CACxC,IAAMa,CAAAA,CAAiB1B,CAAAA,CAAK,SAAA,CAAU,CAAC,CAAA,CAEvC,EAAA,CACE0B,CAAAA,EAAAA,CACCA,CAAAA,CAAe,IAAA,GAAS,oBAAA,EACvBA,CAAAA,CAAe,IAAA,GAAS,yBAAA,CAAA,CAC1B,CACA,IAAMC,CAAAA,CAAY5B,CAAAA,CAAmB2B,CAAc,CAAA,CACnDX,CAAAA,CAAiB,IAAA,CAAK,CACpB,GAAA,CAAKZ,CAAAA,CAAsBuB,CAAc,CAAA,CACzC,sBAAA,CAAwBC,CAAAA,GAAc,aACxC,CAAC,CAAA,CACD,IAAA,CAAK,IAAA,CAAK,CACZ,CACF,CAGF,EAAA,CACE3B,CAAAA,CAAK,IAAA,GAAS,yBAAA,EACdA,CAAAA,CAAK,IAAA,GAAS,oBAAA,EACdA,CAAAA,CAAK,IAAA,GAAS,qBAAA,CACd,CACA,IAAM2B,CAAAA,CAAY5B,CAAAA,CAAmBC,CAAI,CAAA,CACrC2B,CAAAA,EAAAA,CACFZ,CAAAA,CAAiB,IAAA,CAAK,CACpB,GAAA,CAAKZ,CAAAA,CAAsBH,CAAI,CAAA,CAC/B,IAAA,CAAMA,CAAAA,CAAK,IAAA,GAAS,qBAAA,EAChBA,CAAAA,CAAK,IAAA,GAAS,oBAAA,CAAA,CACduB,CAAAA,CAAAvB,CAAAA,CAAK,EAAA,CAAA,EAAL,IAAA,CAAA,KAAA,CAAA,CAAAuB,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,GAAA,CAAA,GACQ,CACJ,GAAA,CAAAC,CAAAA,CACA,IAAA,CAAAC,CAAAA,CACA,sBAAA,CAAAC,CACF,CAAA,GAAKf,CAAAA,CACL,CACA,GAAM,CAAE,MAAA,CAAAgB,CAAAA,CAAQ,IAAA,CAAAC,CAAAA,CAAM,aAAA,CAAAC,CAAc,CAAA,CAAIC,yCAAAA,CAAe,CAAA,CACjDC,CAAAA,CAAsBP,CAAAA,CAAI,IAAA,GAAS,qBAAA,CAGvCO,CAAAA,EACAN,CAAAA,EACAlB,CAAAA,CACK,KAAA,CAAM,CAAA,CAAGiB,CAAAA,CAAI,KAAK,CAAA,CAClB,MAAA,CAAO,IAAI,MAAA,CAAO,CAAA,YAAA,EAAeC,CAAI,CAAA,UAAA,CAAY,CAAC,CAAA,GAAM,CAAA,CAAA,EAE7D,OAAA,CAAQ,IAAA,CACN,CAAA,KAAA,EAAQjB,CAAE,CAAA,YAAA,EAAeiB,CAAI,CAAA,cAAA,EAC3BC,CAAAA,CAAyB,QAAA,CAAW,aACtC,CAAA,yIAAA,CACF,CAAA,CAGF,IAAMM,CAAAA,CAAW,CAAA;AAAA,iBAAA,EACVC,uBAAc,CAAA;AAAA,mBAAA,EACZC,iCAAAA,CAAY,MAAA,CAAAP,CAAAA,CAAQ,IAAA,CAAAC,CAAAA,CAAM,aAAA,CAAAC,CAAc,CAAC,CAAC,CAAA;AAAA,0BAAA,EACnCA,CAAAA,CAAc,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,aAAA,CAAA,CAIxCjB,CAAAA,CAAY,UAAA,CACVY,CAAAA,CAAI,KAAA,CACJ,CAAA,EAAA;AAAwD;AAGxD,uBAAA;AAQE,oCAAA;AAYhB,iBAAA","file":"/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-LI2BURW4.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 { FORMAT_VERSION } from 'tinyest';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { createUnplugin, type UnpluginInstance } from 'unplugin';\nimport babel from './babel.ts';\nimport {\n  type Context,\n  defaultOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  getErrorMessage,\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        },\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          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              console.warn(\n                `File ${id}: function \"${name}\", containing ${\n                  removeJsImplementation ? 'kernel' : 'kernel & js'\n                } directive, might have been referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`,\n              );\n            }\n\n            const metadata = `{\n              v: ${FORMAT_VERSION},\n              ast: ${embedJSON({ params, body, externalNames })},\n              externals: {${externalNames.join(', ')}},\n            }`;\n\n            // Wrap the implementation in a set to `globalThis` to associate the name, AST and externals with the implementation.\n            magicString.appendLeft(\n              def.start,\n              `${isFunctionStatement && name ? `const ${name} = ` : ''}\n              (($) => ((globalThis.__TYPEGPU_META__ ??= new WeakMap()).set(\n                $.f = (`,\n            ).appendRight(\n              def.end,\n              `) , ${metadata}) && $.f))({})`,\n            );\n\n            if (removeJsImplementation) {\n              magicString.overwriteNode(\n                def,\n                `() => {\n                  throw new Error(\\`${getErrorMessage(name)}\\`);\n                }`,\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-LML6XL7I.js

@@ -0,0 +1,10 @@
+import{a as x,b,c as v,d as F,e as d,f as w,g as h}from"./chunk-DO6SOWHC.js";import $ from"defu";import{walk as A}from"estree-walker";import{generateTransform as S,MagicStringAST as I}from"magic-string-ast";import{FORMAT_VERSION as N}from"tinyest";import{transpileFn as P}from"tinyest-for-wgsl";import{createUnplugin as T}from"unplugin";function D(c){if(c.body.type==="BlockStatement"){for(let n of c.body.body)if(n.type==="ExpressionStatement"&&d.includes(n.directive))return n.directive}}function E(c){let n=structuredClone(c);return n.body.type==="BlockStatement"&&(n.body.body=n.body.body.filter(s=>!(s.type==="ExpressionStatement"&&s.directive&&d.includes(s.directive)))),n}var r=T(c=>{let n=$(c,x);return{name:"unplugin-typegpu",enforce:n.enforce,transform:{filter:{id:n},handler(s,p){let g={tgpuAliases:new Set(n.forceTgpuAlias?[n.forceTgpuAlias]:[]),fileId:p},k=this.parse(s,{allowReturnOutsideFunction:!0}),l=[],u=new I(s);A(k,{enter(i,t,m,y){var a;let e=i;if(e.type==="ImportDeclaration"&&v(e,g),e.type==="CallExpression"&&F(e,g)){let o=e.arguments[0];if(o&&(o.type==="FunctionExpression"||o.type==="ArrowFunctionExpression")){let f=D(o);l.push({def:E(o),removeJsImplementation:f!=="kernel & js"}),this.skip()}}if(e.type==="ArrowFunctionExpression"||e.type==="FunctionExpression"||e.type==="FunctionDeclaration"){let o=D(e);o&&(l.push({def:E(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())}}});for(let{def:i,name:t,removeJsImplementation:m}of l){let{params:y,body:e,externalNames:a}=P(i),o=i.type==="FunctionDeclaration";o&&t&&s.slice(0,i.start).search(new RegExp(`(?<![\\w_.])${t}(?![\\w_])`))!==-1&&console.warn(`File ${p}: function "${t}", containing ${m?"kernel":"kernel & js"} directive, might have been referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`);let f=`{
+              v: ${N},
+              ast: ${b({params:y,body:e,externalNames:a})},
+              externals: {${a.join(", ")}},
+            }`;u.appendLeft(i.start,`${o&&t?`const ${t} = `:""}
+              (($) => ((globalThis.__TYPEGPU_META__ ??= new WeakMap()).set(
+                $.f = (`).appendRight(i.end,`) , ${f}) && $.f))({})`),m&&u.overwriteNode(i,`() => {
+                  throw new Error(\`${w(t)}\`);
+                }`)}return S(u,p)}}}}),B=r,V=r.vite,_=r.rollup,G=r.rolldown,L=r.webpack,W=r.rspack,Y=r.esbuild,q=r.farm,z=h;export{B as a,V as b,_ as c,G as d,L as e,W as f,Y as g,q as h,z as i};
+//# sourceMappingURL=chunk-LML6XL7I.js.map

package/chunk-LML6XL7I.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 { FORMAT_VERSION } from 'tinyest';\nimport { transpileFn } from 'tinyest-for-wgsl';\nimport { createUnplugin, type UnpluginInstance } from 'unplugin';\nimport babel from './babel.ts';\nimport {\n  type Context,\n  defaultOptions,\n  embedJSON,\n  gatherTgpuAliases,\n  getErrorMessage,\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        },\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          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              console.warn(\n                `File ${id}: function \"${name}\", containing ${\n                  removeJsImplementation ? 'kernel' : 'kernel & js'\n                } directive, might have been referenced before its usage. Function statements are no longer hoisted after being transformed by the plugin.`,\n              );\n            }\n\n            const metadata = `{\n              v: ${FORMAT_VERSION},\n              ast: ${embedJSON({ params, body, externalNames })},\n              externals: {${externalNames.join(', ')}},\n            }`;\n\n            // Wrap the implementation in a set to `globalThis` to associate the name, AST and externals with the implementation.\n            magicString.appendLeft(\n              def.start,\n              `${isFunctionStatement && name ? `const ${name} = ` : ''}\n              (($) => ((globalThis.__TYPEGPU_META__ ??= new WeakMap()).set(\n                $.f = (`,\n            ).appendRight(\n              def.end,\n              `) , ${metadata}) && $.f))({})`,\n            );\n\n            if (removeJsImplementation) {\n              magicString.overwriteNode(\n                def,\n                `() => {\n                  throw new Error(\\`${getErrorMessage(name)}\\`);\n                }`,\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":"6EACA,OAAOA,MAAU,OACjB,OAAoB,QAAAC,MAAY,gBAChC,OAAS,qBAAAC,EAAmB,kBAAAC,MAAsB,mBAClD,OAAS,kBAAAC,MAAsB,UAC/B,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,OAAQ,CACN,GAAIA,CACN,EACA,QAAQG,EAAMC,EAAI,CAChB,IAAMC,EAAe,CACnB,YAAa,IAAI,IACfL,EAAQ,eAAiB,CAACA,EAAQ,cAAc,EAAI,CAAC,CACvD,EACA,OAAQI,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,IAAMvB,EAAOmB,EAMb,GAJInB,EAAK,OAAS,qBAChBwB,EAAkBxB,EAAMa,CAAG,EAGzBb,EAAK,OAAS,kBACZyB,EAA0BzB,EAAMa,CAAG,EAAG,CACxC,IAAMa,EAAiB1B,EAAK,UAAU,CAAC,EAEvC,GACE0B,IACCA,EAAe,OAAS,sBACvBA,EAAe,OAAS,2BAC1B,CACA,IAAMC,EAAY5B,EAAmB2B,CAAc,EACnDX,EAAiB,KAAK,CACpB,IAAKZ,EAAsBuB,CAAc,EACzC,uBAAwBC,IAAc,aACxC,CAAC,EACD,KAAK,KAAK,CACZ,CACF,CAGF,GACE3B,EAAK,OAAS,2BACdA,EAAK,OAAS,sBACdA,EAAK,OAAS,sBACd,CACA,IAAM2B,EAAY5B,EAAmBC,CAAI,EACrC2B,IACFZ,EAAiB,KAAK,CACpB,IAAKZ,EAAsBH,CAAI,EAC/B,KAAMA,EAAK,OAAS,uBAChBA,EAAK,OAAS,sBACduB,EAAAvB,EAAK,KAAL,YAAAuB,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,OACQ,CACJ,IAAAC,EACA,KAAAC,EACA,uBAAAC,CACF,IAAKf,EACL,CACA,GAAM,CAAE,OAAAgB,EAAQ,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYN,CAAG,EACjDO,EAAsBP,EAAI,OAAS,sBAGvCO,GACAN,GACAlB,EACK,MAAM,EAAGiB,EAAI,KAAK,EAClB,OAAO,IAAI,OAAO,eAAeC,CAAI,YAAY,CAAC,IAAM,IAE7D,QAAQ,KACN,QAAQjB,CAAE,eAAeiB,CAAI,iBAC3BC,EAAyB,SAAW,aACtC,2IACF,EAGF,IAAMM,EAAW;AAAA,mBACVC,CAAc;AAAA,qBACZC,EAAU,CAAE,OAAAP,EAAQ,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAAC;AAAA,4BACnCA,EAAc,KAAK,IAAI,CAAC;AAAA,eAIxCjB,EAAY,WACVY,EAAI,MACJ,GAAGO,GAAuBN,EAAO,SAASA,CAAI,MAAQ,EAAE;AAAA;AAAA,wBAG1D,EAAE,YACAD,EAAI,IACJ,OAAOQ,CAAQ,gBACjB,EAEIN,GACFd,EAAY,cACVY,EACA;AAAA,sCACsBW,EAAgBV,CAAI,CAAC;AAAA,kBAE7C,CAEJ,CAEA,OAAOW,EAAkBxB,EAAaJ,CAAE,CAC1C,CACF,CACF,CACF,CACF,EAIO6B,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","FORMAT_VERSION","transpileFn","createUnplugin","getKernelDirective","node","statement","kernelDirectives","removeKernelDirective","cloned","typegpu","createUnplugin","rawOptions","options","defu","defaultOptions","code","id","ctx","ast","tgslFunctionDefs","magicString","MagicStringAST","walk","_node","_parent","prop","index","_a","gatherTgpuAliases","isShellImplementationCall","implementation","directive","def","name","removeJsImplementation","params","body","externalNames","transpileFn","isFunctionStatement","metadata","FORMAT_VERSION","embedJSON","getErrorMessage","generateTransform","index_default","vitePlugin","rollupPlugin","rolldownPlugin","webpackPlugin","rspackPlugin","esbuildPlugin","farmPlugin","babelPlugin","babel_default"]}

package/esbuild.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');var l=_chunk6XKKJSKWcjs.g;exports.default = l;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');var l=_chunkLI2BURW4cjs.g;exports.default = l;
 //# sourceMappingURL=esbuild.cjs.map

package/esbuild.js

@@ -1,2 +1,2 @@
-import{g as e}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.js";var l=e;export{l as default};
+import{g as e}from"./chunk-LML6XL7I.js";import"./chunk-DO6SOWHC.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 _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');var m=_chunk6XKKJSKWcjs.h;exports.default = m;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');var m=_chunkLI2BURW4cjs.h;exports.default = m;
 //# sourceMappingURL=farm.cjs.map

package/farm.js

@@ -1,2 +1,2 @@
-import{h as r}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.js";var m=r;export{m as default};
+import{h as r}from"./chunk-LML6XL7I.js";import"./chunk-DO6SOWHC.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 _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');exports.babelPlugin = _chunk6XKKJSKWcjs.i; exports.default = _chunk6XKKJSKWcjs.a; exports.esbuildPlugin = _chunk6XKKJSKWcjs.g; exports.farmPlugin = _chunk6XKKJSKWcjs.h; exports.rolldownPlugin = _chunk6XKKJSKWcjs.d; exports.rollupPlugin = _chunk6XKKJSKWcjs.c; exports.rspackPlugin = _chunk6XKKJSKWcjs.f; exports.vitePlugin = _chunk6XKKJSKWcjs.b; exports.webpackPlugin = _chunk6XKKJSKWcjs.e;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');exports.babelPlugin = _chunkLI2BURW4cjs.i; exports.default = _chunkLI2BURW4cjs.a; exports.esbuildPlugin = _chunkLI2BURW4cjs.g; exports.farmPlugin = _chunkLI2BURW4cjs.h; exports.rolldownPlugin = _chunkLI2BURW4cjs.d; exports.rollupPlugin = _chunkLI2BURW4cjs.c; exports.rspackPlugin = _chunkLI2BURW4cjs.f; exports.vitePlugin = _chunkLI2BURW4cjs.b; exports.webpackPlugin = _chunkLI2BURW4cjs.e;
 //# sourceMappingURL=index.cjs.map

package/index.js

@@ -1,2 +1,2 @@
-import{a,b,c,d,e,f,g,h,i}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.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-LML6XL7I.js";import"./chunk-DO6SOWHC.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.1",
+  "version": "0.1.3",
   "description": "Build plugins for TypeGPU, enabling seamless JavaScript -> WGSL transpilation and improved debugging.",
   "keywords": [
     "rollup-plugin",
@@ -85,15 +85,16 @@
   "homepage": "https://typegpu.com",
   "dependencies": {
     "@babel/standalone": "^7.27.0",
+    "defu": "^6.1.4",
     "estree-walker": "^3.0.3",
     "magic-string-ast": "^0.9.1",
     "picomatch": "^4.0.2",
-    "defu": "^6.1.4",
-    "tinyest-for-wgsl": "~0.1.1",
+    "tinyest": "~0.1.0",
+    "tinyest-for-wgsl": "~0.1.2",
     "unplugin": "^2.3.1"
   },
   "peerDependencies": {
-    "typegpu": "^0.5.7"
+    "typegpu": "^0.5.9"
   },
   "main": "./index.js",
   "types": "./index.d.ts",

package/rolldown.cjs

@@ -1,2 +1,2 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');var r=_chunk6XKKJSKWcjs.d;exports.default = r;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');var r=_chunkLI2BURW4cjs.d;exports.default = r;
 //# sourceMappingURL=rolldown.cjs.map

package/rolldown.js

@@ -1,2 +1,2 @@
-import{d as o}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.js";var r=o;export{r as default};
+import{d as o}from"./chunk-LML6XL7I.js";import"./chunk-DO6SOWHC.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 _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');var r=_chunk6XKKJSKWcjs.c;exports.default = r;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');var r=_chunkLI2BURW4cjs.c;exports.default = r;
 //# sourceMappingURL=rollup.cjs.map

package/rollup.js

@@ -1,2 +1,2 @@
-import{c as l}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.js";var r=l;export{r as default};
+import{c as l}from"./chunk-LML6XL7I.js";import"./chunk-DO6SOWHC.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 _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');var p=_chunk6XKKJSKWcjs.f;exports.default = p;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');var p=_chunkLI2BURW4cjs.f;exports.default = p;
 //# sourceMappingURL=rspack.cjs.map

package/rspack.js

@@ -1,2 +1,2 @@
-import{f as r}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.js";var p=r;export{p as default};
+import{f as r}from"./chunk-LML6XL7I.js";import"./chunk-DO6SOWHC.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 _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');var i=_chunk6XKKJSKWcjs.b;exports.default = i;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');var i=_chunkLI2BURW4cjs.b;exports.default = i;
 //# sourceMappingURL=vite.cjs.map

package/vite.js

@@ -1,2 +1,2 @@
-import{b as t}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.js";var i=t;export{i as default};
+import{b as t}from"./chunk-LML6XL7I.js";import"./chunk-DO6SOWHC.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 _chunk6XKKJSKWcjs = require('./chunk-6XKKJSKW.cjs');require('./chunk-YA33RT2L.cjs');var p=_chunk6XKKJSKWcjs.e;exports.default = p;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true});var _chunkLI2BURW4cjs = require('./chunk-LI2BURW4.cjs');require('./chunk-G6NIIU55.cjs');var p=_chunkLI2BURW4cjs.e;exports.default = p;
 //# sourceMappingURL=webpack.cjs.map

package/webpack.js

@@ -1,2 +1,2 @@
-import{e}from"./chunk-IZQXGWE3.js";import"./chunk-Y3TSO2N7.js";var p=e;export{p as default};
+import{e}from"./chunk-LML6XL7I.js";import"./chunk-DO6SOWHC.js";var p=e;export{p as default};
 //# sourceMappingURL=webpack.js.map

package/chunk-6XKKJSKW.cjs

@@ -1,2 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }var _chunkYA33RT2Lcjs = require('./chunk-YA33RT2L.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"&&_chunkYA33RT2Lcjs.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&&_chunkYA33RT2Lcjs.g.includes(s.directive)))),n}var r=_unplugin.createUnplugin.call(void 0, c=>{let n=_defu2.default.call(void 0, c,_chunkYA33RT2Lcjs.b);return{name:"unplugin-typegpu",enforce:n.enforce,transform:{filter:_chunkYA33RT2Lcjs.a.call(void 0, {id:n},n.forceTgpuAlias?{}:{code:_chunkYA33RT2Lcjs.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"&&_chunkYA33RT2Lcjs.e.call(void 0, e,d),e.type==="CallExpression"&&_chunkYA33RT2Lcjs.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,`, ${_chunkYA33RT2Lcjs.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 =_chunkYA33RT2Lcjs.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-6XKKJSKW.cjs.map

package/chunk-6XKKJSKW.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-6XKKJSKW.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-6XKKJSKW.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 { 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 ${\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({ 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-IZQXGWE3.js

@@ -1,2 +0,0 @@
-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-Y3TSO2N7.js";import N from"defu";import{walk as I}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=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 P(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 S(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-IZQXGWE3.js.map

package/chunk-IZQXGWE3.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 { 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 { 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 ${\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({ 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,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,SAAAiB,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYN,CAAG,EACnDO,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,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","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","argNames","body","externalNames","transpileFn","isFunctionStatement","embedJSON","generateTransform","index_default","vitePlugin","rollupPlugin","rolldownPlugin","webpackPlugin","rspackPlugin","esbuildPlugin","farmPlugin","babelPlugin","babel_default"]}

package/chunk-Y3TSO2N7.js

@@ -1,2 +0,0 @@
-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-Y3TSO2N7.js.map

package/chunk-Y3TSO2N7.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  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 { 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: ${\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({ 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: ${\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({ argNames, 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,SAAAK,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYZ,CAAI,EACpDa,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,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,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,SAAAZ,EAAU,KAAAC,EAAM,cAAAC,CAAc,EAAIC,EAAYS,CAAc,EAC9DR,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,SAAAL,EAAU,KAAAC,EAAM,cAAAC,CAAc,CAAC,CAC7C,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","argNames","body","externalNames","transpileFn","tgpuAlias","embedJSON","functionVisitor","path","gatherTgpuAliases","transpiled","expression","isShellImplementationCall","implementation","babel_default","state","code","options","id","filter","createFilterForId","codeFilterRegexes","reg"]}

package/chunk-YA33RT2L.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; } } 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-YA33RT2L.cjs.map

package/chunk-YA33RT2L.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/Users/iwo/Projects/wigsill/packages/unplugin-typegpu/dist/chunk-YA33RT2L.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-YA33RT2L.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 { 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: ${\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({ 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: ${\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({ argNames, 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"]}