script/
routed_promise.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
5use std::rc::Rc;
6
7use ipc_channel::ipc::{self, IpcSender};
8use ipc_channel::router::ROUTER;
9use serde::Serialize;
10use serde::de::DeserializeOwned;
11
12use crate::dom::bindings::refcounted::{Trusted, TrustedPromise};
13use crate::dom::bindings::reflector::DomObject;
14use crate::dom::promise::Promise;
15use crate::script_runtime::CanGc;
16use crate::task_source::TaskSource;
17
18pub(crate) trait RoutedPromiseListener<R: Serialize + DeserializeOwned + Send> {
19    fn handle_response(&self, response: R, promise: &Rc<Promise>, can_gc: CanGc);
20}
21
22pub(crate) struct RoutedPromiseContext<
23    R: Serialize + DeserializeOwned + Send,
24    T: RoutedPromiseListener<R> + DomObject,
25> {
26    trusted: TrustedPromise,
27    receiver: Trusted<T>,
28    _phantom: std::marker::PhantomData<R>,
29}
30
31impl<R: Serialize + DeserializeOwned + Send, T: RoutedPromiseListener<R> + DomObject>
32    RoutedPromiseContext<R, T>
33{
34    fn response(self, response: R, can_gc: CanGc) {
35        let promise = self.trusted.root();
36        self.receiver
37            .root()
38            .handle_response(response, &promise, can_gc);
39    }
40}
41
42pub(crate) fn route_promise<
43    R: Serialize + DeserializeOwned + Send + 'static,
44    T: RoutedPromiseListener<R> + DomObject + 'static,
45>(
46    promise: &Rc<Promise>,
47    receiver: &T,
48    task_source: TaskSource,
49) -> IpcSender<R> {
50    let (action_sender, action_receiver) = ipc::channel().unwrap();
51    let task_source = task_source.to_sendable();
52    let mut trusted: Option<TrustedPromise> = Some(TrustedPromise::new(promise.clone()));
53    let trusted_receiver = Trusted::new(receiver);
54    ROUTER.add_typed_route(
55        action_receiver,
56        Box::new(move |message| {
57            let trusted = if let Some(trusted) = trusted.take() {
58                trusted
59            } else {
60                error!("RoutedPromiseListener callback called twice!");
61                return;
62            };
63
64            let context = RoutedPromiseContext {
65                trusted,
66                receiver: trusted_receiver.clone(),
67                _phantom: Default::default(),
68            };
69            task_source.queue(task!(routed_promise_task: move|| {
70                context.response(message.unwrap(), CanGc::note());
71            }));
72        }),
73    );
74    action_sender
75}