use std::{ ops::{Bound, Range, RangeInclusive}, sync::Arc, }; use buffer_diff::{BufferDiff, BufferDiffSnapshot}; use collections::HashMap; use gpui::{Action, AppContext as _, Entity, EventEmitter, Focusable, Subscription, WeakEntity}; use itertools::Itertools; use language::{Buffer, Capability}; use multi_buffer::{ Anchor, BufferOffset, ExcerptId, ExcerptRange, ExpandExcerptDirection, MultiBuffer, MultiBufferDiffHunk, MultiBufferPoint, MultiBufferSnapshot, PathKey, }; use project::Project; use rope::Point; use settings::DiffViewStyle; use text::{Bias, BufferId, OffsetRangeExt as _, Patch, ToPoint as _}; use ui::{ App, Context, InteractiveElement as _, IntoElement as _, ParentElement as _, Render, Styled as _, Window, div, }; use crate::{ display_map::CompanionExcerptPatch, element::SplitSide, split_editor_view::{SplitEditorState, SplitEditorView}, }; use workspace::{ ActivatePaneLeft, ActivatePaneRight, Item, ToolbarItemLocation, Workspace, item::{BreadcrumbText, ItemBufferKind, ItemEvent, SaveOptions, TabContentParams}, searchable::{SearchEvent, SearchToken, SearchableItem, SearchableItemHandle}, }; use crate::{ Autoscroll, DisplayMap, Editor, EditorEvent, RenderDiffHunkControlsFn, ToggleSoftWrap, actions::{DisableBreakpoint, EditLogBreakpoint, EnableBreakpoint, ToggleBreakpoint}, display_map::Companion, }; use zed_actions::assistant::InlineAssist; pub(crate) fn convert_lhs_rows_to_rhs( lhs_excerpt_to_rhs_excerpt: &HashMap, rhs_snapshot: &MultiBufferSnapshot, lhs_snapshot: &MultiBufferSnapshot, lhs_bounds: (Bound, Bound), ) -> Vec { patches_for_range( lhs_excerpt_to_rhs_excerpt, lhs_snapshot, rhs_snapshot, lhs_bounds, |diff, range, buffer| diff.patch_for_base_text_range(range, buffer), ) } pub(crate) fn convert_rhs_rows_to_lhs( rhs_excerpt_to_lhs_excerpt: &HashMap, lhs_snapshot: &MultiBufferSnapshot, rhs_snapshot: &MultiBufferSnapshot, rhs_bounds: (Bound, Bound), ) -> Vec { patches_for_range( rhs_excerpt_to_lhs_excerpt, rhs_snapshot, lhs_snapshot, rhs_bounds, |diff, range, buffer| diff.patch_for_buffer_range(range, buffer), ) } fn translate_lhs_selections_to_rhs( selections_by_buffer: &HashMap>, Option)>, splittable: &SplittableEditor, cx: &App, ) -> HashMap, (Vec>, Option)> { let rhs_display_map = splittable.rhs_editor.read(cx).display_map.read(cx); let Some(companion) = rhs_display_map.companion() else { return HashMap::default(); }; let companion = companion.read(cx); let mut translated: HashMap, (Vec>, Option)> = HashMap::default(); for (lhs_buffer_id, (ranges, scroll_offset)) in selections_by_buffer { let Some(rhs_buffer_id) = companion.lhs_to_rhs_buffer(*lhs_buffer_id) else { continue; }; let Some(rhs_buffer) = splittable .rhs_editor .read(cx) .buffer() .read(cx) .buffer(rhs_buffer_id) else { continue; }; let Some(diff) = splittable .rhs_editor .read(cx) .buffer() .read(cx) .diff_for(rhs_buffer_id) else { continue; }; let diff_snapshot = diff.read(cx).snapshot(cx); let rhs_buffer_snapshot = rhs_buffer.read(cx).snapshot(); let base_text_buffer = diff.read(cx).base_text_buffer(); let base_text_snapshot = base_text_buffer.read(cx).snapshot(); let translated_ranges: Vec> = ranges .iter() .map(|range| { let start_point = base_text_snapshot.offset_to_point(range.start.0); let end_point = base_text_snapshot.offset_to_point(range.end.0); let rhs_start = diff_snapshot .base_text_point_to_buffer_point(start_point, &rhs_buffer_snapshot); let rhs_end = diff_snapshot.base_text_point_to_buffer_point(end_point, &rhs_buffer_snapshot); BufferOffset(rhs_buffer_snapshot.point_to_offset(rhs_start)) ..BufferOffset(rhs_buffer_snapshot.point_to_offset(rhs_end)) }) .collect(); translated.insert(rhs_buffer, (translated_ranges, *scroll_offset)); } translated } fn translate_lhs_hunks_to_rhs( lhs_hunks: &[MultiBufferDiffHunk], splittable: &SplittableEditor, cx: &App, ) -> Vec { let rhs_display_map = splittable.rhs_editor.read(cx).display_map.read(cx); let Some(companion) = rhs_display_map.companion() else { return vec![]; }; let companion = companion.read(cx); let rhs_snapshot = splittable.rhs_multibuffer.read(cx).snapshot(cx); let rhs_hunks: Vec = rhs_snapshot.diff_hunks().collect(); let mut translated = Vec::new(); for lhs_hunk in lhs_hunks { let Some(rhs_buffer_id) = companion.lhs_to_rhs_buffer(lhs_hunk.buffer_id) else { continue; }; if let Some(rhs_hunk) = rhs_hunks.iter().find(|rhs_hunk| { rhs_hunk.buffer_id == rhs_buffer_id && rhs_hunk.diff_base_byte_range == lhs_hunk.diff_base_byte_range }) { translated.push(rhs_hunk.clone()); } } translated } fn patches_for_range( excerpt_map: &HashMap, source_snapshot: &MultiBufferSnapshot, target_snapshot: &MultiBufferSnapshot, source_bounds: (Bound, Bound), translate_fn: F, ) -> Vec where F: Fn(&BufferDiffSnapshot, RangeInclusive, &text::BufferSnapshot) -> Patch, { struct PendingExcerpt<'a> { source_excerpt_id: ExcerptId, target_excerpt_id: ExcerptId, source_buffer: &'a text::BufferSnapshot, target_buffer: &'a text::BufferSnapshot, buffer_point_range: Range, source_context_range: Range, } let mut result = Vec::new(); let mut current_buffer_id: Option = None; let mut pending_excerpts: Vec = Vec::new(); let mut union_context_start: Option = None; let mut union_context_end: Option = None; let flush_buffer = |pending: &mut Vec, union_start: Point, union_end: Point, result: &mut Vec| { let Some(first) = pending.first() else { return; }; let diff = source_snapshot .diff_for_buffer_id(first.source_buffer.remote_id()) .unwrap(); let rhs_buffer = if first.source_buffer.remote_id() == diff.base_text().remote_id() { first.target_buffer } else { first.source_buffer }; let patch = translate_fn(diff, union_start..=union_end, rhs_buffer); for excerpt in pending.drain(..) { result.push(patch_for_excerpt( source_snapshot, target_snapshot, excerpt.source_excerpt_id, excerpt.target_excerpt_id, excerpt.target_buffer, excerpt.source_context_range, &patch, excerpt.buffer_point_range, )); } }; for (source_buffer, buffer_offset_range, source_excerpt_id, source_context_range) in source_snapshot.range_to_buffer_ranges_with_context(source_bounds) { let target_excerpt_id = excerpt_map.get(&source_excerpt_id).copied().unwrap(); let target_buffer = target_snapshot .buffer_for_excerpt(target_excerpt_id) .unwrap(); let buffer_id = source_buffer.remote_id(); if current_buffer_id != Some(buffer_id) { if let (Some(start), Some(end)) = (union_context_start.take(), union_context_end.take()) { flush_buffer(&mut pending_excerpts, start, end, &mut result); } current_buffer_id = Some(buffer_id); } let buffer_point_range = buffer_offset_range.to_point(source_buffer); let source_context_range = source_context_range.to_point(source_buffer); union_context_start = Some(union_context_start.map_or(source_context_range.start, |s| { s.min(source_context_range.start) })); union_context_end = Some(union_context_end.map_or(source_context_range.end, |e| { e.max(source_context_range.end) })); pending_excerpts.push(PendingExcerpt { source_excerpt_id, target_excerpt_id, source_buffer, target_buffer, buffer_point_range, source_context_range, }); } if let (Some(start), Some(end)) = (union_context_start, union_context_end) { flush_buffer(&mut pending_excerpts, start, end, &mut result); } result } fn patch_for_excerpt( source_snapshot: &MultiBufferSnapshot, target_snapshot: &MultiBufferSnapshot, source_excerpt_id: ExcerptId, target_excerpt_id: ExcerptId, target_buffer: &text::BufferSnapshot, source_context_range: Range, patch: &Patch, source_edited_range: Range, ) -> CompanionExcerptPatch { let source_multibuffer_range = source_snapshot .range_for_excerpt(source_excerpt_id) .unwrap(); let source_excerpt_start_in_multibuffer = source_multibuffer_range.start; let source_excerpt_start_in_buffer = source_context_range.start; let source_excerpt_end_in_buffer = source_context_range.end; let target_multibuffer_range = target_snapshot .range_for_excerpt(target_excerpt_id) .unwrap(); let target_excerpt_start_in_multibuffer = target_multibuffer_range.start; let target_context_range = target_snapshot .context_range_for_excerpt(target_excerpt_id) .unwrap(); let target_excerpt_start_in_buffer = target_context_range.start.to_point(&target_buffer); let target_excerpt_end_in_buffer = target_context_range.end.to_point(&target_buffer); let edits = patch .edits() .iter() .skip_while(|edit| edit.old.end < source_excerpt_start_in_buffer) .take_while(|edit| edit.old.start <= source_excerpt_end_in_buffer) .map(|edit| { let clamped_source_start = edit.old.start.max(source_excerpt_start_in_buffer); let clamped_source_end = edit.old.end.min(source_excerpt_end_in_buffer); let source_multibuffer_start = source_excerpt_start_in_multibuffer + (clamped_source_start - source_excerpt_start_in_buffer); let source_multibuffer_end = source_excerpt_start_in_multibuffer + (clamped_source_end - source_excerpt_start_in_buffer); let clamped_target_start = edit .new .start .max(target_excerpt_start_in_buffer) .min(target_excerpt_end_in_buffer); let clamped_target_end = edit .new .end .max(target_excerpt_start_in_buffer) .min(target_excerpt_end_in_buffer); let target_multibuffer_start = target_excerpt_start_in_multibuffer + (clamped_target_start - target_excerpt_start_in_buffer); let target_multibuffer_end = target_excerpt_start_in_multibuffer + (clamped_target_end - target_excerpt_start_in_buffer); text::Edit { old: source_multibuffer_start..source_multibuffer_end, new: target_multibuffer_start..target_multibuffer_end, } }); let edits = [text::Edit { old: source_excerpt_start_in_multibuffer..source_excerpt_start_in_multibuffer, new: target_excerpt_start_in_multibuffer..target_excerpt_start_in_multibuffer, }] .into_iter() .chain(edits); let mut merged_edits: Vec> = Vec::new(); for edit in edits { if let Some(last) = merged_edits.last_mut() { if edit.new.start <= last.new.end { last.old.end = last.old.end.max(edit.old.end); last.new.end = last.new.end.max(edit.new.end); continue; } } merged_edits.push(edit); } let edited_range = source_excerpt_start_in_multibuffer + (source_edited_range.start - source_excerpt_start_in_buffer) ..source_excerpt_start_in_multibuffer + (source_edited_range.end - source_excerpt_start_in_buffer); let source_excerpt_end = source_excerpt_start_in_multibuffer + (source_excerpt_end_in_buffer - source_excerpt_start_in_buffer); let target_excerpt_end = target_excerpt_start_in_multibuffer + (target_excerpt_end_in_buffer - target_excerpt_start_in_buffer); CompanionExcerptPatch { patch: Patch::new(merged_edits), edited_range, source_excerpt_range: source_excerpt_start_in_multibuffer..source_excerpt_end, target_excerpt_range: target_excerpt_start_in_multibuffer..target_excerpt_end, } } #[derive(Clone, Copy, PartialEq, Eq, Action, Default)] #[action(namespace = editor)] pub struct ToggleSplitDiff; pub struct SplittableEditor { rhs_multibuffer: Entity, rhs_editor: Entity, lhs: Option, workspace: WeakEntity, split_state: Entity, searched_side: Option, _subscriptions: Vec, } struct LhsEditor { multibuffer: Entity, editor: Entity, was_last_focused: bool, _subscriptions: Vec, } impl SplittableEditor { pub fn rhs_editor(&self) -> &Entity { &self.rhs_editor } pub fn lhs_editor(&self) -> Option<&Entity> { self.lhs.as_ref().map(|s| &s.editor) } pub fn is_split(&self) -> bool { self.lhs.is_some() } pub fn set_render_diff_hunk_controls( &self, render_diff_hunk_controls: RenderDiffHunkControlsFn, cx: &mut Context, ) { self.rhs_editor.update(cx, |editor, cx| { editor.set_render_diff_hunk_controls(render_diff_hunk_controls.clone(), cx); }); if let Some(lhs) = &self.lhs { lhs.editor.update(cx, |editor, cx| { editor.set_render_diff_hunk_controls(render_diff_hunk_controls.clone(), cx); }); } } fn focused_side(&self) -> SplitSide { if let Some(lhs) = &self.lhs && lhs.was_last_focused { SplitSide::Left } else { SplitSide::Right } } pub fn focused_editor(&self) -> &Entity { if let Some(lhs) = &self.lhs && lhs.was_last_focused { &lhs.editor } else { &self.rhs_editor } } pub fn new( style: DiffViewStyle, rhs_multibuffer: Entity, project: Entity, workspace: Entity, window: &mut Window, cx: &mut Context, ) -> Self { let rhs_editor = cx.new(|cx| { let mut editor = Editor::for_multibuffer(rhs_multibuffer.clone(), Some(project.clone()), window, cx); editor.set_expand_all_diff_hunks(cx); editor }); // TODO(split-diff) we might want to tag editor events with whether they came from rhs/lhs let subscriptions = vec![ cx.subscribe( &rhs_editor, |this, _, event: &EditorEvent, cx| match event { EditorEvent::ExpandExcerptsRequested { excerpt_ids, lines, direction, } => { this.expand_excerpts(excerpt_ids.iter().copied(), *lines, *direction, cx); } _ => cx.emit(event.clone()), }, ), cx.subscribe(&rhs_editor, |this, _, event: &SearchEvent, cx| { if this.searched_side.is_none() || this.searched_side == Some(SplitSide::Right) { cx.emit(event.clone()); } }), ]; let this = cx.weak_entity(); window.defer(cx, { let workspace = workspace.downgrade(); let rhs_editor = rhs_editor.downgrade(); move |window, cx| { workspace .update(cx, |workspace, cx| { rhs_editor .update(cx, |editor, cx| { editor.added_to_workspace(workspace, window, cx); }) .ok(); }) .ok(); if style == DiffViewStyle::Split { this.update(cx, |this, cx| { this.split(window, cx); }) .ok(); } } }); let split_state = cx.new(|cx| SplitEditorState::new(cx)); Self { rhs_editor, rhs_multibuffer, lhs: None, workspace: workspace.downgrade(), split_state, searched_side: None, _subscriptions: subscriptions, } } pub fn split(&mut self, window: &mut Window, cx: &mut Context) { if self.lhs.is_some() { return; } let Some(workspace) = self.workspace.upgrade() else { return; }; let project = workspace.read(cx).project().clone(); let lhs_multibuffer = cx.new(|cx| { let mut multibuffer = MultiBuffer::new(Capability::ReadOnly); multibuffer.set_all_diff_hunks_expanded(cx); multibuffer }); let render_diff_hunk_controls = self.rhs_editor.read(cx).render_diff_hunk_controls.clone(); let lhs_editor = cx.new(|cx| { let mut editor = Editor::for_multibuffer(lhs_multibuffer.clone(), Some(project.clone()), window, cx); editor.set_number_deleted_lines(true, cx); editor.set_delegate_expand_excerpts(true); editor.set_delegate_stage_and_restore(true); editor.set_delegate_open_excerpts(true); editor.set_show_vertical_scrollbar(false, cx); editor.disable_lsp_data(); editor.disable_runnables(); editor.disable_diagnostics(cx); editor.set_minimap_visibility(crate::MinimapVisibility::Disabled, window, cx); editor }); lhs_editor.update(cx, |editor, cx| { editor.set_render_diff_hunk_controls(render_diff_hunk_controls, cx); }); let mut subscriptions = vec![cx.subscribe_in( &lhs_editor, window, |this, _, event: &EditorEvent, window, cx| match event { EditorEvent::ExpandExcerptsRequested { excerpt_ids, lines, direction, } => { if this.lhs.is_some() { let rhs_display_map = this.rhs_editor.read(cx).display_map.read(cx); let rhs_ids: Vec<_> = excerpt_ids .iter() .filter_map(|id| { rhs_display_map.companion_excerpt_to_my_excerpt(*id, cx) }) .collect(); this.expand_excerpts(rhs_ids.into_iter(), *lines, *direction, cx); } } EditorEvent::StageOrUnstageRequested { stage, hunks } => { if this.lhs.is_some() { let translated = translate_lhs_hunks_to_rhs(hunks, this, cx); if !translated.is_empty() { let stage = *stage; this.rhs_editor.update(cx, |editor, cx| { let chunk_by = translated.into_iter().chunk_by(|h| h.buffer_id); for (buffer_id, hunks) in &chunk_by { editor.do_stage_or_unstage(stage, buffer_id, hunks, cx); } }); } } } EditorEvent::RestoreRequested { hunks } => { if this.lhs.is_some() { let translated = translate_lhs_hunks_to_rhs(hunks, this, cx); if !translated.is_empty() { this.rhs_editor.update(cx, |editor, cx| { editor.restore_diff_hunks(translated, cx); }); } } } EditorEvent::OpenExcerptsRequested { selections_by_buffer, split, } => { if this.lhs.is_some() { let translated = translate_lhs_selections_to_rhs(selections_by_buffer, this, cx); if !translated.is_empty() { let workspace = this.workspace.clone(); let split = *split; Editor::open_buffers_in_workspace( workspace, translated, split, window, cx, ); } } } _ => cx.emit(event.clone()), }, )]; subscriptions.push( cx.subscribe(&lhs_editor, |this, _, event: &SearchEvent, cx| { if this.searched_side == Some(SplitSide::Left) { cx.emit(event.clone()); } }), ); let lhs_focus_handle = lhs_editor.read(cx).focus_handle(cx); subscriptions.push( cx.on_focus_in(&lhs_focus_handle, window, |this, _window, cx| { if let Some(lhs) = &mut this.lhs { if !lhs.was_last_focused { lhs.was_last_focused = true; cx.notify(); } } }), ); let rhs_focus_handle = self.rhs_editor.read(cx).focus_handle(cx); subscriptions.push( cx.on_focus_in(&rhs_focus_handle, window, |this, _window, cx| { if let Some(lhs) = &mut this.lhs { if lhs.was_last_focused { lhs.was_last_focused = false; cx.notify(); } } }), ); let lhs = LhsEditor { editor: lhs_editor, multibuffer: lhs_multibuffer, was_last_focused: false, _subscriptions: subscriptions, }; let rhs_display_map = self.rhs_editor.read(cx).display_map.clone(); let lhs_display_map = lhs.editor.read(cx).display_map.clone(); let rhs_display_map_id = rhs_display_map.entity_id(); self.rhs_editor.update(cx, |editor, cx| { editor.set_delegate_expand_excerpts(true); editor.buffer().update(cx, |rhs_multibuffer, cx| { rhs_multibuffer.set_show_deleted_hunks(false, cx); rhs_multibuffer.set_use_extended_diff_range(true, cx); }) }); let path_diffs: Vec<_> = { let rhs_multibuffer = self.rhs_multibuffer.read(cx); rhs_multibuffer .paths() .filter_map(|path| { let excerpt_id = rhs_multibuffer.excerpts_for_path(path).next()?; let snapshot = rhs_multibuffer.snapshot(cx); let buffer = snapshot.buffer_for_excerpt(excerpt_id)?; let diff = rhs_multibuffer.diff_for(buffer.remote_id())?; Some((path.clone(), diff)) }) .collect() }; let mut companion = Companion::new( rhs_display_map_id, convert_rhs_rows_to_lhs, convert_lhs_rows_to_rhs, ); // stream this for (path, diff) in path_diffs { self.rhs_multibuffer.update(cx, |rhs_multibuffer, cx| { lhs.multibuffer.update(cx, |lhs_multibuffer, lhs_cx| { for (lhs, rhs) in LhsEditor::update_path_excerpts_from_rhs( path.clone(), rhs_multibuffer, lhs_multibuffer, diff.clone(), lhs_cx, ) { companion.add_excerpt_mapping(lhs, rhs); } companion.add_buffer_mapping( diff.read(lhs_cx).base_text(lhs_cx).remote_id(), diff.read(lhs_cx).buffer_id, ); }) }); } let companion = cx.new(|_| companion); rhs_display_map.update(cx, |dm, cx| { dm.set_companion(Some((lhs_display_map, companion.clone())), cx); }); let shared_scroll_anchor = self .rhs_editor .read(cx) .scroll_manager .scroll_anchor_entity(); lhs.editor.update(cx, |editor, _cx| { editor .scroll_manager .set_shared_scroll_anchor(shared_scroll_anchor); }); let this = cx.entity().downgrade(); self.rhs_editor.update(cx, |editor, _cx| { let this = this.clone(); editor.set_on_local_selections_changed(Some(Box::new( move |cursor_position, window, cx| { let this = this.clone(); window.defer(cx, move |window, cx| { this.update(cx, |this, cx| { this.sync_cursor_to_other_side(true, cursor_position, window, cx); }) .ok(); }) }, ))); }); lhs.editor.update(cx, |editor, _cx| { let this = this.clone(); editor.set_on_local_selections_changed(Some(Box::new( move |cursor_position, window, cx| { let this = this.clone(); window.defer(cx, move |window, cx| { this.update(cx, |this, cx| { this.sync_cursor_to_other_side(false, cursor_position, window, cx); }) .ok(); }) }, ))); }); // Copy soft wrap state from rhs (source of truth) to lhs let rhs_soft_wrap_override = self.rhs_editor.read(cx).soft_wrap_mode_override; lhs.editor.update(cx, |editor, cx| { editor.soft_wrap_mode_override = rhs_soft_wrap_override; cx.notify(); }); self.lhs = Some(lhs); cx.notify(); } fn activate_pane_left( &mut self, _: &ActivatePaneLeft, window: &mut Window, cx: &mut Context, ) { if let Some(lhs) = &self.lhs { if !lhs.was_last_focused { lhs.editor.read(cx).focus_handle(cx).focus(window, cx); lhs.editor.update(cx, |editor, cx| { editor.request_autoscroll(Autoscroll::fit(), cx); }); } else { cx.propagate(); } } else { cx.propagate(); } } fn activate_pane_right( &mut self, _: &ActivatePaneRight, window: &mut Window, cx: &mut Context, ) { if let Some(lhs) = &self.lhs { if lhs.was_last_focused { self.rhs_editor.read(cx).focus_handle(cx).focus(window, cx); self.rhs_editor.update(cx, |editor, cx| { editor.request_autoscroll(Autoscroll::fit(), cx); }); } else { cx.propagate(); } } else { cx.propagate(); } } fn sync_cursor_to_other_side( &mut self, from_rhs: bool, source_point: Point, window: &mut Window, cx: &mut Context, ) { let Some(lhs) = &self.lhs else { return; }; let (source_editor, target_editor) = if from_rhs { (&self.rhs_editor, &lhs.editor) } else { (&lhs.editor, &self.rhs_editor) }; let source_snapshot = source_editor.update(cx, |editor, cx| editor.snapshot(window, cx)); let target_snapshot = target_editor.update(cx, |editor, cx| editor.snapshot(window, cx)); let display_point = source_snapshot .display_snapshot .point_to_display_point(source_point, Bias::Right); let display_point = target_snapshot.clip_point(display_point, Bias::Right); let target_point = target_snapshot.display_point_to_point(display_point, Bias::Right); target_editor.update(cx, |editor, cx| { editor.set_suppress_selection_callback(true); editor.change_selections(crate::SelectionEffects::no_scroll(), window, cx, |s| { s.select_ranges([target_point..target_point]); }); editor.set_suppress_selection_callback(false); }); } fn toggle_split(&mut self, _: &ToggleSplitDiff, window: &mut Window, cx: &mut Context) { if self.lhs.is_some() { self.unsplit(window, cx); } else { self.split(window, cx); } } fn intercept_toggle_breakpoint( &mut self, _: &ToggleBreakpoint, _window: &mut Window, cx: &mut Context, ) { // Only block breakpoint actions when the left (lhs) editor has focus if let Some(lhs) = &self.lhs { if lhs.was_last_focused { cx.stop_propagation(); } else { cx.propagate(); } } else { cx.propagate(); } } fn intercept_enable_breakpoint( &mut self, _: &EnableBreakpoint, _window: &mut Window, cx: &mut Context, ) { // Only block breakpoint actions when the left (lhs) editor has focus if let Some(lhs) = &self.lhs { if lhs.was_last_focused { cx.stop_propagation(); } else { cx.propagate(); } } else { cx.propagate(); } } fn intercept_disable_breakpoint( &mut self, _: &DisableBreakpoint, _window: &mut Window, cx: &mut Context, ) { // Only block breakpoint actions when the left (lhs) editor has focus if let Some(lhs) = &self.lhs { if lhs.was_last_focused { cx.stop_propagation(); } else { cx.propagate(); } } else { cx.propagate(); } } fn intercept_edit_log_breakpoint( &mut self, _: &EditLogBreakpoint, _window: &mut Window, cx: &mut Context, ) { // Only block breakpoint actions when the left (lhs) editor has focus if let Some(lhs) = &self.lhs { if lhs.was_last_focused { cx.stop_propagation(); } else { cx.propagate(); } } else { cx.propagate(); } } fn intercept_inline_assist( &mut self, _: &InlineAssist, _window: &mut Window, cx: &mut Context, ) { if self.lhs.is_some() { cx.stop_propagation(); } else { cx.propagate(); } } fn toggle_soft_wrap( &mut self, _: &ToggleSoftWrap, window: &mut Window, cx: &mut Context, ) { if let Some(lhs) = &self.lhs { cx.stop_propagation(); let is_lhs_focused = lhs.was_last_focused; let (focused_editor, other_editor) = if is_lhs_focused { (&lhs.editor, &self.rhs_editor) } else { (&self.rhs_editor, &lhs.editor) }; // Toggle the focused editor focused_editor.update(cx, |editor, cx| { editor.toggle_soft_wrap(&ToggleSoftWrap, window, cx); }); // Copy the soft wrap state from the focused editor to the other editor let soft_wrap_override = focused_editor.read(cx).soft_wrap_mode_override; other_editor.update(cx, |editor, cx| { editor.soft_wrap_mode_override = soft_wrap_override; cx.notify(); }); } else { cx.propagate(); } } fn unsplit(&mut self, _: &mut Window, cx: &mut Context) { let Some(lhs) = self.lhs.take() else { return; }; self.rhs_editor.update(cx, |rhs, cx| { let rhs_snapshot = rhs.display_map.update(cx, |dm, cx| dm.snapshot(cx)); let native_anchor = rhs.scroll_manager.native_anchor(&rhs_snapshot, cx); let rhs_display_map_id = rhs_snapshot.display_map_id; rhs.scroll_manager .scroll_anchor_entity() .update(cx, |shared, _| { shared.scroll_anchor = native_anchor; shared.display_map_id = Some(rhs_display_map_id); }); rhs.set_on_local_selections_changed(None); rhs.set_delegate_expand_excerpts(false); rhs.buffer().update(cx, |buffer, cx| { buffer.set_show_deleted_hunks(true, cx); buffer.set_use_extended_diff_range(false, cx); }); rhs.display_map.update(cx, |dm, cx| { dm.set_companion(None, cx); }); }); lhs.editor.update(cx, |editor, _cx| { editor.set_on_local_selections_changed(None); }); cx.notify(); } pub fn set_excerpts_for_path( &mut self, path: PathKey, buffer: Entity, ranges: impl IntoIterator> + Clone, context_line_count: u32, diff: Entity, cx: &mut Context, ) -> (Vec>, bool) { let rhs_display_map = self.rhs_editor.read(cx).display_map.clone(); let lhs = self.lhs.as_ref(); self.rhs_multibuffer.update(cx, |rhs_multibuffer, cx| { mutate_excerpts_for_paths( rhs_multibuffer, lhs, &rhs_display_map, vec![(path.clone(), diff.clone())], cx, |rhs_multibuffer, cx| { let (anchors, added_a_new_excerpt) = rhs_multibuffer.set_excerpts_for_path( path.clone(), buffer.clone(), ranges, context_line_count, cx, ); if !anchors.is_empty() && rhs_multibuffer .diff_for(buffer.read(cx).remote_id()) .is_none_or(|old_diff| old_diff.entity_id() != diff.entity_id()) { rhs_multibuffer.add_diff(diff.clone(), cx); } (anchors, added_a_new_excerpt) }, ) }) } fn expand_excerpts( &mut self, excerpt_ids: impl Iterator + Clone, lines: u32, direction: ExpandExcerptDirection, cx: &mut Context, ) { let rhs_display_map = self.rhs_editor.read(cx).display_map.clone(); let lhs = self.lhs.as_ref(); self.rhs_multibuffer.update(cx, |rhs_multibuffer, cx| { if lhs.is_some() { let snapshot = rhs_multibuffer.snapshot(cx); let paths_with_diffs: Vec<_> = excerpt_ids .clone() .filter_map(|excerpt_id| { let path = rhs_multibuffer.path_for_excerpt(excerpt_id)?; let buffer = snapshot.buffer_for_excerpt(excerpt_id)?; let diff = rhs_multibuffer.diff_for(buffer.remote_id())?; Some((path, diff)) }) .collect::>() .into_iter() .collect(); mutate_excerpts_for_paths( rhs_multibuffer, lhs, &rhs_display_map, paths_with_diffs, cx, |rhs_multibuffer, cx| { rhs_multibuffer.expand_excerpts(excerpt_ids.clone(), lines, direction, cx); }, ); } else { rhs_multibuffer.expand_excerpts(excerpt_ids, lines, direction, cx); } }); } pub fn remove_excerpts_for_path(&mut self, path: PathKey, cx: &mut Context) { let rhs_display_map = self.rhs_editor.read(cx).display_map.clone(); if let Some(lhs) = &self.lhs { self.rhs_multibuffer.update(cx, |rhs_multibuffer, cx| { let rhs_excerpt_ids: Vec = rhs_multibuffer.excerpts_for_path(&path).collect(); let lhs_excerpt_ids: Vec = lhs.multibuffer.read(cx).excerpts_for_path(&path).collect(); if let Some(companion) = rhs_display_map.read(cx).companion().cloned() { companion.update(cx, |c, _| { c.remove_excerpt_mappings(lhs_excerpt_ids, rhs_excerpt_ids); }); } rhs_multibuffer.remove_excerpts_for_path(path.clone(), cx); }); lhs.multibuffer.update(cx, |lhs_multibuffer, cx| { lhs_multibuffer.remove_excerpts_for_path(path, cx); }); } else { self.rhs_multibuffer.update(cx, |rhs_multibuffer, cx| { rhs_multibuffer.remove_excerpts_for_path(path.clone(), cx); }); } } fn search_token(&self) -> SearchToken { SearchToken::new(self.focused_side() as u64) } fn editor_for_token(&self, token: SearchToken) -> &Entity { if token.value() == SplitSide::Left as u64 { if let Some(lhs) = &self.lhs { return &lhs.editor; } } &self.rhs_editor } } #[cfg(test)] impl SplittableEditor { fn check_invariants(&self, quiesced: bool, cx: &mut App) { use multi_buffer::MultiBufferRow; use text::Bias; use crate::display_map::Block; use crate::display_map::DisplayRow; self.debug_print(cx); let lhs = self.lhs.as_ref().unwrap(); let rhs_excerpts = self.rhs_multibuffer.read(cx).excerpt_ids(); let lhs_excerpts = lhs.multibuffer.read(cx).excerpt_ids(); assert_eq!( lhs_excerpts.len(), rhs_excerpts.len(), "mismatch in excerpt count" ); if quiesced { let rhs_snapshot = lhs .editor .update(cx, |editor, cx| editor.display_snapshot(cx)); let lhs_snapshot = self .rhs_editor .update(cx, |editor, cx| editor.display_snapshot(cx)); let lhs_max_row = lhs_snapshot.max_point().row(); let rhs_max_row = rhs_snapshot.max_point().row(); assert_eq!(lhs_max_row, rhs_max_row, "mismatch in display row count"); let lhs_excerpt_block_rows = lhs_snapshot .blocks_in_range(DisplayRow(0)..lhs_max_row + 1) .filter(|(_, block)| { matches!( block, Block::BufferHeader { .. } | Block::ExcerptBoundary { .. } ) }) .map(|(row, _)| row) .collect::>(); let rhs_excerpt_block_rows = rhs_snapshot .blocks_in_range(DisplayRow(0)..rhs_max_row + 1) .filter(|(_, block)| { matches!( block, Block::BufferHeader { .. } | Block::ExcerptBoundary { .. } ) }) .map(|(row, _)| row) .collect::>(); assert_eq!(lhs_excerpt_block_rows, rhs_excerpt_block_rows); for (lhs_hunk, rhs_hunk) in lhs_snapshot.diff_hunks().zip(rhs_snapshot.diff_hunks()) { assert_eq!( lhs_hunk.diff_base_byte_range, rhs_hunk.diff_base_byte_range, "mismatch in hunks" ); assert_eq!( lhs_hunk.status, rhs_hunk.status, "mismatch in hunk statuses" ); let (lhs_point, rhs_point) = if lhs_hunk.row_range.is_empty() || rhs_hunk.row_range.is_empty() { ( Point::new(lhs_hunk.row_range.end.0, 0), Point::new(rhs_hunk.row_range.end.0, 0), ) } else { ( Point::new(lhs_hunk.row_range.start.0, 0), Point::new(rhs_hunk.row_range.start.0, 0), ) }; let lhs_point = lhs_snapshot.point_to_display_point(lhs_point, Bias::Left); let rhs_point = rhs_snapshot.point_to_display_point(rhs_point, Bias::Left); assert_eq!( lhs_point.row(), rhs_point.row(), "mismatch in hunk position" ); } // Filtering out empty lines is a bit of a hack, to work around a case where // the base text has a trailing newline but the current text doesn't, or vice versa. // In this case, we get the additional newline on one side, but that line is not // marked as added/deleted by rowinfos. self.check_sides_match(cx, |snapshot| { snapshot .buffer_snapshot() .text() .split("\n") .zip(snapshot.buffer_snapshot().row_infos(MultiBufferRow(0))) .filter(|(line, row_info)| !line.is_empty() && row_info.diff_status.is_none()) .map(|(line, _)| line.to_owned()) .collect::>() }); } } #[track_caller] fn check_sides_match( &self, cx: &mut App, mut extract: impl FnMut(&crate::DisplaySnapshot) -> T, ) { let lhs = self.lhs.as_ref().expect("requires split"); let rhs_snapshot = self.rhs_editor.update(cx, |editor, cx| { editor.display_map.update(cx, |map, cx| map.snapshot(cx)) }); let lhs_snapshot = lhs.editor.update(cx, |editor, cx| { editor.display_map.update(cx, |map, cx| map.snapshot(cx)) }); let rhs_t = extract(&rhs_snapshot); let lhs_t = extract(&lhs_snapshot); if rhs_t != lhs_t { self.debug_print(cx); pretty_assertions::assert_eq!(rhs_t, lhs_t); } } fn debug_print(&self, cx: &mut App) { use crate::DisplayRow; use crate::display_map::Block; use buffer_diff::DiffHunkStatusKind; assert!( self.lhs.is_some(), "debug_print is only useful when lhs editor exists" ); let lhs = self.lhs.as_ref().unwrap(); // Get terminal width, default to 80 if unavailable let terminal_width = std::env::var("COLUMNS") .ok() .and_then(|s| s.parse::().ok()) .unwrap_or(80); // Each side gets half the terminal width minus the separator let separator = " │ "; let side_width = (terminal_width - separator.len()) / 2; // Get display snapshots for both editors let lhs_snapshot = lhs.editor.update(cx, |editor, cx| { editor.display_map.update(cx, |map, cx| map.snapshot(cx)) }); let rhs_snapshot = self.rhs_editor.update(cx, |editor, cx| { editor.display_map.update(cx, |map, cx| map.snapshot(cx)) }); let lhs_max_row = lhs_snapshot.max_point().row().0; let rhs_max_row = rhs_snapshot.max_point().row().0; let max_row = lhs_max_row.max(rhs_max_row); // Build a map from display row -> block type string // Each row of a multi-row block gets an entry with the same block type // For spacers, the ID is included in brackets fn build_block_map( snapshot: &crate::DisplaySnapshot, max_row: u32, ) -> std::collections::HashMap { let mut block_map = std::collections::HashMap::new(); for (start_row, block) in snapshot.blocks_in_range(DisplayRow(0)..DisplayRow(max_row + 1)) { let (block_type, height) = match block { Block::Spacer { id, height, is_below: _, } => (format!("SPACER[{}]", id.0), *height), Block::ExcerptBoundary { height, .. } => { ("EXCERPT_BOUNDARY".to_string(), *height) } Block::BufferHeader { height, .. } => ("BUFFER_HEADER".to_string(), *height), Block::FoldedBuffer { height, .. } => ("FOLDED_BUFFER".to_string(), *height), Block::Custom(custom) => { ("CUSTOM_BLOCK".to_string(), custom.height.unwrap_or(1)) } }; for offset in 0..height { block_map.insert(start_row.0 + offset, block_type.clone()); } } block_map } let lhs_blocks = build_block_map(&lhs_snapshot, lhs_max_row); let rhs_blocks = build_block_map(&rhs_snapshot, rhs_max_row); fn display_width(s: &str) -> usize { unicode_width::UnicodeWidthStr::width(s) } fn truncate_line(line: &str, max_width: usize) -> String { let line_width = display_width(line); if line_width <= max_width { return line.to_string(); } if max_width < 9 { let mut result = String::new(); let mut width = 0; for c in line.chars() { let c_width = unicode_width::UnicodeWidthChar::width(c).unwrap_or(0); if width + c_width > max_width { break; } result.push(c); width += c_width; } return result; } let ellipsis = "..."; let target_prefix_width = 3; let target_suffix_width = 3; let mut prefix = String::new(); let mut prefix_width = 0; for c in line.chars() { let c_width = unicode_width::UnicodeWidthChar::width(c).unwrap_or(0); if prefix_width + c_width > target_prefix_width { break; } prefix.push(c); prefix_width += c_width; } let mut suffix_chars: Vec = Vec::new(); let mut suffix_width = 0; for c in line.chars().rev() { let c_width = unicode_width::UnicodeWidthChar::width(c).unwrap_or(0); if suffix_width + c_width > target_suffix_width { break; } suffix_chars.push(c); suffix_width += c_width; } suffix_chars.reverse(); let suffix: String = suffix_chars.into_iter().collect(); format!("{}{}{}", prefix, ellipsis, suffix) } fn pad_to_width(s: &str, target_width: usize) -> String { let current_width = display_width(s); if current_width >= target_width { s.to_string() } else { format!("{}{}", s, " ".repeat(target_width - current_width)) } } // Helper to format a single row for one side // Format: "ln# diff bytes(cumul) text" or block info // Line numbers come from buffer_row in RowInfo (1-indexed for display) fn format_row( row: u32, max_row: u32, snapshot: &crate::DisplaySnapshot, blocks: &std::collections::HashMap, row_infos: &[multi_buffer::RowInfo], cumulative_bytes: &[usize], side_width: usize, ) -> String { // Get row info if available let row_info = row_infos.get(row as usize); // Line number prefix (3 chars + space) // Use buffer_row from RowInfo, which is None for block rows let line_prefix = if row > max_row { " ".to_string() } else if let Some(buffer_row) = row_info.and_then(|info| info.buffer_row) { format!("{:>3} ", buffer_row + 1) // 1-indexed for display } else { " ".to_string() // block rows have no line number }; let content_width = side_width.saturating_sub(line_prefix.len()); if row > max_row { return format!("{}{}", line_prefix, " ".repeat(content_width)); } // Check if this row is a block row if let Some(block_type) = blocks.get(&row) { let block_str = format!("~~~[{}]~~~", block_type); let formatted = format!("{:^width$}", block_str, width = content_width); return format!( "{}{}", line_prefix, truncate_line(&formatted, content_width) ); } // Get line text let line_text = snapshot.line(DisplayRow(row)); let line_bytes = line_text.len(); // Diff status marker let diff_marker = match row_info.and_then(|info| info.diff_status.as_ref()) { Some(status) => match status.kind { DiffHunkStatusKind::Added => "+", DiffHunkStatusKind::Deleted => "-", DiffHunkStatusKind::Modified => "~", }, None => " ", }; // Cumulative bytes let cumulative = cumulative_bytes.get(row as usize).copied().unwrap_or(0); // Format: "diff bytes(cumul) text" - use 3 digits for bytes, 4 for cumulative let info_prefix = format!("{}{:>3}({:>4}) ", diff_marker, line_bytes, cumulative); let text_width = content_width.saturating_sub(info_prefix.len()); let truncated_text = truncate_line(&line_text, text_width); let text_part = pad_to_width(&truncated_text, text_width); format!("{}{}{}", line_prefix, info_prefix, text_part) } // Collect row infos for both sides let lhs_row_infos: Vec<_> = lhs_snapshot .row_infos(DisplayRow(0)) .take((lhs_max_row + 1) as usize) .collect(); let rhs_row_infos: Vec<_> = rhs_snapshot .row_infos(DisplayRow(0)) .take((rhs_max_row + 1) as usize) .collect(); // Calculate cumulative bytes for each side (only counting non-block rows) let mut lhs_cumulative = Vec::with_capacity((lhs_max_row + 1) as usize); let mut cumulative = 0usize; for row in 0..=lhs_max_row { if !lhs_blocks.contains_key(&row) { cumulative += lhs_snapshot.line(DisplayRow(row)).len() + 1; // +1 for newline } lhs_cumulative.push(cumulative); } let mut rhs_cumulative = Vec::with_capacity((rhs_max_row + 1) as usize); cumulative = 0; for row in 0..=rhs_max_row { if !rhs_blocks.contains_key(&row) { cumulative += rhs_snapshot.line(DisplayRow(row)).len() + 1; } rhs_cumulative.push(cumulative); } // Print header eprintln!(); eprintln!("{}", "═".repeat(terminal_width)); let header_left = format!("{:^width$}", "(LHS)", width = side_width); let header_right = format!("{:^width$}", "(RHS)", width = side_width); eprintln!("{}{}{}", header_left, separator, header_right); eprintln!( "{:^width$}{}{:^width$}", "ln# diff len(cum) text", separator, "ln# diff len(cum) text", width = side_width ); eprintln!("{}", "─".repeat(terminal_width)); // Print each row for row in 0..=max_row { let left = format_row( row, lhs_max_row, &lhs_snapshot, &lhs_blocks, &lhs_row_infos, &lhs_cumulative, side_width, ); let right = format_row( row, rhs_max_row, &rhs_snapshot, &rhs_blocks, &rhs_row_infos, &rhs_cumulative, side_width, ); eprintln!("{}{}{}", left, separator, right); } eprintln!("{}", "═".repeat(terminal_width)); eprintln!("Legend: + added, - deleted, ~ modified, ~~~ block/spacer row"); eprintln!(); } fn randomly_edit_excerpts( &mut self, rng: &mut impl rand::Rng, mutation_count: usize, cx: &mut Context, ) { use collections::HashSet; use rand::prelude::*; use std::env; use util::RandomCharIter; let max_buffers = env::var("MAX_BUFFERS") .map(|i| i.parse().expect("invalid `MAX_BUFFERS` variable")) .unwrap_or(4); for _ in 0..mutation_count { let paths = self .rhs_multibuffer .read(cx) .paths() .cloned() .collect::>(); let excerpt_ids = self.rhs_multibuffer.read(cx).excerpt_ids(); if rng.random_bool(0.2) && !excerpt_ids.is_empty() { let mut excerpts = HashSet::default(); for _ in 0..rng.random_range(0..excerpt_ids.len()) { excerpts.extend(excerpt_ids.choose(rng).copied()); } let line_count = rng.random_range(1..5); log::info!("Expanding excerpts {excerpts:?} by {line_count} lines"); self.expand_excerpts( excerpts.iter().cloned(), line_count, ExpandExcerptDirection::UpAndDown, cx, ); continue; } if excerpt_ids.is_empty() || (rng.random_bool(0.8) && paths.len() < max_buffers) { let len = rng.random_range(100..500); let text = RandomCharIter::new(&mut *rng).take(len).collect::(); let buffer = cx.new(|cx| Buffer::local(text, cx)); log::info!( "Creating new buffer {} with text: {:?}", buffer.read(cx).remote_id(), buffer.read(cx).text() ); let buffer_snapshot = buffer.read(cx).snapshot(); let diff = cx.new(|cx| BufferDiff::new_unchanged(&buffer_snapshot, cx)); // Create some initial diff hunks. buffer.update(cx, |buffer, cx| { buffer.randomly_edit(rng, 1, cx); }); let buffer_snapshot = buffer.read(cx).text_snapshot(); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); let path = PathKey::for_buffer(&buffer, cx); let ranges = diff.update(cx, |diff, cx| { diff.snapshot(cx) .hunks(&buffer_snapshot) .map(|hunk| hunk.buffer_range.to_point(&buffer_snapshot)) .collect::>() }); self.set_excerpts_for_path(path, buffer, ranges, 2, diff, cx); } else { log::info!("removing excerpts"); let remove_count = rng.random_range(1..=paths.len()); let paths_to_remove = paths .choose_multiple(rng, remove_count) .cloned() .collect::>(); for path in paths_to_remove { self.remove_excerpts_for_path(path.clone(), cx); } } } } } impl Item for SplittableEditor { type Event = EditorEvent; fn tab_content_text(&self, detail: usize, cx: &App) -> ui::SharedString { self.rhs_editor.read(cx).tab_content_text(detail, cx) } fn tab_tooltip_text(&self, cx: &App) -> Option { self.rhs_editor.read(cx).tab_tooltip_text(cx) } fn tab_icon(&self, window: &Window, cx: &App) -> Option { self.rhs_editor.read(cx).tab_icon(window, cx) } fn tab_content(&self, params: TabContentParams, window: &Window, cx: &App) -> gpui::AnyElement { self.rhs_editor.read(cx).tab_content(params, window, cx) } fn to_item_events(event: &EditorEvent, f: impl FnMut(ItemEvent)) { Editor::to_item_events(event, f) } fn for_each_project_item( &self, cx: &App, f: &mut dyn FnMut(gpui::EntityId, &dyn project::ProjectItem), ) { self.rhs_editor.read(cx).for_each_project_item(cx, f) } fn buffer_kind(&self, cx: &App) -> ItemBufferKind { self.rhs_editor.read(cx).buffer_kind(cx) } fn is_dirty(&self, cx: &App) -> bool { self.rhs_editor.read(cx).is_dirty(cx) } fn has_conflict(&self, cx: &App) -> bool { self.rhs_editor.read(cx).has_conflict(cx) } fn has_deleted_file(&self, cx: &App) -> bool { self.rhs_editor.read(cx).has_deleted_file(cx) } fn capability(&self, cx: &App) -> language::Capability { self.rhs_editor.read(cx).capability(cx) } fn can_save(&self, cx: &App) -> bool { self.rhs_editor.read(cx).can_save(cx) } fn can_save_as(&self, cx: &App) -> bool { self.rhs_editor.read(cx).can_save_as(cx) } fn save( &mut self, options: SaveOptions, project: Entity, window: &mut Window, cx: &mut Context, ) -> gpui::Task> { self.rhs_editor .update(cx, |editor, cx| editor.save(options, project, window, cx)) } fn save_as( &mut self, project: Entity, path: project::ProjectPath, window: &mut Window, cx: &mut Context, ) -> gpui::Task> { self.rhs_editor .update(cx, |editor, cx| editor.save_as(project, path, window, cx)) } fn reload( &mut self, project: Entity, window: &mut Window, cx: &mut Context, ) -> gpui::Task> { self.rhs_editor .update(cx, |editor, cx| editor.reload(project, window, cx)) } fn navigate( &mut self, data: Arc, window: &mut Window, cx: &mut Context, ) -> bool { self.focused_editor() .update(cx, |editor, cx| editor.navigate(data, window, cx)) } fn deactivated(&mut self, window: &mut Window, cx: &mut Context) { self.focused_editor().update(cx, |editor, cx| { editor.deactivated(window, cx); }); } fn added_to_workspace( &mut self, workspace: &mut Workspace, window: &mut Window, cx: &mut Context, ) { self.workspace = workspace.weak_handle(); self.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.added_to_workspace(workspace, window, cx); }); if let Some(lhs) = &self.lhs { lhs.editor.update(cx, |lhs_editor, cx| { lhs_editor.added_to_workspace(workspace, window, cx); }); } } fn as_searchable( &self, handle: &Entity, _: &App, ) -> Option> { Some(Box::new(handle.clone())) } fn breadcrumb_location(&self, cx: &App) -> ToolbarItemLocation { self.rhs_editor.read(cx).breadcrumb_location(cx) } fn breadcrumbs(&self, cx: &App) -> Option> { self.rhs_editor.read(cx).breadcrumbs(cx) } fn pixel_position_of_cursor(&self, cx: &App) -> Option> { self.focused_editor().read(cx).pixel_position_of_cursor(cx) } } impl SearchableItem for SplittableEditor { type Match = Range; fn clear_matches(&mut self, window: &mut Window, cx: &mut Context) { self.rhs_editor.update(cx, |editor, cx| { editor.clear_matches(window, cx); }); if let Some(lhs_editor) = self.lhs_editor() { lhs_editor.update(cx, |editor, cx| { editor.clear_matches(window, cx); }) } } fn update_matches( &mut self, matches: &[Self::Match], active_match_index: Option, token: SearchToken, window: &mut Window, cx: &mut Context, ) { self.editor_for_token(token).update(cx, |editor, cx| { editor.update_matches(matches, active_match_index, token, window, cx); }); } fn search_bar_visibility_changed( &mut self, visible: bool, window: &mut Window, cx: &mut Context, ) { if visible { let side = self.focused_side(); self.searched_side = Some(side); match side { SplitSide::Left => { self.rhs_editor.update(cx, |editor, cx| { editor.clear_matches(window, cx); }); } SplitSide::Right => { if let Some(lhs) = &self.lhs { lhs.editor.update(cx, |editor, cx| { editor.clear_matches(window, cx); }); } } } } else { self.searched_side = None; } } fn query_suggestion(&mut self, window: &mut Window, cx: &mut Context) -> String { self.focused_editor() .update(cx, |editor, cx| editor.query_suggestion(window, cx)) } fn activate_match( &mut self, index: usize, matches: &[Self::Match], token: SearchToken, window: &mut Window, cx: &mut Context, ) { self.editor_for_token(token).update(cx, |editor, cx| { editor.activate_match(index, matches, token, window, cx); }); } fn select_matches( &mut self, matches: &[Self::Match], token: SearchToken, window: &mut Window, cx: &mut Context, ) { self.editor_for_token(token).update(cx, |editor, cx| { editor.select_matches(matches, token, window, cx); }); } fn replace( &mut self, identifier: &Self::Match, query: &project::search::SearchQuery, token: SearchToken, window: &mut Window, cx: &mut Context, ) { self.editor_for_token(token).update(cx, |editor, cx| { editor.replace(identifier, query, token, window, cx); }); } fn find_matches( &mut self, query: Arc, window: &mut Window, cx: &mut Context, ) -> gpui::Task> { self.focused_editor() .update(cx, |editor, cx| editor.find_matches(query, window, cx)) } fn find_matches_with_token( &mut self, query: Arc, window: &mut Window, cx: &mut Context, ) -> gpui::Task<(Vec, SearchToken)> { let token = self.search_token(); let editor = self.focused_editor().downgrade(); cx.spawn_in(window, async move |_, cx| { let Some(matches) = editor .update_in(cx, |editor, window, cx| { editor.find_matches(query, window, cx) }) .ok() else { return (Vec::new(), token); }; (matches.await, token) }) } fn active_match_index( &mut self, direction: workspace::searchable::Direction, matches: &[Self::Match], token: SearchToken, window: &mut Window, cx: &mut Context, ) -> Option { self.editor_for_token(token).update(cx, |editor, cx| { editor.active_match_index(direction, matches, token, window, cx) }) } } impl EventEmitter for SplittableEditor {} impl EventEmitter for SplittableEditor {} impl Focusable for SplittableEditor { fn focus_handle(&self, cx: &App) -> gpui::FocusHandle { self.focused_editor().read(cx).focus_handle(cx) } } // impl Item for SplittableEditor { // type Event = EditorEvent; // fn tab_content_text(&self, detail: usize, cx: &App) -> ui::SharedString { // self.rhs_editor().tab_content_text(detail, cx) // } // fn as_searchable(&self, _this: &Entity, cx: &App) -> Option> { // Some(Box::new(self.last_selected_editor().clone())) // } // } impl Render for SplittableEditor { fn render( &mut self, _window: &mut ui::Window, cx: &mut ui::Context, ) -> impl ui::IntoElement { let inner = if self.lhs.is_some() { let style = self.rhs_editor.read(cx).create_style(cx); SplitEditorView::new(cx.entity(), style, self.split_state.clone()).into_any_element() } else { self.rhs_editor.clone().into_any_element() }; div() .id("splittable-editor") .on_action(cx.listener(Self::toggle_split)) .on_action(cx.listener(Self::activate_pane_left)) .on_action(cx.listener(Self::activate_pane_right)) .on_action(cx.listener(Self::intercept_toggle_breakpoint)) .on_action(cx.listener(Self::intercept_enable_breakpoint)) .on_action(cx.listener(Self::intercept_disable_breakpoint)) .on_action(cx.listener(Self::intercept_edit_log_breakpoint)) .on_action(cx.listener(Self::intercept_inline_assist)) .capture_action(cx.listener(Self::toggle_soft_wrap)) .size_full() .child(inner) } } fn mutate_excerpts_for_paths( rhs_multibuffer: &mut MultiBuffer, lhs: Option<&LhsEditor>, rhs_display_map: &Entity, paths_with_diffs: Vec<(PathKey, Entity)>, cx: &mut Context, mutate: impl FnOnce(&mut MultiBuffer, &mut Context) -> R, ) -> R { let old_rhs_ids: Vec<_> = paths_with_diffs .iter() .map(|(path, _)| { rhs_multibuffer .excerpts_for_path(path) .collect::>() }) .collect(); let result = mutate(rhs_multibuffer, cx); if let Some(lhs) = lhs { for ((path, diff), old_rhs_ids) in paths_with_diffs.into_iter().zip(old_rhs_ids) { lhs.multibuffer.update(cx, |lhs_multibuffer, lhs_cx| { LhsEditor::sync_path_excerpts( path, old_rhs_ids, rhs_multibuffer, lhs_multibuffer, diff, rhs_display_map, lhs_cx, ); }); } } result } impl LhsEditor { fn update_path_excerpts_from_rhs( path_key: PathKey, rhs_multibuffer: &MultiBuffer, lhs_multibuffer: &mut MultiBuffer, diff: Entity, lhs_cx: &mut Context, ) -> Vec<(ExcerptId, ExcerptId)> { let rhs_excerpt_ids: Vec = rhs_multibuffer.excerpts_for_path(&path_key).collect(); let Some(excerpt_id) = rhs_multibuffer.excerpts_for_path(&path_key).next() else { lhs_multibuffer.remove_excerpts_for_path(path_key, lhs_cx); return Vec::new(); }; let rhs_multibuffer_snapshot = rhs_multibuffer.snapshot(lhs_cx); let main_buffer = rhs_multibuffer_snapshot .buffer_for_excerpt(excerpt_id) .unwrap(); let base_text_buffer = diff.read(lhs_cx).base_text_buffer(); let diff_snapshot = diff.read(lhs_cx).snapshot(lhs_cx); let base_text_buffer_snapshot = base_text_buffer.read(lhs_cx).snapshot(); let new = rhs_multibuffer .excerpts_for_buffer(main_buffer.remote_id(), lhs_cx) .into_iter() .map(|(_, excerpt_range)| { let point_range_to_base_text_point_range = |range: Range| { let start = diff_snapshot .buffer_point_to_base_text_range( Point::new(range.start.row, 0), main_buffer, ) .start; let end = diff_snapshot .buffer_point_to_base_text_range(Point::new(range.end.row, 0), main_buffer) .end; let end_column = diff_snapshot.base_text().line_len(end.row); Point::new(start.row, 0)..Point::new(end.row, end_column) }; let rhs = excerpt_range.primary.to_point(main_buffer); let context = excerpt_range.context.to_point(main_buffer); ExcerptRange { primary: point_range_to_base_text_point_range(rhs), context: point_range_to_base_text_point_range(context), } }) .collect(); let (ids, _) = lhs_multibuffer.update_path_excerpts( path_key.clone(), base_text_buffer.clone(), &base_text_buffer_snapshot, new, lhs_cx, ); if !ids.is_empty() && lhs_multibuffer .diff_for(base_text_buffer.read(lhs_cx).remote_id()) .is_none_or(|old_diff| old_diff.entity_id() != diff.entity_id()) { lhs_multibuffer.add_inverted_diff(diff, lhs_cx); } let lhs_excerpt_ids: Vec = lhs_multibuffer.excerpts_for_path(&path_key).collect(); debug_assert_eq!(rhs_excerpt_ids.len(), lhs_excerpt_ids.len()); lhs_excerpt_ids.into_iter().zip(rhs_excerpt_ids).collect() } fn sync_path_excerpts( path_key: PathKey, old_rhs_excerpt_ids: Vec, rhs_multibuffer: &MultiBuffer, lhs_multibuffer: &mut MultiBuffer, diff: Entity, rhs_display_map: &Entity, lhs_cx: &mut Context, ) { let old_lhs_excerpt_ids: Vec = lhs_multibuffer.excerpts_for_path(&path_key).collect(); if let Some(companion) = rhs_display_map.read(lhs_cx).companion().cloned() { companion.update(lhs_cx, |c, _| { c.remove_excerpt_mappings(old_lhs_excerpt_ids, old_rhs_excerpt_ids); }); } let mappings = Self::update_path_excerpts_from_rhs( path_key, rhs_multibuffer, lhs_multibuffer, diff.clone(), lhs_cx, ); let lhs_buffer_id = diff.read(lhs_cx).base_text(lhs_cx).remote_id(); let rhs_buffer_id = diff.read(lhs_cx).buffer_id; if let Some(companion) = rhs_display_map.read(lhs_cx).companion().cloned() { companion.update(lhs_cx, |c, _| { for (lhs, rhs) in mappings { c.add_excerpt_mapping(lhs, rhs); } c.add_buffer_mapping(lhs_buffer_id, rhs_buffer_id); }); } } } #[cfg(test)] mod tests { use std::sync::Arc; use buffer_diff::BufferDiff; use collections::{HashMap, HashSet}; use fs::FakeFs; use gpui::Element as _; use gpui::{AppContext as _, Entity, Pixels, VisualTestContext}; use language::language_settings::SoftWrap; use language::{Buffer, Capability}; use multi_buffer::{MultiBuffer, PathKey}; use pretty_assertions::assert_eq; use project::Project; use rand::rngs::StdRng; use settings::{DiffViewStyle, SettingsStore}; use ui::{VisualContext as _, div, px}; use workspace::MultiWorkspace; use crate::SplittableEditor; use crate::display_map::{ BlockPlacement, BlockProperties, BlockStyle, Crease, FoldPlaceholder, }; use crate::inlays::Inlay; use crate::test::{editor_content_with_blocks_and_width, set_block_content_for_tests}; use multi_buffer::MultiBufferOffset; async fn init_test( cx: &mut gpui::TestAppContext, soft_wrap: SoftWrap, style: DiffViewStyle, ) -> (Entity, &mut VisualTestContext) { cx.update(|cx| { let store = SettingsStore::test(cx); cx.set_global(store); theme::init(theme::LoadThemes::JustBase, cx); crate::init(cx); }); let project = Project::test(FakeFs::new(cx.executor()), [], cx).await; let (multi_workspace, cx) = cx.add_window_view(|window, cx| MultiWorkspace::test_new(project.clone(), window, cx)); let workspace = multi_workspace.read_with(cx, |mw, _| mw.workspace().clone()); let rhs_multibuffer = cx.new(|cx| { let mut multibuffer = MultiBuffer::new(Capability::ReadWrite); multibuffer.set_all_diff_hunks_expanded(cx); multibuffer }); let editor = cx.new_window_entity(|window, cx| { let editor = SplittableEditor::new( style, rhs_multibuffer.clone(), project.clone(), workspace, window, cx, ); editor.rhs_editor.update(cx, |editor, cx| { editor.set_soft_wrap_mode(soft_wrap, cx); }); if let Some(lhs) = &editor.lhs { lhs.editor.update(cx, |editor, cx| { editor.set_soft_wrap_mode(soft_wrap, cx); }); } editor }); (editor, cx) } fn buffer_with_diff( base_text: &str, current_text: &str, cx: &mut VisualTestContext, ) -> (Entity, Entity) { let buffer = cx.new(|cx| Buffer::local(current_text.to_string(), cx)); let diff = cx.new(|cx| { BufferDiff::new_with_base_text(base_text, &buffer.read(cx).text_snapshot(), cx) }); (buffer, diff) } #[track_caller] fn assert_split_content( editor: &Entity, expected_rhs: String, expected_lhs: String, cx: &mut VisualTestContext, ) { assert_split_content_with_widths( editor, px(3000.0), px(3000.0), expected_rhs, expected_lhs, cx, ); } #[track_caller] fn assert_split_content_with_widths( editor: &Entity, rhs_width: Pixels, lhs_width: Pixels, expected_rhs: String, expected_lhs: String, cx: &mut VisualTestContext, ) { let (rhs_editor, lhs_editor) = editor.update(cx, |editor, _cx| { let lhs = editor.lhs.as_ref().expect("should have lhs editor"); (editor.rhs_editor.clone(), lhs.editor.clone()) }); // Make sure both sides learn if the other has soft-wrapped let _ = editor_content_with_blocks_and_width(&rhs_editor, rhs_width, cx); cx.run_until_parked(); let _ = editor_content_with_blocks_and_width(&lhs_editor, lhs_width, cx); cx.run_until_parked(); let rhs_content = editor_content_with_blocks_and_width(&rhs_editor, rhs_width, cx); let lhs_content = editor_content_with_blocks_and_width(&lhs_editor, lhs_width, cx); if rhs_content != expected_rhs || lhs_content != expected_lhs { editor.update(cx, |editor, cx| editor.debug_print(cx)); } assert_eq!(rhs_content, expected_rhs, "rhs"); assert_eq!(lhs_content, expected_lhs, "lhs"); } #[gpui::test(iterations = 100)] async fn test_random_split_editor(mut rng: StdRng, cx: &mut gpui::TestAppContext) { use rand::prelude::*; let (editor, cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let operations = std::env::var("OPERATIONS") .map(|i| i.parse().expect("invalid `OPERATIONS` variable")) .unwrap_or(10); let rng = &mut rng; for _ in 0..operations { let buffers = editor.update(cx, |editor, cx| { editor.rhs_editor.read(cx).buffer().read(cx).all_buffers() }); if buffers.is_empty() { log::info!("adding excerpts to empty multibuffer"); editor.update(cx, |editor, cx| { editor.randomly_edit_excerpts(rng, 2, cx); editor.check_invariants(true, cx); }); continue; } let mut quiesced = false; match rng.random_range(0..100) { 0..=44 => { log::info!("randomly editing multibuffer"); editor.update(cx, |editor, cx| { editor.rhs_multibuffer.update(cx, |multibuffer, cx| { multibuffer.randomly_edit(rng, 5, cx); }) }) } 45..=64 => { log::info!("randomly undoing/redoing in single buffer"); let buffer = buffers.iter().choose(rng).unwrap(); buffer.update(cx, |buffer, cx| { buffer.randomly_undo_redo(rng, cx); }); } 65..=79 => { log::info!("mutating excerpts"); editor.update(cx, |editor, cx| { editor.randomly_edit_excerpts(rng, 2, cx); }); } _ => { log::info!("quiescing"); for buffer in buffers { let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); let diff = editor.update(cx, |editor, cx| { editor .rhs_multibuffer .read(cx) .diff_for(buffer.read(cx).remote_id()) .unwrap() }); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); let diff_snapshot = diff.read_with(cx, |diff, cx| diff.snapshot(cx)); let ranges = diff_snapshot .hunks(&buffer_snapshot) .map(|hunk| hunk.range) .collect::>(); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path(path, buffer, ranges, 2, diff, cx); }); } quiesced = true; } } editor.update(cx, |editor, cx| { editor.check_invariants(quiesced, cx); }); } } #[gpui::test] async fn test_basic_alignment(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc ddd eee fff " .unindent(); let current_text = " aaa ddd eee fff " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa § spacer § spacer ddd eee fff" .unindent(), " § § ----- aaa bbb ccc ddd eee fff" .unindent(), &mut cx, ); buffer.update(cx, |buffer, cx| { buffer.edit([(Point::new(3, 0)..Point::new(3, 3), "FFF")], None, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa § spacer § spacer ddd eee FFF" .unindent(), " § § ----- aaa bbb ccc ddd eee fff" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa § spacer § spacer ddd eee FFF" .unindent(), " § § ----- aaa bbb ccc ddd eee fff" .unindent(), &mut cx, ); } #[gpui::test] async fn test_deleting_unmodified_lines(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text1 = " aaa bbb ccc ddd eee" .unindent(); let base_text2 = " fff ggg hhh iii jjj" .unindent(); let (buffer1, diff1) = buffer_with_diff(&base_text1, &base_text1, &mut cx); let (buffer2, diff2) = buffer_with_diff(&base_text2, &base_text2, &mut cx); editor.update(cx, |editor, cx| { let path1 = PathKey::for_buffer(&buffer1, cx); editor.set_excerpts_for_path( path1, buffer1.clone(), vec![Point::new(0, 0)..buffer1.read(cx).max_point()], 0, diff1.clone(), cx, ); let path2 = PathKey::for_buffer(&buffer2, cx); editor.set_excerpts_for_path( path2, buffer2.clone(), vec![Point::new(0, 0)..buffer2.read(cx).max_point()], 1, diff2.clone(), cx, ); }); cx.run_until_parked(); buffer1.update(cx, |buffer, cx| { buffer.edit( [ (Point::new(0, 0)..Point::new(1, 0), ""), (Point::new(3, 0)..Point::new(4, 0), ""), ], None, cx, ); }); buffer2.update(cx, |buffer, cx| { buffer.edit( [ (Point::new(0, 0)..Point::new(1, 0), ""), (Point::new(3, 0)..Point::new(4, 0), ""), ], None, cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- § spacer bbb ccc § spacer eee § § ----- § spacer ggg hhh § spacer jjj" .unindent(), " § § ----- aaa bbb ccc ddd eee § § ----- fff ggg hhh iii jjj" .unindent(), &mut cx, ); let buffer1_snapshot = buffer1.read_with(cx, |buffer, _| buffer.text_snapshot()); diff1.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer1_snapshot, cx); }); let buffer2_snapshot = buffer2.read_with(cx, |buffer, _| buffer.text_snapshot()); diff2.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer2_snapshot, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- § spacer bbb ccc § spacer eee § § ----- § spacer ggg hhh § spacer jjj" .unindent(), " § § ----- aaa bbb ccc ddd eee § § ----- fff ggg hhh iii jjj" .unindent(), &mut cx, ); } #[gpui::test] async fn test_deleting_added_line(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc ddd " .unindent(); let current_text = " aaa NEW1 NEW2 ccc ddd " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa NEW1 NEW2 ccc ddd" .unindent(), " § § ----- aaa bbb § spacer ccc ddd" .unindent(), &mut cx, ); buffer.update(cx, |buffer, cx| { buffer.edit([(Point::new(2, 0)..Point::new(3, 0), "")], None, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa NEW1 ccc ddd" .unindent(), " § § ----- aaa bbb ccc ddd" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa NEW1 ccc ddd" .unindent(), " § § ----- aaa bbb ccc ddd" .unindent(), &mut cx, ); } #[gpui::test] async fn test_inserting_consecutive_blank_line(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc ddd " .unindent(); let current_text = " aaa bbb CCC ddd " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); buffer.update(cx, |buffer, cx| { buffer.edit([(Point::new(1, 3)..Point::new(1, 3), "\n")], None, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb CCC ddd" .unindent(), " § § ----- aaa bbb § spacer ccc ddd" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb CCC ddd" .unindent(), " § § ----- aaa bbb ccc § spacer ddd" .unindent(), &mut cx, ); } #[gpui::test] async fn test_reverting_deletion_hunk(cx: &mut gpui::TestAppContext) { use git::Restore; use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc ddd eee " .unindent(); let current_text = " aaa ddd eee " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa § spacer § spacer ddd eee" .unindent(), " § § ----- aaa bbb ccc ddd eee" .unindent(), &mut cx, ); let rhs_editor = editor.update(cx, |editor, _cx| editor.rhs_editor.clone()); cx.update_window_entity(&rhs_editor, |editor, window, cx| { editor.change_selections(crate::SelectionEffects::no_scroll(), window, cx, |s| { s.select_ranges([Point::new(1, 0)..Point::new(1, 0)]); }); editor.git_restore(&Restore, window, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc ddd eee" .unindent(), " § § ----- aaa bbb ccc ddd eee" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc ddd eee" .unindent(), " § § ----- aaa bbb ccc ddd eee" .unindent(), &mut cx, ); } #[gpui::test] async fn test_deleting_added_lines(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa old1 old2 old3 old4 zzz " .unindent(); let current_text = " aaa new1 new2 new3 new4 zzz " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); buffer.update(cx, |buffer, cx| { buffer.edit( [ (Point::new(2, 0)..Point::new(3, 0), ""), (Point::new(4, 0)..Point::new(5, 0), ""), ], None, cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa new1 new3 § spacer § spacer zzz" .unindent(), " § § ----- aaa old1 old2 old3 old4 zzz" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa new1 new3 § spacer § spacer zzz" .unindent(), " § § ----- aaa old1 old2 old3 old4 zzz" .unindent(), &mut cx, ); } #[gpui::test] async fn test_soft_wrap_at_end_of_excerpt(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let text = "aaaa bbbb cccc dddd eeee ffff"; let (buffer1, diff1) = buffer_with_diff(text, text, &mut cx); let (buffer2, diff2) = buffer_with_diff(text, text, &mut cx); editor.update(cx, |editor, cx| { let end = Point::new(0, text.len() as u32); let path1 = PathKey::for_buffer(&buffer1, cx); editor.set_excerpts_for_path( path1, buffer1.clone(), vec![Point::new(0, 0)..end], 0, diff1.clone(), cx, ); let path2 = PathKey::for_buffer(&buffer2, cx); editor.set_excerpts_for_path( path2, buffer2.clone(), vec![Point::new(0, 0)..end], 0, diff2.clone(), cx, ); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(200.0), px(400.0), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff" .unindent(), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer" .unindent(), &mut cx, ); } #[gpui::test] async fn test_soft_wrap_before_modification_hunk(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaaa bbbb cccc dddd eeee ffff old line one old line two " .unindent(); let current_text = " aaaa bbbb cccc dddd eeee ffff new line " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(200.0), px(400.0), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff new line § spacer" .unindent(), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer old line one old line two" .unindent(), &mut cx, ); } #[gpui::test] async fn test_soft_wrap_before_deletion_hunk(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaaa bbbb cccc dddd eeee ffff deleted line one deleted line two after " .unindent(); let current_text = " aaaa bbbb cccc dddd eeee ffff after " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer § spacer § spacer § spacer § spacer after" .unindent(), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff deleted line\x20 one deleted line\x20 two after" .unindent(), &mut cx, ); } #[gpui::test] async fn test_soft_wrap_spacer_after_editing_second_line(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let text = " aaaa bbbb cccc dddd eeee ffff short " .unindent(); let (buffer, diff) = buffer_with_diff(&text, &text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer short" .unindent(), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff short" .unindent(), &mut cx, ); buffer.update(cx, |buffer, cx| { buffer.edit([(Point::new(1, 0)..Point::new(1, 5), "modified")], None, cx); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer modified" .unindent(), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff short" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer modified" .unindent(), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff short" .unindent(), &mut cx, ); } #[gpui::test] async fn test_no_base_text(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let (buffer1, diff1) = buffer_with_diff("xxx\nyyy", "xxx\nyyy", &mut cx); let current_text = " aaa bbb ccc " .unindent(); let buffer2 = cx.new(|cx| Buffer::local(current_text.to_string(), cx)); let diff2 = cx.new(|cx| BufferDiff::new(&buffer2.read(cx).text_snapshot(), cx)); editor.update(cx, |editor, cx| { let path1 = PathKey::for_buffer(&buffer1, cx); editor.set_excerpts_for_path( path1, buffer1.clone(), vec![Point::new(0, 0)..buffer1.read(cx).max_point()], 0, diff1.clone(), cx, ); let path2 = PathKey::for_buffer(&buffer2, cx); editor.set_excerpts_for_path( path2, buffer2.clone(), vec![Point::new(0, 0)..buffer2.read(cx).max_point()], 1, diff2.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- xxx yyy § § ----- aaa bbb ccc" .unindent(), " § § ----- xxx yyy § § ----- § spacer § spacer § spacer" .unindent(), &mut cx, ); buffer1.update(cx, |buffer, cx| { buffer.edit([(Point::new(0, 3)..Point::new(0, 3), "z")], None, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- xxxz yyy § § ----- aaa bbb ccc" .unindent(), " § § ----- xxx yyy § § ----- § spacer § spacer § spacer" .unindent(), &mut cx, ); } #[gpui::test] async fn test_deleting_char_in_added_line(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc " .unindent(); let current_text = " NEW1 NEW2 ccc " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- NEW1 NEW2 ccc" .unindent(), " § § ----- aaa bbb ccc" .unindent(), &mut cx, ); buffer.update(cx, |buffer, cx| { buffer.edit([(Point::new(1, 3)..Point::new(1, 4), "")], None, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- NEW1 NEW ccc" .unindent(), " § § ----- aaa bbb ccc" .unindent(), &mut cx, ); } #[gpui::test] async fn test_soft_wrap_spacer_before_added_line(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = "aaaa bbbb cccc dddd eeee ffff\n"; let current_text = " aaaa bbbb cccc dddd eeee ffff added line " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer added line" .unindent(), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff § spacer" .unindent(), &mut cx, ); assert_split_content_with_widths( &editor, px(200.0), px(400.0), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff added line" .unindent(), " § § ----- aaaa bbbb cccc dddd eeee ffff § spacer § spacer § spacer" .unindent(), &mut cx, ); } #[gpui::test] #[ignore] async fn test_joining_added_line_with_unmodified_line(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc ddd eee " .unindent(); let current_text = " aaa NEW eee " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa NEW § spacer § spacer eee" .unindent(), " § § ----- aaa bbb ccc ddd eee" .unindent(), &mut cx, ); buffer.update(cx, |buffer, cx| { buffer.edit([(Point::new(1, 3)..Point::new(2, 0), "")], None, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa § spacer § spacer § spacer NEWeee" .unindent(), " § § ----- aaa bbb ccc ddd eee" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa NEWeee § spacer § spacer § spacer" .unindent(), " § § ----- aaa bbb ccc ddd eee" .unindent(), &mut cx, ); } #[gpui::test] async fn test_added_file_at_end(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = ""; let current_text = " aaaa bbbb cccc dddd eeee ffff bbb ccc " .unindent(); let (buffer, diff) = buffer_with_diff(base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaaa bbbb cccc dddd eeee ffff bbb ccc" .unindent(), " § § ----- § spacer § spacer § spacer" .unindent(), &mut cx, ); assert_split_content_with_widths( &editor, px(200.0), px(200.0), " § § ----- aaaa bbbb\x20 cccc dddd\x20 eeee ffff bbb ccc" .unindent(), " § § ----- § spacer § spacer § spacer § spacer § spacer" .unindent(), &mut cx, ); } #[gpui::test] async fn test_adding_line_to_addition_hunk(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc " .unindent(); let current_text = " aaa bbb xxx yyy ccc " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb xxx yyy ccc" .unindent(), " § § ----- aaa bbb § spacer § spacer ccc" .unindent(), &mut cx, ); buffer.update(cx, |buffer, cx| { buffer.edit([(Point::new(3, 3)..Point::new(3, 3), "\nzzz")], None, cx); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb xxx yyy zzz ccc" .unindent(), " § § ----- aaa bbb § spacer § spacer § spacer ccc" .unindent(), &mut cx, ); } #[gpui::test] async fn test_scrolling(cx: &mut gpui::TestAppContext) { use crate::test::editor_content_with_blocks_and_size; use gpui::size; use rope::Point; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let long_line = "x".repeat(200); let mut lines: Vec = (0..50).map(|i| format!("line {i}")).collect(); lines[25] = long_line; let content = lines.join("\n"); let (buffer, diff) = buffer_with_diff(&content, &content, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); let (rhs_editor, lhs_editor) = editor.update(cx, |editor, _cx| { let lhs = editor.lhs.as_ref().expect("should have lhs editor"); (editor.rhs_editor.clone(), lhs.editor.clone()) }); rhs_editor.update_in(cx, |e, window, cx| { e.set_scroll_position(gpui::Point::new(0., 10.), window, cx); }); let rhs_pos = rhs_editor.update_in(cx, |e, window, cx| e.snapshot(window, cx).scroll_position()); let lhs_pos = lhs_editor.update_in(cx, |e, window, cx| e.snapshot(window, cx).scroll_position()); assert_eq!(rhs_pos.y, 10., "RHS should be scrolled to row 10"); assert_eq!( lhs_pos.y, rhs_pos.y, "LHS should have same scroll position as RHS after set_scroll_position" ); let draw_size = size(px(300.), px(300.)); rhs_editor.update_in(cx, |e, window, cx| { e.change_selections(Some(crate::Autoscroll::fit()).into(), window, cx, |s| { s.select_ranges([Point::new(25, 150)..Point::new(25, 150)]); }); }); let _ = editor_content_with_blocks_and_size(&rhs_editor, draw_size, &mut cx); cx.run_until_parked(); let _ = editor_content_with_blocks_and_size(&lhs_editor, draw_size, &mut cx); cx.run_until_parked(); let rhs_pos = rhs_editor.update_in(cx, |e, window, cx| e.snapshot(window, cx).scroll_position()); let lhs_pos = lhs_editor.update_in(cx, |e, window, cx| e.snapshot(window, cx).scroll_position()); assert!( rhs_pos.y > 0., "RHS should have scrolled vertically to show cursor at row 25" ); assert!( rhs_pos.x > 0., "RHS should have scrolled horizontally to show cursor at column 150" ); assert_eq!( lhs_pos.y, rhs_pos.y, "LHS should have same vertical scroll position as RHS after autoscroll" ); assert_eq!( lhs_pos.x, rhs_pos.x, "LHS should have same horizontal scroll position as RHS after autoscroll" ) } #[gpui::test] async fn test_edit_line_before_soft_wrapped_line_preceding_hunk(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::EditorWidth, DiffViewStyle::Split).await; let base_text = " first line aaaa bbbb cccc dddd eeee ffff original " .unindent(); let current_text = " first line aaaa bbbb cccc dddd eeee ffff modified " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- first line aaaa bbbb cccc dddd eeee ffff § spacer § spacer modified" .unindent(), " § § ----- first line aaaa bbbb\x20 cccc dddd\x20 eeee ffff original" .unindent(), &mut cx, ); buffer.update(cx, |buffer, cx| { buffer.edit( [(Point::new(0, 0)..Point::new(0, 10), "edited first")], None, cx, ); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- edited first aaaa bbbb cccc dddd eeee ffff § spacer § spacer modified" .unindent(), " § § ----- first line aaaa bbbb\x20 cccc dddd\x20 eeee ffff original" .unindent(), &mut cx, ); let buffer_snapshot = buffer.read_with(cx, |buffer, _| buffer.text_snapshot()); diff.update(cx, |diff, cx| { diff.recalculate_diff_sync(&buffer_snapshot, cx); }); cx.run_until_parked(); assert_split_content_with_widths( &editor, px(400.0), px(200.0), " § § ----- edited first aaaa bbbb cccc dddd eeee ffff § spacer § spacer modified" .unindent(), " § § ----- first line aaaa bbbb\x20 cccc dddd\x20 eeee ffff original" .unindent(), &mut cx, ); } #[gpui::test] async fn test_custom_block_sync_between_split_views(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let base_text = " bbb ccc " .unindent(); let current_text = " aaa bbb ccc " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc" .unindent(), " § § ----- § spacer bbb ccc" .unindent(), &mut cx, ); let block_ids = editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); let anchor = snapshot.anchor_before(Point::new(2, 0)); rhs_editor.insert_blocks( [BlockProperties { placement: BlockPlacement::Above(anchor), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }], None, cx, ) }) }); let rhs_editor = editor.read_with(cx, |editor, _| editor.rhs_editor.clone()); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); cx.update(|_, cx| { set_block_content_for_tests(&rhs_editor, block_ids[0], cx, |_| { "custom block".to_string() }); }); let lhs_block_id = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); *mapping.borrow().get(&block_ids[0]).unwrap() }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id, cx, |_| { "custom block".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb § custom block ccc" .unindent(), " § § ----- § spacer bbb § custom block ccc" .unindent(), &mut cx, ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.remove_blocks(HashSet::from_iter(block_ids), None, cx); }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc" .unindent(), " § § ----- § spacer bbb ccc" .unindent(), &mut cx, ); } #[gpui::test] async fn test_custom_block_deletion_and_resplit_sync(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let base_text = " bbb ccc " .unindent(); let current_text = " aaa bbb ccc " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc" .unindent(), " § § ----- § spacer bbb ccc" .unindent(), &mut cx, ); let block_ids = editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); let anchor1 = snapshot.anchor_before(Point::new(2, 0)); let anchor2 = snapshot.anchor_before(Point::new(3, 0)); rhs_editor.insert_blocks( [ BlockProperties { placement: BlockPlacement::Above(anchor1), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }, BlockProperties { placement: BlockPlacement::Above(anchor2), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }, ], None, cx, ) }) }); let rhs_editor = editor.read_with(cx, |editor, _| editor.rhs_editor.clone()); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); cx.update(|_, cx| { set_block_content_for_tests(&rhs_editor, block_ids[0], cx, |_| { "custom block 1".to_string() }); set_block_content_for_tests(&rhs_editor, block_ids[1], cx, |_| { "custom block 2".to_string() }); }); let (lhs_block_id_1, lhs_block_id_2) = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); ( *mapping.borrow().get(&block_ids[0]).unwrap(), *mapping.borrow().get(&block_ids[1]).unwrap(), ) }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id_1, cx, |_| { "custom block 1".to_string() }); set_block_content_for_tests(&lhs_editor, lhs_block_id_2, cx, |_| { "custom block 2".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb § custom block 1 ccc § custom block 2" .unindent(), " § § ----- § spacer bbb § custom block 1 ccc § custom block 2" .unindent(), &mut cx, ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.remove_blocks(HashSet::from_iter([block_ids[0]]), None, cx); }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc § custom block 2" .unindent(), " § § ----- § spacer bbb ccc § custom block 2" .unindent(), &mut cx, ); editor.update_in(cx, |splittable_editor, window, cx| { splittable_editor.unsplit(window, cx); }); cx.run_until_parked(); editor.update_in(cx, |splittable_editor, window, cx| { splittable_editor.split(window, cx); }); cx.run_until_parked(); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); let lhs_block_id_2 = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); *mapping.borrow().get(&block_ids[1]).unwrap() }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id_2, cx, |_| { "custom block 2".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc § custom block 2" .unindent(), " § § ----- § spacer bbb ccc § custom block 2" .unindent(), &mut cx, ); } #[gpui::test] async fn test_custom_block_sync_with_unsplit_start(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let base_text = " bbb ccc " .unindent(); let current_text = " aaa bbb ccc " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); editor.update_in(cx, |splittable_editor, window, cx| { splittable_editor.unsplit(window, cx); }); cx.run_until_parked(); let rhs_editor = editor.read_with(cx, |editor, _| editor.rhs_editor.clone()); let block_ids = editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); let anchor1 = snapshot.anchor_before(Point::new(2, 0)); let anchor2 = snapshot.anchor_before(Point::new(3, 0)); rhs_editor.insert_blocks( [ BlockProperties { placement: BlockPlacement::Above(anchor1), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }, BlockProperties { placement: BlockPlacement::Above(anchor2), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }, ], None, cx, ) }) }); cx.update(|_, cx| { set_block_content_for_tests(&rhs_editor, block_ids[0], cx, |_| { "custom block 1".to_string() }); set_block_content_for_tests(&rhs_editor, block_ids[1], cx, |_| { "custom block 2".to_string() }); }); cx.run_until_parked(); let rhs_content = editor_content_with_blocks_and_width(&rhs_editor, px(3000.0), &mut cx); assert_eq!( rhs_content, " § § ----- aaa bbb § custom block 1 ccc § custom block 2" .unindent(), "rhs content before split" ); editor.update_in(cx, |splittable_editor, window, cx| { splittable_editor.split(window, cx); }); cx.run_until_parked(); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); let (lhs_block_id_1, lhs_block_id_2) = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); ( *mapping.borrow().get(&block_ids[0]).unwrap(), *mapping.borrow().get(&block_ids[1]).unwrap(), ) }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id_1, cx, |_| { "custom block 1".to_string() }); set_block_content_for_tests(&lhs_editor, lhs_block_id_2, cx, |_| { "custom block 2".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb § custom block 1 ccc § custom block 2" .unindent(), " § § ----- § spacer bbb § custom block 1 ccc § custom block 2" .unindent(), &mut cx, ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.remove_blocks(HashSet::from_iter([block_ids[0]]), None, cx); }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc § custom block 2" .unindent(), " § § ----- § spacer bbb ccc § custom block 2" .unindent(), &mut cx, ); editor.update_in(cx, |splittable_editor, window, cx| { splittable_editor.unsplit(window, cx); }); cx.run_until_parked(); editor.update_in(cx, |splittable_editor, window, cx| { splittable_editor.split(window, cx); }); cx.run_until_parked(); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); let lhs_block_id_2 = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); *mapping.borrow().get(&block_ids[1]).unwrap() }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id_2, cx, |_| { "custom block 2".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc § custom block 2" .unindent(), " § § ----- § spacer bbb ccc § custom block 2" .unindent(), &mut cx, ); let new_block_ids = editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); let anchor = snapshot.anchor_before(Point::new(2, 0)); rhs_editor.insert_blocks( [BlockProperties { placement: BlockPlacement::Above(anchor), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }], None, cx, ) }) }); cx.update(|_, cx| { set_block_content_for_tests(&rhs_editor, new_block_ids[0], cx, |_| { "custom block 3".to_string() }); }); let lhs_block_id_3 = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); *mapping.borrow().get(&new_block_ids[0]).unwrap() }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id_3, cx, |_| { "custom block 3".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb § custom block 3 ccc § custom block 2" .unindent(), " § § ----- § spacer bbb § custom block 3 ccc § custom block 2" .unindent(), &mut cx, ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.remove_blocks(HashSet::from_iter([new_block_ids[0]]), None, cx); }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc § custom block 2" .unindent(), " § § ----- § spacer bbb ccc § custom block 2" .unindent(), &mut cx, ); } #[gpui::test] async fn test_buffer_folding_sync(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Unified).await; let base_text1 = " aaa bbb ccc" .unindent(); let current_text1 = " aaa bbb ccc" .unindent(); let base_text2 = " ddd eee fff" .unindent(); let current_text2 = " ddd eee fff" .unindent(); let (buffer1, diff1) = buffer_with_diff(&base_text1, ¤t_text1, &mut cx); let (buffer2, diff2) = buffer_with_diff(&base_text2, ¤t_text2, &mut cx); let buffer1_id = buffer1.read_with(cx, |buffer, _| buffer.remote_id()); let buffer2_id = buffer2.read_with(cx, |buffer, _| buffer.remote_id()); editor.update(cx, |editor, cx| { let path1 = PathKey::for_buffer(&buffer1, cx); editor.set_excerpts_for_path( path1, buffer1.clone(), vec![Point::new(0, 0)..buffer1.read(cx).max_point()], 0, diff1.clone(), cx, ); let path2 = PathKey::for_buffer(&buffer2, cx); editor.set_excerpts_for_path( path2, buffer2.clone(), vec![Point::new(0, 0)..buffer2.read(cx).max_point()], 1, diff2.clone(), cx, ); }); cx.run_until_parked(); editor.update(cx, |editor, cx| { editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.fold_buffer(buffer1_id, cx); }); }); cx.run_until_parked(); let rhs_buffer1_folded = editor.read_with(cx, |editor, cx| { editor.rhs_editor.read(cx).is_buffer_folded(buffer1_id, cx) }); assert!( rhs_buffer1_folded, "buffer1 should be folded in rhs before split" ); editor.update_in(cx, |editor, window, cx| { editor.split(window, cx); }); cx.run_until_parked(); let (rhs_editor, lhs_editor) = editor.read_with(cx, |editor, _cx| { ( editor.rhs_editor.clone(), editor.lhs.as_ref().unwrap().editor.clone(), ) }); let rhs_buffer1_folded = rhs_editor.read_with(cx, |editor, cx| editor.is_buffer_folded(buffer1_id, cx)); assert!( rhs_buffer1_folded, "buffer1 should be folded in rhs after split" ); let base_buffer1_id = diff1.read_with(cx, |diff, cx| diff.base_text(cx).remote_id()); let lhs_buffer1_folded = lhs_editor.read_with(cx, |editor, cx| { editor.is_buffer_folded(base_buffer1_id, cx) }); assert!( lhs_buffer1_folded, "buffer1 should be folded in lhs after split" ); assert_split_content( &editor, " § § ----- § § ----- ddd eee fff" .unindent(), " § § ----- § § ----- ddd eee fff" .unindent(), &mut cx, ); editor.update(cx, |editor, cx| { editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.fold_buffer(buffer2_id, cx); }); }); cx.run_until_parked(); let rhs_buffer2_folded = rhs_editor.read_with(cx, |editor, cx| editor.is_buffer_folded(buffer2_id, cx)); assert!(rhs_buffer2_folded, "buffer2 should be folded in rhs"); let base_buffer2_id = diff2.read_with(cx, |diff, cx| diff.base_text(cx).remote_id()); let lhs_buffer2_folded = lhs_editor.read_with(cx, |editor, cx| { editor.is_buffer_folded(base_buffer2_id, cx) }); assert!(lhs_buffer2_folded, "buffer2 should be folded in lhs"); let rhs_buffer1_still_folded = rhs_editor.read_with(cx, |editor, cx| editor.is_buffer_folded(buffer1_id, cx)); assert!( rhs_buffer1_still_folded, "buffer1 should still be folded in rhs" ); let lhs_buffer1_still_folded = lhs_editor.read_with(cx, |editor, cx| { editor.is_buffer_folded(base_buffer1_id, cx) }); assert!( lhs_buffer1_still_folded, "buffer1 should still be folded in lhs" ); assert_split_content( &editor, " § § ----- § § -----" .unindent(), " § § ----- § § -----" .unindent(), &mut cx, ); } #[gpui::test] async fn test_custom_block_in_middle_of_added_hunk(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let base_text = " ddd eee " .unindent(); let current_text = " aaa bbb ccc ddd eee " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc ddd eee" .unindent(), " § § ----- § spacer § spacer § spacer ddd eee" .unindent(), &mut cx, ); let block_ids = editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); let anchor = snapshot.anchor_before(Point::new(2, 0)); rhs_editor.insert_blocks( [BlockProperties { placement: BlockPlacement::Above(anchor), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }], None, cx, ) }) }); let rhs_editor = editor.read_with(cx, |editor, _| editor.rhs_editor.clone()); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); cx.update(|_, cx| { set_block_content_for_tests(&rhs_editor, block_ids[0], cx, |_| { "custom block".to_string() }); }); let lhs_block_id = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); *mapping.borrow().get(&block_ids[0]).unwrap() }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id, cx, |_| { "custom block".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb § custom block ccc ddd eee" .unindent(), " § § ----- § spacer § spacer § spacer § custom block ddd eee" .unindent(), &mut cx, ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.remove_blocks(HashSet::from_iter(block_ids), None, cx); }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc ddd eee" .unindent(), " § § ----- § spacer § spacer § spacer ddd eee" .unindent(), &mut cx, ); } #[gpui::test] async fn test_custom_block_below_in_middle_of_added_hunk(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let base_text = " ddd eee " .unindent(); let current_text = " aaa bbb ccc ddd eee " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc ddd eee" .unindent(), " § § ----- § spacer § spacer § spacer ddd eee" .unindent(), &mut cx, ); let block_ids = editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); let anchor = snapshot.anchor_after(Point::new(1, 3)); rhs_editor.insert_blocks( [BlockProperties { placement: BlockPlacement::Below(anchor), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }], None, cx, ) }) }); let rhs_editor = editor.read_with(cx, |editor, _| editor.rhs_editor.clone()); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); cx.update(|_, cx| { set_block_content_for_tests(&rhs_editor, block_ids[0], cx, |_| { "custom block".to_string() }); }); let lhs_block_id = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); *mapping.borrow().get(&block_ids[0]).unwrap() }); cx.update(|_, cx| { set_block_content_for_tests(&lhs_editor, lhs_block_id, cx, |_| { "custom block".to_string() }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb § custom block ccc ddd eee" .unindent(), " § § ----- § spacer § spacer § spacer § custom block ddd eee" .unindent(), &mut cx, ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.remove_blocks(HashSet::from_iter(block_ids), None, cx); }); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa bbb ccc ddd eee" .unindent(), " § § ----- § spacer § spacer § spacer ddd eee" .unindent(), &mut cx, ); } #[gpui::test] async fn test_custom_block_resize_syncs_balancing_block(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let base_text = " bbb ccc " .unindent(); let current_text = " aaa bbb ccc " .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); let block_ids = editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); let anchor = snapshot.anchor_before(Point::new(2, 0)); rhs_editor.insert_blocks( [BlockProperties { placement: BlockPlacement::Above(anchor), height: Some(1), style: BlockStyle::Fixed, render: Arc::new(|_| div().into_any()), priority: 0, }], None, cx, ) }) }); let rhs_editor = editor.read_with(cx, |editor, _| editor.rhs_editor.clone()); let lhs_editor = editor.read_with(cx, |editor, _| editor.lhs.as_ref().unwrap().editor.clone()); let lhs_block_id = lhs_editor.read_with(cx, |lhs_editor, cx| { let display_map = lhs_editor.display_map.read(cx); let companion = display_map.companion().unwrap().read(cx); let mapping = companion .custom_block_to_balancing_block(rhs_editor.read(cx).display_map.entity_id()); *mapping.borrow().get(&block_ids[0]).unwrap() }); cx.run_until_parked(); let get_block_height = |editor: &Entity, block_id: crate::CustomBlockId, cx: &mut VisualTestContext| { editor.update_in(cx, |editor, window, cx| { let snapshot = editor.snapshot(window, cx); snapshot .block_for_id(crate::BlockId::Custom(block_id)) .map(|block| block.height()) }) }; assert_eq!( get_block_height(&rhs_editor, block_ids[0], &mut cx), Some(1) ); assert_eq!( get_block_height(&lhs_editor, lhs_block_id, &mut cx), Some(1) ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let mut heights = HashMap::default(); heights.insert(block_ids[0], 3); rhs_editor.resize_blocks(heights, None, cx); }); }); cx.run_until_parked(); assert_eq!( get_block_height(&rhs_editor, block_ids[0], &mut cx), Some(3) ); assert_eq!( get_block_height(&lhs_editor, lhs_block_id, &mut cx), Some(3) ); editor.update(cx, |splittable_editor, cx| { splittable_editor.rhs_editor.update(cx, |rhs_editor, cx| { let mut heights = HashMap::default(); heights.insert(block_ids[0], 5); rhs_editor.resize_blocks(heights, None, cx); }); }); cx.run_until_parked(); assert_eq!( get_block_height(&rhs_editor, block_ids[0], &mut cx), Some(5) ); assert_eq!( get_block_height(&lhs_editor, lhs_block_id, &mut cx), Some(5) ); } #[gpui::test] async fn test_range_folds_removed_on_split(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Unified).await; let base_text = " aaa bbb ccc ddd eee" .unindent(); let current_text = " aaa bbb ccc ddd eee" .unindent(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..buffer.read(cx).max_point()], 0, diff.clone(), cx, ); }); cx.run_until_parked(); editor.update_in(cx, |editor, window, cx| { editor.rhs_editor.update(cx, |rhs_editor, cx| { rhs_editor.fold_creases( vec![Crease::simple( Point::new(1, 0)..Point::new(3, 0), FoldPlaceholder::test(), )], false, window, cx, ); }); }); cx.run_until_parked(); editor.update_in(cx, |editor, window, cx| { editor.split(window, cx); }); cx.run_until_parked(); let (rhs_editor, lhs_editor) = editor.read_with(cx, |editor, _cx| { ( editor.rhs_editor.clone(), editor.lhs.as_ref().unwrap().editor.clone(), ) }); let rhs_has_folds_after_split = rhs_editor.update(cx, |editor, cx| { let snapshot = editor.display_snapshot(cx); snapshot .folds_in_range(MultiBufferOffset(0)..snapshot.buffer_snapshot().len()) .next() .is_some() }); assert!( !rhs_has_folds_after_split, "rhs should not have range folds after split" ); let lhs_has_folds = lhs_editor.update(cx, |editor, cx| { let snapshot = editor.display_snapshot(cx); snapshot .folds_in_range(MultiBufferOffset(0)..snapshot.buffer_snapshot().len()) .next() .is_some() }); assert!(!lhs_has_folds, "lhs should not have any range folds"); } #[gpui::test] async fn test_multiline_inlays_create_spacers(cx: &mut gpui::TestAppContext) { use rope::Point; use unindent::Unindent as _; let (editor, mut cx) = init_test(cx, SoftWrap::None, DiffViewStyle::Split).await; let base_text = " aaa bbb ccc ddd " .unindent(); let current_text = base_text.clone(); let (buffer, diff) = buffer_with_diff(&base_text, ¤t_text, &mut cx); editor.update(cx, |editor, cx| { let path = PathKey::for_buffer(&buffer, cx); editor.set_excerpts_for_path( path, buffer.clone(), vec![Point::new(0, 0)..Point::new(3, 3)], 0, diff.clone(), cx, ); }); cx.run_until_parked(); let rhs_editor = editor.read_with(cx, |e, _| e.rhs_editor.clone()); rhs_editor.update(cx, |rhs_editor, cx| { let snapshot = rhs_editor.buffer().read(cx).snapshot(cx); rhs_editor.splice_inlays( &[], vec![ Inlay::edit_prediction( 0, snapshot.anchor_after(Point::new(0, 3)), "\nINLAY_WITHIN", ), Inlay::edit_prediction( 1, snapshot.anchor_after(Point::new(1, 3)), "\nINLAY_MID_1\nINLAY_MID_2", ), Inlay::edit_prediction( 2, snapshot.anchor_after(Point::new(3, 3)), "\nINLAY_END_1\nINLAY_END_2", ), ], cx, ); }); cx.run_until_parked(); assert_split_content( &editor, " § § ----- aaa INLAY_WITHIN bbb INLAY_MID_1 INLAY_MID_2 ccc ddd INLAY_END_1 INLAY_END_2" .unindent(), " § § ----- aaa § spacer bbb § spacer § spacer ccc ddd § spacer § spacer" .unindent(), &mut cx, ); } }