servo/src/components/script/dom/characterdata.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/. */

//! DOM bindings for `CharacterData`.

use dom::bindings::codegen::InheritTypes::CharacterDataDerived;
use dom::bindings::js::JSRef;
use dom::bindings::error::{Fallible, ErrorResult, IndexSize};
use dom::bindings::utils::{Reflectable, Reflector};
use dom::document::Document;
use dom::eventtarget::{EventTarget, NodeTargetTypeId};
use dom::node::{CommentNodeTypeId, Node, NodeTypeId, TextNodeTypeId, ProcessingInstructionNodeTypeId};
use servo_util::str::DOMString;

#[deriving(Encodable)]
pub struct CharacterData {
    pub node: Node,
    pub data: DOMString,
}

impl CharacterDataDerived for EventTarget {
    fn is_characterdata(&self) -> bool {
        match self.type_id {
            NodeTargetTypeId(TextNodeTypeId) |
            NodeTargetTypeId(CommentNodeTypeId) |
            NodeTargetTypeId(ProcessingInstructionNodeTypeId) => true,
            _ => false
        }
    }
}

impl CharacterData {
    pub fn new_inherited(id: NodeTypeId, data: DOMString, document: &JSRef<Document>) -> CharacterData {
        CharacterData {
            node: Node::new_inherited(id, document),
            data: data
        }
    }
}

pub trait CharacterDataMethods {
    fn Data(&self) -> DOMString;
    fn SetData(&mut self, arg: DOMString) -> ErrorResult;
    fn Length(&self) -> u32;
    fn SubstringData(&self, offset: u32, count: u32) -> Fallible<DOMString>;
    fn AppendData(&mut self, arg: DOMString) -> ErrorResult;
    fn InsertData(&mut self, _offset: u32, _arg: DOMString) -> ErrorResult;
    fn DeleteData(&mut self, _offset: u32, _count: u32) -> ErrorResult;
    fn ReplaceData(&mut self, _offset: u32, _count: u32, _arg: DOMString) -> ErrorResult;
}

impl<'a> CharacterDataMethods for JSRef<'a, CharacterData> {
    fn Data(&self) -> DOMString {
        self.data.clone()
    }

    fn SetData(&mut self, arg: DOMString) -> ErrorResult {
        self.data = arg;
        Ok(())
    }

    fn Length(&self) -> u32 {
        self.data.len() as u32
    }

    fn SubstringData(&self, offset: u32, count: u32) -> Fallible<DOMString> {
        Ok(self.data.slice(offset as uint, count as uint).to_str())
    }

    fn AppendData(&mut self, arg: DOMString) -> ErrorResult {
        self.data.push_str(arg);
        Ok(())
    }

    fn InsertData(&mut self, offset: u32, arg: DOMString) -> ErrorResult {
        self.ReplaceData(offset, 0, arg)
    }

    fn DeleteData(&mut self, offset: u32, count: u32) -> ErrorResult {
        self.ReplaceData(offset, count, "".to_owned())
    }

    fn ReplaceData(&mut self, offset: u32, count: u32, arg: DOMString) -> ErrorResult {
        let length = self.data.len() as u32;
        if offset > length {
            return Err(IndexSize);
        }
        let count = if offset + count > length {
            length - offset
        } else {
            count
        };
        let mut data = self.data.slice(0, offset as uint).to_owned();
        data.push_str(arg);
        data.push_str(self.data.slice((offset + count) as uint, length as uint));
        self.data = data;
        // FIXME: Once we have `Range`, we should implement step7 to step11
        Ok(())
    }
}

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

    fn mut_reflector<'a>(&'a mut self) -> &'a mut Reflector {
        self.node.mut_reflector()
    }
}