feat: support __webpack_exports_info__ (#7205)

crates/rspack_core/src/context_module.rs

@@ -20,6 +20,7 @@ use rspack_sources::{BoxSource, ConcatSource, RawSource, SourceExt};
 use rspack_util::{fx_hash::FxIndexMap, json_stringify, source_map::SourceMapKind};
 use rustc_hash::FxHashMap as HashMap;
 use rustc_hash::FxHashSet as HashSet;
+use swc_core::atoms::Atom;
 
 use crate::{
   block_promise, contextify, get_exports_type_with_strict, impl_module_meta_info,
@@ -141,6 +142,7 @@ pub struct ContextOptions {
   pub replaces: Vec<(String, u32, u32)>,
   pub start: u32,
   pub end: u32,
+  pub referenced_exports: Option<Vec<Atom>>,
 }
 
 #[derive(Debug, PartialEq, Clone)]
@@ -1074,7 +1076,7 @@ impl ContextModule {
             context: options.resource.clone().into(),
             options: options.context_options.clone(),
             resource_identifier: format!("context{}|{}", &options.resource, path.to_string_lossy()),
-            referenced_exports: None,
+            referenced_exports: options.context_options.referenced_exports.clone(),
             dependency_type: DependencyType::ContextElement(options.type_prefix),
           });
         })
@@ -1217,6 +1219,14 @@ fn create_identifier(options: &ContextModuleOptions) -> Identifier {
     id += "|exclude: ";
     id += &exclude.to_source_string();
   }
