fix(rsc): inject AsyncLocalStorage global via transform

Author: hi-ogawa

packages/plugin-rsc/src/plugin.ts

@@ -948,25 +948,40 @@ import.meta.hot.on("rsc:update", () => {
       // make `AsyncLocalStorage` available globally for React request context on edge build (e.g. React.cache, ssr preload)
       // https://github.com/facebook/react/blob/f14d7f0d2597ea25da12bcf97772e8803f2a394c/packages/react-server/src/forks/ReactFlightServerConfig.dom-edge.js#L16-L19
       name: 'rsc:inject-async-local-storage',
-      async configureServer() {
-        const __viteRscAyncHooks = await import('node:async_hooks')
-        ;(globalThis as any).AsyncLocalStorage =
-          __viteRscAyncHooks.AsyncLocalStorage
-      },
-      banner(chunk) {
-        if (
-          (this.environment.name === 'ssr' ||
-            this.environment.name === 'rsc') &&
-          this.environment.mode === 'build' &&
-          chunk.isEntry
-        ) {
-          return `\
-import * as __viteRscAyncHooks from "node:async_hooks";
-globalThis.AsyncLocalStorage = __viteRscAyncHooks.AsyncLocalStorage;
-`
-        }
-        return ''
+      // async configureServer() {
+      //   const __viteRscAyncHooks = await import('node:async_hooks')
+      //   ;(globalThis as any).AsyncLocalStorage =
+      //     __viteRscAyncHooks.AsyncLocalStorage
+      // },
+      transform: {
+        handler(code) {
+          if (
+            this.environment.name !== 'client' &&
+            code.includes('new AsyncLocalStorage()') &&
+            !code.includes('__viteRscAyncHooks')
+          ) {
+            return (
+              `import * as __viteRscAyncHooks from "node:async_hooks";` +
+              `globalThis.AsyncLocalStorage = __viteRscAyncHooks.AsyncLocalStorage;` +
+              code
+            )
+          }
+        },
       },
+      //       banner(chunk) {
+      //         if (
+      //           (this.environment.name === 'ssr' ||
+      //             this.environment.name === 'rsc') &&
+      //           this.environment.mode === 'build' &&
+      //           chunk.isEntry
+      //         ) {
+      //           return `\
+      // import * as __viteRscAyncHooks from "node:async_hooks";
+      // globalThis.AsyncLocalStorage = __viteRscAyncHooks.AsyncLocalStorage;
+      // `
+      //         }
+      //         return ''
+      //       },
     },
     ...vitePluginRscMinimal(rscPluginOptions, manager),
     ...vitePluginFindSourceMapURL(),