tui-test-rs 0.1.0-beta.2

//! Backend-neutral grid vocabulary.
//!
//! Every consumer of the terminal grid (render, assert, monitor, locator)
//! speaks these types and nothing else, so swapping the emulator backend
//! behind [`crate::terminal::emu::Emulator`] is invisible to them. Nothing in
//! this module may depend on a specific emulator crate.

use bitflags::bitflags;
use compact_str::CompactString;
use unicode_width::UnicodeWidthStr;

/// The 16 themeable palette slots (ANSI 0-15).
///
/// Split out from [`Color::Idx`] by *numeric range*, not by how the escape
/// sequence spelled it, so `SGR 31` and `SGR 38;5;1` both land here.
///
/// The backends do not agree on whether that spelling survives parsing:
/// alacritty keeps it (`Named` vs `Indexed`) and so does xterm.js (`CM_P16`
/// vs `CM_P256`), but ghostty flattens both into one `.palette` value. A model
/// that preserved the distinction would therefore be unimplementable on
/// ghostty. The usual reason to want it, painting bold text bright, does not
/// need it either: ghostty keys that off the index (`bold && idx < 8 =>
/// palette[idx + 8]`), which applies to `38;5;1` just as much as to `31`.
///
/// What every supported backend agrees on is that 0-15 are the slots a theme
/// may override, which is the only distinction any consumer here acts on.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[repr(u8)]
pub enum NamedColor {
    Black = 0,
    Red,
    Green,
    Yellow,
    Blue,
    Magenta,
    Cyan,
    White,
    BrightBlack,
    BrightRed,
    BrightGreen,
    BrightYellow,
    BrightBlue,
    BrightMagenta,
    BrightCyan,
    BrightWhite,
}

impl NamedColor {
    pub const ALL: [NamedColor; 16] = [
        NamedColor::Black,
        NamedColor::Red,
        NamedColor::Green,
        NamedColor::Yellow,
        NamedColor::Blue,
        NamedColor::Magenta,
        NamedColor::Cyan,
        NamedColor::White,
        NamedColor::BrightBlack,
        NamedColor::BrightRed,
        NamedColor::BrightGreen,
        NamedColor::BrightYellow,
        NamedColor::BrightBlue,
        NamedColor::BrightMagenta,
        NamedColor::BrightCyan,
        NamedColor::BrightWhite,
    ];

    /// The palette slot this name occupies (0-15).
    pub fn index(self) -> u8 {
        self as u8
    }

    /// The name for a palette slot, or `None` outside 0-15.
    pub fn from_index(i: u8) -> Option<Self> {
        Self::ALL.get(i as usize).copied()
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Color {
    /// A themeable palette slot, ANSI 0-15.
    Named(NamedColor),
    /// A fixed 256-color palette index, 16-255.
    Idx(u8),
    Rgb(u8, u8, u8),
}

impl Color {
    /// Build from a 256-color index, routing 0-15 to [`Color::Named`] so the
    /// same index yields the same value no matter which backend produced it.
    pub fn from_index(i: u8) -> Self {
        match NamedColor::from_index(i) {
            Some(named) => Color::Named(named),
            None => Color::Idx(i),
        }
    }

    /// The 256-color index for this color, approximating RGB.
    pub fn to_index(self) -> u8 {
        match self {
            Color::Named(n) => n.index(),
            Color::Idx(i) => i,
            Color::Rgb(r, g, b) => crate::assert::color::rgb_to_ansi256(r, g, b),
        }
    }
}

/// The shape of a cell's underline.
///
/// [`UnderlineStyle::None`] is a value, not an absence: it is the shape an
/// un-underlined cell has. Wrapping this in an `Option` would give two ways to
/// spell "not underlined" and force every reader through a `map` to reach the
/// shape, which is why the style and its color sit flat on the cell rather
/// than inside a nested struct.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub enum UnderlineStyle {
    #[default]
    None,
    Single,
    Double,
    Curly,
    Dotted,
    Dashed,
}

impl UnderlineStyle {
    /// Is the cell underlined at all?
    pub const fn is_underlined(self) -> bool {
        !matches!(self, UnderlineStyle::None)
    }

