to html

Author: alexkazik

src/dom/element.rs

@@ -1,4 +1,4 @@
-use super::node::Node;
+use super::node::{Node, write_html_list};
 #[cfg(feature = "source-span")]
 use super::span::SourceSpan;
 use crate::dom::vecmap::VecMap;
@@ -92,6 +92,60 @@ impl<'a> Element<'a> {
     {
         let _ = for_each_element_async(self.children.as_mut_slice(), &mut f).await;
     }
+
+    #[inline(always)]
+    pub fn to_html(&self) -> String {
+        let mut result = String::new();
+        self.write_html(&mut result);
+        result
+    }
+
+    pub fn write_html(&self, writer: &mut String) {
+        let e = self;
+
+        writer.push('<');
+        writer.push_str(&e.name);
+        if !e.classes.is_empty() {
+            writer.push_str(" class=\"");
+            writer.push_str(
+                &e.classes
+                    .iter()
+                    .map(|c| c.as_str())
+                    .collect::<Vec<_>>()
+                    .join(" "),
+            );
+            writer.push('"');
+        }
+        if let Some(id) = &e.id {
+            writer.push(' ');
+            writer.push_str("id");
+            writer.push('=');
+            writer.push('"');
+            writer.push_str(id);
+            writer.push('"');
+        }
+        for (k, v) in e.attributes.iter() {
+            writer.push(' ');
+            writer.push_str(k);
+            if let Some(v) = v {
+                writer.push('=');
+                writer.push('"');
+                writer.push_str(v);
+                writer.push('"');
+            }
+        }
+        if e.variant == ElementVariant::Normal {
+            writer.push('>');
+            write_html_list(writer, &e.children);
+            writer.push('<');
+            writer.push('/');
+            writer.push_str(&e.name);
+            writer.push('>');
+        } else {
+            writer.push('/');
+            writer.push('>');
+        }
+    }
 }
 
 pub(crate) fn for_each_element<'a, F>(tree: &mut [Node<'a>], f: &mut F) -> ControlFlow<(), ()>

src/dom/mod.rs

@@ -19,6 +19,7 @@ pub mod vecmap;
 pub mod vecset;
 
 use crate::dom::element::{WalkResult, for_each_element, for_each_element_async};
+use crate::dom::node::write_html_list;
 #[cfg(feature = "source-span")]
 use crate::dom::span::SourceSpan;
 use element::{Element, ElementVariant};
@@ -385,4 +386,16 @@ impl<'a> Dom<'a> {
     {
         let _ = for_each_element_async(self.children.as_mut_slice(), &mut f).await;
     }
+
+    #[inline(always)]
+    pub fn to_html(&self) -> String {
+        let mut result = String::new();
+        self.write_html(&mut result);
+        result
+    }
+
+    #[inline(always)]
+    pub fn write_html(&self, writer: &mut String) {
+        write_html_list(writer, self.children.as_slice());
+    }
 }

src/dom/node.rs

@@ -67,6 +67,34 @@ impl<'a> Node<'a> {
             let _ = for_each_element_async(e.children.as_mut_slice(), &mut f).await;
         }
     }
+
+    #[inline]
+    pub fn to_html(&self) -> String {
+        let mut result = String::new();
+        self.write_html(&mut result);
+        result
+    }
+
+    #[inline]
+    pub fn write_html(&self, writer: &mut String) {
+        match self {
+            Node::Text(s) => {
+                writer.push_str(s);
+            }
+            Node::Element(e) => e.write_html(writer),
+            Node::Comment(c) => {
+                writer.push_str("<!--");
+                writer.push_str(c);
+                writer.push_str("-->");
+            }
+        }
+    }
+}
+
+pub(crate) fn write_html_list(writer: &mut String, list: &[Node]) {
+    for n in list {
+        n.write_html(writer);
+    }
 }
 
 impl<'a> IntoIterator for &'a Node<'a> {