rs-rich 0.0.9

//! Syntax highlighting.
//!
//! Port of `rich/syntax.py`'s renderable surface. A [`Syntax`] highlights a
//! block of source code for a given language and theme, producing colored
//! [`Segment`]s (a solid block: each line is padded to the render width with
//! the theme background).
//!
//! Highlighting goes through a [`CodeHighlighter`]. The default,
//! [`SyntectHighlighter`], uses the `syntect` crate.
//!
//! **Divergence:** upstream uses Pygments; `syntect` ships different grammars
//! and themes. So the *coloring is functional, not byte-identical* to Python
//! rich — see docs/DIVERGENCES.md. Everything else (the renderable protocol,
//! width handling) matches the port's conventions.

use std::collections::BTreeSet;
use std::sync::Arc;

#[cfg(feature = "syntax-cache")]
#[path = "syntax_cache.rs"]
mod cache;
#[path = "syntax_syntect.rs"]
mod syntect_adapter;

pub use syntect_adapter::SyntectHighlighter;

use crate::cells::cell_len;
use crate::color::{Color, ColorSystem, ColorTriplet, ColorType};
use crate::console::{Console, ConsoleOptions};
use crate::console::{Justify, Overflow};
use crate::measure::Measurement;
use crate::protocol::{
    CodeHighlighter, HighlightError, HighlightedCode, HighlightedLine, Renderable,
};
use crate::segment::Segment;
use crate::style::{Style, StyleType};
use crate::text::python_splitlines;
use crate::text::Text;

/// Upstream's `Syntax(tab_size=4)`.
const DEFAULT_TAB_SIZE: usize = 4;

/// A block of syntax-highlighted source code. Mirrors `rich.syntax.Syntax`.
pub struct Syntax {
    code: String,
    language: Option<String>,
    /// `None` means the highlighter's default theme.
    theme: Option<String>,
    word_wrap: bool,
    /// `(top, right, bottom, left)`.
    padding: (usize, usize, usize, usize),
    tab_size: usize,
    /// `None` means [`SyntectHighlighter`].
    highlighter: Option<Arc<dyn CodeHighlighter>>,
    line_numbers: bool,
    start_line: i64,
    line_range: Option<(Option<i64>, Option<i64>)>,
    highlight_lines: BTreeSet<i64>,
    code_width: Option<usize>,
    background_color: Option<String>,
    indent_guides: bool,
    stylized_ranges: Vec<StylizedRange>,
}

/// A `(line, column)` position in the code: 1-based line, 0-based column.
/// Upstream's `SyntaxPosition`: a 1-based line and a 0-based column. Both
/// are signed, as upstream's are (see [`Syntax::stylize_range`]).
pub type SyntaxPosition = (i64, i64);

/// Upstream's `_SyntaxHighlightRange`.
#[derive(Clone, Debug)]
struct StylizedRange {
    style: StyleType,
    start: SyntaxPosition,
    end: SyntaxPosition,
    style_before: bool,
}

/// Upstream's `NUMBERS_COLUMN_DEFAULT_PADDING`.
const NUMBERS_COLUMN_DEFAULT_PADDING: usize = 2;

/// Port of Python's `str.expandtabs(tab_size)`, which `Syntax._process_code`
/// runs over the source before highlighting it.
///
/// A tab advances to the next multiple of `tab_size` **counted in characters,
/// not cells** (CPython's `unicode_expandtabs` walks code points), and the
/// column resets at `\n` and `\r`. `tab_size == 0` deletes the tab, matching
/// CPython's `tabsize <= 0` branch.
///
/// Without this the raw U+0009 reached the terminal, where it jumps to the next
/// 8-cell stop while we had measured it as one cell: a block asked to be 30
/// wide rendered 31-32 cells and tore the background panel.
fn expand_tabs(code: &str, tab_size: usize) -> String {
    if !code.contains('\t') {
        return code.to_string();
    }
    let mut out = String::with_capacity(code.len());
    let mut column = 0usize;
    for ch in code.chars() {
        match ch {
            '\t' => {
                if tab_size > 0 {
                    let advance = tab_size - (column % tab_size);
                    out.extend(std::iter::repeat_n(' ', advance));
                    column += advance;
                }
            }
            '\n' | '\r' => {
                out.push(ch);
                column = 0;
            }
            _ => {
                out.push(ch);
                column += 1;
            }
        }
    }
    out
}

impl Syntax {
    /// Wrap lines wider than the render width instead of cropping them.
    ///
    /// Off by default, matching upstream's `Syntax(word_wrap=False)`: a long
    /// line is cut at the width. Upstream's **CLI** turns this on, which is why
    /// `rich --syntax` does too — cropping a source file silently loses code.
    pub fn word_wrap(mut self, wrap: bool) -> Self {
        self.word_wrap = wrap;
        self
    }

    /// Highlight `code` as `language` (a name or file extension, e.g. `"rust"`
    /// or `"rs"`). Pass an empty/unknown language to render as plain text.
    pub fn new(code: impl Into<String>, language: impl Into<String>) -> Self {
        Syntax {
            word_wrap: false,
            padding: (0, 0, 0, 0),
            tab_size: DEFAULT_TAB_SIZE,
            code: code.into(),
            language: Some(language.into()).filter(|l| !l.is_empty()),
            theme: None,
            highlighter: None,
            line_numbers: false,
            start_line: 1,
            line_range: None,
            highlight_lines: BTreeSet::new(),
            code_width: None,
            background_color: None,
            indent_guides: false,
            stylized_ranges: Vec::new(),
        }
    }

    /// Highlight with `highlighter` instead of the default
    /// [`SyntectHighlighter`]. Theme names are the highlighter's own.
    pub fn highlighter(mut self, highlighter: Arc<dyn CodeHighlighter>) -> Self {
        self.highlighter = Some(highlighter);
        self
    }

    /// How far a tab advances the column, in characters. Upstream's
    /// `Syntax(tab_size=…)`, default 4.
    ///
    /// Tabs are *expanded* to spaces before highlighting (upstream's
    /// `code.expandtabs(self.tab_size)`), so this is the only tab handling in
    /// play — the rendered code contains no U+0009 at all.
    pub fn tab_size(mut self, tab_size: usize) -> Self {
        self.tab_size = tab_size;
        self
    }

    /// Surround the code with `padding` cells of background on every side.
    ///
    /// Upstream's Markdown renders a fenced block as `Syntax(..., padding=1)`,
    /// which is what gives a code block its blank inset row above and below and
    /// its one-column gutter. Without it the code sat flush against the
    /// surrounding text and every document containing a fence diverged.
    pub fn padding(mut self, padding: usize) -> Self {
        self.padding = (padding, padding, padding, padding);
        self
    }

    /// Padding on each side, `(top, right, bottom, left)`, in the code
    /// background (upstream `padding` as a tuple).
    pub fn padding_sides(mut self, padding: (usize, usize, usize, usize)) -> Self {
        self.padding = padding;
        self
    }