    /// The name this style goes by on the wire.
    pub const fn name(self) -> &'static str {
        match self {
            UnderlineStyle::None => "none",
            UnderlineStyle::Single => "single",
            UnderlineStyle::Double => "double",
            UnderlineStyle::Curly => "curly",
            UnderlineStyle::Dotted => "dotted",
            UnderlineStyle::Dashed => "dashed",
        }
    }
}

bitflags! {
    /// Boolean SGR attributes, one bit each: the render and monitor loops copy
    /// and compare a cell's style per column, so keeping it to a single byte
    /// keeps those comparisons to a single integer compare.
    #[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
    pub struct Attrs: u8 {
        const BOLD      = 1 << 0;
        const DIM       = 1 << 1;
        const ITALIC    = 1 << 2;
        const INVERSE   = 1 << 3;
        const INVISIBLE = 1 << 4;
        const STRIKE    = 1 << 5;
        const BLINK     = 1 << 6;
    }
}

/// The grapheme stored in the cell that follows a double-width character.
pub const CONTINUATION: &str = "";

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct EmuCell {
    /// The cell's grapheme. A blank cell holds `" "`; [`CONTINUATION`] (the
    /// empty string) is reserved for the cell trailing a double-width char.
    pub ch: CompactString,
    /// `None` means the terminal's default foreground.
    pub fg: Option<Color>,
    /// `None` means the terminal's default background.
    pub bg: Option<Color>,
    pub underline: UnderlineStyle,
    /// `None` means the underline takes the cell's foreground color. Carried
    /// even when there is no underline, the same way `fg` outlives the
    /// grapheme it colors.
    pub underline_color: Option<Color>,
    pub attrs: Attrs,
}

impl EmuCell {
    /// A blank, unstyled cell.
    pub const fn blank() -> Self {
        EmuCell {
            ch: CompactString::const_new(" "),
            fg: None,
            bg: None,
            underline: UnderlineStyle::None,
            underline_color: None,
            attrs: Attrs::empty(),
        }
    }

    pub fn has(&self, attr: Attrs) -> bool {
        self.attrs.contains(attr)
    }
}

impl Default for EmuCell {
    fn default() -> Self {
        EmuCell::blank()
    }
}

/// How many terminal columns a string occupies.
///
/// A terminal lays text out by column, not by character: a CJK glyph is one
/// `char` but two columns, and a combining mark is one `char` but none.
///
/// Measured over the whole string rather than by summing characters, because
/// a sequence can be narrower than its parts: an emoji joined by zero-width
/// joiners (`\u{200d}`) renders as a single glyph, and a base character
/// followed by a variation selector or a keycap mark is one unit too. Summing
/// per character reports a family emoji as eight columns where a terminal
/// draws two.
pub fn display_width(s: &str) -> usize {
    s.width()
}

/// Shorten `s` to at most `columns` terminal columns, marking a cut with `…`.
///
/// The cut point is found by measuring real prefixes rather than by adding up
/// character widths, so the result is exactly as wide as it was measured to be
/// even when the cut lands inside a sequence.
pub fn truncate_to_columns(s: &str, columns: usize) -> String {
    if columns == 0 {
        return String::new();
    }
    if display_width(s) <= columns {
        return s.to_string();
    }
    // One column is held back for the ellipsis that marks the cut.
    let budget = columns - 1;
    let mut cut = 0;
    for (offset, _) in s.char_indices() {
        if display_width(&s[..offset]) > budget {
            break;
        }
        cut = offset;
    }
    format!("{}\u{2026}", &s[..cut])
}

/// Join a grid of cells into one string per row.
///
/// Continuation cells contribute nothing: a double-width character already
/// carries both its columns, so emitting a filler for the second one would
/// widen the row by one and shift every column after it.
pub fn rows_to_strings(rows: &[Vec<EmuCell>]) -> Vec<String> {
    rows.iter()
        .map(|row| row.iter().map(|c| c.ch.as_str()).collect::<String>())
        .collect()
}

