feat(treeshaking): optimize pure functions

Author: JSerFeng

crates/node_binding/napi-binding.d.ts

@@ -569,6 +569,7 @@ export declare enum BuiltinPluginName {
   EvalSourceMapDevToolPlugin = 'EvalSourceMapDevToolPlugin',
   EvalDevToolModulePlugin = 'EvalDevToolModulePlugin',
   SideEffectsFlagPlugin = 'SideEffectsFlagPlugin',
+  InnerGraphPlugin = 'InnerGraphPlugin',
   FlagDependencyExportsPlugin = 'FlagDependencyExportsPlugin',
   FlagDependencyUsagePlugin = 'FlagDependencyUsagePlugin',
   InlineExportsPlugin = 'InlineExportsPlugin',
@@ -2138,6 +2139,7 @@ useInputFileSystem?: false | Array<RegExp>
 css?: boolean
 lazyBarrel: boolean
 deferImport: boolean
+advancedTreeShaking: boolean
 }
 
 export interface RawExperimentSnapshotOptions {
@@ -2370,6 +2372,7 @@ typeReexportsPresence?: string
  */
 jsx?: boolean
 deferImport?: boolean
+sideEffectsFree?: Array<string>
 }
 
 export interface RawJsonGeneratorOptions {

crates/rspack/src/builder/builder_context.rs

@@ -57,6 +57,7 @@ pub(super) enum BuiltinPluginOptions {
   EnsureChunkConditionsPlugin,
   MergeDuplicateChunksPlugin,
   SideEffectsFlagPlugin,
+  InnerGraphPlugin,
   FlagDependencyExportsPlugin,
   FlagDependencyUsagePlugin(bool),
   ModuleConcatenationPlugin,
@@ -257,6 +258,9 @@ impl BuilderContext {
       BuiltinPluginOptions::SideEffectsFlagPlugin => {
         plugins.push(rspack_plugin_javascript::SideEffectsFlagPlugin::default().boxed());
       }
+      BuiltinPluginOptions::InnerGraphPlugin => {
+        plugins.push(rspack_plugin_javascript::InnerGraphPlugin::default().boxed());
+      }
       BuiltinPluginOptions::FlagDependencyExportsPlugin => {
         plugins.push(rspack_plugin_javascript::FlagDependencyExportsPlugin::default().boxed());
       }

crates/rspack/src/builder/mod.rs

@@ -3601,6 +3601,11 @@ impl OptimizationOptionsBuilder {
         ));
     }
     let inner_graph = d!(self.inner_graph, production);
+    if inner_graph {
+      builder_context
+        .plugins
+        .push(BuiltinPluginOptions::InnerGraphPlugin);
+    }
     if !d!(self.emit_on_errors, !production) {
       builder_context
         .plugins
@@ -3699,6 +3704,7 @@ pub struct ExperimentsBuilder {
   /// Whether to enable async web assembly.
   async_web_assembly: Option<bool>,
   // TODO: lazy compilation
+  advanced_tree_shaking: Option<bool>,
 }
 
 impl From<Experiments> for ExperimentsBuilder {
@@ -3712,6 +3718,7 @@ impl From<Experiments> for ExperimentsBuilder {
       future_defaults: None,
       css: Some(value.css),
       async_web_assembly: None,
+      advanced_tree_shaking: Some(value.advanced_tree_shaking),
     }
   }
 }
@@ -3727,6 +3734,7 @@ impl From<&mut ExperimentsBuilder> for ExperimentsBuilder {
       future_defaults: value.future_defaults.take(),
       css: value.css.take(),
       async_web_assembly: value.async_web_assembly.take(),
+      advanced_tree_shaking: value.advanced_tree_shaking.take(),
     }
   }
 }
@@ -3812,6 +3820,7 @@ impl ExperimentsBuilder {
       css: d!(self.css, false),
       lazy_barrel: true,
       defer_import: false,
+      advanced_tree_shaking: d!(self.advanced_tree_shaking, false),
     })
   }
 }