    /// Number the lines in a gutter (upstream `line_numbers`, default off).
    pub fn line_numbers(mut self, line_numbers: bool) -> Self {
        self.line_numbers = line_numbers;
        self
    }

    /// The number of the first line (upstream `start_line`, default 1). Any
    /// integer, as upstream's: zero and negative numbers are drawn as they are.
    pub fn start_line(mut self, start_line: i64) -> Self {
        self.start_line = start_line;
        self
    }

    /// Render only lines `start..=end` (1-based; `None` leaves that end open).
    /// Upstream `line_range`, with its Python semantics: a start of 0 or less
    /// is the first line, and a negative end counts back from the last line
    /// (`lines[start - 1:end]`) after the highlighter has stopped at the first
    /// line it reached.
    pub fn line_range(mut self, start: Option<i64>, end: Option<i64>) -> Self {
        self.line_range = Some((start, end));
        self
    }

    /// Mark these line numbers with a pointer in the gutter (upstream
    /// `highlight_lines`; shown with `line_numbers`).
    pub fn highlight_lines(mut self, lines: impl IntoIterator<Item = i64>) -> Self {
        self.highlight_lines = lines.into_iter().collect();
        self
    }

    /// A fixed width for the code, excluding line numbers (upstream
    /// `code_width`; default all the available width).
    pub fn code_width(mut self, code_width: usize) -> Self {
        self.code_width = Some(code_width);
        self
    }

    /// Override the theme's background colour (upstream `background_color`).
    pub fn background_color(mut self, color: impl Into<String>) -> Self {
        self.background_color = Some(color.into());
        self
    }

    /// Draw indent guides (upstream `indent_guides`, default off; not on an
    /// ASCII-only console).
    pub fn indent_guides(mut self, indent_guides: bool) -> Self {
        self.indent_guides = indent_guides;
        self
    }

    /// Style a range of the code, from `start` to `end` (`(line, column)`,
    /// 1-based lines, 0-based columns), on top of the highlighting — or
    /// beneath it with `style_before`. Port of `Syntax.stylize_range`.
    pub fn stylize_range(
        &mut self,
        style: impl Into<StyleType>,
        start: SyntaxPosition,
        end: SyntaxPosition,
        style_before: bool,
    ) -> &mut Self {
        self.stylized_ranges.push(StylizedRange {
            style: style.into(),
            start,
            end,
            style_before,
        });
        self
    }

    /// Choose the highlighting theme, by the highlighter's name for it. The
    /// default highlighter offers `syntect`'s themes plus upstream's
    /// `ansi_dark` and `ansi_light`. Unknown names fall back to the
    /// highlighter's default theme.
    pub fn theme(mut self, theme: impl Into<String>) -> Self {
        self.theme = Some(theme.into());
        self
    }

    /// Highlight the (already tab-expanded) `code`, falling back to the default
    /// theme for an unknown one and to plain text if the engine fails, then
    /// validate the result against the source (see [`CodeHighlighter`]).
    ///
    /// The engine is this `Syntax`'s own, else `console`'s default (whose theme
    /// then applies unless this `Syntax` names one), else syntect.
    fn highlighted(&self, code: &str, console: Option<&Console>) -> HighlightedCode {
        let (engine, theme) = self.engine_and_theme(console);
        let language = self.language.as_deref();
        let result = match engine.highlight(code, language, &theme) {
            Err(HighlightError::UnknownTheme(_)) => {
                engine.highlight(code, language, engine.default_theme())
            }
            other => other,
        };
        let highlighted = result.unwrap_or_default();
        validate(code, highlighted)
    }
}

/// Make a highlighter's output safe to render against `code`: one line per
/// `code.split('\n')` element, spans sorted, in range, non-overlapping and on
/// character boundaries, and no hyperlinks in adapter styles.
fn validate(code: &str, mut highlighted: HighlightedCode) -> HighlightedCode {
    let sources: Vec<&str> = code.split('\n').collect();
    highlighted
        .lines
        .resize_with(sources.len(), HighlightedLine::default);
    for (line, source) in highlighted.lines.iter_mut().zip(&sources) {
        let mut end = 0usize;
        line.spans.retain(|span| {
            let keep = span.range.start < span.range.end
                && span.range.start >= end
                && span.range.end <= source.len()
                && source.is_char_boundary(span.range.start)
                && source.is_char_boundary(span.range.end);
            if keep {
                end = span.range.end;
            }
            keep
        });
        for span in &mut line.spans {
            span.style = span.style.update_link(None);
        }
        line.newline_style = line.newline_style.as_ref().map(|s| s.update_link(None));
    }
    highlighted.default_style = highlighted.default_style.update_link(None);
    highlighted
}

/// The pieces of one source line: every span, plus the gaps between them in
/// the default style, in order.
fn line_pieces<'a>(
    source: &'a str,
    line: &HighlightedLine,
    default_style: &Style,
) -> Vec<(&'a str, Style)> {
    let mut pieces = Vec::with_capacity(line.spans.len() + 1);
    let mut position = 0usize;
    for span in &line.spans {
        if span.range.start > position {
            pieces.push((&source[position..span.range.start], default_style.clone()));
        }
        pieces.push((&source[span.range.clone()], span.style.clone()));
        position = span.range.end;
    }
    if position < source.len() {
        pieces.push((&source[position..], default_style.clone()));
    }
    pieces
}

impl Syntax {
    /// The highlighting engine and theme name: this `Syntax`'s own, else
    /// `console`'s default (whose theme then applies unless this `Syntax`
    /// names one), else syntect's.
    fn engine_and_theme(&self, console: Option<&Console>) -> (Arc<dyn CodeHighlighter>, String) {
        use crate::protocol::ConsoleCodeHighlighting;
        let default = match &self.highlighter {
            Some(_) => None,
            None => console.and_then(|console| console.code_highlighting()),
        };
        let engine = self
            .highlighter
            .clone()
            .or_else(|| default.map(|d| d.highlighter.clone()))
            .unwrap_or_else(SyntectHighlighter::shared);
        let theme = self
            .theme
            .as_deref()
            .or_else(|| default.and_then(|d| d.theme.as_deref()))
            .unwrap_or(engine.default_theme())
            .to_string();
        (engine, theme)
    }

    /// The theme's style for a token type, falling back to the default theme
    /// when the chosen one is unknown (as highlighting does).
    fn token_style(&self, console: &Console, token: &str) -> Style {
        let (engine, theme) = self.engine_and_theme(Some(console));
        let theme = if engine.themes().contains(&theme) {
            theme
        } else {
            engine.default_theme().to_string()
        };
        engine.token_style(&theme, token).unwrap_or_default()
    }

    /// `Style(bgcolor=background_color)`, or null.
    fn background_style(&self) -> Style {
        self.background_color
            .as_deref()
            .and_then(|color| Color::parse(color).ok())
            .map_or_else(Style::new, |color| Style::new().with_bgcolor(color))
    }

