Add WithoutAnsi struct that implements Display

Author: ChocolateLoverRaj

src/lib.rs

@@ -101,7 +101,7 @@ pub use crate::utils::{
 #[cfg(all(feature = "ansi-parsing", feature = "alloc"))]
 pub use crate::ansi::strip_ansi_codes;
 #[cfg(feature = "ansi-parsing")]
-pub use crate::ansi::AnsiCodeIterator;
+pub use crate::{ansi::AnsiCodeIterator, without_ansi::WithoutAnsi};
 
 #[cfg(feature = "std")]
 mod common_term;
@@ -120,3 +120,5 @@ mod windows_term;
 
 #[cfg(feature = "ansi-parsing")]
 mod ansi;
+#[cfg(feature = "ansi-parsing")]
+mod without_ansi;

src/without_ansi.rs

@@ -0,0 +1,41 @@
+use core::fmt::Display;
+
+use crate::AnsiCodeIterator;
+
+/// A wrapper struct that implements [`core::fmt::Display`], only displaying non-ansi parts.
+pub struct WithoutAnsi<'a> {
+    str: &'a str,
+}
+impl<'a> WithoutAnsi<'a> {
+    pub fn new(str: &'a str) -> Self {
+        Self { str }
+    }
+}
+impl Display for WithoutAnsi<'_> {
+    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
+        for str in
+            AnsiCodeIterator::new(self.str)
+                .filter_map(|(str, is_ansi)| if is_ansi { None } else { Some(str) })
+        {
+            f.write_str(str)?;
+        }
+        Ok(())
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use core::fmt::Write;
+
+    #[test]
+    fn display_multiple_times() {
+        let str_with_ansi = "\x1b[1;97;41mError\x1b[0m";
+        let without_ansi = WithoutAnsi::new(str_with_ansi);
+        for _ in 0..2 {
+            let mut output = String::default();
+            write!(output, "{without_ansi}").unwrap();
+            assert_eq!(output, "Error");
+        }
+    }
+}