feat: support import.defer() for context module (#12938)

Author: ahabhgk

crates/rspack_core/src/context_module.rs

@@ -230,6 +230,14 @@ impl ImportPhase {
   pub fn is_defer(&self) -> bool {
     matches!(self, ImportPhase::Defer)
   }
+
+  pub fn as_str(&self) -> &'static str {
+    match self {
+      ImportPhase::Evaluation => "evaluation",
+      ImportPhase::Source => "source",
+      ImportPhase::Defer => "defer",
+    }
+  }
 }
 
 impl From<swc_core::ecma::ast::ImportPhase> for ImportPhase {

crates/rspack_core/src/dependency/mod.rs

@@ -430,10 +430,10 @@ where
 
 pub fn render_make_deferred_namespace_mode_from_exports_type(exports_type: ExportsType) -> String {
   match exports_type {
-    ExportsType::Namespace => "0".to_string(),
-    ExportsType::DefaultOnly => "1".to_string(),
+    ExportsType::Namespace => "8".to_string(),
+    ExportsType::DefaultOnly => "0".to_string(),
     ExportsType::DefaultWithNamed => "2".to_string(),
-    ExportsType::Dynamic => "3".to_string(),
+    ExportsType::Dynamic => "6".to_string(),
   }
 }
 
@@ -465,7 +465,7 @@ pub fn get_exports_type_with_strict(
     .get_exports_type(module_graph, module_graph_cache, strict)
 }
 
-fn get_outgoing_async_modules(
+pub fn get_outgoing_async_modules(
   compilation: &Compilation,
   module: &dyn Module,
 ) -> FxIndexSet<ModuleId> {
@@ -1255,7 +1255,10 @@ impl ModuleCodegenRuntimeTemplate {
     let mut appending;
 
     if phase.is_defer() && !target_module.build_meta().has_top_level_await {
-      let mode = render_make_deferred_namespace_mode_from_exports_type(exports_type);
+      let mode = format!(
+        "{} | 16",
+        render_make_deferred_namespace_mode_from_exports_type(exports_type)
+      );
       let async_deps = get_outgoing_async_modules(compilation, target_module.as_ref());
       if !async_deps.is_empty() {
         if let Some(header) = header {

crates/rspack_core/src/runtime_template.rs

@@ -1,4 +1,5 @@
 use rspack_cacheable::{cacheable, cacheable_dyn};
+use rspack_collections::Identifier;
 use rspack_core::{
   AsModuleDependency, ContextDependency, ContextOptions, Dependency, DependencyCategory,
   DependencyCodeGeneration, DependencyId, DependencyRange, DependencyTemplate,
@@ -12,6 +13,16 @@ use super::{
   context_dependency_template_as_require_call, create_resource_identifier_for_context_dependency,
 };
 
+fn create_resource_identifier(options: &ContextOptions) -> Identifier {
+  let mut resource_identifier =
+    create_resource_identifier_for_context_dependency(None, options).to_string();
+  if let Some(attributes) = &options.attributes {
+    resource_identifier
+      .push_str(&serde_json::to_string(attributes).expect("json stringify failed"));
+  }
+  resource_identifier.into()
+}
+
 #[cacheable]
 #[derive(Debug, Clone)]
 pub struct ImportContextDependency {
@@ -32,26 +43,21 @@ impl ImportContextDependency {
     value_range: DependencyRange,
     optional: bool,
   ) -> Self {
-    let mut resource_identifier =
-      create_resource_identifier_for_context_dependency(None, &options).to_string();
-    if let Some(attributes) = &options.attributes {
-      resource_identifier
-        .push_str(&serde_json::to_string(attributes).expect("json stringify failed"));
-    }
     Self {
-      options,
+      id: DependencyId::new(),
+      resource_identifier: create_resource_identifier(&options),
       range,
       value_range,
-      id: DependencyId::new(),
-      resource_identifier: resource_identifier.into(),
       optional,
       critical: None,
       factorize_info: Default::default(),
+      options,
     }
   }
 
   pub fn set_referenced_exports(&mut self, referenced_exports: Vec<Vec<Atom>>) {
     self.options.referenced_exports = Some(referenced_exports);
+    self.resource_identifier = create_resource_identifier(&self.options);
   }
 }
 

crates/rspack_plugin_javascript/src/dependency/context/import_context_dependency.rs

@@ -260,6 +260,7 @@ impl AMDDefineDependencyParserPlugin {
       end: call_span.real_hi(),
       referenced_exports: None,
       attributes: None,
+      phase: None,
     };
     let mut dep = AMDRequireContextDependency::new(options, param_range.into(), parser.in_try);
     *dep.critical_mut() = result.critical;

crates/rspack_plugin_javascript/src/parser_plugin/amd/amd_define_dependency_parser_plugin.rs

@@ -190,6 +190,7 @@ impl AMDRequireDependenciesBlockParserPlugin {
       end: call_span.real_hi(),
       referenced_exports: None,
       attributes: None,
+      phase: None,
     };
     let mut dep = AMDRequireContextDependency::new(options, param_range.into(), parser.in_try);
     *dep.critical_mut() = result.critical;

crates/rspack_plugin_javascript/src/parser_plugin/amd/amd_require_dependencies_block_parser_plugin.rs

@@ -51,6 +51,7 @@ fn create_commonjs_require_context_dependency(
     end: span.real_hi(),
     referenced_exports: None,
     attributes: None,
+    phase: None,
   };
   let mut dep = CommonJsRequireContextDependency::new(
     options,
@@ -96,6 +97,7 @@ fn create_require_resolve_context_dependency(
     end,
     referenced_exports: None,
     attributes: None,
+    phase: None,
   };
   RequireResolveContextDependency::new(options, range, parser.in_try)
 }
@@ -397,6 +399,7 @@ impl CommonJsImportsParserPlugin {
         end,
         referenced_exports: None,
         attributes: None,
+        phase: None,
       },
       DependencyRange::from(span)
         .to_loc(Some(parser.source()))

crates/rspack_plugin_javascript/src/parser_plugin/common_js_imports_parse_plugin.rs

@@ -76,6 +76,7 @@ fn create_import_meta_context_dependency(
       end: node.span().real_hi(),
       referenced_exports: None,
       attributes: None,
+      phase: None,
     }
   } else {
     ContextOptions {
@@ -94,6 +95,7 @@ fn create_import_meta_context_dependency(
       end: node.span().real_hi(),
       referenced_exports: None,
       attributes: None,
+      phase: None,
     }
   };
   Some(ImportMetaContextDependency::new(

crates/rspack_plugin_javascript/src/parser_plugin/import_meta_context_dependency_parser_plugin.rs

@@ -390,10 +390,6 @@ impl JavascriptParserPlugin for ImportParserPlugin {
         return None;
       }
 
-      if phase.is_defer() {
-        parser.add_error(rspack_error::error!("import.defer() is not yet supported for ContextModule (the import path is a dynamic expression).").into());
-      }
-
       let ContextModuleScanResult {
         context,
         reg,
@@ -430,6 +426,7 @@ impl JavascriptParserPlugin for ImportParserPlugin {
           end: import_call_span.real_hi(),
           referenced_exports: exports,
           attributes,
+          phase: Some(phase),
         },
         import_call_span.into(),
         dyn_imported.span().into(),

crates/rspack_plugin_javascript/src/parser_plugin/import_parser_plugin.rs

@@ -88,6 +88,7 @@ impl JavascriptParserPlugin for RequireContextDependencyParserPlugin {
           end: expr.span().real_hi(),
           referenced_exports: None,
           attributes: None,
+          phase: None,
         },
         expr.span.into(),
         parser.in_try,