feat: support chunkGroup.getModulePreOrderIndex (#8588)

crates/node_binding/binding.d.ts

@@ -87,6 +87,8 @@ export declare class JsChunkGroup {
   getParents(): JsChunkGroup[]
   getRuntimeChunk(): JsChunk
   getFiles(): Array<string>
+  getModulePreOrderIndex(module: JsModule): number | null
+  getModulePostOrderIndex(module: JsModule): number | null
 }
 
 export declare class JsCompilation {

crates/rspack_binding_values/src/chunk_group.rs

@@ -6,7 +6,7 @@ use rspack_core::{ChunkGroup, ChunkGroupUkey, Compilation, CompilationId};
 use rspack_napi::OneShotRef;
 use rustc_hash::FxHashMap as HashMap;
 
-use crate::{JsChunkWrapper, JsModuleWrapper};
+use crate::{JsChunkWrapper, JsModule, JsModuleWrapper};
 
 #[napi]
 pub struct JsChunkGroup {
@@ -89,15 +89,15 @@ impl JsChunkGroup {
 impl JsChunkGroup {
   #[napi]
   pub fn is_initial(&self) -> napi::Result<bool> {
-    let (_, chunk_graph) = self.as_ref()?;
-    Ok(chunk_graph.is_initial())
+    let (_, chunk_group) = self.as_ref()?;
+    Ok(chunk_group.is_initial())
   }
 
   #[napi(ts_return_type = "JsChunkGroup[]")]
   pub fn get_parents(&self) -> napi::Result<Vec<JsChunkGroupWrapper>> {
-    let (compilation, chunk_graph) = self.as_ref()?;
+    let (compilation, chunk_group) = self.as_ref()?;
     Ok(
-      chunk_graph
+      chunk_group
         .parents
         .iter()
         .map(|ukey| JsChunkGroupWrapper::new(*ukey, compilation))
@@ -107,16 +107,16 @@ impl JsChunkGroup {
 
   #[napi(ts_return_type = "JsChunk")]
   pub fn get_runtime_chunk(&self) -> napi::Result<JsChunkWrapper> {
-    let (compilation, chunk_graph) = self.as_ref()?;
-    let chunk_ukey = chunk_graph.get_runtime_chunk(&compilation.chunk_group_by_ukey);
+    let (compilation, chunk_group) = self.as_ref()?;
+    let chunk_ukey = chunk_group.get_runtime_chunk(&compilation.chunk_group_by_ukey);
     Ok(JsChunkWrapper::new(chunk_ukey, compilation))
   }
 
   #[napi]
   pub fn get_files(&self) -> napi::Result<Vec<&String>> {
-    let (compilation, chunk_graph) = self.as_ref()?;
+    let (compilation, chunk_group) = self.as_ref()?;
     Ok(
-      chunk_graph
+      chunk_group
         .chunks
         .iter()
         .filter_map(|chunk_ukey| {
@@ -129,6 +129,27 @@ impl JsChunkGroup {
         .collect::<Vec<_>>(),
     )
   }
+
+  #[napi]
+  pub fn get_module_pre_order_index(&self, module: &JsModule) -> napi::Result<Option<u32>> {
+    let (_, chunk_group) = self.as_ref()?;
+    Ok(
+      chunk_group
+        .module_pre_order_index(&module.identifier)
+        .map(|v| v as u32),
+    )
+  }
+
+  #[napi]
+  pub fn get_module_post_order_index(&self, module: &JsModule) -> napi::Result<Option<u32>> {
+    let (_, chunk_group) = self.as_ref()?;
+
+    Ok(
+      chunk_group
+        .module_post_order_index(&module.identifier)
+        .map(|v| v as u32),
+    )
+  }
 }
 
 thread_local! {

crates/rspack_core/src/chunk_group.rs

@@ -76,6 +76,14 @@ impl ChunkGroup {
     self.parents.iter()
   }
 
+  pub fn module_pre_order_index(&self, module_identifier: &ModuleIdentifier) -> Option<usize> {
+    // A module could split into another ChunkGroup, which doesn't have the module_post_order_indices of the module
+    self
+      .module_pre_order_indices
+      .get(module_identifier)
+      .copied()
+  }
+
   pub fn module_post_order_index(&self, module_identifier: &ModuleIdentifier) -> Option<usize> {
     // A module could split into another ChunkGroup, which doesn't have the module_post_order_indices of the module
     self

packages/rspack-test-tools/tests/configCases/chunks/module-order-index/index.js

@@ -0,0 +1,3 @@
+it("should compile", function (done) {
+	done()
+});

packages/rspack-test-tools/tests/configCases/chunks/module-order-index/rspack.config.js

@@ -0,0 +1,24 @@
+/** @type {import("@rspack/core").Configuration} */
+module.exports = {
+	target: "node",
+	entry: {
+		'main': './index.js'
+	},
+	plugins: [
+		{
+			/**@param {import("@rspack/core").Compiler} compiler */
+			apply(compiler) {
+				compiler.hooks.thisCompilation.tap('test', (compilation) => {
+					compilation.hooks.afterSeal.tap('test', () => {
+						let entrypoint = compilation.entrypoints.get('main')
+
+						compilation.chunkGraph.getChunkModules(entrypoint.chunks[0]).forEach((m) => {
+							expect(entrypoint.getModulePreOrderIndex(m)).toBeDefined();
+							expect(entrypoint.getModulePostOrderIndex(m)).toBeDefined();
+						})
+					})
+				})
+			}
+		}
+	]
+};

packages/rspack/etc/core.api.md

@@ -511,6 +511,10 @@ export class ChunkGroup {
     // (undocumented)
     getFiles(): ReadonlyArray<string>;
     // (undocumented)
+    getModulePostOrderIndex(module: Module): number | null;
+    // (undocumented)
+    getModulePreOrderIndex(module: Module): number | null;
+    // (undocumented)
     getParents(): ReadonlyArray<ChunkGroup>;
     // (undocumented)
     readonly index?: number;

packages/rspack/src/ChunkGroup.ts

@@ -74,6 +74,14 @@ export class ChunkGroup {
 	isInitial(): boolean {
 		return this.#inner.isInitial();
 	}
+
+	getModulePreOrderIndex(module: Module) {
+		return this.#inner.getModulePreOrderIndex(Module.__to_binding(module));
+	}
+
+	getModulePostOrderIndex(module: Module) {
+		return this.#inner.getModulePostOrderIndex(Module.__to_binding(module));
+	}
 }
 
 interface ChunkGroupOrigin {