twrite-gpui 0.9.0

use gpui::*;
use twrite_core::{
    HighlightTag, Point as BufferPoint, StyleSpan, StyleValue, UnderlineDecoration,
    split_line_intervals,
};

use std::ops::Range;

use crate::editor::Editor;
use crate::theme::EditorTheme;

/// Visual layout metrics and block-level decorations for a single rendered line.
#[derive(Debug, Clone)]
pub struct LineMetrics {
    /// Font size in pixels for this line.
    pub font_size: Pixels,
    /// Vertical line height in pixels for this line.
    pub line_height: Pixels,
    /// Whether this line is a blockquote.
    pub is_quote: bool,
    /// Whether this line is part of a fenced code block.
    pub is_code_block: bool,
    /// Whether this line is a thematic divider break (---, ***, ___).
    pub is_thematic_break: bool,
    /// Whether this line is a task list item (Some(false) for unchecked, Some(true) for checked).
    pub task_state: Option<bool>,
}

impl LineMetrics {
    /// Calculates layout metrics and block-level decorations based on syntax spans and line text.
    ///
    /// `spans` are the original (pre-concealment) highlight spans for the line. Structural
    /// decorations (`Blockquote`, `HorizontalRule`, `Task*`) are detected solely via
    /// `HighlightTag`s so custom languages work without raw string checks. Code-block
    /// detection relies on a full-line `Code` span rather than fence prefixes.
    pub fn for_line(
        raw_line_text: &str,
        _concealed_display_text: &str,
        spans: &[StyleSpan],
        base_font_size: Pixels,
        base_line_height: Pixels,
    ) -> Self {
        let mut font_size = base_font_size;
        let mut line_height = base_line_height;

        // Lowest heading level wins (level 1 outranks level 6); out-of-range
        // levels keep the base size.
        let heading_level = spans
            .iter()
            .filter_map(|s| match s.style {
                StyleValue::Tag(HighlightTag::Heading(level)) => Some(level),
                _ => None,
            })
            .min();

        match heading_level {
            Some(1) => {
                font_size = base_font_size * 2.0;
                line_height = base_line_height * 1.8;
            }
            Some(2) => {
                font_size = base_font_size * 1.5;
                line_height = base_line_height * 1.4;
            }
            Some(3) => {
                font_size = base_font_size * 1.25;
                line_height = base_line_height * 1.2;
            }
            Some(4) => {
                font_size = base_font_size * 1.125;
                line_height = base_line_height * 1.1;
            }
            Some(5) => {
                font_size = base_font_size * 1.0;
                line_height = base_line_height * 1.0;
            }
            Some(6) => {
                font_size = base_font_size * 0.875;
                line_height = base_line_height * 0.95;
            }
            _ => {}
        }

        let is_quote = spans
            .iter()
            .any(|s| matches!(s.style, StyleValue::Tag(HighlightTag::Blockquote)));

        let is_code_block = spans.iter().any(|s| {
            (raw_line_text.is_empty() || s.range.len() == raw_line_text.len())
                && matches!(s.style, StyleValue::Tag(HighlightTag::Code))
        });

        let is_thematic_break = spans
            .iter()
            .any(|s| matches!(s.style, StyleValue::Tag(HighlightTag::HorizontalRule)));

        let task_state = if spans
            .iter()
            .any(|s| matches!(s.style, StyleValue::Tag(HighlightTag::TaskChecked)))
        {
            Some(true)
        } else if spans
            .iter()
            .any(|s| matches!(s.style, StyleValue::Tag(HighlightTag::TaskUnchecked)))
        {
            Some(false)
        } else {
            None
        };

        Self {
            font_size,
            line_height,
            is_quote,
            is_code_block,
            is_thematic_break,
            task_state,
        }
    }
}

/// Fonts used when building styled text runs for a line.
#[derive(Debug, Clone, Copy)]
pub struct RunFonts<'a> {
    /// Base font for normal spans.
    pub base: &'a Font,
    /// Base font for `Code` spans (bold/italic still derive from it).
    pub code: &'a Font,
}

