implement element writing

1 files changed, 27 insertions, 0 deletions

diff --git a/src/element.rs b/src/element.rs
index 5b5f048..04f2e5e 100644
--- a/src/element.rs
+++ b/src/element.rs
@@ -54,6 +54,33 @@ pub struct Element {
     pub content: Vec<Content>,
 }
 
+pub fn escape_str(s: &str) -> String {
+    let mut string = String::new();
+    for str in s.split_inclusive(|c| c == '<' || c == '&' || c == '>') {
+        if let Some(str) = str.strip_suffix('<') {
+            if !str.is_empty() {
+                string.push_str(str)
+            }
+            string.push_str("&lt;");
+        } else if let Some(str) = str.strip_suffix('&') {
+            if !str.is_empty() {
+                string.push_str(str)
+            }
+            string.push_str("&amp;");
+        } else if let Some(str) = str.strip_suffix('>') {
+            if !str.is_empty() {
+                string.push_str(str)
+            }
+            string.push_str("&gt;");
+        } else {
+            if !str.is_empty() {
+                string.push_str(str)
+            }
+        }
+    }
+    string
+}
+
 // impl<'s> TryFrom<xml::Element<'s>> for Element<'s> {
 //     type Error = Error;