use crate::bindings;use crate::error::generic_error;use crate::module_specifier::ModuleSpecifier;use crate::resolve_import;use crate::resolve_url;use crate::OpState;use anyhow::Error;use futures::future::FutureExt;use futures::stream::FuturesUnordered;use futures::stream::Stream;use futures::stream::StreamFuture;use futures::stream::TryStreamExt;use log::debug;use std::cell::RefCell;use std::collections::HashMap;use std::collections::HashSet;use std::collections::VecDeque;use std::future::Future;use std::pin::Pin;use std::rc::Rc;use std::task::Context;use std::task::Poll;
pub type ModuleId = i32;pub(crate) type ModuleLoadId = i32;
pub const BOM_CHAR: &[u8] = &[0xef, 0xbb, 0xbf];
fn strip_bom(source_code: &[u8]) -> &[u8] { if source_code.starts_with(BOM_CHAR) { &source_code[BOM_CHAR.len()..] } else { source_code }}
const SUPPORTED_TYPE_ASSERTIONS: &[&str] = &["json"];
pub(crate) fn validate_import_assertions( scope: &mut v8::HandleScope, assertions: &HashMap<String, String>,) { for (key, value) in assertions { if key == "type" && !SUPPORTED_TYPE_ASSERTIONS.contains(&value.as_str()) { let message = v8::String::new( scope, &format!("\"{}\" is not a valid module type.", value), ) .unwrap(); let exception = v8::Exception::type_error(scope, message); scope.throw_exception(exception); return; } }}
#[derive(Debug)]pub(crate) enum ImportAssertionsKind { StaticImport, DynamicImport,}
pub(crate) fn parse_import_assertions( scope: &mut v8::HandleScope, import_assertions: v8::Local<v8::FixedArray>, kind: ImportAssertionsKind,) -> HashMap<String, String> { let mut assertions: HashMap<String, String> = HashMap::default();
let assertions_per_line = match kind { ImportAssertionsKind::StaticImport => 3, ImportAssertionsKind::DynamicImport => 2, }; assert_eq!(import_assertions.length() % assertions_per_line, 0); let no_of_assertions = import_assertions.length() / assertions_per_line;
for i in 0..no_of_assertions { let assert_key = import_assertions .get(scope, assertions_per_line * i) .unwrap(); let assert_key_val = v8::Local::<v8::Value>::try_from(assert_key).unwrap(); let assert_value = import_assertions .get(scope, (assertions_per_line * i) + 1) .unwrap(); let assert_value_val = v8::Local::<v8::Value>::try_from(assert_value).unwrap(); assertions.insert( assert_key_val.to_rust_string_lossy(scope), assert_value_val.to_rust_string_lossy(scope), ); }
assertions}
pub(crate) fn get_asserted_module_type_from_assertions( assertions: &HashMap<String, String>,) -> AssertedModuleType { assertions .get("type") .map(|ty| { if ty == "json" { AssertedModuleType::Json } else { AssertedModuleType::JavaScriptOrWasm } }) .unwrap_or(AssertedModuleType::JavaScriptOrWasm)}
#[allow(clippy::unnecessary_wraps)]fn json_module_evaluation_steps<'a>( context: v8::Local<'a, v8::Context>, module: v8::Local<v8::Module>,) -> Option<v8::Local<'a, v8::Value>> { let scope = &mut unsafe { v8::CallbackScope::new(context) }; let tc_scope = &mut v8::TryCatch::new(scope); let module_map = tc_scope .get_slot::<Rc<RefCell<ModuleMap>>>() .unwrap() .clone();
let handle = v8::Global::<v8::Module>::new(tc_scope, module); let value_handle = module_map .borrow_mut() .json_value_store .remove(&handle) .unwrap(); let value_local = v8::Local::new(tc_scope, value_handle);
let name = v8::String::new(tc_scope, "default").unwrap(); assert!( module.set_synthetic_module_export(tc_scope, name, value_local) == Some(true) ); assert!(!tc_scope.has_caught());
let resolver = v8::PromiseResolver::new(tc_scope).unwrap(); let undefined = v8::undefined(tc_scope); resolver.resolve(tc_scope, undefined.into()); Some(resolver.get_promise(tc_scope).into())}
#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]pub enum ModuleType { JavaScript, Json,}
impl std::fmt::Display for ModuleType { fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { match self { Self::JavaScript => write!(f, "JavaScript"), Self::Json => write!(f, "JSON"), } }}
#[derive(Debug, Clone, Eq, PartialEq)]pub struct ModuleSource { pub code: Box<[u8]>, pub module_type: ModuleType, pub module_url_specified: String, pub module_url_found: String,}
pub(crate) type PrepareLoadFuture = dyn Future<Output = (ModuleLoadId, Result<RecursiveModuleLoad, Error>)>;pub type ModuleSourceFuture = dyn Future<Output = Result<ModuleSource, Error>>;
type ModuleLoadFuture = dyn Future<Output = Result<(ModuleRequest, ModuleSource), Error>>;
pub trait ModuleLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error>;
fn load( &self, module_specifier: &ModuleSpecifier, maybe_referrer: Option<ModuleSpecifier>, is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>>;
fn prepare_load( &self, _op_state: Rc<RefCell<OpState>>, _module_specifier: &ModuleSpecifier, _maybe_referrer: Option<String>, _is_dyn_import: bool, ) -> Pin<Box<dyn Future<Output = Result<(), Error>>>> { async { Ok(()) }.boxed_local() }}
pub struct NoopModuleLoader;
impl ModuleLoader for NoopModuleLoader { fn resolve( &self, _specifier: &str, _referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { Err(generic_error("Module loading is not supported")) }
fn load( &self, _module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { async { Err(generic_error("Module loading is not supported")) } .boxed_local() }}
pub struct FsModuleLoader;
impl ModuleLoader for FsModuleLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { Ok(resolve_import(specifier, referrer)?) }
fn load( &self, module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dynamic: bool, ) -> Pin<Box<ModuleSourceFuture>> { let module_specifier = module_specifier.clone(); async move { let path = module_specifier.to_file_path().map_err(|_| { generic_error(format!( "Provided module specifier \"{}\" is not a file URL.", module_specifier )) })?; let module_type = if let Some(extension) = path.extension() { let ext = extension.to_string_lossy().to_lowercase(); if ext == "json" { ModuleType::Json } else { ModuleType::JavaScript } } else { ModuleType::JavaScript };
let code = std::fs::read(path)?; let module = ModuleSource { code: code.into_boxed_slice(), module_type, module_url_specified: module_specifier.to_string(), module_url_found: module_specifier.to_string(), }; Ok(module) } .boxed_local() }}
#[derive(Debug)]enum LoadInit { Main(String), Side(String), DynamicImport(String, String, AssertedModuleType),}
#[derive(Debug, Eq, PartialEq)]pub enum LoadState { Init, LoadingRoot, LoadingImports, Done,}
pub(crate) struct RecursiveModuleLoad { pub id: ModuleLoadId, pub root_module_id: Option<ModuleId>, init: LoadInit, root_asserted_module_type: Option<AssertedModuleType>, root_module_type: Option<ModuleType>, state: LoadState, module_map_rc: Rc<RefCell<ModuleMap>>, pending: FuturesUnordered<Pin<Box<ModuleLoadFuture>>>, visited: HashSet<ModuleRequest>, op_state: Rc<RefCell<OpState>>, loader: Rc<dyn ModuleLoader>,}
impl RecursiveModuleLoad { fn main(specifier: &str, module_map_rc: Rc<RefCell<ModuleMap>>) -> Self { Self::new(LoadInit::Main(specifier.to_string()), module_map_rc) }
fn side(specifier: &str, module_map_rc: Rc<RefCell<ModuleMap>>) -> Self { Self::new(LoadInit::Side(specifier.to_string()), module_map_rc) }
fn dynamic_import( specifier: &str, referrer: &str, asserted_module_type: AssertedModuleType, module_map_rc: Rc<RefCell<ModuleMap>>, ) -> Self { Self::new( LoadInit::DynamicImport( specifier.to_string(), referrer.to_string(), asserted_module_type, ), module_map_rc, ) }
fn new(init: LoadInit, module_map_rc: Rc<RefCell<ModuleMap>>) -> Self { let id = { let mut module_map = module_map_rc.borrow_mut(); let id = module_map.next_load_id; module_map.next_load_id += 1; id }; let op_state = module_map_rc.borrow().op_state.clone(); let loader = module_map_rc.borrow().loader.clone(); let asserted_module_type = match init { LoadInit::DynamicImport(_, _, module_type) => module_type, _ => AssertedModuleType::JavaScriptOrWasm, }; let mut load = Self { id, root_module_id: None, root_asserted_module_type: None, root_module_type: None, init, state: LoadState::Init, module_map_rc: module_map_rc.clone(), op_state, loader, pending: FuturesUnordered::new(), visited: HashSet::new(), }; if let Ok(root_specifier) = load.resolve_root() { if let Some(module_id) = module_map_rc .borrow() .get_id(root_specifier.as_str(), asserted_module_type) { load.root_module_id = Some(module_id); load.root_asserted_module_type = Some(asserted_module_type); load.root_module_type = Some( module_map_rc .borrow() .get_info_by_id(&module_id) .unwrap() .module_type, ); } } load }
fn resolve_root(&self) -> Result<ModuleSpecifier, Error> { match self.init { LoadInit::Main(ref specifier) => { self.loader.resolve(specifier, ".", true) } LoadInit::Side(ref specifier) => { self.loader.resolve(specifier, ".", false) } LoadInit::DynamicImport(ref specifier, ref referrer, _) => { self.loader.resolve(specifier, referrer, false) } } }
async fn prepare(&self) -> Result<(), Error> { let op_state = self.op_state.clone(); let (module_specifier, maybe_referrer) = match self.init { LoadInit::Main(ref specifier) => { let spec = self.loader.resolve(specifier, ".", true)?; (spec, None) } LoadInit::Side(ref specifier) => { let spec = self.loader.resolve(specifier, ".", false)?; (spec, None) } LoadInit::DynamicImport(ref specifier, ref referrer, _) => { let spec = self.loader.resolve(specifier, referrer, false)?; (spec, Some(referrer.to_string())) } };
self .loader .prepare_load( op_state, &module_specifier, maybe_referrer, self.is_dynamic_import(), ) .await }
fn is_currently_loading_main_module(&self) -> bool { !self.is_dynamic_import() && matches!(self.init, LoadInit::Main(..)) && self.state == LoadState::LoadingRoot }
fn is_dynamic_import(&self) -> bool { matches!(self.init, LoadInit::DynamicImport(..)) }
pub(crate) fn register_and_recurse( &mut self, scope: &mut v8::HandleScope, module_request: &ModuleRequest, module_source: &ModuleSource, ) -> Result<(), ModuleError> { let expected_asserted_module_type = module_source.module_type.into(); if module_request.asserted_module_type != expected_asserted_module_type { return Err(ModuleError::Other(generic_error(format!( "Expected a \"{}\" module but loaded a \"{}\" module.", module_request.asserted_module_type, module_source.module_type, )))); }
if module_source.module_url_specified != module_source.module_url_found { self.module_map_rc.borrow_mut().alias( &module_source.module_url_specified, expected_asserted_module_type, &module_source.module_url_found, ); } let maybe_module_id = self.module_map_rc.borrow().get_id( &module_source.module_url_found, expected_asserted_module_type, ); let module_id = match maybe_module_id { Some(id) => { debug!( "Already-registered module fetched again: {}", module_source.module_url_found ); id } None => match module_source.module_type { ModuleType::JavaScript => { self.module_map_rc.borrow_mut().new_es_module( scope, self.is_currently_loading_main_module(), &module_source.module_url_found, &module_source.code, )? } ModuleType::Json => self.module_map_rc.borrow_mut().new_json_module( scope, &module_source.module_url_found, &module_source.code, )?, }, };
let mut already_registered = VecDeque::new(); already_registered.push_back((module_id, module_request.clone())); self.visited.insert(module_request.clone()); while let Some((module_id, module_request)) = already_registered.pop_front() { let referrer = module_request.specifier.clone(); let imports = self .module_map_rc .borrow() .get_requested_modules(module_id) .unwrap() .clone(); for module_request in imports { if !self.visited.contains(&module_request) { if let Some(module_id) = self.module_map_rc.borrow().get_id( module_request.specifier.as_str(), module_request.asserted_module_type, ) { already_registered.push_back((module_id, module_request.clone())); } else { let referrer = referrer.clone(); let request = module_request.clone(); let loader = self.loader.clone(); let is_dynamic_import = self.is_dynamic_import(); let fut = async move { let load_result = loader .load( &request.specifier, Some(referrer.clone()), is_dynamic_import, ) .await; load_result.map(|s| (request, s)) }; self.pending.push(fut.boxed_local()); } self.visited.insert(module_request); } } }
if self.state == LoadState::LoadingRoot { self.root_module_id = Some(module_id); self.root_asserted_module_type = Some(module_source.module_type.into()); self.state = LoadState::LoadingImports; } if self.pending.is_empty() { self.state = LoadState::Done; }
Ok(()) }}
impl Stream for RecursiveModuleLoad { type Item = Result<(ModuleRequest, ModuleSource), Error>;
fn poll_next( self: Pin<&mut Self>, cx: &mut Context, ) -> Poll<Option<Self::Item>> { let inner = self.get_mut(); match inner.state { LoadState::Init => { let module_specifier = match inner.resolve_root() { Ok(url) => url, Err(error) => return Poll::Ready(Some(Err(error))), }; let load_fut = if let Some(_module_id) = inner.root_module_id { let asserted_module_type = inner.root_asserted_module_type.unwrap(); let module_type = inner.root_module_type.unwrap(); let module_request = ModuleRequest { specifier: module_specifier.clone(), asserted_module_type, }; let module_source = ModuleSource { module_url_specified: module_specifier.to_string(), module_url_found: module_specifier.to_string(), code: Default::default(), module_type, }; futures::future::ok((module_request, module_source)).boxed() } else { let maybe_referrer = match inner.init { LoadInit::DynamicImport(_, ref referrer, _) => { resolve_url(referrer).ok() } _ => None, }; let asserted_module_type = match inner.init { LoadInit::DynamicImport(_, _, module_type) => module_type, _ => AssertedModuleType::JavaScriptOrWasm, }; let module_request = ModuleRequest { specifier: module_specifier.clone(), asserted_module_type, }; let loader = inner.loader.clone(); let is_dynamic_import = inner.is_dynamic_import(); async move { let result = loader .load(&module_specifier, maybe_referrer, is_dynamic_import) .await; result.map(|s| (module_request, s)) } .boxed_local() }; inner.pending.push(load_fut); inner.state = LoadState::LoadingRoot; inner.try_poll_next_unpin(cx) } LoadState::LoadingRoot | LoadState::LoadingImports => { match inner.pending.try_poll_next_unpin(cx)? { Poll::Ready(None) => unreachable!(), Poll::Ready(Some(info)) => Poll::Ready(Some(Ok(info))), Poll::Pending => Poll::Pending, } } LoadState::Done => Poll::Ready(None), } }}
#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]pub(crate) enum AssertedModuleType { JavaScriptOrWasm, Json,}
impl From<ModuleType> for AssertedModuleType { fn from(module_type: ModuleType) -> AssertedModuleType { match module_type { ModuleType::JavaScript => AssertedModuleType::JavaScriptOrWasm, ModuleType::Json => AssertedModuleType::Json, } }}
impl std::fmt::Display for AssertedModuleType { fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { match self { Self::JavaScriptOrWasm => write!(f, "JavaScriptOrWasm"), Self::Json => write!(f, "JSON"), } }}
#[derive(Clone, Debug, Eq, Hash, PartialEq)]pub(crate) struct ModuleRequest { pub specifier: ModuleSpecifier, pub asserted_module_type: AssertedModuleType,}
pub(crate) struct ModuleInfo { #[allow(unused)] pub id: ModuleId, pub main: bool, pub name: String, pub requests: Vec<ModuleRequest>, pub module_type: ModuleType,}
enum SymbolicModule { Alias(String), Mod(ModuleId),}
#[derive(Debug)]pub(crate) enum ModuleError { Exception(v8::Global<v8::Value>), Other(Error),}
pub(crate) struct ModuleMap { ids_by_handle: HashMap<v8::Global<v8::Module>, ModuleId>, handles_by_id: HashMap<ModuleId, v8::Global<v8::Module>>, info: HashMap<ModuleId, ModuleInfo>, by_name: HashMap<(String, AssertedModuleType), SymbolicModule>, next_module_id: ModuleId, next_load_id: ModuleLoadId,
pub loader: Rc<dyn ModuleLoader>, op_state: Rc<RefCell<OpState>>, pub(crate) dynamic_import_map: HashMap<ModuleLoadId, v8::Global<v8::PromiseResolver>>, pub(crate) preparing_dynamic_imports: FuturesUnordered<Pin<Box<PrepareLoadFuture>>>, pub(crate) pending_dynamic_imports: FuturesUnordered<StreamFuture<RecursiveModuleLoad>>,
json_value_store: HashMap<v8::Global<v8::Module>, v8::Global<v8::Value>>,}
impl ModuleMap { pub(crate) fn new( loader: Rc<dyn ModuleLoader>, op_state: Rc<RefCell<OpState>>, ) -> ModuleMap { Self { ids_by_handle: HashMap::new(), handles_by_id: HashMap::new(), info: HashMap::new(), by_name: HashMap::new(), next_module_id: 1, next_load_id: 1, loader, op_state, dynamic_import_map: HashMap::new(), preparing_dynamic_imports: FuturesUnordered::new(), pending_dynamic_imports: FuturesUnordered::new(), json_value_store: HashMap::new(), } }
fn get_id( &self, name: &str, asserted_module_type: AssertedModuleType, ) -> Option<ModuleId> { let mut mod_name = name; loop { let symbolic_module = self .by_name .get(&(mod_name.to_string(), asserted_module_type))?; match symbolic_module { SymbolicModule::Alias(target) => { mod_name = target; } SymbolicModule::Mod(mod_id) => return Some(*mod_id), } } }
fn new_json_module( &mut self, scope: &mut v8::HandleScope, name: &str, source: &[u8], ) -> Result<ModuleId, ModuleError> { let name_str = v8::String::new(scope, name).unwrap(); let source_str = v8::String::new_from_utf8( scope, strip_bom(source), v8::NewStringType::Normal, ) .unwrap();
let tc_scope = &mut v8::TryCatch::new(scope);
let parsed_json = match v8::json::parse(tc_scope, source_str) { Some(parsed_json) => parsed_json, None => { assert!(tc_scope.has_caught()); let exception = tc_scope.exception().unwrap(); let exception = v8::Global::new(tc_scope, exception); return Err(ModuleError::Exception(exception)); } };
let export_names = [v8::String::new(tc_scope, "default").unwrap()]; let module = v8::Module::create_synthetic_module( tc_scope, name_str, &export_names, json_module_evaluation_steps, );
let handle = v8::Global::<v8::Module>::new(tc_scope, module); let value_handle = v8::Global::<v8::Value>::new(tc_scope, parsed_json); self.json_value_store.insert(handle.clone(), value_handle);
let id = self.create_module_info(name, ModuleType::Json, handle, false, vec![]);
Ok(id) }
pub(crate) fn new_es_module( &mut self, scope: &mut v8::HandleScope, main: bool, name: &str, source: &[u8], ) -> Result<ModuleId, ModuleError> { let name_str = v8::String::new(scope, name).unwrap(); let source_str = v8::String::new_from_utf8(scope, source, v8::NewStringType::Normal) .unwrap();
let origin = bindings::module_origin(scope, name_str); let source = v8::script_compiler::Source::new(source_str, Some(&origin));
let tc_scope = &mut v8::TryCatch::new(scope);
let maybe_module = v8::script_compiler::compile_module(tc_scope, source);
if tc_scope.has_caught() { assert!(maybe_module.is_none()); let exception = tc_scope.exception().unwrap(); let exception = v8::Global::new(tc_scope, exception); return Err(ModuleError::Exception(exception)); }
let module = maybe_module.unwrap();
let mut requests: Vec<ModuleRequest> = vec![]; let module_requests = module.get_module_requests(); for i in 0..module_requests.length() { let module_request = v8::Local::<v8::ModuleRequest>::try_from( module_requests.get(tc_scope, i).unwrap(), ) .unwrap(); let import_specifier = module_request .get_specifier() .to_rust_string_lossy(tc_scope);
let import_assertions = module_request.get_import_assertions();
let assertions = parse_import_assertions( tc_scope, import_assertions, ImportAssertionsKind::StaticImport, );
validate_import_assertions(tc_scope, &assertions); if tc_scope.has_caught() { let exception = tc_scope.exception().unwrap(); let exception = v8::Global::new(tc_scope, exception); return Err(ModuleError::Exception(exception)); }
let module_specifier = match self.loader.resolve(&import_specifier, name, false) { Ok(s) => s, Err(e) => return Err(ModuleError::Other(e)), }; let asserted_module_type = get_asserted_module_type_from_assertions(&assertions); let request = ModuleRequest { specifier: module_specifier, asserted_module_type, }; requests.push(request); }
if main { let maybe_main_module = self.info.values().find(|module| module.main); if let Some(main_module) = maybe_main_module { return Err(ModuleError::Other(generic_error( format!("Trying to create \"main\" module ({:?}), when one already exists ({:?})", name, main_module.name, )))); } }
let handle = v8::Global::<v8::Module>::new(tc_scope, module); let id = self.create_module_info( name, ModuleType::JavaScript, handle, main, requests, );
Ok(id) }
fn create_module_info( &mut self, name: &str, module_type: ModuleType, handle: v8::Global<v8::Module>, main: bool, requests: Vec<ModuleRequest>, ) -> ModuleId { let id = self.next_module_id; self.next_module_id += 1; self.by_name.insert( (name.to_string(), module_type.into()), SymbolicModule::Mod(id), ); self.handles_by_id.insert(id, handle.clone()); self.ids_by_handle.insert(handle, id); self.info.insert( id, ModuleInfo { id, main, name: name.to_string(), requests, module_type, }, );
id }
fn get_requested_modules(&self, id: ModuleId) -> Option<&Vec<ModuleRequest>> { self.info.get(&id).map(|i| &i.requests) }
fn is_registered( &self, specifier: &ModuleSpecifier, asserted_module_type: AssertedModuleType, ) -> bool { if let Some(id) = self.get_id(specifier.as_str(), asserted_module_type) { let info = self.get_info_by_id(&id).unwrap(); return asserted_module_type == info.module_type.into(); }
false }
fn alias( &mut self, name: &str, asserted_module_type: AssertedModuleType, target: &str, ) { self.by_name.insert( (name.to_string(), asserted_module_type), SymbolicModule::Alias(target.to_string()), ); }
#[cfg(test)] fn is_alias( &self, name: &str, asserted_module_type: AssertedModuleType, ) -> bool { let cond = self.by_name.get(&(name.to_string(), asserted_module_type)); matches!(cond, Some(SymbolicModule::Alias(_))) }
pub(crate) fn get_handle( &self, id: ModuleId, ) -> Option<v8::Global<v8::Module>> { self.handles_by_id.get(&id).cloned() }
pub(crate) fn get_info( &self, global: &v8::Global<v8::Module>, ) -> Option<&ModuleInfo> { if let Some(id) = self.ids_by_handle.get(global) { return self.info.get(id); }
None }
pub(crate) fn get_info_by_id(&self, id: &ModuleId) -> Option<&ModuleInfo> { self.info.get(id) }
pub(crate) async fn load_main( module_map_rc: Rc<RefCell<ModuleMap>>, specifier: &str, ) -> Result<RecursiveModuleLoad, Error> { let load = RecursiveModuleLoad::main(specifier, module_map_rc.clone()); load.prepare().await?; Ok(load) }
pub(crate) async fn load_side( module_map_rc: Rc<RefCell<ModuleMap>>, specifier: &str, ) -> Result<RecursiveModuleLoad, Error> { let load = RecursiveModuleLoad::side(specifier, module_map_rc.clone()); load.prepare().await?; Ok(load) }
pub(crate) fn load_dynamic_import( module_map_rc: Rc<RefCell<ModuleMap>>, specifier: &str, referrer: &str, asserted_module_type: AssertedModuleType, resolver_handle: v8::Global<v8::PromiseResolver>, ) { let load = RecursiveModuleLoad::dynamic_import( specifier, referrer, asserted_module_type, module_map_rc.clone(), ); module_map_rc .borrow_mut() .dynamic_import_map .insert(load.id, resolver_handle); let resolve_result = module_map_rc .borrow() .loader .resolve(specifier, referrer, false); let fut = match resolve_result { Ok(module_specifier) => { if module_map_rc .borrow() .is_registered(&module_specifier, asserted_module_type) { async move { (load.id, Ok(load)) }.boxed_local() } else { async move { (load.id, load.prepare().await.map(|()| load)) } .boxed_local() } } Err(error) => async move { (load.id, Err(error)) }.boxed_local(), }; module_map_rc .borrow_mut() .preparing_dynamic_imports .push(fut); }
pub(crate) fn has_pending_dynamic_imports(&self) -> bool { !(self.preparing_dynamic_imports.is_empty() && self.pending_dynamic_imports.is_empty()) }
pub(crate) fn resolve_callback<'s>( &self, scope: &mut v8::HandleScope<'s>, specifier: &str, referrer: &str, import_assertions: HashMap<String, String>, ) -> Option<v8::Local<'s, v8::Module>> { let resolved_specifier = self .loader .resolve(specifier, referrer, false) .expect("Module should have been already resolved");
let module_type = get_asserted_module_type_from_assertions(&import_assertions);
if let Some(id) = self.get_id(resolved_specifier.as_str(), module_type) { if let Some(handle) = self.get_handle(id) { return Some(v8::Local::new(scope, handle)); } }
None }}
#[cfg(test)]mod tests { use super::*; use crate::error::AnyError; use crate::Extension; use crate::JsRuntime; use crate::RuntimeOptions; use deno_ops::op; use futures::future::FutureExt; use parking_lot::Mutex; use std::fmt; use std::future::Future; use std::io; use std::path::PathBuf; use std::sync::atomic::{AtomicUsize, Ordering}; use std::sync::Arc; mod deno_core { pub use crate::*; }
use crate::runtime::tests::run_in_task;
#[derive(Default)] struct MockLoader { pub loads: Arc<Mutex<Vec<String>>>, }
impl MockLoader { fn new() -> Rc<Self> { Default::default() } }
fn mock_source_code(url: &str) -> Option<(&'static str, &'static str)> { const A_SRC: &str = r#"import { b } from "/b.js";import { c } from "/c.js";if (b() != 'b') throw Error();if (c() != 'c') throw Error();if (!import.meta.main) throw Error();if (import.meta.url != 'file:///a.js') throw Error();"#;
const B_SRC: &str = r#"import { c } from "/c.js";if (c() != 'c') throw Error();export function b() { return 'b'; }if (import.meta.main) throw Error();if (import.meta.url != 'file:///b.js') throw Error();"#;
const C_SRC: &str = r#"import { d } from "/d.js";export function c() { return 'c'; }if (d() != 'd') throw Error();if (import.meta.main) throw Error();if (import.meta.url != 'file:///c.js') throw Error();"#;
const D_SRC: &str = r#"export function d() { return 'd'; }if (import.meta.main) throw Error();if (import.meta.url != 'file:///d.js') throw Error();"#;
const CIRCULAR1_SRC: &str = r#"import "/circular2.js";Deno.core.print("circular1");"#;
const CIRCULAR2_SRC: &str = r#"import "/circular3.js";Deno.core.print("circular2");"#;
const CIRCULAR3_SRC: &str = r#"import "/circular1.js";import "/circular2.js";Deno.core.print("circular3");"#;
const REDIRECT1_SRC: &str = r#"import "./redirect2.js";Deno.core.print("redirect1");"#;
const REDIRECT2_SRC: &str = r#"import "./redirect3.js";Deno.core.print("redirect2");"#;
const REDIRECT3_SRC: &str = r#"Deno.core.print("redirect3");"#;
const MAIN_SRC: &str = r#"// never_ready.js never loads.import "/never_ready.js";// slow.js resolves after one tick.import "/slow.js";"#;
const SLOW_SRC: &str = r#"// Circular import of never_ready.js// Does this trigger two ModuleLoader calls? It shouldn't.import "/never_ready.js";import "/a.js";"#;
const BAD_IMPORT_SRC: &str = r#"import "foo";"#;
let spec: Vec<&str> = url.split("file://").collect(); match spec[1] { "/a.js" => Some((A_SRC, "file:///a.js")), "/b.js" => Some((B_SRC, "file:///b.js")), "/c.js" => Some((C_SRC, "file:///c.js")), "/d.js" => Some((D_SRC, "file:///d.js")), "/circular1.js" => Some((CIRCULAR1_SRC, "file:///circular1.js")), "/circular2.js" => Some((CIRCULAR2_SRC, "file:///circular2.js")), "/circular3.js" => Some((CIRCULAR3_SRC, "file:///circular3.js")), "/redirect1.js" => Some((REDIRECT1_SRC, "file:///redirect1.js")), "/redirect2.js" => Some((REDIRECT2_SRC, "file:///dir/redirect2.js")), "/dir/redirect3.js" => Some((REDIRECT3_SRC, "file:///redirect3.js")), "/slow.js" => Some((SLOW_SRC, "file:///slow.js")), "/never_ready.js" => { Some(("should never be Ready", "file:///never_ready.js")) } "/main.js" => Some((MAIN_SRC, "file:///main.js")), "/bad_import.js" => Some((BAD_IMPORT_SRC, "file:///bad_import.js")), "/main_with_code.js" => Some(("", "file:///main_with_code.js")), _ => None, } }
#[derive(Debug, PartialEq)] enum MockError { ResolveErr, LoadErr, }
impl fmt::Display for MockError { fn fmt(&self, _f: &mut fmt::Formatter) -> fmt::Result { unimplemented!() } }
impl std::error::Error for MockError { fn cause(&self) -> Option<&dyn std::error::Error> { unimplemented!() } }
struct DelayedSourceCodeFuture { url: String, counter: u32, }
impl Future for DelayedSourceCodeFuture { type Output = Result<ModuleSource, Error>;
fn poll(self: Pin<&mut Self>, cx: &mut Context) -> Poll<Self::Output> { let inner = self.get_mut(); inner.counter += 1; if inner.url == "file:///never_ready.js" { return Poll::Pending; } if inner.url == "file:///slow.js" && inner.counter < 2 { cx.waker().wake_by_ref(); return Poll::Pending; } match mock_source_code(&inner.url) { Some(src) => Poll::Ready(Ok(ModuleSource { code: src.0.as_bytes().to_vec().into_boxed_slice(), module_type: ModuleType::JavaScript, module_url_specified: inner.url.clone(), module_url_found: src.1.to_owned(), })), None => Poll::Ready(Err(MockError::LoadErr.into())), } } }
impl ModuleLoader for MockLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_root: bool, ) -> Result<ModuleSpecifier, Error> { let referrer = if referrer == "." { "file:///" } else { referrer };
let output_specifier = match resolve_import(specifier, referrer) { Ok(specifier) => specifier, Err(..) => return Err(MockError::ResolveErr.into()), };
if mock_source_code(&output_specifier.to_string()).is_some() { Ok(output_specifier) } else { Err(MockError::ResolveErr.into()) } }
fn load( &self, module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { let mut loads = self.loads.lock(); loads.push(module_specifier.to_string()); let url = module_specifier.to_string(); DelayedSourceCodeFuture { url, counter: 0 }.boxed() } }
#[test] fn test_recursive_load() { let loader = MockLoader::new(); let loads = loader.loads.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); let spec = resolve_url("file:///a.js").unwrap(); let a_id_fut = runtime.load_main_module(&spec, None); let a_id = futures::executor::block_on(a_id_fut).unwrap();
let _ = runtime.mod_evaluate(a_id); futures::executor::block_on(runtime.run_event_loop(false)).unwrap(); let l = loads.lock(); assert_eq!( l.to_vec(), vec![ "file:///a.js", "file:///b.js", "file:///c.js", "file:///d.js" ] );
let module_map_rc = JsRuntime::module_map(runtime.v8_isolate()); let modules = module_map_rc.borrow();
assert_eq!( modules.get_id("file:///a.js", AssertedModuleType::JavaScriptOrWasm), Some(a_id) ); let b_id = modules .get_id("file:///b.js", AssertedModuleType::JavaScriptOrWasm) .unwrap(); let c_id = modules .get_id("file:///c.js", AssertedModuleType::JavaScriptOrWasm) .unwrap(); let d_id = modules .get_id("file:///d.js", AssertedModuleType::JavaScriptOrWasm) .unwrap(); assert_eq!( modules.get_requested_modules(a_id), Some(&vec![ ModuleRequest { specifier: resolve_url("file:///b.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }, ModuleRequest { specifier: resolve_url("file:///c.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }, ]) ); assert_eq!( modules.get_requested_modules(b_id), Some(&vec![ModuleRequest { specifier: resolve_url("file:///c.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, },]) ); assert_eq!( modules.get_requested_modules(c_id), Some(&vec![ModuleRequest { specifier: resolve_url("file:///d.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, },]) ); assert_eq!(modules.get_requested_modules(d_id), Some(&vec![])); }
#[test] fn test_mods() { #[derive(Default)] struct ModsLoader { pub count: Arc<AtomicUsize>, }
impl ModuleLoader for ModsLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { self.count.fetch_add(1, Ordering::Relaxed); assert_eq!(specifier, "./b.js"); assert_eq!(referrer, "file:///a.js"); let s = resolve_import(specifier, referrer).unwrap(); Ok(s) }
fn load( &self, _module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { unreachable!() } }
let loader = Rc::new(ModsLoader::default());
let resolve_count = loader.count.clone(); static DISPATCH_COUNT: AtomicUsize = AtomicUsize::new(0);
#[op] fn op_test(_: &mut OpState, control: u8) -> Result<u8, AnyError> { DISPATCH_COUNT.fetch_add(1, Ordering::Relaxed); assert_eq!(control, 42); Ok(43) }
let ext = Extension::builder().ops(vec![op_test::decl()]).build();
let mut runtime = JsRuntime::new(RuntimeOptions { extensions: vec![ext], module_loader: Some(loader), ..Default::default() });
runtime .execute_script( "setup.js", r#" function assert(cond) { if (!cond) { throw Error("assert"); } } "#, ) .unwrap();
assert_eq!(DISPATCH_COUNT.load(Ordering::Relaxed), 0);
let module_map_rc = JsRuntime::module_map(runtime.v8_isolate());
let (mod_a, mod_b) = { let scope = &mut runtime.handle_scope(); let mut module_map = module_map_rc.borrow_mut(); let specifier_a = "file:///a.js".to_string(); let mod_a = module_map .new_es_module( scope, true, &specifier_a, br#" import { b } from './b.js' if (b() != 'b') throw Error(); let control = 42; Deno.core.opSync("op_test", control); "#, ) .unwrap();
assert_eq!(DISPATCH_COUNT.load(Ordering::Relaxed), 0); let imports = module_map.get_requested_modules(mod_a); assert_eq!( imports, Some(&vec![ModuleRequest { specifier: resolve_url("file:///b.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, },]) );
let mod_b = module_map .new_es_module( scope, false, "file:///b.js", b"export function b() { return 'b' }", ) .unwrap(); let imports = module_map.get_requested_modules(mod_b).unwrap(); assert_eq!(imports.len(), 0); (mod_a, mod_b) };
runtime.instantiate_module(mod_b).unwrap(); assert_eq!(DISPATCH_COUNT.load(Ordering::Relaxed), 0); assert_eq!(resolve_count.load(Ordering::SeqCst), 1);
runtime.instantiate_module(mod_a).unwrap(); assert_eq!(DISPATCH_COUNT.load(Ordering::Relaxed), 0);
let _ = runtime.mod_evaluate(mod_a); assert_eq!(DISPATCH_COUNT.load(Ordering::Relaxed), 1); }
#[test] fn test_json_module() { #[derive(Default)] struct ModsLoader { pub count: Arc<AtomicUsize>, }
impl ModuleLoader for ModsLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { self.count.fetch_add(1, Ordering::Relaxed); assert_eq!(specifier, "./b.json"); assert_eq!(referrer, "file:///a.js"); let s = resolve_import(specifier, referrer).unwrap(); Ok(s) }
fn load( &self, _module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { unreachable!() } }
let loader = Rc::new(ModsLoader::default());
let resolve_count = loader.count.clone();
let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() });
runtime .execute_script( "setup.js", r#" function assert(cond) { if (!cond) { throw Error("assert"); } } "#, ) .unwrap();
let module_map_rc = JsRuntime::module_map(runtime.v8_isolate());
let (mod_a, mod_b) = { let scope = &mut runtime.handle_scope(); let mut module_map = module_map_rc.borrow_mut(); let specifier_a = "file:///a.js".to_string(); let mod_a = module_map .new_es_module( scope, true, &specifier_a, br#" import jsonData from './b.json' assert {type: "json"}; assert(jsonData.a == "b"); assert(jsonData.c.d == 10); "#, ) .unwrap();
let imports = module_map.get_requested_modules(mod_a); assert_eq!( imports, Some(&vec![ModuleRequest { specifier: resolve_url("file:///b.json").unwrap(), asserted_module_type: AssertedModuleType::Json, },]) );
let mod_b = module_map .new_json_module( scope, "file:///b.json", b"{\"a\": \"b\", \"c\": {\"d\": 10}}", ) .unwrap(); let imports = module_map.get_requested_modules(mod_b).unwrap(); assert_eq!(imports.len(), 0); (mod_a, mod_b) };
runtime.instantiate_module(mod_b).unwrap(); assert_eq!(resolve_count.load(Ordering::SeqCst), 1);
runtime.instantiate_module(mod_a).unwrap();
let receiver = runtime.mod_evaluate(mod_a); futures::executor::block_on(runtime.run_event_loop(false)).unwrap(); futures::executor::block_on(receiver).unwrap().unwrap(); }
#[test] fn dyn_import_err() { #[derive(Clone, Default)] struct DynImportErrLoader { pub count: Arc<AtomicUsize>, }
impl ModuleLoader for DynImportErrLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { self.count.fetch_add(1, Ordering::Relaxed); assert_eq!(specifier, "/foo.js"); assert_eq!(referrer, "file:///dyn_import2.js"); let s = resolve_import(specifier, referrer).unwrap(); Ok(s) }
fn load( &self, _module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { async { Err(io::Error::from(io::ErrorKind::NotFound).into()) }.boxed() } }
run_in_task(|cx| { let loader = Rc::new(DynImportErrLoader::default()); let count = loader.count.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() });
runtime .execute_script( "file:///dyn_import2.js", r#" (async () => { await import("/foo.js"); })(); "#, ) .unwrap();
let result = runtime.poll_event_loop(cx, false); if let Poll::Ready(Ok(_)) = result { unreachable!(); } assert_eq!(count.load(Ordering::Relaxed), 4); }) }
#[derive(Clone, Default)] struct DynImportOkLoader { pub prepare_load_count: Arc<AtomicUsize>, pub resolve_count: Arc<AtomicUsize>, pub load_count: Arc<AtomicUsize>, }
impl ModuleLoader for DynImportOkLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { let c = self.resolve_count.fetch_add(1, Ordering::Relaxed); assert!(c < 7); assert_eq!(specifier, "./b.js"); assert_eq!(referrer, "file:///dyn_import3.js"); let s = resolve_import(specifier, referrer).unwrap(); Ok(s) }
fn load( &self, specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { self.load_count.fetch_add(1, Ordering::Relaxed); let info = ModuleSource { module_url_specified: specifier.to_string(), module_url_found: specifier.to_string(), code: b"export function b() { return 'b' }" .to_vec() .into_boxed_slice(), module_type: ModuleType::JavaScript, }; async move { Ok(info) }.boxed() }
fn prepare_load( &self, _op_state: Rc<RefCell<OpState>>, _module_specifier: &ModuleSpecifier, _maybe_referrer: Option<String>, _is_dyn_import: bool, ) -> Pin<Box<dyn Future<Output = Result<(), Error>>>> { self.prepare_load_count.fetch_add(1, Ordering::Relaxed); async { Ok(()) }.boxed_local() } }
#[test] fn dyn_import_ok() { run_in_task(|cx| { let loader = Rc::new(DynImportOkLoader::default()); let prepare_load_count = loader.prepare_load_count.clone(); let resolve_count = loader.resolve_count.clone(); let load_count = loader.load_count.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() });
runtime .execute_script( "file:///dyn_import3.js", r#" (async () => { let mod = await import("./b.js"); if (mod.b() !== 'b') { throw Error("bad1"); } // And again! mod = await import("./b.js"); if (mod.b() !== 'b') { throw Error("bad2"); } })(); "#, ) .unwrap();
assert!(matches!(runtime.poll_event_loop(cx, false), Poll::Pending)); assert_eq!(prepare_load_count.load(Ordering::Relaxed), 1);
assert!(matches!( runtime.poll_event_loop(cx, false), Poll::Ready(Ok(_)) )); assert_eq!(resolve_count.load(Ordering::Relaxed), 7); assert_eq!(load_count.load(Ordering::Relaxed), 1); assert!(matches!( runtime.poll_event_loop(cx, false), Poll::Ready(Ok(_)) )); assert_eq!(resolve_count.load(Ordering::Relaxed), 7); assert_eq!(load_count.load(Ordering::Relaxed), 1); }) }
#[test] fn dyn_import_borrow_mut_error() { run_in_task(|cx| { let loader = Rc::new(DynImportOkLoader::default()); let prepare_load_count = loader.prepare_load_count.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); runtime .execute_script( "file:///dyn_import3.js", r#" (async () => { let mod = await import("./b.js"); if (mod.b() !== 'b') { throw Error("bad"); } })(); "#, ) .unwrap(); let _ = runtime.poll_event_loop(cx, false); assert_eq!(prepare_load_count.load(Ordering::Relaxed), 1); let _ = runtime.poll_event_loop(cx, false); }) }
#[test] fn dyn_concurrent_circular_import() { #[derive(Clone, Default)] struct DynImportCircularLoader { pub resolve_count: Arc<AtomicUsize>, pub load_count: Arc<AtomicUsize>, }
impl ModuleLoader for DynImportCircularLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { self.resolve_count.fetch_add(1, Ordering::Relaxed); let s = resolve_import(specifier, referrer).unwrap(); Ok(s) }
fn load( &self, specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { self.load_count.fetch_add(1, Ordering::Relaxed); let filename = PathBuf::from(specifier.to_string()) .file_name() .unwrap() .to_string_lossy() .to_string(); let code = match filename.as_str() { "a.js" => "import './b.js';", "b.js" => "import './c.js';\nimport './a.js';", "c.js" => "import './d.js';", "d.js" => "// pass", _ => unreachable!(), }; let info = ModuleSource { module_url_specified: specifier.to_string(), module_url_found: specifier.to_string(), code: code.as_bytes().to_vec().into_boxed_slice(), module_type: ModuleType::JavaScript, }; async move { Ok(info) }.boxed() } }
let loader = Rc::new(DynImportCircularLoader::default()); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() });
runtime .execute_script( "file:///entry.js", "import('./b.js');\nimport('./a.js');", ) .unwrap();
let result = futures::executor::block_on(runtime.run_event_loop(false)); assert!(result.is_ok()); }
#[test] fn test_circular_load() { let loader = MockLoader::new(); let loads = loader.loads.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() });
let fut = async move { let spec = resolve_url("file:///circular1.js").unwrap(); let result = runtime.load_main_module(&spec, None).await; assert!(result.is_ok()); let circular1_id = result.unwrap(); let _ = runtime.mod_evaluate(circular1_id); runtime.run_event_loop(false).await.unwrap();
let l = loads.lock(); assert_eq!( l.to_vec(), vec![ "file:///circular1.js", "file:///circular2.js", "file:///circular3.js" ] );
let module_map_rc = JsRuntime::module_map(runtime.v8_isolate()); let modules = module_map_rc.borrow();
assert_eq!( modules .get_id("file:///circular1.js", AssertedModuleType::JavaScriptOrWasm), Some(circular1_id) ); let circular2_id = modules .get_id("file:///circular2.js", AssertedModuleType::JavaScriptOrWasm) .unwrap();
assert_eq!( modules.get_requested_modules(circular1_id), Some(&vec![ModuleRequest { specifier: resolve_url("file:///circular2.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }]) );
assert_eq!( modules.get_requested_modules(circular2_id), Some(&vec![ModuleRequest { specifier: resolve_url("file:///circular3.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }]) );
assert!(modules .get_id("file:///circular3.js", AssertedModuleType::JavaScriptOrWasm) .is_some()); let circular3_id = modules .get_id("file:///circular3.js", AssertedModuleType::JavaScriptOrWasm) .unwrap(); assert_eq!( modules.get_requested_modules(circular3_id), Some(&vec![ ModuleRequest { specifier: resolve_url("file:///circular1.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }, ModuleRequest { specifier: resolve_url("file:///circular2.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, } ]) ); } .boxed_local();
futures::executor::block_on(fut); }
#[test] fn test_redirect_load() { let loader = MockLoader::new(); let loads = loader.loads.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() });
let fut = async move { let spec = resolve_url("file:///redirect1.js").unwrap(); let result = runtime.load_main_module(&spec, None).await; assert!(result.is_ok()); let redirect1_id = result.unwrap(); let _ = runtime.mod_evaluate(redirect1_id); runtime.run_event_loop(false).await.unwrap(); let l = loads.lock(); assert_eq!( l.to_vec(), vec![ "file:///redirect1.js", "file:///redirect2.js", "file:///dir/redirect3.js" ] );
let module_map_rc = JsRuntime::module_map(runtime.v8_isolate()); let modules = module_map_rc.borrow();
assert_eq!( modules .get_id("file:///redirect1.js", AssertedModuleType::JavaScriptOrWasm), Some(redirect1_id) );
let redirect2_id = modules .get_id( "file:///dir/redirect2.js", AssertedModuleType::JavaScriptOrWasm, ) .unwrap(); assert!(modules.is_alias( "file:///redirect2.js", AssertedModuleType::JavaScriptOrWasm )); assert!(!modules.is_alias( "file:///dir/redirect2.js", AssertedModuleType::JavaScriptOrWasm )); assert_eq!( modules .get_id("file:///redirect2.js", AssertedModuleType::JavaScriptOrWasm), Some(redirect2_id) );
let redirect3_id = modules .get_id("file:///redirect3.js", AssertedModuleType::JavaScriptOrWasm) .unwrap(); assert!(modules.is_alias( "file:///dir/redirect3.js", AssertedModuleType::JavaScriptOrWasm )); assert!(!modules.is_alias( "file:///redirect3.js", AssertedModuleType::JavaScriptOrWasm )); assert_eq!( modules.get_id( "file:///dir/redirect3.js", AssertedModuleType::JavaScriptOrWasm ), Some(redirect3_id) ); } .boxed_local();
futures::executor::block_on(fut); }
#[test] fn slow_never_ready_modules() { run_in_task(|cx| { let loader = MockLoader::new(); let loads = loader.loads.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); let spec = resolve_url("file:///main.js").unwrap(); let mut recursive_load = runtime.load_main_module(&spec, None).boxed_local();
let result = recursive_load.poll_unpin(cx); assert!(result.is_pending());
for _ in 0..10 { let result = recursive_load.poll_unpin(cx); assert!(result.is_pending()); let l = loads.lock(); assert_eq!( l.to_vec(), vec![ "file:///main.js", "file:///never_ready.js", "file:///slow.js", "file:///a.js", "file:///b.js", "file:///c.js", "file:///d.js" ] ); } }) }
#[test] fn loader_disappears_after_error() { run_in_task(|cx| { let loader = MockLoader::new(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); let spec = resolve_url("file:///bad_import.js").unwrap(); let mut load_fut = runtime.load_main_module(&spec, None).boxed_local(); let result = load_fut.poll_unpin(cx); if let Poll::Ready(Err(err)) = result { assert_eq!( err.downcast_ref::<MockError>().unwrap(), &MockError::ResolveErr ); } else { unreachable!(); } }) }
#[test] fn recursive_load_main_with_code() { const MAIN_WITH_CODE_SRC: &str = r#"import { b } from "/b.js";import { c } from "/c.js";if (b() != 'b') throw Error();if (c() != 'c') throw Error();if (!import.meta.main) throw Error();if (import.meta.url != 'file:///main_with_code.js') throw Error();"#;
let loader = MockLoader::new(); let loads = loader.loads.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); let spec = resolve_url("file:///main_with_code.js").unwrap(); let main_id_fut = runtime .load_main_module(&spec, Some(MAIN_WITH_CODE_SRC.to_owned())) .boxed_local(); let main_id = futures::executor::block_on(main_id_fut).unwrap();
let _ = runtime.mod_evaluate(main_id); futures::executor::block_on(runtime.run_event_loop(false)).unwrap();
let l = loads.lock(); assert_eq!( l.to_vec(), vec!["file:///b.js", "file:///c.js", "file:///d.js"] );
let module_map_rc = JsRuntime::module_map(runtime.v8_isolate()); let modules = module_map_rc.borrow();
assert_eq!( modules.get_id( "file:///main_with_code.js", AssertedModuleType::JavaScriptOrWasm ), Some(main_id) ); let b_id = modules .get_id("file:///b.js", AssertedModuleType::JavaScriptOrWasm) .unwrap(); let c_id = modules .get_id("file:///c.js", AssertedModuleType::JavaScriptOrWasm) .unwrap(); let d_id = modules .get_id("file:///d.js", AssertedModuleType::JavaScriptOrWasm) .unwrap();
assert_eq!( modules.get_requested_modules(main_id), Some(&vec![ ModuleRequest { specifier: resolve_url("file:///b.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }, ModuleRequest { specifier: resolve_url("file:///c.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, } ]) ); assert_eq!( modules.get_requested_modules(b_id), Some(&vec![ModuleRequest { specifier: resolve_url("file:///c.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }]) ); assert_eq!( modules.get_requested_modules(c_id), Some(&vec![ModuleRequest { specifier: resolve_url("file:///d.js").unwrap(), asserted_module_type: AssertedModuleType::JavaScriptOrWasm, }]) ); assert_eq!(modules.get_requested_modules(d_id), Some(&vec![])); }
#[test] fn main_and_side_module() { struct ModsLoader {}
let main_specifier = resolve_url("file:///main_module.js").unwrap(); let side_specifier = resolve_url("file:///side_module.js").unwrap();
impl ModuleLoader for ModsLoader { fn resolve( &self, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, Error> { let s = resolve_import(specifier, referrer).unwrap(); Ok(s) }
fn load( &self, module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { let module_source = match module_specifier.as_str() { "file:///main_module.js" => Ok(ModuleSource { module_url_specified: "file:///main_module.js".to_string(), module_url_found: "file:///main_module.js".to_string(), code: b"if (!import.meta.main) throw Error();" .to_vec() .into_boxed_slice(), module_type: ModuleType::JavaScript, }), "file:///side_module.js" => Ok(ModuleSource { module_url_specified: "file:///side_module.js".to_string(), module_url_found: "file:///side_module.js".to_string(), code: b"if (import.meta.main) throw Error();" .to_vec() .into_boxed_slice(), module_type: ModuleType::JavaScript, }), _ => unreachable!(), }; async move { module_source }.boxed() } }
let loader = Rc::new(ModsLoader {}); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() });
let main_id_fut = runtime .load_main_module(&main_specifier, None) .boxed_local(); let main_id = futures::executor::block_on(main_id_fut).unwrap();
let _ = runtime.mod_evaluate(main_id); futures::executor::block_on(runtime.run_event_loop(false)).unwrap();
let side_id_fut = runtime .load_main_module(&side_specifier, None) .boxed_local(); futures::executor::block_on(side_id_fut).unwrap_err();
let side_id_fut = runtime .load_side_module(&side_specifier, None) .boxed_local(); let side_id = futures::executor::block_on(side_id_fut).unwrap();
let _ = runtime.mod_evaluate(side_id); futures::executor::block_on(runtime.run_event_loop(false)).unwrap(); }}