tui-banner 0.1.2

Colorful ASCII art banner renderer for Rust CLI/TUI
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84

use crate::color::{Color, ColorMode};
use crate::grid::Grid;
use crate::terminal::detect_color_mode;

/// Emit ANSI-colored output from a grid.
pub fn emit_ansi(grid: &Grid, color_mode: ColorMode) -> String {
    let mode = match color_mode {
        ColorMode::Auto => detect_color_mode(),
        other => other,
    };

    let mut out = String::new();
    let mut current_fg: Option<Color> = None;

    for (row_idx, row) in grid.rows().iter().enumerate() {
        for cell in row {
            match mode {
                ColorMode::NoColor => {
                    out.push(cell.ch);
                }
                _ => {
                    if cell.fg != current_fg {
                        if let Some(color) = cell.fg {
                            push_fg_code(&mut out, color, mode);
                        } else {
                            out.push_str("\x1b[0m");
                        }
                        current_fg = cell.fg;
                    }
                    out.push(cell.ch);
                }
            }
        }

        if mode != ColorMode::NoColor && current_fg.is_some() {
            out.push_str("\x1b[0m");
            current_fg = None;
        }

        if row_idx + 1 < grid.height() {
            out.push('\n');
        }
    }

    out
}

fn push_fg_code(out: &mut String, color: Color, mode: ColorMode) {
    match mode {
        ColorMode::TrueColor => match color {
            Color::Rgb(r, g, b) => {
                out.push_str(&format!("\x1b[38;2;{};{};{}m", r, g, b));
            }
            Color::Ansi256(code) => {
                out.push_str(&format!("\x1b[38;5;{}m", code));
            }
        },
        ColorMode::Ansi256 => {
            let code = match color {
                Color::Ansi256(v) => v,
                Color::Rgb(r, g, b) => rgb_to_ansi256(r, g, b),
            };
            out.push_str(&format!("\x1b[38;5;{}m", code));
        }
        _ => {}
    }
}

fn rgb_to_ansi256(r: u8, g: u8, b: u8) -> u8 {
    if r == g && g == b {
        if r < 8 {
            return 16;
        }
        if r > 248 {
            return 231;
        }
        return 232 + ((r as u16 - 8) / 10) as u8;
    }

    let rc = (r as u16 * 5 / 255) as u8;
    let gc = (g as u16 * 5 / 255) as u8;
    let bc = (b as u16 * 5 / 255) as u8;
    16 + 36 * rc + 6 * gc + bc
}