syncable_cli/analyzer/display/
utils.rs

1//! String manipulation and visual width calculation utilities
2
3/// Calculate visual width of a string, handling ANSI color codes
4pub fn visual_width(s: &str) -> usize {
5    let mut width = 0;
6    let mut chars = s.chars().peekable();
7
8    while let Some(ch) = chars.next() {
9        if ch == '\x1b' {
10            // Skip ANSI escape sequence
11            if chars.peek() == Some(&'[') {
12                chars.next(); // consume '['
13                for c in chars.by_ref() {
14                    if c.is_ascii_alphabetic() {
15                        break; // End of escape sequence
16                    }
17                }
18            }
19        } else {
20            // Simple width calculation for common cases
21            // Most characters are width 1, some are width 0 or 2
22            width += char_width(ch);
23        }
24    }
25
26    width
27}
28
29/// Simple character width calculation without external dependencies
30pub fn char_width(ch: char) -> usize {
31    match ch {
32        // Control characters have width 0
33        '\u{0000}'..='\u{001F}' | '\u{007F}' => 0,
34        // Combining marks have width 0
35        '\u{0300}'..='\u{036F}' => 0,
36        // Emoji and symbols (width 2)
37        '\u{2600}'..='\u{26FF}' |    // Miscellaneous Symbols
38        '\u{2700}'..='\u{27BF}' |    // Dingbats
39        '\u{1F000}'..='\u{1F02F}' |  // Mahjong Tiles
40        '\u{1F030}'..='\u{1F09F}' |  // Domino Tiles
41        '\u{1F0A0}'..='\u{1F0FF}' |  // Playing Cards
42        '\u{1F100}'..='\u{1F1FF}' |  // Enclosed Alphanumeric Supplement
43        '\u{1F200}'..='\u{1F2FF}' |  // Enclosed Ideographic Supplement
44        '\u{1F300}'..='\u{1F5FF}' |  // Miscellaneous Symbols and Pictographs
45        '\u{1F600}'..='\u{1F64F}' |  // Emoticons
46        '\u{1F650}'..='\u{1F67F}' |  // Ornamental Dingbats
47        '\u{1F680}'..='\u{1F6FF}' |  // Transport and Map Symbols
48        '\u{1F700}'..='\u{1F77F}' |  // Alchemical Symbols
49        '\u{1F780}'..='\u{1F7FF}' |  // Geometric Shapes Extended
50        '\u{1F800}'..='\u{1F8FF}' |  // Supplemental Arrows-C
51        '\u{1F900}'..='\u{1F9FF}' |  // Supplemental Symbols and Pictographs
52        // Full-width characters (common CJK ranges)
53        '\u{1100}'..='\u{115F}' |  // Hangul Jamo
54        '\u{2E80}'..='\u{2EFF}' |  // CJK Radicals
55        '\u{2F00}'..='\u{2FDF}' |  // Kangxi Radicals
56        '\u{2FF0}'..='\u{2FFF}' |  // Ideographic Description
57        '\u{3000}'..='\u{303E}' |  // CJK Symbols and Punctuation
58        '\u{3041}'..='\u{3096}' |  // Hiragana
59        '\u{30A1}'..='\u{30FA}' |  // Katakana
60        '\u{3105}'..='\u{312D}' |  // Bopomofo
61        '\u{3131}'..='\u{318E}' |  // Hangul Compatibility Jamo
62        '\u{3190}'..='\u{31BA}' |  // Kanbun
63        '\u{31C0}'..='\u{31E3}' |  // CJK Strokes
64        '\u{31F0}'..='\u{31FF}' |  // Katakana Phonetic Extensions
65        '\u{3200}'..='\u{32FF}' |  // Enclosed CJK Letters and Months
66        '\u{3300}'..='\u{33FF}' |  // CJK Compatibility
67        '\u{3400}'..='\u{4DBF}' |  // CJK Extension A
68        '\u{4E00}'..='\u{9FFF}' |  // CJK Unified Ideographs
69        '\u{A000}'..='\u{A48C}' |  // Yi Syllables
70        '\u{A490}'..='\u{A4C6}' |  // Yi Radicals
71        '\u{AC00}'..='\u{D7AF}' |  // Hangul Syllables
72        '\u{F900}'..='\u{FAFF}' |  // CJK Compatibility Ideographs
73        '\u{FE10}'..='\u{FE19}' |  // Vertical Forms
74        '\u{FE30}'..='\u{FE6F}' |  // CJK Compatibility Forms
75        '\u{FF00}'..='\u{FF60}' |  // Fullwidth Forms
76        '\u{FFE0}'..='\u{FFE6}' => 2,
77        // Most other printable characters have width 1
78        _ => 1,
79    }
80}
81
82/// Truncate string to specified visual width, preserving color codes
83pub fn truncate_to_width(s: &str, max_width: usize) -> String {
84    let current_visual_width = visual_width(s);
85    if current_visual_width <= max_width {
86        return s.to_string();
87    }
88
89    // For strings with ANSI codes, we need to be more careful
90    if s.contains('\x1b') {
91        // Simple approach: strip ANSI codes, truncate, then re-apply if needed
92        let stripped = strip_ansi_codes(s);
93        if visual_width(&stripped) <= max_width {
94            return s.to_string();
95        }
96
97        // Truncate the stripped version
98        let mut result = String::new();
99        let mut width = 0;
100        for ch in stripped.chars() {
101            let ch_width = char_width(ch);
102            if width + ch_width > max_width.saturating_sub(3) {
103                result.push_str("...");
104                break;
105            }
106            result.push(ch);
107            width += ch_width;
108        }
109        return result;
110    }
111
112    // No ANSI codes - simple truncation
113    let mut result = String::new();
114    let mut width = 0;
115
116    for ch in s.chars() {
117        let ch_width = char_width(ch);
118        if width + ch_width > max_width.saturating_sub(3) {
119            result.push_str("...");
120            break;
121        }
122        result.push(ch);
123        width += ch_width;
124    }
125
126    result
127}
128
129/// Get terminal width, defaulting to 100 if unavailable
130pub fn get_terminal_width() -> usize {
131    term_size::dimensions().map(|(w, _)| w).unwrap_or(100)
132}
133
134/// Smart truncate with single-char ellipsis "…" for cleaner look
135pub fn smart_truncate(s: &str, max_width: usize) -> String {
136    let current_width = visual_width(s);
137    if current_width <= max_width {
138        return s.to_string();
139    }
140
141    // Use single-char ellipsis for cleaner appearance
142    let mut result = String::new();
143    let mut width = 0;
144    let target_width = max_width.saturating_sub(1); // Leave room for "…"
145
146    for ch in strip_ansi_codes(s).chars() {
147        let ch_width = char_width(ch);
148        if width + ch_width > target_width {
149            break;
150        }
151        result.push(ch);
152        width += ch_width;
153    }
154    result.push('…');
155    result
156}
157
158/// Format ports list: deduplicate, limit to max_show, add "+N" if more
159pub fn format_ports_smart(ports: &[u16], max_show: usize) -> String {
160    if ports.is_empty() {
161        return "-".to_string();
162    }
163
164    // Deduplicate and sort
165    let mut unique_ports: Vec<u16> = ports.to_vec();
166    unique_ports.sort_unstable();
167    unique_ports.dedup();
168
169    if unique_ports.len() <= max_show {
170        unique_ports
171            .iter()
172            .map(|p| p.to_string())
173            .collect::<Vec<_>>()
174            .join(", ")
175    } else {
176        let shown: Vec<String> = unique_ports
177            .iter()
178            .take(max_show)
179            .map(|p| p.to_string())
180            .collect();
181        let remaining = unique_ports.len() - max_show;
182        format!("{} +{}", shown.join(", "), remaining)
183    }
184}
185
186/// Format a list of strings smartly: show up to max_show items, add "+N" if more
187/// Each item is truncated to max_item_width if needed
188pub fn format_list_smart(items: &[String], max_show: usize, max_item_width: usize) -> String {
189    if items.is_empty() {
190        return "-".to_string();
191    }
192
193    // Deduplicate while preserving order
194    let mut seen = std::collections::HashSet::new();
195    let unique: Vec<&String> = items
196        .iter()
197        .filter(|item| seen.insert(item.as_str()))
198        .collect();
199
200    if unique.len() <= max_show {
201        unique
202            .iter()
203            .map(|s| {
204                if visual_width(s) > max_item_width {
205                    smart_truncate(s, max_item_width)
206                } else {
207                    s.to_string()
208                }
209            })
210            .collect::<Vec<_>>()
211            .join(", ")
212    } else {
213        let shown: Vec<String> = unique
214            .iter()
215            .take(max_show)
216            .map(|s| {
217                if visual_width(s) > max_item_width {
218                    smart_truncate(s, max_item_width)
219                } else {
220                    s.to_string()
221                }
222            })
223            .collect();
224        let remaining = unique.len() - max_show;
225        format!("{} +{}", shown.join(", "), remaining)
226    }
227}
228
229/// Strip ANSI escape codes from a string
230pub fn strip_ansi_codes(s: &str) -> String {
231    let mut result = String::new();
232    let mut chars = s.chars().peekable();
233
234    while let Some(ch) = chars.next() {
235        if ch == '\x1b' {
236            // Skip ANSI escape sequence
237            if chars.peek() == Some(&'[') {
238                chars.next(); // consume '['
239                for c in chars.by_ref() {
240                    if c.is_ascii_alphabetic() {
241                        break; // End of escape sequence
242                    }
243                }
244            }
245        } else {
246            result.push(ch);
247        }
248    }
249
250    result
251}
252
253#[cfg(test)]
254mod tests {
255    use super::*;
256
257    #[test]
258    fn test_visual_width_basic() {
259        assert_eq!(visual_width("hello"), 5);
260        assert_eq!(visual_width(""), 0);
261        assert_eq!(visual_width("123"), 3);
262    }
263
264    #[test]
265    fn test_visual_width_with_ansi() {
266        assert_eq!(visual_width("\x1b[31mhello\x1b[0m"), 5);
267        assert_eq!(visual_width("\x1b[1;32mtest\x1b[0m"), 4);
268    }
269
270    #[test]
271    fn test_truncate_to_width() {
272        assert_eq!(truncate_to_width("hello world", 5), "he...");
273        assert_eq!(truncate_to_width("hello", 10), "hello");
274        assert_eq!(truncate_to_width("hello world", 8), "hello...");
275    }
276
277    #[test]
278    fn test_strip_ansi_codes() {
279        assert_eq!(strip_ansi_codes("\x1b[31mhello\x1b[0m"), "hello");
280        assert_eq!(strip_ansi_codes("plain text"), "plain text");
281        assert_eq!(
282            strip_ansi_codes("\x1b[1;32mgreen\x1b[0m text"),
283            "green text"
284        );
285    }
286}
syncable_cli/analyzer/display/utils.rs

syncable_cli/analyzer/display/
utils.rs
