feat(worker): bundle worker emit asset file

packages/playground/wasm/__tests__/wasm.spec.ts

@@ -9,3 +9,7 @@ test('should work when output', async () => {
   await page.click('.output-wasm .run')
   await untilUpdated(() => page.textContent('.output-wasm .result'), '24')
 })
+
+test('should work when wasm in worker', async () => {
+  await untilUpdated(() => page.textContent('.worker-wasm .result'), '3')
+})

packages/playground/wasm/index.html

@@ -12,9 +12,20 @@ <h3>When wasm is output, result should be 24</h3>
   <span class="result"></span>
 </div>
 
+<div class="worker-wasm">
+  <h3>worker wasm</h3>
+  <span class="result"></span>
+</div>
+
 <script type="module">
   import light from './light.wasm'
   import heavy from './heavy.wasm'
+  import myWorker from './worker?worker'
+
+  const w = new myWorker()
+  w.addEventListener('message', (ev) => {
+    text('.worker-wasm .result', ev.data.result)
+  })
 
   async function testWasm(init, resultElement) {
     const { exported_func } = await init({
@@ -25,6 +36,10 @@ <h3>When wasm is output, result should be 24</h3>
     exported_func()
   }
 
+  function text(el, text) {
+    document.querySelector(el).textContent = text
+  }
+
   document
     .querySelector('.inline-wasm .run')
     .addEventListener('click', async () =>

packages/playground/wasm/vite.config.ts

@@ -0,0 +1,8 @@
+import { defineConfig } from 'vite'
+export default defineConfig({
+  build: {
+    // make can no emit light.wasm
+    // and emit add.wasm
+    assetsInlineLimit: 80
+  }
+})

packages/playground/wasm/worker.js

@@ -0,0 +1,5 @@
+import init from './add.wasm'
+init().then((exports) => {
+  // eslint-disable-next-line no-undef
+  self.postMessage({ result: exports.add(1, 2) })
+})

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

@@ -6,10 +6,12 @@ import type Rollup from 'rollup'
 import { ENV_PUBLIC_PATH } from '../constants'
 import path from 'path'
 import { onRollupWarning } from '../build'
+import { registerAssetToChunk } from './asset'
 
 const WorkerFileId = 'worker_file'
 
 export async function bundleWorkerEntry(
+  ctx: Rollup.TransformPluginContext,
   config: ResolvedConfig,
   id: string
 ): Promise<Buffer> {
@@ -27,11 +29,20 @@ export async function bundleWorkerEntry(
   })
   let code: string
   try {
-    const { output } = await bundle.generate({
+    const {
+      output: [outputCode, ...outputChunks]
+    } = await bundle.generate({
       format,
       sourcemap: config.build.sourcemap
     })
-    code = output[0].code
+    code = outputCode.code
+    outputChunks.forEach((outputChunk) => {
+      if (outputChunk.type === 'chunk') {
+        registerAssetToChunk(outputChunk, outputChunk.name)
+      } else if (outputChunk.type === 'asset') {
+        ctx.emitFile(outputChunk)
+      }
+    })
   } finally {
     await bundle.close()
   }
@@ -72,7 +83,7 @@ export function webWorkerPlugin(config: ResolvedConfig): Plugin {
 
       let url: string
       if (isBuild) {
-        const code = await bundleWorkerEntry(config, id)
+        const code = await bundleWorkerEntry(this, config, id)
         if (query.inline != null) {
           const { format } = config.worker
           const workerOptions = format === 'es' ? '{type: "module"}' : '{}'

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

@@ -65,7 +65,7 @@ export function workerImportMetaUrlPlugin(config: ResolvedConfig): Plugin {
           const file = path.resolve(path.dirname(id), rawUrl.slice(1, -1))
           let url: string
           if (isBuild) {
-            const content = await bundleWorkerEntry(config, file)
+            const content = await bundleWorkerEntry(this, config, file)
             const basename = path.parse(cleanUrl(file)).name
             const contentHash = getAssetHash(content)
             const fileName = path.posix.join(