ansi-width: Rust中支持ANSI转义字符的字符串宽度计算库

ansi-width是一个专门用于计算包含ANSI转义序列的字符串显示宽度的Rust库。它特别适合终端应用程序开发，能够正确处理彩色文本和各种控制字符的宽度计算。

功能特点

• 支持ANSI转义序列（如颜色代码）
• 正确处理Unicode字符宽度
• 高效计算字符串显示宽度
• 忽略不影响显示的控制字符

安装方法

在Cargo.toml中添加依赖：

[dependencies]
ansi-width = "0.1"

基本使用方法

use ansi_width::ansi_width;

fn main() {
    let text = "\x1b[31m红色\x1b[0m文本";
    let width = ansi_width(text);
    println!("字符串的显示宽度: {}", width); // 输出: 4
}

高级用法

处理多行文本

use ansi_width::ansi_width;

let multiline = "第一行\n\x1b[32m第二行\x1b[0m\n第三行";
for line in multiline.lines() {
    println!("行宽度: {}", ansi_width(line));
}

结合unicode-width处理复杂字符

use ansi_width::ansi_width;

let complex_text = "\x1b[33m🦀\x1b[0m Rust";
println!("宽度: {}", ansi_width(complex_text)); // 输出: 6 (🦀占2个字符宽度)

性能考虑

对于需要频繁计算宽度的场景，可以先将ANSI转义序列剥离：

use ansi_width::{ansi_width, ansi_strip};

let text = "\x1b[1;34m重要信息\x1b[0m";
let stripped = ansi_strip(text);
let width = stripped.chars().map(|c| unicode_width::UnicodeWidthChar::width(c).unwrap_or(0)).sum::<usize>();

实际应用示例

use ansi_width::ansi_width;

fn align_right(text: &str, width: usize) -> String {
    let actual_width = ansi_width(text);
    if actual_width >= width {
        return text.to_string();
    }
    let spaces = " ".repeat(width - actual_width);
    format!("{}{}", spaces, text)
}

let colored_text = "\x1b[35m对齐文本\x1b[0m";
println!("[{}]", align_right(colored_text, 10));

完整示例demo

下面是一个结合多种功能的完整示例：

use ansi_width::{ansi_width, ansi_strip};

fn main() {
    // 基本使用示例
    let colored_text = "\x1b[31mHello\x1b[0m \x1b[32mWorld\x1b[0m";
    println!("基本示例:");
    println!("文本: {}", colored_text);
    println!("显示宽度: {}", ansi_width(colored_text));
    
    // 多行文本处理
    println!("\n多行文本处理:");
    let multiline = "Line1\n\x1b[33mLine2\x1b[0m\nLine3";
    for (i, line) in multiline.lines().enumerate() {
        println!("行{} 宽度: {}", i+1, ansi_width(line));
    }
    
    // 复杂字符处理
    println!("\n复杂字符处理:");
    let emoji_text = "\x1b[34m😊\x1b[0m 表情符号";
    println!("文本: {}", emoji_text);
    println!("显示宽度: {}", ansi_width(emoji_text));
    
    // 性能优化示例
    println!("\n性能优化示例:");
    let performance_text = "\x1b[1;35m性能测试\x1b[0m";
    let stripped = ansi_strip(performance_text);
    println!("原始文本: {}", performance_text);
    println!("剥离ANSI后: {}", stripped);
    
    // 文本对齐应用
    println!("\n文本对齐应用:");
    let align_text = "\x1b[36m对齐演示\x1b[0m";
    println!("[{}]", right_align(align_text, 20));
}

fn right_align(text: &str, width: usize) -> String {
    let actual_width = ansi_width(text);
    if actual_width >= width {
        return text.to_string();
    }
    let spaces = " ".repeat(width - actual_width);
    format!("{}{}", spaces, text)
}

注意事项

1. 某些终端特殊字符可能无法准确计算宽度
2. 组合字符的宽度计算可能因环境而异
3. 对于非常复杂的ANSI序列，建议在实际终端中测试

ansi-width是终端UI开发中的实用工具，特别适合需要精确控制文本对齐和布局的场景。