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::for_each::ForEach;
@@ -81,3 +81,59 @@ impl<'a> ForEach<'a> for Element<'a> {
         self.children.as_mut_slice()
     }
 }
+
+impl Element<'_> {
+    #[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('>');
+        }
+    }
+}

src/dom/mod.rs

@@ -20,6 +20,7 @@ pub mod vecmap;
 pub mod vecset;
 
 use crate::dom::for_each::ForEach;
+use crate::dom::node::write_html_list;
 #[cfg(feature = "source-span")]
 use crate::dom::span::SourceSpan;
 use element::{Element, ElementVariant};
@@ -370,6 +371,18 @@ impl<'a> Dom<'a> {
         }
         Ok(attribute)
     }
+
+    #[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());
+    }
 }
 
 impl<'a> ForEach<'a> for Dom<'a> {

src/dom/node.rs

@@ -62,6 +62,36 @@ impl<'a> ForEach<'a> for Node<'a> {
     }
 }
 
+impl Node<'_> {
+    #[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> {
     type Item = &'a Node<'a>;
     type IntoIter = NodeIntoIterator<'a>;