3 files changed, 36 insertions, 0 deletions
diff --git a/Libraries/LibHTML/DOM/Document.cpp b/Libraries/LibHTML/DOM/Document.cpp
index 36c937e6d7..0e24019598 100644
--- a/Libraries/LibHTML/DOM/Document.cpp
+++ b/Libraries/LibHTML/DOM/Document.cpp
@@ -19,3 +19,18 @@ StyleResolver& Document::style_resolver()
         m_style_resolver = make<StyleResolver>(*this);
     return *m_style_resolver;
 }
+
+void Document::normalize()
+{
+    if (first_child() != nullptr && first_child()->is_element()) {
+        const Element& el = static_cast<const Element&>(*first_child());
+        if (el.tag_name() == "html")
+            return;
+    }
+
+    NonnullRefPtr<Element> body = adopt(*new Element("body"));
+    NonnullRefPtr<Element> html = adopt(*new Element("html"));
+    html->append_child(body);
+    this->donate_all_children_to(body);
+    this->append_child(html);
+}
diff --git a/Libraries/LibHTML/DOM/Document.h b/Libraries/LibHTML/DOM/Document.h
index 3add23acd3..2b3c10e9bd 100644
--- a/Libraries/LibHTML/DOM/Document.h
+++ b/Libraries/LibHTML/DOM/Document.h
@@ -16,6 +16,8 @@ public:
     Document();
     virtual ~Document() override;
 
+    void normalize();
+
     StyleResolver& style_resolver();
 
     void add_sheet(const StyleSheet& sheet) { m_sheets.append(sheet); }
diff --git a/Libraries/LibHTML/TreeNode.h b/Libraries/LibHTML/TreeNode.h
index df3b67cdf7..a4da2f1f9b 100644
--- a/Libraries/LibHTML/TreeNode.h
+++ b/Libraries/LibHTML/TreeNode.h
@@ -34,6 +34,7 @@ public:
     const T* last_child() const { return m_last_child; }
 
     void append_child(NonnullRefPtr<T> node);
+    void donate_all_children_to(T& node);
 
 protected:
     TreeNode() { }
@@ -59,3 +60,21 @@ inline void TreeNode<T>::append_child(NonnullRefPtr<T> node)
     if (!m_first_child)
         m_first_child = m_last_child;
 }
+
+template<typename T>
+inline void TreeNode<T>::donate_all_children_to(T& node)
+{
+    for (T* child = m_first_child; child != nullptr;) {
+        T* next_child = child->m_next_sibling;
+
+        child->m_parent = nullptr;
+        child->m_next_sibling = nullptr;
+        child->m_previous_sibling = nullptr;
+
+        node.append_child(adopt(*child));
+        child = next_child;
+    }
+
+    m_first_child = nullptr;
+    m_last_child = nullptr;
+}