use std::rc::Rc;
use dom_struct::dom_struct;
use js::gc::MutableHandleValue;
use js::jsapi::Heap;
use js::jsval::UndefinedValue;
use js::rust::HandleObject;
use profile_traits::mem::MemoryReportResult;
use script_bindings::conversions::SafeToJSValConvertible;
use script_bindings::error::{Error, Fallible};
use script_bindings::interfaces::ServoInternalsHelpers;
use script_bindings::script_runtime::JSContext;
use script_bindings::str::USVString;
use servo_config::prefs::{self, PrefValue, Preferences};
use servo_constellation_traits::ScriptToConstellationMessage;
use crate::dom::bindings::codegen::Bindings::ServoInternalsBinding::ServoInternalsMethods;
use crate::dom::bindings::import::base::SafeJSContext;
use crate::dom::bindings::reflector::{DomGlobal, Reflector, reflect_dom_object};
use crate::dom::bindings::root::DomRoot;
use crate::dom::globalscope::GlobalScope;
use crate::dom::promise::Promise;
use crate::realms::{AlreadyInRealm, InRealm};
use crate::routed_promise::{RoutedPromiseListener, callback_promise};
use crate::script_runtime::CanGc;
use crate::script_thread::ScriptThread;
fn pref_to_jsval(pref: &PrefValue, cx: JSContext, rval: MutableHandleValue, can_gc: CanGc) {
match pref {
PrefValue::Bool(b) => b.safe_to_jsval(cx, rval, can_gc),
PrefValue::Int(i) => i.safe_to_jsval(cx, rval, can_gc),
PrefValue::UInt(u) => u.safe_to_jsval(cx, rval, can_gc),
PrefValue::Str(s) => s.safe_to_jsval(cx, rval, can_gc),
PrefValue::Float(f) => f.safe_to_jsval(cx, rval, can_gc),
PrefValue::Array(arr) => {
rooted_vec!(let mut js_arr);
for item in arr {
rooted!(in(*cx) let mut js_val = UndefinedValue());
pref_to_jsval(item, cx, js_val.handle_mut(), can_gc);
js_arr.push(Heap::boxed(js_val.get()));
}
js_arr.safe_to_jsval(cx, rval, can_gc);
},
}
}
#[dom_struct]
pub(crate) struct ServoInternals {
reflector_: Reflector,
}
impl ServoInternals {
pub fn new_inherited() -> ServoInternals {
ServoInternals {
reflector_: Reflector::new(),
}
}
pub(crate) fn new(global: &GlobalScope, can_gc: CanGc) -> DomRoot<ServoInternals> {
reflect_dom_object(Box::new(ServoInternals::new_inherited()), global, can_gc)
}
}
impl ServoInternalsMethods<crate::DomTypeHolder> for ServoInternals {
fn ReportMemory(&self, comp: InRealm, can_gc: CanGc) -> Rc<Promise> {
let global = &self.global();
let promise = Promise::new_in_current_realm(comp, can_gc);
let task_source = global.task_manager().dom_manipulation_task_source();
let callback = callback_promise(&promise, self, task_source);
let script_to_constellation_chan = global.script_to_constellation_chan();
if script_to_constellation_chan
.send(ScriptToConstellationMessage::ReportMemory(callback))
.is_err()
{
promise.reject_error(Error::Operation(None), can_gc);
}
promise
}
fn GarbageCollectAllContexts(&self) {
let global = &self.global();
let script_to_constellation_chan = global.script_to_constellation_chan();
let _ = script_to_constellation_chan
.send(ScriptToConstellationMessage::TriggerGarbageCollection);
}
fn PreferenceList(&self) -> Vec<USVString> {
Preferences::all_fields()
.into_iter()
.map(|s| USVString::from(s.to_string()))
.collect()
}
fn PreferenceType(&self, name: USVString) -> Fallible<USVString> {
if !Preferences::exists(&name) {
return Err(Error::NotFound(None));
}
let type_name = Preferences::type_of(&name).split("::").last().unwrap();
Ok(USVString::from(type_name.to_string()))
}
fn DefaultPreferenceValue(
&self,
cx: SafeJSContext,
name: USVString,
rval: MutableHandleValue,
) -> Fallible<()> {
if !Preferences::exists(&name) {
return Err(Error::NotFound(None));
}
let pref = Preferences::default().get_value(&name);
pref_to_jsval(&pref, cx, rval, CanGc::note());
Ok(())
}
fn GetPreference(
&self,
cx: JSContext,
name: USVString,
rval: MutableHandleValue,
) -> Fallible<()> {
if !Preferences::exists(&name) {
return Err(Error::NotFound(None));
}
let pref = prefs::get().get_value(&name);
pref_to_jsval(&pref, cx, rval, CanGc::note());
Ok(())
}
fn GetBoolPreference(&self, name: USVString) -> Fallible<bool> {
if !Preferences::exists(&name) {
return Err(Error::NotFound(None));
}
if let PrefValue::Bool(b) = prefs::get().get_value(&name) {
return Ok(b);
}
Err(Error::TypeMismatch(None))
}
fn GetIntPreference(&self, name: USVString) -> Fallible<i64> {
if !Preferences::exists(&name) {
return Err(Error::NotFound(None));
}
if let PrefValue::Int(i) = prefs::get().get_value(&name) {
return Ok(i);
}
Err(Error::TypeMismatch(None))
}
fn GetStringPreference(&self, name: USVString) -> Fallible<USVString> {
if !Preferences::exists(&name) {
return Err(Error::NotFound(None));
}
if let PrefValue::Str(s) = prefs::get().get_value(&name) {
return Ok(s.into());
}
Err(Error::TypeMismatch(None))
}
fn SetBoolPreference(&self, name: USVString, value: bool) {
let mut current_prefs = prefs::get().clone();
current_prefs.set_value(&name, value.into());
prefs::set(current_prefs);
}
fn SetIntPreference(&self, name: USVString, value: i64) {
let mut current_prefs = prefs::get().clone();
current_prefs.set_value(&name, value.into());
prefs::set(current_prefs);
}
fn SetStringPreference(&self, name: USVString, value: USVString) {
let mut current_prefs = prefs::get().clone();
current_prefs.set_value(&name, value.0.into());
prefs::set(current_prefs);
}
}
impl RoutedPromiseListener<MemoryReportResult> for ServoInternals {
fn handle_response(&self, response: MemoryReportResult, promise: &Rc<Promise>, can_gc: CanGc) {
let stringified = serde_json::to_string(&response.results)
.unwrap_or_else(|_| "{ error: \"failed to create memory report\"}".to_owned());
promise.resolve_native(&stringified, can_gc);
}
}
impl ServoInternalsHelpers for ServoInternals {
#[expect(unsafe_code)]
fn is_servo_internal(cx: JSContext, _global: HandleObject) -> bool {
unsafe {
let in_realm_proof = AlreadyInRealm::assert_for_cx(cx);
let global_scope = GlobalScope::from_context(*cx, InRealm::Already(&in_realm_proof));
let url = global_scope.get_url();
(url.scheme() == "about" && url.as_str() != "about:blank") ||
ScriptThread::is_servo_privileged(url)
}
}
}