/// Builds styled text runs for a single line by blending syntax spans and active selection.
pub fn build_line_text_runs(
    line_text: &str,
    spans: &[StyleSpan],
    selection_line_range: Option<(usize, usize)>,
    fonts: &RunFonts,
    theme: &EditorTheme,
    is_code_block: bool,
    is_checked_task: bool,
) -> Vec<TextRun> {
    if line_text.is_empty() {
        return Vec::new();
    }

    let segments = split_line_intervals(line_text.len(), spans, selection_line_range);
    let mut runs = Vec::with_capacity(segments.len());

    for segment in segments {
        let resolved = segment.style.map(|s| theme.resolve_style(s));

        let color = if is_checked_task && !segment.is_selected {
            theme.syntax.comment
        } else {
            resolved
                .as_ref()
                .map(|r| r.color)
                .unwrap_or(theme.foreground)
        };

        let background_color = if segment.is_selected {
            Some(theme.selection)
        } else if is_code_block {
            None
        } else {
            resolved.as_ref().and_then(|r| r.background)
        };
        let mut run_font = if matches!(segment.style, Some(StyleValue::Tag(HighlightTag::Code))) {
            fonts.code.clone()
        } else {
            fonts.base.clone()
        };
        if let Some(r) = resolved.as_ref() {
            if r.bold {
                run_font.weight = FontWeight::BOLD;
            }
            if r.italic {
                run_font.style = FontStyle::Italic;
            }
        }

        let underline = resolved
            .as_ref()
            .and_then(|r| r.underline)
            .map(|u| match u {
                UnderlineDecoration::Solid => UnderlineStyle {
                    color: Some(color),
                    thickness: px(1.0),
                    wavy: false,
                },
                UnderlineDecoration::Wavy => UnderlineStyle {
                    color: Some(theme.syntax.error),
                    thickness: px(1.2),
                    wavy: true,
                },
            });

        let strikethrough = if is_checked_task {
            Some(StrikethroughStyle {
                thickness: px(1.0),
                color: Some(theme.syntax.comment),
            })
        } else {
            resolved.as_ref().and_then(|r| {
                if r.strikethrough {
                    Some(StrikethroughStyle {
                        color: Some(color),
                        thickness: px(1.0),
                    })
                } else {
                    None
                }
            })
        };

        runs.push(TextRun {
            len: segment.range.end - segment.range.start,
            font: run_font,
            color,
            background_color,
            underline,
            strikethrough,
        });
    }

    let mut merged: Vec<TextRun> = Vec::with_capacity(runs.len());
    for run in runs {
        if let Some(last) = merged.last_mut()
            && last.font == run.font
            && last.color == run.color
            && last.background_color == run.background_color
            && last.underline == run.underline
            && last.strikethrough == run.strikethrough
        {
            last.len += run.len;
            continue;
        }
        merged.push(run);
    }

    merged
}

/// Data computed during prepaint for each visible line.
struct PreparedLine {
    gutter_num: Option<(Point<Pixels>, ShapedLine)>,
    text_origin: Point<Pixels>,
    text_line: WrappedLine,
    line_height: Pixels,
    quote_bar_quad: Option<PaintQuad>,
    code_block_bg_quad: Option<PaintQuad>,
    thematic_break_quad: Option<PaintQuad>,
    empty_selection_quad: Option<PaintQuad>,
    cursor_quad: Option<PaintQuad>,
    task_checkbox_quad: Option<PaintQuad>,
    /// Highlight-all search washes for this line (painted under the text).
    search_match_quads: Vec<PaintQuad>,
}

/// Intersects ascending `matches` with `[line_start, line_end)`, advancing
/// `*cursor` past ranges ending at or before the line. Returns clamped
/// overlapping sub-ranges. Both inputs ascend, so a full frame walks each
/// list once.
fn wash_ranges_for_line(
    matches: &[Range<usize>],
    cursor: &mut usize,
    line_start: usize,
    line_end: usize,
) -> Vec<Range<usize>> {
    while *cursor < matches.len() && matches[*cursor].end <= line_start {
        *cursor += 1;
    }
    let mut out = Vec::new();
    let mut j = *cursor;
    while j < matches.len() && matches[j].start < line_end {
        let start = matches[j].start.max(line_start);
        let end = matches[j].end.min(line_end);
        if end > start {
            out.push(start..end);
        }
        j += 1;
    }
    out
}

