Implement measure_text_width with no allocation

I implemented that while working on optimizing something I don't
remember, it might still be useful.

Author: remi-dupre

Cargo.toml

@@ -19,7 +19,7 @@ ansi-parsing = []
 
 [dependencies]
 libc = "0.2.99"
-once_cell = "1.8"
+once_cell = "1.8,<1.21" # version 1.21 requires Rust >= 1.70
 unicode-width = { version = "0.2", optional = true }
 
 [target.'cfg(windows)'.dependencies]

Makefile

@@ -14,6 +14,8 @@ test:
 	@cargo test
 	@cargo test --all-features
 	@cargo test --no-default-features
+	@cargo test --no-default-features -F ansi-parsing
+	@cargo test --no-default-features -F unicode-width
 
 check-minver:
 	@echo "MINVER CHECK"

src/utils.rs

@@ -9,12 +9,7 @@ use once_cell::sync::Lazy;
 use crate::term::{wants_emoji, Term};
 
 #[cfg(feature = "ansi-parsing")]
-use crate::ansi::{strip_ansi_codes, AnsiCodeIterator};
-
-#[cfg(not(feature = "ansi-parsing"))]
-fn strip_ansi_codes(s: &str) -> &str {
-    s
-}
+use crate::ansi::AnsiCodeIterator;
 
 fn default_colors_enabled(out: &Term) -> bool {
     (out.features().colors_supported()
@@ -71,7 +66,18 @@ pub fn set_colors_enabled_stderr(val: bool) {
 
 /// Measure the width of a string in terminal characters.
 pub fn measure_text_width(s: &str) -> usize {
-    str_width(&strip_ansi_codes(s))
+    dbg!(&s);
+    #[cfg(feature = "ansi-parsing")]
+    {
+        AnsiCodeIterator::new(s)
+            .filter(|(_, is_ansi)| !is_ansi)
+            .map(|(sub, _)| str_width(sub))
+            .sum()
+    }
+    #[cfg(not(feature = "ansi-parsing"))]
+    {
+        str_width(s)
+    }
 }
 
 /// A terminal color.
@@ -937,16 +943,28 @@ fn test_text_width() {
         .bold()
         .force_styling(true)
         .to_string();
+
     assert_eq!(
         measure_text_width(&s),
         if cfg!(feature = "ansi-parsing") {
             3
-        } else if cfg!(feature = "unicode-width") {
-            17
         } else {
             21
         }
     );
+
+    assert_eq!(
+        measure_text_width(&dbg!(style("🐶 <3").red()).to_string()),
+        match (
+            cfg!(feature = "ansi-parsing"),
+            cfg!(feature = "unicode-width")
+        ) {
+            (true, true) => 5,    // "🐶 <3"
+            (true, false) => 4,   // "🐶 <3", no unicode-aware width
+            (false, true) => 14,  // full string
+            (false, false) => 13, // full string, no unicode-aware width
+        }
+    );
 }
 
 #[test]