turn enum into struct

pub enum CollectionTypeId -> pub struct Collection
This commit is contained in:
vectorijk 2015-08-15 02:49:15 -07:00
parent d853aaba65
commit 59d704b53f

View file

@ -24,27 +24,25 @@ pub trait CollectionFilter : JSTraceable {
#[derive(JSTraceable)] #[derive(JSTraceable)]
#[must_root] #[must_root]
pub enum CollectionTypeId { pub struct Collection(JS<Node>, Box<CollectionFilter + 'static>);
Live(JS<Node>, Box<CollectionFilter+'static>)
}
#[dom_struct] #[dom_struct]
#[derive(HeapSizeOf)] #[derive(HeapSizeOf)]
pub struct HTMLCollection { pub struct HTMLCollection {
reflector_: Reflector, reflector_: Reflector,
#[ignore_heap_size_of = "Contains a trait object; can't measure due to #6870"] #[ignore_heap_size_of = "Contains a trait object; can't measure due to #6870"]
collection: CollectionTypeId, collection: Collection,
} }
impl HTMLCollection { impl HTMLCollection {
fn new_inherited(collection: CollectionTypeId) -> HTMLCollection { fn new_inherited(collection: Collection) -> HTMLCollection {
HTMLCollection { HTMLCollection {
reflector_: Reflector::new(), reflector_: Reflector::new(),
collection: collection, collection: collection,
} }
} }
pub fn new(window: &Window, collection: CollectionTypeId) -> Root<HTMLCollection> { pub fn new(window: &Window, collection: Collection) -> Root<HTMLCollection> {
reflect_dom_object(box HTMLCollection::new_inherited(collection), reflect_dom_object(box HTMLCollection::new_inherited(collection),
GlobalRef::Window(window), HTMLCollectionBinding::Wrap) GlobalRef::Window(window), HTMLCollectionBinding::Wrap)
} }
@ -53,7 +51,7 @@ impl HTMLCollection {
impl HTMLCollection { impl HTMLCollection {
pub fn create(window: &Window, root: &Node, pub fn create(window: &Window, root: &Node,
filter: Box<CollectionFilter + 'static>) -> Root<HTMLCollection> { filter: Box<CollectionFilter + 'static>) -> Root<HTMLCollection> {
HTMLCollection::new(window, CollectionTypeId::Live(JS::from_ref(root), filter)) HTMLCollection::new(window, Collection(JS::from_ref(root), filter))
} }
fn all_elements(window: &Window, root: &Node, fn all_elements(window: &Window, root: &Node,
@ -178,31 +176,28 @@ impl HTMLCollection {
impl<'a> HTMLCollectionMethods for &'a HTMLCollection { impl<'a> HTMLCollectionMethods for &'a HTMLCollection {
// https://dom.spec.whatwg.org/#dom-htmlcollection-length // https://dom.spec.whatwg.org/#dom-htmlcollection-length
#[allow(unrooted_must_root)]
fn Length(self) -> u32 { fn Length(self) -> u32 {
match self.collection { let Collection(ref root, ref filter) = self.collection;
CollectionTypeId::Live(ref root, ref filter) => { let root = root.root();
let root = root.root(); HTMLCollection::traverse(root.r())
HTMLCollection::traverse(root.r()) .filter(|element| filter.filter(element.r(), root.r()))
.filter(|element| filter.filter(element.r(), root.r())) .count() as u32
.count() as u32
}
}
} }
// https://dom.spec.whatwg.org/#dom-htmlcollection-item // https://dom.spec.whatwg.org/#dom-htmlcollection-item
#[allow(unrooted_must_root)]
fn Item(self, index: u32) -> Option<Root<Element>> { fn Item(self, index: u32) -> Option<Root<Element>> {
let index = index as usize; let index = index as usize;
match self.collection { let Collection(ref root, ref filter) = self.collection;
CollectionTypeId::Live(ref root, ref filter) => { let root = root.root();
let root = root.root(); HTMLCollection::traverse(root.r())
HTMLCollection::traverse(root.r()) .filter(|element| filter.filter(element.r(), root.r()))
.filter(|element| filter.filter(element.r(), root.r())) .nth(index)
.nth(index)
}
}
} }
// https://dom.spec.whatwg.org/#dom-htmlcollection-nameditem // https://dom.spec.whatwg.org/#dom-htmlcollection-nameditem
#[allow(unrooted_must_root)]
fn NamedItem(self, key: DOMString) -> Option<Root<Element>> { fn NamedItem(self, key: DOMString) -> Option<Root<Element>> {
// Step 1. // Step 1.
if key.is_empty() { if key.is_empty() {
@ -210,16 +205,13 @@ impl<'a> HTMLCollectionMethods for &'a HTMLCollection {
} }
// Step 2. // Step 2.
match self.collection { let Collection(ref root, ref filter) = self.collection;
CollectionTypeId::Live(ref root, ref filter) => { let root = root.root();
let root = root.root(); HTMLCollection::traverse(root.r())
HTMLCollection::traverse(root.r()) .filter(|element| filter.filter(element.r(), root.r()))
.filter(|element| filter.filter(element.r(), root.r())) .find(|elem| {
.find(|elem| { elem.r().get_string_attribute(&atom!("name")) == key ||
elem.r().get_string_attribute(&atom!("name")) == key || elem.r().get_string_attribute(&atom!("id")) == key})
elem.r().get_string_attribute(&atom!("id")) == key })
}
}
} }
// https://dom.spec.whatwg.org/#dom-htmlcollection-item // https://dom.spec.whatwg.org/#dom-htmlcollection-item