LibHTML: Add a convenient way to get from any layout node to the root

1 files changed, 20 insertions, 0 deletions

diff --git a/Libraries/LibHTML/Layout/LayoutNode.cpp b/Libraries/LibHTML/Layout/LayoutNode.cpp
index 49e28bed18..b1ed18b827 100644
--- a/Libraries/LibHTML/Layout/LayoutNode.cpp
+++ b/Libraries/LibHTML/Layout/LayoutNode.cpp
@@ -63,6 +63,26 @@ const Document& LayoutNode::document() const
     return node()->document();
 }
 
+Document& LayoutNode::document()
+{
+    if (is_anonymous())
+        return parent()->document();
+    // FIXME: Remove this const_cast once we give up on the idea of a const link from layout tree to DOM tree.
+    return const_cast<Node*>(node())->document();
+}
+
+const LayoutDocument& LayoutNode::root() const
+{
+    ASSERT(document().layout_node());
+    return *document().layout_node();
+}
+
+LayoutDocument& LayoutNode::root()
+{
+    ASSERT(document().layout_node());
+    return *document().layout_node();
+}
+
 void LayoutNode::split_into_lines(LayoutBlock& container)
 {
     for_each_child([&](auto& child) {