From 225f6d4660d8df920054ba4490a1a04aca7bbe80 Mon Sep 17 00:00:00 2001
From: Dmitry Konishchev <konishchev@gmail.com>
Date: Mon, 4 Nov 2024 13:46:06 +0300
Subject: [PATCH 1/3] Fix a typo

---
 src/csv.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/csv.rs b/src/csv.rs
index e3d116ade..3a6e1199d 100644
--- a/src/csv.rs
+++ b/src/csv.rs
@@ -17,8 +17,8 @@ impl<'a> super::TableSlice<'a> {
     ///
     /// This allows for format customisation.
     pub fn to_csv_writer<W: Write>(&self, mut writer: Writer<W>) -> Result<Writer<W>> {
-        for title in self.titles {
-            writer.write_record(title.iter().map(|c| c.get_content()))?;
+        if let Some(titles) = self.titles {
+            writer.write_record(titles.iter().map(|c| c.get_content()))?;
         }
         for row in self.rows {
             writer.write_record(row.iter().map(|c| c.get_content()))?;

From cedea30668f06b776dcd6dcf2ae896b80ccdec4f Mon Sep 17 00:00:00 2001
From: Dmitry Konishchev <konishchev@gmail.com>
Date: Mon, 4 Nov 2024 14:45:11 +0300
Subject: [PATCH 2/3] Fix invalid cell width calculation with modern
 unicode-width versions

Fixes #165
---
 Cargo.toml   |  3 ++-
 src/utils.rs | 51 ++-------------------------------------------------
 2 files changed, 4 insertions(+), 50 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 65fd38eab..67c347bb8 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -36,9 +36,10 @@ path = "src/main.rs"
 name = "prettytable"
 
 [dependencies]
-unicode-width = "0.1"
+unicode-width = "0.2"
 term = "0.7"
 lazy_static = "1.4"
 is-terminal = "0.4"
 encode_unicode = "1.0"
+strip-ansi-escapes = "0.2"
 csv = { version = "1.1", optional = true }
diff --git a/src/utils.rs b/src/utils.rs
index 1256ced7b..5bdcb05dd 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -3,7 +3,7 @@ use std::fmt;
 use std::io::{Error, ErrorKind, Write};
 use std::str;
 
-use unicode_width::{UnicodeWidthChar, UnicodeWidthStr};
+use unicode_width::UnicodeWidthStr;
 
 use super::format::Alignment;
 
@@ -84,54 +84,7 @@ pub fn print_align<T: Write + ?Sized>(
 /// Return the display width of a unicode string.
 /// This functions takes ANSI-escaped color codes into account.
 pub fn display_width(text: &str) -> usize {
-    #[derive(PartialEq, Eq, Clone, Copy)]
-    enum State {
-        /// We are not inside any terminal escape.
-        Normal,
-        /// We have just seen a \u{1b}
-        EscapeChar,
-        /// We have just seen a [
-        OpenBracket,
-        /// We just ended the escape by seeing an m
-        AfterEscape,
-    }
-
-    let width = UnicodeWidthStr::width(text);
-    let mut state = State::Normal;
-    let mut hidden = 0;
-
-    for c in text.chars() {
-        state = match (state, c) {
-            (State::Normal, '\u{1b}') => State::EscapeChar,
-            (State::EscapeChar, '[') => State::OpenBracket,
-            (State::EscapeChar, _) => State::Normal,
-            (State::OpenBracket, 'm') => State::AfterEscape,
-            _ => state,
-        };
-
-        // We don't count escape characters as hidden as
-        // UnicodeWidthStr::width already considers them.
-        if matches!(state, State::OpenBracket | State::AfterEscape) {
-            // but if we see an escape char *inside* the ANSI escape, we should ignore it.
-            if UnicodeWidthChar::width(c).unwrap_or(0) > 0 {
-                hidden += 1;
-            }
-        }
-
-        if state == State::AfterEscape {
-            state = State::Normal;
-        }
-    }
-
-    assert!(
-        width >= hidden,
-        "internal error: width {} less than hidden {} on string {:?}",
-        width,
-        hidden,
-        text
-    );
-
-    width - hidden
+    UnicodeWidthStr::width(strip_ansi_escapes::strip_str(text).as_str())
 }
 
 /// Wrapper struct which will emit the HTML-escaped version of the contained

From d5e2a480c02b0c4be8d61f4f3a9039c841376433 Mon Sep 17 00:00:00 2001
From: Dmitry Konishchev <konishchev@gmail.com>
Date: Tue, 5 Nov 2024 09:18:31 +0300
Subject: [PATCH 3/3] Add display_width() test

---
 src/utils.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/utils.rs b/src/utils.rs
index 5bdcb05dd..761d3912a 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -184,4 +184,13 @@ mod tests {
         let res = out.write_all(&[0, 255]);
         assert!(res.is_err());
     }
+
+    #[test]
+    fn display_width_calculation() {
+        assert_eq!(display_width("ASCII text"), 10);
+        assert_eq!(display_width(" ASCII\r\ntext with\nspaces "), 24);
+
+        assert_eq!(display_width("Не-ASCII текст"), 14);
+        assert_eq!(display_width("Colored \u{1B}[2mне-ASCII\u{1B}[0m текст"), 22);
+    }
 }