nativelite-ansi 0.4.1

//! A screen buffer and diff-based rendering: build the frame you want as a
//! grid of styled cells, then emit only the bytes that change what the
//! terminal is already showing.

use crate::history::History;
use crate::style::Style;

/// How many terminal columns a [`Cell`] covers, modeling East Asian
/// double-width glyphs and emoji, which occupy two columns.
///
/// This crate does not *compute* width — that needs the Unicode database, which
/// a zero-dependency crate cannot own. Callers pick the width (e.g. from the
/// `uwidth` crate); the renderers here honor whatever they are given.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub enum CellWidth {
    /// The right half of the double-width glyph to its left: it carries no
    /// character of its own and is never emitted, because the terminal advances
    /// the cursor by two columns when it draws the lead glyph.
    Continuation,
    /// A normal single-width cell (the common case, and the default).
    #[default]
    Single,
    /// The left half (lead) of a double-width glyph; the cell holds the
    /// character, and the cell immediately to its right should be a
    /// [`CellWidth::Continuation`].
    Wide,
}

impl CellWidth {
    /// The columns this cell advances the terminal cursor when drawn: `0`, `1`
    /// or `2`.
    pub const fn columns(self) -> usize {
        match self {
            CellWidth::Continuation => 0,
            CellWidth::Single => 1,
            CellWidth::Wide => 2,
        }
    }
}

/// One character cell: a `char`, its [`Style`], and its [`CellWidth`]. The
/// default cell is a single-width space in the default style.
///
/// A double-width glyph is a [`Cell::wide`] lead followed by a
/// [`Cell::continuation`]; build cells with the constructors to keep that
/// pairing. A grid can still hold an unpaired half (a compositor clipping a
/// row through a glyph produces one), so readers of arbitrary cells should
/// expect it.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct Cell {
    pub ch: char,
    pub style: Style,
    pub width: CellWidth,
}

impl Default for Cell {
    fn default() -> Self {
        Cell {
            ch: ' ',
            style: Style::default(),
            width: CellWidth::Single,
        }
    }
}

impl Cell {
    /// A normal single-width cell.
    pub const fn new(ch: char, style: Style) -> Self {
        Cell {
            ch,
            style,
            width: CellWidth::Single,
        }
    }

    /// The lead (left half) of a double-width glyph. Its right neighbor should
    /// be a [`Cell::continuation`].
    pub const fn wide(ch: char, style: Style) -> Self {
        Cell {
            ch,
            style,
            width: CellWidth::Wide,
        }
    }

    /// A continuation (right half) of a double-width glyph: no character of its
    /// own, but carrying `style` so the covered column keeps the right
    /// background. Never emitted — the lead glyph fills both columns.
    pub const fn continuation(style: Style) -> Self {
        Cell {
            ch: ' ',
            style,
            width: CellWidth::Continuation,
        }
    }
}

/// A 0-based cursor position on a [`Screen`]. Named fields, so a row and a
/// column cannot be transposed the way a bare `(usize, usize)` can.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
pub struct Cursor {
    pub row: usize,
    pub col: usize,
}

impl Cursor {
    pub const fn new(row: usize, col: usize) -> Self {
        Cursor { row, col }
    }
}

/// What a physical row's written cells are, so that storing it in the
/// history need not look at each cell.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum Plain {
    /// Nothing written.
    Empty,
    /// Every written cell is a single-width character below U+0100 in this
    /// style, one byte each in `Screen::text`.
    Bytes(Style),
    /// Anything else.
    Mixed,
}

/// A `rows x cols` grid of [`Cell`]s plus a [`Cursor`], both 0-based.
#[derive(Debug, Clone)]
pub struct Screen {
    rows: usize,
    cols: usize,
    /// Row-major, and exactly `rows * cols` long: sized by
    /// [`Screen::new`] / [`Screen::with_history`] and never grown or shrunk
    /// afterwards (`clear` refills in place, `set` and
    /// `copy_cells` overwrite in bounds). The indexers check it in debug builds.
    ///
    /// Rows are reached through a **row map**: logical row `r` lives at
    /// physical row `map[r]`, so scrolling any region rotates a few row
    /// indices instead of moving cells. Nothing outside [`Screen::index`]
    /// knows this.
    ///
    /// Rows are also **cleared lazily**: physical row `p` holds real cells
    /// only in its first `written[p]` columns; every column past that reads
    /// as `blank[p]`. Scrolling a row in or clearing it just resets those two
    /// (O(1)), where filling every cell cost ~half of all emulation time on a
    /// 177-column pane of short agent lines. Writes fill any gap first.
    cells: Vec<Cell>,
    /// Logical row to physical row.
    map: Vec<usize>,
    /// **Scrollback, stored compactly.** When the whole screen scrolls up,
    /// the row leaving the top is encoded (its characters in UTF-8 plus
    /// runs of style) and appended here, and its physical row comes back in
    /// at the bottom, lazily blank. (It was a ring of full rows: 3.5 KB a
    /// line on a 177-column screen, 35 MB for 10,000 lines.)
    hist: History,
    /// Only with history: per physical row, whether its written cells are
    /// plain [`Plain::Bytes`], whose characters `text` (`rows * cols`)
    /// mirrors. Such a row, the usual one, goes into the history as a copy
    /// of its bytes rather than a scan of its cells (~110 ns a line of
    /// 48 cells for the scan).
    plain: Vec<Plain>,
    text: Vec<u8>,
    /// Per physical row: how many leading cells are real.
    written: Vec<usize>,
    /// Per physical row: what every cell past `written` reads as.
    blank: Vec<Cell>,
    /// Per physical row: it wrapped onto the next row (the emulator's
    /// autowrap) rather than ending in a line break of its own, so copying
    /// joins the two. Cleared when the row is cleared or scrolled in.
    wrapped: Vec<bool>,
    /// Where the cursor should rest after rendering. Always on the grid (see
    /// [`Screen::set_cursor`]).
    cursor: Cursor,
}