/// The state struct (T) passed from `prepaint` to `paint`.
struct EditorCanvasPrepaint {
    background_quad: PaintQuad,
    lines: Vec<PreparedLine>,
}

/// The GPUI canvas element responsible for shaping and rendering text runs and quads.
#[derive(IntoElement)]
pub struct EditorCanvas {
    editor: Entity<Editor>,
}

impl EditorCanvas {
    /// Creates a new canvas element linked to the given editor entity.
    pub fn new(editor: Entity<Editor>) -> Self {
        EditorCanvas { editor }
    }
}

impl RenderOnce for EditorCanvas {
    fn render(self, window: &mut Window, _cx: &mut App) -> impl IntoElement {
        window.request_animation_frame();

        let editor_handle = self.editor.clone();

        canvas(
            move |bounds, window, cx| {
                editor_handle.update(cx, |editor, _| {
                    editor.last_bounds = Some(bounds);
                    // One sample per presented frame for the FPS testing HUD.
                    editor.frame_stats.record();
                    // Probe + auto-select when inputs change: explicit families,
                    // code override, or host font. Explicit families are trusted
                    // verbatim (probed for the indicator only); otherwise the
                    // first candidate with bold + italic faces wins.
                    let host = window.text_style().font();
                    let key = (
                        editor.config.font_family.clone(),
                        editor.config.code_font_family.clone(),
                        host.clone(),
                    );
                    if editor.face_probe_key.as_ref() != Some(&key) {
                        let text_system = window.text_system();
                        let mut probe_one = |family: &str| {
                            let mut base = host.clone();
                            // gpui 0.2.2's SharedString is ArcCow<'static, str>:
                            // `&str.into()` would demand a 'static borrow, so
                            // copy through the owning constructor instead.
                            base.family = SharedString::new(family);
                            let base_id = text_system.resolve_font(&base);
                            let mut bold = base.clone();
                            bold.weight = FontWeight::BOLD;
                            let mut italic = base.clone();
                            italic.style = FontStyle::Italic;
                            (
                                text_system.resolve_font(&bold) != base_id,
                                text_system.resolve_font(&italic) != base_id,
                            )
                        };
                        let candidates = editor.config.font_candidates();
                        let (selected, mut availability) = match editor.config.font_family.clone() {
                            Some(explicit) => {
                                let (bold, italic) = probe_one(explicit.as_ref());
                                (
                                    Some(explicit),
                                    crate::editor::FaceAvailability { bold, italic },
                                )
                            }
                            None => match crate::config::EditorConfig::pick_family(
                                &candidates,
                                &mut probe_one,
                            ) {
                                Some(winner) => {
                                    let (bold, italic) = probe_one(winner.as_ref());
                                    (
                                        Some(winner.clone()),
                                        crate::editor::FaceAvailability { bold, italic },
                                    )
                                }
                                None => (
                                    None,
                                    crate::editor::FaceAvailability {
                                        bold: false,
                                        italic: false,
                                    },
                                ),
                            },
                        };
                        // An explicitly overridden code family gets its own say.
                        if let Some(code_family) = &editor.config.code_font_family
                            && Some(code_family) != selected.as_ref()
                        {
                            let (bold, italic) = probe_one(code_family.as_ref());
                            availability.bold &= bold;
                            availability.italic &= italic;
                        }
                        editor.selected_font_family = selected;
                        editor.face_availability = Some(availability);
                        editor.face_probe_key = Some(key);
                    }
                });
                // Take the cache out for the frame: the read guard below borrows
                // the editor immutably, so the cache travels as a local.
                let mut layout_cache =
                    editor_handle.update(cx, |editor, _| std::mem::take(&mut editor.layout_cache));
                let editor = editor_handle.read(cx);
                let theme = editor.theme.clone();
                let config = editor.config.clone();
                let host_font = window.text_style().font();
                let font = editor.resolved_base_font(&host_font);
                let code_font = editor.resolved_code_font(&host_font);

                let gutter_width = if config.line_numbers {
                    px(48.0)
                } else {
                    px(0.0)
                };
                let text_origin_x = bounds.left() + gutter_width + px(12.0);

                // Per-line override point: `cached.allow_wrap` (set by the
                // highlighter) can opt rows out of soft-wrapping below.
                let wrap_width = if config.line_wrap {
                    let available = bounds.size.width - gutter_width - px(24.0);
                    Some(available.max(px(50.0)))
                } else {
                    None
                };

                let total_lines = editor.buffer.len_lines();
                let total_bytes = editor.buffer.len_bytes();
                let scroll_row = editor.scroll_row;

                let cursor_offset = editor.buffer.cursor_offset();
                let cursor_point = editor.buffer.cursor_point();
                let selection = editor.selection;

                let is_all_selected = selection.is_some_and(|s| {
                    let range = s.byte_range();
                    !s.is_empty() && range.start == 0 && range.end == total_bytes
                });

                let mut lines: Vec<PreparedLine> = Vec::new();
                let mut visible_line_layouts: Vec<crate::editor::VisibleLineLayout> = Vec::new();
                let mut current_y = bounds.top();
                let mut computed_cursor_pixel = None;
                // Highlight-all wash, cloned once per frame; the per-line walk
                // below advances linearly through both ascending lists.
                let search_wash: Vec<Range<usize>> = if editor.search_highlight_all {
                    editor.search_matches.clone()
                } else {
                    Vec::new()
                };
                let mut wash_cursor = 0;

                for row in scroll_row..total_lines {
                    if current_y >= bounds.bottom() {
                        break;
                    }

                    let raw_line = editor.buffer.line_to_string(row);
                    let line_text = raw_line.trim_end_matches(['\r', '\n']);
                    let line_start_byte = editor.buffer.point_to_offset(BufferPoint::new(row, 0));
                    let line_end_byte = line_start_byte + line_text.len();

                    let gutter_num = if config.line_numbers {
                        let is_cursor_row = cursor_point.row == row;
                        let line_num_str = format!("{:>3}", row + 1);
                        let num_color = if is_cursor_row {
                            theme.line_number_active
                        } else {
                            theme.line_number
                        };

                        let shaped = window.text_system().shape_line(
                            line_num_str.into(),
                            config.font_size,
                            &[TextRun {
                                len: 3,
                                font: font.clone(),
                                color: num_color,
                                background_color: None,
                                underline: None,
                                strikethrough: None,
                            }],
                            None,
                        );

                        Some((point(bounds.left() + px(8.0), current_y), shaped))
                    } else {
                        None
                    };

                    let cursor_row = cursor_point.row;
                    let highlighter_rev = editor.highlighter_rev;
                    let cached = layout_cache.cached_input(
                        &editor.buffer,
                        editor.highlighter.as_deref(),
                        highlighter_rev,
                        cursor_row,
                        row,
                        line_text,
                    );
                    let spans = &cached.spans;
                    let concealed = &cached.concealed;

                    // Pass original spans: structural tags on concealed bytes would
                    // otherwise be stripped and invisible to layout.
                    let metrics = LineMetrics::for_line(
                        line_text,
                        &concealed.display_text,
                        spans,
                        config.font_size,
                        config.line_height,
                    );

                    let selection_line_range = if let Some(sel) = selection {
                        let sel_range = sel.byte_range();
                        if sel_range.end > line_start_byte && sel_range.start < line_end_byte {
                            let raw_start = sel_range
                                .start
                                .saturating_sub(line_start_byte)
                                .min(line_text.len());
                            let raw_end = (sel_range.end - line_start_byte).min(line_text.len());
                            let sel_start = concealed.source_to_display(raw_start);
                            let sel_end = concealed.source_to_display(raw_end);
                            if sel_end > sel_start {
                                Some((sel_start, sel_end))
                            } else {
                                None
                            }
                        } else {
                            None
                        }
                    } else {
                        None
                    };

                    let has_task = metrics.task_state.is_some();
                    let is_concealed_task =
                        has_task && line_text.len() != concealed.display_text.len();
                    let is_checked_task = is_concealed_task && metrics.task_state == Some(true);

                    let (task_checkbox_quad, line_text_origin_x) = if is_concealed_task {
                        let checked = metrics.task_state.unwrap();
                        let indent = line_text.len() - line_text.trim_start().len();
                        let box_size = px(15.0);
                        let box_x = text_origin_x + px((indent as f32) * 8.0);
                        let box_y = current_y + (metrics.line_height - box_size) / 2.0;

                        if checked {
                            let quad = fill(
                                Bounds::new(point(box_x, box_y), size(box_size, box_size)),
                                theme.syntax.function,
                            )
                            .corner_radii(px(3.5))
                            .border_widths(px(1.5))
                            .border_color(theme.syntax.function);

                            (Some(quad), box_x + px(24.0))
                        } else {
                            let quad = fill(
                                Bounds::new(point(box_x, box_y), size(box_size, box_size)),
                                gpui::hsla(0.65, 0.4, 0.6, 0.1),
                            )
                            .corner_radii(px(3.5))
                            .border_widths(px(1.5))
                            .border_color(theme.syntax.comment);
                            (Some(quad), box_x + px(24.0))
                        }
                    } else {
                        (None, text_origin_x)
                    };

                    let fonts = RunFonts {
                        base: &font,
                        code: &code_font,
                    };
                    let runs = build_line_text_runs(
                        &concealed.display_text,
                        &concealed.spans,
                        selection_line_range,
                        &fonts,
                        &theme,
                        metrics.is_code_block,
                        is_checked_task,
                    );

                    let text_line = window
                        .text_system()
                        .shape_text(
                            concealed.display_text.clone().into(),
                            metrics.font_size,
                            &runs,
                            wrap_width.filter(|_| cached.allow_wrap),
                            None,
                        )
                        .ok()
                        .and_then(|mut l| l.pop())
                        .unwrap_or_default();

                    let line_visual_lines = text_line.wrap_boundaries.len() + 1;
                    let line_total_height = metrics.line_height * line_visual_lines;

                    #[allow(unused_mut)]
                    let mut visible_links = Vec::new();
                    // Link ranges come from the same cached input: no second parse.
                    for (src_range, url) in &cached.link_src {
                        let disp_start = concealed.source_to_display(src_range.start);
                        let disp_end = concealed.source_to_display(src_range.end);
                        if disp_start < disp_end {
                            let start_pt =
                                text_line.position_for_index(disp_start, metrics.line_height);
                            let end_pt =
                                text_line.position_for_index(disp_end, metrics.line_height);
                            if let (Some(s), Some(e)) = (start_pt, end_pt) {
                                let width = if e.x > s.x { e.x - s.x } else { px(20.0) };
                                visible_links.push(crate::editor::VisibleLink {
                                    bounds: Bounds::new(
                                        point(line_text_origin_x + s.x, current_y + s.y),
                                        size(width.max(px(5.0)), metrics.line_height),
                                    ),
                                    url: url.clone(),
                                });
                            }
                        }
                    }

                    let mut search_match_quads = Vec::new();
                    for sub in wash_ranges_for_line(
                        &search_wash,
                        &mut wash_cursor,
                        line_start_byte,
                        line_end_byte,
                    ) {
                        let raw_start = sub
                            .start
                            .saturating_sub(line_start_byte)
                            .min(line_text.len());
                        let raw_end = sub.end.saturating_sub(line_start_byte).min(line_text.len());
                        let disp_start = concealed.source_to_display(raw_start);
                        let disp_end = concealed.source_to_display(raw_end);
                        if disp_end <= disp_start {
                            continue;
                        }
                        let (Some(s), Some(e)) = (
                            text_line.position_for_index(disp_start, metrics.line_height),
                            text_line.position_for_index(disp_end, metrics.line_height),
                        ) else {
                            continue;
                        };
                        // Same visual line: exact wash. Wrapped matches fall
                        // back to the remainder of the first visual line.
                        let (x, width) = if s.y == e.y && e.x > s.x {
                            (s.x, e.x - s.x)
                        } else {
                            let edge =
                                (bounds.right() - px(12.0)).max(line_text_origin_x + s.x + px(4.0));
                            (s.x, edge - (line_text_origin_x + s.x))
                        };
                        search_match_quads.push(fill(
                            Bounds::new(
                                point(line_text_origin_x + x, current_y + s.y),
                                size(width.max(px(4.0)), metrics.line_height),
                            ),
                            theme.search_match,
                        ));
                    }

                    let quote_bar_quad = if metrics.is_quote {
                        Some(fill(
                            Bounds::new(
                                point(bounds.left() + gutter_width + px(4.0), current_y),
                                size(px(3.0), line_total_height),
                            ),
                            theme.syntax.comment,
                        ))
                    } else {
                        None
                    };

                    let code_block_bg_quad = if metrics.is_code_block {
                        let bg_width = (bounds.size.width - gutter_width - px(8.0)).max(px(0.0));
                        Some(fill(
                            Bounds::new(
                                point(bounds.left() + gutter_width + px(4.0), current_y),
                                size(bg_width, line_total_height),
                            ),
                            theme.syntax.code_bg,
                        ))
                    } else {
                        None
                    };

                    let thematic_break_quad = if metrics.is_thematic_break {
                        let width = (bounds.size.width - gutter_width - px(24.0)).max(px(0.0));
                        let line_y = current_y + metrics.line_height / 2.0;
                        Some(fill(
                            Bounds::new(point(text_origin_x, line_y), size(width, px(1.0))),
                            theme.syntax.punctuation,
                        ))
                    } else {
                        None
                    };

                    let cursor_quad = if cursor_point.row == row {
                        let col_in_line = cursor_offset
                            .saturating_sub(line_start_byte)
                            .min(line_text.len());
                        let col_in_display = concealed.source_to_display(col_in_line);
                        let pos = text_line
                            .position_for_index(col_in_display, metrics.line_height)
                            .unwrap_or(point(px(0.0), px(0.0)));

                        computed_cursor_pixel = Some(point(
                            text_origin_x + pos.x,
                            current_y + pos.y + metrics.line_height,
                        ));

                        let should_draw_cursor =
                            !is_all_selected && (!config.cursor_blink || editor.cursor_visible);

                        if should_draw_cursor {
                            let style = if config.block_cursor {
                                twrite_core::CursorStyle::Block
                            } else {
                                editor.cursor_style
                            };

                            match style {
                                twrite_core::CursorStyle::Hidden => None,
                                twrite_core::CursorStyle::Block => Some(fill(
                                    Bounds::new(
                                        point(text_origin_x + pos.x, current_y + pos.y),
                                        size(px(8.5), metrics.line_height),
                                    ),
                                    theme.cursor,
                                )),
                                twrite_core::CursorStyle::Underline => Some(fill(
                                    Bounds::new(
                                        point(
                                            text_origin_x + pos.x,
                                            current_y + pos.y + metrics.line_height - px(2.0),
                                        ),
                                        size(px(8.5), px(2.0)),
                                    ),
                                    theme.cursor,
                                )),
                                twrite_core::CursorStyle::Bar => Some(fill(
                                    Bounds::new(
                                        point(text_origin_x + pos.x, current_y + pos.y),
                                        size(px(2.0), metrics.line_height),
                                    ),
                                    theme.cursor,
                                )),
                            }
                        } else {
                            None
                        }
                    } else {
                        None
                    };

                    let empty_selection_quad = if line_text.is_empty() {
                        if let Some(sel) = selection {
                            let sel_range = sel.byte_range();
                            if sel_range.start <= line_start_byte && sel_range.end > line_start_byte
                            {
                                Some(fill(
                                    Bounds::new(
                                        point(text_origin_x, current_y),
                                        size(px(6.0), metrics.line_height),
                                    ),
                                    theme.selection,
                                ))
                            } else {
                                None
                            }
                        } else {
                            None
                        }
                    } else {
                        None
                    };

                    lines.push(PreparedLine {
                        gutter_num,
                        text_origin: point(line_text_origin_x, current_y),
                        text_line,
                        line_height: metrics.line_height,
                        quote_bar_quad,
                        code_block_bg_quad,
                        thematic_break_quad,
                        empty_selection_quad,
                        cursor_quad,
                        task_checkbox_quad,
                        search_match_quads,
                    });

                    let checkbox_box_x = if has_task {
                        let indent = line_text.len() - line_text.trim_start().len();
                        text_origin_x + px((indent as f32) * 8.0)
                    } else {
                        px(0.0)
                    };

                    visible_line_layouts.push(crate::editor::VisibleLineLayout {
                        row,
                        top: current_y,
                        bottom: current_y + line_total_height,
                        line_start_byte,
                        line_len_bytes: line_text.len(),
                        text_origin_x: line_text_origin_x,
                        line_height: metrics.line_height,
                        is_task_checkbox: has_task,
                        checkbox_box_x,
                        task_state: metrics.task_state,
                        links: visible_links,
                    });

                    current_y += line_total_height;
                }

                editor_handle.update(cx, |editor, _| {
                    editor.layout_cache = layout_cache;
                    editor.last_cursor_pixel = computed_cursor_pixel;
                    editor.visible_lines = visible_line_layouts;
                });

                EditorCanvasPrepaint {
                    background_quad: fill(bounds, theme.background),
                    lines,
                }
            },
            move |_bounds, prepaint, window, cx| {
                window.paint_quad(prepaint.background_quad);

                for line in prepaint.lines {
                    let is_break = line.thematic_break_quad.is_some();

                    if let Some(code_bg) = line.code_block_bg_quad {
                        window.paint_quad(code_bg);
                    }
                    if let Some(quote_bar) = line.quote_bar_quad {
                        window.paint_quad(quote_bar);
                    }
                    if let Some(thematic_break) = line.thematic_break_quad {
                        window.paint_quad(thematic_break);
                    }
                    if let Some(cb_quad) = line.task_checkbox_quad {
                        window.paint_quad(cb_quad);
                    }

                    if let Some((origin, shaped_num)) = line.gutter_num {
                        // gpui 0.2.2 bakes TextAlign::Left into ShapedLine::paint.
                        let _ = shaped_num.paint(origin, line.line_height, window, cx);
                    }

                    if !is_break {
                        let _ = line.text_line.paint_background(
                            line.text_origin,
                            line.line_height,
                            TextAlign::Left,
                            None,
                            window,
                            cx,
                        );
                    }

                    for quad in line.search_match_quads {
                        window.paint_quad(quad);
                    }

                    if let Some(empty_sel) = line.empty_selection_quad {
                        window.paint_quad(empty_sel);
                    }

                    if let Some(cursor_quad) = line.cursor_quad {
                        window.paint_quad(cursor_quad);
                    }

                    if !is_break {
                        let _ = line.text_line.paint(
                            line.text_origin,
                            line.line_height,
                            TextAlign::Left,
                            None,
                            window,
                            cx,
                        );
                    }
                }
            },
        )
        .size_full()
    }
}

