1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112
use std::convert::TryFrom; use wasm_bindgen::JsCast; use crate::console::{Write, Writer}; use crate::GenericElement; pub struct QuerySelectorAll { inner: web_sys::NodeList, } impl QuerySelectorAll { pub(crate) fn new(inner: web_sys::NodeList) -> Self { QuerySelectorAll { inner } } pub fn get(&self, index: usize) -> Option<GenericElement> { u32::try_from(index) .ok() .and_then(|index| self.inner.get(index)) .map(|node| { let element: web_sys::Element = node.unchecked_into(); element.into() }) } pub fn len(&self) -> usize { self.inner.length() as usize } pub fn is_empty(&self) -> bool { self.len() == 0 } pub fn is_not_empty(&self) -> bool { !self.is_empty() } pub fn first(&self) -> Option<GenericElement> { self.get(0) } pub fn last(&self) -> Option<GenericElement> { let len = self.len(); if len > 0 { self.get(len - 1) } else { None } } pub fn iter(&self) -> QuerySelectorAllIter { QuerySelectorAllIter { query_selector_all: self, current: 0, } } } impl Write for QuerySelectorAll { fn write(&self, writer: &mut Writer) { writer.write_1(self.inner.as_ref()) } } impl IntoIterator for QuerySelectorAll { type Item = GenericElement; type IntoIter = QuerySelectorAllIntoIter; fn into_iter(self) -> Self::IntoIter { QuerySelectorAllIntoIter { query_selector_all: self, current: 0, } } } pub struct QuerySelectorAllIter<'a> { query_selector_all: &'a QuerySelectorAll, current: usize, } impl<'a> Iterator for QuerySelectorAllIter<'a> { type Item = GenericElement; fn next(&mut self) -> Option<Self::Item> { let current = self.current; self.current += 1; self.query_selector_all.get(current) } } pub struct QuerySelectorAllIntoIter { query_selector_all: QuerySelectorAll, current: usize, } impl Iterator for QuerySelectorAllIntoIter { type Item = GenericElement; fn next(&mut self) -> Option<Self::Item> { let current = self.current; self.current += 1; self.query_selector_all.get(current) } }