/// Two screens are equal when they *show* the same thing: same size, same
/// cursor, same cell at every logical position. Where each row physically sits
/// in the ring is an implementation detail, so a scrolled screen and a freshly
/// built one with the same content compare equal.
impl PartialEq for Screen {
    fn eq(&self, other: &Self) -> bool {
        if self.rows != other.rows || self.cols != other.cols || self.cursor != other.cursor {
            return false;
        }
        (0..self.rows).all(|r| {
            self.row_wrapped(r) == other.row_wrapped(r)
                && (0..self.cols).all(|c| self.cell(r, c) == other.cell(r, c))
        })
    }
}

impl Eq for Screen {}

impl Screen {
    /// A blank screen (all default cells, cursor at the origin).
    pub fn new(rows: usize, cols: usize) -> Self {
        Screen::with_history(rows, cols, 0)
    }

    /// A blank screen that keeps up to `history` lines scrolled off its top
    /// (see [`Screen::history_len`]). Only a scroll of the whole screen feeds
    /// the history, as in other terminals; a scroll region does not.
    pub fn with_history(rows: usize, cols: usize, history: usize) -> Self {
        Screen {
            rows,
            cols,
            cells: vec![Cell::default(); rows * cols],
            map: (0..rows).collect(),
            hist: History::new(history),
            plain: vec![Plain::Empty; if history > 0 { rows } else { 0 }],
            text: vec![0; if history > 0 { rows * cols } else { 0 }],
            written: vec![0; rows],
            blank: vec![Cell::default(); rows],
            wrapped: vec![false; rows],
            cursor: Cursor::default(),
        }
    }

    /// The same visible rows and cursor, without the history: what a
    /// renderer or a synchronized-output snapshot needs, at the cost of the
    /// visible cells only.
    pub fn visible(&self) -> Screen {
        let mut s = Screen::new(self.rows, self.cols);
        for r in 0..self.rows {
            let p = self.map[r];
            let w = self.written[p];
            s.blank[r] = self.blank[p];
            s.written[r] = w;
            s.wrapped[r] = self.wrapped[p];
            let (src, dst) = (p * self.cols, r * self.cols);
            s.cells[dst..dst + w].copy_from_slice(&self.cells[src..src + w]);
        }
        s.cursor = self.cursor;
        s
    }

    /// Lines of history held (at most the capacity given to
    /// [`Screen::with_history`]).
    pub fn history_len(&self) -> usize {
        self.hist.len()
    }

    /// The history capacity, in lines.
    pub fn history_capacity(&self) -> usize {
        self.hist.capacity()
    }

    /// Hold at most `max` bytes of memory in history (0: no cap), dropping
    /// the oldest lines past it: a bound however wide or colourful lines
    /// are, on top of the line capacity. It may run over by up to one
    /// 96 KiB chunk.
    pub fn set_history_bytes(&mut self, max: usize) {
        self.hist.set_max_bytes(max);
    }

    /// Memory the history holds: its buffers' capacity.
    pub fn history_bytes(&self) -> usize {
        self.hist.bytes()
    }

    /// Lines that have ever entered the history, including ones since
    /// dropped: a monotonic count to anchor a scrolled-back view on.
    pub fn scrolled_lines(&self) -> u64 {
        self.hist.pushed()
    }

    /// A cell of the history: `age` 0 is the line that most recently left
    /// the top of the screen. Out of range returns a default cell. Each call
    /// decodes into the stored line; for whole rows use
    /// [`Screen::history_row`].
    pub fn history_cell(&self, age: usize, col: usize) -> Cell {
        if col >= self.cols {
            return Cell::default();
        }
        self.hist.cell(age, col).unwrap_or_default()
    }

    /// History line `age` (0 the newest) as this screen's `cols` cells, into
    /// `out`, replacing its contents. Returns false, leaving `out` empty,
    /// when the line is not held.
    pub fn history_row(&self, age: usize, out: &mut Vec<Cell>) -> bool {
        self.hist.row(age, self.cols, out)
    }

    /// Whether history line `age` (0 the newest) wrapped onto the next line;
    /// false when not held.
    pub fn history_wrapped(&self, age: usize) -> bool {
        self.hist.wrapped(age)
    }

    /// Whether `row` wrapped onto the row below (autowrap), so the two are
    /// one line of output. Out of range is false.
    pub fn row_wrapped(&self, row: usize) -> bool {
        row < self.rows && self.wrapped[self.physical(row)]
    }

    /// Mark `row` as wrapped onto the row below, or not; the emulator sets
    /// it when a line autowraps. Out of range is ignored.
    pub fn set_row_wrapped(&mut self, row: usize, wrapped: bool) {
        if row < self.rows {
            let p = self.physical(row);
            self.wrapped[p] = wrapped;
        }
    }

    /// Forget the history (as `ESC [3J` asks).
    pub fn clear_history(&mut self) {
        self.hist.clear();
    }

    /// Copy `other`'s history (oldest first) into this screen's, each line
    /// cut or padded to this width: used to keep scrollback across a resize.
    pub fn copy_history_from(&mut self, other: &Screen) {
        let cols = self.cols.min(other.cols);
        let mut row = Vec::with_capacity(other.cols);
        for age in (0..other.history_len()).rev() {
            other.history_row(age, &mut row);
            row.truncate(cols);
            self.hist
                .push(&row, Cell::default(), other.history_wrapped(age));
        }
    }

    pub fn rows(&self) -> usize {
        self.rows
    }

    pub fn cols(&self) -> usize {
        self.cols
    }

    /// Where the cursor rests after rendering.
    #[inline]
    pub fn cursor(&self) -> Cursor {
        self.cursor
    }

    /// Move the cursor, clamped onto the grid: a row or column past the edge
    /// lands on the last row or column (the origin on an empty screen). The
    /// renderers park the terminal cursor here.
    #[inline]
    pub fn set_cursor(&mut self, cursor: Cursor) {
        self.cursor = Cursor {
            row: cursor.row.min(self.rows.saturating_sub(1)),
            col: cursor.col.min(self.cols.saturating_sub(1)),
        };
    }

    /// Reset all cells to default and the cursor to the origin, reusing the
    /// existing allocation. The caller must ensure the screen is already the
    /// right size; use [`Screen::new`] when the size changes.
    pub fn clear(&mut self) {
        // Visible rows only; any history stays (see `clear_history`).
        for r in 0..self.rows {
            self.fill_row(r, Cell::default());
        }
        self.cursor = Cursor::default();
    }

