feat: allow to access environment context in process assets (#2970)

web-infra-dev · Jul 21, 2024 · eebe00d · eebe00d
1 parent 718343c
commit eebe00d
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 15 deletions.
diff --git a/packages/core/src/initPlugins.ts b/packages/core/src/initPlugins.ts
@@ -209,26 +209,32 @@ export function getPluginAPI({
   const processAssets: ProcessAssetsFn = (descriptor, handler) => {
     const name = 'RsbuildProcessAssetsPlugin';
 
-    class RsbuildProcessAssetsPlugin {
-      apply(compiler: Compiler): void {
-        compiler.hooks.compilation.tap(name, (compilation) => {
-          compilation.hooks.processAssets.tapPromise(
-            {
-              name,
-              stage: mapProcessAssetsStage(compiler, descriptor.stage),
-            },
-            async (assets) => handler({ assets, compiler, compilation }),
-          );
-        });
-      }
-    }
-
-    hooks.modifyBundlerChain.tap((chain, { target }) => {
+    hooks.modifyBundlerChain.tap((chain, { target, environment }) => {
       // filter by targets
       if (descriptor.targets && !descriptor.targets.includes(target)) {
         return;
       }
 
+      class RsbuildProcessAssetsPlugin {
+        apply(compiler: Compiler): void {
+          compiler.hooks.compilation.tap(name, (compilation) => {
+            compilation.hooks.processAssets.tapPromise(
+              {
+                name,
+                stage: mapProcessAssetsStage(compiler, descriptor.stage),
+              },
+              async (assets) =>
+                handler({
+                  assets,
+                  compiler,
+                  compilation,
+                  environment,
+                }),
+            );
+          });
+        }
+      }
+
       chain
         .plugin(`RsbuildProcessAssetsPlugin#${processAssetsId++}`)
         .use(RsbuildProcessAssetsPlugin);

diff --git a/packages/core/src/types/plugin.ts b/packages/core/src/types/plugin.ts
@@ -277,6 +277,10 @@ export type ProcessAssetsHandler = (context: {
   assets: Record<string, Rspack.sources.Source>;
   compiler: Rspack.Compiler;
   compilation: Rspack.Compilation;
+  /**
+   * The environment context for current build.
+   */
+  environment: EnvironmentContext;
 }) => Promise<void> | void;
 
 export type ProcessAssetsFn = (