refactor(worker): set virtual file content in load hook (#20160)

Author: sapphi-red

packages/vite/src/node/plugins/worker.ts

@@ -262,60 +262,7 @@ export function webWorkerPlugin(config: ResolvedConfig): Plugin {
     },
 
     load: {
-      handler(id) {
-        if (isBuild && workerOrSharedWorkerRE.test(id)) {
-          return ''
-        }
-      },
-    },
-
-    shouldTransformCachedModule({ id }) {
-      if (isBuild && config.build.watch && workerOrSharedWorkerRE.test(id)) {
-        return true
-      }
-    },
-
-    transform: {
-      async handler(raw, id) {
-        const workerFileMatch = workerFileRE.exec(id)
-        if (workerFileMatch) {
-          // if import worker by worker constructor will have query.type
-          // other type will be import worker by esm
-          const workerType = workerFileMatch[1] as WorkerType
-          let injectEnv = ''
-
-          const scriptPath = JSON.stringify(
-            path.posix.join(config.base, ENV_PUBLIC_PATH),
-          )
-
-          if (workerType === 'classic') {
-            injectEnv = `importScripts(${scriptPath})\n`
-          } else if (workerType === 'module') {
-            injectEnv = `import ${scriptPath}\n`
-          } else if (workerType === 'ignore') {
-            if (isBuild) {
-              injectEnv = ''
-            } else {
-              // dynamic worker type we can't know how import the env
-              // so we copy /@vite/env code of server transform result into file header
-              const environment = this.environment
-              const moduleGraph =
-                environment.mode === 'dev' ? environment.moduleGraph : undefined
-              const module = moduleGraph?.getModuleById(ENV_ENTRY)
-              injectEnv = module?.transformResult?.code || ''
-            }
-          }
-          if (injectEnv) {
-            const s = new MagicString(raw)
-            s.prepend(injectEnv + ';\n')
-            return {
-              code: s.toString(),
-              map: s.generateMap({ hires: 'boundary' }),
-            }
-          }
-          return
-        }
-
+      async handler(id) {
         const workerMatch = workerOrSharedWorkerRE.exec(id)
         if (!workerMatch) return
 
@@ -417,6 +364,54 @@ export function webWorkerPlugin(config: ResolvedConfig): Plugin {
       },
     },
 
+    shouldTransformCachedModule({ id }) {
+      if (isBuild && config.build.watch && workerOrSharedWorkerRE.test(id)) {
+        return true
+      }
+    },
+
+    transform: {
+      async handler(raw, id) {
+        const workerFileMatch = workerFileRE.exec(id)
+        if (workerFileMatch) {
+          // if import worker by worker constructor will have query.type
+          // other type will be import worker by esm
+          const workerType = workerFileMatch[1] as WorkerType
+          let injectEnv = ''
+
+          const scriptPath = JSON.stringify(
+            path.posix.join(config.base, ENV_PUBLIC_PATH),
+          )
+
+          if (workerType === 'classic') {
+            injectEnv = `importScripts(${scriptPath})\n`
+          } else if (workerType === 'module') {
+            injectEnv = `import ${scriptPath}\n`
+          } else if (workerType === 'ignore') {
+            if (isBuild) {
+              injectEnv = ''
+            } else {
+              // dynamic worker type we can't know how import the env
+              // so we copy /@vite/env code of server transform result into file header
+              const environment = this.environment
+              const moduleGraph =
+                environment.mode === 'dev' ? environment.moduleGraph : undefined
+              const module = moduleGraph?.getModuleById(ENV_ENTRY)
+              injectEnv = module?.transformResult?.code || ''
+            }
+          }
+          if (injectEnv) {
+            const s = new MagicString(raw)
+            s.prepend(injectEnv + ';\n')
+            return {
+              code: s.toString(),
+              map: s.generateMap({ hires: 'boundary' }),
+            }
+          }
+        }
+      },
+    },
+
     renderChunk(code, chunk, outputOptions) {
       let s: MagicString
       const result = () => {