+  if let Some(exports) = &options.context_options.referenced_exports {
+    id += "|referencedExports: ";
+    id += &format!(
+      "[{}]",
+      exports.iter().map(|x| format!(r#""{x}""#)).join(",")
+    );
+  }
+
   if let Some(GroupOptions::ChunkGroup(group)) = &options.context_options.group_options {
     if let Some(chunk_name) = &group.name {
       id += "|chunkName: ";

crates/rspack_core/src/dependency/context_element_dependency.rs

@@ -1,6 +1,7 @@
+use itertools::Itertools;
 use swc_core::ecma::atoms::Atom;
 
-use crate::{AsContextDependency, AsDependencyTemplate, Context};
+use crate::{create_exports_object_referenced, AsContextDependency, AsDependencyTemplate, Context};
 use crate::{ContextMode, ContextOptions, Dependency};
 use crate::{DependencyCategory, DependencyId, DependencyType};
 use crate::{ExtendedReferencedExport, ModuleDependency};
@@ -47,9 +48,14 @@ impl Dependency for ContextElementDependency {
     _runtime: Option<&RuntimeSpec>,
   ) -> Vec<ExtendedReferencedExport> {
     if let Some(referenced_exports) = &self.referenced_exports {
-      vec![ReferencedExport::new(referenced_exports.clone(), false).into()]
+      referenced_exports
+        .iter()
+        .map(|export| {
+          ExtendedReferencedExport::Export(ReferencedExport::new(vec![export.clone()], false))
+        })
+        .collect_vec()
     } else {
-      vec![ExtendedReferencedExport::Array(vec![])]
+      create_exports_object_referenced()
     }
   }
 }

crates/rspack_core/src/dependency/mod.rs

@@ -9,7 +9,6 @@ mod dependency_template;
 mod dependency_trait;
 mod dependency_type;
 mod entry;
-mod import_dependency_trait;
 mod loader_import;
 mod module_dependency;
 mod runtime_requirements_dependency;
@@ -29,7 +28,6 @@ pub use dependency_template::*;
 pub use dependency_trait::*;
 pub use dependency_type::DependencyType;
 pub use entry::*;
-pub use import_dependency_trait::ImportDependencyTrait;
 pub use loader_import::*;
 pub use module_dependency::*;
 pub use runtime_requirements_dependency::RuntimeRequirementsDependency;

crates/rspack_core/src/exports_info.rs

@@ -740,11 +740,17 @@ impl ExportsInfo {
       }
       UsedName::Vec(value) => {
         if value.is_empty() {
-          let other_export_info = module_graph.get_export_info_by_id(&self.other_exports_info);
-          return other_export_info.get_used(runtime);
+          return self
+            .other_exports_info
+            .get_export_info(module_graph)
+            .get_used(runtime);
         }
         let info = self.id.get_read_only_export_info(&value[0], module_graph);
-        if let Some(exports_info) = info.get_exports_info(module_graph) {
+        if let Some(exports_info) = info
+          .exports_info
+          .map(|id| id.get_exports_info(module_graph))
+          && value.len() > 1
+        {
           return exports_info.get_used(
             UsedName::Vec(value.iter().skip(1).cloned().collect::<Vec<_>>()),
             runtime,

crates/rspack_plugin_javascript/src/dependency/esm/import_dependency.rs

@@ -1,9 +1,54 @@
-use rspack_core::{module_namespace_promise, DependencyType, ErrorSpan, ImportDependencyTrait};
+use rspack_core::{
+  create_exports_object_referenced, module_namespace_promise, DependencyType, ErrorSpan,
+  ExportsType, ExtendedReferencedExport, ModuleGraph, ReferencedExport,
+};
 use rspack_core::{AsContextDependency, Dependency};
 use rspack_core::{DependencyCategory, DependencyId, DependencyTemplate};
 use rspack_core::{ModuleDependency, TemplateContext, TemplateReplaceSource};
 use swc_core::ecma::atoms::Atom;
 
+pub fn create_import_dependency_referenced_exports(
+  dependency_id: &DependencyId,
+  referenced_exports: &Option<Vec<Atom>>,
+  mg: &ModuleGraph,
+) -> Vec<ExtendedReferencedExport> {
+  if let Some(referenced_exports) = referenced_exports {
+    let mut refs = vec![];
+    for referenced_export in referenced_exports {
+      if referenced_export == "default" {
+        let Some(strict) = mg
+          .get_parent_module(dependency_id)
+          .and_then(|id| mg.module_by_identifier(id))
+          .and_then(|m| m.build_meta())
+          .map(|bm| bm.strict_harmony_module)
+        else {
+          return create_exports_object_referenced();
+        };
+        let Some(imported_module) = mg
+          .module_identifier_by_dependency_id(dependency_id)
+          .and_then(|id| mg.module_by_identifier(id))
+        else {
+          return create_exports_object_referenced();
+        };
+        let exports_type = imported_module.get_exports_type_readonly(mg, strict);
+        if matches!(
+          exports_type,
+          ExportsType::DefaultOnly | ExportsType::DefaultWithNamed
+        ) {
+          return create_exports_object_referenced();
+        }
+      }
+      refs.push(ExtendedReferencedExport::Export(ReferencedExport::new(
+        vec![referenced_export.clone()],
+        false,
+      )));
+    }
+    refs
+  } else {
+    create_exports_object_referenced()
+  }
+}
+
 #[derive(Debug, Clone)]
 pub struct ImportDependency {
   start: u32,
@@ -49,6 +94,14 @@ impl Dependency for ImportDependency {
   fn span(&self) -> Option<ErrorSpan> {
     self.span
   }
+
+  fn get_referenced_exports(
+    &self,
+    module_graph: &rspack_core::ModuleGraph,
+    _runtime: Option<&rspack_core::RuntimeSpec>,
+  ) -> Vec<rspack_core::ExtendedReferencedExport> {
+    create_import_dependency_referenced_exports(&self.id, &self.referenced_exports, module_graph)
+  }
 }
 
 impl ModuleDependency for ImportDependency {
@@ -65,12 +118,6 @@ impl ModuleDependency for ImportDependency {
   }
 }
 
-impl ImportDependencyTrait for ImportDependency {
-  fn referenced_exports(&self) -> Option<&Vec<Atom>> {
-    self.referenced_exports.as_ref()
-  }
-}
-
 impl DependencyTemplate for ImportDependency {
   fn apply(
     &self,

crates/rspack_plugin_javascript/src/dependency/esm/import_eager_dependency.rs

@@ -1,11 +1,12 @@
 use rspack_core::{
   module_namespace_promise, AsContextDependency, Dependency, DependencyCategory, DependencyId,
-  DependencyTemplate, DependencyType, ErrorSpan, ExtendedReferencedExport, ImportDependencyTrait,
-  ModuleDependency, ModuleGraph, ReferencedExport, RuntimeSpec, TemplateContext,
+  DependencyTemplate, DependencyType, ErrorSpan, ModuleDependency, TemplateContext,
   TemplateReplaceSource,
 };
 use swc_core::ecma::atoms::Atom;
 
+use super::import_dependency::create_import_dependency_referenced_exports;
+
 #[derive(Debug, Clone)]
 pub struct ImportEagerDependency {
   start: u32,
@@ -54,14 +55,10 @@ impl Dependency for ImportEagerDependency {
 
   fn get_referenced_exports(
     &self,
-    _module_graph: &ModuleGraph,
-    _runtime: Option<&RuntimeSpec>,
-  ) -> Vec<ExtendedReferencedExport> {
-    if let Some(referenced_exports) = &self.referenced_exports {
-      vec![ReferencedExport::new(referenced_exports.clone(), false).into()]
-    } else {
-      vec![ExtendedReferencedExport::Array(vec![])]
-    }
+    module_graph: &rspack_core::ModuleGraph,
+    _runtime: Option<&rspack_core::RuntimeSpec>,
+  ) -> Vec<rspack_core::ExtendedReferencedExport> {
+    create_import_dependency_referenced_exports(&self.id, &self.referenced_exports, module_graph)
   }
 }
 
@@ -79,12 +76,6 @@ impl ModuleDependency for ImportEagerDependency {
   }
 }
 
-impl ImportDependencyTrait for ImportEagerDependency {
-  fn referenced_exports(&self) -> Option<&Vec<Atom>> {
-    self.referenced_exports.as_ref()
-  }
-}
-
 impl DependencyTemplate for ImportEagerDependency {
   fn apply(
     &self,

crates/rspack_plugin_javascript/src/dependency/esm/provide_dependency.rs

@@ -1,5 +1,8 @@
 use itertools::Itertools;
-use rspack_core::{module_raw, NormalInitFragment, UsedName};
+use rspack_core::{
+  create_exports_object_referenced, module_raw, ExtendedReferencedExport, ModuleGraph,
+  NormalInitFragment, RuntimeSpec, UsedName,
+};
 use rspack_core::{AsContextDependency, Dependency, InitFragmentKey, InitFragmentStage};
 use rspack_core::{DependencyCategory, DependencyId, DependencyTemplate};
 use rspack_core::{DependencyType, ErrorSpan};
@@ -45,6 +48,18 @@ impl Dependency for ProvideDependency {
   fn span(&self) -> Option<ErrorSpan> {
     None
   }
+
+  fn get_referenced_exports(
+    &self,
+    _module_graph: &ModuleGraph,
+    _runtime: Option<&RuntimeSpec>,
+  ) -> Vec<ExtendedReferencedExport> {
+    if self.ids.is_empty() {
+      create_exports_object_referenced()
+    } else {
+      vec![ExtendedReferencedExport::Array(self.ids.clone())]
+    }
+  }
 }
 
 impl ModuleDependency for ProvideDependency {