Skip to main content

script/dom/testing/
testworklet.rs

1/* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
4
5// check-tidy: no specs after this line
6use std::rc::Rc;
7
8use dom_struct::dom_struct;
9use js::context::JSContext;
10use js::realm::CurrentRealm;
11use js::rust::HandleObject;
12use script_bindings::reflector::{Reflector, reflect_dom_object_with_proto_and_cx};
13
14use crate::dom::bindings::codegen::Bindings::TestWorkletBinding::TestWorkletMethods;
15use crate::dom::bindings::codegen::Bindings::WorkletBinding::Worklet_Binding::WorkletMethods;
16use crate::dom::bindings::codegen::Bindings::WorkletBinding::WorkletOptions;
17use crate::dom::bindings::error::Fallible;
18use crate::dom::bindings::reflector::DomGlobal;
19use crate::dom::bindings::root::{Dom, DomRoot};
20use crate::dom::bindings::str::{DOMString, USVString};
21use crate::dom::promise::Promise;
22use crate::dom::window::Window;
23use crate::dom::worklet::Worklet;
24use crate::dom::workletglobalscope::WorkletGlobalScopeType;
25use crate::script_thread::ScriptThread;
26
27#[dom_struct]
28pub(crate) struct TestWorklet {
29    reflector: Reflector,
30    worklet: Dom<Worklet>,
31}
32
33impl TestWorklet {
34    fn new_inherited(worklet: &Worklet) -> TestWorklet {
35        TestWorklet {
36            reflector: Reflector::new(),
37            worklet: Dom::from_ref(worklet),
38        }
39    }
40
41    fn new(
42        cx: &mut JSContext,
43        window: &Window,
44        proto: Option<HandleObject>,
45    ) -> DomRoot<TestWorklet> {
46        let worklet = Worklet::new(cx, window, WorkletGlobalScopeType::Test);
47        reflect_dom_object_with_proto_and_cx(
48            Box::new(TestWorklet::new_inherited(&worklet)),
49            window,
50            proto,
51            cx,
52        )
53    }
54}
55
56impl TestWorkletMethods<crate::DomTypeHolder> for TestWorklet {
57    fn Constructor(
58        cx: &mut JSContext,
59        window: &Window,
60        proto: Option<HandleObject>,
61    ) -> Fallible<DomRoot<TestWorklet>> {
62        Ok(TestWorklet::new(cx, window, proto))
63    }
64
65    fn AddModule(
66        &self,
67        realm: &mut CurrentRealm,
68        module_url: USVString,
69        options: &WorkletOptions,
70    ) -> Rc<Promise> {
71        self.worklet.AddModule(realm, module_url, options)
72    }
73
74    fn Lookup(&self, key: DOMString) -> Option<DOMString> {
75        let id = self.worklet.worklet_id();
76        let pool = ScriptThread::worklet_thread_pool(self.global().image_cache());
77        pool.test_worklet_lookup(id, String::from(key))
78            .map(DOMString::from)
79    }
80}