Add wasm support for test and SSR

Author: upsuper

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

@@ -26,21 +26,32 @@ const wasmHelper = async (opts = {}, url: string) => {
     }
     result = await WebAssembly.instantiate(bytes, opts)
   } else {
-    // https://github.com/mdn/webassembly-examples/issues/5
-    // WebAssembly.instantiateStreaming requires the server to provide the
-    // correct MIME type for .wasm files, which unfortunately doesn't work for
-    // a lot of static file servers, so we just work around it by getting the
-    // raw buffer.
-    const response = await fetch(url)
-    const contentType = response.headers.get('Content-Type') || ''
-    if (
-      'instantiateStreaming' in WebAssembly &&
-      contentType.startsWith('application/wasm')
-    ) {
-      result = await WebAssembly.instantiateStreaming(response, opts)
-    } else {
-      const buffer = await response.arrayBuffer()
+    if (typeof process !== 'undefined' && process.versions?.node) {
+      const fs = await import('node:fs/promises')
+      if (url.startsWith('/@fs/')) {
+        url = url.slice(4)
+      } else if (url.startsWith('/')) {
+        url = url.slice(1)
+      }
+      const buffer = await fs.readFile(url)
       result = await WebAssembly.instantiate(buffer, opts)
+    } else {
+      // https://github.com/mdn/webassembly-examples/issues/5
+      // WebAssembly.instantiateStreaming requires the server to provide the
+      // correct MIME type for .wasm files, which unfortunately doesn't work for
+      // a lot of static file servers, so we just work around it by getting the
+      // raw buffer.
+      const response = await fetch(url)
+      const contentType = response.headers.get('Content-Type') || ''
+      if (
+        'instantiateStreaming' in WebAssembly &&
+        contentType.startsWith('application/wasm')
+      ) {
+        result = await WebAssembly.instantiateStreaming(response, opts)
+      } else {
+        const buffer = await response.arrayBuffer()
+        result = await WebAssembly.instantiate(buffer, opts)
+      }
     }
   }
   return result.instance
@@ -66,7 +77,7 @@ export const wasmHelperPlugin = (): Plugin => {
           return `export default ${wasmHelperCode}`
         }
 
-        const url = await fileToUrl(this, id)
+        const url = (await fileToUrl(this, id)).split('?')[0]
 
         return `
   import initWasm from "${wasmHelperId}"

playground/ssr-wasm/__tests__/serve.ts

@@ -0,0 +1,35 @@
+// this is automatically detected by playground/vitestSetup.ts and will replace
+// the default e2e test serve behavior
+
+import path from 'node:path'
+import kill from 'kill-port'
+import { hmrPorts, ports, rootDir } from '~utils'
+
+export const port = ports['ssr-wasm']
+
+export async function serve(): Promise<{ close(): Promise<void> }> {
+  await kill(port)
+
+  const { createServer } = await import(path.resolve(rootDir, 'server.js'))
+  const { app, vite } = await createServer(rootDir, hmrPorts['ssr-wasm'])
+
+  return new Promise((resolve, reject) => {
+    try {
+      const server = app.listen(port, () => {
+        resolve({
+          // for test teardown
+          async close() {
+            await new Promise((resolve) => {
+              server.close(resolve)
+            })
+            if (vite) {
+              await vite.close()
+            }
+          },
+        })
+      })
+    } catch (e) {
+      reject(e)
+    }
+  })
+}

playground/ssr-wasm/__tests__/ssr-wasm.spec.ts

@@ -0,0 +1,15 @@
+import { expect, test } from 'vitest'
+import { port } from './serve'
+import { page } from '~utils'
+
+const url = `http://localhost:${port}`
+
+test('should work when inlined', async () => {
+  await page.goto(`${url}/static-light`)
+  expect(await page.textContent('.static-light')).toMatch('42')
+})
+
+test('should work when output', async () => {
+  await page.goto(`${url}/static-heavy`)
+  expect(await page.textContent('.static-heavy')).toMatch('24')
+})

playground/ssr-wasm/package.json

@@ -0,0 +1,15 @@
+{
+  "name": "@vitejs/test-ssr-wasm",
+  "private": true,
+  "version": "0.0.0",
+  "type": "module",
+  "scripts": {
+    "dev": "node server",
+    "serve": "NODE_ENV=production node server",
+    "debug": "node --inspect-brk server"
+  },
+  "dependencies": {},
+  "devDependencies": {
+    "express": "^5.1.0"
+  }
+}

playground/ssr-wasm/server.js

@@ -0,0 +1,56 @@
+import express from 'express'
+
+const isTest = process.env.VITEST
+
+export async function createServer(root = process.cwd(), hmrPort) {
+  const app = express()
+
+  /**
+   * @type {import('vite').ViteDevServer}
+   */
+  const vite = await (
+    await import('vite')
+  ).createServer({
+    root,
+    logLevel: isTest ? 'error' : 'info',
+    server: {
+      middlewareMode: true,
+      watch: {
+        // During tests we edit the files too fast and sometimes chokidar
+        // misses change events, so enforce polling for consistency
+        usePolling: true,
+        interval: 100,
+      },
+      hmr: {
+        port: hmrPort,
+      },
+    },
+    appType: 'custom',
+  })
+  // use vite's connect instance as middleware
+  app.use(vite.middlewares)
+
+  app.use('*all', async (req, res, next) => {
+    try {
+      const url = req.originalUrl
+      const render = (await vite.ssrLoadModule('/src/app.js')).render
+      const html = await render(url)
+      res.status(200).set({ 'Content-Type': 'text/html' }).end(html)
+    } catch (e) {
+      vite && vite.ssrFixStacktrace(e)
+      if (isTest) throw e
+      console.log(e.stack)
+      res.status(500).end(e.stack)
+    }
+  })
+
+  return { app, vite }
+}
+
+if (!isTest) {
+  createServer().then(({ app }) =>
+    app.listen(5173, () => {
+      console.log('http://localhost:5173')
+    }),
+  )
+}

playground/ssr-wasm/src/app.js

@@ -0,0 +1,8 @@
+export async function render(url) {
+  switch (url) {
+    case '/static-light':
+      return (await import('./static-light')).render()
+    case '/static-heavy':
+      return (await import('./static-heavy')).render()
+  }
+}

playground/ssr-wasm/src/heavy.wasm

@@ -0,0 +1 @@
+../../wasm/heavy.wasm

playground/ssr-wasm/src/light.wasm

@@ -0,0 +1 @@
+../../wasm/light.wasm

playground/ssr-wasm/src/static-heavy.js

@@ -0,0 +1,12 @@
+import heavy from './heavy.wasm?init'
+
+export async function render() {
+  let result
+  const { exported_func } = await heavy({
+    imports: {
+      imported_func: (res) => (result = res),
+    },
+  }).then((i) => i.exports)
+  exported_func()
+  return `<div class="static-heavy">${result}</div>`
+}

playground/ssr-wasm/src/static-light.js

@@ -0,0 +1,12 @@
+import light from './light.wasm?init'
+
+export async function render() {
+  let result
+  const { exported_func } = await light({
+    imports: {
+      imported_func: (res) => (result = res),
+    },
+  }).then((i) => i.exports)
+  exported_func()
+  return `<div class="static-light">${result}</div>`
+}