added ansi_parse trait so things are more ergonomic

src/traits.rs

@@ -0,0 +1,74 @@
+use crate::enums::{Output};
+use crate::parsers::parse_escape;
+
+pub trait AnsiParser {
+    fn ansi_parse<'a>(&'a self) -> AnsiParseIterator<'a>;
+}
+
+impl AnsiParser for str {
+    fn ansi_parse<'a>(&'a self) -> AnsiParseIterator<'a> {
+        AnsiParseIterator {
+            dat: self
+        }
+    }
+}
+
+impl AnsiParser for String {
+    fn ansi_parse<'a>(&'a self) -> AnsiParseIterator<'a> {
+        AnsiParseIterator {
+            dat: self
+        }
+    }
+}
+
+pub struct AnsiParseIterator<'a> {
+    dat: &'a str,
+}
+
+impl<'a> Iterator for AnsiParseIterator<'a> {
+    type Item = Output<'a>;
+    
+    fn next(&mut self) -> Option<Self::Item> {
+        if self.dat == "" {
+            return None;
+        }
+
+        let pos = self.dat.find('\u{1b}');
+        if let Some(loc) = pos {
+            if loc == 0 {
+                let res = parse_escape(&self.dat[loc..]);
+
+                if let Ok(ret) = res {
+                    self.dat = &ret.0;
+                    Some(ret.1)
+                }else{
+                    let pos = self.dat[(loc+1)..].find('\u{1b}');
+                    if let Some(loc) = pos {
+                        //Added to because it's based one character ahead
+                        let loc = loc+1;
+
+                        let temp = &self.dat[..loc];
+                        self.dat = &self.dat[loc..];
+
+                        Some(Output::TextBlock(temp))
+                    }else{
+                        let temp = self.dat;
+                        self.dat = "";
+
+                        Some(Output::TextBlock(temp))
+                    }
+                }
+
+            }else {
+                let temp = &self.dat[..loc];
+                self.dat = &self.dat[loc..];
+
+                Some(Output::TextBlock(&temp))
+            }
+        }else{
+            let temp = self.dat;
+            self.dat = "";
+            Some(Output::TextBlock(temp))
+        }
+    }
+}