refactor: Unplugin 3 (#2294)

Author: iwoplaza

apps/bun-example/preload.ts

@@ -1,4 +1,4 @@
 import { plugin } from 'bun';
 import typegpu from 'unplugin-typegpu/bun';
 
-void plugin(typegpu({ include: /\.(ts)$/ }));
+void plugin(typegpu());

apps/treeshake-test/bundleWith.ts

@@ -17,7 +17,7 @@ export async function bundleWithEsbuild(entryUrl: URL, outDir: URL): Promise<URL
   const entryFileName = path.basename(entryUrl.pathname, '.ts');
   const outPath = new URL(`${entryFileName}.esbuild.js`, outDir);
   await esbuild({
-    plugins: [esbuildPlugin({})],
+    plugins: [esbuildPlugin()],
     entryPoints: [entryUrl.pathname],
     bundle: true,
     outfile: outPath.pathname,
@@ -40,7 +40,7 @@ export async function bundleWithWebpack(entryPath: URL, outDir: URL): Promise<UR
           path: path.dirname(outPath.pathname),
           filename: path.basename(outPath.pathname),
         },
-        plugins: [webpackPlugin({})],
+        plugins: [webpackPlugin()],
         module: {
           rules: [
             {
@@ -84,7 +84,7 @@ export async function bundleWithTsdown(entryUrl: URL, outDir: URL): Promise<URL>
 
   try {
     await tsdown({
-      plugins: [rolldownPlugin({})],
+      plugins: [rolldownPlugin()],
       outputOptions: {
         name: path.basename(outPath.pathname),
         dir: path.dirname(outPath.pathname),

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

@@ -71,7 +71,7 @@ import { defineConfig } from 'vite';
 import typegpuPlugin from 'unplugin-typegpu/vite';
 
 export default defineConfig({
-  plugins: [typegpuPlugin({})],
+  plugins: [typegpuPlugin()],
 });
 ```
 

package.json

@@ -66,8 +66,9 @@
       "sharp"
     ],
     "overrides": {
-      "rollup": "catalog:",
+      "rollup": "catalog:build",
       "vite": "8.0.2",
+      "@babel/types": "catalog:",
       "typescript": "catalog:types",
       "three": "catalog:example"
     }

packages/tinyest-for-wgsl/package.json

@@ -56,7 +56,7 @@
   },
   "devDependencies": {
     "@babel/parser": "^7.27.0",
-    "@babel/types": "7.26.5",
+    "@babel/types": "catalog:",
     "@typegpu/tgpu-dev-cli": "workspace:*",
     "acorn": "^8.14.1",
     "tsdown": "catalog:build",

packages/tinyest-for-wgsl/tsdown.config.ts

@@ -3,7 +3,9 @@ import { defineConfig } from 'tsdown';
 export default defineConfig({
   entry: ['src/index.ts'],
   outDir: 'dist',
-  inlineOnly: false,
+  deps: {
+    onlyBundle: false,
+  },
   format: ['cjs', 'esm'],
   dts: true,
   platform: 'neutral',

packages/typegpu/tests/unplugin/autoname.test.ts

@@ -175,16 +175,45 @@ describe('autonaming', () => {
   });
 
   it('shellless name carries over to WGSL', () => {
-    function myFun() {
-      'use gpu';
-      return 0;
-    }
+    const scope = () => {
+      function myFun() {
+        'use gpu';
+        return 0;
+      }
 
-    const main = tgpu.fn([])(() => {
-      myFun();
-    });
+      const main = tgpu.fn([])(() => {
+        myFun();
+      });
+
+      return main;
+    };
+
+    expect(scope.toString()).toMatchInlineSnapshot(`
+      "() => {
+      			const myFun = (/*#__PURE__*/($ => (globalThis.__TYPEGPU_META__ ??= new WeakMap()).set($.f = (function myFun() {
+      				"use gpu";
+      				return 0;
+      			}), {
+          v: 1,
+          name: "myFun",
+          ast: {"params":[],"body":[0,[[10,[5,"0"]]]],"externalNames":[]},
+          externals: () => ({}),
+        }) && $.f)({}));
+
+
+      			const main = (/*#__PURE__*/(globalThis.__TYPEGPU_AUTONAME__ ?? (a => a))(__vite_ssr_import_2__.default.fn([])((/*#__PURE__*/($ => (globalThis.__TYPEGPU_META__ ??= new WeakMap()).set($.f = (() => {
+      				myFun();
+      			}), {
+          v: 1,
+          name: undefined,
+          ast: {"params":[],"body":[0,[[6,"myFun",[]]]],"externalNames":["myFun"]},
+          externals: () => ({myFun}),
+        }) && $.f)({}))), "main"));
+      			return main;
+      		}"
+    `);
 
-    expect(tgpu.resolve([main])).toMatchInlineSnapshot(`
+    expect(tgpu.resolve([scope()])).toMatchInlineSnapshot(`
       "fn myFun() -> i32 {
         return 0;
       }

packages/unplugin-typegpu/README.md

@@ -2,18 +2,17 @@
 
 # unplugin-typegpu
 
-🚧 **Under Construction** 🚧 -
 [GitHub](https://github.com/software-mansion/TypeGPU/tree/main/packages/unplugin-typegpu)
 
 Read more about the plugin in the
 [TypeGPU documentation](https://docs.swmansion.com/TypeGPU/tooling/unplugin-typegpu/).
 
 </div>
 
-Build plugins for [TypeGPU](https://typegpu.com) that enable:
+A set of bundler plugins that enhance [TypeGPU](https://typegpu.com) with:
 
-- Seamless JavaScript -> WGSL transpilation
-- [🚧 TODO] Improved debugging with automatic naming of resources
+- JavaScript/TypeScript shader support ('use gpu' directive)
+- Improved debugging with automatic naming of resources
 
 ## Getting Started
 
@@ -48,6 +47,17 @@ export default defineConfig({
 });
 ```
 
+- bun
+
+```ts
+// preload.ts
+
+import { plugin } from 'bun';
+import typegpu from 'unplugin-typegpu/bun';
+
+void plugin(typegpu());
+```
+
 ## TypeGPU is created by Software Mansion
 
 [![swm](https://logo.swmansion.com/logo?color=white&variant=desktop&width=150&tag=typegpu-github 'Software Mansion')](https://swmansion.com)

packages/unplugin-typegpu/package.json

@@ -22,7 +22,8 @@
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/software-mansion/TypeGPU.git"
+    "url": "git+https://github.com/software-mansion/TypeGPU.git",
+    "directory": "packages/unplugin-typegpu"
   },
   "type": "module",
   "sideEffects": false,
@@ -165,29 +166,27 @@
     "prepublishOnly": "tgpu-dev-cli prepack"
   },
   "dependencies": {
-    "@babel/standalone": "^7.28.6",
+    "@babel/parser": "^7.29.0",
+    "@babel/traverse": "^7.29.0",
+    "@babel/types": "catalog:",
+    "ast-kit": "^2.2.0",
     "defu": "^6.1.4",
-    "estree-walker": "^3.0.3",
-    "magic-string-ast": "^1.0.0",
+    "magic-string-ast": "^1.0.3",
     "pathe": "^2.0.3",
     "picomatch": "^4.0.3",
     "tinyest": "workspace:~0.3.1",
     "tinyest-for-wgsl": "workspace:~0.3.2",
-    "unplugin": "^2.3.5"
+    "unplugin": "^3.0.0"
   },
   "devDependencies": {
-    "@babel/template": "^7.27.2",
-    "@babel/types": "^7.26.5",
+    "@babel/standalone": "^7.28.6",
     "@rollup/plugin-virtual": "^3.0.2",
     "@typegpu/tgpu-dev-cli": "workspace:*",
     "@types/babel__standalone": "^7.1.9",
-    "@types/babel__template": "^7.4.4",
     "@types/babel__traverse": "^7.20.7",
-    "@types/bun": "^1.2.22",
+    "@types/bun": "^1.3.11",
     "@types/picomatch": "^4.0.1",
-    "acorn": "^8.14.1",
-    "rolldown": "1.0.0-beta.33",
-    "rollup": "catalog:",
+    "rollup": "catalog:build",
     "tsdown": "catalog:build",
     "typescript": "catalog:types"
   },