refactor: using build_chunk_graph_artifact

Author: hardfist

crates/rspack_binding_api/src/chunk.rs

@@ -20,7 +20,11 @@ pub struct Chunk {
 impl Chunk {
   fn as_ref(&self) -> napi::Result<(&'static Compilation, &'static rspack_core::Chunk)> {
     let compilation = unsafe { self.compilation.as_ref() };
-    if let Some(chunk) = compilation.chunk_by_ukey.get(&self.chunk_ukey) {
+    if let Some(chunk) = compilation
+      .build_chunk_graph_artifact
+      .chunk_by_ukey
+      .get(&self.chunk_ukey)
+    {
       Ok((compilation, chunk))
     } else {
       Err(napi::Error::from_reason(format!(
@@ -163,27 +167,27 @@ impl Chunk {
   #[napi]
   pub fn is_only_initial(&self) -> napi::Result<bool> {
     let (compilation, chunk) = self.as_ref()?;
-    Ok(chunk.is_only_initial(&compilation.chunk_group_by_ukey))
+    Ok(chunk.is_only_initial(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey))
   }
 
   #[napi]
   pub fn can_be_initial(&self) -> napi::Result<bool> {
     let (compilation, chunk) = self.as_ref()?;
-    Ok(chunk.can_be_initial(&compilation.chunk_group_by_ukey))
+    Ok(chunk.can_be_initial(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey))
   }
 
   #[napi]
   pub fn has_runtime(&self) -> napi::Result<bool> {
     let (compilation, chunk) = self.as_ref()?;
-    Ok(chunk.has_runtime(&compilation.chunk_group_by_ukey))
+    Ok(chunk.has_runtime(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey))
   }
 
   #[napi(ts_return_type = "Chunk[]")]
   pub fn get_all_async_chunks(&self) -> napi::Result<Vec<ChunkWrapper>> {
     let (compilation, chunk) = self.as_ref()?;
     Ok(
       chunk
-        .get_all_async_chunks(&compilation.chunk_group_by_ukey)
+        .get_all_async_chunks(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey)
         .into_iter()
         .map(|chunk_ukey| ChunkWrapper::new(chunk_ukey, compilation))
         .collect::<Vec<_>>(),
@@ -195,7 +199,7 @@ impl Chunk {
     let (compilation, chunk) = self.as_ref()?;
     Ok(
       chunk
-        .get_all_initial_chunks(&compilation.chunk_group_by_ukey)
+        .get_all_initial_chunks(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey)
         .into_iter()
         .map(|chunk_ukey| ChunkWrapper::new(chunk_ukey, compilation))
         .collect::<Vec<_>>(),
@@ -207,7 +211,7 @@ impl Chunk {
     let (compilation, chunk) = self.as_ref()?;
     Ok(
       chunk
-        .get_all_referenced_chunks(&compilation.chunk_group_by_ukey)
+        .get_all_referenced_chunks(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey)
         .into_iter()
         .map(|chunk_ukey| ChunkWrapper::new(chunk_ukey, compilation))
         .collect::<Vec<_>>(),
@@ -220,7 +224,12 @@ impl Chunk {
     let mut groups = chunk
       .groups()
       .iter()
-      .filter_map(|group| compilation.chunk_group_by_ukey.get(group))
+      .filter_map(|group| {
+        compilation
+          .build_chunk_graph_artifact
+          .chunk_group_by_ukey
+          .get(group)
+      })
       .collect::<Vec<_>>();
     groups.sort_unstable_by(|a, b| a.index.cmp(&b.index));
     Ok(
@@ -235,7 +244,8 @@ impl Chunk {
   pub fn get_entry_options(&self) -> napi::Result<Option<EntryOptionsDTO>> {
     let (compilation, chunk) = self.as_ref()?;
 
-    let entry_options = chunk.get_entry_options(&compilation.chunk_group_by_ukey);
+    let entry_options =
+      chunk.get_entry_options(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey);
 
     Ok(entry_options.map(|options| EntryOptionsDTO::new(options.clone())))
   }

crates/rspack_binding_api/src/chunk_graph.rs

@@ -48,8 +48,12 @@ impl ChunkGraph {
     let compilation = self.as_ref()?;
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .chunk_graph
-        .has_chunk_entry_dependent_chunks(&chunk.chunk_ukey, &compilation.chunk_group_by_ukey),
+        .has_chunk_entry_dependent_chunks(
+          &chunk.chunk_ukey,
+          &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
+        ),
     )
   }
 
@@ -59,6 +63,7 @@ impl ChunkGraph {
 
     let module_graph = compilation.get_module_graph();
     let modules = compilation
+      .build_chunk_graph_artifact
       .chunk_graph
       .get_chunk_modules(&chunk.chunk_ukey, module_graph);
 
@@ -80,6 +85,7 @@ impl ChunkGraph {
     let compilation = self.as_ref()?;
 
     let modules = compilation
+      .build_chunk_graph_artifact
       .chunk_graph
       .get_chunk_entry_modules(&chunk.chunk_ukey);
     let module_graph = compilation.get_module_graph();
@@ -98,6 +104,7 @@ impl ChunkGraph {
 
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .chunk_graph
         .get_number_of_entry_modules(&chunk.chunk_ukey) as u32,
     )
@@ -111,11 +118,12 @@ impl ChunkGraph {
     let compilation = self.as_ref()?;
 
     let chunks = compilation
+      .build_chunk_graph_artifact
       .chunk_graph
       .get_chunk_entry_dependent_chunks_iterable(
         &chunk.chunk_ukey,
-        &compilation.chunk_by_ukey,
-        &compilation.chunk_group_by_ukey,
+        &compilation.build_chunk_graph_artifact.chunk_by_ukey,
+        &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
       );
 
     Ok(
@@ -135,11 +143,14 @@ impl ChunkGraph {
     let compilation = self.as_ref()?;
     let module_graph = compilation.get_module_graph();
 
-    let chunk_modules = compilation.chunk_graph.get_chunk_modules_by_source_type(
-      &chunk.chunk_ukey,
-      SourceType::from(source_type.as_str()),
-      module_graph,
-    );
+    let chunk_modules = compilation
+      .build_chunk_graph_artifact
+      .chunk_graph
+      .get_chunk_modules_by_source_type(
+        &chunk.chunk_ukey,
+        SourceType::from(source_type.as_str()),
+        module_graph,
+      );
 
     Ok(
       chunk_modules
@@ -155,6 +166,7 @@ impl ChunkGraph {
 
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .chunk_graph
         .get_module_chunks(module.identifier)
         .iter()
@@ -205,8 +217,12 @@ impl ChunkGraph {
     let compilation = self.as_ref()?;
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .chunk_graph
-        .get_block_chunk_group(&js_block.block_id, &compilation.chunk_group_by_ukey)
+        .get_block_chunk_group(
+          &js_block.block_id,
+          &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
+        )
         .map(|chunk_group| ChunkGroupWrapper::new(chunk_group.ukey, compilation)),
     )
   }

crates/rspack_binding_api/src/chunk_group.rs

@@ -21,7 +21,11 @@ pub struct ChunkGroup {
 impl ChunkGroup {
   fn as_ref(&self) -> napi::Result<(&'static Compilation, &'static rspack_core::ChunkGroup)> {
     let compilation = unsafe { self.compilation.as_ref() };
-    if let Some(chunk_group) = compilation.chunk_group_by_ukey.get(&self.chunk_group_ukey) {
+    if let Some(chunk_group) = compilation
+      .build_chunk_graph_artifact
+      .chunk_group_by_ukey
+      .get(&self.chunk_group_ukey)
+    {
       Ok((compilation, chunk_group))
     } else {
       Err(napi::Error::from_reason(format!(
@@ -131,7 +135,8 @@ impl ChunkGroup {
   #[napi(ts_return_type = "Chunk")]
   pub fn get_runtime_chunk(&self) -> napi::Result<ChunkWrapper> {
     let (compilation, chunk_group) = self.as_ref()?;
-    let chunk_ukey = chunk_group.get_runtime_chunk(&compilation.chunk_group_by_ukey);
+    let chunk_ukey =
+      chunk_group.get_runtime_chunk(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey);
     Ok(ChunkWrapper::new(chunk_ukey, compilation))
   }
 
@@ -151,6 +156,7 @@ impl ChunkGroup {
         .iter()
         .filter_map(|chunk_ukey| {
           compilation
+            .build_chunk_graph_artifact
             .chunk_by_ukey
             .get(chunk_ukey)
             .map(|chunk| chunk.files().iter())

crates/rspack_binding_api/src/compilation/chunks.rs

@@ -40,14 +40,15 @@ impl Chunks {
   #[napi(getter)]
   pub fn size(&self) -> napi::Result<u32> {
     let compilation = self.as_ref()?;
-    Ok(compilation.chunk_by_ukey.len() as u32)
+    Ok(compilation.build_chunk_graph_artifact.chunk_by_ukey.len() as u32)
   }
 
   #[napi(js_name = "_values", ts_return_type = "Chunk[]")]
   pub fn values(&self) -> napi::Result<Vec<ChunkWrapper>> {
     let compilation = self.as_ref()?;
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .chunk_by_ukey
         .keys()
         .map(|chunk_ukey| ChunkWrapper::new(*chunk_ukey, compilation))
@@ -58,6 +59,11 @@ impl Chunks {
   #[napi(js_name = "_has")]
   pub fn has(&self, chunk: &Chunk) -> napi::Result<bool> {
     let compilation = self.as_ref()?;
-    Ok(compilation.chunk_by_ukey.contains(&chunk.chunk_ukey))
+    Ok(
+      compilation
+        .build_chunk_graph_artifact
+        .chunk_by_ukey
+        .contains(&chunk.chunk_ukey),
+    )
   }
 }

crates/rspack_binding_api/src/compilation/mod.rs

@@ -240,19 +240,33 @@ impl JsCompilation {
   pub fn get_named_chunk_keys(&self) -> Result<Vec<String>> {
     let compilation = self.as_ref()?;
 
-    Ok(compilation.named_chunks.keys().cloned().collect::<Vec<_>>())
+    Ok(
+      compilation
+        .build_chunk_graph_artifact
+        .named_chunks
+        .keys()
+        .cloned()
+        .collect::<Vec<_>>(),
+    )
   }
 
   #[napi(ts_return_type = "Chunk")]
   pub fn get_named_chunk(&self, name: String) -> Result<Option<ChunkWrapper>> {
     let compilation = self.as_ref()?;
 
-    Ok(compilation.named_chunks.get(&name).and_then(|c| {
+    Ok(
       compilation
-        .chunk_by_ukey
-        .get(c)
-        .map(|chunk| ChunkWrapper::new(chunk.ukey(), compilation))
-    }))
+        .build_chunk_graph_artifact
+        .named_chunks
+        .get(&name)
+        .and_then(|c| {
+          compilation
+            .build_chunk_graph_artifact
+            .chunk_by_ukey
+            .get(c)
+            .map(|chunk| ChunkWrapper::new(chunk.ukey(), compilation))
+        }),
+    )
   }
 
   #[napi]
@@ -261,6 +275,7 @@ impl JsCompilation {
 
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .named_chunk_groups
         .keys()
         .cloned()
@@ -273,6 +288,7 @@ impl JsCompilation {
     let compilation = self.as_ref()?;
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .named_chunk_groups
         .get(&name)
         .map(|ukey| ChunkGroupWrapper::new(*ukey, compilation)),
@@ -394,6 +410,7 @@ impl JsCompilation {
 
     Ok(
       compilation
+        .build_chunk_graph_artifact
         .chunk_group_by_ukey
         .keys()
         .map(|ukey| ChunkGroupWrapper::new(*ukey, compilation))