Parse styles from a string containing ANSI sequences

1 files changed, 15 insertions, 0 deletions

diff --git a/src/ansi/mod.rs b/src/ansi/mod.rs
index 643b47f0..5f90c536 100644
--- a/src/ansi/mod.rs
+++ b/src/ansi/mod.rs
@@ -3,6 +3,7 @@ mod iterator;
 
 use std::borrow::Cow;
 
+use ansi_term::Style;
 use itertools::Itertools;
 use unicode_segmentation::UnicodeSegmentation;
 use unicode_width::UnicodeWidthStr;
@@ -64,6 +65,20 @@ pub fn truncate_str<'a, 'b>(s: &'a str, display_width: usize, tail: &'b str) ->
     Cow::from(format!("{}{}", result, result_tail))
 }
 
+pub fn parse_style_sections(s: &str) -> Vec<(ansi_term::Style, &str)> {
+    let mut sections = Vec::new();
+    let mut curr_style = Style::default();
+    for element in AnsiElementIterator::new(s) {
+        match element {
+            Element::Text(start, end) => sections.push((curr_style, &s[start..end])),
+            Element::Csi(style, _, _) => curr_style = style,
+            _ => {}
+        }
+    }
+    sections
+}
+
+// Return the first CSI element, if any, as an `ansi_term::Style`.
 pub fn parse_first_style(s: &str) -> Option<ansi_term::Style> {
     AnsiElementIterator::new(s).find_map(|el| match el {
         Element::Csi(style, _, _) => Some(style),