fix: wasm file not found error (#677)

Co-authored-by: Copilot <[email protected]>

Author: 9aoy

e2e/wasm/index.test.ts

@@ -0,0 +1,11 @@
+import { expect, test } from '@rstest/core';
+
+// TODO: need to fix below error after bump rspack 1.6.2+
+// WebAssemb ly.instantiate（）： length overflow while decoding section length @+13
+test.fails('WASM factorial', async () => {
+  const { _Z4facti: AsyncFactorial } = await import('./src/factorial.wasm');
+
+  expect(AsyncFactorial(1)).toBe(1);
+  expect(AsyncFactorial(2)).toBe(2);
+  expect(AsyncFactorial(3)).toBe(6);
+});

e2e/wasm/src/factorial.wasm

@@ -0,0 +1 @@
+export function _Z4facti(n: number): number;

e2e/wasm/src/factorial.wasm.d.ts

@@ -15,7 +15,7 @@ const autoExternalNodeModules: (
     return callback();
   }
 
-  if (request.startsWith('@swc/helpers/')) {
+  if (request.startsWith('@swc/helpers/') || request.endsWith('.wasm')) {
     // @swc/helper is a special case (Load by require but resolve to esm)
     return callback();
   }

packages/core/src/core/plugins/external.ts

@@ -48,6 +48,17 @@ class MockRuntimeRspackPlugin {
 
             module.source!.source = Buffer.from(finalSource);
           }
+
+          if (module.name === 'async_wasm_loading') {
+            const finalSource = module.source!.source.toString('utf-8').replace(
+              // Replace readFile with readWasmFile to use the custom WASM file loader
+              // Hard coded in EJS template https://github.com/web-infra-dev/rspack/tree/7df875eb3ca3bb4bcb21836fdf4e6be1f38a057c/crates/rspack_plugin_wasm/src/runtime
+              'readFile(',
+              'readWasmFile(',
+            );
+
+            module.source!.source = Buffer.from(finalSource);
+          }
         },
       );
     });

packages/core/src/core/plugins/mockRuntime.ts

@@ -186,6 +186,22 @@ export const loadModule = ({
       assetFiles,
       interopDefault,
     ),
+    readWasmFile: (
+      wasmPath: string,
+      callback: (err: Error | null, data?: Buffer) => void,
+    ) => {
+      const joinedPath = isRelativePath(wasmPath)
+        ? path.join(path.dirname(distPath), wasmPath)
+        : wasmPath;
+      const content = assetFiles[path.normalize(joinedPath)];
+      if (content) {
+        callback(null, Buffer.from(content, 'utf-8'));
+      } else {
+        callback(
+          new Error(`WASM file ${joinedPath} not found in asset files.`),
+        );
+      }
+    },
     __rstest_dynamic_import__: defineRstestDynamicImport({
       testPath,
       interopDefault,

packages/core/src/runtime/worker/loadModule.ts

@@ -67,8 +67,11 @@ export async function printError(
       fullStack: error.fullStack,
       getSourcemap,
     });
-
-    if (!stackFrames.length && error.stack.length) {
+    if (
+      !stackFrames.length &&
+      !(error.fullStack || isDebug()) &&
+      !error.stack.endsWith(error.message)
+    ) {
       logger.log(
         color.gray(
           "No error stack found, set 'DEBUG=rstest' to show fullStack.",