    /// The flat index of `(row, col)`, which the caller has bounds-checked.
    /// Logical row `row` sits at physical row `map[row]`.
    #[inline]
    fn index(&self, row: usize, col: usize) -> usize {
        debug_assert_eq!(self.cells.len(), self.rows * self.cols);
        debug_assert!(row < self.rows && col < self.cols);
        self.physical(row) * self.cols + col
    }

    /// The physical row holding logical `row`.
    #[inline]
    fn physical(&self, row: usize) -> usize {
        self.map[row]
    }

    /// Make the first `upto` cells of physical row `p` real, filling the
    /// gap past `written[p]` with the row's blank.
    #[inline]
    fn materialize(&mut self, p: usize, upto: usize) {
        let w = self.written[p];
        if w < upto {
            let base = p * self.cols;
            let blank = self.blank[p];
            self.cells[base + w..base + upto].fill(blank);
            self.written[p] = upto;
            if self.track(p, blank) {
                self.text[base + w..base + upto].fill(blank.ch as u8);
            }
        }
    }

    /// With history, fold a write of cells like `cell` into physical row
    /// `p`'s [`Plain`] state; true if the row stays plain, and the caller
    /// then mirrors the characters into `text`.
    #[inline]
    fn track(&mut self, p: usize, cell: Cell) -> bool {
        let Some(state) = self.plain.get_mut(p) else {
            return false;
        };
        let byte = cell.width == CellWidth::Single && (cell.ch as u32) < 0x100;
        *state = match *state {
            _ if !byte => Plain::Mixed,
            Plain::Empty => Plain::Bytes(cell.style),
            Plain::Bytes(s) if s == cell.style => Plain::Bytes(s),
            _ => Plain::Mixed,
        };
        *state != Plain::Mixed
    }

    /// The cell at `(row, col)`; out of bounds returns a default cell.
    pub fn cell(&self, row: usize, col: usize) -> Cell {
        if row < self.rows && col < self.cols {
            let p = self.physical(row);
            if col < self.written[p] {
                self.cells[self.index(row, col)]
            } else {
                self.blank[p]
            }
        } else {
            Cell::default()
        }
    }

    /// Row `row` without a copy: the cells written so far, and the cell every
    /// column past them reads as (the row's blank). The two together are
    /// what [`Screen::cell`] returns column by column, so a compositor or a
    /// diff walks a row as one slice and one fill instead of `cols` lookups.
    /// Out of range: an empty slice and a default cell.
    pub fn row_cells(&self, row: usize) -> (&[Cell], Cell) {
        if row >= self.rows {
            return (&[], Cell::default());
        }
        let p = self.physical(row);
        let base = p * self.cols;
        (&self.cells[base..base + self.written[p]], self.blank[p])
    }

    /// Set one cell. Out-of-bounds writes are ignored.
    pub fn set(&mut self, row: usize, col: usize, cell: Cell) {
        if row < self.rows && col < self.cols {
            let p = self.physical(row);
            self.materialize(p, col + 1);
            let i = self.index(row, col);
            self.cells[i] = cell;
            if self.track(p, cell) {
                self.text[i] = cell.ch as u8;
            }
        }
    }

    /// Write a string across a row starting at `(row, col)`, truncating at
    /// the right edge. Each `char` takes one cell (single-width); callers that
    /// need double-width layout use [`Cell::wide`] with [`Screen::set`] or
    /// [`Screen::copy_cells`].
    pub fn write_str(&mut self, row: usize, col: usize, text: &str, style: Style) {
        for (i, ch) in text.chars().enumerate() {
            self.set(row, col + i, Cell::new(ch, style));
        }
    }

    /// Copy a contiguous run of cells into `row` starting at column `col`,
    /// clipping the run at the right edge (and ignoring an out-of-range row or
    /// start column). Equivalent to calling [`Screen::set`] for each cell in
    /// turn, but with a single bounds check for the whole run — the fast path a
    /// compositor uses to blit a row slice.
    pub fn copy_cells(&mut self, row: usize, col: usize, cells: &[Cell]) {
        if row >= self.rows || col >= self.cols {
            return;
        }
        let n = cells.len().min(self.cols - col);
        let p = self.physical(row);
        self.materialize(p, col);
        let base = self.index(row, col);
        self.cells[base..base + n].copy_from_slice(&cells[..n]);
        self.written[p] = self.written[p].max(col + n);
        if let Some(state) = self.plain.get_mut(p) {
            *state = Plain::Mixed;
        }
    }

    /// Write ASCII `text` into `row` from column `col` in one pass, each byte
    /// a single-width cell in `style`, clipped at the right edge (an
    /// out-of-range row or column writes nothing). Returns the cells written.
    /// The same as [`Screen::copy_cells`] of `Cell::new(b as char, style)`
    /// for each byte, without building the cells first — the hot path of a
    /// terminal printing plain text. Bytes are not checked: callers pass
    /// printable ASCII.
    pub fn write_ascii(&mut self, row: usize, col: usize, text: &[u8], style: Style) -> usize {
        if row >= self.rows || col >= self.cols {
            return 0;
        }
        let n = text.len().min(self.cols - col);
        let p = self.physical(row);
        if col == 0 && n >= self.written[p] {
            // Everything written is replaced.
            if let Some(state) = self.plain.get_mut(p) {
                *state = Plain::Empty;
            }
        }
        self.materialize(p, col);
        let base = self.index(row, col);
        for (cell, &b) in self.cells[base..base + n].iter_mut().zip(text) {
            *cell = Cell::new(b as char, style);
        }
        self.written[p] = self.written[p].max(col + n);
        if n > 0 && self.track(p, Cell::new(' ', style)) {
            self.text[base..base + n].copy_from_slice(&text[..n]);
        }
        n
    }