#[cfg(test)]
mod tests {
    use super::*;

    fn cell(s: &str) -> EmuCell {
        EmuCell {
            ch: CompactString::from(s),
            ..EmuCell::blank()
        }
    }

    #[test]
    fn index_splits_named_from_palette() {
        assert_eq!(Color::from_index(0), Color::Named(NamedColor::Black));
        assert_eq!(Color::from_index(9), Color::Named(NamedColor::BrightRed));
        assert_eq!(Color::from_index(15), Color::Named(NamedColor::BrightWhite));
        assert_eq!(Color::from_index(16), Color::Idx(16));
        assert_eq!(Color::from_index(255), Color::Idx(255));
        for i in 0..=255u8 {
            assert_eq!(Color::from_index(i).to_index(), i, "roundtrip {i}");
        }
    }

    #[test]
    fn continuation_cells_do_not_widen_a_row() {
        let rows = vec![vec![cell("你"), cell(CONTINUATION), cell("a"), cell(" ")]];
        assert_eq!(rows_to_strings(&rows), vec!["你a "]);
    }

    #[test]
    fn blank_is_a_space_not_a_continuation() {
        assert_eq!(EmuCell::blank().ch, " ");
        assert_ne!(EmuCell::blank().ch, CONTINUATION);
    }
}

#[cfg(test)]
mod width_tests {
    use super::*;

    /// A sequence is measured as the glyph it renders as, not as the sum of
    /// its parts. Summing per character is the mistake that draws a frame
    /// around a family emoji six columns too wide.
    #[test]
    fn a_sequence_is_narrower_than_its_characters() {
        for (name, text, columns) in [
            (
                "family",
                "\u{1f468}\u{200d}\u{1f469}\u{200d}\u{1f467}\u{200d}\u{1f466}",
                2,
            ),
            ("skin tone", "\u{1f44d}\u{1f3fd}", 2),
            ("keycap", "1\u{fe0f}\u{20e3}", 2),
            ("heart with a variation selector", "\u{2764}\u{fe0f}", 2),
            ("flag", "\u{1f1fa}\u{1f1f8}", 2),
        ] {
            assert_eq!(display_width(text), columns, "{name} is {columns} columns");
        }
    }

    /// The ordinary cases the grid already relies on.
    #[test]
    fn width_counts_columns_not_characters() {
        assert_eq!(display_width("hello"), 5);
        assert_eq!(
            display_width("\u{4f60}\u{597d}"),
            4,
            "each CJK glyph takes two"
        );
        assert_eq!(display_width("e\u{301}"), 1, "a combining mark adds none");
        assert_eq!(display_width(""), 0);
    }

    /// Truncation never exceeds its budget, whatever it has to cut through.
    ///
    /// The result is measured rather than assumed: a cut inside a sequence
    /// changes how the remainder renders, so only measuring the real prefix
    /// keeps the promise this function makes to a frame drawn around it.
    #[test]
    fn truncation_stays_within_its_budget() {
        for text in [
            "a-very-long-title-that-will-not-fit",
            "\u{4f60}\u{597d}\u{4e16}\u{754c}\u{4f60}\u{597d}",
            "\u{1f468}\u{200d}\u{1f469}\u{200d}\u{1f467}\u{200d}\u{1f466} building",
            "\u{1f680} deploy \u{4f60}\u{597d} done",
        ] {
            for budget in 0..12 {
                let cut = truncate_to_columns(text, budget);
                assert!(
                    display_width(&cut) <= budget,
                    "{text:?} cut to {budget} came out {} wide: {cut:?}",
                    display_width(&cut)
                );
            }
        }
    }

    /// A string that already fits is returned whole, with no ellipsis.
    #[test]
    fn truncation_leaves_a_string_that_fits_alone() {
        assert_eq!(truncate_to_columns("fits", 10), "fits");
        assert_eq!(truncate_to_columns("fits", 4), "fits");
        assert_eq!(
            truncate_to_columns("\u{4f60}\u{597d}", 4),
            "\u{4f60}\u{597d}"
        );
    }
}