feat: 🎸 json tree shaking (#4858)

* feat: 🎸 json tree shaking * chore: 🤖 ck point * chore: 🤖 ck points * chore: 🤖 ck point * chore: 🤖 partial runtime opt * chore: 🤖 rebase * fix: 🐛 mangle issue * chore: 🤖 update * chore: 🤖 clean * chore: 🤖 lint * chore: 🤖 update snap * chore: 🤖 recover * chore: 🤖 move import-named to packages/rspack/tests * chore: 🤖 json.parse opt * fix: 🐛 ci issue * chore: 🤖 update snap * fix: 🐛 compile error * chore: 🤖 snapshot
web-infra-dev · Dec 12, 2023 · 1ca2e0f · 1ca2e0f
1 parent 77f57e5
commit 1ca2e0f
Show file tree

Hide file tree

Showing 19 changed files with 345 additions and 34 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/rspack_core/Cargo.toml b/crates/rspack_core/Cargo.toml
@@ -20,6 +20,7 @@ glob-match = "0.2.1"
 hashlink = { workspace = true }
 indexmap = { workspace = true }
 itertools = { workspace = true }
+json = { workspace = true }
 mime_guess = { workspace = true }
 nodejs-resolver = { version = "0.1.1" }
 once_cell = { workspace = true }

diff --git a/crates/rspack_core/src/exports_info.rs b/crates/rspack_core/src/exports_info.rs
@@ -182,6 +182,8 @@ impl ExportsInfoId {
     changed
   }
 
+  /// # Panic
+  /// this function would panic if name doesn't exists in current exportsInfo
   pub fn get_read_only_export_info<'a>(
     &self,
     name: &JsWord,
@@ -1364,7 +1366,6 @@ impl ExportInfo {
     );
     let new_exports_info = ExportsInfo::new(other_exports_info.id, side_effects_only_info.id);
     let new_exports_info_id = new_exports_info.id;
-
     mg.exports_info_map
       .insert(new_exports_info_id, new_exports_info);
     mg.export_info_map

diff --git a/crates/rspack_core/src/module.rs b/crates/rspack_core/src/module.rs
@@ -4,6 +4,7 @@ use std::path::PathBuf;
 use std::{any::Any, borrow::Cow, fmt::Debug};
 
 use async_trait::async_trait;
+use json::JsonValue;
 use rspack_error::{IntoTWithDiagnosticArray, Result, TWithDiagnosticArray};
 use rspack_hash::{RspackHash, RspackHashDigest};
 use rspack_identifier::{Identifiable, Identifier};
@@ -47,6 +48,7 @@ pub struct BuildInfo {
   pub harmony_named_exports: HashSet<JsWord>,
   pub all_star_exports: Vec<DependencyId>,
   pub need_create_require: bool,
+  pub json_data: Option<JsonValue>,
 }
 
 #[derive(Debug, Default, Clone, Hash, PartialEq, Eq)]

diff --git a/crates/rspack_core/src/utils/visitor.rs b/crates/rspack_core/src/utils/visitor.rs
@@ -194,6 +194,12 @@ pub fn extract_member_expression_chain<'e, T: Into<MaybeExpr<'e>>>(
     }) = expr.prop
     {
       members.push_front((val.value.clone(), val.span.ctxt));
+    } else if let MemberProp::Computed(ComputedPropName {
+      expr: box Expr::Lit(Lit::Num(ref val)),
+      ..
+    }) = expr.prop
+    {
+      members.push_front((val.value.to_string().into(), val.span.ctxt));
     } else if let MemberProp::Ident(ref ident) = expr.prop {
       members.push_front((ident.sym.clone(), ident.span.ctxt));
       members_spans.push_front(ident.span);

diff --git a/crates/rspack_plugin_javascript/src/dependency/esm/harmony_import_specifier_dependency.rs b/crates/rspack_plugin_javascript/src/dependency/esm/harmony_import_specifier_dependency.rs
@@ -262,15 +262,7 @@ impl ModuleDependency for HarmonyImportSpecifierDependency {
   }
 
   fn get_condition(&self) -> Option<DependencyCondition> {
-    // dbg!(
-    //   &self.ids,
-    //   &self.specifier,
-    //   self.request(),
-    //   self.used_by_exports.as_ref()
-    // );
-    let ret = get_dependency_used_by_exports_condition(self.id, self.used_by_exports.as_ref());
-    // dbg!(&ret);
-    ret
+    get_dependency_used_by_exports_condition(self.id, self.used_by_exports.as_ref())
   }
 
   fn get_referenced_exports(

diff --git a/crates/rspack_plugin_javascript/src/visitors/dependency/harmony_import_dependency_scanner.rs b/crates/rspack_plugin_javascript/src/visitors/dependency/harmony_import_dependency_scanner.rs
@@ -510,7 +510,6 @@ impl Visit for HarmonyImportRefDependencyScanner<'_> {
             .map(|item| item.0.clone())
             .collect::<Vec<_>>(),
         );
-        // dbg!(&ids);
         self
           .rewrite_usage_span
           .insert(member_expr.span, ExtraSpanInfo::ReWriteUsedByExports);
@@ -596,6 +595,7 @@ mod test {
       harmony_named_exports: Default::default(),
       all_star_exports: Default::default(),
       need_create_require: false,
+      json_data: None,
     };
     let mut import_map = Default::default();
     let mut deps = vec![];

diff --git a/crates/rspack_plugin_json/src/json_exports_dependency.rs b/crates/rspack_plugin_json/src/json_exports_dependency.rs
@@ -0,0 +1,101 @@
+use json::JsonValue;
+use rspack_core::{
+  AsContextDependency, AsModuleDependency, Dependency, DependencyId, DependencyTemplate,
+  ExportNameOrSpec, ExportSpec, ExportsOfExportsSpec, ExportsSpec, ModuleGraph, TemplateContext,
+  TemplateReplaceSource,
+};
+#[derive(Debug, Clone)]
+pub struct JsonExportsDependency {
+  id: DependencyId,
+  data: JsonValue,
+}
+
+impl JsonExportsDependency {
+  pub fn new(data: JsonValue) -> Self {
+    Self {
+      data,
+      id: DependencyId::new(),
+    }
+  }
+}
+
+impl Dependency for JsonExportsDependency {
+  fn id(&self) -> &rspack_core::DependencyId {
+    &self.id
+  }
+
+  fn dependency_debug_name(&self) -> &'static str {
+    "JsonExportsDependency"
+  }
+
+  fn get_exports(&self, _mg: &ModuleGraph) -> Option<ExportsSpec> {
+    Some(ExportsSpec {
+      exports: get_exports_from_data(&self.data).unwrap_or(ExportsOfExportsSpec::Null),
+      ..Default::default()
+    })
+  }
+}
+
+impl AsModuleDependency for JsonExportsDependency {}
+impl AsContextDependency for JsonExportsDependency {}
+
+impl DependencyTemplate for JsonExportsDependency {
+  fn apply(
+    &self,
+    _source: &mut TemplateReplaceSource,
+    _code_generatable_context: &mut TemplateContext,
+  ) {
+  }
+}
+
+fn get_exports_from_data(data: &JsonValue) -> Option<ExportsOfExportsSpec> {
+  let ret = match data {
+    JsonValue::Null
+    | JsonValue::Short(_)
+    | JsonValue::String(_)
+    | JsonValue::Number(_)
+    | JsonValue::Boolean(_) => {
+      return None;
+    }
+    JsonValue::Object(obj) => ExportsOfExportsSpec::Array(
+      obj
+        .iter()
+        .map(|(k, v)| {
+          ExportNameOrSpec::ExportSpec(ExportSpec {
+            name: k.into(),
+            can_mangle: Some(true),
+            exports: get_exports_from_data(v).map(|item| match item {
+              ExportsOfExportsSpec::True => unreachable!(),
+              ExportsOfExportsSpec::Null => unreachable!(),
+              ExportsOfExportsSpec::Array(arr) => arr,
+            }),
+            ..Default::default()
+          })
+        })
+        .collect::<Vec<_>>(),
+    ),
+    JsonValue::Array(arr) => {
+      if arr.len() > 100 {
+        return None;
+      }
+      ExportsOfExportsSpec::Array(
+        arr
+          .iter()
+          .enumerate()
+          .map(|(i, item)| {
+            ExportNameOrSpec::ExportSpec(ExportSpec {
+              name: format!("{i}").into(),
+              can_mangle: Some(true),
+              exports: get_exports_from_data(item).map(|item| match item {
+                ExportsOfExportsSpec::True | ExportsOfExportsSpec::Null => unreachable!(),
+                ExportsOfExportsSpec::Array(arr) => arr,
+              }),
+              ..Default::default()
+            })
+          })
+          .collect::<Vec<_>>(),
+      )
+    }
+  };
+  Some(ret)
+}