domstringmap.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 dom::bindings::codegen::Bindings::DOMStringMapBinding;
use dom::bindings::codegen::Bindings::DOMStringMapBinding::DOMStringMapMethods;
use dom::bindings::error::ErrorResult;
use dom::bindings::global::GlobalRef;
use dom::bindings::js::{JS, JSRef, Temporary};
use dom::bindings::utils::{Reflectable, Reflector, reflect_dom_object};
use dom::node::window_from_node;
use dom::htmlelement::{HTMLElement, HTMLElementCustomAttributeHelpers};
use servo_util::str::DOMString;

#[dom_struct]
pub struct DOMStringMap {
    reflector_: Reflector,
    element: JS<HTMLElement>,
}

impl DOMStringMap {
    fn new_inherited(element: JSRef<HTMLElement>) -> DOMStringMap {
        DOMStringMap {
            reflector_: Reflector::new(),
            element: JS::from_rooted(element),
        }
    }

    pub fn new(element: JSRef<HTMLElement>) -> Temporary<DOMStringMap> {
        let window = window_from_node(element).root();
        reflect_dom_object(box DOMStringMap::new_inherited(element),
                           GlobalRef::Window(window.root_ref()), DOMStringMapBinding::Wrap)
    }
}

// https://html.spec.whatwg.org/#domstringmap
impl<'a> DOMStringMapMethods for JSRef<'a, DOMStringMap> {
    fn NamedCreator(self, name: DOMString, value: DOMString) -> ErrorResult {
        self.NamedSetter(name, value)
    }

    fn NamedDeleter(self, name: DOMString) {
        let element = self.element.root();
        element.delete_custom_attr(name)
    }

    fn NamedSetter(self, name: DOMString, value: DOMString) -> ErrorResult {
        let element = self.element.root();
        element.set_custom_attr(name, value)
    }

    fn NamedGetter(self, name: DOMString, found: &mut bool) -> DOMString {
        let element = self.element.root();
        match element.get_custom_attr(name) {
            Some(value) => {
                *found = true;
                value.clone()
            },
            None => {
                *found = false;
                String::new()
            }
        }
    }
}

impl Reflectable for DOMStringMap {
    fn reflector<'a>(&'a self) -> &'a Reflector {
        &self.reflector_
    }
}