nshen
diff --git a/‎src/index.ts‎
Lines changed: 18 additions & 9 deletions b/‎src/index.ts‎
Lines changed: 18 additions & 9 deletions
@@ -4,7 +4,10 @@ import chalk from 'chalk';
 import { isString } from 'narrowing';
 import { Plugin } from 'vite';
 
-function vitePluginWasmPack(crates: string[] | string): Plugin {
+function vitePluginWasmPack(
+  crates: string[] | string,
+  isNodeModule?: boolean
+): Plugin {
   const prefix = '@vite-plugin-wasm-pack@';
   const pkg = 'pkg'; // default folder of wasm-pack module
   let config_base: string;
@@ -17,7 +20,10 @@ function vitePluginWasmPack(crates: string[] | string): Plugin {
   const wasmMap = new Map<string, string>(); // { 'my_crate_bg.wasm': '../../wasm-game/pkg/wasm_game_bg.wasm' }
   cratePaths.forEach((cratePath) => {
     const wasmFile = wasmFilename(cratePath);
-    wasmMap.set(wasmFile, path.join(cratePath, pkg, wasmFile));
+    const wasmPath = isNodeModule
+      ? path.join('node_modules', cratePath, wasmFile)
+      : path.join(cratePath, pkg, wasmFile);
+    wasmMap.set(wasmFile, wasmPath);
   });
 
   return {
@@ -52,7 +58,9 @@ function vitePluginWasmPack(crates: string[] | string): Plugin {
 
     async buildStart(inputOptions) {
       for await (const cratePath of cratePaths) {
-        const pkgPath = path.join(cratePath, pkg);
+        const pkgPath = isNodeModule
+          ? path.join('node_modules', cratePath)
+          : path.join(cratePath, pkg);
         const crateName = path.basename(cratePath);
         if (!fs.existsSync(pkgPath)) {
           console.error(
@@ -65,12 +73,13 @@ function vitePluginWasmPack(crates: string[] | string): Plugin {
             `Can't find ${pkgPath}, run 'wasm-pack build ${cratePath} --target web' first`
           );
         }
-        // copy pkg generated by wasm-pack to node_modules
-        try {
-          await fs.copy(pkgPath, path.join('node_modules', crateName));
-        } catch (error) {
-          this.error(`copy crates failed`);
-          return;
+        if (!isNodeModule) {
+          // copy pkg generated by wasm-pack to node_modules
+          try {
+            await fs.copy(pkgPath, path.join('node_modules', crateName));
+          } catch (error) {
+            this.error(`copy crates failed`);
+          }
         }
         // replace default load path with '/assets/xxx.wasm'
         const jsName = crateName.replace('-', '_') + '.js';