use std::thread;
use js::jsapi::{HideScriptedCaller, UnhideScriptedCaller};
use js::rust::Runtime;
use crate::DomTypes;
use crate::interfaces::{DomHelpers, GlobalScopeHelpers};
use crate::root::Dom;
use crate::script_runtime::temp_cx;
#[derive(Debug, Eq, JSTraceable, PartialEq)]
pub enum StackEntryKind {
Incumbent,
Entry,
}
#[cfg_attr(crown, expect(crown::unrooted_must_root))]
#[derive(JSTraceable)]
pub struct StackEntry<D: DomTypes> {
pub global: Dom<D::GlobalScope>,
pub kind: StackEntryKind,
}
pub fn run_a_script<D: DomTypes, R>(global: &D::GlobalScope, f: impl FnOnce() -> R) -> R {
let settings_stack = <D as DomHelpers<D>>::settings_stack();
let _span = settings_stack.with(|stack| {
trace!("Prepare to run script with {:p}", global);
let mut stack = stack.borrow_mut();
stack.push(StackEntry {
global: Dom::from_ref(global),
kind: StackEntryKind::Entry,
});
profile_traits::info_span!("ScriptEvaluate", url = global.get_url().to_string()).entered()
});
let result = f();
let stack_is_empty = settings_stack.with(|stack| {
let mut stack = stack.borrow_mut();
let entry = stack.pop().unwrap();
assert_eq!(
&*entry.global as *const D::GlobalScope, global as *const D::GlobalScope,
"Dropped AutoEntryScript out of order."
);
assert_eq!(entry.kind, StackEntryKind::Entry);
trace!("Clean up after running script with {:p}", &*entry.global);
stack.is_empty()
});
if !thread::panicking() && stack_is_empty {
let mut cx = unsafe { temp_cx() };
global.perform_a_microtask_checkpoint(&mut cx);
}
result
}
pub fn run_a_callback<D: DomTypes, R>(global: &D::GlobalScope, f: impl FnOnce() -> R) -> R {
let settings_stack = <D as DomHelpers<D>>::settings_stack();
let cx = Runtime::get().expect("Creating a new incumbent script after runtime shutdown");
unsafe {
HideScriptedCaller(cx.as_ptr());
}
settings_stack.with(|stack| {
trace!("Prepare to run a callback with {:p}", global);
let mut stack = stack.borrow_mut();
stack.push(StackEntry {
global: Dom::from_ref(global),
kind: StackEntryKind::Incumbent,
});
});
let result = f();
settings_stack.with(|stack| {
let mut stack = stack.borrow_mut();
let entry = stack.pop().unwrap();
assert_eq!(
&*entry.global as *const D::GlobalScope as usize,
global as *const D::GlobalScope as usize,
"Dropped AutoIncumbentScript out of order."
);
assert_eq!(entry.kind, StackEntryKind::Incumbent);
trace!(
"Clean up after running a callback with {:p}",
&*entry.global
);
});
unsafe {
UnhideScriptedCaller(cx.as_ptr());
}
result
}