#[cfg(test)]
mod tests {
    use super::wash_ranges_for_line;
    use std::ops::Range;

    fn matches_vec(ranges: &[Range<usize>]) -> Vec<Range<usize>> {
        ranges.to_vec()
    }

    #[test]
    fn wash_ranges_clip_and_advance_linearly() {
        let matches = matches_vec(&[0..3, 8..11, 20..25]);
        let mut cursor = 0;

        // Line 0..7 overlaps only the first match, clamped.
        assert_eq!(
            wash_ranges_for_line(&matches, &mut cursor, 0, 7),
            vec![0..3]
        );
        // Next line skips the consumed range without rescanning.
        assert_eq!(
            wash_ranges_for_line(&matches, &mut cursor, 7, 15),
            vec![8..11]
        );
        assert_eq!(cursor, 1);
        // No overlap yields nothing but keeps the cursor.
        assert!(wash_ranges_for_line(&matches, &mut cursor, 12, 18).is_empty());
        assert_eq!(
            wash_ranges_for_line(&matches, &mut cursor, 18, 30),
            vec![20..25]
        );
        // Past the end stays empty.
        assert!(wash_ranges_for_line(&matches, &mut cursor, 30, 40).is_empty());
    }

    #[test]
    fn wash_ranges_split_cross_line_matches() {
        let matches = matches_vec(&[5..15, 30..35]);
        let mut cursor = 0;
        assert_eq!(
            wash_ranges_for_line(&matches, &mut cursor, 0, 10),
            vec![5..10]
        );
        assert_eq!(
            wash_ranges_for_line(&matches, &mut cursor, 10, 20),
            vec![10..15]
        );
        assert_eq!(
            wash_ranges_for_line(&matches, &mut cursor, 20, 40),
            vec![30..35]
        );
    }
}