fix: unplugin-typegpu should parse files with TS syntax support (#1516)

Author: iwoplaza

apps/typegpu-docs/src/content/docs/tooling/unplugin-typegpu.mdx

@@ -116,10 +116,16 @@ module.exports = (api) => {
 
 ```ts
 interface Options {
+  /** @default [/\.m?[jt]sx?$/] */
   include?: FilterPattern;
+  /** @default undefined */
   exclude?: FilterPattern;
+  /** @default undefined */
   enforce?: 'post' | 'pre' | undefined;
+  /** @default undefined */
   forceTgpuAlias?: string;
+  /** @default true */
+  autoNamingEnabled?: boolean;
 }
 ```
 

packages/unplugin-typegpu/package.json

@@ -110,11 +110,11 @@
     "@babel/standalone": "^7.27.0",
     "defu": "^6.1.4",
     "estree-walker": "^3.0.3",
-    "magic-string-ast": "^0.9.1",
-    "picomatch": "^4.0.2",
+    "magic-string-ast": "^1.0.0",
+    "picomatch": "^4.0.3",
     "tinyest": "workspace:~0.1.1",
     "tinyest-for-wgsl": "workspace:~0.1.2",
-    "unplugin": "^2.3.1"
+    "unplugin": "^2.3.5"
   },
   "peerDependencies": {
     "typegpu": "workspace:^0.6.0"
@@ -127,7 +127,7 @@
     "@types/babel__standalone": "^7.1.9",
     "@types/babel__template": "^7.4.4",
     "@types/babel__traverse": "^7.20.7",
-    "@types/picomatch": "^4.0.0",
+    "@types/picomatch": "^4.0.1",
     "acorn": "^8.14.1",
     "rollup": "~4.37.0",
     "tsup": "catalog:build",

packages/unplugin-typegpu/src/common.ts

@@ -16,8 +16,8 @@ export interface Options {
   include?: FilterPattern;
   exclude?: FilterPattern;
   enforce?: 'post' | 'pre' | undefined;
-  forceTgpuAlias?: string;
-  autoNamingEnabled?: boolean;
+  forceTgpuAlias?: string | undefined;
+  autoNamingEnabled?: boolean | undefined;
 }
 
 export const defaultOptions = {

packages/unplugin-typegpu/src/index.ts

@@ -100,9 +100,22 @@ const typegpu: UnpluginInstance<Options, false> = createUnplugin(
             autoNamingEnabled: options.autoNamingEnabled,
           };
 
-          const ast = this.parse(code, {
-            allowReturnOutsideFunction: true,
-          }) as Node;
+          let ast: Node;
+          try {
+            ast = this.parse(code, {
+              lang: 'ts',
+              allowReturnOutsideFunction: true,
+            }) as Node;
+          } catch (cause) {
+            console.warn(
+              `[unplugin-typegpu] Failed to parse ${id}. Cause: ${
+                typeof cause === 'object' && cause && 'message' in cause
+                  ? cause.message
+                  : cause
+              }`,
+            );
+            return undefined;
+          }
 
           const tgslFunctionDefs: {
             def: FunctionNode;
@@ -163,20 +176,14 @@ const typegpu: UnpluginInstance<Options, false> = createUnplugin(
             },
           });
 
-          for (
-            const {
-              def,
-              name,
-            } of tgslFunctionDefs
-          ) {
+          for (const { def, name } of tgslFunctionDefs) {
             const { params, body, externalNames } = transpileFn(def);
             const isFunctionStatement = def.type === 'FunctionDeclaration';
 
             if (
               isFunctionStatement &&
               name &&
-              code
-                  .slice(0, def.start)
+              code.slice(0, def.start)
                   .search(new RegExp(`(?<![\\w_.])${name}(?![\\w_])`)) !== -1
             ) {
               console.warn(