    /// Highlight the code into a [`Text`](crate::text::Text) rather than a padded block. Port of
    /// `Syntax.highlight`: the theme background is the text's base style and
    /// every token carries its own style. Tabs are expanded first, as
    /// `_process_code` does. Used by `Markdown(inline_code_lexer=…)`.
    pub fn highlight(&self) -> crate::text::Text {
        self.highlight_text(None)
    }

    /// [`highlight`](Self::highlight) with `console`'s default code
    /// highlighter (see [`ConsoleCodeHighlighting`](crate::protocol::ConsoleCodeHighlighting))
    /// when this `Syntax` has none of its own. Not in upstream.
    pub fn highlight_for(&self, console: &Console) -> crate::text::Text {
        self.highlight_text(Some(console))
    }

    /// `Syntax._process_code` plus Pygments' own preprocessing: tabs are
    /// expanded, then `\r\n` and a lone `\r` become `\n` (Pygments' `Lexer`
    /// does this for every lexer, the plain-text one included). Without it a
    /// lone `\r` was stripped as a control code and joined two lines.
    fn process_code(&self) -> String {
        let mut code = expand_tabs(&self.code, self.tab_size);
        if !code.contains('\r') {
            return code;
        }
        // Upstream appends `\n` to code that lacks one and removes it after
        // highlighting, so a trailing lone `\r` becomes that `\r\n` and goes.
        if code.ends_with('\r') {
            code.pop();
        }
        code.replace("\r\n", "\n").replace('\r', "\n")
    }

    /// Port of `Syntax.highlight(code, line_range)` over this `Syntax`'s
    /// code: the highlighted `Text` in the base style (the theme's
    /// background and `background_color`), with upstream's `justify`,
    /// `tab_size` and `no_wrap`. With a `line_range`, lines before it are
    /// left unstyled and lines after it dropped. `console` supplies a
    /// default code highlighter, as in [`highlight_for`](Self::highlight_for).
    pub fn highlight_range(
        &self,
        line_range: Option<(Option<i64>, Option<i64>)>,
        console: Option<&Console>,
    ) -> crate::text::Text {
        let mut code = self.process_code();
        if !code.ends_with('\n') {
            code.push('\n');
        }
        let highlighted = self.highlighted(&code, console);
        let base_style = highlighted
            .background
            .clone()
            .map_or_else(Style::new, |bg| Style::new().with_bgcolor(bg))
            .combine(&self.background_style());
        let transparent = base_style.bgcolor().is_none();
        let mut text = self.highlighted_text(&code, &highlighted, line_range);
        text.set_base_style(base_style);
        text.set_justify(if transparent {
            Justify::Default
        } else {
            Justify::Left
        });
        text.set_tab_size(Some(self.tab_size));
        text.set_no_wrap(Some(!self.word_wrap));
        text
    }

    fn highlight_text(&self, console: Option<&Console>) -> crate::text::Text {
        let code = self.process_code();
        let highlighted = self.highlighted(&code, console);
        let mut text = self.highlighted_text(&code, &highlighted, None);
        if let Some(background) = &highlighted.background {
            text.set_base_style(Style::new().with_bgcolor(background.clone()));
        }
        text
    }

    /// Port of `Syntax.highlight`'s token loop over already highlighted
    /// `code`: with a `line_range`, lines before it are left unstyled and
    /// lines after it are dropped. Then `background_color` and the stylized
    /// ranges.
    fn highlighted_text(
        &self,
        code: &str,
        highlighted: &HighlightedCode,
        line_range: Option<(Option<i64>, Option<i64>)>,
    ) -> Text {
        let mut text = Text::new("");
        let (line_start, line_end) = line_range.unwrap_or((None, None));
        // `_line_start = line_start - 1 if line_start else 0`; below zero no
        // line is skipped.
        let skip = line_start.filter(|&start| start != 0).map_or(0, |start| {
            usize::try_from(start.saturating_sub(1)).unwrap_or(0)
        });
        let sources: Vec<&str> = code.split('\n').collect();
        let last = sources.len().saturating_sub(1);
        for (index, (source, line)) in sources.iter().zip(&highlighted.lines).enumerate() {
            let mut pieces: Vec<(String, Option<Style>)> =
                line_pieces(source, line, &highlighted.default_style)
                    .into_iter()
                    .map(|(piece, style)| (piece.to_string(), Some(style)))
                    .collect();
            if index != last {
                // The engine's style for the line break. When it matches the
                // last piece, the break joins that piece, as one token.
                let style = line.newline_style.clone().unwrap_or_default();
                match pieces.last_mut() {
                    Some((piece, Some(last_style)))
                        if line.newline_style.is_some() && *last_style == style =>
                    {
                        piece.push('\n');
                    }
                    _ => pieces.push(("\n".to_string(), Some(style))),
                }
            }
            for (piece, style) in pieces {
                // Tokens before the range carry no style (`yield (token, None)`).
                let style = style.filter(|_| index >= skip);
                text.append(piece.as_str(), style.map(Into::into));
            }
            // `if line_end and line_no >= line_end: break`.
            // A negative end is truthy and already passed.
            if line_end.is_some_and(|end| end != 0 && (index as i64) + 1 >= end && index >= skip) {
                break;
            }
        }
        if let Some(color) = &self.background_color {
            let len = text.plain().len();
            text.stylize(StyleType::Name(format!("on {color}")), 0, len);
        }
        self.apply_stylized_ranges(&mut text);
        text
    }

    /// Port of `Syntax._apply_stylized_ranges`: positions are resolved
    /// against the highlighted text; a column past the end of its line is
    /// clamped, a line out of range skips the range.
    fn apply_stylized_ranges(&self, text: &mut Text) {
        if self.stylized_ranges.is_empty() {
            return;
        }
        let plain = text.plain().to_string();
        // Character offsets of each line start, plus `len + 1`.
        let mut offsets = vec![0usize];
        let mut chars = 0usize;
        for ch in plain.chars() {
            chars += 1;
            if ch == '\n' {
                offsets.push(chars);
            }
        }
        offsets.push(chars + 1);
        // Port of `_get_code_index_for_syntax_position`, with Python's
        // negative list indexing: a line before the first counts back from
        // the end of `offsets`, and a negative column back from the line's
        // start. An index Python would raise `IndexError` for is skipped.
        let count = offsets.len() as i64;
        let offset_at = |index: i64| -> Option<i64> {
            let index = if index < 0 {
                count.saturating_add(index)
            } else {
                index
            };
            usize::try_from(index)
                .ok()
                .and_then(|index| offsets.get(index))
                .map(|&offset| offset as i64)
        };
        let index_for = |(line_number, column): SyntaxPosition| -> Option<i64> {
            if line_number > count || count < line_number.saturating_add(1) {
                return None;
            }
            let line_index = line_number.saturating_sub(1);
            let line_length = offset_at(line_index.saturating_add(1))? - offset_at(line_index)? - 1;
            Some(offset_at(line_index)?.saturating_add(column.min(line_length)))
        };
        let length = chars as i64;
        let byte = |char_index: usize| {
            plain
                .char_indices()
                .nth(char_index)
                .map_or(plain.len(), |(at, _)| at)
        };
        for range in &self.stylized_ranges {
            let (Some(start), Some(end)) = (index_for(range.start), index_for(range.end)) else {
                continue;
            };
            // `Text.stylize`: negative offsets count from the end, and an
            // empty or out-of-range span is dropped.
            let start = if start < 0 {
                length.saturating_add(start)
            } else {
                start
            };
            let end = if end < 0 {
                length.saturating_add(end)
            } else {
                end
            };
            if start >= length || end <= start {
                continue;
            }
            let (start, end) = (byte(start.max(0) as usize), byte(end.min(length) as usize));
            if range.style_before {
                text.stylize_before(range.style.clone(), start, end);
            } else {
                text.stylize(range.style.clone(), start, end);
            }
        }
    }
}

