fix: support multiple environment

web-infra-dev · Aug 6, 2024 · f34a8f6 · f34a8f6
1 parent 5ac9308
commit f34a8f6
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/packages/plugin-dts/src/index.ts b/packages/plugin-dts/src/index.ts
@@ -33,7 +33,7 @@ export const pluginDts = (options: pluginDtsOptions): RsbuildPlugin => ({
     options.distPath =
       options.distPath ?? config.output?.distPath?.root ?? 'dist';
 
-    let dtsPromise: Promise<void>;
+    const dtsPromises: Promise<void>[] = [];
 
     api.onBeforeEnvironmentCompile(
       ({ isWatch, isFirstCompile, environment }) => {
@@ -55,19 +55,21 @@ export const pluginDts = (options: pluginDtsOptions): RsbuildPlugin => ({
 
         childProcess.send(dtsGenOptions);
 
-        dtsPromise = new Promise((resolve, reject) => {
-          childProcess.on('message', (message) => {
-            if (message === 'success') {
-              resolve();
-            } else if (message === 'error') {
-              reject(
-                new Error(
-                  `Error occurred in ${environment.name} dts generation`,
-                ),
-              );
-            }
-          });
-        });
+        dtsPromises.push(
+          new Promise((resolve, reject) => {
+            childProcess.on('message', (message) => {
+              if (message === 'success') {
+                resolve();
+              } else if (message === 'error') {
+                reject(
+                  new Error(
+                    `Error occurred in ${environment.name} dts generation`,
+                  ),
+                );
+              }
+            });
+          }),
+        );
       },
     );
 
@@ -76,7 +78,7 @@ export const pluginDts = (options: pluginDtsOptions): RsbuildPlugin => ({
         return;
       }
 
-      await dtsPromise;
+      await Promise.all(dtsPromises);
     });
   },
 });