script/dom/
xmlserializer.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 dom_struct::dom_struct;
6use js::rust::HandleObject;
7use xml5ever::serialize::{SerializeOpts, TraversalScope, serialize};
8
9use crate::dom::bindings::codegen::Bindings::XMLSerializerBinding::XMLSerializerMethods;
10use crate::dom::bindings::error::{Error, Fallible};
11use crate::dom::bindings::reflector::{Reflector, reflect_dom_object_with_proto};
12use crate::dom::bindings::root::{Dom, DomRoot};
13use crate::dom::bindings::str::DOMString;
14use crate::dom::node::Node;
15use crate::dom::window::Window;
16use crate::script_runtime::CanGc;
17
18#[dom_struct]
19pub(crate) struct XMLSerializer {
20    reflector_: Reflector,
21    window: Dom<Window>,
22}
23
24impl XMLSerializer {
25    fn new_inherited(window: &Window) -> XMLSerializer {
26        XMLSerializer {
27            reflector_: Reflector::new(),
28            window: Dom::from_ref(window),
29        }
30    }
31
32    pub(crate) fn new(
33        window: &Window,
34        proto: Option<HandleObject>,
35        can_gc: CanGc,
36    ) -> DomRoot<XMLSerializer> {
37        reflect_dom_object_with_proto(
38            Box::new(XMLSerializer::new_inherited(window)),
39            window,
40            proto,
41            can_gc,
42        )
43    }
44}
45
46impl XMLSerializerMethods<crate::DomTypeHolder> for XMLSerializer {
47    // https://w3c.github.io/DOM-Parsing/#dom-xmlserializer
48    fn Constructor(
49        window: &Window,
50        proto: Option<HandleObject>,
51        can_gc: CanGc,
52    ) -> Fallible<DomRoot<XMLSerializer>> {
53        Ok(XMLSerializer::new(window, proto, can_gc))
54    }
55
56    // https://w3c.github.io/DOM-Parsing/#the-xmlserializer-interface
57    fn SerializeToString(&self, root: &Node) -> Fallible<DOMString> {
58        let mut writer = vec![];
59        match serialize(
60            &mut writer,
61            &root,
62            SerializeOpts {
63                traversal_scope: TraversalScope::IncludeNode,
64            },
65        ) {
66            Ok(_) => Ok(DOMString::from(String::from_utf8(writer).unwrap())),
67            Err(_) => Err(Error::Type(String::from(
68                "root must be a Node or an Attr object",
69            ))),
70        }
71    }
72}