add a method to tell between node types

shouya · Feb 7, 2024 · cbb2979 · cbb2979
1 parent d471546
commit cbb2979
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/js/dom.rs b/src/js/dom.rs
@@ -280,6 +280,16 @@ impl<'js> Node<'js> {
     Ok(parent)
   }
 
+  fn node_type(&self) -> String {
+    let dom = self.dom.borrow();
+    let node = self.node_ref(&dom).unwrap();
+    let val = node.value();
+    match val {
+      scraper::Node::Text(_) => "text".to_string(),
+      scraper::Node::Element(_) => "element".to_string(),
+      _ => "other".to_string(),
+    }
+  }
 
   fn remove(&self) {
     self.node_mut(&mut self.dom.borrow_mut()).unwrap().detach();