    /// Scroll rows `top..=bottom` up by `n`. Rows move up, the top `n` rows of
    /// the region are lost, and `n` rows of `fill` enter at the bottom. Rows
    /// outside the region are untouched. An empty or out-of-range region does
    /// nothing, and `n` is clamped to the region's height.
    ///
    /// Any region, the whole screen included, scrolls by rotating its entries
    /// in the row map and clearing the rows that come in: `region` index moves
    /// and O(1) per incoming row, never `region x cols`.
    ///
    /// A terminal scrolls on every output line. Copying the region cell by cell
    /// fed a 16 MB log at 11 MB/s on a 40x160 grid; moving whole rows still
    /// held a 177x47 pane with a one-row status line (`DECSTBM`) to 21 MiB/s,
    /// against 182 MiB/s scrolling the full screen.
    pub fn scroll_rows_up(&mut self, top: usize, bottom: usize, n: usize, fill: Cell) {
        if top > bottom || bottom >= self.rows || n == 0 {
            return;
        }
        let n = n.min(bottom - top + 1);
        if self.hist.capacity() > 0 && top == 0 && bottom == self.rows - 1 {
            // The whole screen: each departing row is stored in the history
            // before its physical row comes back in at the bottom.
            for r in 0..n {
                let p = self.map[r];
                let (from, to) = (p * self.cols, p * self.cols + self.written[p]);
                let wrapped = self.wrapped[p];
                match self.plain[p] {
                    Plain::Bytes(style) => {
                        self.hist
                            .push_ascii(&self.text[from..to], style, self.blank[p], wrapped)
                    }
                    Plain::Empty => {
                        self.hist
                            .push_ascii(&[], Style::default(), self.blank[p], wrapped)
                    }
                    Plain::Mixed => self
                        .hist
                        .push(&self.cells[from..to], self.blank[p], wrapped),
                }
            }
        }
        self.map[top..=bottom].rotate_left(n);
        for r in bottom + 1 - n..=bottom {
            self.fill_row(r, fill);
        }
    }

    /// Fill one logical row with `fill`: O(1), by making the whole row read
    /// as `fill` (see `written`).
    #[inline]
    fn fill_row(&mut self, row: usize, fill: Cell) {
        let p = self.physical(row);
        self.written[p] = 0;
        self.blank[p] = fill;
        self.wrapped[p] = false;
        if let Some(state) = self.plain.get_mut(p) {
            *state = Plain::Empty;
        }
    }

    /// Scroll rows `top..=bottom` down by `n`: the mirror of
    /// [`Screen::scroll_rows_up`]. The bottom `n` rows of the region are lost,
    /// and `n` rows of `fill` enter at the top.
    pub fn scroll_rows_down(&mut self, top: usize, bottom: usize, n: usize, fill: Cell) {
        if top > bottom || bottom >= self.rows || n == 0 {
            return;
        }
        let n = n.min(bottom - top + 1);
        self.map[top..=bottom].rotate_right(n);
        for r in top..top + n {
            self.fill_row(r, fill);
        }
    }

    /// The bytes that transform a terminal currently showing `self` into one
    /// showing `next`: cursor moves (CUP), minimal SGR transitions, and the
    /// changed characters, ending with a style reset and the cursor parked at
    /// `next`'s cursor. Equal screens produce no bytes at all.
    ///
    /// Both screens must have the same dimensions; if they differ (a resize),
    /// the diff falls back to a full repaint of `next`.
    pub fn diff(&self, next: &Screen) -> Vec<u8> {
        if self.rows != next.rows || self.cols != next.cols {
            return next.render_full();
        }
        if self == next {
            return Vec::new();
        }
        let mut out = String::new();
        emit_changes(Some(self), next, &mut out);
        out.into_bytes()
    }

    /// A from-scratch repaint: clear the screen, then paint every non-default
    /// cell, ending with a style reset and the cursor parked at
    /// this screen's cursor.
    pub fn render_full(&self) -> Vec<u8> {
        let mut out = String::from("\x1b[2J");
        emit_changes(None, self, &mut out);
        out.into_bytes()
    }
}

