feat: support MF share treeshake

crates/node_binding/napi-binding.d.ts

@@ -548,10 +548,13 @@ export declare enum BuiltinPluginName {
   SplitChunksPlugin = 'SplitChunksPlugin',
   RemoveDuplicateModulesPlugin = 'RemoveDuplicateModulesPlugin',
   ShareRuntimePlugin = 'ShareRuntimePlugin',
+  SharedUsedExportsOptimizerPlugin = 'SharedUsedExportsOptimizerPlugin',
   ContainerPlugin = 'ContainerPlugin',
   ContainerReferencePlugin = 'ContainerReferencePlugin',
   ProvideSharedPlugin = 'ProvideSharedPlugin',
   ConsumeSharedPlugin = 'ConsumeSharedPlugin',
+  CollectSharedEntryPlugin = 'CollectSharedEntryPlugin',
+  SharedContainerPlugin = 'SharedContainerPlugin',
   ModuleFederationRuntimePlugin = 'ModuleFederationRuntimePlugin',
   ModuleFederationManifestPlugin = 'ModuleFederationManifestPlugin',
   NamedModuleIdsPlugin = 'NamedModuleIdsPlugin',
@@ -1876,6 +1879,11 @@ export interface RawCircularDependencyRspackPluginOptions {
   onEnd?: () => void
 }
 
+export interface RawCollectShareEntryPluginOptions {
+  consumes: Array<RawConsumeOptions>
+  filename?: string
+}
+
 export interface RawCompilerPlatform {
   web?: boolean | null
   browser?: boolean | null
@@ -1896,6 +1904,7 @@ export interface RawConsumeOptions {
   strictVersion: boolean
   singleton: boolean
   eager: boolean
+  treeShakingMode?: string
 }
 
 export interface RawConsumeSharedPluginOptions {
@@ -2613,6 +2622,12 @@ export interface RawOptimizationOptions {
   avoidEntryIife: boolean
 }
 
+export interface RawOptimizeSharedConfig {
+  shareKey: string
+  treeShaking: boolean
+  usedExports?: Array<string>
+}
+
 export interface RawOptions {
   name?: string
   mode?: undefined | 'production' | 'development' | 'none'
@@ -2712,6 +2727,7 @@ export interface RawProvideOptions {
   singleton?: boolean
   requiredVersion?: string | false | undefined
   strictVersion?: boolean
+  treeShakingMode?: string
 }
 
 export interface RawRelated {
@@ -2846,6 +2862,21 @@ export interface RawRuntimeChunkOptions {
   name: string | ((entrypoint: { name: string }) => string)
 }
 
+export interface RawSharedContainerPluginOptions {
+  name: string
+  request: string
+  version: string
+  fileName?: string
+  library: JsLibraryOptions
+}
+
+export interface RawSharedUsedExportsOptimizerPluginOptions {
+  shared: Array<RawOptimizeSharedConfig>
+  injectTreeShakingUsedExports?: boolean
+  manifestFileName?: string
+  statsFileName?: string
+}
+
 export interface RawSizeLimitsPluginOptions {
   assetFilter?: (assetFilename: string) => boolean
   hints?: "error" | "warning"
@@ -2889,6 +2920,8 @@ export interface RawSplitChunksOptions {
 export interface RawStatsBuildInfo {
   buildVersion: string
   buildName?: string
+  target?: Array<string>
+  plugins?: Array<string>
 }
 
 export interface RawStatsOptions {

crates/rspack_binding_api/src/plugins/interceptor.rs

@@ -81,6 +81,7 @@ use crate::{
     JsContextModuleFactoryAfterResolveDataWrapper, JsContextModuleFactoryAfterResolveResult,
     JsContextModuleFactoryBeforeResolveDataWrapper, JsContextModuleFactoryBeforeResolveResult,
   },
+  dependency::DependencyWrapper,
   html::{
     JsAfterEmitData, JsAfterTemplateExecutionData, JsAlterAssetTagGroupsData, JsAlterAssetTagsData,
     JsBeforeAssetTagGenerationData, JsBeforeEmitData,
@@ -97,7 +98,7 @@ use crate::{
   runtime::{
     JsAdditionalTreeRuntimeRequirementsArg, JsAdditionalTreeRuntimeRequirementsResult,
     JsCreateLinkData, JsCreateScriptData, JsLinkPrefetchData, JsLinkPreloadData, JsRuntimeGlobals,
-    JsRuntimeRequirementInTreeArg, JsRuntimeRequirementInTreeResult,
+    JsRuntimeRequirementInTreeArg, JsRuntimeRequirementInTreeResult, JsRuntimeSpec,
   },
   source::JsSourceToJs,
 };

crates/rspack_binding_api/src/raw_options/raw_builtins/mod.rs

@@ -32,7 +32,11 @@ use napi_derive::napi;
 use raw_dll::{RawDllReferenceAgencyPluginOptions, RawFlagAllModulesAsUsedPluginOptions};
 use raw_ids::RawOccurrenceChunkIdsPluginOptions;
 use raw_lightning_css_minimizer::RawLightningCssMinimizerRspackPluginOptions;
-use raw_mf::{RawModuleFederationManifestPluginOptions, RawModuleFederationRuntimePluginOptions};
+use raw_mf::{
+  RawCollectShareEntryPluginOptions, RawModuleFederationManifestPluginOptions,
+  RawModuleFederationRuntimePluginOptions, RawProvideOptions,
+  RawSharedUsedExportsOptimizerPluginOptions,
+};
 use raw_sri::RawSubresourceIntegrityPluginOptions;
 use rspack_core::{BoxPlugin, Plugin, PluginExt};
 use rspack_error::{Result, ToStringResultToRspackResultExt};
@@ -76,8 +80,9 @@ use rspack_plugin_lightning_css_minimizer::LightningCssMinimizerRspackPlugin;
 use rspack_plugin_limit_chunk_count::LimitChunkCountPlugin;
 use rspack_plugin_merge_duplicate_chunks::MergeDuplicateChunksPlugin;
 use rspack_plugin_mf::{
-  ConsumeSharedPlugin, ContainerPlugin, ContainerReferencePlugin, ModuleFederationManifestPlugin,
-  ModuleFederationRuntimePlugin, ProvideSharedPlugin, ShareRuntimePlugin,
+  CollectSharedEntryPlugin, ConsumeSharedPlugin, ContainerPlugin, ContainerReferencePlugin,
+  ModuleFederationManifestPlugin, ModuleFederationRuntimePlugin, ProvideSharedPlugin,
+  ShareRuntimePlugin, SharedContainerPlugin, SharedUsedExportsOptimizerPlugin,
 };
 use rspack_plugin_module_info_header::ModuleInfoHeaderPlugin;
 use rspack_plugin_module_replacement::{ContextReplacementPlugin, NormalModuleReplacementPlugin};
@@ -118,7 +123,7 @@ use self::{
   raw_limit_chunk_count::RawLimitChunkCountPluginOptions,
   raw_mf::{
     RawConsumeSharedPluginOptions, RawContainerPluginOptions, RawContainerReferencePluginOptions,
-    RawProvideOptions,
+    RawSharedContainerPluginOptions,
   },
   raw_normal_replacement::RawNormalModuleReplacementPluginOptions,
   raw_runtime_chunk::RawRuntimeChunkOptions,
@@ -171,10 +176,13 @@ pub enum BuiltinPluginName {
   SplitChunksPlugin,
   RemoveDuplicateModulesPlugin,
   ShareRuntimePlugin,
+  SharedUsedExportsOptimizerPlugin,
   ContainerPlugin,
   ContainerReferencePlugin,
   ProvideSharedPlugin,
   ConsumeSharedPlugin,
+  CollectSharedEntryPlugin,
+  SharedContainerPlugin,
   ModuleFederationRuntimePlugin,
   ModuleFederationManifestPlugin,
   NamedModuleIdsPlugin,
@@ -471,6 +479,12 @@ impl<'a> BuiltinPlugin<'a> {
         )
         .boxed(),
       ),
+      BuiltinPluginName::SharedUsedExportsOptimizerPlugin => {
+        let options = downcast_into::<RawSharedUsedExportsOptimizerPluginOptions>(self.options)
+          .map_err(|report| napi::Error::from_reason(report.to_string()))?
+          .into();
+        plugins.push(SharedUsedExportsOptimizerPlugin::new(options).boxed());
+      }
       BuiltinPluginName::ContainerPlugin => {
         plugins.push(
           ContainerPlugin::new(
@@ -500,6 +514,18 @@ impl<'a> BuiltinPlugin<'a> {
         provides.sort_unstable_by_key(|(k, _)| k.to_string());
         plugins.push(ProvideSharedPlugin::new(provides).boxed())
       }
+      BuiltinPluginName::CollectSharedEntryPlugin => {
+        let options = downcast_into::<RawCollectShareEntryPluginOptions>(self.options)
+          .map_err(|report| napi::Error::from_reason(report.to_string()))?
+          .into();
+        plugins.push(CollectSharedEntryPlugin::new(options).boxed())
+      }
+      BuiltinPluginName::SharedContainerPlugin => {
+        let options = downcast_into::<RawSharedContainerPluginOptions>(self.options)
+          .map_err(|report| napi::Error::from_reason(report.to_string()))?
+          .into();
+        plugins.push(SharedContainerPlugin::new(options).boxed())
+      }
       BuiltinPluginName::ConsumeSharedPlugin => plugins.push(
         ConsumeSharedPlugin::new(
           downcast_into::<RawConsumeSharedPluginOptions>(self.options)

crates/rspack_binding_api/src/raw_options/raw_builtins/raw_mf.rs

@@ -3,11 +3,12 @@ use std::{collections::HashMap, sync::Arc};
 use napi::Either;
 use napi_derive::napi;
 use rspack_plugin_mf::{
-  ConsumeOptions, ConsumeSharedPluginOptions, ConsumeVersion, ContainerPluginOptions,
-  ContainerReferencePluginOptions, ExposeOptions, ManifestExposeOption, ManifestSharedOption,
-  ModuleFederationManifestPluginOptions, ModuleFederationRuntimeExperimentsOptions,
-  ModuleFederationRuntimePluginOptions, ProvideOptions, ProvideVersion, RemoteAliasTarget,
-  RemoteOptions, StatsBuildInfo,
+  CollectSharedEntryPluginOptions, ConsumeOptions, ConsumeSharedPluginOptions, ConsumeVersion,
+  ContainerPluginOptions, ContainerReferencePluginOptions, ExposeOptions, ManifestExposeOption,
+  ManifestSharedOption, ModuleFederationManifestPluginOptions,
+  ModuleFederationRuntimeExperimentsOptions, ModuleFederationRuntimePluginOptions,
+  OptimizeSharedConfig, ProvideOptions, ProvideVersion, RemoteAliasTarget, RemoteOptions,
+  SharedContainerPluginOptions, SharedUsedExportsOptimizerPluginOptions, StatsBuildInfo,
 };
 
 use crate::options::{
@@ -115,6 +116,7 @@ pub struct RawProvideOptions {
   #[napi(ts_type = "string | false | undefined")]
   pub required_version: Option<RawVersion>,
   pub strict_version: Option<bool>,
+  pub tree_shaking_mode: Option<String>,
 }
 
 impl From<RawProvideOptions> for (String, ProvideOptions) {
@@ -129,11 +131,57 @@ impl From<RawProvideOptions> for (String, ProvideOptions) {
         singleton: value.singleton,
         required_version: value.required_version.map(|v| RawVersionWrapper(v).into()),
         strict_version: value.strict_version,
+        tree_shaking_mode: value.tree_shaking_mode,
       },
     )
   }
 }
 
+#[derive(Debug)]
+#[napi(object)]
+pub struct RawCollectShareEntryPluginOptions {
+  pub consumes: Vec<RawConsumeOptions>,
+  pub filename: Option<String>,
+}
+
+impl From<RawCollectShareEntryPluginOptions> for CollectSharedEntryPluginOptions {
+  fn from(value: RawCollectShareEntryPluginOptions) -> Self {
+    Self {
+      consumes: value
+        .consumes
+        .into_iter()
+        .map(|provide| {
+          let (key, consume_options): (String, ConsumeOptions) = provide.into();
+          (key, std::sync::Arc::new(consume_options))
+        })
+        .collect(),
+      filename: value.filename,
+    }
+  }
+}
+
+#[derive(Debug)]
+#[napi(object)]
+pub struct RawSharedContainerPluginOptions {
+  pub name: String,
+  pub request: String,
+  pub version: String,
+  pub file_name: Option<String>,
+  pub library: JsLibraryOptions,
+}
+
+impl From<RawSharedContainerPluginOptions> for SharedContainerPluginOptions {
+  fn from(value: RawSharedContainerPluginOptions) -> Self {
+    SharedContainerPluginOptions {
+      name: value.name,
+      request: value.request,
+      version: value.version,
+      library: value.library.into(),
+      file_name: value.file_name.clone().map(Into::into),
+    }
+  }
+}
+
 #[derive(Debug)]
 #[napi(object)]
 pub struct RawConsumeSharedPluginOptions {
@@ -155,6 +203,52 @@ impl From<RawConsumeSharedPluginOptions> for ConsumeSharedPluginOptions {
   }
 }
 
+#[derive(Debug)]
+#[napi(object)]
+pub struct RawOptimizeSharedConfig {
+  pub share_key: String,
+  pub tree_shaking: bool,
+  pub used_exports: Option<Vec<String>>,
+}
+
+impl From<RawOptimizeSharedConfig> for OptimizeSharedConfig {
+  fn from(value: RawOptimizeSharedConfig) -> Self {
+    Self {
+      share_key: value.share_key,
+      tree_shaking: value.tree_shaking,
+      used_exports: value.used_exports.unwrap_or_default(),
+    }
+  }
+}
+
+#[derive(Debug)]
+#[napi(object)]
+pub struct RawSharedUsedExportsOptimizerPluginOptions {
+  pub shared: Vec<RawOptimizeSharedConfig>,
+  pub inject_tree_shaking_used_exports: Option<bool>,
+  pub manifest_file_name: Option<String>,
+  pub stats_file_name: Option<String>,
+}
+
+impl From<RawSharedUsedExportsOptimizerPluginOptions> for SharedUsedExportsOptimizerPluginOptions {
+  fn from(value: RawSharedUsedExportsOptimizerPluginOptions) -> Self {
+    Self {
+      shared: value
+        .shared
+        .into_iter()
+        .map(|config| config.into())
+        .collect(),
+      inject_tree_shaking_used_exports: value.inject_tree_shaking_used_exports.unwrap_or(true),
+      manifest_file_name: value
+        .manifest_file_name
+        .and_then(|s| if s.trim().is_empty() { None } else { Some(s) }),
+      stats_file_name: value
+        .stats_file_name
+        .and_then(|s| if s.trim().is_empty() { None } else { Some(s) }),
+    }
+  }
+}
+
 #[derive(Debug)]
 #[napi(object)]
 pub struct RawConsumeOptions {
@@ -169,6 +263,7 @@ pub struct RawConsumeOptions {
   pub strict_version: bool,
   pub singleton: bool,
   pub eager: bool,
+  pub tree_shaking_mode: Option<String>,
 }
 
 impl From<RawConsumeOptions> for (String, ConsumeOptions) {
@@ -185,6 +280,7 @@ impl From<RawConsumeOptions> for (String, ConsumeOptions) {
         strict_version: value.strict_version,
         singleton: value.singleton,
         eager: value.eager,
+        tree_shaking_mode: value.tree_shaking_mode,
       },
     )
   }
@@ -274,6 +370,9 @@ pub struct RawManifestSharedOption {
 pub struct RawStatsBuildInfo {
   pub build_version: String,
   pub build_name: Option<String>,
+  // only appear when enable tree_shaking
+  pub target: Option<Vec<String>>,
+  pub plugins: Option<Vec<String>>,
 }
 
 #[derive(Debug)]
@@ -337,6 +436,8 @@ impl From<RawModuleFederationManifestPluginOptions> for ModuleFederationManifest
       build_info: value.build_info.map(|info| StatsBuildInfo {
         build_version: info.build_version,
         build_name: info.build_name,
+        target: info.target,
+        plugins: info.plugins,
       }),
     }
   }

crates/rspack_core/src/compilation/mod.rs

@@ -71,10 +71,10 @@ use crate::{
   ChunkRenderCacheArtifact, ChunkRenderResult, ChunkUkey, CodeGenerateCacheArtifact,
   CodeGenerationJob, CodeGenerationResult, CodeGenerationResults, CompilationLogger,
   CompilationLogging, CompilerOptions, CompilerPlatform, ConcatenationScope,
-  DependenciesDiagnosticsArtifact, DependencyCodeGeneration, DependencyTemplate,
+  DependenciesDiagnosticsArtifact, DependencyCodeGeneration, DependencyId, DependencyTemplate,
   DependencyTemplateType, DependencyType, DerefOption, Entry, EntryData, EntryOptions,
-  EntryRuntime, Entrypoint, ExecuteModuleId, Filename, ImportPhase, ImportVarMap,
-  ImportedByDeferModulesArtifact, MemoryGCStorage, ModuleFactory, ModuleGraph,
+  EntryRuntime, Entrypoint, ExecuteModuleId, ExtendedReferencedExport, Filename, ImportPhase,
+  ImportVarMap, ImportedByDeferModulesArtifact, MemoryGCStorage, ModuleFactory, ModuleGraph,
   ModuleGraphCacheArtifact, ModuleIdentifier, ModuleIdsArtifact, ModuleStaticCache, PathData,
   ProcessRuntimeRequirementsCacheArtifact, ResolverFactory, RuntimeGlobals, RuntimeKeyMap,
   RuntimeMode, RuntimeModule, RuntimeSpec, RuntimeSpecMap, RuntimeTemplate, SharedPluginDriver,
@@ -97,6 +97,13 @@ define_hook!(CompilationExecuteModule:
   Series(module: &ModuleIdentifier, runtime_modules: &IdentifierSet, code_generation_results: &BindingCell<CodeGenerationResults>, execute_module_id: &ExecuteModuleId));
 define_hook!(CompilationFinishModules: Series(compilation: &mut Compilation, async_modules_artifact: &mut AsyncModulesArtifact));
 define_hook!(CompilationSeal: Series(compilation: &mut Compilation));
+define_hook!(CompilationDependencyReferencedExports: Series(
+  compilation: &Compilation,
+  dependency: &DependencyId,
+  referenced_exports: &Option<Vec<ExtendedReferencedExport>>,
+  runtime: Option<&RuntimeSpec>,
+  module_graph: Option<&ModuleGraph>
+));
 define_hook!(CompilationConcatenationScope: SeriesBail(compilation: &Compilation, curr_module: ModuleIdentifier) -> ConcatenationScope);
 define_hook!(CompilationOptimizeDependencies: SeriesBail(compilation: &Compilation, side_effects_optimize_artifact: &mut SideEffectsOptimizeArtifact,  build_module_graph_artifact: &mut BuildModuleGraphArtifact,
  diagnostics: &mut Vec<Diagnostic>) -> bool);
@@ -136,6 +143,7 @@ pub struct CompilationHooks {
   pub succeed_module: CompilationSucceedModuleHook,
   pub execute_module: CompilationExecuteModuleHook,
   pub finish_modules: CompilationFinishModulesHook,
+  pub dependency_referenced_exports: CompilationDependencyReferencedExportsHook,
   pub seal: CompilationSealHook,
   pub optimize_dependencies: CompilationOptimizeDependenciesHook,
   pub optimize_modules: CompilationOptimizeModulesHook,