Skip to main content
Module

x/deno/core/modules.rs

A modern runtime for JavaScript and TypeScript.
Go to Latest
File
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022
// Copyright 2018-2021 the Deno authors. All rights reserved. MIT license.
use rusty_v8 as v8;
use crate::error::generic_error;use crate::error::AnyError;use crate::module_specifier::ModuleSpecifier;use crate::OpState;use futures::future::FutureExt;use futures::stream::FuturesUnordered;use futures::stream::Stream;use futures::stream::TryStreamExt;use std::cell::RefCell;use std::collections::HashMap;use std::collections::HashSet;use std::future::Future;use std::pin::Pin;use std::rc::Rc;use std::sync::atomic::AtomicI32;use std::sync::atomic::Ordering;use std::task::Context;use std::task::Poll;
lazy_static::lazy_static! { pub static ref NEXT_LOAD_ID: AtomicI32 = AtomicI32::new(0);}
pub type ModuleId = i32;pub type ModuleLoadId = i32;
/// EsModule source code that will be loaded into V8.////// Users can implement `Into<ModuleInfo>` for different file types that/// can be transpiled to valid EsModule.////// Found module URL might be different from specified URL/// used for loading due to redirections (like HTTP 303)./// Eg. Both "https://example.com/a.ts" and/// "https://example.com/b.ts" may point to "https://example.com/c.ts"/// By keeping track of specified and found URL we can alias modules and avoid/// recompiling the same code 3 times.// TODO(bartlomieju): I have a strong opinion we should store all redirects// that happened; not only first and final target. It would simplify a lot// of things throughout the codebase otherwise we may end up requesting// intermediate redirects from file loader.#[derive(Debug, Clone, Eq, PartialEq)]pub struct ModuleSource { pub code: String, pub module_url_specified: String, pub module_url_found: String,}
pub type PrepareLoadFuture = dyn Future<Output = (ModuleLoadId, Result<RecursiveModuleLoad, AnyError>)>;pub type ModuleSourceFuture = dyn Future<Output = Result<ModuleSource, AnyError>>;
pub trait ModuleLoader { /// Returns an absolute URL. /// When implementing an spec-complaint VM, this should be exactly the /// algorithm described here: /// https://html.spec.whatwg.org/multipage/webappapis.html#resolve-a-module-specifier /// /// `is_main` can be used to resolve from current working directory or /// apply import map for child imports. fn resolve( &self, op_state: Rc<RefCell<OpState>>, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, AnyError>;
/// Given ModuleSpecifier, load its source code. /// /// `is_dyn_import` can be used to check permissions or deny /// dynamic imports altogether. fn load( &self, op_state: Rc<RefCell<OpState>>, module_specifier: &ModuleSpecifier, maybe_referrer: Option<ModuleSpecifier>, is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>>;
/// This hook can be used by implementors to do some preparation /// work before starting loading of modules. /// /// For example implementor might download multiple modules in /// parallel and transpile them to final JS sources before /// yielding control back to the runtime. /// /// It's not required to implement this method. fn prepare_load( &self, _op_state: Rc<RefCell<OpState>>, _load_id: ModuleLoadId, _module_specifier: &ModuleSpecifier, _maybe_referrer: Option<String>, _is_dyn_import: bool, ) -> Pin<Box<dyn Future<Output = Result<(), AnyError>>>> { async { Ok(()) }.boxed_local() }}
/// Placeholder structure used when creating/// a runtime that doesn't support module loading.pub struct NoopModuleLoader;
impl ModuleLoader for NoopModuleLoader { fn resolve( &self, _op_state: Rc<RefCell<OpState>>, _specifier: &str, _referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, AnyError> { Err(generic_error("Module loading is not supported")) }
fn load( &self, _op_state: Rc<RefCell<OpState>>, _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() }}
/// Basic file system module loader.////// Note that this loader will **block** event loop/// when loading file as it uses synchronous FS API/// from standard library.pub struct FsModuleLoader;
impl ModuleLoader for FsModuleLoader { fn resolve( &self, _op_state: Rc<RefCell<OpState>>, specifier: &str, referrer: &str, _is_main: bool, ) -> Result<ModuleSpecifier, AnyError> { Ok(crate::resolve_import(specifier, referrer)?) }
fn load( &self, _op_state: Rc<RefCell<OpState>>, 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 code = std::fs::read_to_string(path)?; let module = ModuleSource { code, module_url_specified: module_specifier.to_string(), module_url_found: module_specifier.to_string(), }; Ok(module) } .boxed_local() }}
#[derive(Debug, Eq, PartialEq)]enum Kind { Main, DynamicImport,}
#[derive(Debug, Eq, PartialEq)]pub enum LoadState { ResolveMain(String, Option<String>), ResolveImport(String, String), LoadingRoot, LoadingImports, Done,}
/// This future is used to implement parallel async module loading.pub struct RecursiveModuleLoad { op_state: Rc<RefCell<OpState>>, kind: Kind, // TODO(bartlomieju): in future this value should // be randomized pub id: ModuleLoadId, pub root_module_id: Option<ModuleId>, pub state: LoadState, pub loader: Rc<dyn ModuleLoader>, pub pending: FuturesUnordered<Pin<Box<ModuleSourceFuture>>>, pub is_pending: HashSet<ModuleSpecifier>,}
impl RecursiveModuleLoad { /// Starts a new parallel load of the given URL of the main module. pub fn main( op_state: Rc<RefCell<OpState>>, specifier: &str, code: Option<String>, loader: Rc<dyn ModuleLoader>, ) -> Self { let kind = Kind::Main; let state = LoadState::ResolveMain(specifier.to_owned(), code); Self::new(op_state, kind, state, loader) }
pub fn dynamic_import( op_state: Rc<RefCell<OpState>>, specifier: &str, referrer: &str, loader: Rc<dyn ModuleLoader>, ) -> Self { let kind = Kind::DynamicImport; let state = LoadState::ResolveImport(specifier.to_owned(), referrer.to_owned()); Self::new(op_state, kind, state, loader) }
pub fn is_dynamic_import(&self) -> bool { self.kind != Kind::Main }
fn new( op_state: Rc<RefCell<OpState>>, kind: Kind, state: LoadState, loader: Rc<dyn ModuleLoader>, ) -> Self { Self { id: NEXT_LOAD_ID.fetch_add(1, Ordering::SeqCst), root_module_id: None, op_state, kind, state, loader, pending: FuturesUnordered::new(), is_pending: HashSet::new(), } }
pub async fn prepare(self) -> (ModuleLoadId, Result<Self, AnyError>) { let (module_specifier, maybe_referrer) = match self.state { LoadState::ResolveMain(ref specifier, _) => { let spec = match self .loader .resolve(self.op_state.clone(), specifier, ".", true) { Ok(spec) => spec, Err(e) => return (self.id, Err(e)), }; (spec, None) } LoadState::ResolveImport(ref specifier, ref referrer) => { let spec = match self.loader.resolve( self.op_state.clone(), specifier, referrer, false, ) { Ok(spec) => spec, Err(e) => return (self.id, Err(e)), }; (spec, Some(referrer.to_string())) } _ => unreachable!(), };
let prepare_result = self .loader .prepare_load( self.op_state.clone(), self.id, &module_specifier, maybe_referrer, self.is_dynamic_import(), ) .await;
match prepare_result { Ok(()) => (self.id, Ok(self)), Err(e) => (self.id, Err(e)), } }
fn add_root(&mut self) -> Result<(), AnyError> { let module_specifier = match self.state { LoadState::ResolveMain(ref specifier, _) => { self .loader .resolve(self.op_state.clone(), specifier, ".", true)? } LoadState::ResolveImport(ref specifier, ref referrer) => self .loader .resolve(self.op_state.clone(), specifier, referrer, false)?,
_ => unreachable!(), };
let load_fut = match &self.state { LoadState::ResolveMain(_, Some(code)) => { futures::future::ok(ModuleSource { code: code.to_owned(), module_url_specified: module_specifier.to_string(), module_url_found: module_specifier.to_string(), }) .boxed() } _ => self .loader .load( self.op_state.clone(), &module_specifier, None, self.is_dynamic_import(), ) .boxed_local(), };
self.pending.push(load_fut);
self.state = LoadState::LoadingRoot; Ok(()) }
pub fn add_import( &mut self, specifier: ModuleSpecifier, referrer: ModuleSpecifier, ) { if !self.is_pending.contains(&specifier) { let fut = self.loader.load( self.op_state.clone(), &specifier, Some(referrer), self.is_dynamic_import(), ); self.pending.push(fut.boxed_local()); self.is_pending.insert(specifier); } }}
impl Stream for RecursiveModuleLoad { type Item = Result<ModuleSource, AnyError>;
fn poll_next( self: Pin<&mut Self>, cx: &mut Context, ) -> Poll<Option<Self::Item>> { let inner = self.get_mut(); match inner.state { LoadState::ResolveMain(..) | LoadState::ResolveImport(..) => { if let Err(e) = inner.add_root() { return Poll::Ready(Some(Err(e))); } 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), } }}
pub struct ModuleInfo { pub id: ModuleId, pub main: bool, pub name: String, pub import_specifiers: Vec<ModuleSpecifier>,}
/// A symbolic module entity.enum SymbolicModule { /// This module is an alias to another module. /// This is useful such that multiple names could point to /// the same underlying module (particularly due to redirects). Alias(String), /// This module associates with a V8 module by id. Mod(ModuleId),}
/// A collection of JS modules.#[derive(Default)]pub 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, SymbolicModule>, next_module_id: ModuleId,}
impl ModuleMap { pub fn new() -> ModuleMap { Self { handles_by_id: HashMap::new(), ids_by_handle: HashMap::new(), info: HashMap::new(), by_name: HashMap::new(), next_module_id: 1, } }
/// Get module id, following all aliases in case of module specifier /// that had been redirected. pub fn get_id(&self, name: &str) -> Option<ModuleId> { let mut mod_name = name; loop { let symbolic_module = self.by_name.get(mod_name)?; match symbolic_module { SymbolicModule::Alias(target) => { mod_name = target; } SymbolicModule::Mod(mod_id) => return Some(*mod_id), } } }
pub fn get_children(&self, id: ModuleId) -> Option<&Vec<ModuleSpecifier>> { self.info.get(&id).map(|i| &i.import_specifiers) }
pub fn is_registered(&self, specifier: &ModuleSpecifier) -> bool { self.get_id(&specifier.to_string()).is_some() }
pub fn register( &mut self, name: &str, main: bool, handle: v8::Global<v8::Module>, import_specifiers: Vec<ModuleSpecifier>, ) -> ModuleId { let name = String::from(name); let id = self.next_module_id; self.next_module_id += 1; self .by_name .insert(name.to_string(), 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, import_specifiers, }, ); id }
pub fn alias(&mut self, name: &str, target: &str) { self .by_name .insert(name.to_string(), SymbolicModule::Alias(target.to_string())); }
#[cfg(test)] pub fn is_alias(&self, name: &str) -> bool { let cond = self.by_name.get(name); matches!(cond, Some(SymbolicModule::Alias(_))) }
pub fn get_handle(&self, id: ModuleId) -> Option<v8::Global<v8::Module>> { self.handles_by_id.get(&id).cloned() }
pub 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 fn get_info_by_id(&self, id: &ModuleId) -> Option<&ModuleInfo> { self.info.get(id) }}
#[cfg(test)]mod tests { use super::*; use crate::JsRuntime; use crate::RuntimeOptions; use futures::future::FutureExt; use std::error::Error; use std::fmt; use std::future::Future; use std::sync::Arc; use std::sync::Mutex;
// TODO(ry) Sadly FuturesUnordered requires the current task to be set. So // even though we are only using poll() in these tests and not Tokio, we must // nevertheless run it in the tokio executor. Ideally run_in_task can be // removed in the future. 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)> { // (code, real_module_name) 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")), // pretend redirect - real module name is different than one requested "/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")), // deliberately empty code. "/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 Error for MockError { fn cause(&self) -> Option<&dyn Error> { unimplemented!() } }
struct DelayedSourceCodeFuture { url: String, counter: u32, }
impl Future for DelayedSourceCodeFuture { type Output = Result<ModuleSource, AnyError>;
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 { // TODO(ry) Hopefully in the future we can remove current task // notification. See comment above run_in_task. cx.waker().wake_by_ref(); return Poll::Pending; } match mock_source_code(&inner.url) { Some(src) => Poll::Ready(Ok(ModuleSource { code: src.0.to_owned(), 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, _op_state: Rc<RefCell<OpState>>, specifier: &str, referrer: &str, _is_root: bool, ) -> Result<ModuleSpecifier, AnyError> { let referrer = if referrer == "." { "file:///" } else { referrer };
eprintln!(">> RESOLVING, S: {}, R: {}", specifier, referrer);
let output_specifier = match crate::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, _op_state: Rc<RefCell<OpState>>, module_specifier: &ModuleSpecifier, _maybe_referrer: Option<ModuleSpecifier>, _is_dyn_import: bool, ) -> Pin<Box<ModuleSourceFuture>> { let mut loads = self.loads.lock().unwrap(); loads.push(module_specifier.to_string()); let url = module_specifier.to_string(); DelayedSourceCodeFuture { url, counter: 0 }.boxed() } }
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(); "#;
#[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 = crate::resolve_url("file:///a.js").unwrap(); let a_id_fut = runtime.load_module(&spec, None); let a_id = futures::executor::block_on(a_id_fut).expect("Failed to load");
runtime.mod_evaluate(a_id); futures::executor::block_on(runtime.run_event_loop()).unwrap(); let l = loads.lock().unwrap(); assert_eq!( l.to_vec(), vec![ "file:///a.js", "file:///b.js", "file:///c.js", "file:///d.js" ] );
let state_rc = JsRuntime::state(runtime.v8_isolate()); let state = state_rc.borrow(); let modules = &state.module_map; assert_eq!(modules.get_id("file:///a.js"), Some(a_id)); let b_id = modules.get_id("file:///b.js").unwrap(); let c_id = modules.get_id("file:///c.js").unwrap(); let d_id = modules.get_id("file:///d.js").unwrap(); assert_eq!( modules.get_children(a_id), Some(&vec![ crate::resolve_url("file:///b.js").unwrap(), crate::resolve_url("file:///c.js").unwrap() ]) ); assert_eq!( modules.get_children(b_id), Some(&vec![crate::resolve_url("file:///c.js").unwrap()]) ); assert_eq!( modules.get_children(c_id), Some(&vec![crate::resolve_url("file:///d.js").unwrap()]) ); assert_eq!(modules.get_children(d_id), Some(&vec![])); }
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"); "#;
#[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 = crate::resolve_url("file:///circular1.js").unwrap(); let result = runtime.load_module(&spec, None).await; assert!(result.is_ok()); let circular1_id = result.unwrap(); runtime.mod_evaluate(circular1_id); runtime.run_event_loop().await.unwrap();
let l = loads.lock().unwrap(); assert_eq!( l.to_vec(), vec![ "file:///circular1.js", "file:///circular2.js", "file:///circular3.js" ] );
let state_rc = JsRuntime::state(runtime.v8_isolate()); let state = state_rc.borrow(); let modules = &state.module_map;
assert_eq!(modules.get_id("file:///circular1.js"), Some(circular1_id)); let circular2_id = modules.get_id("file:///circular2.js").unwrap();
assert_eq!( modules.get_children(circular1_id), Some(&vec![crate::resolve_url("file:///circular2.js").unwrap()]) );
assert_eq!( modules.get_children(circular2_id), Some(&vec![crate::resolve_url("file:///circular3.js").unwrap()]) );
assert!(modules.get_id("file:///circular3.js").is_some()); let circular3_id = modules.get_id("file:///circular3.js").unwrap(); assert_eq!( modules.get_children(circular3_id), Some(&vec![ crate::resolve_url("file:///circular1.js").unwrap(), crate::resolve_url("file:///circular2.js").unwrap() ]) ); } .boxed_local();
futures::executor::block_on(fut); }
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"); "#;
#[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 = crate::resolve_url("file:///redirect1.js").unwrap(); let result = runtime.load_module(&spec, None).await; println!(">> result {:?}", result); assert!(result.is_ok()); let redirect1_id = result.unwrap(); runtime.mod_evaluate(redirect1_id); runtime.run_event_loop().await.unwrap(); let l = loads.lock().unwrap(); assert_eq!( l.to_vec(), vec![ "file:///redirect1.js", "file:///redirect2.js", "file:///dir/redirect3.js" ] );
let state_rc = JsRuntime::state(runtime.v8_isolate()); let state = state_rc.borrow(); let modules = &state.module_map;
assert_eq!(modules.get_id("file:///redirect1.js"), Some(redirect1_id));
let redirect2_id = modules.get_id("file:///dir/redirect2.js").unwrap(); assert!(modules.is_alias("file:///redirect2.js")); assert!(!modules.is_alias("file:///dir/redirect2.js")); assert_eq!(modules.get_id("file:///redirect2.js"), Some(redirect2_id));
let redirect3_id = modules.get_id("file:///redirect3.js").unwrap(); assert!(modules.is_alias("file:///dir/redirect3.js")); assert!(!modules.is_alias("file:///redirect3.js")); assert_eq!( modules.get_id("file:///dir/redirect3.js"), Some(redirect3_id) ); } .boxed_local();
futures::executor::block_on(fut); }
// main.js const MAIN_SRC: &str = r#" // never_ready.js never loads. import "/never_ready.js"; // slow.js resolves after one tick. import "/slow.js"; "#;
// 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"; "#;
#[test] fn slow_never_ready_modules() { run_in_task(|mut cx| { let loader = MockLoader::new(); let loads = loader.loads.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); let spec = crate::resolve_url("file:///main.js").unwrap(); let mut recursive_load = runtime.load_module(&spec, None).boxed_local();
let result = recursive_load.poll_unpin(&mut cx); assert!(result.is_pending());
// TODO(ry) Arguably the first time we poll only the following modules // should be loaded: // "file:///main.js", // "file:///never_ready.js", // "file:///slow.js" // But due to current task notification in DelayedSourceCodeFuture they // all get loaded in a single poll. Also see the comment above // run_in_task.
for _ in 0..10 { let result = recursive_load.poll_unpin(&mut cx); assert!(result.is_pending()); let l = loads.lock().unwrap(); 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" ] ); } }) }
// bad_import.js const BAD_IMPORT_SRC: &str = r#" import "foo"; "#;
#[test] fn loader_disappears_after_error() { run_in_task(|mut cx| { let loader = MockLoader::new(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); let spec = crate::resolve_url("file:///bad_import.js").unwrap(); let mut load_fut = runtime.load_module(&spec, None).boxed_local(); let result = load_fut.poll_unpin(&mut cx); if let Poll::Ready(Err(err)) = result { assert_eq!( err.downcast_ref::<MockError>().unwrap(), &MockError::ResolveErr ); } else { unreachable!(); } }) }
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(); "#;
#[test] fn recursive_load_main_with_code() { let loader = MockLoader::new(); let loads = loader.loads.clone(); let mut runtime = JsRuntime::new(RuntimeOptions { module_loader: Some(loader), ..Default::default() }); // In default resolution code should be empty. // Instead we explicitly pass in our own code. // The behavior should be very similar to /a.js. let spec = crate::resolve_url("file:///main_with_code.js").unwrap(); let main_id_fut = runtime .load_module(&spec, Some(MAIN_WITH_CODE_SRC.to_owned())) .boxed_local(); let main_id = futures::executor::block_on(main_id_fut).expect("Failed to load");
runtime.mod_evaluate(main_id); futures::executor::block_on(runtime.run_event_loop()).unwrap();
let l = loads.lock().unwrap(); assert_eq!( l.to_vec(), vec!["file:///b.js", "file:///c.js", "file:///d.js"] );
let state_rc = JsRuntime::state(runtime.v8_isolate()); let state = state_rc.borrow(); let modules = &state.module_map;
assert_eq!(modules.get_id("file:///main_with_code.js"), Some(main_id)); let b_id = modules.get_id("file:///b.js").unwrap(); let c_id = modules.get_id("file:///c.js").unwrap(); let d_id = modules.get_id("file:///d.js").unwrap();
assert_eq!( modules.get_children(main_id), Some(&vec![ crate::resolve_url("file:///b.js").unwrap(), crate::resolve_url("file:///c.js").unwrap() ]) ); assert_eq!( modules.get_children(b_id), Some(&vec![crate::resolve_url("file:///c.js").unwrap()]) ); assert_eq!( modules.get_children(c_id), Some(&vec![crate::resolve_url("file:///d.js").unwrap()]) ); assert_eq!(modules.get_children(d_id), Some(&vec![])); }}