LibWeb: Add NonDocumentTypeChildNode::next_element_in_pre_order()

This is handy for traversing only the elements in a document.

1 files changed, 10 insertions, 0 deletions

diff --git a/Libraries/LibWeb/DOM/NonDocumentTypeChildNode.h b/Libraries/LibWeb/DOM/NonDocumentTypeChildNode.h
index 720edd2429..2e3b3996d9 100644
--- a/Libraries/LibWeb/DOM/NonDocumentTypeChildNode.h
+++ b/Libraries/LibWeb/DOM/NonDocumentTypeChildNode.h
@@ -53,8 +53,18 @@ public:
         return nullptr;
     }
 
+    Element* next_element_in_pre_order()
+    {
+        for (auto* node = static_cast<NodeType*>(this)->next_in_pre_order(); node; node = node->next_in_pre_order()) {
+            if (is<Element>(*node))
+                return downcast<Element>(node);
+        }
+        return nullptr;
+    }
+
     const Element* previous_element_sibling() const { return const_cast<NonDocumentTypeChildNode*>(this)->previous_element_sibling(); }
     const Element* next_element_sibling() const { return const_cast<NonDocumentTypeChildNode*>(this)->next_element_sibling(); }
+    const Element* next_element_in_pre_order() const { return const_cast<NonDocumentTypeChildNode*>(this)->next_element_in_pre_order(); }
 
 protected:
     NonDocumentTypeChildNode() { }