impl Syntax {
    /// Upstream's `_numbers_column_width`.
    fn numbers_column_width(&self) -> usize {
        if !self.line_numbers {
            return 0;
        }
        // Python ints are unbounded: widen so `i64::MAX` does not overflow.
        let last = i128::from(self.start_line) + self.code.matches('\n').count() as i128;
        last.to_string().len() + NUMBERS_COLUMN_DEFAULT_PADDING
    }

    /// `(background, number, highlighted number)` styles for the gutter.
    /// Port of `Syntax._get_number_styles`.
    fn number_styles(&self, console: &Console, base_style: &Style) -> (Style, Style, Style) {
        let dim = |on: bool| Style::parse(if on { "dim" } else { "not dim" }).unwrap_or_default();
        if base_style.bgcolor().is_none() {
            return (Style::new(), dim(true), Style::new());
        }
        if matches!(
            console.color_system(),
            Some(ColorSystem::EightBit | ColorSystem::Truecolor)
        ) {
            let text_style = self.token_style(console, "Text");
            let background = self.background_style();
            let number = base_style
                .combine(&text_style)
                .combine(
                    &Style::new().with_color(self.line_numbers_color(console, base_style, 0.3)),
                )
                .combine(&background);
            let highlight = base_style
                .combine(&text_style)
                .combine(
                    &Style::parse("bold")
                        .unwrap_or_default()
                        .with_color(self.line_numbers_color(console, base_style, 0.9)),
                )
                .combine(&background);
            (base_style.clone(), number, highlight)
        } else {
            (
                base_style.clone(),
                base_style.combine(&dim(true)),
                base_style.combine(&dim(false)),
            )
        }
    }

    /// Port of `Syntax._get_line_numbers_color`: the text colour blended
    /// into the background.
    fn line_numbers_color(&self, console: &Console, base_style: &Style, blend: f64) -> Color {
        let system_defined =
            |color: &Color| !matches!(color.kind, ColorType::EightBit | ColorType::Truecolor);
        let Some(background) = base_style.bgcolor().filter(|color| !system_defined(color)) else {
            return Color::default_color();
        };
        let text_style = self.token_style(console, "Text");
        let Some(foreground) = text_style.color().filter(|color| !system_defined(color)) else {
            return text_style
                .color()
                .cloned()
                .unwrap_or_else(Color::default_color);
        };
        let (Some(bg), Some(fg)) = (background.get_truecolor(), foreground.get_truecolor()) else {
            return Color::default_color();
        };
        let mix = |a: u8, b: u8| (f64::from(a) + (f64::from(b) - f64::from(a)) * blend) as u8;
        let ColorTriplet { red, green, blue } = ColorTriplet::new(
            mix(bg.red, fg.red),
            mix(bg.green, fg.green),
            mix(bg.blue, fg.blue),
        );
        Color::from_rgb(red, green, blue)
    }

