Add support for colored wrapping

Signed-off-by: Maxim Zhiburt <zhiburt@gmail.com>

Cargo.toml

@@ -26,8 +26,7 @@ color = ["papergrid/color", "ansi-cut"]
 [dependencies]
 tabled_derive = {path = "./tabled_derive"}
 papergrid = "0.1.22"
-ansi-cut = { version = "0.1.0", optional = true }
-unicode-segmentation = "1.8.0"
+ansi-cut = { version = "0.1.1", optional = true }
 
 [dev-dependencies]
 owo-colors = "1"

examples/color.rs

@@ -11,6 +11,7 @@ fn main() {
     use owo_colors::OwoColorize;
     use tabled::{Alignment, Column, Format, Head, Modify, Object, Row, Style, Table, Tabled};
 
+    #[allow(clippy::upper_case_acronyms)]
     #[derive(Tabled)]
     struct BSD {
         distribution: &'static str,

src/width.rs

@@ -61,12 +61,14 @@ impl<S: AsRef<str>> CellOption for MaxWidth<S> {
                     let new_content = format!("{}{}", striped_content, filler.as_ref());
                     grid.set(Entity::Cell(row, column), Settings::new().text(new_content))
                 }
-
             }
             Wrap::Wrap => {
                 let wrapped_content = split(content, self.width);
                 if wrapped_content.len() != content.len() {
-                    grid.set(Entity::Cell(row, column), Settings::new().text(wrapped_content))
+                    grid.set(
+                        Entity::Cell(row, column),
+                        Settings::new().text(wrapped_content),
+                    )
                 }
             }
         }
@@ -81,15 +83,26 @@ fn strip(s: &str, width: usize) -> String {
     #[cfg(feature = "color")]
     {
         let max_width = std::cmp::min(s.chars().count(), width);
-        ansi_cut::AnsiCut::cut(&s, ..max_width).to_string()
+        ansi_cut::AnsiCut::cut(&s, ..max_width)
     }
 }
 
 fn split(s: &str, width: usize) -> String {
-    s.chars()
-        .collect::<Vec<char>>()
-        .chunks(width)
-        .map(|chunk| chunk.iter().collect::<String>())
-        .collect::<Vec<String>>()
-        .join("\n")
+    #[cfg(not(feature = "color"))]
+    {
+        s.chars()
+            .collect::<Vec<char>>()
+            .chunks(width)
+            .map(|chunk| chunk.iter().collect::<String>())
+            .collect::<Vec<String>>()
+            .join("\n")
+    }
+    #[cfg(feature = "color")]
+    {
+        if width == 0 {
+            s.to_string()
+        } else {
+            ansi_cut::chunks(s, width).join("\n")
+        }
+    }
 }

tests/width_test.rs

@@ -85,6 +85,41 @@ fn max_width_wrapped() {
     assert_eq!(table, expected);
 }
 
+#[cfg(feature = "color")]
+#[test]
+fn max_width_wrapped_collored() {
+    use owo_colors::OwoColorize;
+
+    let data = &[
+        "asd".red().to_string(),
+        "zxc2".blue().to_string(),
+        "asdasd".on_black().green().to_string(),
+    ];
+
+    let expected = concat!(
+        "| St |\n",
+        "| ri |\n",
+        "| ng |\n",
+        "|----|\n",
+        "| \u{1b}[31mas\u{1b}[0m |\n",
+        "| \u{1b}[31md\u{1b}[0m  |\n",
+        "| \u{1b}[34mzx\u{1b}[0m |\n",
+        "| \u{1b}[34mc2\u{1b}[0m |\n",
+        "| \u{1b}[32m\u{1b}[40mas\u{1b}[0m\u{1b}[0m |\n",
+        "| \u{1b}[32m\u{1b}[40mda\u{1b}[0m\u{1b}[0m |\n",
+        "| \u{1b}[32m\u{1b}[40msd\u{1b}[0m\u{1b}[0m |\n",
+    );
+
+    let table = Table::new(data)
+        .with(Style::github_markdown())
+        .with(Modify::new(Full).with(MaxWidth::wrapping(2)))
+        .to_string();
+
+    println!("{}", table);
+
+    assert_eq!(expected, table);
+}
+
 #[test]
 fn dont_change_content_if_width_is_less_then_max_width() {
     let data = vec![