feat: support more moduleGraph and chunkGraph js api

crates/node_binding/binding.d.ts

@@ -19,13 +19,6 @@ export declare class ExternalObject<T> {
     [K: symbol]: T
   }
 }
-export declare class EntryDataDto {
-  get dependencies(): JsDependency[]
-  get includeDependencies(): JsDependency[]
-  get options(): EntryOptionsDto
-}
-export type EntryDataDTO = EntryDataDto
-
 export declare class EntryOptionsDto {
   get name(): string | undefined
   set name(name: string | undefined)
@@ -76,13 +69,15 @@ export declare class JsChunkGraph {
   getChunkModulesIterableBySourceType(chunk: JsChunk, sourceType: string): JsModule[]
   getModuleChunks(module: JsModule): JsChunk[]
   getModuleId(jsModule: JsModule): string | null
+  getBlockChunkGroup(jsBlock: JsDependenciesBlock): JsChunkGroup | null
 }
 
 export declare class JsChunkGroup {
   get chunks(): JsChunk[]
   get index(): number | undefined
   get name(): string | undefined
   get origins(): Array<JsChunkGroupOrigin>
+  get childrenIterable(): JsChunkGroup[]
   isInitial(): boolean
   getParents(): JsChunkGroup[]
   getRuntimeChunk(): JsChunk
@@ -99,7 +94,7 @@ export declare class JsCompilation {
   getOptimizationBailout(): Array<JsStatsOptimizationBailout>
   getChunks(): JsChunk[]
   getNamedChunkKeys(): Array<string>
-  getNamedChunk(name: string): JsChunk
+  getNamedChunk(name: string): JsChunk | null
   getNamedChunkGroupKeys(): Array<string>
   getNamedChunkGroup(name: string): JsChunkGroup
   setAssetSource(name: string, source: JsCompatSource): void
@@ -110,7 +105,7 @@ export declare class JsCompilation {
   emitAsset(filename: string, source: JsCompatSource, assetInfo: JsAssetInfo): void
   deleteAsset(filename: string): void
   renameAsset(filename: string, newName: string): void
-  get entrypoints(): Record<string, JsChunkGroup>
+  get entrypoints(): JsChunkGroup[]
   get chunkGroups(): JsChunkGroup[]
   get hash(): string | null
   dependencies(): JsDependencies
@@ -140,7 +135,7 @@ export declare class JsCompilation {
   addRuntimeModule(chunk: JsChunk, runtimeModule: JsAddingRuntimeModule): void
   get moduleGraph(): JsModuleGraph
   get chunkGraph(): JsChunkGraph
-  addInclude(args: [string, RawDependency, JsEntryOptions | undefined][], callback: (errMsg: Error | null, results: [string | null, JsModule][]) => void): void
+  addInclude(args: [string, RawDependency, RawEntryOptions | undefined][], callback: (errMsg: Error | null, results: [string | null, JsDependency | null, JsModule | null][]) => void): void
 }
 
 export declare class JsContextModuleFactoryAfterResolveData {
@@ -194,17 +189,26 @@ export declare class JsDependency {
   get request(): string | undefined
   get critical(): boolean
   set critical(val: boolean)
+  get ids(): Array<string> | undefined
 }
 
 export declare class JsEntries {
   clear(): void
   get size(): number
   has(key: string): boolean
-  set(key: string, value: JsEntryData | EntryDataDto): void
+  set(key: string, value: RawEntryData): void
   delete(key: string): boolean
-  get(key: string): EntryDataDto | undefined
+  get(key: string): JsEntryData | undefined
   keys(): Array<string>
-  values(): Array<EntryDataDto>
+  values(): Array<JsEntryData>
+}
+
+export declare class JsEntryData {
+  get dependencies(): JsDependency[]
+  set dependencies(dependencies: Array<JsDependency>)
+  get includeDependencies(): JsDependency[]
+  set includeDependencies(dependencies: Array<JsDependency>)
+  get options(): EntryOptionsDto
 }
 
 export declare class JsExportsInfo {
@@ -215,23 +219,28 @@ export declare class JsExportsInfo {
 }
 
 export declare class JsModule {
-  get context(): string | undefined
+  get constructorName(): string
+  get context(): string | null
   get originalSource(): JsCompatSource | undefined
-  get resource(): string | undefined
+  get resource(): string | null
   get moduleIdentifier(): string
   get nameForCondition(): string | undefined
-  get request(): string | undefined
+  get request(): string | null
   get userRequest(): string | undefined
   set userRequest(val: string)
-  get rawRequest(): string | undefined
+  get rawRequest(): string | null
   get factoryMeta(): JsFactoryMeta | undefined
   get type(): string
-  get layer(): string | undefined
+  get layer(): string | null
   get blocks(): JsDependenciesBlock[]
   get dependencies(): JsDependency[]
   size(ty?: string | undefined | null): number
   get modules(): JsModule[] | undefined
   get useSourceMap(): boolean
+  libIdent(options: JsLibIdentOptions): string | null
+  get resourceResolveData(): JsResourceData | null
+  get matchResource(): string | null
+  get loaders(): Array<string> | undefined
 }
 
 export declare class JsModuleGraph {
@@ -243,11 +252,16 @@ export declare class JsModuleGraph {
   getConnection(dependency: JsDependency): JsModuleGraphConnection | null
   getOutgoingConnections(module: JsModule): JsModuleGraphConnection[]
   getIncomingConnections(module: JsModule): JsModuleGraphConnection[]
+  getParentModule(jsDependency: JsDependency): JsModule | null
+  getParentBlockIndex(jsDependency: JsDependency): number
+  isAsync(module: JsModule): boolean
 }
 
 export declare class JsModuleGraphConnection {
   get dependency(): JsDependency
   get module(): JsModule | null
+  get resolvedModule(): JsModule | null
+  get originModule(): JsModule | null
 }
 
 export declare class JsResolver {
@@ -402,6 +416,7 @@ export interface JsAfterResolveData {
   request: string
   context: string
   issuer: string
+  issuerLayer?: string
   fileDependencies: Array<string>
   contextDependencies: Array<string>
   missingDependencies: Array<string>
@@ -504,6 +519,7 @@ export interface JsBeforeResolveArgs {
   request: string
   context: string
   issuer: string
+  issuerLayer?: string
 }
 
 export interface JsBuildMeta {
@@ -613,12 +629,6 @@ export interface JsDiagnosticLocation {
   length: number
 }
 
-export interface JsEntryData {
-  dependencies: Array<JsDependency>
-  includeDependencies: Array<JsDependency>
-  options: JsEntryOptions
-}
-
 export interface JsEntryOptions {
   name?: string
   runtime?: false | string
@@ -660,6 +670,7 @@ export interface JsFactorizeArgs {
   request: string
   context: string
   issuer: string
+  issuerLayer?: string
 }
 
 export interface JsFactoryMeta {
@@ -687,6 +698,10 @@ export interface JsHtmlPluginTag {
   asset?: string
 }
 
+export interface JsLibIdentOptions {
+  context: string
+}
+
 export interface JsLibraryAuxiliaryComment {
   root?: string
   commonjs?: string
@@ -718,15 +733,13 @@ export interface JsLibraryOptions {
 
 export interface JsLoaderContext {
   resourceData: Readonly<JsResourceData>
-  /** Will be deprecated. Use module.module_identifier instead */
-  _moduleIdentifier: Readonly<string>
   _module: JsModule
   hot: Readonly<boolean>
   /** Content maybe empty in pitching stage */
-  content: null | Buffer
+  content: null | Buffer | string
   additionalData?: any
   __internal__parseMeta: Record<string, string>
-  sourceMap?: Buffer
+  sourceMap?: JsSourceMap
   cacheable: boolean
   fileDependencies: Array<string>
   contextDependencies: Array<string>
@@ -791,6 +804,7 @@ export interface JsResolveArgs {
   request: string
   context: string
   issuer: string
+  issuerLayer?: string
 }
 
 export interface JsResolveForSchemeArgs {
@@ -854,6 +868,16 @@ export interface JsRuntimeRequirementInTreeResult {
   runtimeRequirements: JsRuntimeGlobals
 }
 
+export interface JsSourceMap {
+  version: number
+  file?: string
+  sources: Array<string>
+  sourcesContent?: Array<string>
+  names: Array<string>
+  mappings: string
+  sourceRoot?: string
+}
+
 export interface JsStatsAsset {
   type: string
   name: string
@@ -1361,6 +1385,12 @@ export interface RawDynamicEntryPluginOptions {
   entry: () => Promise<RawEntryDynamicResult[]>
 }
 
+export interface RawEntryData {
+  dependencies: Array<JsDependency>
+  includeDependencies: Array<JsDependency>
+  options: JsEntryOptions
+}
+
 export interface RawEntryDynamicResult {
   import: Array<string>
   options: JsEntryOptions

crates/node_binding/src/plugins/interceptor.rs

@@ -1388,6 +1388,7 @@ impl NormalModuleFactoryBeforeResolve for NormalModuleFactoryBeforeResolveTap {
           .as_ref()
           .map(|issuer| issuer.to_string())
           .unwrap_or_default(),
+        issuer_layer: data.issuer_layer.clone(),
       })
       .await
     {
@@ -1424,6 +1425,7 @@ impl NormalModuleFactoryFactorize for NormalModuleFactoryFactorizeTap {
           .as_ref()
           .map(|issuer| issuer.to_string())
           .unwrap_or_default(),
+        issuer_layer: data.issuer_layer.clone(),
       })
       .await
     {
@@ -1461,6 +1463,7 @@ impl NormalModuleFactoryResolve for NormalModuleFactoryResolveTap {
           .as_ref()
           .map(|issuer| issuer.to_string())
           .unwrap_or_default(),
+        issuer_layer: data.issuer_layer.clone(),
       })
       .await
     {
@@ -1523,6 +1526,7 @@ impl NormalModuleFactoryAfterResolve for NormalModuleFactoryAfterResolveTap {
           .as_ref()
           .map(|issuer| issuer.to_string())
           .unwrap_or_default(),
+        issuer_layer: data.issuer_layer.clone(),
         file_dependencies: data
           .file_dependencies
           .clone()

crates/rspack_binding_values/src/chunk_graph.rs

@@ -4,7 +4,9 @@ use napi::Result;
 use napi_derive::napi;
 use rspack_core::{ChunkGraph, Compilation, SourceType};
 
-use crate::{JsChunk, JsChunkWrapper, JsModule, JsModuleWrapper};
+use crate::{
+  JsChunk, JsChunkGroupWrapper, JsChunkWrapper, JsDependenciesBlock, JsModule, JsModuleWrapper,
+};
 
 #[napi]
 pub struct JsChunkGraph {
@@ -127,4 +129,18 @@ impl JsChunkGraph {
         .map(|module_id| module_id.as_str()),
     )
   }
+
+  #[napi(ts_return_type = "JsChunkGroup | null")]
+  pub fn get_block_chunk_group(
+    &self,
+    js_block: &JsDependenciesBlock,
+  ) -> napi::Result<Option<JsChunkGroupWrapper>> {
+    let compilation = self.as_ref()?;
+    Ok(
+      compilation
+        .chunk_graph
+        .get_block_chunk_group(&js_block.block_id, &compilation.chunk_group_by_ukey)
+        .map(|chunk_group| JsChunkGroupWrapper::new(chunk_group.ukey, compilation)),
+    )
+  }
 }

crates/rspack_binding_values/src/chunk_group.rs

@@ -83,10 +83,18 @@ impl JsChunkGroup {
 
     Ok(js_origins)
   }
-}
 
-#[napi]
-impl JsChunkGroup {
+  #[napi(getter, ts_return_type = "JsChunkGroup[]")]
+  pub fn children_iterable(&self) -> napi::Result<Vec<JsChunkGroupWrapper>> {
+    let (compilation, chunk_graph) = self.as_ref()?;
+    Ok(
+      chunk_graph
+        .children_iterable()
+        .map(|ukey| JsChunkGroupWrapper::new(*ukey, compilation))
+        .collect::<Vec<_>>(),
+    )
+  }
+
   #[napi]
   pub fn is_initial(&self) -> napi::Result<bool> {
     let (_, chunk_graph) = self.as_ref()?;