fix: concatenation should merge module’s chunk init fragments (#7504)

Author: fi3ework

crates/rspack_core/src/concatenated_module.rs

@@ -1655,11 +1655,16 @@ impl ConcatenatedModule {
       let codegen_res = module.code_generation(compilation, runtime, Some(concatenation_scope))?;
       let CodeGenerationResult {
         mut inner,
-        chunk_init_fragments,
+        mut chunk_init_fragments,
         runtime_requirements,
         concatenation_scope,
         ..
       } = codegen_res;
+
+      if let Some(fragments) = codegen_res.data.get::<ChunkInitFragments>() {
+        chunk_init_fragments.extend(fragments.iter().cloned());
+      }
+
       let concatenation_scope = concatenation_scope.expect("should have concatenation_scope");
       let source = inner
         .remove(&SourceType::JavaScript)

packages/rspack-test-tools/tests/configCases/output-module/node-shims-in-concatenated-module/file.js

@@ -0,0 +1,6 @@
+const fileURLToPath = "";
+const file = __filename;
+const dir = __dirname;
+const dir2 = `${__dirname}/`;
+
+export { file, dir, dir2 };

packages/rspack-test-tools/tests/configCases/output-module/node-shims-in-concatenated-module/index.js

@@ -0,0 +1,10 @@
+import { dir, dir2, file } from './file.js'
+
+it("should generate correct __dirname", () => {
+	expect(dir).toMatch(/[\\/]node-shims-in-concatenated-module$/);
+	expect(dir2).toMatch(/[\\/]node-shims-in-concatenated-module\/$/);
+});
+
+it("should generate correct __filename", () => {
+	expect(file).toMatch(/[\\/]main.mjs$/);
+});

packages/rspack-test-tools/tests/configCases/output-module/node-shims-in-concatenated-module/rspack.config.js

@@ -0,0 +1,14 @@
+/** @type {import("@rspack/core").Configuration} */
+module.exports = {
+	entry: {
+		main: "./index.js"
+	},
+	output: {
+		filename: "[name].mjs",
+		module: true
+	},
+	experiments: {
+		outputModule: true
+	},
+	target: "node14"
+};

packages/rspack-test-tools/tests/configCases/output-module/node-shims-in-concatenated-module/test.config.js

@@ -0,0 +1,5 @@
+module.exports = {
+	findBundle() {
+		return "./main.mjs";
+	}
+};