feat(enhanced): chunk matcher to federation runtime (#2349)

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

Author: ScriptedAlchemy

.changeset/green-news-leave.md

@@ -0,0 +1,5 @@
+---
+'@module-federation/enhanced': patch
+---
+
+add chunk matcher logic to federation runtime module

packages/enhanced/src/lib/container/runtime/FederationRuntimeModule.ts

@@ -1,4 +1,10 @@
 import { normalizeWebpackPath } from '@module-federation/sdk/normalize-webpack-path';
+const compileBooleanMatcher = require(
+  normalizeWebpackPath('webpack/lib/util/compileBooleanMatcher'),
+) as typeof import('webpack/lib/util/compileBooleanMatcher');
+const { getUndoPath } = require(
+  normalizeWebpackPath('webpack/lib/util/identifier'),
+) as typeof import('webpack/lib/util/identifier');
 // inspired by react-refresh-webpack-plugin
 import getFederationGlobal from './getFederationGlobal';
 import { NormalizedRuntimeInitOptionsWithOutShared } from '../../../types/runtime';
@@ -27,10 +33,50 @@ class FederationRuntimeModule extends RuntimeModule {
    * @returns {string | null} runtime code
    */
   override generate() {
+    let matcher: string | boolean = false;
+    let rootOutputDir: string | undefined;
+    if (this.compilation && this.chunk) {
+      const jsModulePlugin =
+        this.compilation.compiler.webpack?.javascript
+          ?.JavascriptModulesPlugin ||
+        require(
+          normalizeWebpackPath(
+            'webpack/lib/javascript/JavascriptModulesPlugin',
+          ),
+        );
+      const { chunkHasJs } = jsModulePlugin;
+      if (this.runtimeRequirements.has(RuntimeGlobals.ensureChunkHandlers)) {
+        const conditionMap = this.compilation.chunkGraph.getChunkConditionMap(
+          this.chunk,
+          chunkHasJs,
+        );
+        const hasJsMatcher = compileBooleanMatcher(conditionMap);
+        if (typeof hasJsMatcher === 'boolean') {
+          matcher = hasJsMatcher;
+        } else {
+          matcher = hasJsMatcher('chunkId');
+        }
+        const outputName = this.compilation.getPath(
+          jsModulePlugin.getChunkFilenameTemplate(
+            this.chunk,
+            this.compilation.outputOptions,
+          ),
+          { chunk: this.chunk, contentHashType: 'javascript' },
+        );
+        rootOutputDir = getUndoPath(
+          outputName,
+          this.compilation.outputOptions.path || '',
+          false,
+        );
+      }
+    }
+
     return Template.asString([
       getFederationGlobal(
         Template,
         RuntimeGlobals,
+        matcher,
+        rootOutputDir,
         this.initOptionsWithoutShared,
       ),
     ]);

packages/enhanced/src/lib/container/runtime/getFederationGlobal.ts

@@ -10,6 +10,8 @@ const { Template } = require(
 function getFederationGlobal(
   template: typeof Template,
   runtimeGlobals: typeof RuntimeGlobals,
+  matcher: string | boolean,
+  rootOutputDir: string | undefined,
   initOptionsWithoutShared: NormalizedRuntimeInitOptionsWithOutShared,
 ): string {
   const federationGlobal = getFederationGlobalScope(runtimeGlobals);
@@ -21,6 +23,8 @@ function getFederationGlobal(
       `${federationGlobal} = {`,
       template.indent([
         `initOptions: ${initOptionsStrWithoutShared},`,
+        `chunkMatcher: function(chunkId) {return ${matcher}},`,
+        `rootOutputDir: ${JSON.stringify(rootOutputDir || '')},`,
         `initialConsumes: undefined,`,
         'bundlerRuntimeOptions: {}',
       ]),