crates/rspack/tests/snapshots/defaults__default_options.snap

@@ -1480,6 +1480,7 @@ CompilerOptions {
                                 false,
                             ),
                             defer_import: None,
+                            side_effects_free: None,
                         },
                     ),
                 },
@@ -1518,6 +1519,7 @@ CompilerOptions {
         css: false,
         lazy_barrel: true,
         defer_import: false,
+        advanced_tree_shaking: false,
     },
     node: Some(
         NodeOption {

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

@@ -66,7 +66,7 @@ use rspack_plugin_html::HtmlRspackPlugin;
 use rspack_plugin_ignore::IgnorePlugin;
 use rspack_plugin_javascript::{
   FlagDependencyExportsPlugin, FlagDependencyUsagePlugin, InferAsyncModulesPlugin,
-  InlineExportsPlugin, JsPlugin, MangleExportsPlugin, ModuleConcatenationPlugin,
+  InlineExportsPlugin, InnerGraphPlugin, JsPlugin, MangleExportsPlugin, ModuleConcatenationPlugin,
   SideEffectsFlagPlugin, api_plugin::APIPlugin, define_plugin::DefinePlugin,
   provide_plugin::ProvidePlugin, url_plugin::URLPlugin,
 };
@@ -196,6 +196,7 @@ pub enum BuiltinPluginName {
   EvalSourceMapDevToolPlugin,
   EvalDevToolModulePlugin,
   SideEffectsFlagPlugin,
+  InnerGraphPlugin,
   FlagDependencyExportsPlugin,
   FlagDependencyUsagePlugin,
   InlineExportsPlugin,
@@ -607,6 +608,7 @@ impl<'a> BuiltinPlugin<'a> {
       BuiltinPluginName::SideEffectsFlagPlugin => {
         plugins.push(SideEffectsFlagPlugin::default().boxed())
       }
+      BuiltinPluginName::InnerGraphPlugin => plugins.push(InnerGraphPlugin::default().boxed()),
       BuiltinPluginName::FlagDependencyExportsPlugin => {
         plugins.push(FlagDependencyExportsPlugin::default().boxed())
       }

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

@@ -27,6 +27,7 @@ pub struct RawExperiments {
   pub css: Option<bool>,
   pub lazy_barrel: bool,
   pub defer_import: bool,
+  pub advanced_tree_shaking: bool,
 }
 
 impl From<RawExperiments> for Experiments {
@@ -45,6 +46,7 @@ impl From<RawExperiments> for Experiments {
       css: value.css.unwrap_or(false),
       lazy_barrel: value.lazy_barrel,
       defer_import: value.defer_import,
+      advanced_tree_shaking: value.advanced_tree_shaking,
     }
   }
 }

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

@@ -310,6 +310,7 @@ pub struct RawJavascriptParserOptions {
   /// @experimental
   pub jsx: Option<bool>,
   pub defer_import: Option<bool>,
+  pub side_effects_free: Option<Vec<String>>,
 }
 
 #[napi(object)]
@@ -388,6 +389,9 @@ impl From<RawJavascriptParserOptions> for JavascriptParserOptions {
       commonjs_magic_comments: value.commonjs_magic_comments,
       jsx: value.jsx,
       defer_import: value.defer_import,
+      side_effects_free: value
+        .side_effects_free
+        .map(|functions| functions.into_iter().collect()),
     }
   }
 }

crates/rspack_core/src/dependency/mod.rs

@@ -198,7 +198,7 @@ impl std::fmt::Debug for DependencyCondition {
 }
 
 #[rspack_cacheable::cacheable]
-#[derive(Debug, Clone, Serialize, Default)]
+#[derive(Debug, Clone, Serialize, Default, PartialEq, Eq)]
 pub struct ImportAttributes(FxHashMap<String, String>);
 
 impl FromIterator<(String, String)> for ImportAttributes {