//! This module defines where the text should be displayed in an [`Editor`][Editor]. //! //! Not literally though - rendering, layout and all that jazz is a responsibility of [`EditorElement`][EditorElement]. //! Instead, [`DisplayMap`] decides where Inlays/Inlay hints are displayed, when //! to apply a soft wrap, where to add fold indicators, whether there are any tabs in the buffer that //! we display as spaces and where to display custom blocks (like diagnostics). //! Seems like a lot? That's because it is. [`DisplayMap`] is conceptually made up //! of several smaller structures that form a hierarchy (starting at the bottom): //! - [`InlayMap`] that decides where the [`Inlay`]s should be displayed. //! - [`FoldMap`] that decides where the fold indicators should be; it also tracks parts of a source file that are currently folded. //! - [`TabMap`] that keeps track of hard tabs in a buffer. //! - [`WrapMap`] that handles soft wrapping. //! - [`BlockMap`] that tracks custom blocks such as diagnostics that should be displayed within buffer. //! - [`DisplayMap`] that adds background highlights to the regions of text. //! Each one of those builds on top of preceding map. //! //! [Editor]: crate::Editor //! [EditorElement]: crate::element::EditorElement mod block_map; mod crease_map; mod fold_map; mod inlay_map; pub(crate) mod invisibles; mod tab_map; mod wrap_map; use crate::{ hover_links::InlayHighlight, movement::TextLayoutDetails, EditorStyle, InlayId, RowExt, }; pub use block_map::{ Block, BlockBufferRows, BlockChunks as DisplayChunks, BlockContext, BlockId, BlockMap, BlockPlacement, BlockPoint, BlockProperties, BlockStyle, CustomBlockId, RenderBlock, }; use block_map::{BlockRow, BlockSnapshot}; use collections::{HashMap, HashSet}; pub use crease_map::*; pub use fold_map::{Fold, FoldId, FoldPlaceholder, FoldPoint}; use fold_map::{FoldMap, FoldMapWriter, FoldOffset, FoldSnapshot}; use gpui::{ AnyElement, Font, HighlightStyle, LineLayout, Model, ModelContext, Pixels, UnderlineStyle, }; pub(crate) use inlay_map::Inlay; use inlay_map::{InlayMap, InlaySnapshot}; pub use inlay_map::{InlayOffset, InlayPoint}; use invisibles::{is_invisible, replacement}; use language::{ language_settings::language_settings, ChunkRenderer, OffsetUtf16, Point, Subscription as BufferSubscription, }; use lsp::DiagnosticSeverity; use multi_buffer::{ Anchor, AnchorRangeExt, MultiBuffer, MultiBufferPoint, MultiBufferRow, MultiBufferSnapshot, ToOffset, ToPoint, }; use serde::Deserialize; use std::{ any::TypeId, borrow::Cow, fmt::Debug, iter, num::NonZeroU32, ops::{Add, Range, Sub}, sync::Arc, }; use sum_tree::{Bias, TreeMap}; use tab_map::{TabMap, TabSnapshot}; use text::{Edit, LineIndent}; use ui::{div, px, IntoElement, ParentElement, SharedString, Styled, WindowContext}; use unicode_segmentation::UnicodeSegmentation; use wrap_map::{WrapMap, WrapSnapshot}; #[derive(Copy, Clone, Debug, PartialEq, Eq)] pub enum FoldStatus { Folded, Foldable, } pub type RenderFoldToggle = Arc AnyElement>; pub trait ToDisplayPoint { fn to_display_point(&self, map: &DisplaySnapshot) -> DisplayPoint; } type TextHighlights = TreeMap, Arc<(HighlightStyle, Vec>)>>; type InlayHighlights = TreeMap>; /// Decides how text in a [`MultiBuffer`] should be displayed in a buffer, handling inlay hints, /// folding, hard tabs, soft wrapping, custom blocks (like diagnostics), and highlighting. /// /// See the [module level documentation](self) for more information. pub struct DisplayMap { /// The buffer that we are displaying. buffer: Model, buffer_subscription: BufferSubscription, /// Decides where the [`Inlay`]s should be displayed. inlay_map: InlayMap, /// Decides where the fold indicators should be and tracks parts of a source file that are currently folded. fold_map: FoldMap, /// Keeps track of hard tabs in a buffer. tab_map: TabMap, /// Handles soft wrapping. wrap_map: Model, /// Tracks custom blocks such as diagnostics that should be displayed within buffer. block_map: BlockMap, /// Regions of text that should be highlighted. text_highlights: TextHighlights, /// Regions of inlays that should be highlighted. inlay_highlights: InlayHighlights, /// A container for explicitly foldable ranges, which supersede indentation based fold range suggestions. crease_map: CreaseMap, pub(crate) fold_placeholder: FoldPlaceholder, pub clip_at_line_ends: bool, pub(crate) masked: bool, } impl DisplayMap { #[allow(clippy::too_many_arguments)] pub fn new( buffer: Model, font: Font, font_size: Pixels, wrap_width: Option, show_excerpt_controls: bool, buffer_header_height: u32, excerpt_header_height: u32, excerpt_footer_height: u32, fold_placeholder: FoldPlaceholder, cx: &mut ModelContext, ) -> Self { let buffer_subscription = buffer.update(cx, |buffer, _| buffer.subscribe()); let tab_size = Self::tab_size(&buffer, cx); let buffer_snapshot = buffer.read(cx).snapshot(cx); let crease_map = CreaseMap::new(&buffer_snapshot); let (inlay_map, snapshot) = InlayMap::new(buffer_snapshot); let (fold_map, snapshot) = FoldMap::new(snapshot); let (tab_map, snapshot) = TabMap::new(snapshot, tab_size); let (wrap_map, snapshot) = WrapMap::new(snapshot, font, font_size, wrap_width, cx); let block_map = BlockMap::new( snapshot, show_excerpt_controls, buffer_header_height, excerpt_header_height, excerpt_footer_height, ); cx.observe(&wrap_map, |_, _, cx| cx.notify()).detach(); DisplayMap { buffer, buffer_subscription, fold_map, inlay_map, tab_map, wrap_map, block_map, crease_map, fold_placeholder, text_highlights: Default::default(), inlay_highlights: Default::default(), clip_at_line_ends: false, masked: false, } } pub fn snapshot(&mut self, cx: &mut ModelContext) -> DisplaySnapshot { let buffer_snapshot = self.buffer.read(cx).snapshot(cx); let edits = self.buffer_subscription.consume().into_inner(); let (inlay_snapshot, edits) = self.inlay_map.sync(buffer_snapshot, edits); let (fold_snapshot, edits) = self.fold_map.read(inlay_snapshot.clone(), edits); let tab_size = Self::tab_size(&self.buffer, cx); let (tab_snapshot, edits) = self.tab_map.sync(fold_snapshot.clone(), edits, tab_size); let (wrap_snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(tab_snapshot.clone(), edits, cx)); let block_snapshot = self.block_map.read(wrap_snapshot.clone(), edits).snapshot; DisplaySnapshot { buffer_snapshot: self.buffer.read(cx).snapshot(cx), fold_snapshot, inlay_snapshot, tab_snapshot, wrap_snapshot, block_snapshot, crease_snapshot: self.crease_map.snapshot(), text_highlights: self.text_highlights.clone(), inlay_highlights: self.inlay_highlights.clone(), clip_at_line_ends: self.clip_at_line_ends, masked: self.masked, fold_placeholder: self.fold_placeholder.clone(), } } pub fn set_state(&mut self, other: &DisplaySnapshot, cx: &mut ModelContext) { self.fold( other .folds_in_range(0..other.buffer_snapshot.len()) .map(|fold| { ( fold.range.to_offset(&other.buffer_snapshot), fold.placeholder.clone(), ) }), cx, ); } /// Creates folds for the given ranges. pub fn fold( &mut self, ranges: impl IntoIterator, FoldPlaceholder)>, cx: &mut ModelContext, ) { self.update_fold_map(cx, |fold_map| fold_map.fold(ranges)) } /// Removes any folds with the given ranges. pub fn remove_folds_with_type( &mut self, ranges: impl IntoIterator>, type_id: TypeId, cx: &mut ModelContext, ) { self.update_fold_map(cx, |fold_map| fold_map.remove_folds(ranges, type_id)) } /// Removes any folds whose ranges intersect any of the given ranges. pub fn unfold_intersecting( &mut self, ranges: impl IntoIterator>, inclusive: bool, cx: &mut ModelContext, ) { self.update_fold_map(cx, |fold_map| { fold_map.unfold_intersecting(ranges, inclusive) }) } fn update_fold_map( &mut self, cx: &mut ModelContext, callback: impl FnOnce(&mut FoldMapWriter) -> (FoldSnapshot, Vec>), ) { let snapshot = self.buffer.read(cx).snapshot(cx); let edits = self.buffer_subscription.consume().into_inner(); let tab_size = Self::tab_size(&self.buffer, cx); let (snapshot, edits) = self.inlay_map.sync(snapshot, edits); let (mut fold_map, snapshot, edits) = self.fold_map.write(snapshot, edits); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); self.block_map.read(snapshot, edits); let (snapshot, edits) = callback(&mut fold_map); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); self.block_map.read(snapshot, edits); } pub fn insert_creases( &mut self, creases: impl IntoIterator, cx: &mut ModelContext, ) -> Vec { let snapshot = self.buffer.read(cx).snapshot(cx); self.crease_map.insert(creases, &snapshot) } pub fn remove_creases( &mut self, crease_ids: impl IntoIterator, cx: &mut ModelContext, ) { let snapshot = self.buffer.read(cx).snapshot(cx); self.crease_map.remove(crease_ids, &snapshot) } pub fn insert_blocks( &mut self, blocks: impl IntoIterator>, cx: &mut ModelContext, ) -> Vec { let snapshot = self.buffer.read(cx).snapshot(cx); let edits = self.buffer_subscription.consume().into_inner(); let tab_size = Self::tab_size(&self.buffer, cx); let (snapshot, edits) = self.inlay_map.sync(snapshot, edits); let (snapshot, edits) = self.fold_map.read(snapshot, edits); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); let mut block_map = self.block_map.write(snapshot, edits); block_map.insert(blocks) } pub fn resize_blocks( &mut self, heights: HashMap, cx: &mut ModelContext, ) { let snapshot = self.buffer.read(cx).snapshot(cx); let edits = self.buffer_subscription.consume().into_inner(); let tab_size = Self::tab_size(&self.buffer, cx); let (snapshot, edits) = self.inlay_map.sync(snapshot, edits); let (snapshot, edits) = self.fold_map.read(snapshot, edits); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); let mut block_map = self.block_map.write(snapshot, edits); block_map.resize(heights); } pub fn replace_blocks(&mut self, renderers: HashMap) { self.block_map.replace_blocks(renderers); } pub fn remove_blocks(&mut self, ids: HashSet, cx: &mut ModelContext) { let snapshot = self.buffer.read(cx).snapshot(cx); let edits = self.buffer_subscription.consume().into_inner(); let tab_size = Self::tab_size(&self.buffer, cx); let (snapshot, edits) = self.inlay_map.sync(snapshot, edits); let (snapshot, edits) = self.fold_map.read(snapshot, edits); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); let mut block_map = self.block_map.write(snapshot, edits); block_map.remove(ids); } pub fn row_for_block( &mut self, block_id: CustomBlockId, cx: &mut ModelContext, ) -> Option { let snapshot = self.buffer.read(cx).snapshot(cx); let edits = self.buffer_subscription.consume().into_inner(); let tab_size = Self::tab_size(&self.buffer, cx); let (snapshot, edits) = self.inlay_map.sync(snapshot, edits); let (snapshot, edits) = self.fold_map.read(snapshot, edits); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); let block_map = self.block_map.read(snapshot, edits); let block_row = block_map.row_for_block(block_id)?; Some(DisplayRow(block_row.0)) } pub fn highlight_text( &mut self, type_id: TypeId, ranges: Vec>, style: HighlightStyle, ) { self.text_highlights .insert(Some(type_id), Arc::new((style, ranges))); } pub(crate) fn highlight_inlays( &mut self, type_id: TypeId, highlights: Vec, style: HighlightStyle, ) { for highlight in highlights { let update = self.inlay_highlights.update(&type_id, |highlights| { highlights.insert(highlight.inlay, (style, highlight.clone())) }); if update.is_none() { self.inlay_highlights.insert( type_id, TreeMap::from_ordered_entries([(highlight.inlay, (style, highlight))]), ); } } } pub fn text_highlights(&self, type_id: TypeId) -> Option<(HighlightStyle, &[Range])> { let highlights = self.text_highlights.get(&Some(type_id))?; Some((highlights.0, &highlights.1)) } pub fn clear_highlights(&mut self, type_id: TypeId) -> bool { let mut cleared = self.text_highlights.remove(&Some(type_id)).is_some(); cleared |= self.inlay_highlights.remove(&type_id).is_some(); cleared } pub fn set_font(&self, font: Font, font_size: Pixels, cx: &mut ModelContext) -> bool { self.wrap_map .update(cx, |map, cx| map.set_font_with_size(font, font_size, cx)) } pub fn set_wrap_width(&self, width: Option, cx: &mut ModelContext) -> bool { self.wrap_map .update(cx, |map, cx| map.set_wrap_width(width, cx)) } pub(crate) fn current_inlays(&self) -> impl Iterator { self.inlay_map.current_inlays() } pub(crate) fn splice_inlays( &mut self, to_remove: Vec, to_insert: Vec, cx: &mut ModelContext, ) { if to_remove.is_empty() && to_insert.is_empty() { return; } let buffer_snapshot = self.buffer.read(cx).snapshot(cx); let edits = self.buffer_subscription.consume().into_inner(); let (snapshot, edits) = self.inlay_map.sync(buffer_snapshot, edits); let (snapshot, edits) = self.fold_map.read(snapshot, edits); let tab_size = Self::tab_size(&self.buffer, cx); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); self.block_map.read(snapshot, edits); let (snapshot, edits) = self.inlay_map.splice(to_remove, to_insert); let (snapshot, edits) = self.fold_map.read(snapshot, edits); let (snapshot, edits) = self.tab_map.sync(snapshot, edits, tab_size); let (snapshot, edits) = self .wrap_map .update(cx, |map, cx| map.sync(snapshot, edits, cx)); self.block_map.read(snapshot, edits); } fn tab_size(buffer: &Model, cx: &mut ModelContext) -> NonZeroU32 { let buffer = buffer.read(cx).as_singleton().map(|buffer| buffer.read(cx)); let language = buffer .and_then(|buffer| buffer.language()) .map(|l| l.name()); let file = buffer.and_then(|buffer| buffer.file()); language_settings(language, file, cx).tab_size } #[cfg(test)] pub fn is_rewrapping(&self, cx: &gpui::AppContext) -> bool { self.wrap_map.read(cx).is_rewrapping() } pub fn show_excerpt_controls(&self) -> bool { self.block_map.show_excerpt_controls() } } #[derive(Debug, Default)] pub(crate) struct Highlights<'a> { pub text_highlights: Option<&'a TextHighlights>, pub inlay_highlights: Option<&'a InlayHighlights>, pub styles: HighlightStyles, } #[derive(Default, Debug, Clone, Copy)] pub struct HighlightStyles { pub inlay_hint: Option, pub suggestion: Option, } pub struct HighlightedChunk<'a> { pub text: &'a str, pub style: Option, pub is_tab: bool, pub renderer: Option, } impl<'a> HighlightedChunk<'a> { fn highlight_invisibles( self, editor_style: &'a EditorStyle, ) -> impl Iterator + 'a { let mut chars = self.text.chars().peekable(); let mut text = self.text; let style = self.style; let is_tab = self.is_tab; let renderer = self.renderer; iter::from_fn(move || { let mut prefix_len = 0; while let Some(&ch) = chars.peek() { if !is_invisible(ch) { prefix_len += ch.len_utf8(); chars.next(); continue; } if prefix_len > 0 { let (prefix, suffix) = text.split_at(prefix_len); text = suffix; return Some(HighlightedChunk { text: prefix, style, is_tab, renderer: renderer.clone(), }); } chars.next(); let (prefix, suffix) = text.split_at(ch.len_utf8()); text = suffix; if let Some(replacement) = replacement(ch) { let background = editor_style.status.hint_background; let underline = editor_style.status.hint; return Some(HighlightedChunk { text: prefix, style: None, is_tab: false, renderer: Some(ChunkRenderer { render: Arc::new(move |_| { div() .child(replacement) .bg(background) .text_decoration_1() .text_decoration_color(underline) .into_any_element() }), constrain_width: false, }), }); } else { let invisible_highlight = HighlightStyle { background_color: Some(editor_style.status.hint_background), underline: Some(UnderlineStyle { color: Some(editor_style.status.hint), thickness: px(1.), wavy: false, }), ..Default::default() }; let invisible_style = if let Some(mut style) = style { style.highlight(invisible_highlight); style } else { invisible_highlight }; return Some(HighlightedChunk { text: prefix, style: Some(invisible_style), is_tab: false, renderer: renderer.clone(), }); } } if !text.is_empty() { let remainder = text; text = ""; Some(HighlightedChunk { text: remainder, style, is_tab, renderer: renderer.clone(), }) } else { None } }) } } #[derive(Clone)] pub struct DisplaySnapshot { pub buffer_snapshot: MultiBufferSnapshot, pub fold_snapshot: FoldSnapshot, pub crease_snapshot: CreaseSnapshot, inlay_snapshot: InlaySnapshot, tab_snapshot: TabSnapshot, wrap_snapshot: WrapSnapshot, block_snapshot: BlockSnapshot, text_highlights: TextHighlights, inlay_highlights: InlayHighlights, clip_at_line_ends: bool, masked: bool, pub(crate) fold_placeholder: FoldPlaceholder, } impl DisplaySnapshot { #[cfg(test)] pub fn fold_count(&self) -> usize { self.fold_snapshot.fold_count() } pub fn is_empty(&self) -> bool { self.buffer_snapshot.len() == 0 } pub fn buffer_rows( &self, start_row: DisplayRow, ) -> impl Iterator> + '_ { self.block_snapshot .buffer_rows(BlockRow(start_row.0)) .map(|row| row.map(|row| MultiBufferRow(row.0))) } pub fn max_buffer_row(&self) -> MultiBufferRow { self.buffer_snapshot.max_buffer_row() } pub fn prev_line_boundary(&self, mut point: MultiBufferPoint) -> (Point, DisplayPoint) { loop { let mut inlay_point = self.inlay_snapshot.to_inlay_point(point); let mut fold_point = self.fold_snapshot.to_fold_point(inlay_point, Bias::Left); fold_point.0.column = 0; inlay_point = fold_point.to_inlay_point(&self.fold_snapshot); point = self.inlay_snapshot.to_buffer_point(inlay_point); let mut display_point = self.point_to_display_point(point, Bias::Left); *display_point.column_mut() = 0; let next_point = self.display_point_to_point(display_point, Bias::Left); if next_point == point { return (point, display_point); } point = next_point; } } pub fn next_line_boundary(&self, mut point: MultiBufferPoint) -> (Point, DisplayPoint) { loop { let mut inlay_point = self.inlay_snapshot.to_inlay_point(point); let mut fold_point = self.fold_snapshot.to_fold_point(inlay_point, Bias::Right); fold_point.0.column = self.fold_snapshot.line_len(fold_point.row()); inlay_point = fold_point.to_inlay_point(&self.fold_snapshot); point = self.inlay_snapshot.to_buffer_point(inlay_point); let mut display_point = self.point_to_display_point(point, Bias::Right); *display_point.column_mut() = self.line_len(display_point.row()); let next_point = self.display_point_to_point(display_point, Bias::Right); if next_point == point { return (point, display_point); } point = next_point; } } // used by line_mode selections and tries to match vim behavior pub fn expand_to_line(&self, range: Range) -> Range { let new_start = if range.start.row == 0 { MultiBufferPoint::new(0, 0) } else if range.start.row == self.max_buffer_row().0 || (range.end.column > 0 && range.end.row == self.max_buffer_row().0) { MultiBufferPoint::new( range.start.row - 1, self.buffer_snapshot .line_len(MultiBufferRow(range.start.row - 1)), ) } else { self.prev_line_boundary(range.start).0 }; let new_end = if range.end.column == 0 { range.end } else if range.end.row < self.max_buffer_row().0 { self.buffer_snapshot .clip_point(MultiBufferPoint::new(range.end.row + 1, 0), Bias::Left) } else { self.buffer_snapshot.max_point() }; new_start..new_end } pub fn point_to_display_point(&self, point: MultiBufferPoint, bias: Bias) -> DisplayPoint { let inlay_point = self.inlay_snapshot.to_inlay_point(point); let fold_point = self.fold_snapshot.to_fold_point(inlay_point, bias); let tab_point = self.tab_snapshot.to_tab_point(fold_point); let wrap_point = self.wrap_snapshot.tab_point_to_wrap_point(tab_point); let block_point = self.block_snapshot.to_block_point(wrap_point); DisplayPoint(block_point) } pub fn display_point_to_point(&self, point: DisplayPoint, bias: Bias) -> Point { self.inlay_snapshot .to_buffer_point(self.display_point_to_inlay_point(point, bias)) } pub fn display_point_to_inlay_offset(&self, point: DisplayPoint, bias: Bias) -> InlayOffset { self.inlay_snapshot .to_offset(self.display_point_to_inlay_point(point, bias)) } pub fn anchor_to_inlay_offset(&self, anchor: Anchor) -> InlayOffset { self.inlay_snapshot .to_inlay_offset(anchor.to_offset(&self.buffer_snapshot)) } pub fn display_point_to_anchor(&self, point: DisplayPoint, bias: Bias) -> Anchor { self.buffer_snapshot .anchor_at(point.to_offset(self, bias), bias) } fn display_point_to_inlay_point(&self, point: DisplayPoint, bias: Bias) -> InlayPoint { let block_point = point.0; let wrap_point = self.block_snapshot.to_wrap_point(block_point, bias); let tab_point = self.wrap_snapshot.to_tab_point(wrap_point); let fold_point = self.tab_snapshot.to_fold_point(tab_point, bias).0; fold_point.to_inlay_point(&self.fold_snapshot) } pub fn display_point_to_fold_point(&self, point: DisplayPoint, bias: Bias) -> FoldPoint { let block_point = point.0; let wrap_point = self.block_snapshot.to_wrap_point(block_point, bias); let tab_point = self.wrap_snapshot.to_tab_point(wrap_point); self.tab_snapshot.to_fold_point(tab_point, bias).0 } pub fn fold_point_to_display_point(&self, fold_point: FoldPoint) -> DisplayPoint { let tab_point = self.tab_snapshot.to_tab_point(fold_point); let wrap_point = self.wrap_snapshot.tab_point_to_wrap_point(tab_point); let block_point = self.block_snapshot.to_block_point(wrap_point); DisplayPoint(block_point) } pub fn max_point(&self) -> DisplayPoint { DisplayPoint(self.block_snapshot.max_point()) } /// Returns text chunks starting at the given display row until the end of the file pub fn text_chunks(&self, display_row: DisplayRow) -> impl Iterator { self.block_snapshot .chunks( display_row.0..self.max_point().row().next_row().0, false, self.masked, Highlights::default(), ) .map(|h| h.text) } /// Returns text chunks starting at the end of the given display row in reverse until the start of the file pub fn reverse_text_chunks(&self, display_row: DisplayRow) -> impl Iterator { (0..=display_row.0).rev().flat_map(move |row| { self.block_snapshot .chunks(row..row + 1, false, self.masked, Highlights::default()) .map(|h| h.text) .collect::>() .into_iter() .rev() }) } pub fn chunks( &self, display_rows: Range, language_aware: bool, highlight_styles: HighlightStyles, ) -> DisplayChunks<'_> { self.block_snapshot.chunks( display_rows.start.0..display_rows.end.0, language_aware, self.masked, Highlights { text_highlights: Some(&self.text_highlights), inlay_highlights: Some(&self.inlay_highlights), styles: highlight_styles, }, ) } pub fn highlighted_chunks<'a>( &'a self, display_rows: Range, language_aware: bool, editor_style: &'a EditorStyle, ) -> impl Iterator> { self.chunks( display_rows, language_aware, HighlightStyles { inlay_hint: Some(editor_style.inlay_hints_style), suggestion: Some(editor_style.suggestions_style), }, ) .flat_map(|chunk| { let mut highlight_style = chunk .syntax_highlight_id .and_then(|id| id.style(&editor_style.syntax)); if let Some(chunk_highlight) = chunk.highlight_style { if let Some(highlight_style) = highlight_style.as_mut() { highlight_style.highlight(chunk_highlight); } else { highlight_style = Some(chunk_highlight); } } let mut diagnostic_highlight = HighlightStyle::default(); if chunk.is_unnecessary { diagnostic_highlight.fade_out = Some(editor_style.unnecessary_code_fade); } if let Some(severity) = chunk.diagnostic_severity { // Omit underlines for HINT/INFO diagnostics on 'unnecessary' code. if severity <= DiagnosticSeverity::WARNING || !chunk.is_unnecessary { let diagnostic_color = super::diagnostic_style(severity, &editor_style.status); diagnostic_highlight.underline = Some(UnderlineStyle { color: Some(diagnostic_color), thickness: 1.0.into(), wavy: true, }); } } if let Some(highlight_style) = highlight_style.as_mut() { highlight_style.highlight(diagnostic_highlight); } else { highlight_style = Some(diagnostic_highlight); } HighlightedChunk { text: chunk.text, style: highlight_style, is_tab: chunk.is_tab, renderer: chunk.renderer, } .highlight_invisibles(editor_style) }) } pub fn layout_row( &self, display_row: DisplayRow, TextLayoutDetails { text_system, editor_style, rem_size, scroll_anchor: _, visible_rows: _, vertical_scroll_margin: _, }: &TextLayoutDetails, ) -> Arc { let mut runs = Vec::new(); let mut line = String::new(); let range = display_row..display_row.next_row(); for chunk in self.highlighted_chunks(range, false, editor_style) { line.push_str(chunk.text); let text_style = if let Some(style) = chunk.style { Cow::Owned(editor_style.text.clone().highlight(style)) } else { Cow::Borrowed(&editor_style.text) }; runs.push(text_style.to_run(chunk.text.len())) } if line.ends_with('\n') { line.pop(); if let Some(last_run) = runs.last_mut() { last_run.len -= 1; if last_run.len == 0 { runs.pop(); } } } let font_size = editor_style.text.font_size.to_pixels(*rem_size); text_system .layout_line(&line, font_size, &runs) .expect("we expect the font to be loaded because it's rendered by the editor") } pub fn x_for_display_point( &self, display_point: DisplayPoint, text_layout_details: &TextLayoutDetails, ) -> Pixels { let line = self.layout_row(display_point.row(), text_layout_details); line.x_for_index(display_point.column() as usize) } pub fn display_column_for_x( &self, display_row: DisplayRow, x: Pixels, details: &TextLayoutDetails, ) -> u32 { let layout_line = self.layout_row(display_row, details); layout_line.closest_index_for_x(x) as u32 } pub fn grapheme_at(&self, mut point: DisplayPoint) -> Option { point = DisplayPoint(self.block_snapshot.clip_point(point.0, Bias::Left)); let chars = self .text_chunks(point.row()) .flat_map(str::chars) .skip_while({ let mut column = 0; move |char| { let at_point = column >= point.column(); column += char.len_utf8() as u32; !at_point } }) .take_while({ let mut prev = false; move |char| { let now = char.is_ascii(); let end = char.is_ascii() && (char.is_ascii_whitespace() || prev); prev = now; !end } }); chars.collect::().graphemes(true).next().map(|s| { if let Some(invisible) = s.chars().next().filter(|&c| is_invisible(c)) { replacement(invisible).unwrap_or(s).to_owned().into() } else if s == "\n" { " ".into() } else { s.to_owned().into() } }) } pub fn buffer_chars_at(&self, mut offset: usize) -> impl Iterator + '_ { self.buffer_snapshot.chars_at(offset).map(move |ch| { let ret = (ch, offset); offset += ch.len_utf8(); ret }) } pub fn reverse_buffer_chars_at( &self, mut offset: usize, ) -> impl Iterator + '_ { self.buffer_snapshot .reversed_chars_at(offset) .map(move |ch| { offset -= ch.len_utf8(); (ch, offset) }) } pub fn clip_point(&self, point: DisplayPoint, bias: Bias) -> DisplayPoint { let mut clipped = self.block_snapshot.clip_point(point.0, bias); if self.clip_at_line_ends { clipped = self.clip_at_line_end(DisplayPoint(clipped)).0 } DisplayPoint(clipped) } pub fn clip_ignoring_line_ends(&self, point: DisplayPoint, bias: Bias) -> DisplayPoint { DisplayPoint(self.block_snapshot.clip_point(point.0, bias)) } pub fn clip_at_line_end(&self, point: DisplayPoint) -> DisplayPoint { let mut point = point.0; if point.column == self.line_len(DisplayRow(point.row)) { point.column = point.column.saturating_sub(1); point = self.block_snapshot.clip_point(point, Bias::Left); } DisplayPoint(point) } pub fn folds_in_range(&self, range: Range) -> impl Iterator where T: ToOffset, { self.fold_snapshot.folds_in_range(range) } pub fn blocks_in_range( &self, rows: Range, ) -> impl Iterator { self.block_snapshot .blocks_in_range(rows.start.0..rows.end.0) .map(|(row, block)| (DisplayRow(row), block)) } pub fn block_for_id(&self, id: BlockId) -> Option { self.block_snapshot.block_for_id(id) } pub fn intersects_fold(&self, offset: T) -> bool { self.fold_snapshot.intersects_fold(offset) } pub fn is_line_folded(&self, buffer_row: MultiBufferRow) -> bool { self.fold_snapshot.is_line_folded(buffer_row) } pub fn is_block_line(&self, display_row: DisplayRow) -> bool { self.block_snapshot.is_block_line(BlockRow(display_row.0)) } pub fn soft_wrap_indent(&self, display_row: DisplayRow) -> Option { let wrap_row = self .block_snapshot .to_wrap_point(BlockPoint::new(display_row.0, 0), Bias::Left) .row(); self.wrap_snapshot.soft_wrap_indent(wrap_row) } pub fn text(&self) -> String { self.text_chunks(DisplayRow(0)).collect() } pub fn line(&self, display_row: DisplayRow) -> String { let mut result = String::new(); for chunk in self.text_chunks(display_row) { if let Some(ix) = chunk.find('\n') { result.push_str(&chunk[0..ix]); break; } else { result.push_str(chunk); } } result } pub fn line_indent_for_buffer_row(&self, buffer_row: MultiBufferRow) -> LineIndent { let (buffer, range) = self .buffer_snapshot .buffer_line_for_row(buffer_row) .unwrap(); buffer.line_indent_for_row(range.start.row) } pub fn line_len(&self, row: DisplayRow) -> u32 { self.block_snapshot.line_len(BlockRow(row.0)) } pub fn longest_row(&self) -> DisplayRow { DisplayRow(self.block_snapshot.longest_row()) } pub fn starts_indent(&self, buffer_row: MultiBufferRow) -> bool { let max_row = self.buffer_snapshot.max_buffer_row(); if buffer_row >= max_row { return false; } let line_indent = self.line_indent_for_buffer_row(buffer_row); if line_indent.is_line_blank() { return false; } (buffer_row.0 + 1..=max_row.0) .find_map(|next_row| { let next_line_indent = self.line_indent_for_buffer_row(MultiBufferRow(next_row)); if next_line_indent.raw_len() > line_indent.raw_len() { Some(true) } else if !next_line_indent.is_line_blank() { Some(false) } else { None } }) .unwrap_or(false) } pub fn foldable_range( &self, buffer_row: MultiBufferRow, ) -> Option<(Range, FoldPlaceholder)> { let start = MultiBufferPoint::new(buffer_row.0, self.buffer_snapshot.line_len(buffer_row)); if let Some(crease) = self .crease_snapshot .query_row(buffer_row, &self.buffer_snapshot) { Some(( crease.range.to_point(&self.buffer_snapshot), crease.placeholder.clone(), )) } else if self.starts_indent(MultiBufferRow(start.row)) && !self.is_line_folded(MultiBufferRow(start.row)) { let start_line_indent = self.line_indent_for_buffer_row(buffer_row); let max_point = self.buffer_snapshot.max_point(); let mut end = None; for row in (buffer_row.0 + 1)..=max_point.row { let line_indent = self.line_indent_for_buffer_row(MultiBufferRow(row)); if !line_indent.is_line_blank() && line_indent.raw_len() <= start_line_indent.raw_len() { let prev_row = row - 1; end = Some(Point::new( prev_row, self.buffer_snapshot.line_len(MultiBufferRow(prev_row)), )); break; } } let mut row_before_line_breaks = end.unwrap_or(max_point); while row_before_line_breaks.row > start.row && self .buffer_snapshot .is_line_blank(MultiBufferRow(row_before_line_breaks.row)) { row_before_line_breaks.row -= 1; } row_before_line_breaks = Point::new( row_before_line_breaks.row, self.buffer_snapshot .line_len(MultiBufferRow(row_before_line_breaks.row)), ); Some((start..row_before_line_breaks, self.fold_placeholder.clone())) } else { None } } #[cfg(any(test, feature = "test-support"))] pub fn text_highlight_ranges( &self, ) -> Option>)>> { let type_id = TypeId::of::(); self.text_highlights.get(&Some(type_id)).cloned() } #[allow(unused)] #[cfg(any(test, feature = "test-support"))] pub(crate) fn inlay_highlights( &self, ) -> Option<&TreeMap> { let type_id = TypeId::of::(); self.inlay_highlights.get(&type_id) } pub fn buffer_header_height(&self) -> u32 { self.block_snapshot.buffer_header_height } pub fn excerpt_footer_height(&self) -> u32 { self.block_snapshot.excerpt_footer_height } pub fn excerpt_header_height(&self) -> u32 { self.block_snapshot.excerpt_header_height } } #[derive(Copy, Clone, Default, Eq, Ord, PartialOrd, PartialEq)] pub struct DisplayPoint(BlockPoint); impl Debug for DisplayPoint { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { f.write_fmt(format_args!( "DisplayPoint({}, {})", self.row().0, self.column() )) } } impl Add for DisplayPoint { type Output = Self; fn add(self, other: Self) -> Self::Output { DisplayPoint(BlockPoint(self.0 .0 + other.0 .0)) } } impl Sub for DisplayPoint { type Output = Self; fn sub(self, other: Self) -> Self::Output { DisplayPoint(BlockPoint(self.0 .0 - other.0 .0)) } } #[derive(Debug, Copy, Clone, Default, Eq, Ord, PartialOrd, PartialEq, Deserialize, Hash)] #[serde(transparent)] pub struct DisplayRow(pub u32); impl Add for DisplayRow { type Output = Self; fn add(self, other: Self) -> Self::Output { DisplayRow(self.0 + other.0) } } impl Add for DisplayRow { type Output = Self; fn add(self, other: u32) -> Self::Output { DisplayRow(self.0 + other) } } impl Sub for DisplayRow { type Output = Self; fn sub(self, other: Self) -> Self::Output { DisplayRow(self.0 - other.0) } } impl Sub for DisplayRow { type Output = Self; fn sub(self, other: u32) -> Self::Output { DisplayRow(self.0 - other) } } impl DisplayPoint { pub fn new(row: DisplayRow, column: u32) -> Self { Self(BlockPoint(Point::new(row.0, column))) } pub fn zero() -> Self { Self::new(DisplayRow(0), 0) } pub fn is_zero(&self) -> bool { self.0.is_zero() } pub fn row(self) -> DisplayRow { DisplayRow(self.0.row) } pub fn column(self) -> u32 { self.0.column } pub fn row_mut(&mut self) -> &mut u32 { &mut self.0.row } pub fn column_mut(&mut self) -> &mut u32 { &mut self.0.column } pub fn to_point(self, map: &DisplaySnapshot) -> Point { map.display_point_to_point(self, Bias::Left) } pub fn to_offset(self, map: &DisplaySnapshot, bias: Bias) -> usize { let wrap_point = map.block_snapshot.to_wrap_point(self.0, bias); let tab_point = map.wrap_snapshot.to_tab_point(wrap_point); let fold_point = map.tab_snapshot.to_fold_point(tab_point, bias).0; let inlay_point = fold_point.to_inlay_point(&map.fold_snapshot); map.inlay_snapshot .to_buffer_offset(map.inlay_snapshot.to_offset(inlay_point)) } } impl ToDisplayPoint for usize { fn to_display_point(&self, map: &DisplaySnapshot) -> DisplayPoint { map.point_to_display_point(self.to_point(&map.buffer_snapshot), Bias::Left) } } impl ToDisplayPoint for OffsetUtf16 { fn to_display_point(&self, map: &DisplaySnapshot) -> DisplayPoint { self.to_offset(&map.buffer_snapshot).to_display_point(map) } } impl ToDisplayPoint for Point { fn to_display_point(&self, map: &DisplaySnapshot) -> DisplayPoint { map.point_to_display_point(*self, Bias::Left) } } impl ToDisplayPoint for Anchor { fn to_display_point(&self, map: &DisplaySnapshot) -> DisplayPoint { self.to_point(&map.buffer_snapshot).to_display_point(map) } } #[cfg(test)] pub mod tests { use super::*; use crate::{movement, test::marked_display_snapshot}; use block_map::BlockPlacement; use gpui::{ div, font, observe, px, AppContext, BorrowAppContext, Context, Element, Hsla, Rgba, }; use language::{ language_settings::{AllLanguageSettings, AllLanguageSettingsContent}, Buffer, Diagnostic, DiagnosticEntry, DiagnosticSet, Language, LanguageConfig, LanguageMatcher, }; use lsp::LanguageServerId; use project::Project; use rand::{prelude::*, Rng}; use settings::SettingsStore; use smol::stream::StreamExt; use std::{env, sync::Arc}; use text::PointUtf16; use theme::{LoadThemes, SyntaxTheme}; use unindent::Unindent as _; use util::test::{marked_text_ranges, sample_text}; use Bias::*; #[gpui::test(iterations = 100)] async fn test_random_display_map(cx: &mut gpui::TestAppContext, mut rng: StdRng) { cx.background_executor.set_block_on_ticks(0..=50); let operations = env::var("OPERATIONS") .map(|i| i.parse().expect("invalid `OPERATIONS` variable")) .unwrap_or(10); let mut tab_size = rng.gen_range(1..=4); let buffer_start_excerpt_header_height = rng.gen_range(1..=5); let excerpt_header_height = rng.gen_range(1..=5); let font_size = px(14.0); let max_wrap_width = 300.0; let mut wrap_width = if rng.gen_bool(0.1) { None } else { Some(px(rng.gen_range(0.0..=max_wrap_width))) }; log::info!("tab size: {}", tab_size); log::info!("wrap width: {:?}", wrap_width); cx.update(|cx| { init_test(cx, |s| s.defaults.tab_size = NonZeroU32::new(tab_size)); }); let buffer = cx.update(|cx| { if rng.gen() { let len = rng.gen_range(0..10); let text = util::RandomCharIter::new(&mut rng) .take(len) .collect::(); MultiBuffer::build_simple(&text, cx) } else { MultiBuffer::build_random(&mut rng, cx) } }); let map = cx.new_model(|cx| { DisplayMap::new( buffer.clone(), font("Helvetica"), font_size, wrap_width, true, buffer_start_excerpt_header_height, excerpt_header_height, 0, FoldPlaceholder::test(), cx, ) }); let mut notifications = observe(&map, cx); let mut fold_count = 0; let mut blocks = Vec::new(); let snapshot = map.update(cx, |map, cx| map.snapshot(cx)); log::info!("buffer text: {:?}", snapshot.buffer_snapshot.text()); log::info!("fold text: {:?}", snapshot.fold_snapshot.text()); log::info!("tab text: {:?}", snapshot.tab_snapshot.text()); log::info!("wrap text: {:?}", snapshot.wrap_snapshot.text()); log::info!("block text: {:?}", snapshot.block_snapshot.text()); log::info!("display text: {:?}", snapshot.text()); for _i in 0..operations { match rng.gen_range(0..100) { 0..=19 => { wrap_width = if rng.gen_bool(0.2) { None } else { Some(px(rng.gen_range(0.0..=max_wrap_width))) }; log::info!("setting wrap width to {:?}", wrap_width); map.update(cx, |map, cx| map.set_wrap_width(wrap_width, cx)); } 20..=29 => { let mut tab_sizes = vec![1, 2, 3, 4]; tab_sizes.remove((tab_size - 1) as usize); tab_size = *tab_sizes.choose(&mut rng).unwrap(); log::info!("setting tab size to {:?}", tab_size); cx.update(|cx| { cx.update_global::(|store, cx| { store.update_user_settings::(cx, |s| { s.defaults.tab_size = NonZeroU32::new(tab_size); }); }); }); } 30..=44 => { map.update(cx, |map, cx| { if rng.gen() || blocks.is_empty() { let buffer = map.snapshot(cx).buffer_snapshot; let block_properties = (0..rng.gen_range(1..=1)) .map(|_| { let position = buffer.anchor_after(buffer.clip_offset( rng.gen_range(0..=buffer.len()), Bias::Left, )); let placement = if rng.gen() { BlockPlacement::Above(position) } else { BlockPlacement::Below(position) }; let height = rng.gen_range(1..5); log::info!( "inserting block {:?} with height {}", placement.as_ref().map(|p| p.to_point(&buffer)), height ); let priority = rng.gen_range(1..100); BlockProperties { placement, style: BlockStyle::Fixed, height, render: Box::new(|_| div().into_any()), priority, } }) .collect::>(); blocks.extend(map.insert_blocks(block_properties, cx)); } else { blocks.shuffle(&mut rng); let remove_count = rng.gen_range(1..=4.min(blocks.len())); let block_ids_to_remove = (0..remove_count) .map(|_| blocks.remove(rng.gen_range(0..blocks.len()))) .collect(); log::info!("removing block ids {:?}", block_ids_to_remove); map.remove_blocks(block_ids_to_remove, cx); } }); } 45..=79 => { let mut ranges = Vec::new(); for _ in 0..rng.gen_range(1..=3) { buffer.read_with(cx, |buffer, cx| { let buffer = buffer.read(cx); let end = buffer.clip_offset(rng.gen_range(0..=buffer.len()), Right); let start = buffer.clip_offset(rng.gen_range(0..=end), Left); ranges.push(start..end); }); } if rng.gen() && fold_count > 0 { log::info!("unfolding ranges: {:?}", ranges); map.update(cx, |map, cx| { map.unfold_intersecting(ranges, true, cx); }); } else { log::info!("folding ranges: {:?}", ranges); map.update(cx, |map, cx| { map.fold( ranges .into_iter() .map(|range| (range, FoldPlaceholder::test())), cx, ); }); } } _ => { buffer.update(cx, |buffer, cx| buffer.randomly_mutate(&mut rng, 5, cx)); } } if map.read_with(cx, |map, cx| map.is_rewrapping(cx)) { notifications.next().await.unwrap(); } let snapshot = map.update(cx, |map, cx| map.snapshot(cx)); fold_count = snapshot.fold_count(); log::info!("buffer text: {:?}", snapshot.buffer_snapshot.text()); log::info!("fold text: {:?}", snapshot.fold_snapshot.text()); log::info!("tab text: {:?}", snapshot.tab_snapshot.text()); log::info!("wrap text: {:?}", snapshot.wrap_snapshot.text()); log::info!("block text: {:?}", snapshot.block_snapshot.text()); log::info!("display text: {:?}", snapshot.text()); // Line boundaries let buffer = &snapshot.buffer_snapshot; for _ in 0..5 { let row = rng.gen_range(0..=buffer.max_point().row); let column = rng.gen_range(0..=buffer.line_len(MultiBufferRow(row))); let point = buffer.clip_point(Point::new(row, column), Left); let (prev_buffer_bound, prev_display_bound) = snapshot.prev_line_boundary(point); let (next_buffer_bound, next_display_bound) = snapshot.next_line_boundary(point); assert!(prev_buffer_bound <= point); assert!(next_buffer_bound >= point); assert_eq!(prev_buffer_bound.column, 0); assert_eq!(prev_display_bound.column(), 0); if next_buffer_bound < buffer.max_point() { assert_eq!(buffer.chars_at(next_buffer_bound).next(), Some('\n')); } assert_eq!( prev_display_bound, prev_buffer_bound.to_display_point(&snapshot), "row boundary before {:?}. reported buffer row boundary: {:?}", point, prev_buffer_bound ); assert_eq!( next_display_bound, next_buffer_bound.to_display_point(&snapshot), "display row boundary after {:?}. reported buffer row boundary: {:?}", point, next_buffer_bound ); assert_eq!( prev_buffer_bound, prev_display_bound.to_point(&snapshot), "row boundary before {:?}. reported display row boundary: {:?}", point, prev_display_bound ); assert_eq!( next_buffer_bound, next_display_bound.to_point(&snapshot), "row boundary after {:?}. reported display row boundary: {:?}", point, next_display_bound ); } // Movement let min_point = snapshot.clip_point(DisplayPoint::new(DisplayRow(0), 0), Left); let max_point = snapshot.clip_point(snapshot.max_point(), Right); for _ in 0..5 { let row = rng.gen_range(0..=snapshot.max_point().row().0); let column = rng.gen_range(0..=snapshot.line_len(DisplayRow(row))); let point = snapshot.clip_point(DisplayPoint::new(DisplayRow(row), column), Left); log::info!("Moving from point {:?}", point); let moved_right = movement::right(&snapshot, point); log::info!("Right {:?}", moved_right); if point < max_point { assert!(moved_right > point); if point.column() == snapshot.line_len(point.row()) || snapshot.soft_wrap_indent(point.row()).is_some() && point.column() == snapshot.line_len(point.row()) - 1 { assert!(moved_right.row() > point.row()); } } else { assert_eq!(moved_right, point); } let moved_left = movement::left(&snapshot, point); log::info!("Left {:?}", moved_left); if point > min_point { assert!(moved_left < point); if point.column() == 0 { assert!(moved_left.row() < point.row()); } } else { assert_eq!(moved_left, point); } } } } #[cfg(target_os = "macos")] #[gpui::test(retries = 5)] async fn test_soft_wraps(cx: &mut gpui::TestAppContext) { cx.background_executor .set_block_on_ticks(usize::MAX..=usize::MAX); cx.update(|cx| { init_test(cx, |_| {}); }); let mut cx = crate::test::editor_test_context::EditorTestContext::new(cx).await; let editor = cx.editor.clone(); let window = cx.window; _ = cx.update_window(window, |_, cx| { let text_layout_details = editor.update(cx, |editor, cx| editor.text_layout_details(cx)); let font_size = px(12.0); let wrap_width = Some(px(64.)); let text = "one two three four five\nsix seven eight"; let buffer = MultiBuffer::build_simple(text, cx); let map = cx.new_model(|cx| { DisplayMap::new( buffer.clone(), font("Helvetica"), font_size, wrap_width, true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); let snapshot = map.update(cx, |map, cx| map.snapshot(cx)); assert_eq!( snapshot.text_chunks(DisplayRow(0)).collect::(), "one two \nthree four \nfive\nsix seven \neight" ); assert_eq!( snapshot.clip_point(DisplayPoint::new(DisplayRow(0), 8), Bias::Left), DisplayPoint::new(DisplayRow(0), 7) ); assert_eq!( snapshot.clip_point(DisplayPoint::new(DisplayRow(0), 8), Bias::Right), DisplayPoint::new(DisplayRow(1), 0) ); assert_eq!( movement::right(&snapshot, DisplayPoint::new(DisplayRow(0), 7)), DisplayPoint::new(DisplayRow(1), 0) ); assert_eq!( movement::left(&snapshot, DisplayPoint::new(DisplayRow(1), 0)), DisplayPoint::new(DisplayRow(0), 7) ); let x = snapshot .x_for_display_point(DisplayPoint::new(DisplayRow(1), 10), &text_layout_details); assert_eq!( movement::up( &snapshot, DisplayPoint::new(DisplayRow(1), 10), language::SelectionGoal::None, false, &text_layout_details, ), ( DisplayPoint::new(DisplayRow(0), 7), language::SelectionGoal::HorizontalPosition(x.0) ) ); assert_eq!( movement::down( &snapshot, DisplayPoint::new(DisplayRow(0), 7), language::SelectionGoal::HorizontalPosition(x.0), false, &text_layout_details ), ( DisplayPoint::new(DisplayRow(1), 10), language::SelectionGoal::HorizontalPosition(x.0) ) ); assert_eq!( movement::down( &snapshot, DisplayPoint::new(DisplayRow(1), 10), language::SelectionGoal::HorizontalPosition(x.0), false, &text_layout_details ), ( DisplayPoint::new(DisplayRow(2), 4), language::SelectionGoal::HorizontalPosition(x.0) ) ); let ix = snapshot.buffer_snapshot.text().find("seven").unwrap(); buffer.update(cx, |buffer, cx| { buffer.edit([(ix..ix, "and ")], None, cx); }); let snapshot = map.update(cx, |map, cx| map.snapshot(cx)); assert_eq!( snapshot.text_chunks(DisplayRow(1)).collect::(), "three four \nfive\nsix and \nseven eight" ); // Re-wrap on font size changes map.update(cx, |map, cx| { map.set_font(font("Helvetica"), px(font_size.0 + 3.), cx) }); let snapshot = map.update(cx, |map, cx| map.snapshot(cx)); assert_eq!( snapshot.text_chunks(DisplayRow(1)).collect::(), "three \nfour five\nsix and \nseven \neight" ) }); } #[gpui::test] fn test_text_chunks(cx: &mut gpui::AppContext) { init_test(cx, |_| {}); let text = sample_text(6, 6, 'a'); let buffer = MultiBuffer::build_simple(&text, cx); let font_size = px(14.0); let map = cx.new_model(|cx| { DisplayMap::new( buffer.clone(), font("Helvetica"), font_size, None, true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); buffer.update(cx, |buffer, cx| { buffer.edit( vec![ ( MultiBufferPoint::new(1, 0)..MultiBufferPoint::new(1, 0), "\t", ), ( MultiBufferPoint::new(1, 1)..MultiBufferPoint::new(1, 1), "\t", ), ( MultiBufferPoint::new(2, 1)..MultiBufferPoint::new(2, 1), "\t", ), ], None, cx, ) }); assert_eq!( map.update(cx, |map, cx| map.snapshot(cx)) .text_chunks(DisplayRow(1)) .collect::() .lines() .next(), Some(" b bbbbb") ); assert_eq!( map.update(cx, |map, cx| map.snapshot(cx)) .text_chunks(DisplayRow(2)) .collect::() .lines() .next(), Some("c ccccc") ); } #[gpui::test] async fn test_chunks(cx: &mut gpui::TestAppContext) { let text = r#" fn outer() {} mod module { fn inner() {} }"# .unindent(); let theme = SyntaxTheme::new_test(vec![("mod.body", Hsla::red()), ("fn.name", Hsla::blue())]); let language = Arc::new( Language::new( LanguageConfig { name: "Test".into(), matcher: LanguageMatcher { path_suffixes: vec![".test".to_string()], ..Default::default() }, ..Default::default() }, Some(tree_sitter_rust::LANGUAGE.into()), ) .with_highlights_query( r#" (mod_item name: (identifier) body: _ @mod.body) (function_item name: (identifier) @fn.name) "#, ) .unwrap(), ); language.set_theme(&theme); cx.update(|cx| init_test(cx, |s| s.defaults.tab_size = Some(2.try_into().unwrap()))); let buffer = cx.new_model(|cx| Buffer::local(text, cx).with_language(language, cx)); cx.condition(&buffer, |buf, _| !buf.is_parsing()).await; let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx)); let font_size = px(14.0); let map = cx.new_model(|cx| { DisplayMap::new( buffer, font("Helvetica"), font_size, None, true, 1, 1, 1, FoldPlaceholder::test(), cx, ) }); assert_eq!( cx.update(|cx| syntax_chunks(DisplayRow(0)..DisplayRow(5), &map, &theme, cx)), vec![ ("fn ".to_string(), None), ("outer".to_string(), Some(Hsla::blue())), ("() {}\n\nmod module ".to_string(), None), ("{\n fn ".to_string(), Some(Hsla::red())), ("inner".to_string(), Some(Hsla::blue())), ("() {}\n}".to_string(), Some(Hsla::red())), ] ); assert_eq!( cx.update(|cx| syntax_chunks(DisplayRow(3)..DisplayRow(5), &map, &theme, cx)), vec![ (" fn ".to_string(), Some(Hsla::red())), ("inner".to_string(), Some(Hsla::blue())), ("() {}\n}".to_string(), Some(Hsla::red())), ] ); map.update(cx, |map, cx| { map.fold( vec![( MultiBufferPoint::new(0, 6)..MultiBufferPoint::new(3, 2), FoldPlaceholder::test(), )], cx, ) }); assert_eq!( cx.update(|cx| syntax_chunks(DisplayRow(0)..DisplayRow(2), &map, &theme, cx)), vec![ ("fn ".to_string(), None), ("out".to_string(), Some(Hsla::blue())), ("⋯".to_string(), None), (" fn ".to_string(), Some(Hsla::red())), ("inner".to_string(), Some(Hsla::blue())), ("() {}\n}".to_string(), Some(Hsla::red())), ] ); } #[gpui::test] async fn test_chunks_with_syntax_highlighting_across_blocks(cx: &mut gpui::TestAppContext) { cx.background_executor .set_block_on_ticks(usize::MAX..=usize::MAX); let text = r#" const A: &str = " one two three "; const B: &str = "four"; "# .unindent(); let theme = SyntaxTheme::new_test(vec![ ("string", Hsla::red()), ("punctuation", Hsla::blue()), ("keyword", Hsla::green()), ]); let language = Arc::new( Language::new( LanguageConfig { name: "Rust".into(), ..Default::default() }, Some(tree_sitter_rust::LANGUAGE.into()), ) .with_highlights_query( r#" (string_literal) @string "const" @keyword [":" ";"] @punctuation "#, ) .unwrap(), ); language.set_theme(&theme); cx.update(|cx| init_test(cx, |_| {})); let buffer = cx.new_model(|cx| Buffer::local(text, cx).with_language(language, cx)); cx.condition(&buffer, |buf, _| !buf.is_parsing()).await; let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx)); let buffer_snapshot = buffer.read_with(cx, |buffer, cx| buffer.snapshot(cx)); let map = cx.new_model(|cx| { DisplayMap::new( buffer, font("Courier"), px(16.0), None, true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); // Insert a block in the middle of a multi-line string literal map.update(cx, |map, cx| { map.insert_blocks( [BlockProperties { placement: BlockPlacement::Below( buffer_snapshot.anchor_before(Point::new(1, 0)), ), height: 1, style: BlockStyle::Sticky, render: Box::new(|_| div().into_any()), priority: 0, }], cx, ) }); pretty_assertions::assert_eq!( cx.update(|cx| syntax_chunks(DisplayRow(0)..DisplayRow(7), &map, &theme, cx)), [ ("const".into(), Some(Hsla::green())), (" A".into(), None), (":".into(), Some(Hsla::blue())), (" &str = ".into(), None), ("\"\n one\n".into(), Some(Hsla::red())), ("\n".into(), None), (" two\n three\n\"".into(), Some(Hsla::red())), (";".into(), Some(Hsla::blue())), ("\n".into(), None), ("const".into(), Some(Hsla::green())), (" B".into(), None), (":".into(), Some(Hsla::blue())), (" &str = ".into(), None), ("\"four\"".into(), Some(Hsla::red())), (";".into(), Some(Hsla::blue())), ("\n".into(), None), ] ); } #[gpui::test] async fn test_chunks_with_diagnostics_across_blocks(cx: &mut gpui::TestAppContext) { cx.background_executor .set_block_on_ticks(usize::MAX..=usize::MAX); let text = r#" struct A { b: usize; } const c: usize = 1; "# .unindent(); cx.update(|cx| init_test(cx, |_| {})); let buffer = cx.new_model(|cx| Buffer::local(text, cx)); buffer.update(cx, |buffer, cx| { buffer.update_diagnostics( LanguageServerId(0), DiagnosticSet::new( [DiagnosticEntry { range: PointUtf16::new(0, 0)..PointUtf16::new(2, 1), diagnostic: Diagnostic { severity: DiagnosticSeverity::ERROR, group_id: 1, message: "hi".into(), ..Default::default() }, }], buffer, ), cx, ) }); let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx)); let buffer_snapshot = buffer.read_with(cx, |buffer, cx| buffer.snapshot(cx)); let map = cx.new_model(|cx| { DisplayMap::new( buffer, font("Courier"), px(16.0), None, true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); let black = gpui::black().to_rgb(); let red = gpui::red().to_rgb(); // Insert a block in the middle of a multi-line diagnostic. map.update(cx, |map, cx| { map.highlight_text( TypeId::of::(), vec![ buffer_snapshot.anchor_before(Point::new(3, 9)) ..buffer_snapshot.anchor_after(Point::new(3, 14)), buffer_snapshot.anchor_before(Point::new(3, 17)) ..buffer_snapshot.anchor_after(Point::new(3, 18)), ], red.into(), ); map.insert_blocks( [BlockProperties { placement: BlockPlacement::Below( buffer_snapshot.anchor_before(Point::new(1, 0)), ), height: 1, style: BlockStyle::Sticky, render: Box::new(|_| div().into_any()), priority: 0, }], cx, ) }); let snapshot = map.update(cx, |map, cx| map.snapshot(cx)); let mut chunks = Vec::<(String, Option, Rgba)>::new(); for chunk in snapshot.chunks(DisplayRow(0)..DisplayRow(5), true, Default::default()) { let color = chunk .highlight_style .and_then(|style| style.color) .map_or(black, |color| color.to_rgb()); if let Some((last_chunk, last_severity, last_color)) = chunks.last_mut() { if *last_severity == chunk.diagnostic_severity && *last_color == color { last_chunk.push_str(chunk.text); continue; } } chunks.push((chunk.text.to_string(), chunk.diagnostic_severity, color)); } assert_eq!( chunks, [ ( "struct A {\n b: usize;\n".into(), Some(DiagnosticSeverity::ERROR), black ), ("\n".into(), None, black), ("}".into(), Some(DiagnosticSeverity::ERROR), black), ("\nconst c: ".into(), None, black), ("usize".into(), None, red), (" = ".into(), None, black), ("1".into(), None, red), (";\n".into(), None, black), ] ); } #[gpui::test] async fn test_point_translation_with_replace_blocks(cx: &mut gpui::TestAppContext) { cx.background_executor .set_block_on_ticks(usize::MAX..=usize::MAX); cx.update(|cx| init_test(cx, |_| {})); let buffer = cx.update(|cx| MultiBuffer::build_simple("abcde\nfghij\nklmno\npqrst", cx)); let buffer_snapshot = buffer.read_with(cx, |buffer, cx| buffer.snapshot(cx)); let map = cx.new_model(|cx| { DisplayMap::new( buffer.clone(), font("Courier"), px(16.0), None, true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); let snapshot = map.update(cx, |map, cx| { map.insert_blocks( [BlockProperties { placement: BlockPlacement::Replace( buffer_snapshot.anchor_before(Point::new(1, 2)) ..buffer_snapshot.anchor_after(Point::new(2, 3)), ), height: 4, style: BlockStyle::Fixed, render: Box::new(|_| div().into_any()), priority: 0, }], cx, ); map.snapshot(cx) }); assert_eq!(snapshot.text(), "abcde\n\n\n\n\npqrst"); let point_to_display_points = [ (Point::new(1, 0), DisplayPoint::new(DisplayRow(1), 0)), (Point::new(2, 0), DisplayPoint::new(DisplayRow(1), 0)), (Point::new(3, 0), DisplayPoint::new(DisplayRow(5), 0)), ]; for (buffer_point, display_point) in point_to_display_points { assert_eq!( snapshot.point_to_display_point(buffer_point, Bias::Left), display_point, "point_to_display_point({:?}, Bias::Left)", buffer_point ); assert_eq!( snapshot.point_to_display_point(buffer_point, Bias::Right), display_point, "point_to_display_point({:?}, Bias::Right)", buffer_point ); } let display_points_to_points = [ ( DisplayPoint::new(DisplayRow(1), 0), Point::new(1, 0), Point::new(2, 5), ), ( DisplayPoint::new(DisplayRow(2), 0), Point::new(1, 0), Point::new(2, 5), ), ( DisplayPoint::new(DisplayRow(3), 0), Point::new(1, 0), Point::new(2, 5), ), ( DisplayPoint::new(DisplayRow(4), 0), Point::new(1, 0), Point::new(2, 5), ), ( DisplayPoint::new(DisplayRow(5), 0), Point::new(3, 0), Point::new(3, 0), ), ]; for (display_point, left_buffer_point, right_buffer_point) in display_points_to_points { assert_eq!( snapshot.display_point_to_point(display_point, Bias::Left), left_buffer_point, "display_point_to_point({:?}, Bias::Left)", display_point ); assert_eq!( snapshot.display_point_to_point(display_point, Bias::Right), right_buffer_point, "display_point_to_point({:?}, Bias::Right)", display_point ); } } // todo(linux) fails due to pixel differences in text rendering #[cfg(target_os = "macos")] #[gpui::test] async fn test_chunks_with_soft_wrapping(cx: &mut gpui::TestAppContext) { cx.background_executor .set_block_on_ticks(usize::MAX..=usize::MAX); let text = r#" fn outer() {} mod module { fn inner() {} }"# .unindent(); let theme = SyntaxTheme::new_test(vec![("mod.body", Hsla::red()), ("fn.name", Hsla::blue())]); let language = Arc::new( Language::new( LanguageConfig { name: "Test".into(), matcher: LanguageMatcher { path_suffixes: vec![".test".to_string()], ..Default::default() }, ..Default::default() }, Some(tree_sitter_rust::LANGUAGE.into()), ) .with_highlights_query( r#" (mod_item name: (identifier) body: _ @mod.body) (function_item name: (identifier) @fn.name) "#, ) .unwrap(), ); language.set_theme(&theme); cx.update(|cx| init_test(cx, |_| {})); let buffer = cx.new_model(|cx| Buffer::local(text, cx).with_language(language, cx)); cx.condition(&buffer, |buf, _| !buf.is_parsing()).await; let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx)); let font_size = px(16.0); let map = cx.new_model(|cx| { DisplayMap::new( buffer, font("Courier"), font_size, Some(px(40.0)), true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); assert_eq!( cx.update(|cx| syntax_chunks(DisplayRow(0)..DisplayRow(5), &map, &theme, cx)), [ ("fn \n".to_string(), None), ("oute\nr".to_string(), Some(Hsla::blue())), ("() \n{}\n\n".to_string(), None), ] ); assert_eq!( cx.update(|cx| syntax_chunks(DisplayRow(3)..DisplayRow(5), &map, &theme, cx)), [("{}\n\n".to_string(), None)] ); map.update(cx, |map, cx| { map.fold( vec![( MultiBufferPoint::new(0, 6)..MultiBufferPoint::new(3, 2), FoldPlaceholder::test(), )], cx, ) }); assert_eq!( cx.update(|cx| syntax_chunks(DisplayRow(1)..DisplayRow(4), &map, &theme, cx)), [ ("out".to_string(), Some(Hsla::blue())), ("⋯\n".to_string(), None), (" \nfn ".to_string(), Some(Hsla::red())), ("i\n".to_string(), Some(Hsla::blue())) ] ); } #[gpui::test] async fn test_chunks_with_text_highlights(cx: &mut gpui::TestAppContext) { cx.update(|cx| init_test(cx, |_| {})); let theme = SyntaxTheme::new_test(vec![("operator", Hsla::red()), ("string", Hsla::green())]); let language = Arc::new( Language::new( LanguageConfig { name: "Test".into(), matcher: LanguageMatcher { path_suffixes: vec![".test".to_string()], ..Default::default() }, ..Default::default() }, Some(tree_sitter_rust::LANGUAGE.into()), ) .with_highlights_query( r#" ":" @operator (string_literal) @string "#, ) .unwrap(), ); language.set_theme(&theme); let (text, highlighted_ranges) = marked_text_ranges(r#"constˇ «a»: B = "c «d»""#, false); let buffer = cx.new_model(|cx| Buffer::local(text, cx).with_language(language, cx)); cx.condition(&buffer, |buf, _| !buf.is_parsing()).await; let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx)); let buffer_snapshot = buffer.read_with(cx, |buffer, cx| buffer.snapshot(cx)); let font_size = px(16.0); let map = cx.new_model(|cx| { DisplayMap::new( buffer, font("Courier"), font_size, None, true, 1, 1, 1, FoldPlaceholder::test(), cx, ) }); enum MyType {} let style = HighlightStyle { color: Some(Hsla::blue()), ..Default::default() }; map.update(cx, |map, _cx| { map.highlight_text( TypeId::of::(), highlighted_ranges .into_iter() .map(|range| { buffer_snapshot.anchor_before(range.start) ..buffer_snapshot.anchor_before(range.end) }) .collect(), style, ); }); assert_eq!( cx.update(|cx| chunks(DisplayRow(0)..DisplayRow(10), &map, &theme, cx)), [ ("const ".to_string(), None, None), ("a".to_string(), None, Some(Hsla::blue())), (":".to_string(), Some(Hsla::red()), None), (" B = ".to_string(), None, None), ("\"c ".to_string(), Some(Hsla::green()), None), ("d".to_string(), Some(Hsla::green()), Some(Hsla::blue())), ("\"".to_string(), Some(Hsla::green()), None), ] ); } #[gpui::test] fn test_clip_point(cx: &mut gpui::AppContext) { init_test(cx, |_| {}); fn assert(text: &str, shift_right: bool, bias: Bias, cx: &mut gpui::AppContext) { let (unmarked_snapshot, mut markers) = marked_display_snapshot(text, cx); match bias { Bias::Left => { if shift_right { *markers[1].column_mut() += 1; } assert_eq!(unmarked_snapshot.clip_point(markers[1], bias), markers[0]) } Bias::Right => { if shift_right { *markers[0].column_mut() += 1; } assert_eq!(unmarked_snapshot.clip_point(markers[0], bias), markers[1]) } }; } use Bias::{Left, Right}; assert("ˇˇα", false, Left, cx); assert("ˇˇα", true, Left, cx); assert("ˇˇα", false, Right, cx); assert("ˇαˇ", true, Right, cx); assert("ˇˇ✋", false, Left, cx); assert("ˇˇ✋", true, Left, cx); assert("ˇˇ✋", false, Right, cx); assert("ˇ✋ˇ", true, Right, cx); assert("ˇˇ🍐", false, Left, cx); assert("ˇˇ🍐", true, Left, cx); assert("ˇˇ🍐", false, Right, cx); assert("ˇ🍐ˇ", true, Right, cx); assert("ˇˇ\t", false, Left, cx); assert("ˇˇ\t", true, Left, cx); assert("ˇˇ\t", false, Right, cx); assert("ˇ\tˇ", true, Right, cx); assert(" ˇˇ\t", false, Left, cx); assert(" ˇˇ\t", true, Left, cx); assert(" ˇˇ\t", false, Right, cx); assert(" ˇ\tˇ", true, Right, cx); assert(" ˇˇ\t", false, Left, cx); assert(" ˇˇ\t", false, Right, cx); } #[gpui::test] fn test_clip_at_line_ends(cx: &mut gpui::AppContext) { init_test(cx, |_| {}); fn assert(text: &str, cx: &mut gpui::AppContext) { let (mut unmarked_snapshot, markers) = marked_display_snapshot(text, cx); unmarked_snapshot.clip_at_line_ends = true; assert_eq!( unmarked_snapshot.clip_point(markers[1], Bias::Left), markers[0] ); } assert("ˇˇ", cx); assert("ˇaˇ", cx); assert("aˇbˇ", cx); assert("aˇαˇ", cx); } #[gpui::test] fn test_creases(cx: &mut gpui::AppContext) { init_test(cx, |_| {}); let text = "aaa\nbbb\nccc\nddd\neee\nfff\nggg\nhhh\niii\njjj\nkkk\nlll"; let buffer = MultiBuffer::build_simple(text, cx); let font_size = px(14.0); cx.new_model(|cx| { let mut map = DisplayMap::new( buffer.clone(), font("Helvetica"), font_size, None, true, 1, 1, 0, FoldPlaceholder::test(), cx, ); let snapshot = map.buffer.read(cx).snapshot(cx); let range = snapshot.anchor_before(Point::new(2, 0))..snapshot.anchor_after(Point::new(3, 3)); map.crease_map.insert( [Crease::new( range, FoldPlaceholder::test(), |_row, _status, _toggle, _cx| div(), |_row, _status, _cx| div(), )], &map.buffer.read(cx).snapshot(cx), ); map }); } #[gpui::test] fn test_tabs_with_multibyte_chars(cx: &mut gpui::AppContext) { init_test(cx, |_| {}); let text = "✅\t\tα\nβ\t\n🏀β\t\tγ"; let buffer = MultiBuffer::build_simple(text, cx); let font_size = px(14.0); let map = cx.new_model(|cx| { DisplayMap::new( buffer.clone(), font("Helvetica"), font_size, None, true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); let map = map.update(cx, |map, cx| map.snapshot(cx)); assert_eq!(map.text(), "✅ α\nβ \n🏀β γ"); assert_eq!( map.text_chunks(DisplayRow(0)).collect::(), "✅ α\nβ \n🏀β γ" ); assert_eq!( map.text_chunks(DisplayRow(1)).collect::(), "β \n🏀β γ" ); assert_eq!( map.text_chunks(DisplayRow(2)).collect::(), "🏀β γ" ); let point = MultiBufferPoint::new(0, "✅\t\t".len() as u32); let display_point = DisplayPoint::new(DisplayRow(0), "✅ ".len() as u32); assert_eq!(point.to_display_point(&map), display_point); assert_eq!(display_point.to_point(&map), point); let point = MultiBufferPoint::new(1, "β\t".len() as u32); let display_point = DisplayPoint::new(DisplayRow(1), "β ".len() as u32); assert_eq!(point.to_display_point(&map), display_point); assert_eq!(display_point.to_point(&map), point,); let point = MultiBufferPoint::new(2, "🏀β\t\t".len() as u32); let display_point = DisplayPoint::new(DisplayRow(2), "🏀β ".len() as u32); assert_eq!(point.to_display_point(&map), display_point); assert_eq!(display_point.to_point(&map), point,); // Display points inside of expanded tabs assert_eq!( DisplayPoint::new(DisplayRow(0), "✅ ".len() as u32).to_point(&map), MultiBufferPoint::new(0, "✅\t".len() as u32), ); assert_eq!( DisplayPoint::new(DisplayRow(0), "✅ ".len() as u32).to_point(&map), MultiBufferPoint::new(0, "✅".len() as u32), ); // Clipping display points inside of multi-byte characters assert_eq!( map.clip_point( DisplayPoint::new(DisplayRow(0), "✅".len() as u32 - 1), Left ), DisplayPoint::new(DisplayRow(0), 0) ); assert_eq!( map.clip_point( DisplayPoint::new(DisplayRow(0), "✅".len() as u32 - 1), Bias::Right ), DisplayPoint::new(DisplayRow(0), "✅".len() as u32) ); } #[gpui::test] fn test_max_point(cx: &mut gpui::AppContext) { init_test(cx, |_| {}); let buffer = MultiBuffer::build_simple("aaa\n\t\tbbb", cx); let font_size = px(14.0); let map = cx.new_model(|cx| { DisplayMap::new( buffer.clone(), font("Helvetica"), font_size, None, true, 1, 1, 0, FoldPlaceholder::test(), cx, ) }); assert_eq!( map.update(cx, |map, cx| map.snapshot(cx)).max_point(), DisplayPoint::new(DisplayRow(1), 11) ) } fn syntax_chunks( rows: Range, map: &Model, theme: &SyntaxTheme, cx: &mut AppContext, ) -> Vec<(String, Option)> { chunks(rows, map, theme, cx) .into_iter() .map(|(text, color, _)| (text, color)) .collect() } fn chunks( rows: Range, map: &Model, theme: &SyntaxTheme, cx: &mut AppContext, ) -> Vec<(String, Option, Option)> { let snapshot = map.update(cx, |map, cx| map.snapshot(cx)); let mut chunks: Vec<(String, Option, Option)> = Vec::new(); for chunk in snapshot.chunks(rows, true, HighlightStyles::default()) { let syntax_color = chunk .syntax_highlight_id .and_then(|id| id.style(theme)?.color); let highlight_color = chunk.highlight_style.and_then(|style| style.color); if let Some((last_chunk, last_syntax_color, last_highlight_color)) = chunks.last_mut() { if syntax_color == *last_syntax_color && highlight_color == *last_highlight_color { last_chunk.push_str(chunk.text); continue; } } chunks.push((chunk.text.to_string(), syntax_color, highlight_color)); } chunks } fn init_test(cx: &mut AppContext, f: impl Fn(&mut AllLanguageSettingsContent)) { let settings = SettingsStore::test(cx); cx.set_global(settings); language::init(cx); crate::init(cx); Project::init_settings(cx); theme::init(LoadThemes::JustBase, cx); cx.update_global::(|store, cx| { store.update_user_settings::(cx, f); }); } }