-
-
Notifications
You must be signed in to change notification settings - Fork 595
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: support analysing AMD module format (#8389)
- Loading branch information
Showing
85 changed files
with
2,686 additions
and
22 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
302 changes: 302 additions & 0 deletions
302
crates/rspack_plugin_javascript/src/dependency/amd/amd_define_dependency.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,302 @@ | ||
| use bitflags::bitflags; | ||
| use rspack_core::{ | ||
| AffectType, AsContextDependency, AsModuleDependency, Compilation, Dependency, DependencyCategory, | ||
| DependencyId, DependencyTemplate, DependencyType, RuntimeGlobals, RuntimeSpec, TemplateContext, | ||
| TemplateReplaceSource, | ||
| }; | ||
| use rspack_util::{atom::Atom, json_stringify}; | ||
|
|
||
| use super::local_module::LocalModule; | ||
|
|
||
| bitflags! { | ||
| #[derive(Debug, Clone, Copy, Eq, PartialEq, Hash)] | ||
| struct Branch: u8 { | ||
| const L = 1 << 0; | ||
| const A = 1 << 1; | ||
| const O = 1 << 2; | ||
| const F = 1 << 3; | ||
| } | ||
| } | ||
|
|
||
| impl Branch { | ||
| pub fn get_requests(&self) -> RuntimeGlobals { | ||
| match *self { | ||
| f if f == Branch::F => { | ||
| RuntimeGlobals::REQUIRE | RuntimeGlobals::EXPORTS | RuntimeGlobals::MODULE | ||
| } | ||
| o if o == Branch::O => RuntimeGlobals::MODULE, | ||
| o_f if o_f == (Branch::O | Branch::F) => { | ||
| RuntimeGlobals::REQUIRE | RuntimeGlobals::EXPORTS | RuntimeGlobals::MODULE | ||
| } | ||
| a_f if a_f == (Branch::A | Branch::F) => RuntimeGlobals::EXPORTS | RuntimeGlobals::MODULE, | ||
| a_o if a_o == (Branch::A | Branch::O) => RuntimeGlobals::MODULE, | ||
| a_o_f if a_o_f == (Branch::A | Branch::O | Branch::F) => { | ||
| RuntimeGlobals::EXPORTS | RuntimeGlobals::MODULE | ||
| } | ||
| l_f if l_f == (Branch::L | Branch::F) => RuntimeGlobals::REQUIRE | RuntimeGlobals::MODULE, | ||
| l_o if l_o == (Branch::L | Branch::O) => RuntimeGlobals::empty(), | ||
| l_o_f if l_o_f == (Branch::L | Branch::O | Branch::F) => { | ||
| RuntimeGlobals::REQUIRE | RuntimeGlobals::MODULE | ||
| } | ||
| l_a_f if l_a_f == (Branch::L | Branch::A | Branch::F) => RuntimeGlobals::empty(), | ||
| l_a_o if l_a_o == (Branch::L | Branch::A | Branch::O) => RuntimeGlobals::empty(), | ||
| l_a_o_f if l_a_o_f == (Branch::L | Branch::A | Branch::O | Branch::F) => { | ||
| RuntimeGlobals::empty() | ||
| } | ||
| _ => RuntimeGlobals::empty(), | ||
| } | ||
| } | ||
|
|
||
| pub fn get_definition(&self, local_module_var: &Option<String>) -> String { | ||
| let name = match local_module_var { | ||
| Some(name) => name, | ||
| None => "XXX", | ||
| }; | ||
| match *self { | ||
| f if f == Branch::F => "var __WEBPACK_AMD_DEFINE_RESULT__;".to_string(), | ||
| o if o == Branch::O => "".to_string(), | ||
| o_f if o_f == (Branch::O | Branch::F) => { | ||
| "var __WEBPACK_AMD_DEFINE_FACTORY__, __WEBPACK_AMD_DEFINE_RESULT__;".to_string() | ||
| } | ||
| a_f if a_f == (Branch::A | Branch::F) => { | ||
| "var __WEBPACK_AMD_DEFINE_ARRAY__, __WEBPACK_AMD_DEFINE_RESULT__;".to_string() | ||
| } | ||
| a_o if a_o == (Branch::A | Branch::O) => "".to_string(), | ||
| a_o_f if a_o_f == (Branch::A | Branch::O | Branch::F) => "var __WEBPACK_AMD_DEFINE_FACTORY__, __WEBPACK_AMD_DEFINE_ARRAY__, __WEBPACK_AMD_DEFINE_RESULT__;".to_string(), | ||
| l_f if l_f == (Branch::L | Branch::F) => { | ||
| format!("var {}, {}module;", name, name) | ||
| }, | ||
| l_o if l_o == (Branch::L | Branch::O) => { | ||
| format!("var {};", name) | ||
| }, | ||
| l_o_f if l_o_f == (Branch::L | Branch::O | Branch::F) => { | ||
| format!("var {}, {}factory, {}module;", name, name, name) | ||
| }, | ||
| l_a_f if l_a_f == (Branch::L | Branch::A | Branch::F) => { | ||
| format!("var __WEBPACK_AMD_DEFINE_ARRAY__, {}, {}exports;", name, name) | ||
| }, | ||
| l_a_o if l_a_o == (Branch::L | Branch::A | Branch::O)=> { | ||
| format!("var {};", name) | ||
| }, | ||
| l_a_o_f if l_a_o_f == (Branch::L | Branch::A | Branch::O | Branch::F) => { | ||
| format!("var {}array, {}factory, {}exports, {};", name, name, name, name) | ||
| }, | ||
| _ => "".to_string(), | ||
| } | ||
| } | ||
|
|
||
| pub fn get_content( | ||
| &self, | ||
| local_module_var: &Option<String>, | ||
| named_module: &Option<Atom>, | ||
| ) -> String { | ||
| let local_module_var = match local_module_var { | ||
| Some(name) => name, | ||
| None => "XXX", | ||
| }; | ||
| let named_module = match named_module { | ||
| Some(name) => name, | ||
| None => "YYY", | ||
| }; | ||
| match *self { | ||
| f if f == Branch::F => { | ||
| format!( | ||
| "!(__WEBPACK_AMD_DEFINE_RESULT__ = (#).call(exports, {require}, exports, module), | ||
| __WEBPACK_AMD_DEFINE_RESULT__ !== undefined && (module.exports = __WEBPACK_AMD_DEFINE_RESULT__))", | ||
| require = RuntimeGlobals::REQUIRE.name() | ||
| ) | ||
| } | ||
| o if o == Branch::O => "!(module.exports = #)".to_string(), | ||
| o_f if o_f == (Branch::O | Branch::F) => { | ||
| format!( | ||
| "!(__WEBPACK_AMD_DEFINE_FACTORY__ = (#), | ||
| __WEBPACK_AMD_DEFINE_RESULT__ = (typeof __WEBPACK_AMD_DEFINE_FACTORY__ === 'function' ? | ||
| (__WEBPACK_AMD_DEFINE_FACTORY__.call(exports, {require}, exports, module)) : | ||
| __WEBPACK_AMD_DEFINE_FACTORY__), | ||
| __WEBPACK_AMD_DEFINE_RESULT__ !== undefined && (module.exports = __WEBPACK_AMD_DEFINE_RESULT__))", | ||
| require = RuntimeGlobals::REQUIRE.name() | ||
| ) | ||
| } | ||
| a_f if a_f == (Branch::A | Branch::F) => "!(__WEBPACK_AMD_DEFINE_ARRAY__ = #, __WEBPACK_AMD_DEFINE_RESULT__ = (#).apply(exports, __WEBPACK_AMD_DEFINE_ARRAY__), | ||
| __WEBPACK_AMD_DEFINE_RESULT__ !== undefined && (module.exports = __WEBPACK_AMD_DEFINE_RESULT__))".to_string(), | ||
| a_o if a_o == (Branch::A | Branch::O) => "!(#, module.exports = #)".to_string(), | ||
| a_o_f if a_o_f == (Branch::A | Branch::O | Branch::F) => { | ||
| "!(__WEBPACK_AMD_DEFINE_ARRAY__ = #, __WEBPACK_AMD_DEFINE_FACTORY__ = (#), | ||
| __WEBPACK_AMD_DEFINE_RESULT__ = (typeof __WEBPACK_AMD_DEFINE_FACTORY__ === 'function' ? | ||
| (__WEBPACK_AMD_DEFINE_FACTORY__.apply(exports, __WEBPACK_AMD_DEFINE_ARRAY__)) : __WEBPACK_AMD_DEFINE_FACTORY__), | ||
| __WEBPACK_AMD_DEFINE_RESULT__ !== undefined && (module.exports = __WEBPACK_AMD_DEFINE_RESULT__))".to_string() | ||
| } | ||
| l_f if l_f == (Branch::L | Branch::F) => { | ||
| format!( | ||
| "!({var_name}module = {{ id: {module_id}, exports: {{}}, loaded: false }}, {var_name} = (#).call({var_name}module.exports, {require}, {var_name}module.exports, {var_name}module), {var_name}module.loaded = true, {var_name} === undefined && ({var_name} = {var_name}module.exports))", | ||
| var_name = local_module_var, | ||
| module_id = json_stringify(named_module), | ||
| require = RuntimeGlobals::REQUIRE.name(), | ||
| ) | ||
| } | ||
| l_o if l_o == (Branch::L | Branch::O) => format!("!({} = #)", local_module_var), | ||
| l_o_f if l_o_f == (Branch::L | Branch::O | Branch::F) => { | ||
| format!( | ||
| "!({var_name}factory = (#), (typeof {var_name}factory === 'function' ? (({var_name}module = {{ id: {module_id}, exports: {{}}, loaded: false }}), ({var_name} = {var_name}factory.call({var_name}module.exports, {require}, {var_name}module.exports, {var_name}module)), ({var_name}module.loaded = true), {var_name} === undefined && ({var_name} = {var_name}module.exports)) : {var_name} = {var_name}factory))", | ||
| var_name = local_module_var, | ||
| module_id = json_stringify(named_module), | ||
| require = RuntimeGlobals::REQUIRE.name(), | ||
| ) | ||
| } | ||
| l_a_f if l_a_f == (Branch::L | Branch::A | Branch::F) => format!("!(__WEBPACK_AMD_DEFINE_ARRAY__ = #, {} = (#).apply({}exports = {{}}, __WEBPACK_AMD_DEFINE_ARRAY__), {} === undefined && ({} = {}exports))", local_module_var, local_module_var, local_module_var, local_module_var, local_module_var), | ||
| l_a_o if l_a_o == (Branch::L | Branch::A | Branch::O) => format!("!(#, {} = #)", local_module_var), | ||
| l_a_o_f if l_a_o_f == (Branch::L | Branch::A | Branch::O | Branch::F) => format!( | ||
| "!({var_name}array = #, {var_name}factory = (#), | ||
| (typeof {var_name}factory === 'function' ? | ||
| (({var_name} = {var_name}factory.apply({var_name}exports = {{}}, {var_name}array)), {var_name} === undefined && ({var_name} = {var_name}exports)) : | ||
| ({var_name} = {var_name}factory) | ||
| ))", | ||
| var_name = local_module_var, | ||
| ), | ||
| _ => "".to_string(), | ||
| } | ||
| } | ||
| } | ||
|
|
||
| #[derive(Debug, Clone)] | ||
| pub struct AmdDefineDependency { | ||
| id: DependencyId, | ||
| range: (u32, u32), | ||
| array_range: Option<(u32, u32)>, | ||
| function_range: Option<(u32, u32)>, | ||
| object_range: Option<(u32, u32)>, | ||
| named_module: Option<Atom>, | ||
| local_module: Option<LocalModule>, | ||
| } | ||
|
|
||
| impl AmdDefineDependency { | ||
| pub fn new( | ||
| range: (u32, u32), | ||
| array_range: Option<(u32, u32)>, | ||
| function_range: Option<(u32, u32)>, | ||
| object_range: Option<(u32, u32)>, | ||
| named_module: Option<Atom>, | ||
| local_module: Option<LocalModule>, | ||
| ) -> Self { | ||
| Self { | ||
| id: DependencyId::new(), | ||
| range, | ||
| array_range, | ||
| function_range, | ||
| object_range, | ||
| named_module, | ||
| local_module, | ||
| } | ||
| } | ||
|
|
||
| pub fn get_local_module_mut(&mut self) -> Option<&mut LocalModule> { | ||
| self.local_module.as_mut() | ||
| } | ||
| } | ||
|
|
||
| impl Dependency for AmdDefineDependency { | ||
| fn id(&self) -> &DependencyId { | ||
| &self.id | ||
| } | ||
|
|
||
| fn category(&self) -> &DependencyCategory { | ||
| &DependencyCategory::Amd | ||
| } | ||
|
|
||
| fn dependency_type(&self) -> &DependencyType { | ||
| &DependencyType::AmdDefine | ||
| } | ||
|
|
||
| fn could_affect_referencing_module(&self) -> AffectType { | ||
| AffectType::False | ||
| } | ||
| } | ||
|
|
||
| impl AmdDefineDependency { | ||
| fn local_module_var(&self) -> Option<String> { | ||
| self.local_module.as_ref().and_then(|m| { | ||
| if m.is_used() { | ||
| Some(m.variable_name()) | ||
| } else { | ||
| None | ||
| } | ||
| }) | ||
| } | ||
|
|
||
| fn branch(&self) -> Branch { | ||
| let mut ret = Branch::empty(); | ||
| if self.local_module.as_ref().is_some_and(|m| m.is_used()) { | ||
| ret |= Branch::L; | ||
| } | ||
| if self.array_range.is_some() { | ||
| ret |= Branch::A; | ||
| } | ||
| if self.object_range.is_some() { | ||
| ret |= Branch::O; | ||
| } | ||
| if self.function_range.is_some() { | ||
| ret |= Branch::F; | ||
| } | ||
| ret | ||
| } | ||
| } | ||
|
|
||
| impl DependencyTemplate for AmdDefineDependency { | ||
| fn apply( | ||
| &self, | ||
| source: &mut TemplateReplaceSource, | ||
| code_generatable_context: &mut TemplateContext, | ||
| ) { | ||
| let branch = self.branch(); | ||
| code_generatable_context | ||
| .runtime_requirements | ||
| .insert(branch.get_requests()); | ||
|
|
||
| let local_module_var = self.local_module_var(); | ||
|
|
||
| let text = branch.get_content(&local_module_var, &self.named_module); | ||
| let definition = branch.get_definition(&local_module_var); | ||
|
|
||
| let mut texts = text.split('#'); | ||
|
|
||
| if !definition.is_empty() { | ||
| source.insert(0, &definition, None); | ||
| } | ||
|
|
||
| let mut current = self.range.0; | ||
| if let Some(array_range) = self.array_range { | ||
| source.replace(current, array_range.0, texts.next().unwrap_or(""), None); | ||
| current = array_range.1; | ||
| } | ||
|
|
||
| if let Some(object_range) = self.object_range { | ||
| source.replace(current, object_range.0, texts.next().unwrap_or(""), None); | ||
| current = object_range.1; | ||
| } else if let Some(function_range) = self.function_range { | ||
| source.replace(current, function_range.0, texts.next().unwrap_or(""), None); | ||
| current = function_range.1; | ||
| } | ||
|
|
||
| source.replace(current, self.range.1, texts.next().unwrap_or(""), None); | ||
|
|
||
| if texts.next().is_some() { | ||
| panic!("Implementation error"); | ||
| } | ||
| } | ||
|
|
||
| fn dependency_id(&self) -> Option<DependencyId> { | ||
| Some(self.id) | ||
| } | ||
|
|
||
| fn update_hash( | ||
| &self, | ||
| _hasher: &mut dyn std::hash::Hasher, | ||
| _compilation: &Compilation, | ||
| _runtime: Option<&RuntimeSpec>, | ||
| ) { | ||
| } | ||
| } | ||
|
|
||
| impl AsModuleDependency for AmdDefineDependency {} | ||
|
|
||
| impl AsContextDependency for AmdDefineDependency {} |
Oops, something went wrong.
96f500dThere was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
📝 Ran ecosystem CI: Open
96f500dThere was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
📝 Benchmark detail: Open