/// Emit the cell changes from `prev` to `next` (`None` means "cleared
/// screen": every non-default cell of `next` is painted).
///
/// Row suffix optimisation: when the trailing cells of a row are all default
/// in `next` but were not in `prev`, a single `\x1b[K]` (Erase to EOL)
/// replaces one write-per-blank-cell. Style is reset before the erase so the
/// terminal fills with the default background.
fn emit_changes(prev: Option<&Screen>, next: &Screen, out: &mut String) {
    use std::fmt::Write;
    let mut style = Style::default();
    // Where the terminal cursor is right now, when known. Printing advances
    // it; at the right edge the position becomes ambiguous (pending wrap),
    // so it is dropped to force an explicit move next time.
    let mut at: Option<(usize, usize)> = None;
    for row in 0..next.rows {
        // Find the exclusive right bound of the non-default suffix in `next`.
        // Columns in [last_content, cols) are all Cell::default() in `next`.
        let last_content = (0..next.cols)
            .rev()
            .find(|&c| next.cell(row, c) != Cell::default())
            .map_or(0, |c| c + 1);

        // Emit changed cells up to (but not including) the trailing blank region.
        for col in 0..last_content {
            let target = next.cell(row, col);
            // A width-0 cell is the right half (continuation) of a double-width
            // glyph: it carries no character of its own and emits nothing. The
            // lead cell to its left drew the glyph and the terminal advanced the
            // cursor across both columns.
            if target.width == CellWidth::Continuation {
                continue;
            }
            let same = match prev {
                Some(p) => p.cell(row, col) == target,
                None => target == Cell::default(),
            };
            if same {
                continue;
            }
            if at != Some((row, col)) {
                write!(out, "\x1b[{};{}H", row + 1, col + 1).unwrap();
            }
            out.push_str(&style.transition_to(&target.style));
            style = target.style;
            out.push(target.ch);
            // The terminal advances the cursor by the glyph's column width, so a
            // width-2 lead lands `at` two columns on and its continuation column
            // is then skipped. (A single-width cell advances by one, exactly as
            // before — ASCII output is byte-for-byte unchanged.) At or past the
            // right edge the position is ambiguous (pending wrap), so drop it to
            // force an explicit move next time.
            let advance = target.width.columns();
            at = if col + advance < next.cols {
                Some((row, col + advance))
            } else {
                None
            };
        }

        // If prev had non-default content in the trailing blank region of next,
        // erase it with a single EL rather than writing each blank cell.
        // Not used in render_full (prev == None) because `\x1b[2J` already
        // cleared the screen before emit_changes is called.
        if last_content < next.cols {
            let needs_erase = match prev {
                Some(p) => (last_content..next.cols).any(|c| p.cell(row, c) != Cell::default()),
                None => false,
            };
            if needs_erase {
                // Reset SGR before EL so the terminal erases with the default
                // background, not whatever the last cell's style was.
                out.push_str(&style.transition_to(&Style::default()));
                style = Style::default();
                if at != Some((row, last_content)) {
                    write!(out, "\x1b[{};{}H", row + 1, last_content + 1).unwrap();
                }
                out.push_str("\x1b[K");
                // Cursor stays at (row, last_content); next row needs an explicit
                // move regardless, so mark position unknown.
                at = None;
            }
        }
    }
    out.push_str(&style.transition_to(&Style::default()));
    let Cursor { row, col } = next.cursor;
    write!(out, "\x1b[{};{}H", row + 1, col + 1).unwrap();
}

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

    #[test]
    fn row_cells_agrees_with_cell_before_and_after_scrolling() {
        use crate::Color;
        let mut s = Screen::with_history(4, 10, 8);
        let red = Style {
            fg: Color::Indexed(1),
            ..Style::default()
        };
        s.write_str(0, 0, "abc", Style::default());
        s.write_str(1, 2, "de", red);
        s.copy_cells(
            3,
            0,
            &[Cell::wide('\u{4e2d}', red), Cell::continuation(red)],
        );
        let check = |s: &Screen| {
            for r in 0..s.rows() {
                let (written, fill) = s.row_cells(r);
                assert!(written.len() <= s.cols());
                for c in 0..s.cols() {
                    let want = s.cell(r, c);
                    let got = written.get(c).copied().unwrap_or(fill);
                    assert_eq!(got, want, "row {r} col {c}");
                }
            }
        };
        check(&s);
        // Rows written in full, then scrolled: the row map moves, the
        // slices follow.
        s.write_str(2, 0, "0123456789", red);
        check(&s);
        s.scroll_rows_up(0, 3, 1, Cell::new('~', red));
        check(&s);
        let (written, fill) = s.row_cells(3);
        assert!(written.is_empty(), "a scrolled-in row is lazily blank");
        assert_eq!(fill, Cell::new('~', red));
        assert_eq!(s.row_cells(4), (&[][..], Cell::default()));
    }
    use crate::style::Style;

    /// A screen whose every cell names its own position, so any move is visible.
    fn numbered(rows: usize, cols: usize) -> Screen {
        let mut s = Screen::new(rows, cols);
        for r in 0..rows {
            for c in 0..cols {
                let ch = char::from_u32(0x4E00 + (r * 64 + c) as u32).unwrap();
                s.set(r, c, Cell::new(ch, Style::default()));
            }
        }
        s
    }

    /// The per-cell scroll this replaces, as the reference.
    fn reference_up(s: &mut Screen, top: usize, bottom: usize, n: usize, fill: Cell) {
        let n = n.min(bottom - top + 1);
        for r in top..=bottom {
            for c in 0..s.cols() {
                let cell = if r + n <= bottom {
                    s.cell(r + n, c)
                } else {
                    fill
                };
                s.set(r, c, cell);
            }
        }
    }

    fn reference_down(s: &mut Screen, top: usize, bottom: usize, n: usize, fill: Cell) {
        let n = n.min(bottom - top + 1);
        for r in (top..=bottom).rev() {
            for c in 0..s.cols() {
                let cell = if r >= top + n { s.cell(r - n, c) } else { fill };
                s.set(r, c, cell);
            }
        }
    }

    #[test]
    fn region_scrolls_match_the_per_cell_reference_everywhere() {
        let fill = Cell::new(' ', Style::default());
        for (rows, cols) in [(1, 1), (5, 3), (8, 7)] {
            for top in 0..rows {
                for bottom in top..rows {
                    for n in 0..=(bottom - top + 2) {
                        let (mut fast, mut slow) = (numbered(rows, cols), numbered(rows, cols));
                        fast.scroll_rows_up(top, bottom, n, fill);
                        if n > 0 {
                            reference_up(&mut slow, top, bottom, n, fill);
                        }
                        assert_eq!(fast, slow, "up {rows}x{cols} {top}..={bottom} by {n}");

                        let (mut fast, mut slow) = (numbered(rows, cols), numbered(rows, cols));
                        fast.scroll_rows_down(top, bottom, n, fill);
                        if n > 0 {
                            reference_down(&mut slow, top, bottom, n, fill);
                        }
                        assert_eq!(fast, slow, "down {rows}x{cols} {top}..={bottom} by {n}");
                    }
                }
            }
        }
    }

    /// The row map is invisible: a screen scrolled into a state equals a
    /// screen built that way, its rows read back in order, and a clear resets it.
    #[test]
    fn a_fully_scrolled_screen_equals_one_built_that_way() {
        let fill = Cell::new(' ', Style::default());
        let mut s = numbered(4, 3);
        // Scroll a whole screen's worth and then some, so origin wraps.
        for _ in 0..7 {
            s.scroll_rows_up(0, 3, 1, fill);
        }
        let blank = Screen::new(4, 3);
        assert_eq!(s, blank, "everything scrolled off: a blank screen");

        let mut s = numbered(4, 3);
        s.scroll_rows_up(0, 3, 2, fill);
        let mut want = Screen::new(4, 3);
        for r in 0..2 {
            for c in 0..3 {
                want.set(r, c, numbered(4, 3).cell(r + 2, c));
            }
        }
        assert_eq!(s, want, "two rows up, two blank rows in");
        assert_eq!(s.render_full(), want.render_full(), "same bytes");

        // A wrapped ring still writes, reads and clears by logical position.
        s.set(0, 0, Cell::new('A', Style::default()));
        assert_eq!(s.cell(0, 0).ch, 'A');
        s.clear();
        assert_eq!(s, Screen::new(4, 3));
    }

    /// Long random mixes of full-screen scrolls, region scrolls (up and down)
    /// and writes agree with the per-cell reference at every step, so the row
    /// map stays a faithful permutation however it has been rotated.
    #[test]
    fn mixed_scrolls_and_writes_match_the_reference_over_many_steps() {
        let mut seed = 0x9E37_79B9_7F4A_7C15u64;
        let mut next = |n: usize| {
            seed ^= seed << 13;
            seed ^= seed >> 7;
            seed ^= seed << 17;
            (seed % n as u64) as usize
        };
        for (rows, cols) in [(1, 3), (6, 5), (24, 9)] {
            let (mut fast, mut slow) = (numbered(rows, cols), numbered(rows, cols));
            for step in 0..2_000 {
                let fill = Cell::new(char::from(b'a' + (step % 26) as u8), Style::default());
                let (top, bottom) = if next(3) == 0 {
                    (0, rows - 1)
                } else {
                    let a = next(rows);
                    (a, a + next(rows - a))
                };
                let n = 1 + next(3);
                match next(4) {
                    0 => {
                        fast.scroll_rows_up(top, bottom, n, fill);
                        reference_up(&mut slow, top, bottom, n, fill);
                    }
                    1 => {
                        fast.scroll_rows_down(top, bottom, n, fill);
                        reference_down(&mut slow, top, bottom, n, fill);
                    }
                    _ => {
                        let (r, c) = (next(rows), next(cols));
                        fast.set(r, c, fill);
                        slow.set(r, c, fill);
                    }
                }
                assert_eq!(fast, slow, "{rows}x{cols}, step {step}");
            }
        }
    }

    #[test]
    fn write_ascii_matches_copy_cells_and_clips() {
        let style = Style {
            bold: true,
            ..Style::default()
        };
        let (mut a, mut b) = (numbered(3, 8), numbered(3, 8));
        // Scroll first so the row is lazily blank with a styled fill.
        let fill = Cell::new(' ', style);
        a.scroll_rows_up(0, 2, 1, fill);
        b.scroll_rows_up(0, 2, 1, fill);
        let text = b"hello, world";
        assert_eq!(a.write_ascii(2, 3, text, style), 5, "clipped at the edge");
        let cells: Vec<Cell> = text.iter().map(|&c| Cell::new(c as char, style)).collect();
        b.copy_cells(2, 3, &cells);
        assert_eq!(a, b);
        assert_eq!(a.cell(2, 0), fill, "the gap before the text keeps the fill");
        assert_eq!(a.write_ascii(3, 0, b"x", style), 0);
        assert_eq!(a.write_ascii(0, 8, b"x", style), 0);
    }

    /// A row's wrap flag leaves with it into the history and is clear on the
    /// row that takes its place; setting it out of range is ignored.
    #[test]
    fn a_wrap_flag_follows_its_row_into_history_and_clears_on_reuse() {
        let fill = Cell::default();
        let mut s = Screen::with_history(3, 5, 10);
        s.write_str(0, 0, "hello", Style::default());
        s.set_row_wrapped(0, true);
        assert!(s.row_wrapped(0) && !s.row_wrapped(1));
        assert!(s.visible().row_wrapped(0), "a snapshot keeps it");
        let mut other = Screen::with_history(3, 5, 10);
        other.write_str(0, 0, "hello", Style::default());
        assert_ne!(s, other, "the flag is part of what a screen holds");
        // Scrolled off the top: the history keeps it; the row that comes back
        // in at the bottom starts unwrapped.
        s.scroll_rows_up(0, 2, 1, fill);
        assert!(s.history_wrapped(0));
        assert!(!s.row_wrapped(2), "reused row cleared");
        let mut copy = Screen::with_history(3, 5, 10);
        copy.copy_history_from(&s);
        assert!(copy.history_wrapped(0), "a resize's history copy keeps it");
        s.set_row_wrapped(1, true);
        s.clear();
        assert!(!s.row_wrapped(1));
        s.set_row_wrapped(9, true); // out of range: ignored
        assert!(!s.row_wrapped(9));
    }

    /// A screen with history scrolls exactly like one without, and every line
    /// that left the top is in the history, newest first, until the ring is
    /// full and drops the oldest.
    #[test]
    fn history_keeps_what_scrolls_off_the_top() {
        let fill = Cell::new(' ', Style::default());
        let line = |k: usize| -> Vec<Cell> {
            format!("line {k:03}")
                .chars()
                .map(|c| Cell::new(c, Style::default()))
                .collect()
        };
        let (mut plain, mut hist) = (Screen::new(4, 10), Screen::with_history(4, 10, 5));
        for k in 0..12 {
            for s in [&mut plain, &mut hist] {
                s.scroll_rows_up(0, 3, 1, fill);
                s.copy_cells(3, 0, &line(k));
            }
            assert_eq!(
                hist, plain,
                "the visible screen is unchanged by history, step {k}"
            );
        }
        // 12 scrolls: the 4 initial blank rows, then lines 0..=7, left the
        // top (lines 8..=11 are on screen); the ring keeps the newest 5.
        assert_eq!(hist.history_len(), 5);
        assert_eq!(hist.scrolled_lines(), 12);
        let text = |age| -> String { (0..8).map(|c| hist.history_cell(age, c).ch).collect() };
        assert_eq!(text(0), "line 007");
        assert_eq!(text(4), "line 003");
        assert_eq!(hist.history_cell(5, 0), Cell::default(), "past the ring");
        // A scroll region does not feed the history.
        hist.scroll_rows_up(1, 3, 1, fill);
        assert_eq!(hist.history_len(), 5);
        assert_eq!(hist.scrolled_lines(), 12);
        // visible() drops the history, keeps what shows.
        let v = hist.visible();
        assert_eq!(v, hist);
        assert_eq!(v.history_len(), 0);
        // History survives a copy into another size, cut to its width.
        let mut wide = Screen::with_history(3, 6, 10);
        wide.copy_history_from(&hist);
        assert_eq!(wide.history_len(), 5);
        let t: String = (0..6).map(|c| wide.history_cell(0, c).ch).collect();
        assert_eq!(t, "line 0");
        hist.clear_history();
        assert_eq!(hist.history_len(), 0);
        // After clearing, scrolling reuses the freed rows.
        for k in 0..3 {
            hist.scroll_rows_up(0, 3, 1, fill);
            hist.copy_cells(3, 0, &line(100 + k));
        }
        assert_eq!(hist.history_len(), 3);
    }

    /// Every row that scrolls off comes back from the history cell for cell:
    /// styles, wide and continuation halves, other characters, and the lazy
    /// blank past what was written, over many random rows and scroll counts.
    #[test]
    fn history_rows_come_back_exactly_as_they_left() {
        let (rows, cols, keep) = (5, 12, 400);
        let mut seed = 0x2545_f491_4f6c_dd1du64;
        let mut next = move |n: u64| {
            seed ^= seed << 13;
            seed ^= seed >> 7;
            seed ^= seed << 17;
            seed % n
        };
        let styles = [
            Style::default(),
            Style {
                fg: crate::style::Color::Indexed(9),
                bold: true,
                ..Style::default()
            },
            Style {
                bg: crate::style::Color::Rgb(10, 20, 30),
                italic: true,
                strike: true,
                ..Style::default()
            },
        ];
        let chars = ['a', 'Z', ' ', 'é', '─', '😀'];
        let mut s = Screen::with_history(rows, cols, keep);
        let mut left: Vec<Vec<Cell>> = Vec::new();
        for step in 0..3000 {
            let row = next(rows as u64) as usize;
            let col = next(cols as u64) as usize;
            // Mostly one style and plain text, so rows often stay plain.
            let style = styles[if next(4) == 0 { next(3) as usize } else { 0 }];
            match next(10) {
                0 => s.set(row, col, Cell::wide('漢', style)),
                1 => s.set(row, col, Cell::continuation(style)),
                2..=4 => s.set(row, col, Cell::new(chars[next(6) as usize], style)),
                _ => {
                    let text = &b"ls -la \xe9x"[..next(10) as usize];
                    s.write_ascii(row, if next(2) == 0 { 0 } else { col }, text, style);
                }
            }
            if next(3) == 0 {
                let n = 1 + next(3) as usize;
                for r in 0..n.min(rows) {
                    left.push((0..cols).map(|c| s.cell(r, c)).collect());
                }
                s.scroll_rows_up(0, rows - 1, n, Cell::new(' ', styles[step % 3]));
            }
        }
        assert_eq!(s.scrolled_lines(), left.len() as u64);
        assert_eq!(s.history_len(), keep.min(left.len()));
        let mut out = Vec::new();
        for age in 0..s.history_len() {
            assert!(s.history_row(age, &mut out));
            let want = &left[left.len() - 1 - age];
            assert_eq!(&out, want, "age {age}");
            for (c, cell) in want.iter().enumerate() {
                assert_eq!(s.history_cell(age, c), *cell, "age {age} col {c}");
            }
        }
        assert!(!s.history_row(s.history_len(), &mut out));
    }

    #[test]
    fn an_out_of_range_region_scroll_changes_nothing() {
        let fill = Cell::new('x', Style::default());
        let mut s = numbered(4, 4);
        let before = s.clone();
        s.scroll_rows_up(3, 1, 1, fill);
        s.scroll_rows_up(0, 4, 1, fill);
        s.scroll_rows_down(2, 9, 1, fill);
        assert_eq!(s, before);
    }

    // ── helpers ──────────────────────────────────────────────────────────────

    fn filled_row(rows: usize, cols: usize, row: usize, ch: char) -> Screen {
        let mut s = Screen::new(rows, cols);
        for c in 0..cols {
            s.set(row, c, Cell::new(ch, Style::default()));
        }
        s
    }

    fn contains_el(bytes: &[u8]) -> bool {
        let s = std::str::from_utf8(bytes).unwrap_or("");
        s.contains("\x1b[K")
    }

    // ── Screen::clear() ──────────────────────────────────────────────────────

    #[test]
    fn clear_resets_all_cells_and_cursor_without_realloc() {
        let mut s = Screen::new(4, 6);
        // Dirty every cell with a non-default character and style.
        let dirty_style = Style {
            bold: true,
            ..Style::default()
        };
        for r in 0..4 {
            for c in 0..6 {
                s.set(r, c, Cell::new('X', dirty_style));
            }
        }
        // Move the cursor off the origin.
        s.set_cursor(Cursor::new(3, 5));

        s.clear();

        // Dimensions must be unchanged.
        assert_eq!(s.rows(), 4);
        assert_eq!(s.cols(), 6);
        // Cursor must be at the origin.
        assert_eq!(s.cursor(), Cursor::new(0, 0));
        // Every cell must equal the default.
        let blank = Cell::default();
        for r in 0..4 {
            for c in 0..6 {
                assert_eq!(
                    s.cell(r, c),
                    blank,
                    "cell ({r},{c}) was not reset by clear()"
                );
            }
        }
    }

    // ── emit_changes / diff: erase-to-EOL optimisation ───────────────────────

    #[test]
    fn diff_emits_el_when_row_tail_clears_to_blank() {
        // prev: row 0 fully filled with 'X'; next: only first 3 cols are 'X'.
        // The tail cols [3..8) revert to blank → diff should contain EL.
        let prev = filled_row(2, 8, 0, 'X');
        let mut next = Screen::new(2, 8);
        for c in 0..3 {
            next.set(0, c, Cell::new('X', Style::default()));
        }
        let diff = prev.diff(&next);
        assert!(
            contains_el(&diff),
            "expected \\x1b[K in diff when tail blanks out"
        );
    }

    #[test]
    fn diff_no_el_when_tail_was_already_blank_in_prev() {
        // Both screens have content only in cols [0, 3); tail was already blank.
        // No cells changed in the tail, so no EL should be emitted.
        let mut prev = Screen::new(2, 8);
        let mut next = Screen::new(2, 8);
        for c in 0..3 {
            prev.set(0, c, Cell::new('A', Style::default()));
            next.set(0, c, Cell::new('B', Style::default()));
        }
        let diff = prev.diff(&next);
        assert!(
            !contains_el(&diff),
            "unexpected \\x1b[K when tail was already blank"
        );
    }

    #[test]
    fn render_full_never_emits_el() {
        // render_full follows \x1b[2J with individual cell writes; it must NOT
        // use EL (the screen is already clear, so EL would be redundant, and
        // the contract is that render_full never relies on prior state).
        let s = filled_row(2, 8, 0, 'X');
        let bytes = s.render_full();
        assert!(!contains_el(&bytes), "render_full must not emit \\x1b[K");
    }

    #[test]
    fn diff_el_visual_correctness_via_round_trip() {
        // Prove that a diff containing EL produces the correct visual result:
        // simulate applying the diff to a "terminal" modelled as a Screen copy.
        // prev → diff(next) → the result must equal next.
        let prev = filled_row(1, 6, 0, 'Z');
        let mut next = Screen::new(1, 6);
        next.set(0, 0, Cell::new('A', Style::default()));
        // next has 'A' at (0,0) and blanks at (0,1..5).
        assert!(contains_el(&prev.diff(&next)));
        // The diff-then-apply path is tested implicitly: if diff produces EL,
        // but the NEXT diff of (next, next) is empty, then prev+diff == next.
        let empty = next.diff(&next);
        assert!(empty.is_empty(), "diff of identical screens must be empty");
    }

    #[test]
    fn diff_entire_row_blank_in_next_uses_el() {
        // prev: entire row 1 filled; next: row 1 is all blank.
        let prev = filled_row(3, 5, 1, 'Y');
        let next = Screen::new(3, 5);
        let diff = prev.diff(&next);
        assert!(
            contains_el(&diff),
            "entire-row blank transition should use EL"
        );
    }

    // ── double-width cells: Cell.width, continuation, width-aware emit ────────

    #[test]
    fn cell_constructors_set_width() {
        let st = Style::default();
        assert_eq!(Cell::default().width, CellWidth::Single);
        assert_eq!(Cell::new('a', st).width, CellWidth::Single);
        assert_eq!(Cell::wide('世', st).width, CellWidth::Wide);
        let cont = Cell::continuation(st);
        assert_eq!(cont.width, CellWidth::Continuation);
        let cols: Vec<usize> = [CellWidth::Continuation, CellWidth::Single, CellWidth::Wide]
            .iter()
            .map(|w| w.columns())
            .collect();
        assert_eq!(cols, [0, 1, 2]);
        assert_eq!(cont.ch, ' ');
    }

    #[test]
    fn wide_lead_emits_glyph_once_no_continuation_char() {
        // A width-2 lead followed by a width-0 continuation must emit the glyph
        // exactly once and NOTHING for the continuation column (the terminal
        // advances the cursor by two when it draws the wide glyph).
        let prev = Screen::new(1, 4);
        let mut next = Screen::new(1, 4);
        next.set(0, 0, Cell::wide('世', Style::default()));
        next.set(0, 1, Cell::continuation(Style::default()));
        let diff = prev.diff(&next);
        let s = std::str::from_utf8(&diff).unwrap();
        assert_eq!(s.matches('世').count(), 1, "glyph must appear exactly once");
        assert!(
            !s.contains(' '),
            "continuation column must emit no space: {s:?}"
        );
        // Exact bytes: CUP to (1,1), the glyph, then park cursor at (1,1).
        assert_eq!(diff, "\x1b[1;1H世\x1b[1;1H".as_bytes());
    }

    #[test]
    fn wide_lead_advances_cursor_by_two() {
        // After a width-2 lead, a cell two columns on needs no fresh CUP — the
        // emitter tracks that the terminal advanced the cursor across both
        // halves of the wide glyph. If accounting were off by one, a CUP escape
        // would appear between the glyph and 'X'.
        let prev = Screen::new(1, 6);
        let mut next = Screen::new(1, 6);
        next.set(0, 0, Cell::wide('世', Style::default()));
        next.set(0, 1, Cell::continuation(Style::default()));
        next.set(0, 2, Cell::new('X', Style::default()));
        let diff = prev.diff(&next);
        assert_eq!(diff, "\x1b[1;1H世X\x1b[1;1H".as_bytes());
    }

    #[test]
    fn ascii_diff_is_byte_identical_regression() {
        // Guard: the width model must not change single-width output at all.
        let mut old = Screen::new(2, 10);
        old.write_str(0, 0, "hello", Style::default());
        let mut new = old.clone();
        new.set(0, 1, Cell::new('a', Style::default()));
        assert_eq!(old.diff(&new), b"\x1b[1;2Ha\x1b[1;1H");
    }

    #[test]
    fn set_cursor_clamps_onto_the_grid() {
        let mut s = Screen::new(4, 6);
        s.set_cursor(Cursor::new(2, 3));
        assert_eq!(s.cursor(), Cursor::new(2, 3));
        s.set_cursor(Cursor::new(99, 99));
        assert_eq!(s.cursor(), Cursor::new(3, 5));
        // An empty screen has no cell to stand on: the origin.
        let mut empty = Screen::new(0, 0);
        empty.set_cursor(Cursor::new(5, 5));
        assert_eq!(empty.cursor(), Cursor::new(0, 0));
    }

    #[test]
    fn copy_cells_matches_repeated_set() {
        let st = Style::default();
        let run = [
            Cell::new('a', st),
            Cell::wide('世', st),
            Cell::continuation(st),
            Cell::new('b', st),
        ];
        let mut via_copy = Screen::new(2, 8);
        via_copy.copy_cells(1, 2, &run);
        let mut via_set = Screen::new(2, 8);
        for (i, cell) in run.iter().enumerate() {
            via_set.set(1, 2 + i, *cell);
        }
        assert_eq!(via_copy, via_set);
    }

    #[test]
    fn copy_cells_clips_at_edge_and_ignores_out_of_range() {
        let st = Style::default();
        let run = [Cell::new('x', st); 5];
        let mut s = Screen::new(2, 4);
        // 5-cell run starting at col 2: only cols 2 and 3 are written.
        s.copy_cells(0, 2, &run);
        assert_eq!(s.cell(0, 2).ch, 'x');
        assert_eq!(s.cell(0, 3).ch, 'x');
        assert_eq!(s.cell(0, 1), Cell::default());
        // Out-of-range row or start column are silent no-ops (must not panic).
        s.copy_cells(9, 0, &run);
        s.copy_cells(0, 9, &run);
    }
}