    /// Port of `Syntax._get_syntax`: the code's lines, without the padding.
    fn syntax_lines(&self, console: &Console, options: &ConsoleOptions) -> Vec<Vec<Segment>> {
        let (_, pad_right, _, pad_left) = self.padding;
        let horizontal_padding = pad_left + pad_right;
        let numbers_column_width = self.numbers_column_width();
        let code_width = match self.code_width {
            Some(code_width) => code_width,
            None => {
                let width = if self.line_numbers {
                    options.max_width.saturating_sub(numbers_column_width + 1)
                } else {
                    options.max_width
                };
                width.saturating_sub(horizontal_padding)
            }
        };

        // `_process_code`: the code always ends with a newline to highlight.
        let ends_on_nl = self.code.ends_with('\n');
        let mut code = self.process_code();
        if !code.ends_with('\n') {
            code.push('\n');
        }
        let highlighted = self.highlighted(&code, Some(console));
        let background_style = self.background_style();
        let base_style = highlighted
            .background
            .clone()
            .map_or_else(Style::new, |bg| Style::new().with_bgcolor(bg))
            .combine(&background_style);
        let transparent = base_style.bgcolor().is_none();
        let mut text = self.highlighted_text(&code, &highlighted, self.line_range);
        text.set_base_style(base_style.clone());
        text.set_justify(if transparent {
            Justify::Default
        } else {
            Justify::Left
        });
        text.set_tab_size(Some(self.tab_size));
        text.set_no_wrap(Some(!self.word_wrap));
        let guide_style = || {
            base_style
                .combine(&self.token_style(console, "Comment"))
                .combine(&Style::parse("dim").unwrap_or_default())
                .combine(&background_style)
        };
        let render_text = |text: &Text, width: usize, justify: Justify| {
            let overflow = text
                .get_overflow()
                .or(options.overflow)
                .unwrap_or(Overflow::Fold);
            let no_wrap = text.get_no_wrap().or(options.no_wrap).unwrap_or(false);
            let mut lines = text.render_lines_wrapped_tabs(
                console.theme(),
                console.base_style(),
                Some(width),
                justify,
                overflow,
                no_wrap,
                self.tab_size.max(1),
            );
            if lines.is_empty() {
                lines.push(Vec::new());
            }
            lines
        };

        if !self.line_numbers && !self.word_wrap && self.line_range.is_none() {
            if !ends_on_nl && text.plain().ends_with('\n') {
                text.right_crop(1);
            }
            if self.indent_guides && !options.ascii_only() {
                text = text.with_indent_guides(Some(self.tab_size), "│", guide_style());
                text.set_overflow(Some(Overflow::Crop));
            }
            if code_width == 0 {
                return Vec::new();
            }
            if transparent {
                // `console.render(text, options.update(width=code_width))`.
                return render_text(&text, code_width, text.get_justify());
            }
            // `render_lines(text, width=code_width, justify="left",
            // style=background_style, pad=True)`.
            return render_text(&text, code_width, Justify::Left)
                .into_iter()
                .map(|line| {
                    let line = if background_style.is_null() {
                        line
                    } else {
                        Segment::apply_style(&line, &background_style)
                    };
                    Segment::adjust_line_length(&line, code_width, Some(background_style.clone()))
                })
                .collect();
        }

        let (start_line, end_line) = self.line_range.unwrap_or((None, None));
        // `line_offset = max(0, start_line - 1)` when `start_line` is truthy.
        let line_offset = start_line.filter(|&start| start != 0).map_or(0, |start| {
            usize::try_from(start.saturating_sub(1)).unwrap_or(0)
        });
        let mut lines = text.split("\n", false, ends_on_nl);
        if self.line_range.is_some() {
            if line_offset > lines.len() {
                return Vec::new();
            }
            // `lines[line_offset:end_line]`, with Python's slice bounds.
            let len = lines.len() as i64;
            let end = match end_line {
                None => len,
                Some(end) if end < 0 => (len + end).max(0),
                Some(end) => end.min(len),
            } as usize;
            lines = if end > line_offset {
                lines[line_offset..end].to_vec()
            } else {
                Vec::new()
            };
        }
        if self.indent_guides && !options.ascii_only() {
            let style = guide_style().combine(&Style::parse("not italic").unwrap_or_default());
            lines = Text::new("\n")
                .join(&lines)
                .with_indent_guides(Some(self.tab_size), "│", style)
                .split("\n", false, true);
        }

        let (background, number_style, highlight_number_style) =
            self.number_styles(console, &base_style);
        let line_pointer = if options.legacy_windows { "> " } else { "❱ " };
        let mut out: Vec<Vec<Segment>> = Vec::new();
        for (index, line) in lines.iter().enumerate() {
            // Widened, as Python ints are unbounded (`start_line` may be
            // `i64::MAX`).
            let line_no = i128::from(self.start_line) + (line_offset + index) as i128;
            let wrapped_lines: Vec<Vec<Segment>> = if self.word_wrap {
                if code_width == 0 {
                    Vec::new()
                } else {
                    render_text(line, code_width, line.get_justify())
                        .into_iter()
                        .map(|wrapped| {
                            let wrapped = if background.is_null() {
                                wrapped
                            } else {
                                Segment::apply_style(&wrapped, &background)
                            };
                            if transparent {
                                wrapped
                            } else {
                                Segment::adjust_line_length(
                                    &wrapped,
                                    code_width,
                                    Some(background.clone()),
                                )
                            }
                        })
                        .collect()
                }
            } else {
                let segments = line.render(console.theme(), console.base_style());
                if options.no_wrap == Some(true) {
                    vec![segments]
                } else if transparent {
                    vec![crop_line(&segments, code_width)]
                } else {
                    vec![Segment::adjust_line_length(
                        &segments,
                        code_width,
                        Some(background.clone()),
                    )]
                }
            };
            if self.line_numbers {
                let left_pad = Segment::new(
                    " ".repeat(numbers_column_width + 1),
                    Some(background.clone()),
                );
                for (first, wrapped) in wrapped_lines.into_iter().enumerate() {
                    let mut row = Vec::new();
                    if first == 0 {
                        let line_column = format!(
                            "{:>width$} ",
                            line_no,
                            width = numbers_column_width.saturating_sub(2)
                        );
                        if i64::try_from(line_no)
                            .is_ok_and(|line_no| self.highlight_lines.contains(&line_no))
                        {
                            row.push(Segment::new(
                                line_pointer,
                                Some(Style::parse("red").unwrap_or_default()),
                            ));
                            row.push(Segment::new(
                                line_column,
                                Some(highlight_number_style.clone()),
                            ));
                        } else {
                            row.push(Segment::new("  ", Some(highlight_number_style.clone())));
                            row.push(Segment::new(line_column, Some(number_style.clone())));
                        }
                    } else {
                        row.push(left_pad.clone());
                    }
                    row.extend(wrapped);
                    out.push(row);
                }
            } else {
                out.extend(wrapped_lines);
            }
        }
        out
    }
}

impl Syntax {}

/// `Segment.adjust_line_length(…, pad=False)`: crop a line to `width`, never
/// pad it.
fn crop_line(line: &[Segment], width: usize) -> Vec<Segment> {
    let length: usize = line.iter().map(Segment::cell_length).sum();
    if length > width {
        Segment::adjust_line_length(line, width, None)
    } else {
        line.to_vec()
    }
}

/// Lines already rendered, as upstream's `Segments(…)` under `Padding`.
struct RenderedLines(Vec<Vec<Segment>>);

impl Renderable for RenderedLines {
    fn rich_render(&self, _console: &Console, _options: &ConsoleOptions) -> Vec<Segment> {
        join_lines(self.0.clone())
    }

    /// No `__rich_measure__`: `Measurement(0, max_width)`.
    fn measure(&self, _console: &Console, options: &ConsoleOptions) -> Measurement {
        Measurement::new(0, options.max_width)
    }
}

/// Lines joined by newlines. An empty last line is marked with an empty
/// segment, so a container splitting the stream keeps it (see
/// [`Segment::split_lines`]).
fn join_lines(lines: Vec<Vec<Segment>>) -> Vec<Segment> {
    let mut segments = Vec::new();
    let last = lines.len().saturating_sub(1);
    for (index, line) in lines.into_iter().enumerate() {
        // Even when it is the only line: `Syntax("")` prints a blank line.
        if index == last && line.is_empty() {
            segments.push(Segment::new("", None));
        }
        segments.extend(line);
        if index != last {
            segments.push(Segment::line());
        }
    }
    segments
}

impl Renderable for Syntax {
    /// Port of `Syntax.__rich_measure__`. Like upstream it measures the raw
    /// source, where a tab counts as zero cells.
    fn measure(&self, _console: &Console, _options: &ConsoleOptions) -> Measurement {
        let (_, right, _, left) = self.padding;
        let padding = left + right;
        let numbers = self.numbers_column_width();
        if let Some(code_width) = self.code_width {
            return Measurement::new(numbers, code_width + numbers + padding + 1);
        }
        let widest = python_splitlines(&self.code)
            .into_iter()
            .map(cell_len)
            .max()
            .unwrap_or(0);
        let mut width = numbers + padding + widest;
        if self.line_numbers {
            width += 1;
        }
        Measurement::new(numbers, width)
    }

