feat(tree-shaking):skip-module according to side-effects

Cargo.toml

@@ -6,3 +6,4 @@ members  = ["crates/*"]
 serde      = { version = "1.0.171", features = ["derive"] }
 serde_json = "1.0.100"
 cached     = "0.46.1"
+swc_core   = { version = "=0.83.19", default-features = false }

crates/core/Cargo.toml

@@ -7,7 +7,7 @@ license    = "MIT"
 
 [dependencies]
 tokio = { version = "1", features = ["full"] }
-swc_core = { version = "=0.83.19", features = [
+swc_core = { workspace = true, features = [
   "__ecma",
   "__visit",
   "__utils",

crates/mako/Cargo.toml

@@ -15,6 +15,7 @@ mako_core  = { path = "../core" }
 serde      = { workspace = true }
 serde_json = { workspace = true }
 cached     = { workspace = true }
+swc_core   = { workspace = true, features = ["swc_ecma_quote_macros"] }
 
 [dev-dependencies]
 insta                       = { version = "1.30.0", features = ["yaml"] }

crates/mako/src/analyze_deps.rs

@@ -5,11 +5,10 @@ use mako_core::anyhow::{anyhow, Result};
 use crate::compiler::Context;
 use crate::module::{Dependency, ModuleAst};
 use crate::plugin::PluginDepAnalyzeParam;
-use crate::task::Task;
 
 pub fn analyze_deps(
     ast: &ModuleAst,
-    task: &Task,
+    file_path: &String,
     context: &Arc<Context>,
 ) -> Result<Vec<Dependency>> {
     mako_core::mako_profile_function!();
@@ -32,7 +31,7 @@ pub fn analyze_deps(
             return Err(anyhow!(
                 "webpack loader syntax is not supported, since found dep {:?} in {:?}",
                 dep.source,
-                task.path,
+                file_path
             ));
         }
     }

crates/mako/src/build.rs

@@ -267,7 +267,7 @@ module.exports = new Promise((resolve, reject) => {{
 
         // 在此之前需要把所有依赖都和模块关联起来，并且需要使用 resolved source
         // analyze deps
-        let deps = analyze_deps(&ast, &task, context)?;
+        let deps = analyze_deps(&ast, &task.path, context)?;
 
         // resolve
         let mut dep_resolve_err = None;

crates/mako/src/config.rs

@@ -249,6 +249,11 @@ pub struct MinifishConfig {
     pub meta_path: Option<PathBuf>,
     pub inject: Option<HashMap<String, InjectItem>>,
 }
+#[derive(Deserialize, Serialize, Debug)]
+#[serde(rename_all = "camelCase")]
+pub struct OptimizationConfig {
+    pub skip_module: Option<bool>,
+}
 
 #[derive(Deserialize, Serialize, Debug)]
 #[serde(rename_all = "camelCase")]
@@ -301,6 +306,7 @@ pub struct Config {
     pub optimize_package_imports: bool,
     pub emotion: bool,
     pub flex_bugs: bool,
+    pub optimization: Option<OptimizationConfig>,
 }
 
 pub(crate) fn hash_config(c: &Config) -> u64 {
@@ -457,6 +463,7 @@ const DEFAULT_CONFIG: &str = r#"
     "optimizePackageImports": false,
     "emotion": false,
     "flexBugs": false,
+    "optimization": { "skipModule": true }
 }
 "#;
 

crates/mako/src/lib.rs

@@ -1,5 +1,6 @@
 #![feature(box_patterns)]
 #![feature(hasher_prefixfree_extras)]
+#![feature(let_chains)]
 
 mod analyze_deps;
 mod ast;

crates/mako/src/main.rs

@@ -1,4 +1,5 @@
 #![feature(box_patterns)]
+#![feature(let_chains)]
 
 use std::sync::Arc;
 

crates/mako/src/module_graph.rs

@@ -238,6 +238,14 @@ impl ModuleGraph {
         targets
     }
 
+    pub fn remove_dependencies(&mut self, module_id: &ModuleId) {
+        let mut edges = self.get_edges(module_id, Direction::Outgoing);
+
+        while let Some((edge_index, _)) = edges.next(&self.graph) {
+            self.graph.remove_edge(edge_index);
+        }
+    }
+
     pub fn get_dependency_module_by_source(
         &self,
         module_id: &ModuleId,

crates/mako/src/plugins/farm_tree_shake/module.rs

@@ -1,3 +1,5 @@
+mod import_exports;
+
 use std::collections::{HashMap, HashSet};
 
 use mako_core::swc_common::SyntaxContext;
@@ -129,6 +131,22 @@ pub struct TreeShakeModule {
 }
 
 impl TreeShakeModule {
+    pub fn update_stmt_graph(&mut self, module: &SwcModule) {
+        let stmt_graph = StatementGraph::new(module, self.unresolved_ctxt);
+
+        self.stmt_graph = stmt_graph;
+    }
+
+    pub fn has_side_effect(&self) -> bool {
+        if let Some(described_side_effects) = self.described_side_effects {
+            if !described_side_effects {
+                return false;
+            }
+        }
+
+        self.side_effects
+    }
+
     pub fn update_side_effect(&mut self) -> bool {
         let mut side_effect_stmts = vec![];
 
@@ -236,8 +254,6 @@ impl TreeShakeModule {
     pub fn new(module: &Module, order: usize, _module_graph: &ModuleGraph) -> Self {
         let module_info = module.info.as_ref().unwrap();
 
-        let side_effect_map: HashMap<String, bool> = Default::default();
-
         let mut unresolved_ctxt = SyntaxContext::empty();
         // 1. generate statement graph
         let mut module_system = ModuleSystem::CommonJS;
@@ -251,7 +267,7 @@ impl TreeShakeModule {
                 if is_esm {
                     module_system = ModuleSystem::ESModule;
                     unresolved_ctxt = unresolved_ctxt.apply_mark(module.unresolved_mark);
-                    StatementGraph::new(&module.ast, &side_effect_map, unresolved_ctxt)
+                    StatementGraph::new(&module.ast, unresolved_ctxt)
                 } else {
                     StatementGraph::empty()
                 }
@@ -396,7 +412,7 @@ impl TreeShakeModule {
                 for export_info in self.exports() {
                     for sp in export_info.specifiers {
                         match sp {
-                            ExportSpecifierInfo::Default => {
+                            ExportSpecifierInfo::Default(_) => {
                                 used_idents.push((UsedIdent::Default, export_info.stmt_id));
                             }
                             ExportSpecifierInfo::Named { local, .. } => {
@@ -443,7 +459,7 @@ impl TreeShakeModule {
                     for export_info in export_infos {
                         for sp in export_info.specifiers {
                             match sp {
-                                ExportSpecifierInfo::Default => {
+                                ExportSpecifierInfo::Default(_) => {
                                     if ident == "default" {
                                         used_idents.push((UsedIdent::Default, export_info.stmt_id));
                                     }
@@ -508,3 +524,10 @@ pub fn is_ident_equal(ident1: &str, ident2: &str) -> bool {
         split1[0] == split2[0]
     }
 }
+
+pub fn is_ident_sym_equal(ident1: &str, ident2: &str) -> bool {
+    let split1 = ident1.split('#').collect::<Vec<_>>();
+    let split2 = ident2.split('#').collect::<Vec<_>>();
+
+    split1[0] == split2[0]
+}