servo/components/script/dom/messagechannel.rs

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

use crate::dom::bindings::codegen::Bindings::MessageChannelBinding::{MessageChannelMethods, Wrap};
use crate::dom::bindings::error::{Error, Fallible};
use crate::dom::bindings::reflector::{reflect_dom_object, Reflector};
use crate::dom::bindings::root::{Dom, DomRoot};
use crate::dom::globalscope::GlobalScope;
use crate::dom::messageport::MessagePort;
use dom_struct::dom_struct;

#[dom_struct]
pub struct MessageChannel {
    reflector_: Reflector,
    port1: Dom<MessagePort>,
    port2: Dom<MessagePort>,
}

impl MessageChannel {
    /// <https://html.spec.whatwg.org/multipage/#dom-messagechannel>
    #[allow(non_snake_case)]
    pub fn Constructor(global: &GlobalScope) -> Fallible<DomRoot<MessageChannel>> {
        let incumbent = GlobalScope::incumbent().ok_or(Error::InvalidState)?;

        // Step 1
        let port1 = MessagePort::new(&incumbent);

        // Step 2
        let port2 = MessagePort::new(&incumbent);

        incumbent.track_message_port(&*port1, None);
        incumbent.track_message_port(&*port2, None);

        // Step 3
        incumbent.entangle_ports(
            port1.message_port_id().clone(),
            port2.message_port_id().clone(),
        );

        // Steps 4-6
        let channel = reflect_dom_object(
            Box::new(MessageChannel {
                reflector_: Reflector::new(),
                port1: Dom::from_ref(&port1),
                port2: Dom::from_ref(&port2),
            }),
            global,
            Wrap,
        );

        // Step 7
        Ok(channel)
    }
}

impl MessageChannelMethods for MessageChannel {
    /// <https://html.spec.whatwg.org/multipage/#dom-messagechannel-port1>
    fn Port1(&self) -> DomRoot<MessagePort> {
        DomRoot::from_ref(&*self.port1)
    }

    /// <https://html.spec.whatwg.org/multipage/#dom-messagechannel-port2>
    fn Port2(&self) -> DomRoot<MessagePort> {
        DomRoot::from_ref(&*self.port2)
    }
}