    /// Port of `Syntax.__rich_console__`: the lines, inside `Padding` in the
    /// code's base style when there is any padding.
    fn rich_render(&self, console: &Console, options: &ConsoleOptions) -> Vec<Segment> {
        // The theme's background alone: highlighting no code finds it cheaply.
        let theme_background = self.highlighted("", Some(console)).background;
        let lines = self.syntax_lines(console, options);
        let (top, right, bottom, left) = self.padding;
        if top == 0 && right == 0 && bottom == 0 && left == 0 {
            return join_lines(lines);
        }
        let base_style = theme_background
            .map_or_else(Style::new, |bg| Style::new().with_bgcolor(bg))
            .combine(&self.background_style());
        crate::padding::Padding::new(Box::new(RenderedLines(lines)), self.padding)
            .style(base_style)
            .rich_render(console, options)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::color::ColorSystem;
    use std::sync::Arc;

    fn render(code: &str, lang: &str, width: usize) -> String {
        Console::builder()
            .force_terminal(true)
            .color_system(Some(ColorSystem::Truecolor))
            .width(width)
            .no_color(false)
            .build()
            .render_to_string(&Syntax::new(code, lang))
    }

    #[test]
    fn measured_syntax_still_prints_at_full_width() {
        // Upstream renders a printed Syntax at the console width (its background
        // pads every row); only str/Text shrink to their measurement.
        let console = Console::builder().width(30).color_system(None).build();
        let syntax = Syntax::new("x = 1", "python");
        assert_eq!(syntax.measure(&console, &console.options()).maximum, 5);
        let out = console.render_to_string(&syntax);
        assert!(!out.contains('\x1b'), "{out:?}");
        assert_eq!(cell_len(out.lines().next().unwrap()), 30, "{out:?}");
    }

    #[test]
    fn a_lone_carriage_return_breaks_the_line_like_pygments() {
        // Expected output captured from rich 15.0.0 (color_system=None) with
        // `Console.print`; `render_to_string` leaves out its final newline.
        let render = |renderable: &dyn Renderable, width| {
            let console = Console::builder().width(width).color_system(None).build();
            console.render_to_string(renderable) + "\n"
        };
        assert_eq!(
            render(&Syntax::new("ab\rcd\r", "python"), 10),
            "ab        \ncd        \n"
        );
        assert_eq!(
            render(&Syntax::new("ab\r\ncd", "text"), 10),
            "ab        \ncd        \n"
        );
        assert_eq!(
            render(
                &crate::panel::Panel::fit(Box::new(Syntax::new("ab\rcd", "python"))),
                20
            ),
            "╭────╮\n│ ab │\n│ cd │\n╰────╯\n"
        );
        assert_eq!(
            render(&crate::markdown::Markdown::new("```python\nab\rcd\n```"), 20),
            "                    \n ab                 \n cd                 \n                    \n"
        );
    }

    #[test]
    fn splitlines_matches_python() {
        assert_eq!(
            python_splitlines("a\r\nb\rc\u{2028}d\n"),
            ["a", "b", "c", "d"]
        );
        assert_eq!(python_splitlines("\n\n"), ["", ""]);
        assert!(python_splitlines("").is_empty());
    }

    #[test]
    fn highlights_rust_keyword() {
        // Functional (not byte-parity): assert the code text survives and the
        // output is colored (contains SGR sequences).
        let out = render("fn main() {}", "rust", 20);
        assert!(out.contains("fn"));
        assert!(out.contains("main"));
        assert!(out.contains('\x1b'), "expected ANSI color codes");
    }

    #[test]
    fn multiple_lines_are_separated() {
        let out = render("let x = 1;\nlet y = 2;", "rust", 20);
        assert_eq!(out.matches('\n').count(), 1);
        assert!(out.contains("let"));
    }

    #[test]
    fn unknown_language_renders_plain() {
        // No panic, code preserved, still padded/colored to a block.
        let out = render("just some text", "nonsense-lang", 20);
        assert!(out.contains("just some text"));
    }

    #[test]
    fn word_wrap_is_off_by_default_matching_upstream() {
        // Measured against upstream: Syntax(word_wrap=False) at width 80 keeps
        // 80 of 300 characters. The default must not diverge from that.
        let code = "A".repeat(300);
        let out = render(&code, "python", 80);
        assert_eq!(out.matches('A').count(), 80, "default should crop");
    }

    #[test]
    fn word_wrap_keeps_every_character() {
        let code = "A".repeat(300);
        let console = Console::builder().width(80).color_system(None).build();
        let out = console.render_to_string(&Syntax::new(code.as_str(), "python").word_wrap(true));
        assert_eq!(
            out.matches('A').count(),
            300,
            "wrapping must not lose characters:
{out}"
        );
    }

    /// Syntax emits segments directly rather than going through `Text`, so the
    /// shared `strip_control_codes` never ran and `rich -x` leaked backspaces
    /// and BELs that `rich -m` did not.
    #[test]
    fn control_codes_are_stripped_from_highlighted_code() {
        let out = render("let x = 1;\u{7}\u{8}\u{b}\u{c}", "rust", 40);
        for code in ['\u{7}', '\u{8}', '\u{b}', '\u{c}'] {
            assert!(
                !out.contains(code),
                "control code {code:?} reached the output"
            );
        }
        assert!(out.contains("let"), "content lost with the control codes");
    }

    /// A blank source line has no segments, and folding an empty row yielded
    /// zero rows rather than one empty one — so wrapping silently deleted every
    /// blank line in the file, and the loss was baked into exports.
    #[test]
    fn word_wrap_keeps_blank_lines() {
        let console = Console::builder().width(20).color_system(None).build();
        let out =
            console.render_to_string(&Syntax::new("a = 1\n\nb = 2\n", "python").word_wrap(true));
        let rows: Vec<&str> = out.trim_end_matches('\n').split('\n').collect();
        // Four rows, not three: upstream splits with Python's `str.split("\n")`,
        // so the trailing newline contributes a final empty row —
        // `"a = 1\n\nb = 2\n".split("\n") == ["a = 1", "", "b = 2", ""]`, and
        // rich 15.0.0 prints four padded rows for it. This assertion previously
        // said three, pinning our own missing-row bug as the expectation.
        assert_eq!(rows.len(), 4, "blank line lost: {rows:?}");
        assert!(
            rows[1].trim().is_empty(),
            "middle row should be blank: {rows:?}"
        );
        assert!(
            rows[3].trim().is_empty(),
            "trailing row should be blank: {rows:?}"
        );
    }

    /// `Syntax._process_code` runs `code.expandtabs(self.tab_size)` before
    /// anything is highlighted. We emitted the raw U+0009 and measured it as one
    /// cell, so a tabbed line reached the terminal 31-32 cells wide against a
    /// requested 30 and tore the background block.
    ///
    /// Both expectations captured verbatim from real rich 15.0.0.
    #[test]
    fn tabs_are_expanded_before_highlighting() {
        let console = Console::builder().width(30).color_system(None).build();
        let out = console.render_to_string(&Syntax::new(
            "def f():\n\tif x:\n\t\treturn 1\n\treturn 0",
            "python",
        ));
        assert_eq!(
            out.split('\n').collect::<Vec<_>>(),
            [
                "def f():                      ",
                "    if x:                     ",
                "        return 1              ",
                "    return 0                  ",
            ]
        );
        assert!(!out.contains('\t'), "a raw tab survived: {out:?}");
    }

    /// A tab advances to the next multiple of the tab size, so it is *not* a
    /// fixed run of spaces — the width of the text before it decides.
    #[test]
    fn a_tab_advances_to_the_next_tab_stop() {
        let console = Console::builder().width(20).color_system(None).build();
        let out = console.render_to_string(&Syntax::new(
            "a\tb\tc\nab\tcd\tef\nabcd\tefgh\tijkl",
            "python",
        ));
        assert_eq!(
            out.split('\n').collect::<Vec<_>>(),
            [
                "a   b   c           ",
                "ab  cd  ef          ",
                "abcd    efgh    ijkl",
            ]
        );
    }

    /// Every row must occupy exactly the requested width *on screen*.
    ///
    /// Measuring against [`cell_len`] cannot catch this: it counted a raw tab as
    /// one cell and the padding was computed the same way, so the row looked
    /// exactly `width` wide to us while the terminal advanced the tab to the
    /// next 8-cell stop and the block overran by seven.
    #[test]
    fn a_tabbed_line_measures_the_requested_width() {
        /// Width as the *terminal* renders it: a tab jumps to the next 8-cell
        /// stop, which is the only measure that reveals the defect.
        fn screen_width(row: &str) -> usize {
            let mut column = 0usize;
            for ch in row.chars() {
                column += if ch == '\t' {
                    8 - (column % 8)
                } else {
                    cell_len(ch.encode_utf8(&mut [0u8; 4]))
                };
            }
            column
        }

        for width in [10usize, 20, 30, 40] {
            let console = Console::builder().width(width).color_system(None).build();
            let out = console.render_to_string(&Syntax::new("\tvalue = compute(a, b)", "python"));
            for row in out.split('\n') {
                assert_eq!(screen_width(row), width, "row {row:?} at width {width}");
            }
        }
    }

    /// `str.expandtabs` counts *characters*, not cells, and resets its column at
    /// `\n` and `\r`.
    #[test]
    fn expand_tabs_matches_pythons_str_expandtabs() {
        // Left column verified against CPython's `str.expandtabs(4)`.
        for (input, expected) in [
            ("a\tb", "a   b"),
            ("ab\tb", "ab  b"),
            ("abc\tb", "abc b"),
            ("abcd\tb", "abcd    b"),
            ("\t", "    "),
            ("a\nbb\tc", "a\nbb  c"),
            ("a\rbb\tc", "a\rbb  c"),
            // A wide char counts as one column, exactly as in Python.
            ("\u{4e2d}\tx", "\u{4e2d}   x"),
        ] {
            assert_eq!(expand_tabs(input, 4), expected, "input {input:?}");
        }
        // `tabsize <= 0` deletes the tab (CPython's own branch).
        assert_eq!(expand_tabs("a\tb", 0), "ab");
    }

    /// Upstream's word_wrap breaks at word boundaries; we folded wherever the
    /// row filled up, splitting identifiers mid-word.
    #[test]
    fn word_wrap_breaks_between_words() {
        let console = Console::builder().width(30).color_system(None).build();
        // This exact line is the one character-folding splits as `z` / `eta`,
        // which is what makes the assertion discriminating.
        let code = "result = compute_total(alpha, beta, gamma, delta, epsilon, zeta, eta, theta)\n";
        let out = console.render_to_string(&Syntax::new(code, "python").word_wrap(true));
        // Every identifier must survive on a single row. Folding mid-word split
        // `epsilon` across the break as `e` / `psilon`.
        for word in [
            "compute_total",
            "alpha",
            "gamma",
            "epsilon",
            "zeta",
            "theta",
        ] {
            assert!(
                out.split('\n').any(|row| row.contains(word)),
                "{word:?} was split across rows: {out:?}"
            );
        }
    }

    // ---- CodeHighlighter (#522, #523) ---------------------------------------

    use crate::protocol::HighlightSpan;

    /// A highlighter that returns exactly the lines it was built with.
    struct Fixed(Vec<HighlightedLine>);

    impl CodeHighlighter for Fixed {
        fn highlight(
            &self,
            _code: &str,
            _language: Option<&str>,
            _theme: &str,
        ) -> Result<HighlightedCode, HighlightError> {
            Ok(HighlightedCode {
                lines: self.0.clone(),
                ..Default::default()
            })
        }
        fn default_theme(&self) -> &str {
            "fixed"
        }
        fn themes(&self) -> Vec<String> {
            vec!["fixed".into()]
        }
        fn languages(&self) -> Vec<String> {
            Vec::new()
        }
    }

    /// Styles every line by theme: `bold` (its default) or `underline`.
    struct ByTheme;

    impl CodeHighlighter for ByTheme {
        fn highlight(
            &self,
            code: &str,
            _language: Option<&str>,
            theme: &str,
        ) -> Result<HighlightedCode, HighlightError> {
            let style = match theme {
                "bold" | "underline" => Style::parse(theme).unwrap(),
                other => return Err(HighlightError::UnknownTheme(other.into())),
            };
            let lines = code
                .split('\n')
                .map(|line| HighlightedLine {
                    spans: (!line.is_empty())
                        .then(|| HighlightSpan {
                            range: 0..line.len(),
                            style: style.clone(),
                        })
                        .into_iter()
                        .collect(),
                    newline_style: None,
                })
                .collect();
            Ok(HighlightedCode {
                lines,
                ..Default::default()
            })
        }
        fn default_theme(&self) -> &str {
            "bold"
        }
        fn themes(&self) -> Vec<String> {
            vec!["bold".into(), "underline".into()]
        }
        fn languages(&self) -> Vec<String> {
            Vec::new()
        }
    }

    /// The console's default highlighter and theme apply to a `Syntax` (and
    /// to Markdown code) without its own; the `Syntax`'s own highlighter or
    /// theme wins; with no default, output is unchanged.
    #[test]
    fn a_console_default_highlighter_applies_where_none_is_given() {
        use crate::markdown::Markdown;
        use crate::protocol::{CodeHighlighting, ConsoleCodeHighlighting};
        let plain = || {
            Console::builder()
                .width(20)
                .force_terminal(true)
                .color_system(Some(crate::color::ColorSystem::Truecolor))
                .build()
        };
        let with = |theme: Option<&str>| {
            let mut console = plain();
            console.set_code_highlighting(Some(CodeHighlighting {
                highlighter: Arc::new(ByTheme),
                theme: theme.map(str::to_string),
            }));
            console
        };
        let syntax = || Syntax::new("x = 1", "python");

        assert!(with(None)
            .render_to_string(&syntax())
            .contains("\x1b[1mx = 1"));
        assert!(with(Some("underline"))
            .render_to_string(&syntax())
            .contains("\x1b[4mx = 1"));
        // The Syntax's own theme beats the console's.
        assert!(with(Some("underline"))
            .render_to_string(&syntax().theme("bold"))
            .contains("\x1b[1mx = 1"));
        // The Syntax's own highlighter beats the console's, and the console's
        // theme (a name of another engine) does not follow it.
        let own = with(Some("underline"))
            .render_to_string(&syntax().highlighter(SyntectHighlighter::shared()));
        assert_eq!(own, plain().render_to_string(&syntax()));
        // Markdown code blocks follow the console.
        let markdown = Markdown::new("```python\nx = 1\n```");
        assert!(with(None)
            .render_to_string(&markdown)
            .contains("\x1b[1mx = 1"));
        // `highlight_for` sees the console; `highlight` does not.
        assert!(!syntax().highlight_for(&with(None)).spans().is_empty());
        assert_eq!(
            syntax().highlight_for(&plain()).spans(),
            syntax().highlight().spans()
        );
        // With no default, nothing changes.
        let mut cleared = with(None);
        cleared.set_code_highlighting(None);
        assert_eq!(
            cleared.render_to_string(&syntax()),
            plain().render_to_string(&syntax())
        );
    }

    /// Always fails with an engine error.
    struct Broken;

    impl CodeHighlighter for Broken {
        fn highlight(
            &self,
            _code: &str,
            _language: Option<&str>,
            _theme: &str,
        ) -> Result<HighlightedCode, HighlightError> {
            Err(HighlightError::Engine("boom".into()))
        }
        fn default_theme(&self) -> &str {
            "x"
        }
        fn themes(&self) -> Vec<String> {
            vec!["x".into()]
        }
        fn languages(&self) -> Vec<String> {
            Vec::new()
        }
    }

    fn span(range: std::ops::Range<usize>, style: &str) -> HighlightSpan {
        HighlightSpan {
            range,
            style: Style::parse(style).unwrap(),
        }
    }

    fn line(spans: Vec<HighlightSpan>) -> HighlightedLine {
        HighlightedLine {
            spans,
            newline_style: None,
        }
    }

    fn render_with(code: &str, highlighter: Arc<dyn CodeHighlighter>, width: usize) -> String {
        Console::builder()
            .force_terminal(true)
            .color_system(Some(ColorSystem::Standard))
            .width(width)
            .build()
            .render_to_string(&Syntax::new(code, "python").highlighter(highlighter))
    }

    #[test]
    fn a_custom_highlighter_drives_the_rendered_styles() {
        let fixed = Fixed(vec![line(vec![span(0..1, "red"), span(4..5, "bold")])]);
        let out = render_with("x = 1", Arc::new(fixed), 7);
        // `x` red, ` = ` in the default (unstyled) gap, `1` bold. A theme
        // with no background is transparent: upstream renders it unpadded.
        assert_eq!(out, "\u{1b}[31mx\u{1b}[0m = \u{1b}[1m1\u{1b}[0m");
    }

    #[test]
    fn invalid_spans_are_dropped_not_rendered() {
        let fixed = Fixed(vec![line(vec![
            span(0..3, "red"),
            span(1..2, "green"), // overlaps the first
            span(3..99, "blue"), // past the end of the line
            span(std::ops::Range { start: 2, end: 1 }, "yellow"), // reversed
        ])]);
        let out = render_with("abcdef", Arc::new(fixed), 6);
        assert_eq!(out, "\u{1b}[31mabc\u{1b}[0mdef");
        // A span ending inside a multi-byte character is dropped too.
        let fixed = Fixed(vec![line(vec![span(0..1, "red")])]);
        let out = render_with("é", Arc::new(fixed), 1);
        assert_eq!(out, "é");
    }

    #[test]
    fn missing_and_extra_lines_are_reconciled_with_the_source() {
        // One line returned for three source lines: the rest render unstyled.
        let fixed = Fixed(vec![line(vec![span(0..1, "red")])]);
        let out = render_with("a\nb\nc", Arc::new(fixed), 1);
        assert_eq!(out, "\u{1b}[31ma\u{1b}[0m\nb\nc");
        // Extra lines beyond the source are ignored.
        let fixed = Fixed(vec![
            line(vec![]),
            line(vec![]),
            line(vec![span(0..1, "red")]),
        ]);
        assert_eq!(render_with("a", Arc::new(fixed), 1), "a");
    }

    #[test]
    fn highlighter_styles_cannot_add_links_or_text() {
        let style = Style::parse("red")
            .unwrap()
            .with_link("https://evil.example/\u{1b}]0;x\u{7}");
        let fixed = Fixed(vec![line(vec![HighlightSpan { range: 0..1, style }])]);
        let out = render_with("a\u{8}", Arc::new(fixed), 1);
        assert!(!out.contains("\u{1b}]8"), "hyperlink escaped: {out:?}");
        assert!(!out.contains("evil"), "{out:?}");
        assert!(!out.contains('\u{8}'), "control code survived: {out:?}");
    }

    #[test]
    fn an_engine_failure_renders_the_source_unstyled() {
        let out = render_with("print(1)", Arc::new(Broken), 8);
        assert_eq!(out, "print(1)");
    }

    #[test]
    fn an_unknown_theme_falls_back_to_the_default_theme() {
        let console = Console::builder()
            .force_terminal(true)
            .color_system(Some(ColorSystem::Truecolor))
            .width(20)
            .build();
        let default = console.render_to_string(&Syntax::new("x = 1", "python"));
        let unknown =
            console.render_to_string(&Syntax::new("x = 1", "python").theme("no-such-theme"));
        assert_eq!(default, unknown);
    }

    #[test]
    fn highlight_text_uses_the_same_highlighter() {
        let fixed = Fixed(vec![line(vec![span(0..1, "red")]), line(vec![])]);
        let text = Syntax::new("ab\ncd", "python")
            .highlighter(Arc::new(fixed))
            .highlight();
        assert_eq!(text.plain(), "ab\ncd");
        assert_eq!(text.spans()[0].start, 0);
        assert_eq!(text.spans()[0].end, 1);
    }

    /// Upstream's `ANSI_DARK`/`ANSI_LIGHT` colours, by Pygments token class,
    /// applied through the TextMate scopes syntect reports.
    #[test]
    fn ansi_themes_use_upstream_palette_colours() {
        let code = "# note\n@wraps\ndef greet(name):\n    return f\"hi {name}\" and 42\n";
        let render = |theme: &str, system: ColorSystem| {
            Console::builder()
                .force_terminal(true)
                .color_system(Some(system))
                .width(40)
                .build()
                .render_to_string(&Syntax::new(code, "python").theme(theme))
        };
        let dark = render("ansi_dark", ColorSystem::Truecolor);
        // No RGB colour and no background anywhere: only the 16 palette colours.
        assert!(!dark.contains("38;2;") && !dark.contains("48;"), "{dark:?}");
        assert!(dark.contains("\u{1b}[2m#"), "comment is dim: {dark:?}");
        assert!(
            dark.contains("\u{1b}[1;95m@"),
            "decorator bold bright_magenta: {dark:?}"
        );
        assert!(
            dark.contains("\u{1b}[33mf"),
            "string prefix is part of the string: {dark:?}"
        );
        assert!(
            dark.contains("\u{1b}[94mdef"),
            "keyword bright_blue: {dark:?}"
        );
        assert!(
            dark.contains("\u{1b}[92mgreet"),
            "function bright_green: {dark:?}"
        );
        assert!(
            dark.contains("\u{1b}[94m42"),
            "number bright_blue: {dark:?}"
        );
        assert!(
            dark.contains("\u{1b}[95mand"),
            "operator word bright_magenta: {dark:?}"
        );
        assert!(dark.contains("\u{1b}[33m"), "string yellow: {dark:?}");
        let light = render("ansi_light", ColorSystem::Truecolor);
        assert!(light.contains("\u{1b}[34mdef"), "keyword blue: {light:?}");
        assert!(
            light.contains("\u{1b}[32mgreet"),
            "function green: {light:?}"
        );
        // A 16-colour console gets exactly the same codes.
        assert_eq!(render("ansi_dark", ColorSystem::Standard), dark);
    }
}