1pub mod display_map;
2mod element;
3mod hover_popover;
4pub mod items;
5pub mod movement;
6mod multi_buffer;
7pub mod selections_collection;
8
9#[cfg(any(test, feature = "test-support"))]
10pub mod test;
11
12use aho_corasick::AhoCorasick;
13use anyhow::Result;
14use clock::ReplicaId;
15use collections::{BTreeMap, Bound, HashMap, HashSet, VecDeque};
16pub use display_map::DisplayPoint;
17use display_map::*;
18pub use element::*;
19use fuzzy::{StringMatch, StringMatchCandidate};
20use gpui::{
21 actions,
22 color::Color,
23 elements::*,
24 executor,
25 fonts::{self, HighlightStyle, TextStyle},
26 geometry::vector::{vec2f, Vector2F},
27 impl_actions, impl_internal_actions,
28 platform::CursorStyle,
29 text_layout, AppContext, AsyncAppContext, ClipboardItem, Element, ElementBox, Entity,
30 ModelHandle, MutableAppContext, RenderContext, Task, View, ViewContext, ViewHandle,
31 WeakViewHandle,
32};
33use hover_popover::{hide_hover, HoverState};
34pub use language::{char_kind, CharKind};
35use language::{
36 BracketPair, Buffer, CodeAction, CodeLabel, Completion, Diagnostic, DiagnosticSeverity,
37 IndentKind, IndentSize, Language, OffsetRangeExt, Point, Selection, SelectionGoal,
38 TransactionId,
39};
40use multi_buffer::MultiBufferChunks;
41pub use multi_buffer::{
42 Anchor, AnchorRangeExt, ExcerptId, ExcerptRange, MultiBuffer, MultiBufferSnapshot, ToOffset,
43 ToPoint,
44};
45use ordered_float::OrderedFloat;
46use project::{Project, ProjectPath, ProjectTransaction};
47use selections_collection::{resolve_multiple, MutableSelectionsCollection, SelectionsCollection};
48use serde::{Deserialize, Serialize};
49use settings::Settings;
50use smallvec::SmallVec;
51use smol::Timer;
52use snippet::Snippet;
53use std::{
54 any::TypeId,
55 borrow::Cow,
56 cmp::{self, Ordering, Reverse},
57 mem,
58 ops::{Deref, DerefMut, Range, RangeInclusive},
59 sync::Arc,
60 time::{Duration, Instant},
61};
62pub use sum_tree::Bias;
63use theme::{DiagnosticStyle, Theme};
64use util::{post_inc, ResultExt, TryFutureExt};
65use workspace::{ItemNavHistory, Workspace};
66
67const CURSOR_BLINK_INTERVAL: Duration = Duration::from_millis(500);
68const MAX_LINE_LEN: usize = 1024;
69const MIN_NAVIGATION_HISTORY_ROW_DELTA: i64 = 10;
70const MAX_SELECTION_HISTORY_LEN: usize = 1024;
71
72#[derive(Clone, Deserialize, PartialEq)]
73pub struct SelectNext {
74 #[serde(default)]
75 pub replace_newest: bool,
76}
77
78#[derive(Clone, PartialEq)]
79pub struct GoToDiagnostic(pub Direction);
80
81#[derive(Clone, PartialEq)]
82pub struct Scroll(pub Vector2F);
83
84#[derive(Clone, PartialEq)]
85pub struct Select(pub SelectPhase);
86
87#[derive(Clone, Debug, PartialEq)]
88pub struct Jump {
89 path: ProjectPath,
90 position: Point,
91 anchor: language::Anchor,
92}
93
94#[derive(Clone, Deserialize, PartialEq)]
95pub struct Input(pub String);
96
97#[derive(Clone, Deserialize, PartialEq)]
98pub struct SelectToBeginningOfLine {
99 #[serde(default)]
100 stop_at_soft_wraps: bool,
101}
102
103#[derive(Clone, Deserialize, PartialEq)]
104pub struct SelectToEndOfLine {
105 #[serde(default)]
106 stop_at_soft_wraps: bool,
107}
108
109#[derive(Clone, Deserialize, PartialEq)]
110pub struct ToggleCodeActions {
111 #[serde(default)]
112 pub deployed_from_indicator: bool,
113}
114
115#[derive(Clone, Default, Deserialize, PartialEq)]
116pub struct ConfirmCompletion {
117 #[serde(default)]
118 pub item_ix: Option<usize>,
119}
120
121#[derive(Clone, Default, Deserialize, PartialEq)]
122pub struct ConfirmCodeAction {
123 #[serde(default)]
124 pub item_ix: Option<usize>,
125}
126
127actions!(
128 editor,
129 [
130 Cancel,
131 Backspace,
132 Delete,
133 Newline,
134 GoToNextDiagnostic,
135 GoToPrevDiagnostic,
136 Indent,
137 Outdent,
138 DeleteLine,
139 DeleteToPreviousWordStart,
140 DeleteToPreviousSubwordStart,
141 DeleteToNextWordEnd,
142 DeleteToNextSubwordEnd,
143 DeleteToBeginningOfLine,
144 DeleteToEndOfLine,
145 CutToEndOfLine,
146 DuplicateLine,
147 MoveLineUp,
148 MoveLineDown,
149 Transpose,
150 Cut,
151 Copy,
152 Paste,
153 Undo,
154 Redo,
155 MoveUp,
156 MoveDown,
157 MoveLeft,
158 MoveRight,
159 MoveToPreviousWordStart,
160 MoveToPreviousSubwordStart,
161 MoveToNextWordEnd,
162 MoveToNextSubwordEnd,
163 MoveToBeginningOfLine,
164 MoveToEndOfLine,
165 MoveToBeginning,
166 MoveToEnd,
167 SelectUp,
168 SelectDown,
169 SelectLeft,
170 SelectRight,
171 SelectToPreviousWordStart,
172 SelectToPreviousSubwordStart,
173 SelectToNextWordEnd,
174 SelectToNextSubwordEnd,
175 SelectToBeginning,
176 SelectToEnd,
177 SelectAll,
178 SelectLine,
179 SplitSelectionIntoLines,
180 AddSelectionAbove,
181 AddSelectionBelow,
182 Tab,
183 TabPrev,
184 ToggleComments,
185 SelectLargerSyntaxNode,
186 SelectSmallerSyntaxNode,
187 GoToDefinition,
188 MoveToEnclosingBracket,
189 UndoSelection,
190 RedoSelection,
191 FindAllReferences,
192 Rename,
193 ConfirmRename,
194 PageUp,
195 PageDown,
196 Fold,
197 UnfoldLines,
198 FoldSelectedRanges,
199 ShowCompletions,
200 OpenExcerpts,
201 RestartLanguageServer,
202 Hover,
203 ]
204);
205
206impl_actions!(
207 editor,
208 [
209 Input,
210 SelectNext,
211 SelectToBeginningOfLine,
212 SelectToEndOfLine,
213 ToggleCodeActions,
214 ConfirmCompletion,
215 ConfirmCodeAction,
216 ]
217);
218
219impl_internal_actions!(editor, [Scroll, Select, Jump]);
220
221enum DocumentHighlightRead {}
222enum DocumentHighlightWrite {}
223
224#[derive(Copy, Clone, PartialEq, Eq)]
225pub enum Direction {
226 Prev,
227 Next,
228}
229
230pub fn init(cx: &mut MutableAppContext) {
231 cx.add_action(Editor::new_file);
232 cx.add_action(|this: &mut Editor, action: &Scroll, cx| this.set_scroll_position(action.0, cx));
233 cx.add_action(Editor::select);
234 cx.add_action(Editor::cancel);
235 cx.add_action(Editor::handle_input);
236 cx.add_action(Editor::newline);
237 cx.add_action(Editor::backspace);
238 cx.add_action(Editor::delete);
239 cx.add_action(Editor::tab);
240 cx.add_action(Editor::tab_prev);
241 cx.add_action(Editor::indent);
242 cx.add_action(Editor::outdent);
243 cx.add_action(Editor::delete_line);
244 cx.add_action(Editor::delete_to_previous_word_start);
245 cx.add_action(Editor::delete_to_previous_subword_start);
246 cx.add_action(Editor::delete_to_next_word_end);
247 cx.add_action(Editor::delete_to_next_subword_end);
248 cx.add_action(Editor::delete_to_beginning_of_line);
249 cx.add_action(Editor::delete_to_end_of_line);
250 cx.add_action(Editor::cut_to_end_of_line);
251 cx.add_action(Editor::duplicate_line);
252 cx.add_action(Editor::move_line_up);
253 cx.add_action(Editor::move_line_down);
254 cx.add_action(Editor::transpose);
255 cx.add_action(Editor::cut);
256 cx.add_action(Editor::copy);
257 cx.add_action(Editor::paste);
258 cx.add_action(Editor::undo);
259 cx.add_action(Editor::redo);
260 cx.add_action(Editor::move_up);
261 cx.add_action(Editor::move_down);
262 cx.add_action(Editor::move_left);
263 cx.add_action(Editor::move_right);
264 cx.add_action(Editor::move_to_previous_word_start);
265 cx.add_action(Editor::move_to_previous_subword_start);
266 cx.add_action(Editor::move_to_next_word_end);
267 cx.add_action(Editor::move_to_next_subword_end);
268 cx.add_action(Editor::move_to_beginning_of_line);
269 cx.add_action(Editor::move_to_end_of_line);
270 cx.add_action(Editor::move_to_beginning);
271 cx.add_action(Editor::move_to_end);
272 cx.add_action(Editor::select_up);
273 cx.add_action(Editor::select_down);
274 cx.add_action(Editor::select_left);
275 cx.add_action(Editor::select_right);
276 cx.add_action(Editor::select_to_previous_word_start);
277 cx.add_action(Editor::select_to_previous_subword_start);
278 cx.add_action(Editor::select_to_next_word_end);
279 cx.add_action(Editor::select_to_next_subword_end);
280 cx.add_action(Editor::select_to_beginning_of_line);
281 cx.add_action(Editor::select_to_end_of_line);
282 cx.add_action(Editor::select_to_beginning);
283 cx.add_action(Editor::select_to_end);
284 cx.add_action(Editor::select_all);
285 cx.add_action(Editor::select_line);
286 cx.add_action(Editor::split_selection_into_lines);
287 cx.add_action(Editor::add_selection_above);
288 cx.add_action(Editor::add_selection_below);
289 cx.add_action(Editor::select_next);
290 cx.add_action(Editor::toggle_comments);
291 cx.add_action(Editor::select_larger_syntax_node);
292 cx.add_action(Editor::select_smaller_syntax_node);
293 cx.add_action(Editor::move_to_enclosing_bracket);
294 cx.add_action(Editor::undo_selection);
295 cx.add_action(Editor::redo_selection);
296 cx.add_action(Editor::go_to_next_diagnostic);
297 cx.add_action(Editor::go_to_prev_diagnostic);
298 cx.add_action(Editor::go_to_definition);
299 cx.add_action(Editor::page_up);
300 cx.add_action(Editor::page_down);
301 cx.add_action(Editor::fold);
302 cx.add_action(Editor::unfold_lines);
303 cx.add_action(Editor::fold_selected_ranges);
304 cx.add_action(Editor::show_completions);
305 cx.add_action(Editor::toggle_code_actions);
306 cx.add_action(Editor::open_excerpts);
307 cx.add_action(Editor::jump);
308 cx.add_action(Editor::restart_language_server);
309 cx.add_async_action(Editor::confirm_completion);
310 cx.add_async_action(Editor::confirm_code_action);
311 cx.add_async_action(Editor::rename);
312 cx.add_async_action(Editor::confirm_rename);
313 cx.add_async_action(Editor::find_all_references);
314
315 hover_popover::init(cx);
316
317 workspace::register_project_item::<Editor>(cx);
318 workspace::register_followable_item::<Editor>(cx);
319}
320
321trait InvalidationRegion {
322 fn ranges(&self) -> &[Range<Anchor>];
323}
324
325#[derive(Clone, Debug, PartialEq)]
326pub enum SelectPhase {
327 Begin {
328 position: DisplayPoint,
329 add: bool,
330 click_count: usize,
331 },
332 BeginColumnar {
333 position: DisplayPoint,
334 overshoot: u32,
335 },
336 Extend {
337 position: DisplayPoint,
338 click_count: usize,
339 },
340 Update {
341 position: DisplayPoint,
342 overshoot: u32,
343 scroll_position: Vector2F,
344 },
345 End,
346}
347
348#[derive(Clone, Debug)]
349pub enum SelectMode {
350 Character,
351 Word(Range<Anchor>),
352 Line(Range<Anchor>),
353 All,
354}
355
356#[derive(PartialEq, Eq)]
357pub enum Autoscroll {
358 Fit,
359 Center,
360 Newest,
361}
362
363#[derive(Copy, Clone, PartialEq, Eq)]
364pub enum EditorMode {
365 SingleLine,
366 AutoHeight { max_lines: usize },
367 Full,
368}
369
370#[derive(Clone)]
371pub enum SoftWrap {
372 None,
373 EditorWidth,
374 Column(u32),
375}
376
377#[derive(Clone)]
378pub struct EditorStyle {
379 pub text: TextStyle,
380 pub placeholder_text: Option<TextStyle>,
381 pub theme: theme::Editor,
382}
383
384type CompletionId = usize;
385
386pub type GetFieldEditorTheme = fn(&theme::Theme) -> theme::FieldEditor;
387
388type OverrideTextStyle = dyn Fn(&EditorStyle) -> Option<HighlightStyle>;
389
390pub struct Editor {
391 handle: WeakViewHandle<Self>,
392 buffer: ModelHandle<MultiBuffer>,
393 display_map: ModelHandle<DisplayMap>,
394 pub selections: SelectionsCollection,
395 columnar_selection_tail: Option<Anchor>,
396 add_selections_state: Option<AddSelectionsState>,
397 select_next_state: Option<SelectNextState>,
398 selection_history: SelectionHistory,
399 autoclose_stack: InvalidationStack<BracketPairState>,
400 snippet_stack: InvalidationStack<SnippetState>,
401 select_larger_syntax_node_stack: Vec<Box<[Selection<usize>]>>,
402 active_diagnostics: Option<ActiveDiagnosticGroup>,
403 scroll_position: Vector2F,
404 scroll_top_anchor: Anchor,
405 autoscroll_request: Option<(Autoscroll, bool)>,
406 soft_wrap_mode_override: Option<settings::SoftWrap>,
407 get_field_editor_theme: Option<GetFieldEditorTheme>,
408 override_text_style: Option<Box<OverrideTextStyle>>,
409 project: Option<ModelHandle<Project>>,
410 focused: bool,
411 show_local_cursors: bool,
412 show_local_selections: bool,
413 blink_epoch: usize,
414 blinking_paused: bool,
415 mode: EditorMode,
416 vertical_scroll_margin: f32,
417 placeholder_text: Option<Arc<str>>,
418 highlighted_rows: Option<Range<u32>>,
419 background_highlights: BTreeMap<TypeId, (fn(&Theme) -> Color, Vec<Range<Anchor>>)>,
420 nav_history: Option<ItemNavHistory>,
421 context_menu: Option<ContextMenu>,
422 completion_tasks: Vec<(CompletionId, Task<Option<()>>)>,
423 next_completion_id: CompletionId,
424 available_code_actions: Option<(ModelHandle<Buffer>, Arc<[CodeAction]>)>,
425 code_actions_task: Option<Task<()>>,
426 document_highlights_task: Option<Task<()>>,
427 pending_rename: Option<RenameState>,
428 searchable: bool,
429 cursor_shape: CursorShape,
430 keymap_context_layers: BTreeMap<TypeId, gpui::keymap::Context>,
431 input_enabled: bool,
432 leader_replica_id: Option<u16>,
433 hover_state: HoverState,
434}
435
436pub struct EditorSnapshot {
437 pub mode: EditorMode,
438 pub display_snapshot: DisplaySnapshot,
439 pub placeholder_text: Option<Arc<str>>,
440 is_focused: bool,
441 scroll_position: Vector2F,
442 scroll_top_anchor: Anchor,
443}
444
445#[derive(Clone, Debug)]
446struct SelectionHistoryEntry {
447 selections: Arc<[Selection<Anchor>]>,
448 select_next_state: Option<SelectNextState>,
449 add_selections_state: Option<AddSelectionsState>,
450}
451
452enum SelectionHistoryMode {
453 Normal,
454 Undoing,
455 Redoing,
456}
457
458impl Default for SelectionHistoryMode {
459 fn default() -> Self {
460 Self::Normal
461 }
462}
463
464#[derive(Default)]
465struct SelectionHistory {
466 selections_by_transaction:
467 HashMap<TransactionId, (Arc<[Selection<Anchor>]>, Option<Arc<[Selection<Anchor>]>>)>,
468 mode: SelectionHistoryMode,
469 undo_stack: VecDeque<SelectionHistoryEntry>,
470 redo_stack: VecDeque<SelectionHistoryEntry>,
471}
472
473impl SelectionHistory {
474 fn insert_transaction(
475 &mut self,
476 transaction_id: TransactionId,
477 selections: Arc<[Selection<Anchor>]>,
478 ) {
479 self.selections_by_transaction
480 .insert(transaction_id, (selections, None));
481 }
482
483 fn transaction(
484 &self,
485 transaction_id: TransactionId,
486 ) -> Option<&(Arc<[Selection<Anchor>]>, Option<Arc<[Selection<Anchor>]>>)> {
487 self.selections_by_transaction.get(&transaction_id)
488 }
489
490 fn transaction_mut(
491 &mut self,
492 transaction_id: TransactionId,
493 ) -> Option<&mut (Arc<[Selection<Anchor>]>, Option<Arc<[Selection<Anchor>]>>)> {
494 self.selections_by_transaction.get_mut(&transaction_id)
495 }
496
497 fn push(&mut self, entry: SelectionHistoryEntry) {
498 if !entry.selections.is_empty() {
499 match self.mode {
500 SelectionHistoryMode::Normal => {
501 self.push_undo(entry);
502 self.redo_stack.clear();
503 }
504 SelectionHistoryMode::Undoing => self.push_redo(entry),
505 SelectionHistoryMode::Redoing => self.push_undo(entry),
506 }
507 }
508 }
509
510 fn push_undo(&mut self, entry: SelectionHistoryEntry) {
511 if self
512 .undo_stack
513 .back()
514 .map_or(true, |e| e.selections != entry.selections)
515 {
516 self.undo_stack.push_back(entry);
517 if self.undo_stack.len() > MAX_SELECTION_HISTORY_LEN {
518 self.undo_stack.pop_front();
519 }
520 }
521 }
522
523 fn push_redo(&mut self, entry: SelectionHistoryEntry) {
524 if self
525 .redo_stack
526 .back()
527 .map_or(true, |e| e.selections != entry.selections)
528 {
529 self.redo_stack.push_back(entry);
530 if self.redo_stack.len() > MAX_SELECTION_HISTORY_LEN {
531 self.redo_stack.pop_front();
532 }
533 }
534 }
535}
536
537#[derive(Clone, Debug)]
538struct AddSelectionsState {
539 above: bool,
540 stack: Vec<usize>,
541}
542
543#[derive(Clone, Debug)]
544struct SelectNextState {
545 query: AhoCorasick,
546 wordwise: bool,
547 done: bool,
548}
549
550struct BracketPairState {
551 ranges: Vec<Range<Anchor>>,
552 pair: BracketPair,
553}
554
555#[derive(Debug)]
556struct SnippetState {
557 ranges: Vec<Vec<Range<Anchor>>>,
558 active_index: usize,
559}
560
561pub struct RenameState {
562 pub range: Range<Anchor>,
563 pub old_name: Arc<str>,
564 pub editor: ViewHandle<Editor>,
565 block_id: BlockId,
566}
567
568struct InvalidationStack<T>(Vec<T>);
569
570enum ContextMenu {
571 Completions(CompletionsMenu),
572 CodeActions(CodeActionsMenu),
573}
574
575impl ContextMenu {
576 fn select_prev(&mut self, cx: &mut ViewContext<Editor>) -> bool {
577 if self.visible() {
578 match self {
579 ContextMenu::Completions(menu) => menu.select_prev(cx),
580 ContextMenu::CodeActions(menu) => menu.select_prev(cx),
581 }
582 true
583 } else {
584 false
585 }
586 }
587
588 fn select_next(&mut self, cx: &mut ViewContext<Editor>) -> bool {
589 if self.visible() {
590 match self {
591 ContextMenu::Completions(menu) => menu.select_next(cx),
592 ContextMenu::CodeActions(menu) => menu.select_next(cx),
593 }
594 true
595 } else {
596 false
597 }
598 }
599
600 fn visible(&self) -> bool {
601 match self {
602 ContextMenu::Completions(menu) => menu.visible(),
603 ContextMenu::CodeActions(menu) => menu.visible(),
604 }
605 }
606
607 fn render(
608 &self,
609 cursor_position: DisplayPoint,
610 style: EditorStyle,
611 cx: &mut RenderContext<Editor>,
612 ) -> (DisplayPoint, ElementBox) {
613 match self {
614 ContextMenu::Completions(menu) => (cursor_position, menu.render(style, cx)),
615 ContextMenu::CodeActions(menu) => menu.render(cursor_position, style, cx),
616 }
617 }
618}
619
620struct CompletionsMenu {
621 id: CompletionId,
622 initial_position: Anchor,
623 buffer: ModelHandle<Buffer>,
624 completions: Arc<[Completion]>,
625 match_candidates: Vec<StringMatchCandidate>,
626 matches: Arc<[StringMatch]>,
627 selected_item: usize,
628 list: UniformListState,
629}
630
631impl CompletionsMenu {
632 fn select_prev(&mut self, cx: &mut ViewContext<Editor>) {
633 if self.selected_item > 0 {
634 self.selected_item -= 1;
635 self.list.scroll_to(ScrollTarget::Show(self.selected_item));
636 }
637 cx.notify();
638 }
639
640 fn select_next(&mut self, cx: &mut ViewContext<Editor>) {
641 if self.selected_item + 1 < self.matches.len() {
642 self.selected_item += 1;
643 self.list.scroll_to(ScrollTarget::Show(self.selected_item));
644 }
645 cx.notify();
646 }
647
648 fn visible(&self) -> bool {
649 !self.matches.is_empty()
650 }
651
652 fn render(&self, style: EditorStyle, cx: &mut RenderContext<Editor>) -> ElementBox {
653 enum CompletionTag {}
654
655 let completions = self.completions.clone();
656 let matches = self.matches.clone();
657 let selected_item = self.selected_item;
658 let container_style = style.autocomplete.container;
659 UniformList::new(
660 self.list.clone(),
661 matches.len(),
662 cx,
663 move |_, range, items, cx| {
664 let start_ix = range.start;
665 for (ix, mat) in matches[range].iter().enumerate() {
666 let completion = &completions[mat.candidate_id];
667 let item_ix = start_ix + ix;
668 items.push(
669 MouseEventHandler::new::<CompletionTag, _, _>(
670 mat.candidate_id,
671 cx,
672 |state, _| {
673 let item_style = if item_ix == selected_item {
674 style.autocomplete.selected_item
675 } else if state.hovered {
676 style.autocomplete.hovered_item
677 } else {
678 style.autocomplete.item
679 };
680
681 Text::new(completion.label.text.clone(), style.text.clone())
682 .with_soft_wrap(false)
683 .with_highlights(combine_syntax_and_fuzzy_match_highlights(
684 &completion.label.text,
685 style.text.color.into(),
686 styled_runs_for_code_label(
687 &completion.label,
688 &style.syntax,
689 ),
690 &mat.positions,
691 ))
692 .contained()
693 .with_style(item_style)
694 .boxed()
695 },
696 )
697 .with_cursor_style(CursorStyle::PointingHand)
698 .on_mouse_down(move |_, cx| {
699 cx.dispatch_action(ConfirmCompletion {
700 item_ix: Some(item_ix),
701 });
702 })
703 .boxed(),
704 );
705 }
706 },
707 )
708 .with_width_from_item(
709 self.matches
710 .iter()
711 .enumerate()
712 .max_by_key(|(_, mat)| {
713 self.completions[mat.candidate_id]
714 .label
715 .text
716 .chars()
717 .count()
718 })
719 .map(|(ix, _)| ix),
720 )
721 .contained()
722 .with_style(container_style)
723 .boxed()
724 }
725
726 pub async fn filter(&mut self, query: Option<&str>, executor: Arc<executor::Background>) {
727 let mut matches = if let Some(query) = query {
728 fuzzy::match_strings(
729 &self.match_candidates,
730 query,
731 false,
732 100,
733 &Default::default(),
734 executor,
735 )
736 .await
737 } else {
738 self.match_candidates
739 .iter()
740 .enumerate()
741 .map(|(candidate_id, candidate)| StringMatch {
742 candidate_id,
743 score: Default::default(),
744 positions: Default::default(),
745 string: candidate.string.clone(),
746 })
747 .collect()
748 };
749 matches.sort_unstable_by_key(|mat| {
750 (
751 Reverse(OrderedFloat(mat.score)),
752 self.completions[mat.candidate_id].sort_key(),
753 )
754 });
755
756 for mat in &mut matches {
757 let filter_start = self.completions[mat.candidate_id].label.filter_range.start;
758 for position in &mut mat.positions {
759 *position += filter_start;
760 }
761 }
762
763 self.matches = matches.into();
764 }
765}
766
767#[derive(Clone)]
768struct CodeActionsMenu {
769 actions: Arc<[CodeAction]>,
770 buffer: ModelHandle<Buffer>,
771 selected_item: usize,
772 list: UniformListState,
773 deployed_from_indicator: bool,
774}
775
776impl CodeActionsMenu {
777 fn select_prev(&mut self, cx: &mut ViewContext<Editor>) {
778 if self.selected_item > 0 {
779 self.selected_item -= 1;
780 cx.notify()
781 }
782 }
783
784 fn select_next(&mut self, cx: &mut ViewContext<Editor>) {
785 if self.selected_item + 1 < self.actions.len() {
786 self.selected_item += 1;
787 cx.notify()
788 }
789 }
790
791 fn visible(&self) -> bool {
792 !self.actions.is_empty()
793 }
794
795 fn render(
796 &self,
797 mut cursor_position: DisplayPoint,
798 style: EditorStyle,
799 cx: &mut RenderContext<Editor>,
800 ) -> (DisplayPoint, ElementBox) {
801 enum ActionTag {}
802
803 let container_style = style.autocomplete.container;
804 let actions = self.actions.clone();
805 let selected_item = self.selected_item;
806 let element = UniformList::new(
807 self.list.clone(),
808 actions.len(),
809 cx,
810 move |_, range, items, cx| {
811 let start_ix = range.start;
812 for (ix, action) in actions[range].iter().enumerate() {
813 let item_ix = start_ix + ix;
814 items.push(
815 MouseEventHandler::new::<ActionTag, _, _>(item_ix, cx, |state, _| {
816 let item_style = if item_ix == selected_item {
817 style.autocomplete.selected_item
818 } else if state.hovered {
819 style.autocomplete.hovered_item
820 } else {
821 style.autocomplete.item
822 };
823
824 Text::new(action.lsp_action.title.clone(), style.text.clone())
825 .with_soft_wrap(false)
826 .contained()
827 .with_style(item_style)
828 .boxed()
829 })
830 .with_cursor_style(CursorStyle::PointingHand)
831 .on_mouse_down(move |_, cx| {
832 cx.dispatch_action(ConfirmCodeAction {
833 item_ix: Some(item_ix),
834 });
835 })
836 .boxed(),
837 );
838 }
839 },
840 )
841 .with_width_from_item(
842 self.actions
843 .iter()
844 .enumerate()
845 .max_by_key(|(_, action)| action.lsp_action.title.chars().count())
846 .map(|(ix, _)| ix),
847 )
848 .contained()
849 .with_style(container_style)
850 .boxed();
851
852 if self.deployed_from_indicator {
853 *cursor_position.column_mut() = 0;
854 }
855
856 (cursor_position, element)
857 }
858}
859
860#[derive(Debug)]
861struct ActiveDiagnosticGroup {
862 primary_range: Range<Anchor>,
863 primary_message: String,
864 blocks: HashMap<BlockId, Diagnostic>,
865 is_valid: bool,
866}
867
868#[derive(Serialize, Deserialize)]
869pub struct ClipboardSelection {
870 pub len: usize,
871 pub is_entire_line: bool,
872}
873
874#[derive(Debug)]
875pub struct NavigationData {
876 // Matching offsets for anchor and scroll_top_anchor allows us to recreate the anchor if the buffer
877 // has since been closed
878 cursor_anchor: Anchor,
879 cursor_position: Point,
880 scroll_position: Vector2F,
881 scroll_top_anchor: Anchor,
882 scroll_top_row: u32,
883}
884
885pub struct EditorCreated(pub ViewHandle<Editor>);
886
887impl Editor {
888 pub fn single_line(
889 field_editor_style: Option<GetFieldEditorTheme>,
890 cx: &mut ViewContext<Self>,
891 ) -> Self {
892 let buffer = cx.add_model(|cx| Buffer::new(0, String::new(), cx));
893 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
894 Self::new(
895 EditorMode::SingleLine,
896 buffer,
897 None,
898 field_editor_style,
899 None,
900 cx,
901 )
902 }
903
904 pub fn auto_height(
905 max_lines: usize,
906 field_editor_style: Option<GetFieldEditorTheme>,
907 cx: &mut ViewContext<Self>,
908 ) -> Self {
909 let buffer = cx.add_model(|cx| Buffer::new(0, String::new(), cx));
910 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
911 Self::new(
912 EditorMode::AutoHeight { max_lines },
913 buffer,
914 None,
915 field_editor_style,
916 None,
917 cx,
918 )
919 }
920
921 pub fn for_buffer(
922 buffer: ModelHandle<Buffer>,
923 project: Option<ModelHandle<Project>>,
924 cx: &mut ViewContext<Self>,
925 ) -> Self {
926 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
927 Self::new(EditorMode::Full, buffer, project, None, None, cx)
928 }
929
930 pub fn for_multibuffer(
931 buffer: ModelHandle<MultiBuffer>,
932 project: Option<ModelHandle<Project>>,
933 cx: &mut ViewContext<Self>,
934 ) -> Self {
935 Self::new(EditorMode::Full, buffer, project, None, None, cx)
936 }
937
938 pub fn clone(&self, cx: &mut ViewContext<Self>) -> Self {
939 let mut clone = Self::new(
940 self.mode,
941 self.buffer.clone(),
942 self.project.clone(),
943 self.get_field_editor_theme,
944 Some(self.selections.clone()),
945 cx,
946 );
947 clone.scroll_position = self.scroll_position;
948 clone.scroll_top_anchor = self.scroll_top_anchor.clone();
949 clone.searchable = self.searchable;
950 clone
951 }
952
953 fn new(
954 mode: EditorMode,
955 buffer: ModelHandle<MultiBuffer>,
956 project: Option<ModelHandle<Project>>,
957 get_field_editor_theme: Option<GetFieldEditorTheme>,
958 selections: Option<SelectionsCollection>,
959 cx: &mut ViewContext<Self>,
960 ) -> Self {
961 let display_map = cx.add_model(|cx| {
962 let settings = cx.global::<Settings>();
963 let style = build_style(&*settings, get_field_editor_theme, None, cx);
964 DisplayMap::new(
965 buffer.clone(),
966 style.text.font_id,
967 style.text.font_size,
968 None,
969 2,
970 1,
971 cx,
972 )
973 });
974 cx.observe(&buffer, Self::on_buffer_changed).detach();
975 cx.subscribe(&buffer, Self::on_buffer_event).detach();
976 cx.observe(&display_map, Self::on_display_map_changed)
977 .detach();
978
979 let selections = selections
980 .unwrap_or_else(|| SelectionsCollection::new(display_map.clone(), buffer.clone()));
981
982 let mut this = Self {
983 handle: cx.weak_handle(),
984 buffer,
985 display_map,
986 selections,
987 columnar_selection_tail: None,
988 add_selections_state: None,
989 select_next_state: None,
990 selection_history: Default::default(),
991 autoclose_stack: Default::default(),
992 snippet_stack: Default::default(),
993 select_larger_syntax_node_stack: Vec::new(),
994 active_diagnostics: None,
995 soft_wrap_mode_override: None,
996 get_field_editor_theme,
997 project,
998 scroll_position: Vector2F::zero(),
999 scroll_top_anchor: Anchor::min(),
1000 autoscroll_request: None,
1001 focused: false,
1002 show_local_cursors: false,
1003 show_local_selections: true,
1004 blink_epoch: 0,
1005 blinking_paused: false,
1006 mode,
1007 vertical_scroll_margin: 3.0,
1008 placeholder_text: None,
1009 highlighted_rows: None,
1010 background_highlights: Default::default(),
1011 nav_history: None,
1012 context_menu: None,
1013 completion_tasks: Default::default(),
1014 next_completion_id: 0,
1015 available_code_actions: Default::default(),
1016 code_actions_task: Default::default(),
1017
1018 document_highlights_task: Default::default(),
1019 pending_rename: Default::default(),
1020 searchable: true,
1021 override_text_style: None,
1022 cursor_shape: Default::default(),
1023 keymap_context_layers: Default::default(),
1024 input_enabled: true,
1025 leader_replica_id: None,
1026 hover_state: Default::default(),
1027 };
1028 this.end_selection(cx);
1029
1030 let editor_created_event = EditorCreated(cx.handle());
1031 cx.emit_global(editor_created_event);
1032
1033 this
1034 }
1035
1036 pub fn new_file(
1037 workspace: &mut Workspace,
1038 _: &workspace::NewFile,
1039 cx: &mut ViewContext<Workspace>,
1040 ) {
1041 let project = workspace.project().clone();
1042 if project.read(cx).is_remote() {
1043 cx.propagate_action();
1044 } else if let Some(buffer) = project
1045 .update(cx, |project, cx| project.create_buffer("", None, cx))
1046 .log_err()
1047 {
1048 workspace.add_item(
1049 Box::new(cx.add_view(|cx| Editor::for_buffer(buffer, Some(project.clone()), cx))),
1050 cx,
1051 );
1052 }
1053 }
1054
1055 pub fn replica_id(&self, cx: &AppContext) -> ReplicaId {
1056 self.buffer.read(cx).replica_id()
1057 }
1058
1059 pub fn leader_replica_id(&self) -> Option<ReplicaId> {
1060 self.leader_replica_id
1061 }
1062
1063 pub fn buffer(&self) -> &ModelHandle<MultiBuffer> {
1064 &self.buffer
1065 }
1066
1067 pub fn title(&self, cx: &AppContext) -> String {
1068 self.buffer().read(cx).title(cx)
1069 }
1070
1071 pub fn snapshot(&mut self, cx: &mut MutableAppContext) -> EditorSnapshot {
1072 EditorSnapshot {
1073 mode: self.mode,
1074 display_snapshot: self.display_map.update(cx, |map, cx| map.snapshot(cx)),
1075 scroll_position: self.scroll_position,
1076 scroll_top_anchor: self.scroll_top_anchor.clone(),
1077 placeholder_text: self.placeholder_text.clone(),
1078 is_focused: self
1079 .handle
1080 .upgrade(cx)
1081 .map_or(false, |handle| handle.is_focused(cx)),
1082 }
1083 }
1084
1085 pub fn language_at<'a, T: ToOffset>(
1086 &self,
1087 point: T,
1088 cx: &'a AppContext,
1089 ) -> Option<&'a Arc<Language>> {
1090 self.buffer.read(cx).language_at(point, cx)
1091 }
1092
1093 fn style(&self, cx: &AppContext) -> EditorStyle {
1094 build_style(
1095 cx.global::<Settings>(),
1096 self.get_field_editor_theme,
1097 self.override_text_style.as_deref(),
1098 cx,
1099 )
1100 }
1101
1102 pub fn mode(&self) -> EditorMode {
1103 self.mode
1104 }
1105
1106 pub fn set_placeholder_text(
1107 &mut self,
1108 placeholder_text: impl Into<Arc<str>>,
1109 cx: &mut ViewContext<Self>,
1110 ) {
1111 self.placeholder_text = Some(placeholder_text.into());
1112 cx.notify();
1113 }
1114
1115 pub fn set_vertical_scroll_margin(&mut self, margin_rows: usize, cx: &mut ViewContext<Self>) {
1116 self.vertical_scroll_margin = margin_rows as f32;
1117 cx.notify();
1118 }
1119
1120 pub fn set_scroll_position(&mut self, scroll_position: Vector2F, cx: &mut ViewContext<Self>) {
1121 self.set_scroll_position_internal(scroll_position, true, cx);
1122 }
1123
1124 fn set_scroll_position_internal(
1125 &mut self,
1126 scroll_position: Vector2F,
1127 local: bool,
1128 cx: &mut ViewContext<Self>,
1129 ) {
1130 let map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1131
1132 if scroll_position.y() == 0. {
1133 self.scroll_top_anchor = Anchor::min();
1134 self.scroll_position = scroll_position;
1135 } else {
1136 let scroll_top_buffer_offset =
1137 DisplayPoint::new(scroll_position.y() as u32, 0).to_offset(&map, Bias::Right);
1138 let anchor = map
1139 .buffer_snapshot
1140 .anchor_at(scroll_top_buffer_offset, Bias::Right);
1141 self.scroll_position = vec2f(
1142 scroll_position.x(),
1143 scroll_position.y() - anchor.to_display_point(&map).row() as f32,
1144 );
1145 self.scroll_top_anchor = anchor;
1146 }
1147
1148 self.autoscroll_request.take();
1149 hide_hover(self, cx);
1150
1151 cx.emit(Event::ScrollPositionChanged { local });
1152 cx.notify();
1153 }
1154
1155 fn set_scroll_top_anchor(
1156 &mut self,
1157 anchor: Anchor,
1158 position: Vector2F,
1159 cx: &mut ViewContext<Self>,
1160 ) {
1161 self.scroll_top_anchor = anchor;
1162 self.scroll_position = position;
1163 cx.emit(Event::ScrollPositionChanged { local: false });
1164 cx.notify();
1165 }
1166
1167 pub fn set_cursor_shape(&mut self, cursor_shape: CursorShape, cx: &mut ViewContext<Self>) {
1168 self.cursor_shape = cursor_shape;
1169 cx.notify();
1170 }
1171
1172 pub fn set_clip_at_line_ends(&mut self, clip: bool, cx: &mut ViewContext<Self>) {
1173 if self.display_map.read(cx).clip_at_line_ends != clip {
1174 self.display_map
1175 .update(cx, |map, _| map.clip_at_line_ends = clip);
1176 }
1177 }
1178
1179 pub fn set_keymap_context_layer<Tag: 'static>(&mut self, context: gpui::keymap::Context) {
1180 self.keymap_context_layers
1181 .insert(TypeId::of::<Tag>(), context);
1182 }
1183
1184 pub fn remove_keymap_context_layer<Tag: 'static>(&mut self) {
1185 self.keymap_context_layers.remove(&TypeId::of::<Tag>());
1186 }
1187
1188 pub fn set_input_enabled(&mut self, input_enabled: bool) {
1189 self.input_enabled = input_enabled;
1190 }
1191
1192 pub fn scroll_position(&self, cx: &mut ViewContext<Self>) -> Vector2F {
1193 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1194 compute_scroll_position(&display_map, self.scroll_position, &self.scroll_top_anchor)
1195 }
1196
1197 pub fn clamp_scroll_left(&mut self, max: f32) -> bool {
1198 if max < self.scroll_position.x() {
1199 self.scroll_position.set_x(max);
1200 true
1201 } else {
1202 false
1203 }
1204 }
1205
1206 pub fn autoscroll_vertically(
1207 &mut self,
1208 viewport_height: f32,
1209 line_height: f32,
1210 cx: &mut ViewContext<Self>,
1211 ) -> bool {
1212 let visible_lines = viewport_height / line_height;
1213 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1214 let mut scroll_position =
1215 compute_scroll_position(&display_map, self.scroll_position, &self.scroll_top_anchor);
1216 let max_scroll_top = if matches!(self.mode, EditorMode::AutoHeight { .. }) {
1217 (display_map.max_point().row() as f32 - visible_lines + 1.).max(0.)
1218 } else {
1219 display_map.max_point().row().saturating_sub(1) as f32
1220 };
1221 if scroll_position.y() > max_scroll_top {
1222 scroll_position.set_y(max_scroll_top);
1223 self.set_scroll_position(scroll_position, cx);
1224 }
1225
1226 let (autoscroll, local) = if let Some(autoscroll) = self.autoscroll_request.take() {
1227 autoscroll
1228 } else {
1229 return false;
1230 };
1231
1232 let first_cursor_top;
1233 let last_cursor_bottom;
1234 if let Some(highlighted_rows) = &self.highlighted_rows {
1235 first_cursor_top = highlighted_rows.start as f32;
1236 last_cursor_bottom = first_cursor_top + 1.;
1237 } else if autoscroll == Autoscroll::Newest {
1238 let newest_selection = self.selections.newest::<Point>(cx);
1239 first_cursor_top = newest_selection.head().to_display_point(&display_map).row() as f32;
1240 last_cursor_bottom = first_cursor_top + 1.;
1241 } else {
1242 let selections = self.selections.all::<Point>(cx);
1243 first_cursor_top = selections
1244 .first()
1245 .unwrap()
1246 .head()
1247 .to_display_point(&display_map)
1248 .row() as f32;
1249 last_cursor_bottom = selections
1250 .last()
1251 .unwrap()
1252 .head()
1253 .to_display_point(&display_map)
1254 .row() as f32
1255 + 1.0;
1256 }
1257
1258 let margin = if matches!(self.mode, EditorMode::AutoHeight { .. }) {
1259 0.
1260 } else {
1261 ((visible_lines - (last_cursor_bottom - first_cursor_top)) / 2.0).floor()
1262 };
1263 if margin < 0.0 {
1264 return false;
1265 }
1266
1267 match autoscroll {
1268 Autoscroll::Fit | Autoscroll::Newest => {
1269 let margin = margin.min(self.vertical_scroll_margin);
1270 let target_top = (first_cursor_top - margin).max(0.0);
1271 let target_bottom = last_cursor_bottom + margin;
1272 let start_row = scroll_position.y();
1273 let end_row = start_row + visible_lines;
1274
1275 if target_top < start_row {
1276 scroll_position.set_y(target_top);
1277 self.set_scroll_position_internal(scroll_position, local, cx);
1278 } else if target_bottom >= end_row {
1279 scroll_position.set_y(target_bottom - visible_lines);
1280 self.set_scroll_position_internal(scroll_position, local, cx);
1281 }
1282 }
1283 Autoscroll::Center => {
1284 scroll_position.set_y((first_cursor_top - margin).max(0.0));
1285 self.set_scroll_position_internal(scroll_position, local, cx);
1286 }
1287 }
1288
1289 true
1290 }
1291
1292 pub fn autoscroll_horizontally(
1293 &mut self,
1294 start_row: u32,
1295 viewport_width: f32,
1296 scroll_width: f32,
1297 max_glyph_width: f32,
1298 layouts: &[text_layout::Line],
1299 cx: &mut ViewContext<Self>,
1300 ) -> bool {
1301 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1302 let selections = self.selections.all::<Point>(cx);
1303
1304 let mut target_left;
1305 let mut target_right;
1306
1307 if self.highlighted_rows.is_some() {
1308 target_left = 0.0_f32;
1309 target_right = 0.0_f32;
1310 } else {
1311 target_left = std::f32::INFINITY;
1312 target_right = 0.0_f32;
1313 for selection in selections {
1314 let head = selection.head().to_display_point(&display_map);
1315 if head.row() >= start_row && head.row() < start_row + layouts.len() as u32 {
1316 let start_column = head.column().saturating_sub(3);
1317 let end_column = cmp::min(display_map.line_len(head.row()), head.column() + 3);
1318 target_left = target_left.min(
1319 layouts[(head.row() - start_row) as usize]
1320 .x_for_index(start_column as usize),
1321 );
1322 target_right = target_right.max(
1323 layouts[(head.row() - start_row) as usize].x_for_index(end_column as usize)
1324 + max_glyph_width,
1325 );
1326 }
1327 }
1328 }
1329
1330 target_right = target_right.min(scroll_width);
1331
1332 if target_right - target_left > viewport_width {
1333 return false;
1334 }
1335
1336 let scroll_left = self.scroll_position.x() * max_glyph_width;
1337 let scroll_right = scroll_left + viewport_width;
1338
1339 if target_left < scroll_left {
1340 self.scroll_position.set_x(target_left / max_glyph_width);
1341 true
1342 } else if target_right > scroll_right {
1343 self.scroll_position
1344 .set_x((target_right - viewport_width) / max_glyph_width);
1345 true
1346 } else {
1347 false
1348 }
1349 }
1350
1351 fn selections_did_change(
1352 &mut self,
1353 local: bool,
1354 old_cursor_position: &Anchor,
1355 cx: &mut ViewContext<Self>,
1356 ) {
1357 if self.focused && self.leader_replica_id.is_none() {
1358 self.buffer.update(cx, |buffer, cx| {
1359 buffer.set_active_selections(
1360 &self.selections.disjoint_anchors(),
1361 self.selections.line_mode,
1362 cx,
1363 )
1364 });
1365 }
1366
1367 let display_map = self
1368 .display_map
1369 .update(cx, |display_map, cx| display_map.snapshot(cx));
1370 let buffer = &display_map.buffer_snapshot;
1371 self.add_selections_state = None;
1372 self.select_next_state = None;
1373 self.select_larger_syntax_node_stack.clear();
1374 self.autoclose_stack
1375 .invalidate(&self.selections.disjoint_anchors(), buffer);
1376 self.snippet_stack
1377 .invalidate(&self.selections.disjoint_anchors(), buffer);
1378 self.take_rename(false, cx);
1379
1380 let new_cursor_position = self.selections.newest_anchor().head();
1381
1382 self.push_to_nav_history(
1383 old_cursor_position.clone(),
1384 Some(new_cursor_position.to_point(buffer)),
1385 cx,
1386 );
1387
1388 if local {
1389 let new_cursor_position = self.selections.newest_anchor().head();
1390 let completion_menu = match self.context_menu.as_mut() {
1391 Some(ContextMenu::Completions(menu)) => Some(menu),
1392 _ => {
1393 self.context_menu.take();
1394 None
1395 }
1396 };
1397
1398 if let Some(completion_menu) = completion_menu {
1399 let cursor_position = new_cursor_position.to_offset(buffer);
1400 let (word_range, kind) =
1401 buffer.surrounding_word(completion_menu.initial_position.clone());
1402 if kind == Some(CharKind::Word)
1403 && word_range.to_inclusive().contains(&cursor_position)
1404 {
1405 let query = Self::completion_query(buffer, cursor_position);
1406 cx.background()
1407 .block(completion_menu.filter(query.as_deref(), cx.background().clone()));
1408 self.show_completions(&ShowCompletions, cx);
1409 } else {
1410 self.hide_context_menu(cx);
1411 }
1412 }
1413
1414 hide_hover(self, cx);
1415
1416 if old_cursor_position.to_display_point(&display_map).row()
1417 != new_cursor_position.to_display_point(&display_map).row()
1418 {
1419 self.available_code_actions.take();
1420 }
1421 self.refresh_code_actions(cx);
1422 self.refresh_document_highlights(cx);
1423 }
1424
1425 self.pause_cursor_blinking(cx);
1426 cx.emit(Event::SelectionsChanged { local });
1427 cx.notify();
1428 }
1429
1430 pub fn change_selections<R>(
1431 &mut self,
1432 autoscroll: Option<Autoscroll>,
1433 cx: &mut ViewContext<Self>,
1434 change: impl FnOnce(&mut MutableSelectionsCollection<'_>) -> R,
1435 ) -> R {
1436 let old_cursor_position = self.selections.newest_anchor().head();
1437 self.push_to_selection_history();
1438
1439 let (changed, result) = self.selections.change_with(cx, change);
1440
1441 if changed {
1442 if let Some(autoscroll) = autoscroll {
1443 self.request_autoscroll(autoscroll, cx);
1444 }
1445 self.selections_did_change(true, &old_cursor_position, cx);
1446 }
1447
1448 result
1449 }
1450
1451 pub fn edit<I, S, T>(&mut self, edits: I, cx: &mut ViewContext<Self>)
1452 where
1453 I: IntoIterator<Item = (Range<S>, T)>,
1454 S: ToOffset,
1455 T: Into<Arc<str>>,
1456 {
1457 self.buffer.update(cx, |buffer, cx| buffer.edit(edits, cx));
1458 }
1459
1460 pub fn edit_with_autoindent<I, S, T>(&mut self, edits: I, cx: &mut ViewContext<Self>)
1461 where
1462 I: IntoIterator<Item = (Range<S>, T)>,
1463 S: ToOffset,
1464 T: Into<Arc<str>>,
1465 {
1466 self.buffer
1467 .update(cx, |buffer, cx| buffer.edit_with_autoindent(edits, cx));
1468 }
1469
1470 fn select(&mut self, Select(phase): &Select, cx: &mut ViewContext<Self>) {
1471 self.hide_context_menu(cx);
1472
1473 match phase {
1474 SelectPhase::Begin {
1475 position,
1476 add,
1477 click_count,
1478 } => self.begin_selection(*position, *add, *click_count, cx),
1479 SelectPhase::BeginColumnar {
1480 position,
1481 overshoot,
1482 } => self.begin_columnar_selection(*position, *overshoot, cx),
1483 SelectPhase::Extend {
1484 position,
1485 click_count,
1486 } => self.extend_selection(*position, *click_count, cx),
1487 SelectPhase::Update {
1488 position,
1489 overshoot,
1490 scroll_position,
1491 } => self.update_selection(*position, *overshoot, *scroll_position, cx),
1492 SelectPhase::End => self.end_selection(cx),
1493 }
1494 }
1495
1496 fn extend_selection(
1497 &mut self,
1498 position: DisplayPoint,
1499 click_count: usize,
1500 cx: &mut ViewContext<Self>,
1501 ) {
1502 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1503 let tail = self.selections.newest::<usize>(cx).tail();
1504 self.begin_selection(position, false, click_count, cx);
1505
1506 let position = position.to_offset(&display_map, Bias::Left);
1507 let tail_anchor = display_map.buffer_snapshot.anchor_before(tail);
1508
1509 let mut pending_selection = self
1510 .selections
1511 .pending_anchor()
1512 .expect("extend_selection not called with pending selection");
1513 if position >= tail {
1514 pending_selection.start = tail_anchor.clone();
1515 } else {
1516 pending_selection.end = tail_anchor.clone();
1517 pending_selection.reversed = true;
1518 }
1519
1520 let mut pending_mode = self.selections.pending_mode().unwrap();
1521 match &mut pending_mode {
1522 SelectMode::Word(range) | SelectMode::Line(range) => {
1523 *range = tail_anchor.clone()..tail_anchor
1524 }
1525 _ => {}
1526 }
1527
1528 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
1529 s.set_pending(pending_selection, pending_mode)
1530 });
1531 }
1532
1533 fn begin_selection(
1534 &mut self,
1535 position: DisplayPoint,
1536 add: bool,
1537 click_count: usize,
1538 cx: &mut ViewContext<Self>,
1539 ) {
1540 if !self.focused {
1541 cx.focus_self();
1542 cx.emit(Event::Activate);
1543 }
1544
1545 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1546 let buffer = &display_map.buffer_snapshot;
1547 let newest_selection = self.selections.newest_anchor().clone();
1548 let position = display_map.clip_point(position, Bias::Left);
1549
1550 let start;
1551 let end;
1552 let mode;
1553 match click_count {
1554 1 => {
1555 start = buffer.anchor_before(position.to_point(&display_map));
1556 end = start.clone();
1557 mode = SelectMode::Character;
1558 }
1559 2 => {
1560 let range = movement::surrounding_word(&display_map, position);
1561 start = buffer.anchor_before(range.start.to_point(&display_map));
1562 end = buffer.anchor_before(range.end.to_point(&display_map));
1563 mode = SelectMode::Word(start.clone()..end.clone());
1564 }
1565 3 => {
1566 let position = display_map
1567 .clip_point(position, Bias::Left)
1568 .to_point(&display_map);
1569 let line_start = display_map.prev_line_boundary(position).0;
1570 let next_line_start = buffer.clip_point(
1571 display_map.next_line_boundary(position).0 + Point::new(1, 0),
1572 Bias::Left,
1573 );
1574 start = buffer.anchor_before(line_start);
1575 end = buffer.anchor_before(next_line_start);
1576 mode = SelectMode::Line(start.clone()..end.clone());
1577 }
1578 _ => {
1579 start = buffer.anchor_before(0);
1580 end = buffer.anchor_before(buffer.len());
1581 mode = SelectMode::All;
1582 }
1583 }
1584
1585 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
1586 if !add {
1587 s.clear_disjoint();
1588 } else if click_count > 1 {
1589 s.delete(newest_selection.id)
1590 }
1591
1592 s.set_pending_range(start..end, mode);
1593 });
1594 }
1595
1596 fn begin_columnar_selection(
1597 &mut self,
1598 position: DisplayPoint,
1599 overshoot: u32,
1600 cx: &mut ViewContext<Self>,
1601 ) {
1602 if !self.focused {
1603 cx.focus_self();
1604 cx.emit(Event::Activate);
1605 }
1606
1607 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1608 let tail = self.selections.newest::<Point>(cx).tail();
1609 self.columnar_selection_tail = Some(display_map.buffer_snapshot.anchor_before(tail));
1610
1611 self.select_columns(
1612 tail.to_display_point(&display_map),
1613 position,
1614 overshoot,
1615 &display_map,
1616 cx,
1617 );
1618 }
1619
1620 fn update_selection(
1621 &mut self,
1622 position: DisplayPoint,
1623 overshoot: u32,
1624 scroll_position: Vector2F,
1625 cx: &mut ViewContext<Self>,
1626 ) {
1627 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
1628
1629 if let Some(tail) = self.columnar_selection_tail.as_ref() {
1630 let tail = tail.to_display_point(&display_map);
1631 self.select_columns(tail, position, overshoot, &display_map, cx);
1632 } else if let Some(mut pending) = self.selections.pending_anchor().clone() {
1633 let buffer = self.buffer.read(cx).snapshot(cx);
1634 let head;
1635 let tail;
1636 let mode = self.selections.pending_mode().unwrap();
1637 match &mode {
1638 SelectMode::Character => {
1639 head = position.to_point(&display_map);
1640 tail = pending.tail().to_point(&buffer);
1641 }
1642 SelectMode::Word(original_range) => {
1643 let original_display_range = original_range.start.to_display_point(&display_map)
1644 ..original_range.end.to_display_point(&display_map);
1645 let original_buffer_range = original_display_range.start.to_point(&display_map)
1646 ..original_display_range.end.to_point(&display_map);
1647 if movement::is_inside_word(&display_map, position)
1648 || original_display_range.contains(&position)
1649 {
1650 let word_range = movement::surrounding_word(&display_map, position);
1651 if word_range.start < original_display_range.start {
1652 head = word_range.start.to_point(&display_map);
1653 } else {
1654 head = word_range.end.to_point(&display_map);
1655 }
1656 } else {
1657 head = position.to_point(&display_map);
1658 }
1659
1660 if head <= original_buffer_range.start {
1661 tail = original_buffer_range.end;
1662 } else {
1663 tail = original_buffer_range.start;
1664 }
1665 }
1666 SelectMode::Line(original_range) => {
1667 let original_range = original_range.to_point(&display_map.buffer_snapshot);
1668
1669 let position = display_map
1670 .clip_point(position, Bias::Left)
1671 .to_point(&display_map);
1672 let line_start = display_map.prev_line_boundary(position).0;
1673 let next_line_start = buffer.clip_point(
1674 display_map.next_line_boundary(position).0 + Point::new(1, 0),
1675 Bias::Left,
1676 );
1677
1678 if line_start < original_range.start {
1679 head = line_start
1680 } else {
1681 head = next_line_start
1682 }
1683
1684 if head <= original_range.start {
1685 tail = original_range.end;
1686 } else {
1687 tail = original_range.start;
1688 }
1689 }
1690 SelectMode::All => {
1691 return;
1692 }
1693 };
1694
1695 if head < tail {
1696 pending.start = buffer.anchor_before(head);
1697 pending.end = buffer.anchor_before(tail);
1698 pending.reversed = true;
1699 } else {
1700 pending.start = buffer.anchor_before(tail);
1701 pending.end = buffer.anchor_before(head);
1702 pending.reversed = false;
1703 }
1704
1705 self.change_selections(None, cx, |s| {
1706 s.set_pending(pending, mode);
1707 });
1708 } else {
1709 log::error!("update_selection dispatched with no pending selection");
1710 return;
1711 }
1712
1713 self.set_scroll_position(scroll_position, cx);
1714 cx.notify();
1715 }
1716
1717 fn end_selection(&mut self, cx: &mut ViewContext<Self>) {
1718 self.columnar_selection_tail.take();
1719 if self.selections.pending_anchor().is_some() {
1720 let selections = self.selections.all::<usize>(cx);
1721 self.change_selections(None, cx, |s| {
1722 s.select(selections);
1723 s.clear_pending();
1724 });
1725 }
1726 }
1727
1728 fn select_columns(
1729 &mut self,
1730 tail: DisplayPoint,
1731 head: DisplayPoint,
1732 overshoot: u32,
1733 display_map: &DisplaySnapshot,
1734 cx: &mut ViewContext<Self>,
1735 ) {
1736 let start_row = cmp::min(tail.row(), head.row());
1737 let end_row = cmp::max(tail.row(), head.row());
1738 let start_column = cmp::min(tail.column(), head.column() + overshoot);
1739 let end_column = cmp::max(tail.column(), head.column() + overshoot);
1740 let reversed = start_column < tail.column();
1741
1742 let selection_ranges = (start_row..=end_row)
1743 .filter_map(|row| {
1744 if start_column <= display_map.line_len(row) && !display_map.is_block_line(row) {
1745 let start = display_map
1746 .clip_point(DisplayPoint::new(row, start_column), Bias::Left)
1747 .to_point(&display_map);
1748 let end = display_map
1749 .clip_point(DisplayPoint::new(row, end_column), Bias::Right)
1750 .to_point(&display_map);
1751 if reversed {
1752 Some(end..start)
1753 } else {
1754 Some(start..end)
1755 }
1756 } else {
1757 None
1758 }
1759 })
1760 .collect::<Vec<_>>();
1761
1762 self.change_selections(None, cx, |s| {
1763 s.select_ranges(selection_ranges);
1764 });
1765 cx.notify();
1766 }
1767
1768 pub fn is_selecting(&self) -> bool {
1769 self.selections.pending_anchor().is_some() || self.columnar_selection_tail.is_some()
1770 }
1771
1772 pub fn cancel(&mut self, _: &Cancel, cx: &mut ViewContext<Self>) {
1773 if self.take_rename(false, cx).is_some() {
1774 return;
1775 }
1776
1777 if hide_hover(self, cx) {
1778 return;
1779 }
1780
1781 if self.hide_context_menu(cx).is_some() {
1782 return;
1783 }
1784
1785 if self.snippet_stack.pop().is_some() {
1786 return;
1787 }
1788
1789 if self.mode == EditorMode::Full {
1790 if self.active_diagnostics.is_some() {
1791 self.dismiss_diagnostics(cx);
1792 return;
1793 }
1794
1795 if self.change_selections(Some(Autoscroll::Fit), cx, |s| s.try_cancel()) {
1796 return;
1797 }
1798 }
1799
1800 cx.propagate_action();
1801 }
1802
1803 pub fn handle_input(&mut self, action: &Input, cx: &mut ViewContext<Self>) {
1804 if !self.input_enabled {
1805 cx.propagate_action();
1806 return;
1807 }
1808
1809 let text = action.0.as_ref();
1810 if !self.skip_autoclose_end(text, cx) {
1811 self.transact(cx, |this, cx| {
1812 if !this.surround_with_bracket_pair(text, cx) {
1813 this.insert(text, cx);
1814 this.autoclose_bracket_pairs(cx);
1815 }
1816 });
1817 self.trigger_completion_on_input(text, cx);
1818 }
1819 }
1820
1821 pub fn newline(&mut self, _: &Newline, cx: &mut ViewContext<Self>) {
1822 self.transact(cx, |this, cx| {
1823 let (edits, selection_fixup_info): (Vec<_>, Vec<_>) = {
1824 let selections = this.selections.all::<usize>(cx);
1825
1826 let buffer = this.buffer.read(cx).snapshot(cx);
1827 selections
1828 .iter()
1829 .map(|selection| {
1830 let start_point = selection.start.to_point(&buffer);
1831 let mut indent = buffer.indent_size_for_line(start_point.row);
1832 indent.len = cmp::min(indent.len, start_point.column);
1833 let start = selection.start;
1834 let end = selection.end;
1835
1836 let mut insert_extra_newline = false;
1837 if let Some(language) = buffer.language() {
1838 let leading_whitespace_len = buffer
1839 .reversed_chars_at(start)
1840 .take_while(|c| c.is_whitespace() && *c != '\n')
1841 .map(|c| c.len_utf8())
1842 .sum::<usize>();
1843
1844 let trailing_whitespace_len = buffer
1845 .chars_at(end)
1846 .take_while(|c| c.is_whitespace() && *c != '\n')
1847 .map(|c| c.len_utf8())
1848 .sum::<usize>();
1849
1850 insert_extra_newline = language.brackets().iter().any(|pair| {
1851 let pair_start = pair.start.trim_end();
1852 let pair_end = pair.end.trim_start();
1853
1854 pair.newline
1855 && buffer
1856 .contains_str_at(end + trailing_whitespace_len, pair_end)
1857 && buffer.contains_str_at(
1858 (start - leading_whitespace_len)
1859 .saturating_sub(pair_start.len()),
1860 pair_start,
1861 )
1862 });
1863 }
1864
1865 let mut new_text = String::with_capacity(1 + indent.len as usize);
1866 new_text.push('\n');
1867 new_text.extend(indent.chars());
1868 if insert_extra_newline {
1869 new_text = new_text.repeat(2);
1870 }
1871
1872 let anchor = buffer.anchor_after(end);
1873 let new_selection = selection.map(|_| anchor.clone());
1874 (
1875 (start..end, new_text),
1876 (insert_extra_newline, new_selection),
1877 )
1878 })
1879 .unzip()
1880 };
1881
1882 this.buffer.update(cx, |buffer, cx| {
1883 buffer.edit_with_autoindent(edits, cx);
1884 });
1885 let buffer = this.buffer.read(cx).snapshot(cx);
1886 let new_selections = selection_fixup_info
1887 .into_iter()
1888 .map(|(extra_newline_inserted, new_selection)| {
1889 let mut cursor = new_selection.end.to_point(&buffer);
1890 if extra_newline_inserted {
1891 cursor.row -= 1;
1892 cursor.column = buffer.line_len(cursor.row);
1893 }
1894 new_selection.map(|_| cursor.clone())
1895 })
1896 .collect();
1897
1898 this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(new_selections));
1899 });
1900 }
1901
1902 pub fn insert(&mut self, text: &str, cx: &mut ViewContext<Self>) {
1903 let text: Arc<str> = text.into();
1904 self.transact(cx, |this, cx| {
1905 let old_selections = this.selections.all_adjusted(cx);
1906 let selection_anchors = this.buffer.update(cx, |buffer, cx| {
1907 let anchors = {
1908 let snapshot = buffer.read(cx);
1909 old_selections
1910 .iter()
1911 .map(|s| {
1912 let anchor = snapshot.anchor_after(s.end);
1913 s.map(|_| anchor.clone())
1914 })
1915 .collect::<Vec<_>>()
1916 };
1917 buffer.edit_with_autoindent(
1918 old_selections
1919 .iter()
1920 .map(|s| (s.start..s.end, text.clone())),
1921 cx,
1922 );
1923 anchors
1924 });
1925
1926 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
1927 s.select_anchors(selection_anchors);
1928 })
1929 });
1930 }
1931
1932 fn trigger_completion_on_input(&mut self, text: &str, cx: &mut ViewContext<Self>) {
1933 let selection = self.selections.newest_anchor();
1934 if self
1935 .buffer
1936 .read(cx)
1937 .is_completion_trigger(selection.head(), text, cx)
1938 {
1939 self.show_completions(&ShowCompletions, cx);
1940 } else {
1941 self.hide_context_menu(cx);
1942 }
1943 }
1944
1945 fn surround_with_bracket_pair(&mut self, text: &str, cx: &mut ViewContext<Self>) -> bool {
1946 let snapshot = self.buffer.read(cx).snapshot(cx);
1947 if let Some(pair) = snapshot
1948 .language()
1949 .and_then(|language| language.brackets().iter().find(|b| b.start == text))
1950 .cloned()
1951 {
1952 if self
1953 .selections
1954 .all::<usize>(cx)
1955 .iter()
1956 .any(|selection| selection.is_empty())
1957 {
1958 return false;
1959 }
1960
1961 let mut selections = self.selections.disjoint_anchors().to_vec();
1962 for selection in &mut selections {
1963 selection.end = selection.end.bias_left(&snapshot);
1964 }
1965 drop(snapshot);
1966
1967 self.buffer.update(cx, |buffer, cx| {
1968 let pair_start: Arc<str> = pair.start.clone().into();
1969 let pair_end: Arc<str> = pair.end.clone().into();
1970 buffer.edit(
1971 selections
1972 .iter()
1973 .map(|s| (s.start.clone()..s.start.clone(), pair_start.clone()))
1974 .chain(
1975 selections
1976 .iter()
1977 .map(|s| (s.end.clone()..s.end.clone(), pair_end.clone())),
1978 ),
1979 cx,
1980 );
1981 });
1982
1983 let snapshot = self.buffer.read(cx).read(cx);
1984 for selection in &mut selections {
1985 selection.end = selection.end.bias_right(&snapshot);
1986 }
1987 drop(snapshot);
1988
1989 self.change_selections(None, cx, |s| s.select_anchors(selections));
1990 true
1991 } else {
1992 false
1993 }
1994 }
1995
1996 fn autoclose_bracket_pairs(&mut self, cx: &mut ViewContext<Self>) {
1997 let selections = self.selections.all::<usize>(cx);
1998 let mut bracket_pair_state = None;
1999 let mut new_selections = None;
2000 self.buffer.update(cx, |buffer, cx| {
2001 let mut snapshot = buffer.snapshot(cx);
2002 let left_biased_selections = selections
2003 .iter()
2004 .map(|selection| selection.map(|p| snapshot.anchor_before(p)))
2005 .collect::<Vec<_>>();
2006
2007 let autoclose_pair = snapshot.language().and_then(|language| {
2008 let first_selection_start = selections.first().unwrap().start;
2009 let pair = language.brackets().iter().find(|pair| {
2010 snapshot.contains_str_at(
2011 first_selection_start.saturating_sub(pair.start.len()),
2012 &pair.start,
2013 )
2014 });
2015 pair.and_then(|pair| {
2016 let should_autoclose = selections.iter().all(|selection| {
2017 // Ensure all selections are parked at the end of a pair start.
2018 if snapshot.contains_str_at(
2019 selection.start.saturating_sub(pair.start.len()),
2020 &pair.start,
2021 ) {
2022 snapshot
2023 .chars_at(selection.start)
2024 .next()
2025 .map_or(true, |c| language.should_autoclose_before(c))
2026 } else {
2027 false
2028 }
2029 });
2030
2031 if should_autoclose {
2032 Some(pair.clone())
2033 } else {
2034 None
2035 }
2036 })
2037 });
2038
2039 if let Some(pair) = autoclose_pair {
2040 let selection_ranges = selections
2041 .iter()
2042 .map(|selection| {
2043 let start = selection.start.to_offset(&snapshot);
2044 start..start
2045 })
2046 .collect::<SmallVec<[_; 32]>>();
2047
2048 let pair_end: Arc<str> = pair.end.clone().into();
2049 buffer.edit(
2050 selection_ranges
2051 .iter()
2052 .map(|range| (range.clone(), pair_end.clone())),
2053 cx,
2054 );
2055 snapshot = buffer.snapshot(cx);
2056
2057 new_selections = Some(
2058 resolve_multiple::<usize, _>(left_biased_selections.iter(), &snapshot)
2059 .collect::<Vec<_>>(),
2060 );
2061
2062 if pair.end.len() == 1 {
2063 let mut delta = 0;
2064 bracket_pair_state = Some(BracketPairState {
2065 ranges: selections
2066 .iter()
2067 .map(move |selection| {
2068 let offset = selection.start + delta;
2069 delta += 1;
2070 snapshot.anchor_before(offset)..snapshot.anchor_after(offset)
2071 })
2072 .collect(),
2073 pair,
2074 });
2075 }
2076 }
2077 });
2078
2079 if let Some(new_selections) = new_selections {
2080 self.change_selections(None, cx, |s| {
2081 s.select(new_selections);
2082 });
2083 }
2084 if let Some(bracket_pair_state) = bracket_pair_state {
2085 self.autoclose_stack.push(bracket_pair_state);
2086 }
2087 }
2088
2089 fn skip_autoclose_end(&mut self, text: &str, cx: &mut ViewContext<Self>) -> bool {
2090 let buffer = self.buffer.read(cx).snapshot(cx);
2091 let old_selections = self.selections.all::<usize>(cx);
2092 let autoclose_pair = if let Some(autoclose_pair) = self.autoclose_stack.last() {
2093 autoclose_pair
2094 } else {
2095 return false;
2096 };
2097 if text != autoclose_pair.pair.end {
2098 return false;
2099 }
2100
2101 debug_assert_eq!(old_selections.len(), autoclose_pair.ranges.len());
2102
2103 if old_selections
2104 .iter()
2105 .zip(autoclose_pair.ranges.iter().map(|r| r.to_offset(&buffer)))
2106 .all(|(selection, autoclose_range)| {
2107 let autoclose_range_end = autoclose_range.end.to_offset(&buffer);
2108 selection.is_empty() && selection.start == autoclose_range_end
2109 })
2110 {
2111 let new_selections = old_selections
2112 .into_iter()
2113 .map(|selection| {
2114 let cursor = selection.start + 1;
2115 Selection {
2116 id: selection.id,
2117 start: cursor,
2118 end: cursor,
2119 reversed: false,
2120 goal: SelectionGoal::None,
2121 }
2122 })
2123 .collect();
2124 self.autoclose_stack.pop();
2125 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
2126 s.select(new_selections);
2127 });
2128 true
2129 } else {
2130 false
2131 }
2132 }
2133
2134 fn completion_query(buffer: &MultiBufferSnapshot, position: impl ToOffset) -> Option<String> {
2135 let offset = position.to_offset(buffer);
2136 let (word_range, kind) = buffer.surrounding_word(offset);
2137 if offset > word_range.start && kind == Some(CharKind::Word) {
2138 Some(
2139 buffer
2140 .text_for_range(word_range.start..offset)
2141 .collect::<String>(),
2142 )
2143 } else {
2144 None
2145 }
2146 }
2147
2148 fn show_completions(&mut self, _: &ShowCompletions, cx: &mut ViewContext<Self>) {
2149 if self.pending_rename.is_some() {
2150 return;
2151 }
2152
2153 let project = if let Some(project) = self.project.clone() {
2154 project
2155 } else {
2156 return;
2157 };
2158
2159 let position = self.selections.newest_anchor().head();
2160 let (buffer, buffer_position) = if let Some(output) = self
2161 .buffer
2162 .read(cx)
2163 .text_anchor_for_position(position.clone(), cx)
2164 {
2165 output
2166 } else {
2167 return;
2168 };
2169
2170 let query = Self::completion_query(&self.buffer.read(cx).read(cx), position.clone());
2171 let completions = project.update(cx, |project, cx| {
2172 project.completions(&buffer, buffer_position.clone(), cx)
2173 });
2174
2175 let id = post_inc(&mut self.next_completion_id);
2176 let task = cx.spawn_weak(|this, mut cx| {
2177 async move {
2178 let completions = completions.await?;
2179 if completions.is_empty() {
2180 return Ok(());
2181 }
2182
2183 let mut menu = CompletionsMenu {
2184 id,
2185 initial_position: position,
2186 match_candidates: completions
2187 .iter()
2188 .enumerate()
2189 .map(|(id, completion)| {
2190 StringMatchCandidate::new(
2191 id,
2192 completion.label.text[completion.label.filter_range.clone()].into(),
2193 )
2194 })
2195 .collect(),
2196 buffer,
2197 completions: completions.into(),
2198 matches: Vec::new().into(),
2199 selected_item: 0,
2200 list: Default::default(),
2201 };
2202
2203 menu.filter(query.as_deref(), cx.background()).await;
2204
2205 if let Some(this) = this.upgrade(&cx) {
2206 this.update(&mut cx, |this, cx| {
2207 match this.context_menu.as_ref() {
2208 None => {}
2209 Some(ContextMenu::Completions(prev_menu)) => {
2210 if prev_menu.id > menu.id {
2211 return;
2212 }
2213 }
2214 _ => return,
2215 }
2216
2217 this.completion_tasks.retain(|(id, _)| *id > menu.id);
2218 if this.focused {
2219 this.show_context_menu(ContextMenu::Completions(menu), cx);
2220 }
2221
2222 cx.notify();
2223 });
2224 }
2225 Ok::<_, anyhow::Error>(())
2226 }
2227 .log_err()
2228 });
2229 self.completion_tasks.push((id, task));
2230 }
2231
2232 pub fn confirm_completion(
2233 &mut self,
2234 action: &ConfirmCompletion,
2235 cx: &mut ViewContext<Self>,
2236 ) -> Option<Task<Result<()>>> {
2237 use language::ToOffset as _;
2238
2239 let completions_menu = if let ContextMenu::Completions(menu) = self.hide_context_menu(cx)? {
2240 menu
2241 } else {
2242 return None;
2243 };
2244
2245 let mat = completions_menu
2246 .matches
2247 .get(action.item_ix.unwrap_or(completions_menu.selected_item))?;
2248 let buffer_handle = completions_menu.buffer;
2249 let completion = completions_menu.completions.get(mat.candidate_id)?;
2250
2251 let snippet;
2252 let text;
2253 if completion.is_snippet() {
2254 snippet = Some(Snippet::parse(&completion.new_text).log_err()?);
2255 text = snippet.as_ref().unwrap().text.clone();
2256 } else {
2257 snippet = None;
2258 text = completion.new_text.clone();
2259 };
2260 let selections = self.selections.all::<usize>(cx);
2261 let buffer = buffer_handle.read(cx);
2262 let old_range = completion.old_range.to_offset(&buffer);
2263 let old_text = buffer.text_for_range(old_range.clone()).collect::<String>();
2264
2265 let newest_selection = self.selections.newest_anchor();
2266 if newest_selection.start.buffer_id != Some(buffer_handle.id()) {
2267 return None;
2268 }
2269
2270 let lookbehind = newest_selection
2271 .start
2272 .text_anchor
2273 .to_offset(buffer)
2274 .saturating_sub(old_range.start);
2275 let lookahead = old_range
2276 .end
2277 .saturating_sub(newest_selection.end.text_anchor.to_offset(buffer));
2278 let mut common_prefix_len = old_text
2279 .bytes()
2280 .zip(text.bytes())
2281 .take_while(|(a, b)| a == b)
2282 .count();
2283
2284 let snapshot = self.buffer.read(cx).snapshot(cx);
2285 let mut ranges = Vec::new();
2286 for selection in &selections {
2287 if snapshot.contains_str_at(selection.start.saturating_sub(lookbehind), &old_text) {
2288 let start = selection.start.saturating_sub(lookbehind);
2289 let end = selection.end + lookahead;
2290 ranges.push(start + common_prefix_len..end);
2291 } else {
2292 common_prefix_len = 0;
2293 ranges.clear();
2294 ranges.extend(selections.iter().map(|s| {
2295 if s.id == newest_selection.id {
2296 old_range.clone()
2297 } else {
2298 s.start..s.end
2299 }
2300 }));
2301 break;
2302 }
2303 }
2304 let text = &text[common_prefix_len..];
2305
2306 self.transact(cx, |this, cx| {
2307 if let Some(mut snippet) = snippet {
2308 snippet.text = text.to_string();
2309 for tabstop in snippet.tabstops.iter_mut().flatten() {
2310 tabstop.start -= common_prefix_len as isize;
2311 tabstop.end -= common_prefix_len as isize;
2312 }
2313
2314 this.insert_snippet(&ranges, snippet, cx).log_err();
2315 } else {
2316 this.buffer.update(cx, |buffer, cx| {
2317 buffer
2318 .edit_with_autoindent(ranges.iter().map(|range| (range.clone(), text)), cx);
2319 });
2320 }
2321 });
2322
2323 let project = self.project.clone()?;
2324 let apply_edits = project.update(cx, |project, cx| {
2325 project.apply_additional_edits_for_completion(
2326 buffer_handle,
2327 completion.clone(),
2328 true,
2329 cx,
2330 )
2331 });
2332 Some(cx.foreground().spawn(async move {
2333 apply_edits.await?;
2334 Ok(())
2335 }))
2336 }
2337
2338 pub fn toggle_code_actions(&mut self, action: &ToggleCodeActions, cx: &mut ViewContext<Self>) {
2339 if matches!(
2340 self.context_menu.as_ref(),
2341 Some(ContextMenu::CodeActions(_))
2342 ) {
2343 self.context_menu.take();
2344 cx.notify();
2345 return;
2346 }
2347
2348 let deployed_from_indicator = action.deployed_from_indicator;
2349 let mut task = self.code_actions_task.take();
2350 cx.spawn_weak(|this, mut cx| async move {
2351 while let Some(prev_task) = task {
2352 prev_task.await;
2353 task = this
2354 .upgrade(&cx)
2355 .and_then(|this| this.update(&mut cx, |this, _| this.code_actions_task.take()));
2356 }
2357
2358 if let Some(this) = this.upgrade(&cx) {
2359 this.update(&mut cx, |this, cx| {
2360 if this.focused {
2361 if let Some((buffer, actions)) = this.available_code_actions.clone() {
2362 this.show_context_menu(
2363 ContextMenu::CodeActions(CodeActionsMenu {
2364 buffer,
2365 actions,
2366 selected_item: Default::default(),
2367 list: Default::default(),
2368 deployed_from_indicator,
2369 }),
2370 cx,
2371 );
2372 }
2373 }
2374 })
2375 }
2376 Ok::<_, anyhow::Error>(())
2377 })
2378 .detach_and_log_err(cx);
2379 }
2380
2381 pub fn confirm_code_action(
2382 workspace: &mut Workspace,
2383 action: &ConfirmCodeAction,
2384 cx: &mut ViewContext<Workspace>,
2385 ) -> Option<Task<Result<()>>> {
2386 let editor = workspace.active_item(cx)?.act_as::<Editor>(cx)?;
2387 let actions_menu = if let ContextMenu::CodeActions(menu) =
2388 editor.update(cx, |editor, cx| editor.hide_context_menu(cx))?
2389 {
2390 menu
2391 } else {
2392 return None;
2393 };
2394 let action_ix = action.item_ix.unwrap_or(actions_menu.selected_item);
2395 let action = actions_menu.actions.get(action_ix)?.clone();
2396 let title = action.lsp_action.title.clone();
2397 let buffer = actions_menu.buffer;
2398
2399 let apply_code_actions = workspace.project().clone().update(cx, |project, cx| {
2400 project.apply_code_action(buffer, action, true, cx)
2401 });
2402 Some(cx.spawn(|workspace, cx| async move {
2403 let project_transaction = apply_code_actions.await?;
2404 Self::open_project_transaction(editor, workspace, project_transaction, title, cx).await
2405 }))
2406 }
2407
2408 async fn open_project_transaction(
2409 this: ViewHandle<Editor>,
2410 workspace: ViewHandle<Workspace>,
2411 transaction: ProjectTransaction,
2412 title: String,
2413 mut cx: AsyncAppContext,
2414 ) -> Result<()> {
2415 let replica_id = this.read_with(&cx, |this, cx| this.replica_id(cx));
2416
2417 let mut entries = transaction.0.into_iter().collect::<Vec<_>>();
2418 entries.sort_unstable_by_key(|(buffer, _)| {
2419 buffer.read_with(&cx, |buffer, _| buffer.file().map(|f| f.path().clone()))
2420 });
2421
2422 // If the project transaction's edits are all contained within this editor, then
2423 // avoid opening a new editor to display them.
2424
2425 if let Some((buffer, transaction)) = entries.first() {
2426 if entries.len() == 1 {
2427 let excerpt = this.read_with(&cx, |editor, cx| {
2428 editor
2429 .buffer()
2430 .read(cx)
2431 .excerpt_containing(editor.selections.newest_anchor().head(), cx)
2432 });
2433 if let Some((_, excerpted_buffer, excerpt_range)) = excerpt {
2434 if excerpted_buffer == *buffer {
2435 let snapshot = buffer.read_with(&cx, |buffer, _| buffer.snapshot());
2436 let excerpt_range = excerpt_range.to_offset(&snapshot);
2437 if snapshot
2438 .edited_ranges_for_transaction(transaction)
2439 .all(|range| {
2440 excerpt_range.start <= range.start && excerpt_range.end >= range.end
2441 })
2442 {
2443 return Ok(());
2444 }
2445 }
2446 }
2447 }
2448 } else {
2449 return Ok(());
2450 }
2451
2452 let mut ranges_to_highlight = Vec::new();
2453 let excerpt_buffer = cx.add_model(|cx| {
2454 let mut multibuffer = MultiBuffer::new(replica_id).with_title(title);
2455 for (buffer, transaction) in &entries {
2456 let snapshot = buffer.read(cx).snapshot();
2457 ranges_to_highlight.extend(
2458 multibuffer.push_excerpts_with_context_lines(
2459 buffer.clone(),
2460 snapshot
2461 .edited_ranges_for_transaction::<usize>(transaction)
2462 .collect(),
2463 1,
2464 cx,
2465 ),
2466 );
2467 }
2468 multibuffer.push_transaction(entries.iter().map(|(b, t)| (b, t)));
2469 multibuffer
2470 });
2471
2472 workspace.update(&mut cx, |workspace, cx| {
2473 let project = workspace.project().clone();
2474 let editor =
2475 cx.add_view(|cx| Editor::for_multibuffer(excerpt_buffer, Some(project), cx));
2476 workspace.add_item(Box::new(editor.clone()), cx);
2477 editor.update(cx, |editor, cx| {
2478 editor.highlight_background::<Self>(
2479 ranges_to_highlight,
2480 |theme| theme.editor.highlighted_line_background,
2481 cx,
2482 );
2483 });
2484 });
2485
2486 Ok(())
2487 }
2488
2489 fn refresh_code_actions(&mut self, cx: &mut ViewContext<Self>) -> Option<()> {
2490 let project = self.project.as_ref()?;
2491 let buffer = self.buffer.read(cx);
2492 let newest_selection = self.selections.newest_anchor().clone();
2493 let (start_buffer, start) = buffer.text_anchor_for_position(newest_selection.start, cx)?;
2494 let (end_buffer, end) = buffer.text_anchor_for_position(newest_selection.end, cx)?;
2495 if start_buffer != end_buffer {
2496 return None;
2497 }
2498
2499 let actions = project.update(cx, |project, cx| {
2500 project.code_actions(&start_buffer, start..end, cx)
2501 });
2502 self.code_actions_task = Some(cx.spawn_weak(|this, mut cx| async move {
2503 let actions = actions.await;
2504 if let Some(this) = this.upgrade(&cx) {
2505 this.update(&mut cx, |this, cx| {
2506 this.available_code_actions = actions.log_err().and_then(|actions| {
2507 if actions.is_empty() {
2508 None
2509 } else {
2510 Some((start_buffer, actions.into()))
2511 }
2512 });
2513 cx.notify();
2514 })
2515 }
2516 }));
2517 None
2518 }
2519
2520 fn refresh_document_highlights(&mut self, cx: &mut ViewContext<Self>) -> Option<()> {
2521 if self.pending_rename.is_some() {
2522 return None;
2523 }
2524
2525 let project = self.project.as_ref()?;
2526 let buffer = self.buffer.read(cx);
2527 let newest_selection = self.selections.newest_anchor().clone();
2528 let cursor_position = newest_selection.head();
2529 let (cursor_buffer, cursor_buffer_position) =
2530 buffer.text_anchor_for_position(cursor_position.clone(), cx)?;
2531 let (tail_buffer, _) = buffer.text_anchor_for_position(newest_selection.tail(), cx)?;
2532 if cursor_buffer != tail_buffer {
2533 return None;
2534 }
2535
2536 let highlights = project.update(cx, |project, cx| {
2537 project.document_highlights(&cursor_buffer, cursor_buffer_position, cx)
2538 });
2539
2540 self.document_highlights_task = Some(cx.spawn_weak(|this, mut cx| async move {
2541 let highlights = highlights.log_err().await;
2542 if let Some((this, highlights)) = this.upgrade(&cx).zip(highlights) {
2543 this.update(&mut cx, |this, cx| {
2544 if this.pending_rename.is_some() {
2545 return;
2546 }
2547
2548 let buffer_id = cursor_position.buffer_id;
2549 let buffer = this.buffer.read(cx);
2550 if !buffer
2551 .text_anchor_for_position(cursor_position, cx)
2552 .map_or(false, |(buffer, _)| buffer == cursor_buffer)
2553 {
2554 return;
2555 }
2556
2557 let cursor_buffer_snapshot = cursor_buffer.read(cx);
2558 let mut write_ranges = Vec::new();
2559 let mut read_ranges = Vec::new();
2560 for highlight in highlights {
2561 for (excerpt_id, excerpt_range) in
2562 buffer.excerpts_for_buffer(&cursor_buffer, cx)
2563 {
2564 let start = highlight
2565 .range
2566 .start
2567 .max(&excerpt_range.context.start, cursor_buffer_snapshot);
2568 let end = highlight
2569 .range
2570 .end
2571 .min(&excerpt_range.context.end, cursor_buffer_snapshot);
2572 if start.cmp(&end, cursor_buffer_snapshot).is_ge() {
2573 continue;
2574 }
2575
2576 let range = Anchor {
2577 buffer_id,
2578 excerpt_id: excerpt_id.clone(),
2579 text_anchor: start,
2580 }..Anchor {
2581 buffer_id,
2582 excerpt_id,
2583 text_anchor: end,
2584 };
2585 if highlight.kind == lsp::DocumentHighlightKind::WRITE {
2586 write_ranges.push(range);
2587 } else {
2588 read_ranges.push(range);
2589 }
2590 }
2591 }
2592
2593 this.highlight_background::<DocumentHighlightRead>(
2594 read_ranges,
2595 |theme| theme.editor.document_highlight_read_background,
2596 cx,
2597 );
2598 this.highlight_background::<DocumentHighlightWrite>(
2599 write_ranges,
2600 |theme| theme.editor.document_highlight_write_background,
2601 cx,
2602 );
2603 cx.notify();
2604 });
2605 }
2606 }));
2607 None
2608 }
2609
2610 pub fn render_code_actions_indicator(
2611 &self,
2612 style: &EditorStyle,
2613 cx: &mut RenderContext<Self>,
2614 ) -> Option<ElementBox> {
2615 if self.available_code_actions.is_some() {
2616 enum Tag {}
2617 Some(
2618 MouseEventHandler::new::<Tag, _, _>(0, cx, |_, _| {
2619 Svg::new("icons/zap.svg")
2620 .with_color(style.code_actions_indicator)
2621 .boxed()
2622 })
2623 .with_cursor_style(CursorStyle::PointingHand)
2624 .with_padding(Padding::uniform(3.))
2625 .on_mouse_down(|_, cx| {
2626 cx.dispatch_action(ToggleCodeActions {
2627 deployed_from_indicator: true,
2628 });
2629 })
2630 .boxed(),
2631 )
2632 } else {
2633 None
2634 }
2635 }
2636
2637 pub fn context_menu_visible(&self) -> bool {
2638 self.context_menu
2639 .as_ref()
2640 .map_or(false, |menu| menu.visible())
2641 }
2642
2643 pub fn render_context_menu(
2644 &self,
2645 cursor_position: DisplayPoint,
2646 style: EditorStyle,
2647 cx: &mut RenderContext<Editor>,
2648 ) -> Option<(DisplayPoint, ElementBox)> {
2649 self.context_menu
2650 .as_ref()
2651 .map(|menu| menu.render(cursor_position, style, cx))
2652 }
2653
2654 fn show_context_menu(&mut self, menu: ContextMenu, cx: &mut ViewContext<Self>) {
2655 if !matches!(menu, ContextMenu::Completions(_)) {
2656 self.completion_tasks.clear();
2657 }
2658 self.context_menu = Some(menu);
2659 cx.notify();
2660 }
2661
2662 fn hide_context_menu(&mut self, cx: &mut ViewContext<Self>) -> Option<ContextMenu> {
2663 cx.notify();
2664 self.completion_tasks.clear();
2665 self.context_menu.take()
2666 }
2667
2668 pub fn insert_snippet(
2669 &mut self,
2670 insertion_ranges: &[Range<usize>],
2671 snippet: Snippet,
2672 cx: &mut ViewContext<Self>,
2673 ) -> Result<()> {
2674 let tabstops = self.buffer.update(cx, |buffer, cx| {
2675 let snippet_text: Arc<str> = snippet.text.clone().into();
2676 buffer.edit_with_autoindent(
2677 insertion_ranges
2678 .iter()
2679 .cloned()
2680 .map(|range| (range, snippet_text.clone())),
2681 cx,
2682 );
2683
2684 let snapshot = &*buffer.read(cx);
2685 let snippet = &snippet;
2686 snippet
2687 .tabstops
2688 .iter()
2689 .map(|tabstop| {
2690 let mut tabstop_ranges = tabstop
2691 .iter()
2692 .flat_map(|tabstop_range| {
2693 let mut delta = 0 as isize;
2694 insertion_ranges.iter().map(move |insertion_range| {
2695 let insertion_start = insertion_range.start as isize + delta;
2696 delta +=
2697 snippet.text.len() as isize - insertion_range.len() as isize;
2698
2699 let start = snapshot.anchor_before(
2700 (insertion_start + tabstop_range.start) as usize,
2701 );
2702 let end = snapshot
2703 .anchor_after((insertion_start + tabstop_range.end) as usize);
2704 start..end
2705 })
2706 })
2707 .collect::<Vec<_>>();
2708 tabstop_ranges.sort_unstable_by(|a, b| a.start.cmp(&b.start, snapshot));
2709 tabstop_ranges
2710 })
2711 .collect::<Vec<_>>()
2712 });
2713
2714 if let Some(tabstop) = tabstops.first() {
2715 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
2716 s.select_ranges(tabstop.iter().cloned());
2717 });
2718 self.snippet_stack.push(SnippetState {
2719 active_index: 0,
2720 ranges: tabstops,
2721 });
2722 }
2723
2724 Ok(())
2725 }
2726
2727 pub fn move_to_next_snippet_tabstop(&mut self, cx: &mut ViewContext<Self>) -> bool {
2728 self.move_to_snippet_tabstop(Bias::Right, cx)
2729 }
2730
2731 pub fn move_to_prev_snippet_tabstop(&mut self, cx: &mut ViewContext<Self>) -> bool {
2732 self.move_to_snippet_tabstop(Bias::Left, cx)
2733 }
2734
2735 pub fn move_to_snippet_tabstop(&mut self, bias: Bias, cx: &mut ViewContext<Self>) -> bool {
2736 if let Some(mut snippet) = self.snippet_stack.pop() {
2737 match bias {
2738 Bias::Left => {
2739 if snippet.active_index > 0 {
2740 snippet.active_index -= 1;
2741 } else {
2742 self.snippet_stack.push(snippet);
2743 return false;
2744 }
2745 }
2746 Bias::Right => {
2747 if snippet.active_index + 1 < snippet.ranges.len() {
2748 snippet.active_index += 1;
2749 } else {
2750 self.snippet_stack.push(snippet);
2751 return false;
2752 }
2753 }
2754 }
2755 if let Some(current_ranges) = snippet.ranges.get(snippet.active_index) {
2756 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
2757 s.select_anchor_ranges(current_ranges.into_iter().cloned())
2758 });
2759 // If snippet state is not at the last tabstop, push it back on the stack
2760 if snippet.active_index + 1 < snippet.ranges.len() {
2761 self.snippet_stack.push(snippet);
2762 }
2763 return true;
2764 }
2765 }
2766
2767 false
2768 }
2769
2770 pub fn clear(&mut self, cx: &mut ViewContext<Self>) {
2771 self.transact(cx, |this, cx| {
2772 this.select_all(&SelectAll, cx);
2773 this.insert("", cx);
2774 });
2775 }
2776
2777 pub fn backspace(&mut self, _: &Backspace, cx: &mut ViewContext<Self>) {
2778 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
2779 let mut selections = self.selections.all::<Point>(cx);
2780 if !self.selections.line_mode {
2781 for selection in &mut selections {
2782 if selection.is_empty() {
2783 let old_head = selection.head();
2784 let mut new_head =
2785 movement::left(&display_map, old_head.to_display_point(&display_map))
2786 .to_point(&display_map);
2787 if let Some((buffer, line_buffer_range)) = display_map
2788 .buffer_snapshot
2789 .buffer_line_for_row(old_head.row)
2790 {
2791 let indent_size = buffer.indent_size_for_line(line_buffer_range.start.row);
2792 let language_name = buffer.language().map(|language| language.name());
2793 let indent_len = match indent_size.kind {
2794 IndentKind::Space => {
2795 cx.global::<Settings>().tab_size(language_name.as_deref())
2796 }
2797 IndentKind::Tab => 1,
2798 };
2799 if old_head.column <= indent_size.len && old_head.column > 0 {
2800 new_head = cmp::min(
2801 new_head,
2802 Point::new(
2803 old_head.row,
2804 ((old_head.column - 1) / indent_len) * indent_len,
2805 ),
2806 );
2807 }
2808 }
2809
2810 selection.set_head(new_head, SelectionGoal::None);
2811 }
2812 }
2813 }
2814
2815 self.transact(cx, |this, cx| {
2816 this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(selections));
2817 this.insert("", cx);
2818 });
2819 }
2820
2821 pub fn delete(&mut self, _: &Delete, cx: &mut ViewContext<Self>) {
2822 self.transact(cx, |this, cx| {
2823 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
2824 let line_mode = s.line_mode;
2825 s.move_with(|map, selection| {
2826 if selection.is_empty() && !line_mode {
2827 let cursor = movement::right(map, selection.head());
2828 selection.set_head(cursor, SelectionGoal::None);
2829 }
2830 })
2831 });
2832 this.insert(&"", cx);
2833 });
2834 }
2835
2836 pub fn tab_prev(&mut self, _: &TabPrev, cx: &mut ViewContext<Self>) {
2837 if self.move_to_prev_snippet_tabstop(cx) {
2838 return;
2839 }
2840
2841 self.outdent(&Outdent, cx);
2842 }
2843
2844 pub fn tab(&mut self, _: &Tab, cx: &mut ViewContext<Self>) {
2845 if self.move_to_next_snippet_tabstop(cx) {
2846 return;
2847 }
2848
2849 let mut selections = self.selections.all_adjusted(cx);
2850 if selections.iter().all(|s| s.is_empty()) {
2851 self.transact(cx, |this, cx| {
2852 this.buffer.update(cx, |buffer, cx| {
2853 for selection in &mut selections {
2854 let language_name =
2855 buffer.language_at(selection.start, cx).map(|l| l.name());
2856 let settings = cx.global::<Settings>();
2857 let tab_size = if settings.hard_tabs(language_name.as_deref()) {
2858 IndentSize::tab()
2859 } else {
2860 let tab_size = settings.tab_size(language_name.as_deref());
2861 let char_column = buffer
2862 .read(cx)
2863 .text_for_range(Point::new(selection.start.row, 0)..selection.start)
2864 .flat_map(str::chars)
2865 .count();
2866 let chars_to_next_tab_stop = tab_size - (char_column as u32 % tab_size);
2867 IndentSize::spaces(chars_to_next_tab_stop)
2868 };
2869 buffer.edit(
2870 [(
2871 selection.start..selection.start,
2872 tab_size.chars().collect::<String>(),
2873 )],
2874 cx,
2875 );
2876 selection.start.column += tab_size.len;
2877 selection.end = selection.start;
2878 }
2879 });
2880 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
2881 s.select(selections);
2882 });
2883 });
2884 } else {
2885 self.indent(&Indent, cx);
2886 }
2887 }
2888
2889 pub fn indent(&mut self, _: &Indent, cx: &mut ViewContext<Self>) {
2890 let mut selections = self.selections.all::<Point>(cx);
2891 self.transact(cx, |this, cx| {
2892 let mut last_indent = None;
2893 this.buffer.update(cx, |buffer, cx| {
2894 let snapshot = buffer.snapshot(cx);
2895 for selection in &mut selections {
2896 let language_name = buffer.language_at(selection.start, cx).map(|l| l.name());
2897 let settings = &cx.global::<Settings>();
2898 let tab_size = settings.tab_size(language_name.as_deref());
2899 let indent_kind = if settings.hard_tabs(language_name.as_deref()) {
2900 IndentKind::Tab
2901 } else {
2902 IndentKind::Space
2903 };
2904
2905 let mut start_row = selection.start.row;
2906 let mut end_row = selection.end.row + 1;
2907
2908 // If a selection ends at the beginning of a line, don't indent
2909 // that last line.
2910 if selection.end.column == 0 {
2911 end_row -= 1;
2912 }
2913
2914 // Avoid re-indenting a row that has already been indented by a
2915 // previous selection, but still update this selection's column
2916 // to reflect that indentation.
2917 if let Some((last_indent_row, last_indent_len)) = last_indent {
2918 if last_indent_row == selection.start.row {
2919 selection.start.column += last_indent_len;
2920 start_row += 1;
2921 }
2922 if last_indent_row == selection.end.row {
2923 selection.end.column += last_indent_len;
2924 }
2925 }
2926
2927 for row in start_row..end_row {
2928 let current_indent = snapshot.indent_size_for_line(row);
2929 let indent_delta = match (current_indent.kind, indent_kind) {
2930 (IndentKind::Space, IndentKind::Space) => {
2931 let columns_to_next_tab_stop =
2932 tab_size - (current_indent.len % tab_size);
2933 IndentSize::spaces(columns_to_next_tab_stop)
2934 }
2935 (IndentKind::Tab, IndentKind::Space) => IndentSize::spaces(tab_size),
2936 (_, IndentKind::Tab) => IndentSize::tab(),
2937 };
2938
2939 let row_start = Point::new(row, 0);
2940 buffer.edit(
2941 [(
2942 row_start..row_start,
2943 indent_delta.chars().collect::<String>(),
2944 )],
2945 cx,
2946 );
2947
2948 // Update this selection's endpoints to reflect the indentation.
2949 if row == selection.start.row {
2950 selection.start.column += indent_delta.len;
2951 }
2952 if row == selection.end.row {
2953 selection.end.column += indent_delta.len as u32;
2954 }
2955
2956 last_indent = Some((row, indent_delta.len));
2957 }
2958 }
2959 });
2960
2961 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
2962 s.select(selections);
2963 });
2964 });
2965 }
2966
2967 pub fn outdent(&mut self, _: &Outdent, cx: &mut ViewContext<Self>) {
2968 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
2969 let selections = self.selections.all::<Point>(cx);
2970 let mut deletion_ranges = Vec::new();
2971 let mut last_outdent = None;
2972 {
2973 let buffer = self.buffer.read(cx);
2974 let snapshot = buffer.snapshot(cx);
2975 for selection in &selections {
2976 let language_name = buffer.language_at(selection.start, cx).map(|l| l.name());
2977 let tab_size = cx.global::<Settings>().tab_size(language_name.as_deref());
2978 let mut rows = selection.spanned_rows(false, &display_map);
2979
2980 // Avoid re-outdenting a row that has already been outdented by a
2981 // previous selection.
2982 if let Some(last_row) = last_outdent {
2983 if last_row == rows.start {
2984 rows.start += 1;
2985 }
2986 }
2987
2988 for row in rows {
2989 let indent_size = snapshot.indent_size_for_line(row);
2990 if indent_size.len > 0 {
2991 let deletion_len = match indent_size.kind {
2992 IndentKind::Space => {
2993 let columns_to_prev_tab_stop = indent_size.len % tab_size;
2994 if columns_to_prev_tab_stop == 0 {
2995 tab_size
2996 } else {
2997 columns_to_prev_tab_stop
2998 }
2999 }
3000 IndentKind::Tab => 1,
3001 };
3002 deletion_ranges.push(Point::new(row, 0)..Point::new(row, deletion_len));
3003 last_outdent = Some(row);
3004 }
3005 }
3006 }
3007 }
3008
3009 self.transact(cx, |this, cx| {
3010 this.buffer.update(cx, |buffer, cx| {
3011 let empty_str: Arc<str> = "".into();
3012 buffer.edit(
3013 deletion_ranges
3014 .into_iter()
3015 .map(|range| (range, empty_str.clone())),
3016 cx,
3017 );
3018 });
3019 let selections = this.selections.all::<usize>(cx);
3020 this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(selections));
3021 });
3022 }
3023
3024 pub fn delete_line(&mut self, _: &DeleteLine, cx: &mut ViewContext<Self>) {
3025 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
3026 let selections = self.selections.all::<Point>(cx);
3027
3028 let mut new_cursors = Vec::new();
3029 let mut edit_ranges = Vec::new();
3030 let mut selections = selections.iter().peekable();
3031 while let Some(selection) = selections.next() {
3032 let mut rows = selection.spanned_rows(false, &display_map);
3033 let goal_display_column = selection.head().to_display_point(&display_map).column();
3034
3035 // Accumulate contiguous regions of rows that we want to delete.
3036 while let Some(next_selection) = selections.peek() {
3037 let next_rows = next_selection.spanned_rows(false, &display_map);
3038 if next_rows.start <= rows.end {
3039 rows.end = next_rows.end;
3040 selections.next().unwrap();
3041 } else {
3042 break;
3043 }
3044 }
3045
3046 let buffer = &display_map.buffer_snapshot;
3047 let mut edit_start = Point::new(rows.start, 0).to_offset(&buffer);
3048 let edit_end;
3049 let cursor_buffer_row;
3050 if buffer.max_point().row >= rows.end {
3051 // If there's a line after the range, delete the \n from the end of the row range
3052 // and position the cursor on the next line.
3053 edit_end = Point::new(rows.end, 0).to_offset(&buffer);
3054 cursor_buffer_row = rows.end;
3055 } else {
3056 // If there isn't a line after the range, delete the \n from the line before the
3057 // start of the row range and position the cursor there.
3058 edit_start = edit_start.saturating_sub(1);
3059 edit_end = buffer.len();
3060 cursor_buffer_row = rows.start.saturating_sub(1);
3061 }
3062
3063 let mut cursor = Point::new(cursor_buffer_row, 0).to_display_point(&display_map);
3064 *cursor.column_mut() =
3065 cmp::min(goal_display_column, display_map.line_len(cursor.row()));
3066
3067 new_cursors.push((
3068 selection.id,
3069 buffer.anchor_after(cursor.to_point(&display_map)),
3070 ));
3071 edit_ranges.push(edit_start..edit_end);
3072 }
3073
3074 self.transact(cx, |this, cx| {
3075 let buffer = this.buffer.update(cx, |buffer, cx| {
3076 let empty_str: Arc<str> = "".into();
3077 buffer.edit(
3078 edit_ranges
3079 .into_iter()
3080 .map(|range| (range, empty_str.clone())),
3081 cx,
3082 );
3083 buffer.snapshot(cx)
3084 });
3085 let new_selections = new_cursors
3086 .into_iter()
3087 .map(|(id, cursor)| {
3088 let cursor = cursor.to_point(&buffer);
3089 Selection {
3090 id,
3091 start: cursor,
3092 end: cursor,
3093 reversed: false,
3094 goal: SelectionGoal::None,
3095 }
3096 })
3097 .collect();
3098
3099 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3100 s.select(new_selections);
3101 });
3102 });
3103 }
3104
3105 pub fn duplicate_line(&mut self, _: &DuplicateLine, cx: &mut ViewContext<Self>) {
3106 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
3107 let buffer = &display_map.buffer_snapshot;
3108 let selections = self.selections.all::<Point>(cx);
3109
3110 let mut edits = Vec::new();
3111 let mut selections_iter = selections.iter().peekable();
3112 while let Some(selection) = selections_iter.next() {
3113 // Avoid duplicating the same lines twice.
3114 let mut rows = selection.spanned_rows(false, &display_map);
3115
3116 while let Some(next_selection) = selections_iter.peek() {
3117 let next_rows = next_selection.spanned_rows(false, &display_map);
3118 if next_rows.start <= rows.end - 1 {
3119 rows.end = next_rows.end;
3120 selections_iter.next().unwrap();
3121 } else {
3122 break;
3123 }
3124 }
3125
3126 // Copy the text from the selected row region and splice it at the start of the region.
3127 let start = Point::new(rows.start, 0);
3128 let end = Point::new(rows.end - 1, buffer.line_len(rows.end - 1));
3129 let text = buffer
3130 .text_for_range(start..end)
3131 .chain(Some("\n"))
3132 .collect::<String>();
3133 edits.push((start..start, text));
3134 }
3135
3136 self.transact(cx, |this, cx| {
3137 this.buffer.update(cx, |buffer, cx| {
3138 buffer.edit(edits, cx);
3139 });
3140
3141 this.request_autoscroll(Autoscroll::Fit, cx);
3142 });
3143 }
3144
3145 pub fn move_line_up(&mut self, _: &MoveLineUp, cx: &mut ViewContext<Self>) {
3146 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
3147 let buffer = self.buffer.read(cx).snapshot(cx);
3148
3149 let mut edits = Vec::new();
3150 let mut unfold_ranges = Vec::new();
3151 let mut refold_ranges = Vec::new();
3152
3153 let selections = self.selections.all::<Point>(cx);
3154 let mut selections = selections.iter().peekable();
3155 let mut contiguous_row_selections = Vec::new();
3156 let mut new_selections = Vec::new();
3157
3158 while let Some(selection) = selections.next() {
3159 // Find all the selections that span a contiguous row range
3160 contiguous_row_selections.push(selection.clone());
3161 let start_row = selection.start.row;
3162 let mut end_row = if selection.end.column > 0 || selection.is_empty() {
3163 display_map.next_line_boundary(selection.end).0.row + 1
3164 } else {
3165 selection.end.row
3166 };
3167
3168 while let Some(next_selection) = selections.peek() {
3169 if next_selection.start.row <= end_row {
3170 end_row = if next_selection.end.column > 0 || next_selection.is_empty() {
3171 display_map.next_line_boundary(next_selection.end).0.row + 1
3172 } else {
3173 next_selection.end.row
3174 };
3175 contiguous_row_selections.push(selections.next().unwrap().clone());
3176 } else {
3177 break;
3178 }
3179 }
3180
3181 // Move the text spanned by the row range to be before the line preceding the row range
3182 if start_row > 0 {
3183 let range_to_move = Point::new(start_row - 1, buffer.line_len(start_row - 1))
3184 ..Point::new(end_row - 1, buffer.line_len(end_row - 1));
3185 let insertion_point = display_map
3186 .prev_line_boundary(Point::new(start_row - 1, 0))
3187 .0;
3188
3189 // Don't move lines across excerpts
3190 if buffer
3191 .excerpt_boundaries_in_range((
3192 Bound::Excluded(insertion_point),
3193 Bound::Included(range_to_move.end),
3194 ))
3195 .next()
3196 .is_none()
3197 {
3198 let text = buffer
3199 .text_for_range(range_to_move.clone())
3200 .flat_map(|s| s.chars())
3201 .skip(1)
3202 .chain(['\n'])
3203 .collect::<String>();
3204
3205 edits.push((
3206 buffer.anchor_after(range_to_move.start)
3207 ..buffer.anchor_before(range_to_move.end),
3208 String::new(),
3209 ));
3210 let insertion_anchor = buffer.anchor_after(insertion_point);
3211 edits.push((insertion_anchor.clone()..insertion_anchor, text));
3212
3213 let row_delta = range_to_move.start.row - insertion_point.row + 1;
3214
3215 // Move selections up
3216 new_selections.extend(contiguous_row_selections.drain(..).map(
3217 |mut selection| {
3218 selection.start.row -= row_delta;
3219 selection.end.row -= row_delta;
3220 selection
3221 },
3222 ));
3223
3224 // Move folds up
3225 unfold_ranges.push(range_to_move.clone());
3226 for fold in display_map.folds_in_range(
3227 buffer.anchor_before(range_to_move.start)
3228 ..buffer.anchor_after(range_to_move.end),
3229 ) {
3230 let mut start = fold.start.to_point(&buffer);
3231 let mut end = fold.end.to_point(&buffer);
3232 start.row -= row_delta;
3233 end.row -= row_delta;
3234 refold_ranges.push(start..end);
3235 }
3236 }
3237 }
3238
3239 // If we didn't move line(s), preserve the existing selections
3240 new_selections.extend(contiguous_row_selections.drain(..));
3241 }
3242
3243 self.transact(cx, |this, cx| {
3244 this.unfold_ranges(unfold_ranges, true, cx);
3245 this.buffer.update(cx, |buffer, cx| {
3246 for (range, text) in edits {
3247 buffer.edit([(range, text)], cx);
3248 }
3249 });
3250 this.fold_ranges(refold_ranges, cx);
3251 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3252 s.select(new_selections);
3253 })
3254 });
3255 }
3256
3257 pub fn move_line_down(&mut self, _: &MoveLineDown, cx: &mut ViewContext<Self>) {
3258 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
3259 let buffer = self.buffer.read(cx).snapshot(cx);
3260
3261 let mut edits = Vec::new();
3262 let mut unfold_ranges = Vec::new();
3263 let mut refold_ranges = Vec::new();
3264
3265 let selections = self.selections.all::<Point>(cx);
3266 let mut selections = selections.iter().peekable();
3267 let mut contiguous_row_selections = Vec::new();
3268 let mut new_selections = Vec::new();
3269
3270 while let Some(selection) = selections.next() {
3271 // Find all the selections that span a contiguous row range
3272 contiguous_row_selections.push(selection.clone());
3273 let start_row = selection.start.row;
3274 let mut end_row = if selection.end.column > 0 || selection.is_empty() {
3275 display_map.next_line_boundary(selection.end).0.row + 1
3276 } else {
3277 selection.end.row
3278 };
3279
3280 while let Some(next_selection) = selections.peek() {
3281 if next_selection.start.row <= end_row {
3282 end_row = if next_selection.end.column > 0 || next_selection.is_empty() {
3283 display_map.next_line_boundary(next_selection.end).0.row + 1
3284 } else {
3285 next_selection.end.row
3286 };
3287 contiguous_row_selections.push(selections.next().unwrap().clone());
3288 } else {
3289 break;
3290 }
3291 }
3292
3293 // Move the text spanned by the row range to be after the last line of the row range
3294 if end_row <= buffer.max_point().row {
3295 let range_to_move = Point::new(start_row, 0)..Point::new(end_row, 0);
3296 let insertion_point = display_map.next_line_boundary(Point::new(end_row, 0)).0;
3297
3298 // Don't move lines across excerpt boundaries
3299 if buffer
3300 .excerpt_boundaries_in_range((
3301 Bound::Excluded(range_to_move.start),
3302 Bound::Included(insertion_point),
3303 ))
3304 .next()
3305 .is_none()
3306 {
3307 let mut text = String::from("\n");
3308 text.extend(buffer.text_for_range(range_to_move.clone()));
3309 text.pop(); // Drop trailing newline
3310 edits.push((
3311 buffer.anchor_after(range_to_move.start)
3312 ..buffer.anchor_before(range_to_move.end),
3313 String::new(),
3314 ));
3315 let insertion_anchor = buffer.anchor_after(insertion_point);
3316 edits.push((insertion_anchor.clone()..insertion_anchor, text));
3317
3318 let row_delta = insertion_point.row - range_to_move.end.row + 1;
3319
3320 // Move selections down
3321 new_selections.extend(contiguous_row_selections.drain(..).map(
3322 |mut selection| {
3323 selection.start.row += row_delta;
3324 selection.end.row += row_delta;
3325 selection
3326 },
3327 ));
3328
3329 // Move folds down
3330 unfold_ranges.push(range_to_move.clone());
3331 for fold in display_map.folds_in_range(
3332 buffer.anchor_before(range_to_move.start)
3333 ..buffer.anchor_after(range_to_move.end),
3334 ) {
3335 let mut start = fold.start.to_point(&buffer);
3336 let mut end = fold.end.to_point(&buffer);
3337 start.row += row_delta;
3338 end.row += row_delta;
3339 refold_ranges.push(start..end);
3340 }
3341 }
3342 }
3343
3344 // If we didn't move line(s), preserve the existing selections
3345 new_selections.extend(contiguous_row_selections.drain(..));
3346 }
3347
3348 self.transact(cx, |this, cx| {
3349 this.unfold_ranges(unfold_ranges, true, cx);
3350 this.buffer.update(cx, |buffer, cx| {
3351 for (range, text) in edits {
3352 buffer.edit([(range, text)], cx);
3353 }
3354 });
3355 this.fold_ranges(refold_ranges, cx);
3356 this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(new_selections));
3357 });
3358 }
3359
3360 pub fn transpose(&mut self, _: &Transpose, cx: &mut ViewContext<Self>) {
3361 self.transact(cx, |this, cx| {
3362 let edits = this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3363 let mut edits: Vec<(Range<usize>, String)> = Default::default();
3364 let line_mode = s.line_mode;
3365 s.move_with(|display_map, selection| {
3366 if !selection.is_empty() || line_mode {
3367 return;
3368 }
3369
3370 let mut head = selection.head();
3371 let mut transpose_offset = head.to_offset(display_map, Bias::Right);
3372 if head.column() == display_map.line_len(head.row()) {
3373 transpose_offset = display_map
3374 .buffer_snapshot
3375 .clip_offset(transpose_offset.saturating_sub(1), Bias::Left);
3376 }
3377
3378 if transpose_offset == 0 {
3379 return;
3380 }
3381
3382 *head.column_mut() += 1;
3383 head = display_map.clip_point(head, Bias::Right);
3384 selection.collapse_to(head, SelectionGoal::Column(head.column()));
3385
3386 let transpose_start = display_map
3387 .buffer_snapshot
3388 .clip_offset(transpose_offset.saturating_sub(1), Bias::Left);
3389 if edits.last().map_or(true, |e| e.0.end <= transpose_start) {
3390 let transpose_end = display_map
3391 .buffer_snapshot
3392 .clip_offset(transpose_offset + 1, Bias::Right);
3393 if let Some(ch) =
3394 display_map.buffer_snapshot.chars_at(transpose_start).next()
3395 {
3396 edits.push((transpose_start..transpose_offset, String::new()));
3397 edits.push((transpose_end..transpose_end, ch.to_string()));
3398 }
3399 }
3400 });
3401 edits
3402 });
3403 this.buffer.update(cx, |buffer, cx| buffer.edit(edits, cx));
3404 let selections = this.selections.all::<usize>(cx);
3405 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3406 s.select(selections);
3407 });
3408 });
3409 }
3410
3411 pub fn cut(&mut self, _: &Cut, cx: &mut ViewContext<Self>) {
3412 let mut text = String::new();
3413 let buffer = self.buffer.read(cx).snapshot(cx);
3414 let mut selections = self.selections.all::<Point>(cx);
3415 let mut clipboard_selections = Vec::with_capacity(selections.len());
3416 {
3417 let max_point = buffer.max_point();
3418 for selection in &mut selections {
3419 let is_entire_line = selection.is_empty() || self.selections.line_mode;
3420 if is_entire_line {
3421 selection.start = Point::new(selection.start.row, 0);
3422 selection.end = cmp::min(max_point, Point::new(selection.end.row + 1, 0));
3423 selection.goal = SelectionGoal::None;
3424 }
3425 let mut len = 0;
3426 for chunk in buffer.text_for_range(selection.start..selection.end) {
3427 text.push_str(chunk);
3428 len += chunk.len();
3429 }
3430 clipboard_selections.push(ClipboardSelection {
3431 len,
3432 is_entire_line,
3433 });
3434 }
3435 }
3436
3437 self.transact(cx, |this, cx| {
3438 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3439 s.select(selections);
3440 });
3441 this.insert("", cx);
3442 cx.write_to_clipboard(ClipboardItem::new(text).with_metadata(clipboard_selections));
3443 });
3444 }
3445
3446 pub fn copy(&mut self, _: &Copy, cx: &mut ViewContext<Self>) {
3447 let selections = self.selections.all::<Point>(cx);
3448 let buffer = self.buffer.read(cx).read(cx);
3449 let mut text = String::new();
3450
3451 let mut clipboard_selections = Vec::with_capacity(selections.len());
3452 {
3453 let max_point = buffer.max_point();
3454 for selection in selections.iter() {
3455 let mut start = selection.start;
3456 let mut end = selection.end;
3457 let is_entire_line = selection.is_empty() || self.selections.line_mode;
3458 if is_entire_line {
3459 start = Point::new(start.row, 0);
3460 end = cmp::min(max_point, Point::new(end.row + 1, 0));
3461 }
3462 let mut len = 0;
3463 for chunk in buffer.text_for_range(start..end) {
3464 text.push_str(chunk);
3465 len += chunk.len();
3466 }
3467 clipboard_selections.push(ClipboardSelection {
3468 len,
3469 is_entire_line,
3470 });
3471 }
3472 }
3473
3474 cx.write_to_clipboard(ClipboardItem::new(text).with_metadata(clipboard_selections));
3475 }
3476
3477 pub fn paste(&mut self, _: &Paste, cx: &mut ViewContext<Self>) {
3478 self.transact(cx, |this, cx| {
3479 if let Some(item) = cx.as_mut().read_from_clipboard() {
3480 let mut clipboard_text = Cow::Borrowed(item.text());
3481 if let Some(mut clipboard_selections) = item.metadata::<Vec<ClipboardSelection>>() {
3482 let old_selections = this.selections.all::<usize>(cx);
3483 let all_selections_were_entire_line =
3484 clipboard_selections.iter().all(|s| s.is_entire_line);
3485 if clipboard_selections.len() != old_selections.len() {
3486 let mut newline_separated_text = String::new();
3487 let mut clipboard_selections = clipboard_selections.drain(..).peekable();
3488 let mut ix = 0;
3489 while let Some(clipboard_selection) = clipboard_selections.next() {
3490 newline_separated_text
3491 .push_str(&clipboard_text[ix..ix + clipboard_selection.len]);
3492 ix += clipboard_selection.len;
3493 if clipboard_selections.peek().is_some() {
3494 newline_separated_text.push('\n');
3495 }
3496 }
3497 clipboard_text = Cow::Owned(newline_separated_text);
3498 }
3499
3500 this.buffer.update(cx, |buffer, cx| {
3501 let snapshot = buffer.read(cx);
3502 let mut start_offset = 0;
3503 let mut edits = Vec::new();
3504 let line_mode = this.selections.line_mode;
3505 for (ix, selection) in old_selections.iter().enumerate() {
3506 let to_insert;
3507 let entire_line;
3508 if let Some(clipboard_selection) = clipboard_selections.get(ix) {
3509 let end_offset = start_offset + clipboard_selection.len;
3510 to_insert = &clipboard_text[start_offset..end_offset];
3511 entire_line = clipboard_selection.is_entire_line;
3512 start_offset = end_offset;
3513 } else {
3514 to_insert = clipboard_text.as_str();
3515 entire_line = all_selections_were_entire_line;
3516 }
3517
3518 // If the corresponding selection was empty when this slice of the
3519 // clipboard text was written, then the entire line containing the
3520 // selection was copied. If this selection is also currently empty,
3521 // then paste the line before the current line of the buffer.
3522 let range = if selection.is_empty() && !line_mode && entire_line {
3523 let column = selection.start.to_point(&snapshot).column as usize;
3524 let line_start = selection.start - column;
3525 line_start..line_start
3526 } else {
3527 selection.range()
3528 };
3529
3530 edits.push((range, to_insert));
3531 }
3532 drop(snapshot);
3533 buffer.edit_with_autoindent(edits, cx);
3534 });
3535
3536 let selections = this.selections.all::<usize>(cx);
3537 this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(selections));
3538 } else {
3539 this.insert(&clipboard_text, cx);
3540 }
3541 }
3542 });
3543 }
3544
3545 pub fn undo(&mut self, _: &Undo, cx: &mut ViewContext<Self>) {
3546 if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.undo(cx)) {
3547 if let Some((selections, _)) = self.selection_history.transaction(tx_id).cloned() {
3548 self.change_selections(None, cx, |s| {
3549 s.select_anchors(selections.to_vec());
3550 });
3551 }
3552 self.request_autoscroll(Autoscroll::Fit, cx);
3553 cx.emit(Event::Edited);
3554 }
3555 }
3556
3557 pub fn redo(&mut self, _: &Redo, cx: &mut ViewContext<Self>) {
3558 if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.redo(cx)) {
3559 if let Some((_, Some(selections))) = self.selection_history.transaction(tx_id).cloned()
3560 {
3561 self.change_selections(None, cx, |s| {
3562 s.select_anchors(selections.to_vec());
3563 });
3564 }
3565 self.request_autoscroll(Autoscroll::Fit, cx);
3566 cx.emit(Event::Edited);
3567 }
3568 }
3569
3570 pub fn finalize_last_transaction(&mut self, cx: &mut ViewContext<Self>) {
3571 self.buffer
3572 .update(cx, |buffer, cx| buffer.finalize_last_transaction(cx));
3573 }
3574
3575 pub fn move_left(&mut self, _: &MoveLeft, cx: &mut ViewContext<Self>) {
3576 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3577 let line_mode = s.line_mode;
3578 s.move_with(|map, selection| {
3579 let cursor = if selection.is_empty() && !line_mode {
3580 movement::left(map, selection.start)
3581 } else {
3582 selection.start
3583 };
3584 selection.collapse_to(cursor, SelectionGoal::None);
3585 });
3586 })
3587 }
3588
3589 pub fn select_left(&mut self, _: &SelectLeft, cx: &mut ViewContext<Self>) {
3590 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3591 s.move_heads_with(|map, head, _| (movement::left(map, head), SelectionGoal::None));
3592 })
3593 }
3594
3595 pub fn move_right(&mut self, _: &MoveRight, cx: &mut ViewContext<Self>) {
3596 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3597 let line_mode = s.line_mode;
3598 s.move_with(|map, selection| {
3599 let cursor = if selection.is_empty() && !line_mode {
3600 movement::right(map, selection.end)
3601 } else {
3602 selection.end
3603 };
3604 selection.collapse_to(cursor, SelectionGoal::None)
3605 });
3606 })
3607 }
3608
3609 pub fn select_right(&mut self, _: &SelectRight, cx: &mut ViewContext<Self>) {
3610 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3611 s.move_heads_with(|map, head, _| (movement::right(map, head), SelectionGoal::None));
3612 })
3613 }
3614
3615 pub fn move_up(&mut self, _: &MoveUp, cx: &mut ViewContext<Self>) {
3616 if self.take_rename(true, cx).is_some() {
3617 return;
3618 }
3619
3620 if let Some(context_menu) = self.context_menu.as_mut() {
3621 if context_menu.select_prev(cx) {
3622 return;
3623 }
3624 }
3625
3626 if matches!(self.mode, EditorMode::SingleLine) {
3627 cx.propagate_action();
3628 return;
3629 }
3630
3631 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3632 let line_mode = s.line_mode;
3633 s.move_with(|map, selection| {
3634 if !selection.is_empty() && !line_mode {
3635 selection.goal = SelectionGoal::None;
3636 }
3637 let (cursor, goal) = movement::up(&map, selection.start, selection.goal, false);
3638 selection.collapse_to(cursor, goal);
3639 });
3640 })
3641 }
3642
3643 pub fn select_up(&mut self, _: &SelectUp, cx: &mut ViewContext<Self>) {
3644 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3645 s.move_heads_with(|map, head, goal| movement::up(map, head, goal, false))
3646 })
3647 }
3648
3649 pub fn move_down(&mut self, _: &MoveDown, cx: &mut ViewContext<Self>) {
3650 self.take_rename(true, cx);
3651
3652 if let Some(context_menu) = self.context_menu.as_mut() {
3653 if context_menu.select_next(cx) {
3654 return;
3655 }
3656 }
3657
3658 if matches!(self.mode, EditorMode::SingleLine) {
3659 cx.propagate_action();
3660 return;
3661 }
3662
3663 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3664 let line_mode = s.line_mode;
3665 s.move_with(|map, selection| {
3666 if !selection.is_empty() && !line_mode {
3667 selection.goal = SelectionGoal::None;
3668 }
3669 let (cursor, goal) = movement::down(&map, selection.end, selection.goal, false);
3670 selection.collapse_to(cursor, goal);
3671 });
3672 });
3673 }
3674
3675 pub fn select_down(&mut self, _: &SelectDown, cx: &mut ViewContext<Self>) {
3676 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3677 s.move_heads_with(|map, head, goal| movement::down(map, head, goal, false))
3678 });
3679 }
3680
3681 pub fn move_to_previous_word_start(
3682 &mut self,
3683 _: &MoveToPreviousWordStart,
3684 cx: &mut ViewContext<Self>,
3685 ) {
3686 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3687 s.move_cursors_with(|map, head, _| {
3688 (
3689 movement::previous_word_start(map, head),
3690 SelectionGoal::None,
3691 )
3692 });
3693 })
3694 }
3695
3696 pub fn move_to_previous_subword_start(
3697 &mut self,
3698 _: &MoveToPreviousSubwordStart,
3699 cx: &mut ViewContext<Self>,
3700 ) {
3701 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3702 s.move_cursors_with(|map, head, _| {
3703 (
3704 movement::previous_subword_start(map, head),
3705 SelectionGoal::None,
3706 )
3707 });
3708 })
3709 }
3710
3711 pub fn select_to_previous_word_start(
3712 &mut self,
3713 _: &SelectToPreviousWordStart,
3714 cx: &mut ViewContext<Self>,
3715 ) {
3716 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3717 s.move_heads_with(|map, head, _| {
3718 (
3719 movement::previous_word_start(map, head),
3720 SelectionGoal::None,
3721 )
3722 });
3723 })
3724 }
3725
3726 pub fn select_to_previous_subword_start(
3727 &mut self,
3728 _: &SelectToPreviousSubwordStart,
3729 cx: &mut ViewContext<Self>,
3730 ) {
3731 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3732 s.move_heads_with(|map, head, _| {
3733 (
3734 movement::previous_subword_start(map, head),
3735 SelectionGoal::None,
3736 )
3737 });
3738 })
3739 }
3740
3741 pub fn delete_to_previous_word_start(
3742 &mut self,
3743 _: &DeleteToPreviousWordStart,
3744 cx: &mut ViewContext<Self>,
3745 ) {
3746 self.transact(cx, |this, cx| {
3747 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3748 let line_mode = s.line_mode;
3749 s.move_with(|map, selection| {
3750 if selection.is_empty() && !line_mode {
3751 let cursor = movement::previous_word_start(map, selection.head());
3752 selection.set_head(cursor, SelectionGoal::None);
3753 }
3754 });
3755 });
3756 this.insert("", cx);
3757 });
3758 }
3759
3760 pub fn delete_to_previous_subword_start(
3761 &mut self,
3762 _: &DeleteToPreviousSubwordStart,
3763 cx: &mut ViewContext<Self>,
3764 ) {
3765 self.transact(cx, |this, cx| {
3766 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3767 let line_mode = s.line_mode;
3768 s.move_with(|map, selection| {
3769 if selection.is_empty() && !line_mode {
3770 let cursor = movement::previous_subword_start(map, selection.head());
3771 selection.set_head(cursor, SelectionGoal::None);
3772 }
3773 });
3774 });
3775 this.insert("", cx);
3776 });
3777 }
3778
3779 pub fn move_to_next_word_end(&mut self, _: &MoveToNextWordEnd, cx: &mut ViewContext<Self>) {
3780 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3781 s.move_cursors_with(|map, head, _| {
3782 (movement::next_word_end(map, head), SelectionGoal::None)
3783 });
3784 })
3785 }
3786
3787 pub fn move_to_next_subword_end(
3788 &mut self,
3789 _: &MoveToNextSubwordEnd,
3790 cx: &mut ViewContext<Self>,
3791 ) {
3792 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3793 s.move_cursors_with(|map, head, _| {
3794 (movement::next_subword_end(map, head), SelectionGoal::None)
3795 });
3796 })
3797 }
3798
3799 pub fn select_to_next_word_end(&mut self, _: &SelectToNextWordEnd, cx: &mut ViewContext<Self>) {
3800 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3801 s.move_heads_with(|map, head, _| {
3802 (movement::next_word_end(map, head), SelectionGoal::None)
3803 });
3804 })
3805 }
3806
3807 pub fn select_to_next_subword_end(
3808 &mut self,
3809 _: &SelectToNextSubwordEnd,
3810 cx: &mut ViewContext<Self>,
3811 ) {
3812 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3813 s.move_heads_with(|map, head, _| {
3814 (movement::next_subword_end(map, head), SelectionGoal::None)
3815 });
3816 })
3817 }
3818
3819 pub fn delete_to_next_word_end(&mut self, _: &DeleteToNextWordEnd, cx: &mut ViewContext<Self>) {
3820 self.transact(cx, |this, cx| {
3821 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3822 let line_mode = s.line_mode;
3823 s.move_with(|map, selection| {
3824 if selection.is_empty() && !line_mode {
3825 let cursor = movement::next_word_end(map, selection.head());
3826 selection.set_head(cursor, SelectionGoal::None);
3827 }
3828 });
3829 });
3830 this.insert("", cx);
3831 });
3832 }
3833
3834 pub fn delete_to_next_subword_end(
3835 &mut self,
3836 _: &DeleteToNextSubwordEnd,
3837 cx: &mut ViewContext<Self>,
3838 ) {
3839 self.transact(cx, |this, cx| {
3840 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3841 s.move_with(|map, selection| {
3842 if selection.is_empty() {
3843 let cursor = movement::next_subword_end(map, selection.head());
3844 selection.set_head(cursor, SelectionGoal::None);
3845 }
3846 });
3847 });
3848 this.insert("", cx);
3849 });
3850 }
3851
3852 pub fn move_to_beginning_of_line(
3853 &mut self,
3854 _: &MoveToBeginningOfLine,
3855 cx: &mut ViewContext<Self>,
3856 ) {
3857 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3858 s.move_cursors_with(|map, head, _| {
3859 (
3860 movement::line_beginning(map, head, true),
3861 SelectionGoal::None,
3862 )
3863 });
3864 })
3865 }
3866
3867 pub fn select_to_beginning_of_line(
3868 &mut self,
3869 action: &SelectToBeginningOfLine,
3870 cx: &mut ViewContext<Self>,
3871 ) {
3872 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3873 s.move_heads_with(|map, head, _| {
3874 (
3875 movement::line_beginning(map, head, action.stop_at_soft_wraps),
3876 SelectionGoal::None,
3877 )
3878 });
3879 });
3880 }
3881
3882 pub fn delete_to_beginning_of_line(
3883 &mut self,
3884 _: &DeleteToBeginningOfLine,
3885 cx: &mut ViewContext<Self>,
3886 ) {
3887 self.transact(cx, |this, cx| {
3888 this.change_selections(Some(Autoscroll::Fit), cx, |s| {
3889 s.move_with(|_, selection| {
3890 selection.reversed = true;
3891 });
3892 });
3893
3894 this.select_to_beginning_of_line(
3895 &SelectToBeginningOfLine {
3896 stop_at_soft_wraps: false,
3897 },
3898 cx,
3899 );
3900 this.backspace(&Backspace, cx);
3901 });
3902 }
3903
3904 pub fn move_to_end_of_line(&mut self, _: &MoveToEndOfLine, cx: &mut ViewContext<Self>) {
3905 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3906 s.move_cursors_with(|map, head, _| {
3907 (movement::line_end(map, head, true), SelectionGoal::None)
3908 });
3909 })
3910 }
3911
3912 pub fn select_to_end_of_line(
3913 &mut self,
3914 action: &SelectToEndOfLine,
3915 cx: &mut ViewContext<Self>,
3916 ) {
3917 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3918 s.move_heads_with(|map, head, _| {
3919 (
3920 movement::line_end(map, head, action.stop_at_soft_wraps),
3921 SelectionGoal::None,
3922 )
3923 });
3924 })
3925 }
3926
3927 pub fn delete_to_end_of_line(&mut self, _: &DeleteToEndOfLine, cx: &mut ViewContext<Self>) {
3928 self.transact(cx, |this, cx| {
3929 this.select_to_end_of_line(
3930 &SelectToEndOfLine {
3931 stop_at_soft_wraps: false,
3932 },
3933 cx,
3934 );
3935 this.delete(&Delete, cx);
3936 });
3937 }
3938
3939 pub fn cut_to_end_of_line(&mut self, _: &CutToEndOfLine, cx: &mut ViewContext<Self>) {
3940 self.transact(cx, |this, cx| {
3941 this.select_to_end_of_line(
3942 &SelectToEndOfLine {
3943 stop_at_soft_wraps: false,
3944 },
3945 cx,
3946 );
3947 this.cut(&Cut, cx);
3948 });
3949 }
3950
3951 pub fn move_to_beginning(&mut self, _: &MoveToBeginning, cx: &mut ViewContext<Self>) {
3952 if matches!(self.mode, EditorMode::SingleLine) {
3953 cx.propagate_action();
3954 return;
3955 }
3956
3957 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3958 s.select_ranges(vec![0..0]);
3959 });
3960 }
3961
3962 pub fn select_to_beginning(&mut self, _: &SelectToBeginning, cx: &mut ViewContext<Self>) {
3963 let mut selection = self.selections.last::<Point>(cx);
3964 selection.set_head(Point::zero(), SelectionGoal::None);
3965
3966 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3967 s.select(vec![selection]);
3968 });
3969 }
3970
3971 pub fn move_to_end(&mut self, _: &MoveToEnd, cx: &mut ViewContext<Self>) {
3972 if matches!(self.mode, EditorMode::SingleLine) {
3973 cx.propagate_action();
3974 return;
3975 }
3976
3977 let cursor = self.buffer.read(cx).read(cx).len();
3978 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
3979 s.select_ranges(vec![cursor..cursor])
3980 });
3981 }
3982
3983 pub fn set_nav_history(&mut self, nav_history: Option<ItemNavHistory>) {
3984 self.nav_history = nav_history;
3985 }
3986
3987 pub fn nav_history(&self) -> Option<&ItemNavHistory> {
3988 self.nav_history.as_ref()
3989 }
3990
3991 fn push_to_nav_history(
3992 &self,
3993 position: Anchor,
3994 new_position: Option<Point>,
3995 cx: &mut ViewContext<Self>,
3996 ) {
3997 if let Some(nav_history) = &self.nav_history {
3998 let buffer = self.buffer.read(cx).read(cx);
3999 let point = position.to_point(&buffer);
4000 let scroll_top_row = self.scroll_top_anchor.to_point(&buffer).row;
4001 drop(buffer);
4002
4003 if let Some(new_position) = new_position {
4004 let row_delta = (new_position.row as i64 - point.row as i64).abs();
4005 if row_delta < MIN_NAVIGATION_HISTORY_ROW_DELTA {
4006 return;
4007 }
4008 }
4009
4010 nav_history.push(Some(NavigationData {
4011 cursor_anchor: position,
4012 cursor_position: point,
4013 scroll_position: self.scroll_position,
4014 scroll_top_anchor: self.scroll_top_anchor.clone(),
4015 scroll_top_row,
4016 }));
4017 }
4018 }
4019
4020 pub fn select_to_end(&mut self, _: &SelectToEnd, cx: &mut ViewContext<Self>) {
4021 let buffer = self.buffer.read(cx).snapshot(cx);
4022 let mut selection = self.selections.first::<usize>(cx);
4023 selection.set_head(buffer.len(), SelectionGoal::None);
4024 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4025 s.select(vec![selection]);
4026 });
4027 }
4028
4029 pub fn select_all(&mut self, _: &SelectAll, cx: &mut ViewContext<Self>) {
4030 let end = self.buffer.read(cx).read(cx).len();
4031 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4032 s.select_ranges(vec![0..end]);
4033 });
4034 }
4035
4036 pub fn select_line(&mut self, _: &SelectLine, cx: &mut ViewContext<Self>) {
4037 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
4038 let mut selections = self.selections.all::<Point>(cx);
4039 let max_point = display_map.buffer_snapshot.max_point();
4040 for selection in &mut selections {
4041 let rows = selection.spanned_rows(true, &display_map);
4042 selection.start = Point::new(rows.start, 0);
4043 selection.end = cmp::min(max_point, Point::new(rows.end, 0));
4044 selection.reversed = false;
4045 }
4046 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4047 s.select(selections);
4048 });
4049 }
4050
4051 pub fn split_selection_into_lines(
4052 &mut self,
4053 _: &SplitSelectionIntoLines,
4054 cx: &mut ViewContext<Self>,
4055 ) {
4056 let mut to_unfold = Vec::new();
4057 let mut new_selection_ranges = Vec::new();
4058 {
4059 let selections = self.selections.all::<Point>(cx);
4060 let buffer = self.buffer.read(cx).read(cx);
4061 for selection in selections {
4062 for row in selection.start.row..selection.end.row {
4063 let cursor = Point::new(row, buffer.line_len(row));
4064 new_selection_ranges.push(cursor..cursor);
4065 }
4066 new_selection_ranges.push(selection.end..selection.end);
4067 to_unfold.push(selection.start..selection.end);
4068 }
4069 }
4070 self.unfold_ranges(to_unfold, true, cx);
4071 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4072 s.select_ranges(new_selection_ranges);
4073 });
4074 }
4075
4076 pub fn add_selection_above(&mut self, _: &AddSelectionAbove, cx: &mut ViewContext<Self>) {
4077 self.add_selection(true, cx);
4078 }
4079
4080 pub fn add_selection_below(&mut self, _: &AddSelectionBelow, cx: &mut ViewContext<Self>) {
4081 self.add_selection(false, cx);
4082 }
4083
4084 fn add_selection(&mut self, above: bool, cx: &mut ViewContext<Self>) {
4085 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
4086 let mut selections = self.selections.all::<Point>(cx);
4087 let mut state = self.add_selections_state.take().unwrap_or_else(|| {
4088 let oldest_selection = selections.iter().min_by_key(|s| s.id).unwrap().clone();
4089 let range = oldest_selection.display_range(&display_map).sorted();
4090 let columns = cmp::min(range.start.column(), range.end.column())
4091 ..cmp::max(range.start.column(), range.end.column());
4092
4093 selections.clear();
4094 let mut stack = Vec::new();
4095 for row in range.start.row()..=range.end.row() {
4096 if let Some(selection) = self.selections.build_columnar_selection(
4097 &display_map,
4098 row,
4099 &columns,
4100 oldest_selection.reversed,
4101 ) {
4102 stack.push(selection.id);
4103 selections.push(selection);
4104 }
4105 }
4106
4107 if above {
4108 stack.reverse();
4109 }
4110
4111 AddSelectionsState { above, stack }
4112 });
4113
4114 let last_added_selection = *state.stack.last().unwrap();
4115 let mut new_selections = Vec::new();
4116 if above == state.above {
4117 let end_row = if above {
4118 0
4119 } else {
4120 display_map.max_point().row()
4121 };
4122
4123 'outer: for selection in selections {
4124 if selection.id == last_added_selection {
4125 let range = selection.display_range(&display_map).sorted();
4126 debug_assert_eq!(range.start.row(), range.end.row());
4127 let mut row = range.start.row();
4128 let columns = if let SelectionGoal::ColumnRange { start, end } = selection.goal
4129 {
4130 start..end
4131 } else {
4132 cmp::min(range.start.column(), range.end.column())
4133 ..cmp::max(range.start.column(), range.end.column())
4134 };
4135
4136 while row != end_row {
4137 if above {
4138 row -= 1;
4139 } else {
4140 row += 1;
4141 }
4142
4143 if let Some(new_selection) = self.selections.build_columnar_selection(
4144 &display_map,
4145 row,
4146 &columns,
4147 selection.reversed,
4148 ) {
4149 state.stack.push(new_selection.id);
4150 if above {
4151 new_selections.push(new_selection);
4152 new_selections.push(selection);
4153 } else {
4154 new_selections.push(selection);
4155 new_selections.push(new_selection);
4156 }
4157
4158 continue 'outer;
4159 }
4160 }
4161 }
4162
4163 new_selections.push(selection);
4164 }
4165 } else {
4166 new_selections = selections;
4167 new_selections.retain(|s| s.id != last_added_selection);
4168 state.stack.pop();
4169 }
4170
4171 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4172 s.select(new_selections);
4173 });
4174 if state.stack.len() > 1 {
4175 self.add_selections_state = Some(state);
4176 }
4177 }
4178
4179 pub fn select_next(&mut self, action: &SelectNext, cx: &mut ViewContext<Self>) {
4180 self.push_to_selection_history();
4181 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
4182 let buffer = &display_map.buffer_snapshot;
4183 let mut selections = self.selections.all::<usize>(cx);
4184 if let Some(mut select_next_state) = self.select_next_state.take() {
4185 let query = &select_next_state.query;
4186 if !select_next_state.done {
4187 let first_selection = selections.iter().min_by_key(|s| s.id).unwrap();
4188 let last_selection = selections.iter().max_by_key(|s| s.id).unwrap();
4189 let mut next_selected_range = None;
4190
4191 let bytes_after_last_selection =
4192 buffer.bytes_in_range(last_selection.end..buffer.len());
4193 let bytes_before_first_selection = buffer.bytes_in_range(0..first_selection.start);
4194 let query_matches = query
4195 .stream_find_iter(bytes_after_last_selection)
4196 .map(|result| (last_selection.end, result))
4197 .chain(
4198 query
4199 .stream_find_iter(bytes_before_first_selection)
4200 .map(|result| (0, result)),
4201 );
4202 for (start_offset, query_match) in query_matches {
4203 let query_match = query_match.unwrap(); // can only fail due to I/O
4204 let offset_range =
4205 start_offset + query_match.start()..start_offset + query_match.end();
4206 let display_range = offset_range.start.to_display_point(&display_map)
4207 ..offset_range.end.to_display_point(&display_map);
4208
4209 if !select_next_state.wordwise
4210 || (!movement::is_inside_word(&display_map, display_range.start)
4211 && !movement::is_inside_word(&display_map, display_range.end))
4212 {
4213 next_selected_range = Some(offset_range);
4214 break;
4215 }
4216 }
4217
4218 if let Some(next_selected_range) = next_selected_range {
4219 self.unfold_ranges([next_selected_range.clone()], false, cx);
4220 self.change_selections(Some(Autoscroll::Newest), cx, |s| {
4221 if action.replace_newest {
4222 s.delete(s.newest_anchor().id);
4223 }
4224 s.insert_range(next_selected_range);
4225 });
4226 } else {
4227 select_next_state.done = true;
4228 }
4229 }
4230
4231 self.select_next_state = Some(select_next_state);
4232 } else if selections.len() == 1 {
4233 let selection = selections.last_mut().unwrap();
4234 if selection.start == selection.end {
4235 let word_range = movement::surrounding_word(
4236 &display_map,
4237 selection.start.to_display_point(&display_map),
4238 );
4239 selection.start = word_range.start.to_offset(&display_map, Bias::Left);
4240 selection.end = word_range.end.to_offset(&display_map, Bias::Left);
4241 selection.goal = SelectionGoal::None;
4242 selection.reversed = false;
4243
4244 let query = buffer
4245 .text_for_range(selection.start..selection.end)
4246 .collect::<String>();
4247 let select_state = SelectNextState {
4248 query: AhoCorasick::new_auto_configured(&[query]),
4249 wordwise: true,
4250 done: false,
4251 };
4252 self.unfold_ranges([selection.start..selection.end], false, cx);
4253 self.change_selections(Some(Autoscroll::Newest), cx, |s| {
4254 s.select(selections);
4255 });
4256 self.select_next_state = Some(select_state);
4257 } else {
4258 let query = buffer
4259 .text_for_range(selection.start..selection.end)
4260 .collect::<String>();
4261 self.select_next_state = Some(SelectNextState {
4262 query: AhoCorasick::new_auto_configured(&[query]),
4263 wordwise: false,
4264 done: false,
4265 });
4266 self.select_next(action, cx);
4267 }
4268 }
4269 }
4270
4271 pub fn toggle_comments(&mut self, _: &ToggleComments, cx: &mut ViewContext<Self>) {
4272 self.transact(cx, |this, cx| {
4273 let mut selections = this.selections.all::<Point>(cx);
4274 let mut all_selection_lines_are_comments = true;
4275 let mut edit_ranges = Vec::new();
4276 let mut last_toggled_row = None;
4277 this.buffer.update(cx, |buffer, cx| {
4278 // TODO: Handle selections that cross excerpts
4279 for selection in &mut selections {
4280 // Get the line comment prefix. Split its trailing whitespace into a separate string,
4281 // as that portion won't be used for detecting if a line is a comment.
4282 let full_comment_prefix: Arc<str> = if let Some(prefix) = buffer
4283 .language_at(selection.start, cx)
4284 .and_then(|l| l.line_comment_prefix())
4285 {
4286 prefix.into()
4287 } else {
4288 return;
4289 };
4290 let comment_prefix = full_comment_prefix.trim_end_matches(' ');
4291 let comment_prefix_whitespace = &full_comment_prefix[comment_prefix.len()..];
4292 edit_ranges.clear();
4293 let snapshot = buffer.snapshot(cx);
4294
4295 let end_row =
4296 if selection.end.row > selection.start.row && selection.end.column == 0 {
4297 selection.end.row
4298 } else {
4299 selection.end.row + 1
4300 };
4301
4302 for row in selection.start.row..end_row {
4303 // If multiple selections contain a given row, avoid processing that
4304 // row more than once.
4305 if last_toggled_row == Some(row) {
4306 continue;
4307 } else {
4308 last_toggled_row = Some(row);
4309 }
4310
4311 if snapshot.is_line_blank(row) {
4312 continue;
4313 }
4314
4315 let start = Point::new(row, snapshot.indent_size_for_line(row).len);
4316 let mut line_bytes = snapshot
4317 .bytes_in_range(start..snapshot.max_point())
4318 .flatten()
4319 .copied();
4320
4321 // If this line currently begins with the line comment prefix, then record
4322 // the range containing the prefix.
4323 if all_selection_lines_are_comments
4324 && line_bytes
4325 .by_ref()
4326 .take(comment_prefix.len())
4327 .eq(comment_prefix.bytes())
4328 {
4329 // Include any whitespace that matches the comment prefix.
4330 let matching_whitespace_len = line_bytes
4331 .zip(comment_prefix_whitespace.bytes())
4332 .take_while(|(a, b)| a == b)
4333 .count()
4334 as u32;
4335 let end = Point::new(
4336 row,
4337 start.column
4338 + comment_prefix.len() as u32
4339 + matching_whitespace_len,
4340 );
4341 edit_ranges.push(start..end);
4342 }
4343 // If this line does not begin with the line comment prefix, then record
4344 // the position where the prefix should be inserted.
4345 else {
4346 all_selection_lines_are_comments = false;
4347 edit_ranges.push(start..start);
4348 }
4349 }
4350
4351 if !edit_ranges.is_empty() {
4352 if all_selection_lines_are_comments {
4353 let empty_str: Arc<str> = "".into();
4354 buffer.edit(
4355 edit_ranges
4356 .iter()
4357 .cloned()
4358 .map(|range| (range, empty_str.clone())),
4359 cx,
4360 );
4361 } else {
4362 let min_column =
4363 edit_ranges.iter().map(|r| r.start.column).min().unwrap();
4364 let edits = edit_ranges.iter().map(|range| {
4365 let position = Point::new(range.start.row, min_column);
4366 (position..position, full_comment_prefix.clone())
4367 });
4368 buffer.edit(edits, cx);
4369 }
4370 }
4371 }
4372 });
4373
4374 let selections = this.selections.all::<usize>(cx);
4375 this.change_selections(Some(Autoscroll::Fit), cx, |s| s.select(selections));
4376 });
4377 }
4378
4379 pub fn select_larger_syntax_node(
4380 &mut self,
4381 _: &SelectLargerSyntaxNode,
4382 cx: &mut ViewContext<Self>,
4383 ) {
4384 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
4385 let buffer = self.buffer.read(cx).snapshot(cx);
4386 let old_selections = self.selections.all::<usize>(cx).into_boxed_slice();
4387
4388 let mut stack = mem::take(&mut self.select_larger_syntax_node_stack);
4389 let mut selected_larger_node = false;
4390 let new_selections = old_selections
4391 .iter()
4392 .map(|selection| {
4393 let old_range = selection.start..selection.end;
4394 let mut new_range = old_range.clone();
4395 while let Some(containing_range) =
4396 buffer.range_for_syntax_ancestor(new_range.clone())
4397 {
4398 new_range = containing_range;
4399 if !display_map.intersects_fold(new_range.start)
4400 && !display_map.intersects_fold(new_range.end)
4401 {
4402 break;
4403 }
4404 }
4405
4406 selected_larger_node |= new_range != old_range;
4407 Selection {
4408 id: selection.id,
4409 start: new_range.start,
4410 end: new_range.end,
4411 goal: SelectionGoal::None,
4412 reversed: selection.reversed,
4413 }
4414 })
4415 .collect::<Vec<_>>();
4416
4417 if selected_larger_node {
4418 stack.push(old_selections);
4419 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4420 s.select(new_selections);
4421 });
4422 }
4423 self.select_larger_syntax_node_stack = stack;
4424 }
4425
4426 pub fn select_smaller_syntax_node(
4427 &mut self,
4428 _: &SelectSmallerSyntaxNode,
4429 cx: &mut ViewContext<Self>,
4430 ) {
4431 let mut stack = mem::take(&mut self.select_larger_syntax_node_stack);
4432 if let Some(selections) = stack.pop() {
4433 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4434 s.select(selections.to_vec());
4435 });
4436 }
4437 self.select_larger_syntax_node_stack = stack;
4438 }
4439
4440 pub fn move_to_enclosing_bracket(
4441 &mut self,
4442 _: &MoveToEnclosingBracket,
4443 cx: &mut ViewContext<Self>,
4444 ) {
4445 let buffer = self.buffer.read(cx).snapshot(cx);
4446 let mut selections = self.selections.all::<usize>(cx);
4447 for selection in &mut selections {
4448 if let Some((open_range, close_range)) =
4449 buffer.enclosing_bracket_ranges(selection.start..selection.end)
4450 {
4451 let close_range = close_range.to_inclusive();
4452 let destination = if close_range.contains(&selection.start)
4453 && close_range.contains(&selection.end)
4454 {
4455 open_range.end
4456 } else {
4457 *close_range.start()
4458 };
4459 selection.start = destination;
4460 selection.end = destination;
4461 }
4462 }
4463
4464 self.change_selections(Some(Autoscroll::Fit), cx, |s| {
4465 s.select(selections);
4466 });
4467 }
4468
4469 pub fn undo_selection(&mut self, _: &UndoSelection, cx: &mut ViewContext<Self>) {
4470 self.end_selection(cx);
4471 self.selection_history.mode = SelectionHistoryMode::Undoing;
4472 if let Some(entry) = self.selection_history.undo_stack.pop_back() {
4473 self.change_selections(None, cx, |s| s.select_anchors(entry.selections.to_vec()));
4474 self.select_next_state = entry.select_next_state;
4475 self.add_selections_state = entry.add_selections_state;
4476 self.request_autoscroll(Autoscroll::Newest, cx);
4477 }
4478 self.selection_history.mode = SelectionHistoryMode::Normal;
4479 }
4480
4481 pub fn redo_selection(&mut self, _: &RedoSelection, cx: &mut ViewContext<Self>) {
4482 self.end_selection(cx);
4483 self.selection_history.mode = SelectionHistoryMode::Redoing;
4484 if let Some(entry) = self.selection_history.redo_stack.pop_back() {
4485 self.change_selections(None, cx, |s| s.select_anchors(entry.selections.to_vec()));
4486 self.select_next_state = entry.select_next_state;
4487 self.add_selections_state = entry.add_selections_state;
4488 self.request_autoscroll(Autoscroll::Newest, cx);
4489 }
4490 self.selection_history.mode = SelectionHistoryMode::Normal;
4491 }
4492
4493 fn go_to_next_diagnostic(&mut self, _: &GoToNextDiagnostic, cx: &mut ViewContext<Self>) {
4494 self.go_to_diagnostic(Direction::Next, cx)
4495 }
4496
4497 fn go_to_prev_diagnostic(&mut self, _: &GoToPrevDiagnostic, cx: &mut ViewContext<Self>) {
4498 self.go_to_diagnostic(Direction::Prev, cx)
4499 }
4500
4501 pub fn go_to_diagnostic(&mut self, direction: Direction, cx: &mut ViewContext<Self>) {
4502 let buffer = self.buffer.read(cx).snapshot(cx);
4503 let selection = self.selections.newest::<usize>(cx);
4504 let mut active_primary_range = self.active_diagnostics.as_ref().map(|active_diagnostics| {
4505 active_diagnostics
4506 .primary_range
4507 .to_offset(&buffer)
4508 .to_inclusive()
4509 });
4510 let mut search_start = if let Some(active_primary_range) = active_primary_range.as_ref() {
4511 if active_primary_range.contains(&selection.head()) {
4512 *active_primary_range.end()
4513 } else {
4514 selection.head()
4515 }
4516 } else {
4517 selection.head()
4518 };
4519
4520 loop {
4521 let mut diagnostics = if direction == Direction::Prev {
4522 buffer.diagnostics_in_range::<_, usize>(0..search_start, true)
4523 } else {
4524 buffer.diagnostics_in_range::<_, usize>(search_start..buffer.len(), false)
4525 };
4526 let group = diagnostics.find_map(|entry| {
4527 if entry.diagnostic.is_primary
4528 && entry.diagnostic.severity <= DiagnosticSeverity::WARNING
4529 && !entry.range.is_empty()
4530 && Some(entry.range.end) != active_primary_range.as_ref().map(|r| *r.end())
4531 {
4532 Some((entry.range, entry.diagnostic.group_id))
4533 } else {
4534 None
4535 }
4536 });
4537
4538 if let Some((primary_range, group_id)) = group {
4539 self.activate_diagnostics(group_id, cx);
4540 self.change_selections(Some(Autoscroll::Center), cx, |s| {
4541 s.select(vec![Selection {
4542 id: selection.id,
4543 start: primary_range.start,
4544 end: primary_range.start,
4545 reversed: false,
4546 goal: SelectionGoal::None,
4547 }]);
4548 });
4549 break;
4550 } else {
4551 // Cycle around to the start of the buffer, potentially moving back to the start of
4552 // the currently active diagnostic.
4553 active_primary_range.take();
4554 if direction == Direction::Prev {
4555 if search_start == buffer.len() {
4556 break;
4557 } else {
4558 search_start = buffer.len();
4559 }
4560 } else {
4561 if search_start == 0 {
4562 break;
4563 } else {
4564 search_start = 0;
4565 }
4566 }
4567 }
4568 }
4569 }
4570
4571 pub fn go_to_definition(
4572 workspace: &mut Workspace,
4573 _: &GoToDefinition,
4574 cx: &mut ViewContext<Workspace>,
4575 ) {
4576 let active_item = workspace.active_item(cx);
4577 let editor_handle = if let Some(editor) = active_item
4578 .as_ref()
4579 .and_then(|item| item.act_as::<Self>(cx))
4580 {
4581 editor
4582 } else {
4583 return;
4584 };
4585
4586 let editor = editor_handle.read(cx);
4587 let buffer = editor.buffer.read(cx);
4588 let head = editor.selections.newest::<usize>(cx).head();
4589 let (buffer, head) = if let Some(text_anchor) = buffer.text_anchor_for_position(head, cx) {
4590 text_anchor
4591 } else {
4592 return;
4593 };
4594
4595 let project = workspace.project().clone();
4596 let definitions = project.update(cx, |project, cx| project.definition(&buffer, head, cx));
4597 cx.spawn(|workspace, mut cx| async move {
4598 let definitions = definitions.await?;
4599 workspace.update(&mut cx, |workspace, cx| {
4600 let nav_history = workspace.active_pane().read(cx).nav_history().clone();
4601 for definition in definitions {
4602 let range = definition.range.to_offset(definition.buffer.read(cx));
4603
4604 let target_editor_handle = workspace.open_project_item(definition.buffer, cx);
4605 target_editor_handle.update(cx, |target_editor, cx| {
4606 // When selecting a definition in a different buffer, disable the nav history
4607 // to avoid creating a history entry at the previous cursor location.
4608 if editor_handle != target_editor_handle {
4609 nav_history.borrow_mut().disable();
4610 }
4611 target_editor.change_selections(Some(Autoscroll::Center), cx, |s| {
4612 s.select_ranges([range]);
4613 });
4614
4615 nav_history.borrow_mut().enable();
4616 });
4617 }
4618 });
4619
4620 Ok::<(), anyhow::Error>(())
4621 })
4622 .detach_and_log_err(cx);
4623 }
4624
4625 pub fn find_all_references(
4626 workspace: &mut Workspace,
4627 _: &FindAllReferences,
4628 cx: &mut ViewContext<Workspace>,
4629 ) -> Option<Task<Result<()>>> {
4630 let active_item = workspace.active_item(cx)?;
4631 let editor_handle = active_item.act_as::<Self>(cx)?;
4632
4633 let editor = editor_handle.read(cx);
4634 let buffer = editor.buffer.read(cx);
4635 let head = editor.selections.newest::<usize>(cx).head();
4636 let (buffer, head) = buffer.text_anchor_for_position(head, cx)?;
4637 let replica_id = editor.replica_id(cx);
4638
4639 let project = workspace.project().clone();
4640 let references = project.update(cx, |project, cx| project.references(&buffer, head, cx));
4641 Some(cx.spawn(|workspace, mut cx| async move {
4642 let mut locations = references.await?;
4643 if locations.is_empty() {
4644 return Ok(());
4645 }
4646
4647 locations.sort_by_key(|location| location.buffer.id());
4648 let mut locations = locations.into_iter().peekable();
4649 let mut ranges_to_highlight = Vec::new();
4650
4651 let excerpt_buffer = cx.add_model(|cx| {
4652 let mut symbol_name = None;
4653 let mut multibuffer = MultiBuffer::new(replica_id);
4654 while let Some(location) = locations.next() {
4655 let buffer = location.buffer.read(cx);
4656 let mut ranges_for_buffer = Vec::new();
4657 let range = location.range.to_offset(buffer);
4658 ranges_for_buffer.push(range.clone());
4659 if symbol_name.is_none() {
4660 symbol_name = Some(buffer.text_for_range(range).collect::<String>());
4661 }
4662
4663 while let Some(next_location) = locations.peek() {
4664 if next_location.buffer == location.buffer {
4665 ranges_for_buffer.push(next_location.range.to_offset(buffer));
4666 locations.next();
4667 } else {
4668 break;
4669 }
4670 }
4671
4672 ranges_for_buffer.sort_by_key(|range| (range.start, Reverse(range.end)));
4673 ranges_to_highlight.extend(multibuffer.push_excerpts_with_context_lines(
4674 location.buffer.clone(),
4675 ranges_for_buffer,
4676 1,
4677 cx,
4678 ));
4679 }
4680 multibuffer.with_title(format!("References to `{}`", symbol_name.unwrap()))
4681 });
4682
4683 workspace.update(&mut cx, |workspace, cx| {
4684 let editor =
4685 cx.add_view(|cx| Editor::for_multibuffer(excerpt_buffer, Some(project), cx));
4686 editor.update(cx, |editor, cx| {
4687 editor.highlight_background::<Self>(
4688 ranges_to_highlight,
4689 |theme| theme.editor.highlighted_line_background,
4690 cx,
4691 );
4692 });
4693 workspace.add_item(Box::new(editor), cx);
4694 });
4695
4696 Ok(())
4697 }))
4698 }
4699
4700 pub fn rename(&mut self, _: &Rename, cx: &mut ViewContext<Self>) -> Option<Task<Result<()>>> {
4701 use language::ToOffset as _;
4702
4703 let project = self.project.clone()?;
4704 let selection = self.selections.newest_anchor().clone();
4705 let (cursor_buffer, cursor_buffer_position) = self
4706 .buffer
4707 .read(cx)
4708 .text_anchor_for_position(selection.head(), cx)?;
4709 let (tail_buffer, _) = self
4710 .buffer
4711 .read(cx)
4712 .text_anchor_for_position(selection.tail(), cx)?;
4713 if tail_buffer != cursor_buffer {
4714 return None;
4715 }
4716
4717 let snapshot = cursor_buffer.read(cx).snapshot();
4718 let cursor_buffer_offset = cursor_buffer_position.to_offset(&snapshot);
4719 let prepare_rename = project.update(cx, |project, cx| {
4720 project.prepare_rename(cursor_buffer, cursor_buffer_offset, cx)
4721 });
4722
4723 Some(cx.spawn(|this, mut cx| async move {
4724 let rename_range = if let Some(range) = prepare_rename.await? {
4725 Some(range)
4726 } else {
4727 this.read_with(&cx, |this, cx| {
4728 let buffer = this.buffer.read(cx).snapshot(cx);
4729 let mut buffer_highlights = this
4730 .document_highlights_for_position(selection.head(), &buffer)
4731 .filter(|highlight| {
4732 highlight.start.excerpt_id() == selection.head().excerpt_id()
4733 && highlight.end.excerpt_id() == selection.head().excerpt_id()
4734 });
4735 buffer_highlights
4736 .next()
4737 .map(|highlight| highlight.start.text_anchor..highlight.end.text_anchor)
4738 })
4739 };
4740 if let Some(rename_range) = rename_range {
4741 let rename_buffer_range = rename_range.to_offset(&snapshot);
4742 let cursor_offset_in_rename_range =
4743 cursor_buffer_offset.saturating_sub(rename_buffer_range.start);
4744
4745 this.update(&mut cx, |this, cx| {
4746 this.take_rename(false, cx);
4747 let style = this.style(cx);
4748 let buffer = this.buffer.read(cx).read(cx);
4749 let cursor_offset = selection.head().to_offset(&buffer);
4750 let rename_start = cursor_offset.saturating_sub(cursor_offset_in_rename_range);
4751 let rename_end = rename_start + rename_buffer_range.len();
4752 let range = buffer.anchor_before(rename_start)..buffer.anchor_after(rename_end);
4753 let mut old_highlight_id = None;
4754 let old_name: Arc<str> = buffer
4755 .chunks(rename_start..rename_end, true)
4756 .map(|chunk| {
4757 if old_highlight_id.is_none() {
4758 old_highlight_id = chunk.syntax_highlight_id;
4759 }
4760 chunk.text
4761 })
4762 .collect::<String>()
4763 .into();
4764
4765 drop(buffer);
4766
4767 // Position the selection in the rename editor so that it matches the current selection.
4768 this.show_local_selections = false;
4769 let rename_editor = cx.add_view(|cx| {
4770 let mut editor = Editor::single_line(None, cx);
4771 if let Some(old_highlight_id) = old_highlight_id {
4772 editor.override_text_style =
4773 Some(Box::new(move |style| old_highlight_id.style(&style.syntax)));
4774 }
4775 editor
4776 .buffer
4777 .update(cx, |buffer, cx| buffer.edit([(0..0, old_name.clone())], cx));
4778 editor.select_all(&SelectAll, cx);
4779 editor
4780 });
4781
4782 let ranges = this
4783 .clear_background_highlights::<DocumentHighlightWrite>(cx)
4784 .into_iter()
4785 .flat_map(|(_, ranges)| ranges)
4786 .chain(
4787 this.clear_background_highlights::<DocumentHighlightRead>(cx)
4788 .into_iter()
4789 .flat_map(|(_, ranges)| ranges),
4790 )
4791 .collect();
4792
4793 this.highlight_text::<Rename>(
4794 ranges,
4795 HighlightStyle {
4796 fade_out: Some(style.rename_fade),
4797 ..Default::default()
4798 },
4799 cx,
4800 );
4801 cx.focus(&rename_editor);
4802 let block_id = this.insert_blocks(
4803 [BlockProperties {
4804 style: BlockStyle::Flex,
4805 position: range.start.clone(),
4806 height: 1,
4807 render: Arc::new({
4808 let editor = rename_editor.clone();
4809 move |cx: &mut BlockContext| {
4810 ChildView::new(editor.clone())
4811 .contained()
4812 .with_padding_left(cx.anchor_x)
4813 .boxed()
4814 }
4815 }),
4816 disposition: BlockDisposition::Below,
4817 }],
4818 cx,
4819 )[0];
4820 this.pending_rename = Some(RenameState {
4821 range,
4822 old_name,
4823 editor: rename_editor,
4824 block_id,
4825 });
4826 });
4827 }
4828
4829 Ok(())
4830 }))
4831 }
4832
4833 pub fn confirm_rename(
4834 workspace: &mut Workspace,
4835 _: &ConfirmRename,
4836 cx: &mut ViewContext<Workspace>,
4837 ) -> Option<Task<Result<()>>> {
4838 let editor = workspace.active_item(cx)?.act_as::<Editor>(cx)?;
4839
4840 let (buffer, range, old_name, new_name) = editor.update(cx, |editor, cx| {
4841 let rename = editor.take_rename(false, cx)?;
4842 let buffer = editor.buffer.read(cx);
4843 let (start_buffer, start) =
4844 buffer.text_anchor_for_position(rename.range.start.clone(), cx)?;
4845 let (end_buffer, end) =
4846 buffer.text_anchor_for_position(rename.range.end.clone(), cx)?;
4847 if start_buffer == end_buffer {
4848 let new_name = rename.editor.read(cx).text(cx);
4849 Some((start_buffer, start..end, rename.old_name, new_name))
4850 } else {
4851 None
4852 }
4853 })?;
4854
4855 let rename = workspace.project().clone().update(cx, |project, cx| {
4856 project.perform_rename(
4857 buffer.clone(),
4858 range.start.clone(),
4859 new_name.clone(),
4860 true,
4861 cx,
4862 )
4863 });
4864
4865 Some(cx.spawn(|workspace, mut cx| async move {
4866 let project_transaction = rename.await?;
4867 Self::open_project_transaction(
4868 editor.clone(),
4869 workspace,
4870 project_transaction,
4871 format!("Rename: {} → {}", old_name, new_name),
4872 cx.clone(),
4873 )
4874 .await?;
4875
4876 editor.update(&mut cx, |editor, cx| {
4877 editor.refresh_document_highlights(cx);
4878 });
4879 Ok(())
4880 }))
4881 }
4882
4883 fn take_rename(
4884 &mut self,
4885 moving_cursor: bool,
4886 cx: &mut ViewContext<Self>,
4887 ) -> Option<RenameState> {
4888 let rename = self.pending_rename.take()?;
4889 self.remove_blocks([rename.block_id].into_iter().collect(), cx);
4890 self.clear_text_highlights::<Rename>(cx);
4891 self.show_local_selections = true;
4892
4893 if moving_cursor {
4894 let rename_editor = rename.editor.read(cx);
4895 let cursor_in_rename_editor = rename_editor.selections.newest::<usize>(cx).head();
4896
4897 // Update the selection to match the position of the selection inside
4898 // the rename editor.
4899 let snapshot = self.buffer.read(cx).read(cx);
4900 let rename_range = rename.range.to_offset(&snapshot);
4901 let cursor_in_editor = snapshot
4902 .clip_offset(rename_range.start + cursor_in_rename_editor, Bias::Left)
4903 .min(rename_range.end);
4904 drop(snapshot);
4905
4906 self.change_selections(None, cx, |s| {
4907 s.select_ranges(vec![cursor_in_editor..cursor_in_editor])
4908 });
4909 }
4910
4911 Some(rename)
4912 }
4913
4914 #[cfg(any(test, feature = "test-support"))]
4915 pub fn pending_rename(&self) -> Option<&RenameState> {
4916 self.pending_rename.as_ref()
4917 }
4918
4919 fn restart_language_server(&mut self, _: &RestartLanguageServer, cx: &mut ViewContext<Self>) {
4920 if let Some(project) = self.project.clone() {
4921 self.buffer.update(cx, |multi_buffer, cx| {
4922 project.update(cx, |project, cx| {
4923 project.restart_language_servers_for_buffers(multi_buffer.all_buffers(), cx);
4924 });
4925 })
4926 }
4927 }
4928
4929 fn refresh_active_diagnostics(&mut self, cx: &mut ViewContext<Editor>) {
4930 if let Some(active_diagnostics) = self.active_diagnostics.as_mut() {
4931 let buffer = self.buffer.read(cx).snapshot(cx);
4932 let primary_range_start = active_diagnostics.primary_range.start.to_offset(&buffer);
4933 let is_valid = buffer
4934 .diagnostics_in_range::<_, usize>(active_diagnostics.primary_range.clone(), false)
4935 .any(|entry| {
4936 entry.diagnostic.is_primary
4937 && !entry.range.is_empty()
4938 && entry.range.start == primary_range_start
4939 && entry.diagnostic.message == active_diagnostics.primary_message
4940 });
4941
4942 if is_valid != active_diagnostics.is_valid {
4943 active_diagnostics.is_valid = is_valid;
4944 let mut new_styles = HashMap::default();
4945 for (block_id, diagnostic) in &active_diagnostics.blocks {
4946 new_styles.insert(
4947 *block_id,
4948 diagnostic_block_renderer(diagnostic.clone(), is_valid),
4949 );
4950 }
4951 self.display_map
4952 .update(cx, |display_map, _| display_map.replace_blocks(new_styles));
4953 }
4954 }
4955 }
4956
4957 fn activate_diagnostics(&mut self, group_id: usize, cx: &mut ViewContext<Self>) {
4958 self.dismiss_diagnostics(cx);
4959 self.active_diagnostics = self.display_map.update(cx, |display_map, cx| {
4960 let buffer = self.buffer.read(cx).snapshot(cx);
4961
4962 let mut primary_range = None;
4963 let mut primary_message = None;
4964 let mut group_end = Point::zero();
4965 let diagnostic_group = buffer
4966 .diagnostic_group::<Point>(group_id)
4967 .map(|entry| {
4968 if entry.range.end > group_end {
4969 group_end = entry.range.end;
4970 }
4971 if entry.diagnostic.is_primary {
4972 primary_range = Some(entry.range.clone());
4973 primary_message = Some(entry.diagnostic.message.clone());
4974 }
4975 entry
4976 })
4977 .collect::<Vec<_>>();
4978 let primary_range = primary_range.unwrap();
4979 let primary_message = primary_message.unwrap();
4980 let primary_range =
4981 buffer.anchor_after(primary_range.start)..buffer.anchor_before(primary_range.end);
4982
4983 let blocks = display_map
4984 .insert_blocks(
4985 diagnostic_group.iter().map(|entry| {
4986 let diagnostic = entry.diagnostic.clone();
4987 let message_height = diagnostic.message.lines().count() as u8;
4988 BlockProperties {
4989 style: BlockStyle::Fixed,
4990 position: buffer.anchor_after(entry.range.start),
4991 height: message_height,
4992 render: diagnostic_block_renderer(diagnostic, true),
4993 disposition: BlockDisposition::Below,
4994 }
4995 }),
4996 cx,
4997 )
4998 .into_iter()
4999 .zip(diagnostic_group.into_iter().map(|entry| entry.diagnostic))
5000 .collect();
5001
5002 Some(ActiveDiagnosticGroup {
5003 primary_range,
5004 primary_message,
5005 blocks,
5006 is_valid: true,
5007 })
5008 });
5009 }
5010
5011 fn dismiss_diagnostics(&mut self, cx: &mut ViewContext<Self>) {
5012 if let Some(active_diagnostic_group) = self.active_diagnostics.take() {
5013 self.display_map.update(cx, |display_map, cx| {
5014 display_map.remove_blocks(active_diagnostic_group.blocks.into_keys().collect(), cx);
5015 });
5016 cx.notify();
5017 }
5018 }
5019
5020 pub fn set_selections_from_remote(
5021 &mut self,
5022 selections: Vec<Selection<Anchor>>,
5023 cx: &mut ViewContext<Self>,
5024 ) {
5025 let old_cursor_position = self.selections.newest_anchor().head();
5026 self.selections.change_with(cx, |s| {
5027 s.select_anchors(selections);
5028 });
5029 self.selections_did_change(false, &old_cursor_position, cx);
5030 }
5031
5032 fn push_to_selection_history(&mut self) {
5033 self.selection_history.push(SelectionHistoryEntry {
5034 selections: self.selections.disjoint_anchors().clone(),
5035 select_next_state: self.select_next_state.clone(),
5036 add_selections_state: self.add_selections_state.clone(),
5037 });
5038 }
5039
5040 pub fn request_autoscroll(&mut self, autoscroll: Autoscroll, cx: &mut ViewContext<Self>) {
5041 self.autoscroll_request = Some((autoscroll, true));
5042 cx.notify();
5043 }
5044
5045 fn request_autoscroll_remotely(&mut self, autoscroll: Autoscroll, cx: &mut ViewContext<Self>) {
5046 self.autoscroll_request = Some((autoscroll, false));
5047 cx.notify();
5048 }
5049
5050 pub fn transact(
5051 &mut self,
5052 cx: &mut ViewContext<Self>,
5053 update: impl FnOnce(&mut Self, &mut ViewContext<Self>),
5054 ) {
5055 self.start_transaction_at(Instant::now(), cx);
5056 update(self, cx);
5057 self.end_transaction_at(Instant::now(), cx);
5058 }
5059
5060 fn start_transaction_at(&mut self, now: Instant, cx: &mut ViewContext<Self>) {
5061 self.end_selection(cx);
5062 if let Some(tx_id) = self
5063 .buffer
5064 .update(cx, |buffer, cx| buffer.start_transaction_at(now, cx))
5065 {
5066 self.selection_history
5067 .insert_transaction(tx_id, self.selections.disjoint_anchors().clone());
5068 }
5069 }
5070
5071 fn end_transaction_at(&mut self, now: Instant, cx: &mut ViewContext<Self>) {
5072 if let Some(tx_id) = self
5073 .buffer
5074 .update(cx, |buffer, cx| buffer.end_transaction_at(now, cx))
5075 {
5076 if let Some((_, end_selections)) = self.selection_history.transaction_mut(tx_id) {
5077 *end_selections = Some(self.selections.disjoint_anchors().clone());
5078 } else {
5079 log::error!("unexpectedly ended a transaction that wasn't started by this editor");
5080 }
5081
5082 cx.emit(Event::Edited);
5083 }
5084 }
5085
5086 pub fn page_up(&mut self, _: &PageUp, _: &mut ViewContext<Self>) {
5087 log::info!("Editor::page_up");
5088 }
5089
5090 pub fn page_down(&mut self, _: &PageDown, _: &mut ViewContext<Self>) {
5091 log::info!("Editor::page_down");
5092 }
5093
5094 pub fn fold(&mut self, _: &Fold, cx: &mut ViewContext<Self>) {
5095 let mut fold_ranges = Vec::new();
5096
5097 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5098 let selections = self.selections.all::<Point>(cx);
5099 for selection in selections {
5100 let range = selection.display_range(&display_map).sorted();
5101 let buffer_start_row = range.start.to_point(&display_map).row;
5102
5103 for row in (0..=range.end.row()).rev() {
5104 if self.is_line_foldable(&display_map, row) && !display_map.is_line_folded(row) {
5105 let fold_range = self.foldable_range_for_line(&display_map, row);
5106 if fold_range.end.row >= buffer_start_row {
5107 fold_ranges.push(fold_range);
5108 if row <= range.start.row() {
5109 break;
5110 }
5111 }
5112 }
5113 }
5114 }
5115
5116 self.fold_ranges(fold_ranges, cx);
5117 }
5118
5119 pub fn unfold_lines(&mut self, _: &UnfoldLines, cx: &mut ViewContext<Self>) {
5120 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5121 let buffer = &display_map.buffer_snapshot;
5122 let selections = self.selections.all::<Point>(cx);
5123 let ranges = selections
5124 .iter()
5125 .map(|s| {
5126 let range = s.display_range(&display_map).sorted();
5127 let mut start = range.start.to_point(&display_map);
5128 let mut end = range.end.to_point(&display_map);
5129 start.column = 0;
5130 end.column = buffer.line_len(end.row);
5131 start..end
5132 })
5133 .collect::<Vec<_>>();
5134 self.unfold_ranges(ranges, true, cx);
5135 }
5136
5137 fn is_line_foldable(&self, display_map: &DisplaySnapshot, display_row: u32) -> bool {
5138 let max_point = display_map.max_point();
5139 if display_row >= max_point.row() {
5140 false
5141 } else {
5142 let (start_indent, is_blank) = display_map.line_indent(display_row);
5143 if is_blank {
5144 false
5145 } else {
5146 for display_row in display_row + 1..=max_point.row() {
5147 let (indent, is_blank) = display_map.line_indent(display_row);
5148 if !is_blank {
5149 return indent > start_indent;
5150 }
5151 }
5152 false
5153 }
5154 }
5155 }
5156
5157 fn foldable_range_for_line(
5158 &self,
5159 display_map: &DisplaySnapshot,
5160 start_row: u32,
5161 ) -> Range<Point> {
5162 let max_point = display_map.max_point();
5163
5164 let (start_indent, _) = display_map.line_indent(start_row);
5165 let start = DisplayPoint::new(start_row, display_map.line_len(start_row));
5166 let mut end = None;
5167 for row in start_row + 1..=max_point.row() {
5168 let (indent, is_blank) = display_map.line_indent(row);
5169 if !is_blank && indent <= start_indent {
5170 end = Some(DisplayPoint::new(row - 1, display_map.line_len(row - 1)));
5171 break;
5172 }
5173 }
5174
5175 let end = end.unwrap_or(max_point);
5176 return start.to_point(display_map)..end.to_point(display_map);
5177 }
5178
5179 pub fn fold_selected_ranges(&mut self, _: &FoldSelectedRanges, cx: &mut ViewContext<Self>) {
5180 let selections = self.selections.all::<Point>(cx);
5181 let ranges = selections.into_iter().map(|s| s.start..s.end);
5182 self.fold_ranges(ranges, cx);
5183 }
5184
5185 pub fn fold_ranges<T: ToOffset>(
5186 &mut self,
5187 ranges: impl IntoIterator<Item = Range<T>>,
5188 cx: &mut ViewContext<Self>,
5189 ) {
5190 let mut ranges = ranges.into_iter().peekable();
5191 if ranges.peek().is_some() {
5192 self.display_map.update(cx, |map, cx| map.fold(ranges, cx));
5193 self.request_autoscroll(Autoscroll::Fit, cx);
5194 cx.notify();
5195 }
5196 }
5197
5198 pub fn unfold_ranges<T: ToOffset>(
5199 &mut self,
5200 ranges: impl IntoIterator<Item = Range<T>>,
5201 inclusive: bool,
5202 cx: &mut ViewContext<Self>,
5203 ) {
5204 let mut ranges = ranges.into_iter().peekable();
5205 if ranges.peek().is_some() {
5206 self.display_map
5207 .update(cx, |map, cx| map.unfold(ranges, inclusive, cx));
5208 self.request_autoscroll(Autoscroll::Fit, cx);
5209 cx.notify();
5210 }
5211 }
5212
5213 pub fn insert_blocks(
5214 &mut self,
5215 blocks: impl IntoIterator<Item = BlockProperties<Anchor>>,
5216 cx: &mut ViewContext<Self>,
5217 ) -> Vec<BlockId> {
5218 let blocks = self
5219 .display_map
5220 .update(cx, |display_map, cx| display_map.insert_blocks(blocks, cx));
5221 self.request_autoscroll(Autoscroll::Fit, cx);
5222 blocks
5223 }
5224
5225 pub fn replace_blocks(
5226 &mut self,
5227 blocks: HashMap<BlockId, RenderBlock>,
5228 cx: &mut ViewContext<Self>,
5229 ) {
5230 self.display_map
5231 .update(cx, |display_map, _| display_map.replace_blocks(blocks));
5232 self.request_autoscroll(Autoscroll::Fit, cx);
5233 }
5234
5235 pub fn remove_blocks(&mut self, block_ids: HashSet<BlockId>, cx: &mut ViewContext<Self>) {
5236 self.display_map.update(cx, |display_map, cx| {
5237 display_map.remove_blocks(block_ids, cx)
5238 });
5239 }
5240
5241 pub fn longest_row(&self, cx: &mut MutableAppContext) -> u32 {
5242 self.display_map
5243 .update(cx, |map, cx| map.snapshot(cx))
5244 .longest_row()
5245 }
5246
5247 pub fn max_point(&self, cx: &mut MutableAppContext) -> DisplayPoint {
5248 self.display_map
5249 .update(cx, |map, cx| map.snapshot(cx))
5250 .max_point()
5251 }
5252
5253 pub fn text(&self, cx: &AppContext) -> String {
5254 self.buffer.read(cx).read(cx).text()
5255 }
5256
5257 pub fn set_text(&mut self, text: impl Into<Arc<str>>, cx: &mut ViewContext<Self>) {
5258 self.transact(cx, |this, cx| {
5259 this.buffer
5260 .read(cx)
5261 .as_singleton()
5262 .expect("you can only call set_text on editors for singleton buffers")
5263 .update(cx, |buffer, cx| buffer.set_text(text, cx));
5264 });
5265 }
5266
5267 pub fn display_text(&self, cx: &mut MutableAppContext) -> String {
5268 self.display_map
5269 .update(cx, |map, cx| map.snapshot(cx))
5270 .text()
5271 }
5272
5273 pub fn soft_wrap_mode(&self, cx: &AppContext) -> SoftWrap {
5274 let language_name = self
5275 .buffer
5276 .read(cx)
5277 .as_singleton()
5278 .and_then(|singleton_buffer| singleton_buffer.read(cx).language())
5279 .map(|l| l.name());
5280
5281 let settings = cx.global::<Settings>();
5282 let mode = self
5283 .soft_wrap_mode_override
5284 .unwrap_or_else(|| settings.soft_wrap(language_name.as_deref()));
5285 match mode {
5286 settings::SoftWrap::None => SoftWrap::None,
5287 settings::SoftWrap::EditorWidth => SoftWrap::EditorWidth,
5288 settings::SoftWrap::PreferredLineLength => {
5289 SoftWrap::Column(settings.preferred_line_length(language_name.as_deref()))
5290 }
5291 }
5292 }
5293
5294 pub fn set_soft_wrap_mode(&mut self, mode: settings::SoftWrap, cx: &mut ViewContext<Self>) {
5295 self.soft_wrap_mode_override = Some(mode);
5296 cx.notify();
5297 }
5298
5299 pub fn set_wrap_width(&self, width: Option<f32>, cx: &mut MutableAppContext) -> bool {
5300 self.display_map
5301 .update(cx, |map, cx| map.set_wrap_width(width, cx))
5302 }
5303
5304 pub fn highlight_rows(&mut self, rows: Option<Range<u32>>) {
5305 self.highlighted_rows = rows;
5306 }
5307
5308 pub fn highlighted_rows(&self) -> Option<Range<u32>> {
5309 self.highlighted_rows.clone()
5310 }
5311
5312 pub fn highlight_background<T: 'static>(
5313 &mut self,
5314 ranges: Vec<Range<Anchor>>,
5315 color_fetcher: fn(&Theme) -> Color,
5316 cx: &mut ViewContext<Self>,
5317 ) {
5318 self.background_highlights
5319 .insert(TypeId::of::<T>(), (color_fetcher, ranges));
5320 cx.notify();
5321 }
5322
5323 pub fn clear_background_highlights<T: 'static>(
5324 &mut self,
5325 cx: &mut ViewContext<Self>,
5326 ) -> Option<(fn(&Theme) -> Color, Vec<Range<Anchor>>)> {
5327 cx.notify();
5328 self.background_highlights.remove(&TypeId::of::<T>())
5329 }
5330
5331 #[cfg(feature = "test-support")]
5332 pub fn all_background_highlights(
5333 &mut self,
5334 cx: &mut ViewContext<Self>,
5335 ) -> Vec<(Range<DisplayPoint>, Color)> {
5336 let snapshot = self.snapshot(cx);
5337 let buffer = &snapshot.buffer_snapshot;
5338 let start = buffer.anchor_before(0);
5339 let end = buffer.anchor_after(buffer.len());
5340 let theme = cx.global::<Settings>().theme.as_ref();
5341 self.background_highlights_in_range(start..end, &snapshot, theme)
5342 }
5343
5344 fn document_highlights_for_position<'a>(
5345 &'a self,
5346 position: Anchor,
5347 buffer: &'a MultiBufferSnapshot,
5348 ) -> impl 'a + Iterator<Item = &Range<Anchor>> {
5349 let read_highlights = self
5350 .background_highlights
5351 .get(&TypeId::of::<DocumentHighlightRead>())
5352 .map(|h| &h.1);
5353 let write_highlights = self
5354 .background_highlights
5355 .get(&TypeId::of::<DocumentHighlightRead>())
5356 .map(|h| &h.1);
5357 let left_position = position.bias_left(buffer);
5358 let right_position = position.bias_right(buffer);
5359 read_highlights
5360 .into_iter()
5361 .chain(write_highlights)
5362 .flat_map(move |ranges| {
5363 let start_ix = match ranges.binary_search_by(|probe| {
5364 let cmp = probe.end.cmp(&left_position, &buffer);
5365 if cmp.is_ge() {
5366 Ordering::Greater
5367 } else {
5368 Ordering::Less
5369 }
5370 }) {
5371 Ok(i) | Err(i) => i,
5372 };
5373
5374 let right_position = right_position.clone();
5375 ranges[start_ix..]
5376 .iter()
5377 .take_while(move |range| range.start.cmp(&right_position, &buffer).is_le())
5378 })
5379 }
5380
5381 pub fn background_highlights_in_range(
5382 &self,
5383 search_range: Range<Anchor>,
5384 display_snapshot: &DisplaySnapshot,
5385 theme: &Theme,
5386 ) -> Vec<(Range<DisplayPoint>, Color)> {
5387 let mut results = Vec::new();
5388 let buffer = &display_snapshot.buffer_snapshot;
5389 for (color_fetcher, ranges) in self.background_highlights.values() {
5390 let color = color_fetcher(theme);
5391 let start_ix = match ranges.binary_search_by(|probe| {
5392 let cmp = probe.end.cmp(&search_range.start, &buffer);
5393 if cmp.is_gt() {
5394 Ordering::Greater
5395 } else {
5396 Ordering::Less
5397 }
5398 }) {
5399 Ok(i) | Err(i) => i,
5400 };
5401 for range in &ranges[start_ix..] {
5402 if range.start.cmp(&search_range.end, &buffer).is_ge() {
5403 break;
5404 }
5405 let start = range
5406 .start
5407 .to_point(buffer)
5408 .to_display_point(display_snapshot);
5409 let end = range
5410 .end
5411 .to_point(buffer)
5412 .to_display_point(display_snapshot);
5413 results.push((start..end, color))
5414 }
5415 }
5416 results
5417 }
5418
5419 pub fn highlight_text<T: 'static>(
5420 &mut self,
5421 ranges: Vec<Range<Anchor>>,
5422 style: HighlightStyle,
5423 cx: &mut ViewContext<Self>,
5424 ) {
5425 self.display_map.update(cx, |map, _| {
5426 map.highlight_text(TypeId::of::<T>(), ranges, style)
5427 });
5428 cx.notify();
5429 }
5430
5431 pub fn clear_text_highlights<T: 'static>(
5432 &mut self,
5433 cx: &mut ViewContext<Self>,
5434 ) -> Option<Arc<(HighlightStyle, Vec<Range<Anchor>>)>> {
5435 cx.notify();
5436 self.display_map
5437 .update(cx, |map, _| map.clear_text_highlights(TypeId::of::<T>()))
5438 }
5439
5440 fn next_blink_epoch(&mut self) -> usize {
5441 self.blink_epoch += 1;
5442 self.blink_epoch
5443 }
5444
5445 fn pause_cursor_blinking(&mut self, cx: &mut ViewContext<Self>) {
5446 if !self.focused {
5447 return;
5448 }
5449
5450 self.show_local_cursors = true;
5451 cx.notify();
5452
5453 let epoch = self.next_blink_epoch();
5454 cx.spawn(|this, mut cx| {
5455 let this = this.downgrade();
5456 async move {
5457 Timer::after(CURSOR_BLINK_INTERVAL).await;
5458 if let Some(this) = this.upgrade(&cx) {
5459 this.update(&mut cx, |this, cx| this.resume_cursor_blinking(epoch, cx))
5460 }
5461 }
5462 })
5463 .detach();
5464 }
5465
5466 fn resume_cursor_blinking(&mut self, epoch: usize, cx: &mut ViewContext<Self>) {
5467 if epoch == self.blink_epoch {
5468 self.blinking_paused = false;
5469 self.blink_cursors(epoch, cx);
5470 }
5471 }
5472
5473 fn blink_cursors(&mut self, epoch: usize, cx: &mut ViewContext<Self>) {
5474 if epoch == self.blink_epoch && self.focused && !self.blinking_paused {
5475 self.show_local_cursors = !self.show_local_cursors;
5476 cx.notify();
5477
5478 let epoch = self.next_blink_epoch();
5479 cx.spawn(|this, mut cx| {
5480 let this = this.downgrade();
5481 async move {
5482 Timer::after(CURSOR_BLINK_INTERVAL).await;
5483 if let Some(this) = this.upgrade(&cx) {
5484 this.update(&mut cx, |this, cx| this.blink_cursors(epoch, cx));
5485 }
5486 }
5487 })
5488 .detach();
5489 }
5490 }
5491
5492 pub fn show_local_cursors(&self) -> bool {
5493 self.show_local_cursors && self.focused
5494 }
5495
5496 fn on_buffer_changed(&mut self, _: ModelHandle<MultiBuffer>, cx: &mut ViewContext<Self>) {
5497 cx.notify();
5498 }
5499
5500 fn on_buffer_event(
5501 &mut self,
5502 _: ModelHandle<MultiBuffer>,
5503 event: &language::Event,
5504 cx: &mut ViewContext<Self>,
5505 ) {
5506 match event {
5507 language::Event::Edited => {
5508 self.refresh_active_diagnostics(cx);
5509 self.refresh_code_actions(cx);
5510 cx.emit(Event::BufferEdited);
5511 }
5512 language::Event::Reparsed => cx.emit(Event::Reparsed),
5513 language::Event::Dirtied => cx.emit(Event::Dirtied),
5514 language::Event::Saved => cx.emit(Event::Saved),
5515 language::Event::FileHandleChanged => cx.emit(Event::TitleChanged),
5516 language::Event::Reloaded => cx.emit(Event::TitleChanged),
5517 language::Event::Closed => cx.emit(Event::Closed),
5518 language::Event::DiagnosticsUpdated => {
5519 self.refresh_active_diagnostics(cx);
5520 }
5521 _ => {}
5522 }
5523 }
5524
5525 fn on_display_map_changed(&mut self, _: ModelHandle<DisplayMap>, cx: &mut ViewContext<Self>) {
5526 cx.notify();
5527 }
5528
5529 pub fn set_searchable(&mut self, searchable: bool) {
5530 self.searchable = searchable;
5531 }
5532
5533 pub fn searchable(&self) -> bool {
5534 self.searchable
5535 }
5536
5537 fn open_excerpts(workspace: &mut Workspace, _: &OpenExcerpts, cx: &mut ViewContext<Workspace>) {
5538 let active_item = workspace.active_item(cx);
5539 let editor_handle = if let Some(editor) = active_item
5540 .as_ref()
5541 .and_then(|item| item.act_as::<Self>(cx))
5542 {
5543 editor
5544 } else {
5545 cx.propagate_action();
5546 return;
5547 };
5548
5549 let editor = editor_handle.read(cx);
5550 let buffer = editor.buffer.read(cx);
5551 if buffer.is_singleton() {
5552 cx.propagate_action();
5553 return;
5554 }
5555
5556 let mut new_selections_by_buffer = HashMap::default();
5557 for selection in editor.selections.all::<usize>(cx) {
5558 for (buffer, mut range) in
5559 buffer.range_to_buffer_ranges(selection.start..selection.end, cx)
5560 {
5561 if selection.reversed {
5562 mem::swap(&mut range.start, &mut range.end);
5563 }
5564 new_selections_by_buffer
5565 .entry(buffer)
5566 .or_insert(Vec::new())
5567 .push(range)
5568 }
5569 }
5570
5571 editor_handle.update(cx, |editor, cx| {
5572 editor.push_to_nav_history(editor.selections.newest_anchor().head(), None, cx);
5573 });
5574 let nav_history = workspace.active_pane().read(cx).nav_history().clone();
5575 nav_history.borrow_mut().disable();
5576
5577 // We defer the pane interaction because we ourselves are a workspace item
5578 // and activating a new item causes the pane to call a method on us reentrantly,
5579 // which panics if we're on the stack.
5580 cx.defer(move |workspace, cx| {
5581 for (buffer, ranges) in new_selections_by_buffer.into_iter() {
5582 let editor = workspace.open_project_item::<Self>(buffer, cx);
5583 editor.update(cx, |editor, cx| {
5584 editor.change_selections(Some(Autoscroll::Newest), cx, |s| {
5585 s.select_ranges(ranges);
5586 });
5587 });
5588 }
5589
5590 nav_history.borrow_mut().enable();
5591 });
5592 }
5593
5594 fn jump(workspace: &mut Workspace, action: &Jump, cx: &mut ViewContext<Workspace>) {
5595 let editor = workspace.open_path(action.path.clone(), true, cx);
5596 let position = action.position;
5597 let anchor = action.anchor;
5598 cx.spawn_weak(|_, mut cx| async move {
5599 let editor = editor.await.log_err()?.downcast::<Editor>()?;
5600 editor.update(&mut cx, |editor, cx| {
5601 let buffer = editor.buffer().read(cx).as_singleton()?;
5602 let buffer = buffer.read(cx);
5603 let cursor = if buffer.can_resolve(&anchor) {
5604 language::ToPoint::to_point(&anchor, buffer)
5605 } else {
5606 buffer.clip_point(position, Bias::Left)
5607 };
5608
5609 let nav_history = editor.nav_history.take();
5610 editor.change_selections(Some(Autoscroll::Newest), cx, |s| {
5611 s.select_ranges([cursor..cursor]);
5612 });
5613 editor.nav_history = nav_history;
5614
5615 Some(())
5616 })?;
5617 Some(())
5618 })
5619 .detach()
5620 }
5621}
5622
5623impl EditorSnapshot {
5624 pub fn is_focused(&self) -> bool {
5625 self.is_focused
5626 }
5627
5628 pub fn placeholder_text(&self) -> Option<&Arc<str>> {
5629 self.placeholder_text.as_ref()
5630 }
5631
5632 pub fn scroll_position(&self) -> Vector2F {
5633 compute_scroll_position(
5634 &self.display_snapshot,
5635 self.scroll_position,
5636 &self.scroll_top_anchor,
5637 )
5638 }
5639}
5640
5641impl Deref for EditorSnapshot {
5642 type Target = DisplaySnapshot;
5643
5644 fn deref(&self) -> &Self::Target {
5645 &self.display_snapshot
5646 }
5647}
5648
5649fn compute_scroll_position(
5650 snapshot: &DisplaySnapshot,
5651 mut scroll_position: Vector2F,
5652 scroll_top_anchor: &Anchor,
5653) -> Vector2F {
5654 if *scroll_top_anchor != Anchor::min() {
5655 let scroll_top = scroll_top_anchor.to_display_point(snapshot).row() as f32;
5656 scroll_position.set_y(scroll_top + scroll_position.y());
5657 } else {
5658 scroll_position.set_y(0.);
5659 }
5660 scroll_position
5661}
5662
5663#[derive(Copy, Clone, Debug, PartialEq, Eq)]
5664pub enum Event {
5665 Activate,
5666 BufferEdited,
5667 Edited,
5668 Reparsed,
5669 Blurred,
5670 Dirtied,
5671 Saved,
5672 TitleChanged,
5673 SelectionsChanged { local: bool },
5674 ScrollPositionChanged { local: bool },
5675 Closed,
5676}
5677
5678pub struct EditorFocused(pub ViewHandle<Editor>);
5679pub struct EditorBlurred(pub ViewHandle<Editor>);
5680pub struct EditorReleased(pub WeakViewHandle<Editor>);
5681
5682impl Entity for Editor {
5683 type Event = Event;
5684
5685 fn release(&mut self, cx: &mut MutableAppContext) {
5686 cx.emit_global(EditorReleased(self.handle.clone()));
5687 }
5688}
5689
5690impl View for Editor {
5691 fn render(&mut self, cx: &mut RenderContext<Self>) -> ElementBox {
5692 let style = self.style(cx);
5693 let font_changed = self.display_map.update(cx, |map, cx| {
5694 map.set_font(style.text.font_id, style.text.font_size, cx)
5695 });
5696
5697 // If the
5698 if font_changed {
5699 let handle = self.handle.clone();
5700 cx.defer(move |cx| {
5701 if let Some(editor) = handle.upgrade(cx) {
5702 editor.update(cx, |editor, cx| {
5703 hide_hover(editor, cx);
5704 })
5705 }
5706 });
5707 }
5708
5709 EditorElement::new(self.handle.clone(), style.clone(), self.cursor_shape).boxed()
5710 }
5711
5712 fn ui_name() -> &'static str {
5713 "Editor"
5714 }
5715
5716 fn on_focus(&mut self, cx: &mut ViewContext<Self>) {
5717 let focused_event = EditorFocused(cx.handle());
5718 cx.emit_global(focused_event);
5719 if let Some(rename) = self.pending_rename.as_ref() {
5720 cx.focus(&rename.editor);
5721 } else {
5722 self.focused = true;
5723 self.blink_cursors(self.blink_epoch, cx);
5724 self.buffer.update(cx, |buffer, cx| {
5725 buffer.finalize_last_transaction(cx);
5726 if self.leader_replica_id.is_none() {
5727 buffer.set_active_selections(
5728 &self.selections.disjoint_anchors(),
5729 self.selections.line_mode,
5730 cx,
5731 );
5732 }
5733 });
5734 }
5735 }
5736
5737 fn on_blur(&mut self, cx: &mut ViewContext<Self>) {
5738 let blurred_event = EditorBlurred(cx.handle());
5739 cx.emit_global(blurred_event);
5740 self.focused = false;
5741 self.buffer
5742 .update(cx, |buffer, cx| buffer.remove_active_selections(cx));
5743 self.hide_context_menu(cx);
5744 cx.emit(Event::Blurred);
5745 cx.notify();
5746 }
5747
5748 fn keymap_context(&self, _: &AppContext) -> gpui::keymap::Context {
5749 let mut context = Self::default_keymap_context();
5750 let mode = match self.mode {
5751 EditorMode::SingleLine => "single_line",
5752 EditorMode::AutoHeight { .. } => "auto_height",
5753 EditorMode::Full => "full",
5754 };
5755 context.map.insert("mode".into(), mode.into());
5756 if self.pending_rename.is_some() {
5757 context.set.insert("renaming".into());
5758 }
5759 match self.context_menu.as_ref() {
5760 Some(ContextMenu::Completions(_)) => {
5761 context.set.insert("showing_completions".into());
5762 }
5763 Some(ContextMenu::CodeActions(_)) => {
5764 context.set.insert("showing_code_actions".into());
5765 }
5766 None => {}
5767 }
5768
5769 for layer in self.keymap_context_layers.values() {
5770 context.extend(layer);
5771 }
5772
5773 context
5774 }
5775}
5776
5777fn build_style(
5778 settings: &Settings,
5779 get_field_editor_theme: Option<GetFieldEditorTheme>,
5780 override_text_style: Option<&OverrideTextStyle>,
5781 cx: &AppContext,
5782) -> EditorStyle {
5783 let font_cache = cx.font_cache();
5784
5785 let mut theme = settings.theme.editor.clone();
5786 let mut style = if let Some(get_field_editor_theme) = get_field_editor_theme {
5787 let field_editor_theme = get_field_editor_theme(&settings.theme);
5788 theme.text_color = field_editor_theme.text.color;
5789 theme.selection = field_editor_theme.selection;
5790 theme.background = field_editor_theme
5791 .container
5792 .background_color
5793 .unwrap_or_default();
5794 EditorStyle {
5795 text: field_editor_theme.text,
5796 placeholder_text: field_editor_theme.placeholder_text,
5797 theme,
5798 }
5799 } else {
5800 let font_family_id = settings.buffer_font_family;
5801 let font_family_name = cx.font_cache().family_name(font_family_id).unwrap();
5802 let font_properties = Default::default();
5803 let font_id = font_cache
5804 .select_font(font_family_id, &font_properties)
5805 .unwrap();
5806 let font_size = settings.buffer_font_size;
5807 EditorStyle {
5808 text: TextStyle {
5809 color: settings.theme.editor.text_color,
5810 font_family_name,
5811 font_family_id,
5812 font_id,
5813 font_size,
5814 font_properties,
5815 underline: Default::default(),
5816 },
5817 placeholder_text: None,
5818 theme,
5819 }
5820 };
5821
5822 if let Some(highlight_style) = override_text_style.and_then(|build_style| build_style(&style)) {
5823 if let Some(highlighted) = style
5824 .text
5825 .clone()
5826 .highlight(highlight_style, font_cache)
5827 .log_err()
5828 {
5829 style.text = highlighted;
5830 }
5831 }
5832
5833 style
5834}
5835
5836trait SelectionExt {
5837 fn offset_range(&self, buffer: &MultiBufferSnapshot) -> Range<usize>;
5838 fn point_range(&self, buffer: &MultiBufferSnapshot) -> Range<Point>;
5839 fn display_range(&self, map: &DisplaySnapshot) -> Range<DisplayPoint>;
5840 fn spanned_rows(&self, include_end_if_at_line_start: bool, map: &DisplaySnapshot)
5841 -> Range<u32>;
5842}
5843
5844impl<T: ToPoint + ToOffset> SelectionExt for Selection<T> {
5845 fn point_range(&self, buffer: &MultiBufferSnapshot) -> Range<Point> {
5846 let start = self.start.to_point(buffer);
5847 let end = self.end.to_point(buffer);
5848 if self.reversed {
5849 end..start
5850 } else {
5851 start..end
5852 }
5853 }
5854
5855 fn offset_range(&self, buffer: &MultiBufferSnapshot) -> Range<usize> {
5856 let start = self.start.to_offset(buffer);
5857 let end = self.end.to_offset(buffer);
5858 if self.reversed {
5859 end..start
5860 } else {
5861 start..end
5862 }
5863 }
5864
5865 fn display_range(&self, map: &DisplaySnapshot) -> Range<DisplayPoint> {
5866 let start = self
5867 .start
5868 .to_point(&map.buffer_snapshot)
5869 .to_display_point(map);
5870 let end = self
5871 .end
5872 .to_point(&map.buffer_snapshot)
5873 .to_display_point(map);
5874 if self.reversed {
5875 end..start
5876 } else {
5877 start..end
5878 }
5879 }
5880
5881 fn spanned_rows(
5882 &self,
5883 include_end_if_at_line_start: bool,
5884 map: &DisplaySnapshot,
5885 ) -> Range<u32> {
5886 let start = self.start.to_point(&map.buffer_snapshot);
5887 let mut end = self.end.to_point(&map.buffer_snapshot);
5888 if !include_end_if_at_line_start && start.row != end.row && end.column == 0 {
5889 end.row -= 1;
5890 }
5891
5892 let buffer_start = map.prev_line_boundary(start).0;
5893 let buffer_end = map.next_line_boundary(end).0;
5894 buffer_start.row..buffer_end.row + 1
5895 }
5896}
5897
5898impl<T: InvalidationRegion> InvalidationStack<T> {
5899 fn invalidate<S>(&mut self, selections: &[Selection<S>], buffer: &MultiBufferSnapshot)
5900 where
5901 S: Clone + ToOffset,
5902 {
5903 while let Some(region) = self.last() {
5904 let all_selections_inside_invalidation_ranges =
5905 if selections.len() == region.ranges().len() {
5906 selections
5907 .iter()
5908 .zip(region.ranges().iter().map(|r| r.to_offset(&buffer)))
5909 .all(|(selection, invalidation_range)| {
5910 let head = selection.head().to_offset(&buffer);
5911 invalidation_range.start <= head && invalidation_range.end >= head
5912 })
5913 } else {
5914 false
5915 };
5916
5917 if all_selections_inside_invalidation_ranges {
5918 break;
5919 } else {
5920 self.pop();
5921 }
5922 }
5923 }
5924}
5925
5926impl<T> Default for InvalidationStack<T> {
5927 fn default() -> Self {
5928 Self(Default::default())
5929 }
5930}
5931
5932impl<T> Deref for InvalidationStack<T> {
5933 type Target = Vec<T>;
5934
5935 fn deref(&self) -> &Self::Target {
5936 &self.0
5937 }
5938}
5939
5940impl<T> DerefMut for InvalidationStack<T> {
5941 fn deref_mut(&mut self) -> &mut Self::Target {
5942 &mut self.0
5943 }
5944}
5945
5946impl InvalidationRegion for BracketPairState {
5947 fn ranges(&self) -> &[Range<Anchor>] {
5948 &self.ranges
5949 }
5950}
5951
5952impl InvalidationRegion for SnippetState {
5953 fn ranges(&self) -> &[Range<Anchor>] {
5954 &self.ranges[self.active_index]
5955 }
5956}
5957
5958impl Deref for EditorStyle {
5959 type Target = theme::Editor;
5960
5961 fn deref(&self) -> &Self::Target {
5962 &self.theme
5963 }
5964}
5965
5966pub fn diagnostic_block_renderer(diagnostic: Diagnostic, is_valid: bool) -> RenderBlock {
5967 let mut highlighted_lines = Vec::new();
5968 for line in diagnostic.message.lines() {
5969 highlighted_lines.push(highlight_diagnostic_message(line));
5970 }
5971
5972 Arc::new(move |cx: &mut BlockContext| {
5973 let settings = cx.global::<Settings>();
5974 let theme = &settings.theme.editor;
5975 let style = diagnostic_style(diagnostic.severity, is_valid, theme);
5976 let font_size = (style.text_scale_factor * settings.buffer_font_size).round();
5977 Flex::column()
5978 .with_children(highlighted_lines.iter().map(|(line, highlights)| {
5979 Label::new(
5980 line.clone(),
5981 style.message.clone().with_font_size(font_size),
5982 )
5983 .with_highlights(highlights.clone())
5984 .contained()
5985 .with_margin_left(cx.anchor_x)
5986 .boxed()
5987 }))
5988 .aligned()
5989 .left()
5990 .boxed()
5991 })
5992}
5993
5994pub fn highlight_diagnostic_message(message: &str) -> (String, Vec<usize>) {
5995 let mut message_without_backticks = String::new();
5996 let mut prev_offset = 0;
5997 let mut inside_block = false;
5998 let mut highlights = Vec::new();
5999 for (match_ix, (offset, _)) in message
6000 .match_indices('`')
6001 .chain([(message.len(), "")])
6002 .enumerate()
6003 {
6004 message_without_backticks.push_str(&message[prev_offset..offset]);
6005 if inside_block {
6006 highlights.extend(prev_offset - match_ix..offset - match_ix);
6007 }
6008
6009 inside_block = !inside_block;
6010 prev_offset = offset + 1;
6011 }
6012
6013 (message_without_backticks, highlights)
6014}
6015
6016pub fn diagnostic_style(
6017 severity: DiagnosticSeverity,
6018 valid: bool,
6019 theme: &theme::Editor,
6020) -> DiagnosticStyle {
6021 match (severity, valid) {
6022 (DiagnosticSeverity::ERROR, true) => theme.error_diagnostic.clone(),
6023 (DiagnosticSeverity::ERROR, false) => theme.invalid_error_diagnostic.clone(),
6024 (DiagnosticSeverity::WARNING, true) => theme.warning_diagnostic.clone(),
6025 (DiagnosticSeverity::WARNING, false) => theme.invalid_warning_diagnostic.clone(),
6026 (DiagnosticSeverity::INFORMATION, true) => theme.information_diagnostic.clone(),
6027 (DiagnosticSeverity::INFORMATION, false) => theme.invalid_information_diagnostic.clone(),
6028 (DiagnosticSeverity::HINT, true) => theme.hint_diagnostic.clone(),
6029 (DiagnosticSeverity::HINT, false) => theme.invalid_hint_diagnostic.clone(),
6030 _ => theme.invalid_hint_diagnostic.clone(),
6031 }
6032}
6033
6034pub fn combine_syntax_and_fuzzy_match_highlights(
6035 text: &str,
6036 default_style: HighlightStyle,
6037 syntax_ranges: impl Iterator<Item = (Range<usize>, HighlightStyle)>,
6038 match_indices: &[usize],
6039) -> Vec<(Range<usize>, HighlightStyle)> {
6040 let mut result = Vec::new();
6041 let mut match_indices = match_indices.iter().copied().peekable();
6042
6043 for (range, mut syntax_highlight) in syntax_ranges.chain([(usize::MAX..0, Default::default())])
6044 {
6045 syntax_highlight.weight = None;
6046
6047 // Add highlights for any fuzzy match characters before the next
6048 // syntax highlight range.
6049 while let Some(&match_index) = match_indices.peek() {
6050 if match_index >= range.start {
6051 break;
6052 }
6053 match_indices.next();
6054 let end_index = char_ix_after(match_index, text);
6055 let mut match_style = default_style;
6056 match_style.weight = Some(fonts::Weight::BOLD);
6057 result.push((match_index..end_index, match_style));
6058 }
6059
6060 if range.start == usize::MAX {
6061 break;
6062 }
6063
6064 // Add highlights for any fuzzy match characters within the
6065 // syntax highlight range.
6066 let mut offset = range.start;
6067 while let Some(&match_index) = match_indices.peek() {
6068 if match_index >= range.end {
6069 break;
6070 }
6071
6072 match_indices.next();
6073 if match_index > offset {
6074 result.push((offset..match_index, syntax_highlight));
6075 }
6076
6077 let mut end_index = char_ix_after(match_index, text);
6078 while let Some(&next_match_index) = match_indices.peek() {
6079 if next_match_index == end_index && next_match_index < range.end {
6080 end_index = char_ix_after(next_match_index, text);
6081 match_indices.next();
6082 } else {
6083 break;
6084 }
6085 }
6086
6087 let mut match_style = syntax_highlight;
6088 match_style.weight = Some(fonts::Weight::BOLD);
6089 result.push((match_index..end_index, match_style));
6090 offset = end_index;
6091 }
6092
6093 if offset < range.end {
6094 result.push((offset..range.end, syntax_highlight));
6095 }
6096 }
6097
6098 fn char_ix_after(ix: usize, text: &str) -> usize {
6099 ix + text[ix..].chars().next().unwrap().len_utf8()
6100 }
6101
6102 result
6103}
6104
6105pub fn styled_runs_for_code_label<'a>(
6106 label: &'a CodeLabel,
6107 syntax_theme: &'a theme::SyntaxTheme,
6108) -> impl 'a + Iterator<Item = (Range<usize>, HighlightStyle)> {
6109 let fade_out = HighlightStyle {
6110 fade_out: Some(0.35),
6111 ..Default::default()
6112 };
6113
6114 let mut prev_end = label.filter_range.end;
6115 label
6116 .runs
6117 .iter()
6118 .enumerate()
6119 .flat_map(move |(ix, (range, highlight_id))| {
6120 let style = if let Some(style) = highlight_id.style(syntax_theme) {
6121 style
6122 } else {
6123 return Default::default();
6124 };
6125 let mut muted_style = style.clone();
6126 muted_style.highlight(fade_out);
6127
6128 let mut runs = SmallVec::<[(Range<usize>, HighlightStyle); 3]>::new();
6129 if range.start >= label.filter_range.end {
6130 if range.start > prev_end {
6131 runs.push((prev_end..range.start, fade_out));
6132 }
6133 runs.push((range.clone(), muted_style));
6134 } else if range.end <= label.filter_range.end {
6135 runs.push((range.clone(), style));
6136 } else {
6137 runs.push((range.start..label.filter_range.end, style));
6138 runs.push((label.filter_range.end..range.end, muted_style));
6139 }
6140 prev_end = cmp::max(prev_end, range.end);
6141
6142 if ix + 1 == label.runs.len() && label.text.len() > prev_end {
6143 runs.push((prev_end..label.text.len(), fade_out));
6144 }
6145
6146 runs
6147 })
6148}
6149
6150#[cfg(test)]
6151mod tests {
6152 use crate::test::{
6153 assert_text_with_selections, build_editor, select_ranges, EditorTestContext,
6154 };
6155
6156 use super::*;
6157 use futures::StreamExt;
6158 use gpui::{
6159 geometry::rect::RectF,
6160 platform::{WindowBounds, WindowOptions},
6161 };
6162 use indoc::indoc;
6163 use language::{FakeLspAdapter, LanguageConfig};
6164 use lsp::FakeLanguageServer;
6165 use project::FakeFs;
6166 use settings::LanguageOverride;
6167 use std::{cell::RefCell, rc::Rc, time::Instant};
6168 use text::Point;
6169 use unindent::Unindent;
6170 use util::{
6171 assert_set_eq,
6172 test::{marked_text_by, marked_text_ranges, marked_text_ranges_by, sample_text},
6173 };
6174 use workspace::{FollowableItem, ItemHandle};
6175
6176 #[gpui::test]
6177 fn test_edit_events(cx: &mut MutableAppContext) {
6178 cx.set_global(Settings::test(cx));
6179 let buffer = cx.add_model(|cx| language::Buffer::new(0, "123456", cx));
6180
6181 let events = Rc::new(RefCell::new(Vec::new()));
6182 let (_, editor1) = cx.add_window(Default::default(), {
6183 let events = events.clone();
6184 |cx| {
6185 cx.subscribe(&cx.handle(), move |_, _, event, _| {
6186 if matches!(event, Event::Edited | Event::BufferEdited | Event::Dirtied) {
6187 events.borrow_mut().push(("editor1", *event));
6188 }
6189 })
6190 .detach();
6191 Editor::for_buffer(buffer.clone(), None, cx)
6192 }
6193 });
6194 let (_, editor2) = cx.add_window(Default::default(), {
6195 let events = events.clone();
6196 |cx| {
6197 cx.subscribe(&cx.handle(), move |_, _, event, _| {
6198 if matches!(event, Event::Edited | Event::BufferEdited | Event::Dirtied) {
6199 events.borrow_mut().push(("editor2", *event));
6200 }
6201 })
6202 .detach();
6203 Editor::for_buffer(buffer.clone(), None, cx)
6204 }
6205 });
6206 assert_eq!(mem::take(&mut *events.borrow_mut()), []);
6207
6208 // Mutating editor 1 will emit an `Edited` event only for that editor.
6209 editor1.update(cx, |editor, cx| editor.insert("X", cx));
6210 assert_eq!(
6211 mem::take(&mut *events.borrow_mut()),
6212 [
6213 ("editor1", Event::Edited),
6214 ("editor1", Event::BufferEdited),
6215 ("editor2", Event::BufferEdited),
6216 ("editor1", Event::Dirtied),
6217 ("editor2", Event::Dirtied)
6218 ]
6219 );
6220
6221 // Mutating editor 2 will emit an `Edited` event only for that editor.
6222 editor2.update(cx, |editor, cx| editor.delete(&Delete, cx));
6223 assert_eq!(
6224 mem::take(&mut *events.borrow_mut()),
6225 [
6226 ("editor2", Event::Edited),
6227 ("editor1", Event::BufferEdited),
6228 ("editor2", Event::BufferEdited),
6229 ]
6230 );
6231
6232 // Undoing on editor 1 will emit an `Edited` event only for that editor.
6233 editor1.update(cx, |editor, cx| editor.undo(&Undo, cx));
6234 assert_eq!(
6235 mem::take(&mut *events.borrow_mut()),
6236 [
6237 ("editor1", Event::Edited),
6238 ("editor1", Event::BufferEdited),
6239 ("editor2", Event::BufferEdited),
6240 ]
6241 );
6242
6243 // Redoing on editor 1 will emit an `Edited` event only for that editor.
6244 editor1.update(cx, |editor, cx| editor.redo(&Redo, cx));
6245 assert_eq!(
6246 mem::take(&mut *events.borrow_mut()),
6247 [
6248 ("editor1", Event::Edited),
6249 ("editor1", Event::BufferEdited),
6250 ("editor2", Event::BufferEdited),
6251 ]
6252 );
6253
6254 // Undoing on editor 2 will emit an `Edited` event only for that editor.
6255 editor2.update(cx, |editor, cx| editor.undo(&Undo, cx));
6256 assert_eq!(
6257 mem::take(&mut *events.borrow_mut()),
6258 [
6259 ("editor2", Event::Edited),
6260 ("editor1", Event::BufferEdited),
6261 ("editor2", Event::BufferEdited),
6262 ]
6263 );
6264
6265 // Redoing on editor 2 will emit an `Edited` event only for that editor.
6266 editor2.update(cx, |editor, cx| editor.redo(&Redo, cx));
6267 assert_eq!(
6268 mem::take(&mut *events.borrow_mut()),
6269 [
6270 ("editor2", Event::Edited),
6271 ("editor1", Event::BufferEdited),
6272 ("editor2", Event::BufferEdited),
6273 ]
6274 );
6275
6276 // No event is emitted when the mutation is a no-op.
6277 editor2.update(cx, |editor, cx| {
6278 editor.change_selections(None, cx, |s| s.select_ranges([0..0]));
6279
6280 editor.backspace(&Backspace, cx);
6281 });
6282 assert_eq!(mem::take(&mut *events.borrow_mut()), []);
6283 }
6284
6285 #[gpui::test]
6286 fn test_undo_redo_with_selection_restoration(cx: &mut MutableAppContext) {
6287 cx.set_global(Settings::test(cx));
6288 let mut now = Instant::now();
6289 let buffer = cx.add_model(|cx| language::Buffer::new(0, "123456", cx));
6290 let group_interval = buffer.read(cx).transaction_group_interval();
6291 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
6292 let (_, editor) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
6293
6294 editor.update(cx, |editor, cx| {
6295 editor.start_transaction_at(now, cx);
6296 editor.change_selections(None, cx, |s| s.select_ranges([2..4]));
6297
6298 editor.insert("cd", cx);
6299 editor.end_transaction_at(now, cx);
6300 assert_eq!(editor.text(cx), "12cd56");
6301 assert_eq!(editor.selections.ranges(cx), vec![4..4]);
6302
6303 editor.start_transaction_at(now, cx);
6304 editor.change_selections(None, cx, |s| s.select_ranges([4..5]));
6305 editor.insert("e", cx);
6306 editor.end_transaction_at(now, cx);
6307 assert_eq!(editor.text(cx), "12cde6");
6308 assert_eq!(editor.selections.ranges(cx), vec![5..5]);
6309
6310 now += group_interval + Duration::from_millis(1);
6311 editor.change_selections(None, cx, |s| s.select_ranges([2..2]));
6312
6313 // Simulate an edit in another editor
6314 buffer.update(cx, |buffer, cx| {
6315 buffer.start_transaction_at(now, cx);
6316 buffer.edit([(0..1, "a")], cx);
6317 buffer.edit([(1..1, "b")], cx);
6318 buffer.end_transaction_at(now, cx);
6319 });
6320
6321 assert_eq!(editor.text(cx), "ab2cde6");
6322 assert_eq!(editor.selections.ranges(cx), vec![3..3]);
6323
6324 // Last transaction happened past the group interval in a different editor.
6325 // Undo it individually and don't restore selections.
6326 editor.undo(&Undo, cx);
6327 assert_eq!(editor.text(cx), "12cde6");
6328 assert_eq!(editor.selections.ranges(cx), vec![2..2]);
6329
6330 // First two transactions happened within the group interval in this editor.
6331 // Undo them together and restore selections.
6332 editor.undo(&Undo, cx);
6333 editor.undo(&Undo, cx); // Undo stack is empty here, so this is a no-op.
6334 assert_eq!(editor.text(cx), "123456");
6335 assert_eq!(editor.selections.ranges(cx), vec![0..0]);
6336
6337 // Redo the first two transactions together.
6338 editor.redo(&Redo, cx);
6339 assert_eq!(editor.text(cx), "12cde6");
6340 assert_eq!(editor.selections.ranges(cx), vec![5..5]);
6341
6342 // Redo the last transaction on its own.
6343 editor.redo(&Redo, cx);
6344 assert_eq!(editor.text(cx), "ab2cde6");
6345 assert_eq!(editor.selections.ranges(cx), vec![6..6]);
6346
6347 // Test empty transactions.
6348 editor.start_transaction_at(now, cx);
6349 editor.end_transaction_at(now, cx);
6350 editor.undo(&Undo, cx);
6351 assert_eq!(editor.text(cx), "12cde6");
6352 });
6353 }
6354
6355 #[gpui::test]
6356 fn test_selection_with_mouse(cx: &mut gpui::MutableAppContext) {
6357 cx.set_global(Settings::test(cx));
6358
6359 let buffer = MultiBuffer::build_simple("aaaaaa\nbbbbbb\ncccccc\nddddddd\n", cx);
6360 let (_, editor) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
6361 editor.update(cx, |view, cx| {
6362 view.begin_selection(DisplayPoint::new(2, 2), false, 1, cx);
6363 });
6364 assert_eq!(
6365 editor.update(cx, |view, cx| view.selections.display_ranges(cx)),
6366 [DisplayPoint::new(2, 2)..DisplayPoint::new(2, 2)]
6367 );
6368
6369 editor.update(cx, |view, cx| {
6370 view.update_selection(DisplayPoint::new(3, 3), 0, Vector2F::zero(), cx);
6371 });
6372
6373 assert_eq!(
6374 editor.update(cx, |view, cx| view.selections.display_ranges(cx)),
6375 [DisplayPoint::new(2, 2)..DisplayPoint::new(3, 3)]
6376 );
6377
6378 editor.update(cx, |view, cx| {
6379 view.update_selection(DisplayPoint::new(1, 1), 0, Vector2F::zero(), cx);
6380 });
6381
6382 assert_eq!(
6383 editor.update(cx, |view, cx| view.selections.display_ranges(cx)),
6384 [DisplayPoint::new(2, 2)..DisplayPoint::new(1, 1)]
6385 );
6386
6387 editor.update(cx, |view, cx| {
6388 view.end_selection(cx);
6389 view.update_selection(DisplayPoint::new(3, 3), 0, Vector2F::zero(), cx);
6390 });
6391
6392 assert_eq!(
6393 editor.update(cx, |view, cx| view.selections.display_ranges(cx)),
6394 [DisplayPoint::new(2, 2)..DisplayPoint::new(1, 1)]
6395 );
6396
6397 editor.update(cx, |view, cx| {
6398 view.begin_selection(DisplayPoint::new(3, 3), true, 1, cx);
6399 view.update_selection(DisplayPoint::new(0, 0), 0, Vector2F::zero(), cx);
6400 });
6401
6402 assert_eq!(
6403 editor.update(cx, |view, cx| view.selections.display_ranges(cx)),
6404 [
6405 DisplayPoint::new(2, 2)..DisplayPoint::new(1, 1),
6406 DisplayPoint::new(3, 3)..DisplayPoint::new(0, 0)
6407 ]
6408 );
6409
6410 editor.update(cx, |view, cx| {
6411 view.end_selection(cx);
6412 });
6413
6414 assert_eq!(
6415 editor.update(cx, |view, cx| view.selections.display_ranges(cx)),
6416 [DisplayPoint::new(3, 3)..DisplayPoint::new(0, 0)]
6417 );
6418 }
6419
6420 #[gpui::test]
6421 fn test_canceling_pending_selection(cx: &mut gpui::MutableAppContext) {
6422 cx.set_global(Settings::test(cx));
6423 let buffer = MultiBuffer::build_simple("aaaaaa\nbbbbbb\ncccccc\ndddddd\n", cx);
6424 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
6425
6426 view.update(cx, |view, cx| {
6427 view.begin_selection(DisplayPoint::new(2, 2), false, 1, cx);
6428 assert_eq!(
6429 view.selections.display_ranges(cx),
6430 [DisplayPoint::new(2, 2)..DisplayPoint::new(2, 2)]
6431 );
6432 });
6433
6434 view.update(cx, |view, cx| {
6435 view.update_selection(DisplayPoint::new(3, 3), 0, Vector2F::zero(), cx);
6436 assert_eq!(
6437 view.selections.display_ranges(cx),
6438 [DisplayPoint::new(2, 2)..DisplayPoint::new(3, 3)]
6439 );
6440 });
6441
6442 view.update(cx, |view, cx| {
6443 view.cancel(&Cancel, cx);
6444 view.update_selection(DisplayPoint::new(1, 1), 0, Vector2F::zero(), cx);
6445 assert_eq!(
6446 view.selections.display_ranges(cx),
6447 [DisplayPoint::new(2, 2)..DisplayPoint::new(3, 3)]
6448 );
6449 });
6450 }
6451
6452 #[gpui::test]
6453 fn test_clone_with_selections(cx: &mut gpui::MutableAppContext) {
6454 let (text, selection_ranges) = marked_text_ranges(indoc! {"
6455 The qu[ick brown
6456 fox jum]ps over
6457 the lazy dog
6458 "});
6459 cx.set_global(Settings::test(cx));
6460 let buffer = MultiBuffer::build_simple(&text, cx);
6461
6462 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
6463
6464 let cloned_editor = view.update(cx, |view, cx| {
6465 view.change_selections(None, cx, |s| s.select_ranges(selection_ranges.clone()));
6466 view.clone(cx)
6467 });
6468
6469 assert_set_eq!(cloned_editor.selections.ranges(cx), selection_ranges);
6470 }
6471
6472 #[gpui::test]
6473 fn test_navigation_history(cx: &mut gpui::MutableAppContext) {
6474 cx.set_global(Settings::test(cx));
6475 use workspace::Item;
6476 let nav_history = Rc::new(RefCell::new(workspace::NavHistory::default()));
6477 let buffer = MultiBuffer::build_simple(&sample_text(300, 5, 'a'), cx);
6478
6479 cx.add_window(Default::default(), |cx| {
6480 let mut editor = build_editor(buffer.clone(), cx);
6481 editor.nav_history = Some(ItemNavHistory::new(nav_history.clone(), &cx.handle()));
6482
6483 // Move the cursor a small distance.
6484 // Nothing is added to the navigation history.
6485 editor.change_selections(None, cx, |s| {
6486 s.select_display_ranges([DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0)])
6487 });
6488 editor.change_selections(None, cx, |s| {
6489 s.select_display_ranges([DisplayPoint::new(3, 0)..DisplayPoint::new(3, 0)])
6490 });
6491 assert!(nav_history.borrow_mut().pop_backward().is_none());
6492
6493 // Move the cursor a large distance.
6494 // The history can jump back to the previous position.
6495 editor.change_selections(None, cx, |s| {
6496 s.select_display_ranges([DisplayPoint::new(13, 0)..DisplayPoint::new(13, 3)])
6497 });
6498 let nav_entry = nav_history.borrow_mut().pop_backward().unwrap();
6499 editor.navigate(nav_entry.data.unwrap(), cx);
6500 assert_eq!(nav_entry.item.id(), cx.view_id());
6501 assert_eq!(
6502 editor.selections.display_ranges(cx),
6503 &[DisplayPoint::new(3, 0)..DisplayPoint::new(3, 0)]
6504 );
6505 assert!(nav_history.borrow_mut().pop_backward().is_none());
6506
6507 // Move the cursor a small distance via the mouse.
6508 // Nothing is added to the navigation history.
6509 editor.begin_selection(DisplayPoint::new(5, 0), false, 1, cx);
6510 editor.end_selection(cx);
6511 assert_eq!(
6512 editor.selections.display_ranges(cx),
6513 &[DisplayPoint::new(5, 0)..DisplayPoint::new(5, 0)]
6514 );
6515 assert!(nav_history.borrow_mut().pop_backward().is_none());
6516
6517 // Move the cursor a large distance via the mouse.
6518 // The history can jump back to the previous position.
6519 editor.begin_selection(DisplayPoint::new(15, 0), false, 1, cx);
6520 editor.end_selection(cx);
6521 assert_eq!(
6522 editor.selections.display_ranges(cx),
6523 &[DisplayPoint::new(15, 0)..DisplayPoint::new(15, 0)]
6524 );
6525 let nav_entry = nav_history.borrow_mut().pop_backward().unwrap();
6526 editor.navigate(nav_entry.data.unwrap(), cx);
6527 assert_eq!(nav_entry.item.id(), cx.view_id());
6528 assert_eq!(
6529 editor.selections.display_ranges(cx),
6530 &[DisplayPoint::new(5, 0)..DisplayPoint::new(5, 0)]
6531 );
6532 assert!(nav_history.borrow_mut().pop_backward().is_none());
6533
6534 // Set scroll position to check later
6535 editor.set_scroll_position(Vector2F::new(5.5, 5.5), cx);
6536 let original_scroll_position = editor.scroll_position;
6537 let original_scroll_top_anchor = editor.scroll_top_anchor.clone();
6538
6539 // Jump to the end of the document and adjust scroll
6540 editor.move_to_end(&MoveToEnd, cx);
6541 editor.set_scroll_position(Vector2F::new(-2.5, -0.5), cx);
6542 assert_ne!(editor.scroll_position, original_scroll_position);
6543 assert_ne!(editor.scroll_top_anchor, original_scroll_top_anchor);
6544
6545 let nav_entry = nav_history.borrow_mut().pop_backward().unwrap();
6546 editor.navigate(nav_entry.data.unwrap(), cx);
6547 assert_eq!(editor.scroll_position, original_scroll_position);
6548 assert_eq!(editor.scroll_top_anchor, original_scroll_top_anchor);
6549
6550 // Ensure we don't panic when navigation data contains invalid anchors *and* points.
6551 let mut invalid_anchor = editor.scroll_top_anchor.clone();
6552 invalid_anchor.text_anchor.buffer_id = Some(999);
6553 let invalid_point = Point::new(9999, 0);
6554 editor.navigate(
6555 Box::new(NavigationData {
6556 cursor_anchor: invalid_anchor.clone(),
6557 cursor_position: invalid_point,
6558 scroll_top_anchor: invalid_anchor.clone(),
6559 scroll_top_row: invalid_point.row,
6560 scroll_position: Default::default(),
6561 }),
6562 cx,
6563 );
6564 assert_eq!(
6565 editor.selections.display_ranges(cx),
6566 &[editor.max_point(cx)..editor.max_point(cx)]
6567 );
6568 assert_eq!(
6569 editor.scroll_position(cx),
6570 vec2f(0., editor.max_point(cx).row() as f32)
6571 );
6572
6573 editor
6574 });
6575 }
6576
6577 #[gpui::test]
6578 fn test_cancel(cx: &mut gpui::MutableAppContext) {
6579 cx.set_global(Settings::test(cx));
6580 let buffer = MultiBuffer::build_simple("aaaaaa\nbbbbbb\ncccccc\ndddddd\n", cx);
6581 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
6582
6583 view.update(cx, |view, cx| {
6584 view.begin_selection(DisplayPoint::new(3, 4), false, 1, cx);
6585 view.update_selection(DisplayPoint::new(1, 1), 0, Vector2F::zero(), cx);
6586 view.end_selection(cx);
6587
6588 view.begin_selection(DisplayPoint::new(0, 1), true, 1, cx);
6589 view.update_selection(DisplayPoint::new(0, 3), 0, Vector2F::zero(), cx);
6590 view.end_selection(cx);
6591 assert_eq!(
6592 view.selections.display_ranges(cx),
6593 [
6594 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 3),
6595 DisplayPoint::new(3, 4)..DisplayPoint::new(1, 1),
6596 ]
6597 );
6598 });
6599
6600 view.update(cx, |view, cx| {
6601 view.cancel(&Cancel, cx);
6602 assert_eq!(
6603 view.selections.display_ranges(cx),
6604 [DisplayPoint::new(3, 4)..DisplayPoint::new(1, 1)]
6605 );
6606 });
6607
6608 view.update(cx, |view, cx| {
6609 view.cancel(&Cancel, cx);
6610 assert_eq!(
6611 view.selections.display_ranges(cx),
6612 [DisplayPoint::new(1, 1)..DisplayPoint::new(1, 1)]
6613 );
6614 });
6615 }
6616
6617 #[gpui::test]
6618 fn test_fold(cx: &mut gpui::MutableAppContext) {
6619 cx.set_global(Settings::test(cx));
6620 let buffer = MultiBuffer::build_simple(
6621 &"
6622 impl Foo {
6623 // Hello!
6624
6625 fn a() {
6626 1
6627 }
6628
6629 fn b() {
6630 2
6631 }
6632
6633 fn c() {
6634 3
6635 }
6636 }
6637 "
6638 .unindent(),
6639 cx,
6640 );
6641 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
6642
6643 view.update(cx, |view, cx| {
6644 view.change_selections(None, cx, |s| {
6645 s.select_display_ranges([DisplayPoint::new(8, 0)..DisplayPoint::new(12, 0)]);
6646 });
6647 view.fold(&Fold, cx);
6648 assert_eq!(
6649 view.display_text(cx),
6650 "
6651 impl Foo {
6652 // Hello!
6653
6654 fn a() {
6655 1
6656 }
6657
6658 fn b() {…
6659 }
6660
6661 fn c() {…
6662 }
6663 }
6664 "
6665 .unindent(),
6666 );
6667
6668 view.fold(&Fold, cx);
6669 assert_eq!(
6670 view.display_text(cx),
6671 "
6672 impl Foo {…
6673 }
6674 "
6675 .unindent(),
6676 );
6677
6678 view.unfold_lines(&UnfoldLines, cx);
6679 assert_eq!(
6680 view.display_text(cx),
6681 "
6682 impl Foo {
6683 // Hello!
6684
6685 fn a() {
6686 1
6687 }
6688
6689 fn b() {…
6690 }
6691
6692 fn c() {…
6693 }
6694 }
6695 "
6696 .unindent(),
6697 );
6698
6699 view.unfold_lines(&UnfoldLines, cx);
6700 assert_eq!(view.display_text(cx), buffer.read(cx).read(cx).text());
6701 });
6702 }
6703
6704 #[gpui::test]
6705 fn test_move_cursor(cx: &mut gpui::MutableAppContext) {
6706 cx.set_global(Settings::test(cx));
6707 let buffer = MultiBuffer::build_simple(&sample_text(6, 6, 'a'), cx);
6708 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
6709
6710 buffer.update(cx, |buffer, cx| {
6711 buffer.edit(
6712 vec![
6713 (Point::new(1, 0)..Point::new(1, 0), "\t"),
6714 (Point::new(1, 1)..Point::new(1, 1), "\t"),
6715 ],
6716 cx,
6717 );
6718 });
6719
6720 view.update(cx, |view, cx| {
6721 assert_eq!(
6722 view.selections.display_ranges(cx),
6723 &[DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0)]
6724 );
6725
6726 view.move_down(&MoveDown, cx);
6727 assert_eq!(
6728 view.selections.display_ranges(cx),
6729 &[DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0)]
6730 );
6731
6732 view.move_right(&MoveRight, cx);
6733 assert_eq!(
6734 view.selections.display_ranges(cx),
6735 &[DisplayPoint::new(1, 4)..DisplayPoint::new(1, 4)]
6736 );
6737
6738 view.move_left(&MoveLeft, cx);
6739 assert_eq!(
6740 view.selections.display_ranges(cx),
6741 &[DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0)]
6742 );
6743
6744 view.move_up(&MoveUp, cx);
6745 assert_eq!(
6746 view.selections.display_ranges(cx),
6747 &[DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0)]
6748 );
6749
6750 view.move_to_end(&MoveToEnd, cx);
6751 assert_eq!(
6752 view.selections.display_ranges(cx),
6753 &[DisplayPoint::new(5, 6)..DisplayPoint::new(5, 6)]
6754 );
6755
6756 view.move_to_beginning(&MoveToBeginning, cx);
6757 assert_eq!(
6758 view.selections.display_ranges(cx),
6759 &[DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0)]
6760 );
6761
6762 view.change_selections(None, cx, |s| {
6763 s.select_display_ranges([DisplayPoint::new(0, 1)..DisplayPoint::new(0, 2)]);
6764 });
6765 view.select_to_beginning(&SelectToBeginning, cx);
6766 assert_eq!(
6767 view.selections.display_ranges(cx),
6768 &[DisplayPoint::new(0, 1)..DisplayPoint::new(0, 0)]
6769 );
6770
6771 view.select_to_end(&SelectToEnd, cx);
6772 assert_eq!(
6773 view.selections.display_ranges(cx),
6774 &[DisplayPoint::new(0, 1)..DisplayPoint::new(5, 6)]
6775 );
6776 });
6777 }
6778
6779 #[gpui::test]
6780 fn test_move_cursor_multibyte(cx: &mut gpui::MutableAppContext) {
6781 cx.set_global(Settings::test(cx));
6782 let buffer = MultiBuffer::build_simple("ⓐⓑⓒⓓⓔ\nabcde\nαβγδε\n", cx);
6783 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
6784
6785 assert_eq!('ⓐ'.len_utf8(), 3);
6786 assert_eq!('α'.len_utf8(), 2);
6787
6788 view.update(cx, |view, cx| {
6789 view.fold_ranges(
6790 vec![
6791 Point::new(0, 6)..Point::new(0, 12),
6792 Point::new(1, 2)..Point::new(1, 4),
6793 Point::new(2, 4)..Point::new(2, 8),
6794 ],
6795 cx,
6796 );
6797 assert_eq!(view.display_text(cx), "ⓐⓑ…ⓔ\nab…e\nαβ…ε\n");
6798
6799 view.move_right(&MoveRight, cx);
6800 assert_eq!(
6801 view.selections.display_ranges(cx),
6802 &[empty_range(0, "ⓐ".len())]
6803 );
6804 view.move_right(&MoveRight, cx);
6805 assert_eq!(
6806 view.selections.display_ranges(cx),
6807 &[empty_range(0, "ⓐⓑ".len())]
6808 );
6809 view.move_right(&MoveRight, cx);
6810 assert_eq!(
6811 view.selections.display_ranges(cx),
6812 &[empty_range(0, "ⓐⓑ…".len())]
6813 );
6814
6815 view.move_down(&MoveDown, cx);
6816 assert_eq!(
6817 view.selections.display_ranges(cx),
6818 &[empty_range(1, "ab…".len())]
6819 );
6820 view.move_left(&MoveLeft, cx);
6821 assert_eq!(
6822 view.selections.display_ranges(cx),
6823 &[empty_range(1, "ab".len())]
6824 );
6825 view.move_left(&MoveLeft, cx);
6826 assert_eq!(
6827 view.selections.display_ranges(cx),
6828 &[empty_range(1, "a".len())]
6829 );
6830
6831 view.move_down(&MoveDown, cx);
6832 assert_eq!(
6833 view.selections.display_ranges(cx),
6834 &[empty_range(2, "α".len())]
6835 );
6836 view.move_right(&MoveRight, cx);
6837 assert_eq!(
6838 view.selections.display_ranges(cx),
6839 &[empty_range(2, "αβ".len())]
6840 );
6841 view.move_right(&MoveRight, cx);
6842 assert_eq!(
6843 view.selections.display_ranges(cx),
6844 &[empty_range(2, "αβ…".len())]
6845 );
6846 view.move_right(&MoveRight, cx);
6847 assert_eq!(
6848 view.selections.display_ranges(cx),
6849 &[empty_range(2, "αβ…ε".len())]
6850 );
6851
6852 view.move_up(&MoveUp, cx);
6853 assert_eq!(
6854 view.selections.display_ranges(cx),
6855 &[empty_range(1, "ab…e".len())]
6856 );
6857 view.move_up(&MoveUp, cx);
6858 assert_eq!(
6859 view.selections.display_ranges(cx),
6860 &[empty_range(0, "ⓐⓑ…ⓔ".len())]
6861 );
6862 view.move_left(&MoveLeft, cx);
6863 assert_eq!(
6864 view.selections.display_ranges(cx),
6865 &[empty_range(0, "ⓐⓑ…".len())]
6866 );
6867 view.move_left(&MoveLeft, cx);
6868 assert_eq!(
6869 view.selections.display_ranges(cx),
6870 &[empty_range(0, "ⓐⓑ".len())]
6871 );
6872 view.move_left(&MoveLeft, cx);
6873 assert_eq!(
6874 view.selections.display_ranges(cx),
6875 &[empty_range(0, "ⓐ".len())]
6876 );
6877 });
6878 }
6879
6880 #[gpui::test]
6881 fn test_move_cursor_different_line_lengths(cx: &mut gpui::MutableAppContext) {
6882 cx.set_global(Settings::test(cx));
6883 let buffer = MultiBuffer::build_simple("ⓐⓑⓒⓓⓔ\nabcd\nαβγ\nabcd\nⓐⓑⓒⓓⓔ\n", cx);
6884 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
6885 view.update(cx, |view, cx| {
6886 view.change_selections(None, cx, |s| {
6887 s.select_display_ranges([empty_range(0, "ⓐⓑⓒⓓⓔ".len())]);
6888 });
6889 view.move_down(&MoveDown, cx);
6890 assert_eq!(
6891 view.selections.display_ranges(cx),
6892 &[empty_range(1, "abcd".len())]
6893 );
6894
6895 view.move_down(&MoveDown, cx);
6896 assert_eq!(
6897 view.selections.display_ranges(cx),
6898 &[empty_range(2, "αβγ".len())]
6899 );
6900
6901 view.move_down(&MoveDown, cx);
6902 assert_eq!(
6903 view.selections.display_ranges(cx),
6904 &[empty_range(3, "abcd".len())]
6905 );
6906
6907 view.move_down(&MoveDown, cx);
6908 assert_eq!(
6909 view.selections.display_ranges(cx),
6910 &[empty_range(4, "ⓐⓑⓒⓓⓔ".len())]
6911 );
6912
6913 view.move_up(&MoveUp, cx);
6914 assert_eq!(
6915 view.selections.display_ranges(cx),
6916 &[empty_range(3, "abcd".len())]
6917 );
6918
6919 view.move_up(&MoveUp, cx);
6920 assert_eq!(
6921 view.selections.display_ranges(cx),
6922 &[empty_range(2, "αβγ".len())]
6923 );
6924 });
6925 }
6926
6927 #[gpui::test]
6928 fn test_beginning_end_of_line(cx: &mut gpui::MutableAppContext) {
6929 cx.set_global(Settings::test(cx));
6930 let buffer = MultiBuffer::build_simple("abc\n def", cx);
6931 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
6932 view.update(cx, |view, cx| {
6933 view.change_selections(None, cx, |s| {
6934 s.select_display_ranges([
6935 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 1),
6936 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 4),
6937 ]);
6938 });
6939 });
6940
6941 view.update(cx, |view, cx| {
6942 view.move_to_beginning_of_line(&MoveToBeginningOfLine, cx);
6943 assert_eq!(
6944 view.selections.display_ranges(cx),
6945 &[
6946 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0),
6947 DisplayPoint::new(1, 2)..DisplayPoint::new(1, 2),
6948 ]
6949 );
6950 });
6951
6952 view.update(cx, |view, cx| {
6953 view.move_to_beginning_of_line(&MoveToBeginningOfLine, cx);
6954 assert_eq!(
6955 view.selections.display_ranges(cx),
6956 &[
6957 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0),
6958 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0),
6959 ]
6960 );
6961 });
6962
6963 view.update(cx, |view, cx| {
6964 view.move_to_beginning_of_line(&MoveToBeginningOfLine, cx);
6965 assert_eq!(
6966 view.selections.display_ranges(cx),
6967 &[
6968 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0),
6969 DisplayPoint::new(1, 2)..DisplayPoint::new(1, 2),
6970 ]
6971 );
6972 });
6973
6974 view.update(cx, |view, cx| {
6975 view.move_to_end_of_line(&MoveToEndOfLine, cx);
6976 assert_eq!(
6977 view.selections.display_ranges(cx),
6978 &[
6979 DisplayPoint::new(0, 3)..DisplayPoint::new(0, 3),
6980 DisplayPoint::new(1, 5)..DisplayPoint::new(1, 5),
6981 ]
6982 );
6983 });
6984
6985 // Moving to the end of line again is a no-op.
6986 view.update(cx, |view, cx| {
6987 view.move_to_end_of_line(&MoveToEndOfLine, cx);
6988 assert_eq!(
6989 view.selections.display_ranges(cx),
6990 &[
6991 DisplayPoint::new(0, 3)..DisplayPoint::new(0, 3),
6992 DisplayPoint::new(1, 5)..DisplayPoint::new(1, 5),
6993 ]
6994 );
6995 });
6996
6997 view.update(cx, |view, cx| {
6998 view.move_left(&MoveLeft, cx);
6999 view.select_to_beginning_of_line(
7000 &SelectToBeginningOfLine {
7001 stop_at_soft_wraps: true,
7002 },
7003 cx,
7004 );
7005 assert_eq!(
7006 view.selections.display_ranges(cx),
7007 &[
7008 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 0),
7009 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 2),
7010 ]
7011 );
7012 });
7013
7014 view.update(cx, |view, cx| {
7015 view.select_to_beginning_of_line(
7016 &SelectToBeginningOfLine {
7017 stop_at_soft_wraps: true,
7018 },
7019 cx,
7020 );
7021 assert_eq!(
7022 view.selections.display_ranges(cx),
7023 &[
7024 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 0),
7025 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 0),
7026 ]
7027 );
7028 });
7029
7030 view.update(cx, |view, cx| {
7031 view.select_to_beginning_of_line(
7032 &SelectToBeginningOfLine {
7033 stop_at_soft_wraps: true,
7034 },
7035 cx,
7036 );
7037 assert_eq!(
7038 view.selections.display_ranges(cx),
7039 &[
7040 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 0),
7041 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 2),
7042 ]
7043 );
7044 });
7045
7046 view.update(cx, |view, cx| {
7047 view.select_to_end_of_line(
7048 &SelectToEndOfLine {
7049 stop_at_soft_wraps: true,
7050 },
7051 cx,
7052 );
7053 assert_eq!(
7054 view.selections.display_ranges(cx),
7055 &[
7056 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 3),
7057 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 5),
7058 ]
7059 );
7060 });
7061
7062 view.update(cx, |view, cx| {
7063 view.delete_to_end_of_line(&DeleteToEndOfLine, cx);
7064 assert_eq!(view.display_text(cx), "ab\n de");
7065 assert_eq!(
7066 view.selections.display_ranges(cx),
7067 &[
7068 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 2),
7069 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 4),
7070 ]
7071 );
7072 });
7073
7074 view.update(cx, |view, cx| {
7075 view.delete_to_beginning_of_line(&DeleteToBeginningOfLine, cx);
7076 assert_eq!(view.display_text(cx), "\n");
7077 assert_eq!(
7078 view.selections.display_ranges(cx),
7079 &[
7080 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0),
7081 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0),
7082 ]
7083 );
7084 });
7085 }
7086
7087 #[gpui::test]
7088 fn test_prev_next_word_boundary(cx: &mut gpui::MutableAppContext) {
7089 cx.set_global(Settings::test(cx));
7090 let buffer = MultiBuffer::build_simple("use std::str::{foo, bar}\n\n {baz.qux()}", cx);
7091 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7092 view.update(cx, |view, cx| {
7093 view.change_selections(None, cx, |s| {
7094 s.select_display_ranges([
7095 DisplayPoint::new(0, 11)..DisplayPoint::new(0, 11),
7096 DisplayPoint::new(2, 4)..DisplayPoint::new(2, 4),
7097 ])
7098 });
7099
7100 view.move_to_previous_word_start(&MoveToPreviousWordStart, cx);
7101 assert_selection_ranges(
7102 "use std::<>str::{foo, bar}\n\n {[]baz.qux()}",
7103 vec![('<', '>'), ('[', ']')],
7104 view,
7105 cx,
7106 );
7107
7108 view.move_to_previous_word_start(&MoveToPreviousWordStart, cx);
7109 assert_selection_ranges(
7110 "use std<>::str::{foo, bar}\n\n []{baz.qux()}",
7111 vec![('<', '>'), ('[', ']')],
7112 view,
7113 cx,
7114 );
7115
7116 view.move_to_previous_word_start(&MoveToPreviousWordStart, cx);
7117 assert_selection_ranges(
7118 "use <>std::str::{foo, bar}\n\n[] {baz.qux()}",
7119 vec![('<', '>'), ('[', ']')],
7120 view,
7121 cx,
7122 );
7123
7124 view.move_to_previous_word_start(&MoveToPreviousWordStart, cx);
7125 assert_selection_ranges(
7126 "<>use std::str::{foo, bar}\n[]\n {baz.qux()}",
7127 vec![('<', '>'), ('[', ']')],
7128 view,
7129 cx,
7130 );
7131
7132 view.move_to_previous_word_start(&MoveToPreviousWordStart, cx);
7133 assert_selection_ranges(
7134 "<>use std::str::{foo, bar[]}\n\n {baz.qux()}",
7135 vec![('<', '>'), ('[', ']')],
7136 view,
7137 cx,
7138 );
7139
7140 view.move_to_next_word_end(&MoveToNextWordEnd, cx);
7141 assert_selection_ranges(
7142 "use<> std::str::{foo, bar}[]\n\n {baz.qux()}",
7143 vec![('<', '>'), ('[', ']')],
7144 view,
7145 cx,
7146 );
7147
7148 view.move_to_next_word_end(&MoveToNextWordEnd, cx);
7149 assert_selection_ranges(
7150 "use std<>::str::{foo, bar}\n[]\n {baz.qux()}",
7151 vec![('<', '>'), ('[', ']')],
7152 view,
7153 cx,
7154 );
7155
7156 view.move_to_next_word_end(&MoveToNextWordEnd, cx);
7157 assert_selection_ranges(
7158 "use std::<>str::{foo, bar}\n\n {[]baz.qux()}",
7159 vec![('<', '>'), ('[', ']')],
7160 view,
7161 cx,
7162 );
7163
7164 view.move_right(&MoveRight, cx);
7165 view.select_to_previous_word_start(&SelectToPreviousWordStart, cx);
7166 assert_selection_ranges(
7167 "use std::>s<tr::{foo, bar}\n\n {]b[az.qux()}",
7168 vec![('<', '>'), ('[', ']')],
7169 view,
7170 cx,
7171 );
7172
7173 view.select_to_previous_word_start(&SelectToPreviousWordStart, cx);
7174 assert_selection_ranges(
7175 "use std>::s<tr::{foo, bar}\n\n ]{b[az.qux()}",
7176 vec![('<', '>'), ('[', ']')],
7177 view,
7178 cx,
7179 );
7180
7181 view.select_to_next_word_end(&SelectToNextWordEnd, cx);
7182 assert_selection_ranges(
7183 "use std::>s<tr::{foo, bar}\n\n {]b[az.qux()}",
7184 vec![('<', '>'), ('[', ']')],
7185 view,
7186 cx,
7187 );
7188 });
7189 }
7190
7191 #[gpui::test]
7192 fn test_prev_next_word_bounds_with_soft_wrap(cx: &mut gpui::MutableAppContext) {
7193 cx.set_global(Settings::test(cx));
7194 let buffer = MultiBuffer::build_simple("use one::{\n two::three::four::five\n};", cx);
7195 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7196
7197 view.update(cx, |view, cx| {
7198 view.set_wrap_width(Some(140.), cx);
7199 assert_eq!(
7200 view.display_text(cx),
7201 "use one::{\n two::three::\n four::five\n};"
7202 );
7203
7204 view.change_selections(None, cx, |s| {
7205 s.select_display_ranges([DisplayPoint::new(1, 7)..DisplayPoint::new(1, 7)]);
7206 });
7207
7208 view.move_to_next_word_end(&MoveToNextWordEnd, cx);
7209 assert_eq!(
7210 view.selections.display_ranges(cx),
7211 &[DisplayPoint::new(1, 9)..DisplayPoint::new(1, 9)]
7212 );
7213
7214 view.move_to_next_word_end(&MoveToNextWordEnd, cx);
7215 assert_eq!(
7216 view.selections.display_ranges(cx),
7217 &[DisplayPoint::new(1, 14)..DisplayPoint::new(1, 14)]
7218 );
7219
7220 view.move_to_next_word_end(&MoveToNextWordEnd, cx);
7221 assert_eq!(
7222 view.selections.display_ranges(cx),
7223 &[DisplayPoint::new(2, 4)..DisplayPoint::new(2, 4)]
7224 );
7225
7226 view.move_to_next_word_end(&MoveToNextWordEnd, cx);
7227 assert_eq!(
7228 view.selections.display_ranges(cx),
7229 &[DisplayPoint::new(2, 8)..DisplayPoint::new(2, 8)]
7230 );
7231
7232 view.move_to_previous_word_start(&MoveToPreviousWordStart, cx);
7233 assert_eq!(
7234 view.selections.display_ranges(cx),
7235 &[DisplayPoint::new(2, 4)..DisplayPoint::new(2, 4)]
7236 );
7237
7238 view.move_to_previous_word_start(&MoveToPreviousWordStart, cx);
7239 assert_eq!(
7240 view.selections.display_ranges(cx),
7241 &[DisplayPoint::new(1, 14)..DisplayPoint::new(1, 14)]
7242 );
7243 });
7244 }
7245
7246 #[gpui::test]
7247 fn test_delete_to_beginning_of_line(cx: &mut gpui::MutableAppContext) {
7248 cx.set_global(Settings::test(cx));
7249 let (text, ranges) = marked_text_ranges("one [two three] four");
7250 let buffer = MultiBuffer::build_simple(&text, cx);
7251
7252 let (_, editor) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
7253
7254 editor.update(cx, |editor, cx| {
7255 editor.change_selections(None, cx, |s| s.select_ranges(ranges));
7256 editor.delete_to_beginning_of_line(&DeleteToBeginningOfLine, cx);
7257 assert_eq!(editor.text(cx), " four");
7258 });
7259 }
7260
7261 #[gpui::test]
7262 fn test_delete_to_word_boundary(cx: &mut gpui::MutableAppContext) {
7263 cx.set_global(Settings::test(cx));
7264 let buffer = MultiBuffer::build_simple("one two three four", cx);
7265 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
7266
7267 view.update(cx, |view, cx| {
7268 view.change_selections(None, cx, |s| {
7269 s.select_display_ranges([
7270 // an empty selection - the preceding word fragment is deleted
7271 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 2),
7272 // characters selected - they are deleted
7273 DisplayPoint::new(0, 9)..DisplayPoint::new(0, 12),
7274 ])
7275 });
7276 view.delete_to_previous_word_start(&DeleteToPreviousWordStart, cx);
7277 });
7278
7279 assert_eq!(buffer.read(cx).read(cx).text(), "e two te four");
7280
7281 view.update(cx, |view, cx| {
7282 view.change_selections(None, cx, |s| {
7283 s.select_display_ranges([
7284 // an empty selection - the following word fragment is deleted
7285 DisplayPoint::new(0, 3)..DisplayPoint::new(0, 3),
7286 // characters selected - they are deleted
7287 DisplayPoint::new(0, 9)..DisplayPoint::new(0, 10),
7288 ])
7289 });
7290 view.delete_to_next_word_end(&DeleteToNextWordEnd, cx);
7291 });
7292
7293 assert_eq!(buffer.read(cx).read(cx).text(), "e t te our");
7294 }
7295
7296 #[gpui::test]
7297 fn test_newline(cx: &mut gpui::MutableAppContext) {
7298 cx.set_global(Settings::test(cx));
7299 let buffer = MultiBuffer::build_simple("aaaa\n bbbb\n", cx);
7300 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
7301
7302 view.update(cx, |view, cx| {
7303 view.change_selections(None, cx, |s| {
7304 s.select_display_ranges([
7305 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 2),
7306 DisplayPoint::new(1, 2)..DisplayPoint::new(1, 2),
7307 DisplayPoint::new(1, 6)..DisplayPoint::new(1, 6),
7308 ])
7309 });
7310
7311 view.newline(&Newline, cx);
7312 assert_eq!(view.text(cx), "aa\naa\n \n bb\n bb\n");
7313 });
7314 }
7315
7316 #[gpui::test]
7317 fn test_newline_with_old_selections(cx: &mut gpui::MutableAppContext) {
7318 cx.set_global(Settings::test(cx));
7319 let buffer = MultiBuffer::build_simple(
7320 "
7321 a
7322 b(
7323 X
7324 )
7325 c(
7326 X
7327 )
7328 "
7329 .unindent()
7330 .as_str(),
7331 cx,
7332 );
7333
7334 let (_, editor) = cx.add_window(Default::default(), |cx| {
7335 let mut editor = build_editor(buffer.clone(), cx);
7336 editor.change_selections(None, cx, |s| {
7337 s.select_ranges([
7338 Point::new(2, 4)..Point::new(2, 5),
7339 Point::new(5, 4)..Point::new(5, 5),
7340 ])
7341 });
7342 editor
7343 });
7344
7345 // Edit the buffer directly, deleting ranges surrounding the editor's selections
7346 buffer.update(cx, |buffer, cx| {
7347 buffer.edit(
7348 [
7349 (Point::new(1, 2)..Point::new(3, 0), ""),
7350 (Point::new(4, 2)..Point::new(6, 0), ""),
7351 ],
7352 cx,
7353 );
7354 assert_eq!(
7355 buffer.read(cx).text(),
7356 "
7357 a
7358 b()
7359 c()
7360 "
7361 .unindent()
7362 );
7363 });
7364
7365 editor.update(cx, |editor, cx| {
7366 assert_eq!(
7367 editor.selections.ranges(cx),
7368 &[
7369 Point::new(1, 2)..Point::new(1, 2),
7370 Point::new(2, 2)..Point::new(2, 2),
7371 ],
7372 );
7373
7374 editor.newline(&Newline, cx);
7375 assert_eq!(
7376 editor.text(cx),
7377 "
7378 a
7379 b(
7380 )
7381 c(
7382 )
7383 "
7384 .unindent()
7385 );
7386
7387 // The selections are moved after the inserted newlines
7388 assert_eq!(
7389 editor.selections.ranges(cx),
7390 &[
7391 Point::new(2, 0)..Point::new(2, 0),
7392 Point::new(4, 0)..Point::new(4, 0),
7393 ],
7394 );
7395 });
7396 }
7397
7398 #[gpui::test]
7399 fn test_insert_with_old_selections(cx: &mut gpui::MutableAppContext) {
7400 cx.set_global(Settings::test(cx));
7401 let buffer = MultiBuffer::build_simple("a( X ), b( Y ), c( Z )", cx);
7402 let (_, editor) = cx.add_window(Default::default(), |cx| {
7403 let mut editor = build_editor(buffer.clone(), cx);
7404 editor.change_selections(None, cx, |s| s.select_ranges([3..4, 11..12, 19..20]));
7405 editor
7406 });
7407
7408 // Edit the buffer directly, deleting ranges surrounding the editor's selections
7409 buffer.update(cx, |buffer, cx| {
7410 buffer.edit([(2..5, ""), (10..13, ""), (18..21, "")], cx);
7411 assert_eq!(buffer.read(cx).text(), "a(), b(), c()".unindent());
7412 });
7413
7414 editor.update(cx, |editor, cx| {
7415 assert_eq!(editor.selections.ranges(cx), &[2..2, 7..7, 12..12],);
7416
7417 editor.insert("Z", cx);
7418 assert_eq!(editor.text(cx), "a(Z), b(Z), c(Z)");
7419
7420 // The selections are moved after the inserted characters
7421 assert_eq!(editor.selections.ranges(cx), &[3..3, 9..9, 15..15],);
7422 });
7423 }
7424
7425 #[gpui::test]
7426 async fn test_indent_outdent(cx: &mut gpui::TestAppContext) {
7427 let mut cx = EditorTestContext::new(cx).await;
7428
7429 cx.set_state(indoc! {"
7430 [one} [two}
7431 three
7432 four"});
7433 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7434 cx.assert_editor_state(indoc! {"
7435 [one} [two}
7436 three
7437 four"});
7438
7439 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7440 cx.assert_editor_state(indoc! {"
7441 [one} [two}
7442 three
7443 four"});
7444
7445 // select across line ending
7446 cx.set_state(indoc! {"
7447 one two
7448 t[hree
7449 } four"});
7450 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7451 cx.assert_editor_state(indoc! {"
7452 one two
7453 t[hree
7454 } four"});
7455
7456 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7457 cx.assert_editor_state(indoc! {"
7458 one two
7459 t[hree
7460 } four"});
7461
7462 // Ensure that indenting/outdenting works when the cursor is at column 0.
7463 cx.set_state(indoc! {"
7464 one two
7465 |three
7466 four"});
7467 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7468 cx.assert_editor_state(indoc! {"
7469 one two
7470 |three
7471 four"});
7472
7473 cx.set_state(indoc! {"
7474 one two
7475 | three
7476 four"});
7477 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7478 cx.assert_editor_state(indoc! {"
7479 one two
7480 |three
7481 four"});
7482 }
7483
7484 #[gpui::test]
7485 async fn test_indent_outdent_with_hard_tabs(cx: &mut gpui::TestAppContext) {
7486 let mut cx = EditorTestContext::new(cx).await;
7487 cx.update(|cx| {
7488 cx.update_global::<Settings, _, _>(|settings, _| {
7489 settings.hard_tabs = true;
7490 });
7491 });
7492
7493 // select two ranges on one line
7494 cx.set_state(indoc! {"
7495 [one} [two}
7496 three
7497 four"});
7498 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7499 cx.assert_editor_state(indoc! {"
7500 \t[one} [two}
7501 three
7502 four"});
7503 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7504 cx.assert_editor_state(indoc! {"
7505 \t\t[one} [two}
7506 three
7507 four"});
7508 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7509 cx.assert_editor_state(indoc! {"
7510 \t[one} [two}
7511 three
7512 four"});
7513 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7514 cx.assert_editor_state(indoc! {"
7515 [one} [two}
7516 three
7517 four"});
7518
7519 // select across a line ending
7520 cx.set_state(indoc! {"
7521 one two
7522 t[hree
7523 }four"});
7524 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7525 cx.assert_editor_state(indoc! {"
7526 one two
7527 \tt[hree
7528 }four"});
7529 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7530 cx.assert_editor_state(indoc! {"
7531 one two
7532 \t\tt[hree
7533 }four"});
7534 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7535 cx.assert_editor_state(indoc! {"
7536 one two
7537 \tt[hree
7538 }four"});
7539 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7540 cx.assert_editor_state(indoc! {"
7541 one two
7542 t[hree
7543 }four"});
7544
7545 // Ensure that indenting/outdenting works when the cursor is at column 0.
7546 cx.set_state(indoc! {"
7547 one two
7548 |three
7549 four"});
7550 cx.assert_editor_state(indoc! {"
7551 one two
7552 |three
7553 four"});
7554 cx.update_editor(|e, cx| e.tab(&Tab, cx));
7555 cx.assert_editor_state(indoc! {"
7556 one two
7557 \t|three
7558 four"});
7559 cx.update_editor(|e, cx| e.tab_prev(&TabPrev, cx));
7560 cx.assert_editor_state(indoc! {"
7561 one two
7562 |three
7563 four"});
7564 }
7565
7566 #[gpui::test]
7567 fn test_indent_outdent_with_excerpts(cx: &mut gpui::MutableAppContext) {
7568 cx.set_global(
7569 Settings::test(cx)
7570 .with_overrides(
7571 "TOML",
7572 LanguageOverride {
7573 tab_size: Some(2),
7574 ..Default::default()
7575 },
7576 )
7577 .with_overrides(
7578 "Rust",
7579 LanguageOverride {
7580 tab_size: Some(4),
7581 ..Default::default()
7582 },
7583 ),
7584 );
7585 let toml_language = Arc::new(Language::new(
7586 LanguageConfig {
7587 name: "TOML".into(),
7588 ..Default::default()
7589 },
7590 None,
7591 ));
7592 let rust_language = Arc::new(Language::new(
7593 LanguageConfig {
7594 name: "Rust".into(),
7595 ..Default::default()
7596 },
7597 None,
7598 ));
7599
7600 let toml_buffer = cx
7601 .add_model(|cx| Buffer::new(0, "a = 1\nb = 2\n", cx).with_language(toml_language, cx));
7602 let rust_buffer = cx.add_model(|cx| {
7603 Buffer::new(0, "const c: usize = 3;\n", cx).with_language(rust_language, cx)
7604 });
7605 let multibuffer = cx.add_model(|cx| {
7606 let mut multibuffer = MultiBuffer::new(0);
7607 multibuffer.push_excerpts(
7608 toml_buffer.clone(),
7609 [ExcerptRange {
7610 context: Point::new(0, 0)..Point::new(2, 0),
7611 primary: None,
7612 }],
7613 cx,
7614 );
7615 multibuffer.push_excerpts(
7616 rust_buffer.clone(),
7617 [ExcerptRange {
7618 context: Point::new(0, 0)..Point::new(1, 0),
7619 primary: None,
7620 }],
7621 cx,
7622 );
7623 multibuffer
7624 });
7625
7626 cx.add_window(Default::default(), |cx| {
7627 let mut editor = build_editor(multibuffer, cx);
7628
7629 assert_eq!(
7630 editor.text(cx),
7631 indoc! {"
7632 a = 1
7633 b = 2
7634
7635 const c: usize = 3;
7636 "}
7637 );
7638
7639 select_ranges(
7640 &mut editor,
7641 indoc! {"
7642 [a] = 1
7643 b = 2
7644
7645 [const c:] usize = 3;
7646 "},
7647 cx,
7648 );
7649
7650 editor.tab(&Tab, cx);
7651 assert_text_with_selections(
7652 &mut editor,
7653 indoc! {"
7654 [a] = 1
7655 b = 2
7656
7657 [const c:] usize = 3;
7658 "},
7659 cx,
7660 );
7661 editor.tab_prev(&TabPrev, cx);
7662 assert_text_with_selections(
7663 &mut editor,
7664 indoc! {"
7665 [a] = 1
7666 b = 2
7667
7668 [const c:] usize = 3;
7669 "},
7670 cx,
7671 );
7672
7673 editor
7674 });
7675 }
7676
7677 #[gpui::test]
7678 async fn test_backspace(cx: &mut gpui::TestAppContext) {
7679 let mut cx = EditorTestContext::new(cx).await;
7680 // Basic backspace
7681 cx.set_state(indoc! {"
7682 on|e two three
7683 fou[r} five six
7684 seven {eight nine
7685 ]ten"});
7686 cx.update_editor(|e, cx| e.backspace(&Backspace, cx));
7687 cx.assert_editor_state(indoc! {"
7688 o|e two three
7689 fou| five six
7690 seven |ten"});
7691
7692 // Test backspace inside and around indents
7693 cx.set_state(indoc! {"
7694 zero
7695 |one
7696 |two
7697 | | | three
7698 | | four"});
7699 cx.update_editor(|e, cx| e.backspace(&Backspace, cx));
7700 cx.assert_editor_state(indoc! {"
7701 zero
7702 |one
7703 |two
7704 | three| four"});
7705
7706 // Test backspace with line_mode set to true
7707 cx.update_editor(|e, _| e.selections.line_mode = true);
7708 cx.set_state(indoc! {"
7709 The |quick |brown
7710 fox jumps over
7711 the lazy dog
7712 |The qu[ick b}rown"});
7713 cx.update_editor(|e, cx| e.backspace(&Backspace, cx));
7714 cx.assert_editor_state(indoc! {"
7715 |fox jumps over
7716 the lazy dog|"});
7717 }
7718
7719 #[gpui::test]
7720 async fn test_delete(cx: &mut gpui::TestAppContext) {
7721 let mut cx = EditorTestContext::new(cx).await;
7722
7723 cx.set_state(indoc! {"
7724 on|e two three
7725 fou[r} five six
7726 seven {eight nine
7727 ]ten"});
7728 cx.update_editor(|e, cx| e.delete(&Delete, cx));
7729 cx.assert_editor_state(indoc! {"
7730 on| two three
7731 fou| five six
7732 seven |ten"});
7733
7734 // Test backspace with line_mode set to true
7735 cx.update_editor(|e, _| e.selections.line_mode = true);
7736 cx.set_state(indoc! {"
7737 The |quick |brown
7738 fox {jum]ps over
7739 the lazy dog
7740 |The qu[ick b}rown"});
7741 cx.update_editor(|e, cx| e.backspace(&Backspace, cx));
7742 cx.assert_editor_state("|the lazy dog|");
7743 }
7744
7745 #[gpui::test]
7746 fn test_delete_line(cx: &mut gpui::MutableAppContext) {
7747 cx.set_global(Settings::test(cx));
7748 let buffer = MultiBuffer::build_simple("abc\ndef\nghi\n", cx);
7749 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7750 view.update(cx, |view, cx| {
7751 view.change_selections(None, cx, |s| {
7752 s.select_display_ranges([
7753 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 1),
7754 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 1),
7755 DisplayPoint::new(3, 0)..DisplayPoint::new(3, 0),
7756 ])
7757 });
7758 view.delete_line(&DeleteLine, cx);
7759 assert_eq!(view.display_text(cx), "ghi");
7760 assert_eq!(
7761 view.selections.display_ranges(cx),
7762 vec![
7763 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0),
7764 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 1)
7765 ]
7766 );
7767 });
7768
7769 cx.set_global(Settings::test(cx));
7770 let buffer = MultiBuffer::build_simple("abc\ndef\nghi\n", cx);
7771 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7772 view.update(cx, |view, cx| {
7773 view.change_selections(None, cx, |s| {
7774 s.select_display_ranges([DisplayPoint::new(2, 0)..DisplayPoint::new(0, 1)])
7775 });
7776 view.delete_line(&DeleteLine, cx);
7777 assert_eq!(view.display_text(cx), "ghi\n");
7778 assert_eq!(
7779 view.selections.display_ranges(cx),
7780 vec![DisplayPoint::new(0, 1)..DisplayPoint::new(0, 1)]
7781 );
7782 });
7783 }
7784
7785 #[gpui::test]
7786 fn test_duplicate_line(cx: &mut gpui::MutableAppContext) {
7787 cx.set_global(Settings::test(cx));
7788 let buffer = MultiBuffer::build_simple("abc\ndef\nghi\n", cx);
7789 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7790 view.update(cx, |view, cx| {
7791 view.change_selections(None, cx, |s| {
7792 s.select_display_ranges([
7793 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 1),
7794 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 2),
7795 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0),
7796 DisplayPoint::new(3, 0)..DisplayPoint::new(3, 0),
7797 ])
7798 });
7799 view.duplicate_line(&DuplicateLine, cx);
7800 assert_eq!(view.display_text(cx), "abc\nabc\ndef\ndef\nghi\n\n");
7801 assert_eq!(
7802 view.selections.display_ranges(cx),
7803 vec![
7804 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 1),
7805 DisplayPoint::new(1, 2)..DisplayPoint::new(1, 2),
7806 DisplayPoint::new(3, 0)..DisplayPoint::new(3, 0),
7807 DisplayPoint::new(6, 0)..DisplayPoint::new(6, 0),
7808 ]
7809 );
7810 });
7811
7812 let buffer = MultiBuffer::build_simple("abc\ndef\nghi\n", cx);
7813 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7814 view.update(cx, |view, cx| {
7815 view.change_selections(None, cx, |s| {
7816 s.select_display_ranges([
7817 DisplayPoint::new(0, 1)..DisplayPoint::new(1, 1),
7818 DisplayPoint::new(1, 2)..DisplayPoint::new(2, 1),
7819 ])
7820 });
7821 view.duplicate_line(&DuplicateLine, cx);
7822 assert_eq!(view.display_text(cx), "abc\ndef\nghi\nabc\ndef\nghi\n");
7823 assert_eq!(
7824 view.selections.display_ranges(cx),
7825 vec![
7826 DisplayPoint::new(3, 1)..DisplayPoint::new(4, 1),
7827 DisplayPoint::new(4, 2)..DisplayPoint::new(5, 1),
7828 ]
7829 );
7830 });
7831 }
7832
7833 #[gpui::test]
7834 fn test_move_line_up_down(cx: &mut gpui::MutableAppContext) {
7835 cx.set_global(Settings::test(cx));
7836 let buffer = MultiBuffer::build_simple(&sample_text(10, 5, 'a'), cx);
7837 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7838 view.update(cx, |view, cx| {
7839 view.fold_ranges(
7840 vec![
7841 Point::new(0, 2)..Point::new(1, 2),
7842 Point::new(2, 3)..Point::new(4, 1),
7843 Point::new(7, 0)..Point::new(8, 4),
7844 ],
7845 cx,
7846 );
7847 view.change_selections(None, cx, |s| {
7848 s.select_display_ranges([
7849 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 1),
7850 DisplayPoint::new(3, 1)..DisplayPoint::new(3, 1),
7851 DisplayPoint::new(3, 2)..DisplayPoint::new(4, 3),
7852 DisplayPoint::new(5, 0)..DisplayPoint::new(5, 2),
7853 ])
7854 });
7855 assert_eq!(
7856 view.display_text(cx),
7857 "aa…bbb\nccc…eeee\nfffff\nggggg\n…i\njjjjj"
7858 );
7859
7860 view.move_line_up(&MoveLineUp, cx);
7861 assert_eq!(
7862 view.display_text(cx),
7863 "aa…bbb\nccc…eeee\nggggg\n…i\njjjjj\nfffff"
7864 );
7865 assert_eq!(
7866 view.selections.display_ranges(cx),
7867 vec![
7868 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 1),
7869 DisplayPoint::new(2, 1)..DisplayPoint::new(2, 1),
7870 DisplayPoint::new(2, 2)..DisplayPoint::new(3, 3),
7871 DisplayPoint::new(4, 0)..DisplayPoint::new(4, 2)
7872 ]
7873 );
7874 });
7875
7876 view.update(cx, |view, cx| {
7877 view.move_line_down(&MoveLineDown, cx);
7878 assert_eq!(
7879 view.display_text(cx),
7880 "ccc…eeee\naa…bbb\nfffff\nggggg\n…i\njjjjj"
7881 );
7882 assert_eq!(
7883 view.selections.display_ranges(cx),
7884 vec![
7885 DisplayPoint::new(1, 1)..DisplayPoint::new(1, 1),
7886 DisplayPoint::new(3, 1)..DisplayPoint::new(3, 1),
7887 DisplayPoint::new(3, 2)..DisplayPoint::new(4, 3),
7888 DisplayPoint::new(5, 0)..DisplayPoint::new(5, 2)
7889 ]
7890 );
7891 });
7892
7893 view.update(cx, |view, cx| {
7894 view.move_line_down(&MoveLineDown, cx);
7895 assert_eq!(
7896 view.display_text(cx),
7897 "ccc…eeee\nfffff\naa…bbb\nggggg\n…i\njjjjj"
7898 );
7899 assert_eq!(
7900 view.selections.display_ranges(cx),
7901 vec![
7902 DisplayPoint::new(2, 1)..DisplayPoint::new(2, 1),
7903 DisplayPoint::new(3, 1)..DisplayPoint::new(3, 1),
7904 DisplayPoint::new(3, 2)..DisplayPoint::new(4, 3),
7905 DisplayPoint::new(5, 0)..DisplayPoint::new(5, 2)
7906 ]
7907 );
7908 });
7909
7910 view.update(cx, |view, cx| {
7911 view.move_line_up(&MoveLineUp, cx);
7912 assert_eq!(
7913 view.display_text(cx),
7914 "ccc…eeee\naa…bbb\nggggg\n…i\njjjjj\nfffff"
7915 );
7916 assert_eq!(
7917 view.selections.display_ranges(cx),
7918 vec![
7919 DisplayPoint::new(1, 1)..DisplayPoint::new(1, 1),
7920 DisplayPoint::new(2, 1)..DisplayPoint::new(2, 1),
7921 DisplayPoint::new(2, 2)..DisplayPoint::new(3, 3),
7922 DisplayPoint::new(4, 0)..DisplayPoint::new(4, 2)
7923 ]
7924 );
7925 });
7926 }
7927
7928 #[gpui::test]
7929 fn test_move_line_up_down_with_blocks(cx: &mut gpui::MutableAppContext) {
7930 cx.set_global(Settings::test(cx));
7931 let buffer = MultiBuffer::build_simple(&sample_text(10, 5, 'a'), cx);
7932 let snapshot = buffer.read(cx).snapshot(cx);
7933 let (_, editor) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
7934 editor.update(cx, |editor, cx| {
7935 editor.insert_blocks(
7936 [BlockProperties {
7937 style: BlockStyle::Fixed,
7938 position: snapshot.anchor_after(Point::new(2, 0)),
7939 disposition: BlockDisposition::Below,
7940 height: 1,
7941 render: Arc::new(|_| Empty::new().boxed()),
7942 }],
7943 cx,
7944 );
7945 editor.change_selections(None, cx, |s| {
7946 s.select_ranges([Point::new(2, 0)..Point::new(2, 0)])
7947 });
7948 editor.move_line_down(&MoveLineDown, cx);
7949 });
7950 }
7951
7952 #[gpui::test]
7953 fn test_transpose(cx: &mut gpui::MutableAppContext) {
7954 cx.set_global(Settings::test(cx));
7955
7956 cx.add_window(Default::default(), |cx| {
7957 let mut editor = build_editor(MultiBuffer::build_simple("abc", cx), cx);
7958
7959 editor.change_selections(None, cx, |s| s.select_ranges([1..1]));
7960 editor.transpose(&Default::default(), cx);
7961 assert_eq!(editor.text(cx), "bac");
7962 assert_eq!(editor.selections.ranges(cx), [2..2]);
7963
7964 editor.transpose(&Default::default(), cx);
7965 assert_eq!(editor.text(cx), "bca");
7966 assert_eq!(editor.selections.ranges(cx), [3..3]);
7967
7968 editor.transpose(&Default::default(), cx);
7969 assert_eq!(editor.text(cx), "bac");
7970 assert_eq!(editor.selections.ranges(cx), [3..3]);
7971
7972 editor
7973 })
7974 .1;
7975
7976 cx.add_window(Default::default(), |cx| {
7977 let mut editor = build_editor(MultiBuffer::build_simple("abc\nde", cx), cx);
7978
7979 editor.change_selections(None, cx, |s| s.select_ranges([3..3]));
7980 editor.transpose(&Default::default(), cx);
7981 assert_eq!(editor.text(cx), "acb\nde");
7982 assert_eq!(editor.selections.ranges(cx), [3..3]);
7983
7984 editor.change_selections(None, cx, |s| s.select_ranges([4..4]));
7985 editor.transpose(&Default::default(), cx);
7986 assert_eq!(editor.text(cx), "acbd\ne");
7987 assert_eq!(editor.selections.ranges(cx), [5..5]);
7988
7989 editor.transpose(&Default::default(), cx);
7990 assert_eq!(editor.text(cx), "acbde\n");
7991 assert_eq!(editor.selections.ranges(cx), [6..6]);
7992
7993 editor.transpose(&Default::default(), cx);
7994 assert_eq!(editor.text(cx), "acbd\ne");
7995 assert_eq!(editor.selections.ranges(cx), [6..6]);
7996
7997 editor
7998 })
7999 .1;
8000
8001 cx.add_window(Default::default(), |cx| {
8002 let mut editor = build_editor(MultiBuffer::build_simple("abc\nde", cx), cx);
8003
8004 editor.change_selections(None, cx, |s| s.select_ranges([1..1, 2..2, 4..4]));
8005 editor.transpose(&Default::default(), cx);
8006 assert_eq!(editor.text(cx), "bacd\ne");
8007 assert_eq!(editor.selections.ranges(cx), [2..2, 3..3, 5..5]);
8008
8009 editor.transpose(&Default::default(), cx);
8010 assert_eq!(editor.text(cx), "bcade\n");
8011 assert_eq!(editor.selections.ranges(cx), [3..3, 4..4, 6..6]);
8012
8013 editor.transpose(&Default::default(), cx);
8014 assert_eq!(editor.text(cx), "bcda\ne");
8015 assert_eq!(editor.selections.ranges(cx), [4..4, 6..6]);
8016
8017 editor.transpose(&Default::default(), cx);
8018 assert_eq!(editor.text(cx), "bcade\n");
8019 assert_eq!(editor.selections.ranges(cx), [4..4, 6..6]);
8020
8021 editor.transpose(&Default::default(), cx);
8022 assert_eq!(editor.text(cx), "bcaed\n");
8023 assert_eq!(editor.selections.ranges(cx), [5..5, 6..6]);
8024
8025 editor
8026 })
8027 .1;
8028
8029 cx.add_window(Default::default(), |cx| {
8030 let mut editor = build_editor(MultiBuffer::build_simple("🍐🏀✋", cx), cx);
8031
8032 editor.change_selections(None, cx, |s| s.select_ranges([4..4]));
8033 editor.transpose(&Default::default(), cx);
8034 assert_eq!(editor.text(cx), "🏀🍐✋");
8035 assert_eq!(editor.selections.ranges(cx), [8..8]);
8036
8037 editor.transpose(&Default::default(), cx);
8038 assert_eq!(editor.text(cx), "🏀✋🍐");
8039 assert_eq!(editor.selections.ranges(cx), [11..11]);
8040
8041 editor.transpose(&Default::default(), cx);
8042 assert_eq!(editor.text(cx), "🏀🍐✋");
8043 assert_eq!(editor.selections.ranges(cx), [11..11]);
8044
8045 editor
8046 })
8047 .1;
8048 }
8049
8050 #[gpui::test]
8051 async fn test_clipboard(cx: &mut gpui::TestAppContext) {
8052 let mut cx = EditorTestContext::new(cx).await;
8053
8054 cx.set_state("[one✅ }two [three }four [five }six ");
8055 cx.update_editor(|e, cx| e.cut(&Cut, cx));
8056 cx.assert_editor_state("|two |four |six ");
8057
8058 // Paste with three cursors. Each cursor pastes one slice of the clipboard text.
8059 cx.set_state("two |four |six |");
8060 cx.update_editor(|e, cx| e.paste(&Paste, cx));
8061 cx.assert_editor_state("two one✅ |four three |six five |");
8062
8063 // Paste again but with only two cursors. Since the number of cursors doesn't
8064 // match the number of slices in the clipboard, the entire clipboard text
8065 // is pasted at each cursor.
8066 cx.set_state("|two one✅ four three six five |");
8067 cx.update_editor(|e, cx| {
8068 e.handle_input(&Input("( ".into()), cx);
8069 e.paste(&Paste, cx);
8070 e.handle_input(&Input(") ".into()), cx);
8071 });
8072 cx.assert_editor_state(indoc! {"
8073 ( one✅
8074 three
8075 five ) |two one✅ four three six five ( one✅
8076 three
8077 five ) |"});
8078
8079 // Cut with three selections, one of which is full-line.
8080 cx.set_state(indoc! {"
8081 1[2}3
8082 4|567
8083 [8}9"});
8084 cx.update_editor(|e, cx| e.cut(&Cut, cx));
8085 cx.assert_editor_state(indoc! {"
8086 1|3
8087 |9"});
8088
8089 // Paste with three selections, noticing how the copied selection that was full-line
8090 // gets inserted before the second cursor.
8091 cx.set_state(indoc! {"
8092 1|3
8093 9|
8094 [o}ne"});
8095 cx.update_editor(|e, cx| e.paste(&Paste, cx));
8096 cx.assert_editor_state(indoc! {"
8097 12|3
8098 4567
8099 9|
8100 8|ne"});
8101
8102 // Copy with a single cursor only, which writes the whole line into the clipboard.
8103 cx.set_state(indoc! {"
8104 The quick brown
8105 fox ju|mps over
8106 the lazy dog"});
8107 cx.update_editor(|e, cx| e.copy(&Copy, cx));
8108 cx.assert_clipboard_content(Some("fox jumps over\n"));
8109
8110 // Paste with three selections, noticing how the copied full-line selection is inserted
8111 // before the empty selections but replaces the selection that is non-empty.
8112 cx.set_state(indoc! {"
8113 T|he quick brown
8114 [fo}x jumps over
8115 t|he lazy dog"});
8116 cx.update_editor(|e, cx| e.paste(&Paste, cx));
8117 cx.assert_editor_state(indoc! {"
8118 fox jumps over
8119 T|he quick brown
8120 fox jumps over
8121 |x jumps over
8122 fox jumps over
8123 t|he lazy dog"});
8124 }
8125
8126 #[gpui::test]
8127 fn test_select_all(cx: &mut gpui::MutableAppContext) {
8128 cx.set_global(Settings::test(cx));
8129 let buffer = MultiBuffer::build_simple("abc\nde\nfgh", cx);
8130 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
8131 view.update(cx, |view, cx| {
8132 view.select_all(&SelectAll, cx);
8133 assert_eq!(
8134 view.selections.display_ranges(cx),
8135 &[DisplayPoint::new(0, 0)..DisplayPoint::new(2, 3)]
8136 );
8137 });
8138 }
8139
8140 #[gpui::test]
8141 fn test_select_line(cx: &mut gpui::MutableAppContext) {
8142 cx.set_global(Settings::test(cx));
8143 let buffer = MultiBuffer::build_simple(&sample_text(6, 5, 'a'), cx);
8144 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
8145 view.update(cx, |view, cx| {
8146 view.change_selections(None, cx, |s| {
8147 s.select_display_ranges([
8148 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 1),
8149 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 2),
8150 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0),
8151 DisplayPoint::new(4, 2)..DisplayPoint::new(4, 2),
8152 ])
8153 });
8154 view.select_line(&SelectLine, cx);
8155 assert_eq!(
8156 view.selections.display_ranges(cx),
8157 vec![
8158 DisplayPoint::new(0, 0)..DisplayPoint::new(2, 0),
8159 DisplayPoint::new(4, 0)..DisplayPoint::new(5, 0),
8160 ]
8161 );
8162 });
8163
8164 view.update(cx, |view, cx| {
8165 view.select_line(&SelectLine, cx);
8166 assert_eq!(
8167 view.selections.display_ranges(cx),
8168 vec![
8169 DisplayPoint::new(0, 0)..DisplayPoint::new(3, 0),
8170 DisplayPoint::new(4, 0)..DisplayPoint::new(5, 5),
8171 ]
8172 );
8173 });
8174
8175 view.update(cx, |view, cx| {
8176 view.select_line(&SelectLine, cx);
8177 assert_eq!(
8178 view.selections.display_ranges(cx),
8179 vec![DisplayPoint::new(0, 0)..DisplayPoint::new(5, 5)]
8180 );
8181 });
8182 }
8183
8184 #[gpui::test]
8185 fn test_split_selection_into_lines(cx: &mut gpui::MutableAppContext) {
8186 cx.set_global(Settings::test(cx));
8187 let buffer = MultiBuffer::build_simple(&sample_text(9, 5, 'a'), cx);
8188 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
8189 view.update(cx, |view, cx| {
8190 view.fold_ranges(
8191 vec![
8192 Point::new(0, 2)..Point::new(1, 2),
8193 Point::new(2, 3)..Point::new(4, 1),
8194 Point::new(7, 0)..Point::new(8, 4),
8195 ],
8196 cx,
8197 );
8198 view.change_selections(None, cx, |s| {
8199 s.select_display_ranges([
8200 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 1),
8201 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 2),
8202 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0),
8203 DisplayPoint::new(4, 4)..DisplayPoint::new(4, 4),
8204 ])
8205 });
8206 assert_eq!(view.display_text(cx), "aa…bbb\nccc…eeee\nfffff\nggggg\n…i");
8207 });
8208
8209 view.update(cx, |view, cx| {
8210 view.split_selection_into_lines(&SplitSelectionIntoLines, cx);
8211 assert_eq!(
8212 view.display_text(cx),
8213 "aaaaa\nbbbbb\nccc…eeee\nfffff\nggggg\n…i"
8214 );
8215 assert_eq!(
8216 view.selections.display_ranges(cx),
8217 [
8218 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 1),
8219 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 2),
8220 DisplayPoint::new(2, 0)..DisplayPoint::new(2, 0),
8221 DisplayPoint::new(5, 4)..DisplayPoint::new(5, 4)
8222 ]
8223 );
8224 });
8225
8226 view.update(cx, |view, cx| {
8227 view.change_selections(None, cx, |s| {
8228 s.select_display_ranges([DisplayPoint::new(5, 0)..DisplayPoint::new(0, 1)])
8229 });
8230 view.split_selection_into_lines(&SplitSelectionIntoLines, cx);
8231 assert_eq!(
8232 view.display_text(cx),
8233 "aaaaa\nbbbbb\nccccc\nddddd\neeeee\nfffff\nggggg\nhhhhh\niiiii"
8234 );
8235 assert_eq!(
8236 view.selections.display_ranges(cx),
8237 [
8238 DisplayPoint::new(0, 5)..DisplayPoint::new(0, 5),
8239 DisplayPoint::new(1, 5)..DisplayPoint::new(1, 5),
8240 DisplayPoint::new(2, 5)..DisplayPoint::new(2, 5),
8241 DisplayPoint::new(3, 5)..DisplayPoint::new(3, 5),
8242 DisplayPoint::new(4, 5)..DisplayPoint::new(4, 5),
8243 DisplayPoint::new(5, 5)..DisplayPoint::new(5, 5),
8244 DisplayPoint::new(6, 5)..DisplayPoint::new(6, 5),
8245 DisplayPoint::new(7, 0)..DisplayPoint::new(7, 0)
8246 ]
8247 );
8248 });
8249 }
8250
8251 #[gpui::test]
8252 fn test_add_selection_above_below(cx: &mut gpui::MutableAppContext) {
8253 cx.set_global(Settings::test(cx));
8254 let buffer = MultiBuffer::build_simple("abc\ndefghi\n\njk\nlmno\n", cx);
8255 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
8256
8257 view.update(cx, |view, cx| {
8258 view.change_selections(None, cx, |s| {
8259 s.select_display_ranges([DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3)])
8260 });
8261 });
8262 view.update(cx, |view, cx| {
8263 view.add_selection_above(&AddSelectionAbove, cx);
8264 assert_eq!(
8265 view.selections.display_ranges(cx),
8266 vec![
8267 DisplayPoint::new(0, 3)..DisplayPoint::new(0, 3),
8268 DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3)
8269 ]
8270 );
8271 });
8272
8273 view.update(cx, |view, cx| {
8274 view.add_selection_above(&AddSelectionAbove, cx);
8275 assert_eq!(
8276 view.selections.display_ranges(cx),
8277 vec![
8278 DisplayPoint::new(0, 3)..DisplayPoint::new(0, 3),
8279 DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3)
8280 ]
8281 );
8282 });
8283
8284 view.update(cx, |view, cx| {
8285 view.add_selection_below(&AddSelectionBelow, cx);
8286 assert_eq!(
8287 view.selections.display_ranges(cx),
8288 vec![DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3)]
8289 );
8290
8291 view.undo_selection(&UndoSelection, cx);
8292 assert_eq!(
8293 view.selections.display_ranges(cx),
8294 vec![
8295 DisplayPoint::new(0, 3)..DisplayPoint::new(0, 3),
8296 DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3)
8297 ]
8298 );
8299
8300 view.redo_selection(&RedoSelection, cx);
8301 assert_eq!(
8302 view.selections.display_ranges(cx),
8303 vec![DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3)]
8304 );
8305 });
8306
8307 view.update(cx, |view, cx| {
8308 view.add_selection_below(&AddSelectionBelow, cx);
8309 assert_eq!(
8310 view.selections.display_ranges(cx),
8311 vec![
8312 DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3),
8313 DisplayPoint::new(4, 3)..DisplayPoint::new(4, 3)
8314 ]
8315 );
8316 });
8317
8318 view.update(cx, |view, cx| {
8319 view.add_selection_below(&AddSelectionBelow, cx);
8320 assert_eq!(
8321 view.selections.display_ranges(cx),
8322 vec![
8323 DisplayPoint::new(1, 3)..DisplayPoint::new(1, 3),
8324 DisplayPoint::new(4, 3)..DisplayPoint::new(4, 3)
8325 ]
8326 );
8327 });
8328
8329 view.update(cx, |view, cx| {
8330 view.change_selections(None, cx, |s| {
8331 s.select_display_ranges([DisplayPoint::new(1, 4)..DisplayPoint::new(1, 3)])
8332 });
8333 });
8334 view.update(cx, |view, cx| {
8335 view.add_selection_below(&AddSelectionBelow, cx);
8336 assert_eq!(
8337 view.selections.display_ranges(cx),
8338 vec![
8339 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 3),
8340 DisplayPoint::new(4, 4)..DisplayPoint::new(4, 3)
8341 ]
8342 );
8343 });
8344
8345 view.update(cx, |view, cx| {
8346 view.add_selection_below(&AddSelectionBelow, cx);
8347 assert_eq!(
8348 view.selections.display_ranges(cx),
8349 vec![
8350 DisplayPoint::new(1, 4)..DisplayPoint::new(1, 3),
8351 DisplayPoint::new(4, 4)..DisplayPoint::new(4, 3)
8352 ]
8353 );
8354 });
8355
8356 view.update(cx, |view, cx| {
8357 view.add_selection_above(&AddSelectionAbove, cx);
8358 assert_eq!(
8359 view.selections.display_ranges(cx),
8360 vec![DisplayPoint::new(1, 4)..DisplayPoint::new(1, 3)]
8361 );
8362 });
8363
8364 view.update(cx, |view, cx| {
8365 view.add_selection_above(&AddSelectionAbove, cx);
8366 assert_eq!(
8367 view.selections.display_ranges(cx),
8368 vec![DisplayPoint::new(1, 4)..DisplayPoint::new(1, 3)]
8369 );
8370 });
8371
8372 view.update(cx, |view, cx| {
8373 view.change_selections(None, cx, |s| {
8374 s.select_display_ranges([DisplayPoint::new(0, 1)..DisplayPoint::new(1, 4)])
8375 });
8376 view.add_selection_below(&AddSelectionBelow, cx);
8377 assert_eq!(
8378 view.selections.display_ranges(cx),
8379 vec![
8380 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 3),
8381 DisplayPoint::new(1, 1)..DisplayPoint::new(1, 4),
8382 DisplayPoint::new(3, 1)..DisplayPoint::new(3, 2),
8383 ]
8384 );
8385 });
8386
8387 view.update(cx, |view, cx| {
8388 view.add_selection_below(&AddSelectionBelow, cx);
8389 assert_eq!(
8390 view.selections.display_ranges(cx),
8391 vec![
8392 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 3),
8393 DisplayPoint::new(1, 1)..DisplayPoint::new(1, 4),
8394 DisplayPoint::new(3, 1)..DisplayPoint::new(3, 2),
8395 DisplayPoint::new(4, 1)..DisplayPoint::new(4, 4),
8396 ]
8397 );
8398 });
8399
8400 view.update(cx, |view, cx| {
8401 view.add_selection_above(&AddSelectionAbove, cx);
8402 assert_eq!(
8403 view.selections.display_ranges(cx),
8404 vec![
8405 DisplayPoint::new(0, 1)..DisplayPoint::new(0, 3),
8406 DisplayPoint::new(1, 1)..DisplayPoint::new(1, 4),
8407 DisplayPoint::new(3, 1)..DisplayPoint::new(3, 2),
8408 ]
8409 );
8410 });
8411
8412 view.update(cx, |view, cx| {
8413 view.change_selections(None, cx, |s| {
8414 s.select_display_ranges([DisplayPoint::new(4, 3)..DisplayPoint::new(1, 1)])
8415 });
8416 });
8417 view.update(cx, |view, cx| {
8418 view.add_selection_above(&AddSelectionAbove, cx);
8419 assert_eq!(
8420 view.selections.display_ranges(cx),
8421 vec![
8422 DisplayPoint::new(0, 3)..DisplayPoint::new(0, 1),
8423 DisplayPoint::new(1, 3)..DisplayPoint::new(1, 1),
8424 DisplayPoint::new(3, 2)..DisplayPoint::new(3, 1),
8425 DisplayPoint::new(4, 3)..DisplayPoint::new(4, 1),
8426 ]
8427 );
8428 });
8429
8430 view.update(cx, |view, cx| {
8431 view.add_selection_below(&AddSelectionBelow, cx);
8432 assert_eq!(
8433 view.selections.display_ranges(cx),
8434 vec![
8435 DisplayPoint::new(1, 3)..DisplayPoint::new(1, 1),
8436 DisplayPoint::new(3, 2)..DisplayPoint::new(3, 1),
8437 DisplayPoint::new(4, 3)..DisplayPoint::new(4, 1),
8438 ]
8439 );
8440 });
8441 }
8442
8443 #[gpui::test]
8444 fn test_select_next(cx: &mut gpui::MutableAppContext) {
8445 cx.set_global(Settings::test(cx));
8446
8447 let (text, ranges) = marked_text_ranges("[abc]\n[abc] [abc]\ndefabc\n[abc]");
8448 let buffer = MultiBuffer::build_simple(&text, cx);
8449 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(buffer, cx));
8450
8451 view.update(cx, |view, cx| {
8452 view.change_selections(None, cx, |s| {
8453 s.select_ranges([ranges[1].start + 1..ranges[1].start + 1])
8454 });
8455 view.select_next(
8456 &SelectNext {
8457 replace_newest: false,
8458 },
8459 cx,
8460 );
8461 assert_eq!(view.selections.ranges(cx), &ranges[1..2]);
8462
8463 view.select_next(
8464 &SelectNext {
8465 replace_newest: false,
8466 },
8467 cx,
8468 );
8469 assert_eq!(view.selections.ranges(cx), &ranges[1..3]);
8470
8471 view.undo_selection(&UndoSelection, cx);
8472 assert_eq!(view.selections.ranges(cx), &ranges[1..2]);
8473
8474 view.redo_selection(&RedoSelection, cx);
8475 assert_eq!(view.selections.ranges(cx), &ranges[1..3]);
8476
8477 view.select_next(
8478 &SelectNext {
8479 replace_newest: false,
8480 },
8481 cx,
8482 );
8483 assert_eq!(view.selections.ranges(cx), &ranges[1..4]);
8484
8485 view.select_next(
8486 &SelectNext {
8487 replace_newest: false,
8488 },
8489 cx,
8490 );
8491 assert_eq!(view.selections.ranges(cx), &ranges[0..4]);
8492 });
8493 }
8494
8495 #[gpui::test]
8496 async fn test_select_larger_smaller_syntax_node(cx: &mut gpui::TestAppContext) {
8497 cx.update(|cx| cx.set_global(Settings::test(cx)));
8498 let language = Arc::new(Language::new(
8499 LanguageConfig::default(),
8500 Some(tree_sitter_rust::language()),
8501 ));
8502
8503 let text = r#"
8504 use mod1::mod2::{mod3, mod4};
8505
8506 fn fn_1(param1: bool, param2: &str) {
8507 let var1 = "text";
8508 }
8509 "#
8510 .unindent();
8511
8512 let buffer = cx.add_model(|cx| Buffer::new(0, text, cx).with_language(language, cx));
8513 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
8514 let (_, view) = cx.add_window(|cx| build_editor(buffer, cx));
8515 view.condition(&cx, |view, cx| !view.buffer.read(cx).is_parsing(cx))
8516 .await;
8517
8518 view.update(cx, |view, cx| {
8519 view.change_selections(None, cx, |s| {
8520 s.select_display_ranges([
8521 DisplayPoint::new(0, 25)..DisplayPoint::new(0, 25),
8522 DisplayPoint::new(2, 24)..DisplayPoint::new(2, 12),
8523 DisplayPoint::new(3, 18)..DisplayPoint::new(3, 18),
8524 ]);
8525 });
8526 view.select_larger_syntax_node(&SelectLargerSyntaxNode, cx);
8527 });
8528 assert_eq!(
8529 view.update(cx, |view, cx| { view.selections.display_ranges(cx) }),
8530 &[
8531 DisplayPoint::new(0, 23)..DisplayPoint::new(0, 27),
8532 DisplayPoint::new(2, 35)..DisplayPoint::new(2, 7),
8533 DisplayPoint::new(3, 15)..DisplayPoint::new(3, 21),
8534 ]
8535 );
8536
8537 view.update(cx, |view, cx| {
8538 view.select_larger_syntax_node(&SelectLargerSyntaxNode, cx);
8539 });
8540 assert_eq!(
8541 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8542 &[
8543 DisplayPoint::new(0, 16)..DisplayPoint::new(0, 28),
8544 DisplayPoint::new(4, 1)..DisplayPoint::new(2, 0),
8545 ]
8546 );
8547
8548 view.update(cx, |view, cx| {
8549 view.select_larger_syntax_node(&SelectLargerSyntaxNode, cx);
8550 });
8551 assert_eq!(
8552 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8553 &[DisplayPoint::new(5, 0)..DisplayPoint::new(0, 0)]
8554 );
8555
8556 // Trying to expand the selected syntax node one more time has no effect.
8557 view.update(cx, |view, cx| {
8558 view.select_larger_syntax_node(&SelectLargerSyntaxNode, cx);
8559 });
8560 assert_eq!(
8561 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8562 &[DisplayPoint::new(5, 0)..DisplayPoint::new(0, 0)]
8563 );
8564
8565 view.update(cx, |view, cx| {
8566 view.select_smaller_syntax_node(&SelectSmallerSyntaxNode, cx);
8567 });
8568 assert_eq!(
8569 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8570 &[
8571 DisplayPoint::new(0, 16)..DisplayPoint::new(0, 28),
8572 DisplayPoint::new(4, 1)..DisplayPoint::new(2, 0),
8573 ]
8574 );
8575
8576 view.update(cx, |view, cx| {
8577 view.select_smaller_syntax_node(&SelectSmallerSyntaxNode, cx);
8578 });
8579 assert_eq!(
8580 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8581 &[
8582 DisplayPoint::new(0, 23)..DisplayPoint::new(0, 27),
8583 DisplayPoint::new(2, 35)..DisplayPoint::new(2, 7),
8584 DisplayPoint::new(3, 15)..DisplayPoint::new(3, 21),
8585 ]
8586 );
8587
8588 view.update(cx, |view, cx| {
8589 view.select_smaller_syntax_node(&SelectSmallerSyntaxNode, cx);
8590 });
8591 assert_eq!(
8592 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8593 &[
8594 DisplayPoint::new(0, 25)..DisplayPoint::new(0, 25),
8595 DisplayPoint::new(2, 24)..DisplayPoint::new(2, 12),
8596 DisplayPoint::new(3, 18)..DisplayPoint::new(3, 18),
8597 ]
8598 );
8599
8600 // Trying to shrink the selected syntax node one more time has no effect.
8601 view.update(cx, |view, cx| {
8602 view.select_smaller_syntax_node(&SelectSmallerSyntaxNode, cx);
8603 });
8604 assert_eq!(
8605 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8606 &[
8607 DisplayPoint::new(0, 25)..DisplayPoint::new(0, 25),
8608 DisplayPoint::new(2, 24)..DisplayPoint::new(2, 12),
8609 DisplayPoint::new(3, 18)..DisplayPoint::new(3, 18),
8610 ]
8611 );
8612
8613 // Ensure that we keep expanding the selection if the larger selection starts or ends within
8614 // a fold.
8615 view.update(cx, |view, cx| {
8616 view.fold_ranges(
8617 vec![
8618 Point::new(0, 21)..Point::new(0, 24),
8619 Point::new(3, 20)..Point::new(3, 22),
8620 ],
8621 cx,
8622 );
8623 view.select_larger_syntax_node(&SelectLargerSyntaxNode, cx);
8624 });
8625 assert_eq!(
8626 view.update(cx, |view, cx| view.selections.display_ranges(cx)),
8627 &[
8628 DisplayPoint::new(0, 16)..DisplayPoint::new(0, 28),
8629 DisplayPoint::new(2, 35)..DisplayPoint::new(2, 7),
8630 DisplayPoint::new(3, 4)..DisplayPoint::new(3, 23),
8631 ]
8632 );
8633 }
8634
8635 #[gpui::test]
8636 async fn test_autoindent_selections(cx: &mut gpui::TestAppContext) {
8637 cx.update(|cx| cx.set_global(Settings::test(cx)));
8638 let language = Arc::new(
8639 Language::new(
8640 LanguageConfig {
8641 brackets: vec![
8642 BracketPair {
8643 start: "{".to_string(),
8644 end: "}".to_string(),
8645 close: false,
8646 newline: true,
8647 },
8648 BracketPair {
8649 start: "(".to_string(),
8650 end: ")".to_string(),
8651 close: false,
8652 newline: true,
8653 },
8654 ],
8655 ..Default::default()
8656 },
8657 Some(tree_sitter_rust::language()),
8658 )
8659 .with_indents_query(
8660 r#"
8661 (_ "(" ")" @end) @indent
8662 (_ "{" "}" @end) @indent
8663 "#,
8664 )
8665 .unwrap(),
8666 );
8667
8668 let text = "fn a() {}";
8669
8670 let buffer = cx.add_model(|cx| Buffer::new(0, text, cx).with_language(language, cx));
8671 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
8672 let (_, editor) = cx.add_window(|cx| build_editor(buffer, cx));
8673 editor
8674 .condition(&cx, |editor, cx| !editor.buffer.read(cx).is_parsing(cx))
8675 .await;
8676
8677 editor.update(cx, |editor, cx| {
8678 editor.change_selections(None, cx, |s| s.select_ranges([5..5, 8..8, 9..9]));
8679 editor.newline(&Newline, cx);
8680 assert_eq!(editor.text(cx), "fn a(\n \n) {\n \n}\n");
8681 assert_eq!(
8682 editor.selections.ranges(cx),
8683 &[
8684 Point::new(1, 4)..Point::new(1, 4),
8685 Point::new(3, 4)..Point::new(3, 4),
8686 Point::new(5, 0)..Point::new(5, 0)
8687 ]
8688 );
8689 });
8690 }
8691
8692 #[gpui::test]
8693 async fn test_autoclose_pairs(cx: &mut gpui::TestAppContext) {
8694 cx.update(|cx| cx.set_global(Settings::test(cx)));
8695 let language = Arc::new(Language::new(
8696 LanguageConfig {
8697 brackets: vec![
8698 BracketPair {
8699 start: "{".to_string(),
8700 end: "}".to_string(),
8701 close: true,
8702 newline: true,
8703 },
8704 BracketPair {
8705 start: "/*".to_string(),
8706 end: " */".to_string(),
8707 close: true,
8708 newline: true,
8709 },
8710 ],
8711 autoclose_before: "})]".to_string(),
8712 ..Default::default()
8713 },
8714 Some(tree_sitter_rust::language()),
8715 ));
8716
8717 let text = r#"
8718 a
8719
8720 /
8721
8722 "#
8723 .unindent();
8724
8725 let buffer = cx.add_model(|cx| Buffer::new(0, text, cx).with_language(language, cx));
8726 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
8727 let (_, view) = cx.add_window(|cx| build_editor(buffer, cx));
8728 view.condition(&cx, |view, cx| !view.buffer.read(cx).is_parsing(cx))
8729 .await;
8730
8731 view.update(cx, |view, cx| {
8732 view.change_selections(None, cx, |s| {
8733 s.select_display_ranges([
8734 DisplayPoint::new(0, 0)..DisplayPoint::new(0, 1),
8735 DisplayPoint::new(1, 0)..DisplayPoint::new(1, 0),
8736 ])
8737 });
8738
8739 view.handle_input(&Input("{".to_string()), cx);
8740 view.handle_input(&Input("{".to_string()), cx);
8741 view.handle_input(&Input("{".to_string()), cx);
8742 assert_eq!(
8743 view.text(cx),
8744 "
8745 {{{}}}
8746 {{{}}}
8747 /
8748
8749 "
8750 .unindent()
8751 );
8752
8753 view.move_right(&MoveRight, cx);
8754 view.handle_input(&Input("}".to_string()), cx);
8755 view.handle_input(&Input("}".to_string()), cx);
8756 view.handle_input(&Input("}".to_string()), cx);
8757 assert_eq!(
8758 view.text(cx),
8759 "
8760 {{{}}}}
8761 {{{}}}}
8762 /
8763
8764 "
8765 .unindent()
8766 );
8767
8768 view.undo(&Undo, cx);
8769 view.handle_input(&Input("/".to_string()), cx);
8770 view.handle_input(&Input("*".to_string()), cx);
8771 assert_eq!(
8772 view.text(cx),
8773 "
8774 /* */
8775 /* */
8776 /
8777
8778 "
8779 .unindent()
8780 );
8781
8782 view.undo(&Undo, cx);
8783 view.change_selections(None, cx, |s| {
8784 s.select_display_ranges([
8785 DisplayPoint::new(2, 1)..DisplayPoint::new(2, 1),
8786 DisplayPoint::new(3, 0)..DisplayPoint::new(3, 0),
8787 ])
8788 });
8789 view.handle_input(&Input("*".to_string()), cx);
8790 assert_eq!(
8791 view.text(cx),
8792 "
8793 a
8794
8795 /*
8796 *
8797 "
8798 .unindent()
8799 );
8800
8801 // Don't autoclose if the next character isn't whitespace and isn't
8802 // listed in the language's "autoclose_before" section.
8803 view.finalize_last_transaction(cx);
8804 view.change_selections(None, cx, |s| {
8805 s.select_display_ranges([DisplayPoint::new(0, 0)..DisplayPoint::new(0, 0)])
8806 });
8807 view.handle_input(&Input("{".to_string()), cx);
8808 assert_eq!(
8809 view.text(cx),
8810 "
8811 {a
8812
8813 /*
8814 *
8815 "
8816 .unindent()
8817 );
8818
8819 view.undo(&Undo, cx);
8820 view.change_selections(None, cx, |s| {
8821 s.select_display_ranges([DisplayPoint::new(0, 0)..DisplayPoint::new(0, 1)])
8822 });
8823 view.handle_input(&Input("{".to_string()), cx);
8824 assert_eq!(
8825 view.text(cx),
8826 "
8827 {a}
8828
8829 /*
8830 *
8831 "
8832 .unindent()
8833 );
8834 assert_eq!(
8835 view.selections.display_ranges(cx),
8836 [DisplayPoint::new(0, 1)..DisplayPoint::new(0, 2)]
8837 );
8838 });
8839 }
8840
8841 #[gpui::test]
8842 async fn test_snippets(cx: &mut gpui::TestAppContext) {
8843 cx.update(|cx| cx.set_global(Settings::test(cx)));
8844
8845 let (text, insertion_ranges) = marked_text_ranges(indoc! {"
8846 a.| b
8847 a.| b
8848 a.| b"});
8849 let buffer = cx.update(|cx| MultiBuffer::build_simple(&text, cx));
8850 let (_, editor) = cx.add_window(|cx| build_editor(buffer, cx));
8851
8852 editor.update(cx, |editor, cx| {
8853 let snippet = Snippet::parse("f(${1:one}, ${2:two}, ${1:three})$0").unwrap();
8854
8855 editor
8856 .insert_snippet(&insertion_ranges, snippet, cx)
8857 .unwrap();
8858
8859 fn assert(editor: &mut Editor, cx: &mut ViewContext<Editor>, marked_text_ranges: &str) {
8860 let range_markers = ('<', '>');
8861 let (expected_text, mut selection_ranges_lookup) =
8862 marked_text_ranges_by(marked_text_ranges, vec![range_markers.clone().into()]);
8863 let selection_ranges = selection_ranges_lookup
8864 .remove(&range_markers.into())
8865 .unwrap();
8866 assert_eq!(editor.text(cx), expected_text);
8867 assert_eq!(editor.selections.ranges::<usize>(cx), selection_ranges);
8868 }
8869 assert(
8870 editor,
8871 cx,
8872 indoc! {"
8873 a.f(<one>, two, <three>) b
8874 a.f(<one>, two, <three>) b
8875 a.f(<one>, two, <three>) b"},
8876 );
8877
8878 // Can't move earlier than the first tab stop
8879 assert!(!editor.move_to_prev_snippet_tabstop(cx));
8880 assert(
8881 editor,
8882 cx,
8883 indoc! {"
8884 a.f(<one>, two, <three>) b
8885 a.f(<one>, two, <three>) b
8886 a.f(<one>, two, <three>) b"},
8887 );
8888
8889 assert!(editor.move_to_next_snippet_tabstop(cx));
8890 assert(
8891 editor,
8892 cx,
8893 indoc! {"
8894 a.f(one, <two>, three) b
8895 a.f(one, <two>, three) b
8896 a.f(one, <two>, three) b"},
8897 );
8898
8899 editor.move_to_prev_snippet_tabstop(cx);
8900 assert(
8901 editor,
8902 cx,
8903 indoc! {"
8904 a.f(<one>, two, <three>) b
8905 a.f(<one>, two, <three>) b
8906 a.f(<one>, two, <three>) b"},
8907 );
8908
8909 assert!(editor.move_to_next_snippet_tabstop(cx));
8910 assert(
8911 editor,
8912 cx,
8913 indoc! {"
8914 a.f(one, <two>, three) b
8915 a.f(one, <two>, three) b
8916 a.f(one, <two>, three) b"},
8917 );
8918 assert!(editor.move_to_next_snippet_tabstop(cx));
8919 assert(
8920 editor,
8921 cx,
8922 indoc! {"
8923 a.f(one, two, three)<> b
8924 a.f(one, two, three)<> b
8925 a.f(one, two, three)<> b"},
8926 );
8927
8928 // As soon as the last tab stop is reached, snippet state is gone
8929 editor.move_to_prev_snippet_tabstop(cx);
8930 assert(
8931 editor,
8932 cx,
8933 indoc! {"
8934 a.f(one, two, three)<> b
8935 a.f(one, two, three)<> b
8936 a.f(one, two, three)<> b"},
8937 );
8938 });
8939 }
8940
8941 #[gpui::test]
8942 async fn test_document_format_during_save(cx: &mut gpui::TestAppContext) {
8943 cx.foreground().forbid_parking();
8944
8945 let mut language = Language::new(
8946 LanguageConfig {
8947 name: "Rust".into(),
8948 path_suffixes: vec!["rs".to_string()],
8949 ..Default::default()
8950 },
8951 Some(tree_sitter_rust::language()),
8952 );
8953 let mut fake_servers = language.set_fake_lsp_adapter(FakeLspAdapter {
8954 capabilities: lsp::ServerCapabilities {
8955 document_formatting_provider: Some(lsp::OneOf::Left(true)),
8956 ..Default::default()
8957 },
8958 ..Default::default()
8959 });
8960
8961 let fs = FakeFs::new(cx.background().clone());
8962 fs.insert_file("/file.rs", Default::default()).await;
8963
8964 let project = Project::test(fs, ["/file.rs".as_ref()], cx).await;
8965 project.update(cx, |project, _| project.languages().add(Arc::new(language)));
8966 let buffer = project
8967 .update(cx, |project, cx| project.open_local_buffer("/file.rs", cx))
8968 .await
8969 .unwrap();
8970
8971 cx.foreground().start_waiting();
8972 let fake_server = fake_servers.next().await.unwrap();
8973
8974 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
8975 let (_, editor) = cx.add_window(|cx| build_editor(buffer, cx));
8976 editor.update(cx, |editor, cx| editor.set_text("one\ntwo\nthree\n", cx));
8977 assert!(cx.read(|cx| editor.is_dirty(cx)));
8978
8979 let save = cx.update(|cx| editor.save(project.clone(), cx));
8980 fake_server
8981 .handle_request::<lsp::request::Formatting, _, _>(move |params, _| async move {
8982 assert_eq!(
8983 params.text_document.uri,
8984 lsp::Url::from_file_path("/file.rs").unwrap()
8985 );
8986 assert_eq!(params.options.tab_size, 4);
8987 Ok(Some(vec![lsp::TextEdit::new(
8988 lsp::Range::new(lsp::Position::new(0, 3), lsp::Position::new(1, 0)),
8989 ", ".to_string(),
8990 )]))
8991 })
8992 .next()
8993 .await;
8994 cx.foreground().start_waiting();
8995 save.await.unwrap();
8996 assert_eq!(
8997 editor.read_with(cx, |editor, cx| editor.text(cx)),
8998 "one, two\nthree\n"
8999 );
9000 assert!(!cx.read(|cx| editor.is_dirty(cx)));
9001
9002 editor.update(cx, |editor, cx| editor.set_text("one\ntwo\nthree\n", cx));
9003 assert!(cx.read(|cx| editor.is_dirty(cx)));
9004
9005 // Ensure we can still save even if formatting hangs.
9006 fake_server.handle_request::<lsp::request::Formatting, _, _>(move |params, _| async move {
9007 assert_eq!(
9008 params.text_document.uri,
9009 lsp::Url::from_file_path("/file.rs").unwrap()
9010 );
9011 futures::future::pending::<()>().await;
9012 unreachable!()
9013 });
9014 let save = cx.update(|cx| editor.save(project.clone(), cx));
9015 cx.foreground().advance_clock(items::FORMAT_TIMEOUT);
9016 cx.foreground().start_waiting();
9017 save.await.unwrap();
9018 assert_eq!(
9019 editor.read_with(cx, |editor, cx| editor.text(cx)),
9020 "one\ntwo\nthree\n"
9021 );
9022 assert!(!cx.read(|cx| editor.is_dirty(cx)));
9023
9024 // Set rust language override and assert overriden tabsize is sent to language server
9025 cx.update(|cx| {
9026 cx.update_global::<Settings, _, _>(|settings, _| {
9027 settings.language_overrides.insert(
9028 "Rust".into(),
9029 LanguageOverride {
9030 tab_size: Some(8),
9031 ..Default::default()
9032 },
9033 );
9034 })
9035 });
9036
9037 let save = cx.update(|cx| editor.save(project.clone(), cx));
9038 fake_server
9039 .handle_request::<lsp::request::Formatting, _, _>(move |params, _| async move {
9040 assert_eq!(
9041 params.text_document.uri,
9042 lsp::Url::from_file_path("/file.rs").unwrap()
9043 );
9044 assert_eq!(params.options.tab_size, 8);
9045 Ok(Some(vec![]))
9046 })
9047 .next()
9048 .await;
9049 cx.foreground().start_waiting();
9050 save.await.unwrap();
9051 }
9052
9053 #[gpui::test]
9054 async fn test_range_format_during_save(cx: &mut gpui::TestAppContext) {
9055 cx.foreground().forbid_parking();
9056
9057 let mut language = Language::new(
9058 LanguageConfig {
9059 name: "Rust".into(),
9060 path_suffixes: vec!["rs".to_string()],
9061 ..Default::default()
9062 },
9063 Some(tree_sitter_rust::language()),
9064 );
9065 let mut fake_servers = language.set_fake_lsp_adapter(FakeLspAdapter {
9066 capabilities: lsp::ServerCapabilities {
9067 document_range_formatting_provider: Some(lsp::OneOf::Left(true)),
9068 ..Default::default()
9069 },
9070 ..Default::default()
9071 });
9072
9073 let fs = FakeFs::new(cx.background().clone());
9074 fs.insert_file("/file.rs", Default::default()).await;
9075
9076 let project = Project::test(fs, ["/file.rs".as_ref()], cx).await;
9077 project.update(cx, |project, _| project.languages().add(Arc::new(language)));
9078 let buffer = project
9079 .update(cx, |project, cx| project.open_local_buffer("/file.rs", cx))
9080 .await
9081 .unwrap();
9082
9083 cx.foreground().start_waiting();
9084 let fake_server = fake_servers.next().await.unwrap();
9085
9086 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
9087 let (_, editor) = cx.add_window(|cx| build_editor(buffer, cx));
9088 editor.update(cx, |editor, cx| editor.set_text("one\ntwo\nthree\n", cx));
9089 assert!(cx.read(|cx| editor.is_dirty(cx)));
9090
9091 let save = cx.update(|cx| editor.save(project.clone(), cx));
9092 fake_server
9093 .handle_request::<lsp::request::RangeFormatting, _, _>(move |params, _| async move {
9094 assert_eq!(
9095 params.text_document.uri,
9096 lsp::Url::from_file_path("/file.rs").unwrap()
9097 );
9098 assert_eq!(params.options.tab_size, 4);
9099 Ok(Some(vec![lsp::TextEdit::new(
9100 lsp::Range::new(lsp::Position::new(0, 3), lsp::Position::new(1, 0)),
9101 ", ".to_string(),
9102 )]))
9103 })
9104 .next()
9105 .await;
9106 cx.foreground().start_waiting();
9107 save.await.unwrap();
9108 assert_eq!(
9109 editor.read_with(cx, |editor, cx| editor.text(cx)),
9110 "one, two\nthree\n"
9111 );
9112 assert!(!cx.read(|cx| editor.is_dirty(cx)));
9113
9114 editor.update(cx, |editor, cx| editor.set_text("one\ntwo\nthree\n", cx));
9115 assert!(cx.read(|cx| editor.is_dirty(cx)));
9116
9117 // Ensure we can still save even if formatting hangs.
9118 fake_server.handle_request::<lsp::request::RangeFormatting, _, _>(
9119 move |params, _| async move {
9120 assert_eq!(
9121 params.text_document.uri,
9122 lsp::Url::from_file_path("/file.rs").unwrap()
9123 );
9124 futures::future::pending::<()>().await;
9125 unreachable!()
9126 },
9127 );
9128 let save = cx.update(|cx| editor.save(project.clone(), cx));
9129 cx.foreground().advance_clock(items::FORMAT_TIMEOUT);
9130 cx.foreground().start_waiting();
9131 save.await.unwrap();
9132 assert_eq!(
9133 editor.read_with(cx, |editor, cx| editor.text(cx)),
9134 "one\ntwo\nthree\n"
9135 );
9136 assert!(!cx.read(|cx| editor.is_dirty(cx)));
9137
9138 // Set rust language override and assert overriden tabsize is sent to language server
9139 cx.update(|cx| {
9140 cx.update_global::<Settings, _, _>(|settings, _| {
9141 settings.language_overrides.insert(
9142 "Rust".into(),
9143 LanguageOverride {
9144 tab_size: Some(8),
9145 ..Default::default()
9146 },
9147 );
9148 })
9149 });
9150
9151 let save = cx.update(|cx| editor.save(project.clone(), cx));
9152 fake_server
9153 .handle_request::<lsp::request::RangeFormatting, _, _>(move |params, _| async move {
9154 assert_eq!(
9155 params.text_document.uri,
9156 lsp::Url::from_file_path("/file.rs").unwrap()
9157 );
9158 assert_eq!(params.options.tab_size, 8);
9159 Ok(Some(vec![]))
9160 })
9161 .next()
9162 .await;
9163 cx.foreground().start_waiting();
9164 save.await.unwrap();
9165 }
9166
9167 #[gpui::test]
9168 async fn test_completion(cx: &mut gpui::TestAppContext) {
9169 let mut language = Language::new(
9170 LanguageConfig {
9171 name: "Rust".into(),
9172 path_suffixes: vec!["rs".to_string()],
9173 ..Default::default()
9174 },
9175 Some(tree_sitter_rust::language()),
9176 );
9177 let mut fake_servers = language.set_fake_lsp_adapter(FakeLspAdapter {
9178 capabilities: lsp::ServerCapabilities {
9179 completion_provider: Some(lsp::CompletionOptions {
9180 trigger_characters: Some(vec![".".to_string(), ":".to_string()]),
9181 ..Default::default()
9182 }),
9183 ..Default::default()
9184 },
9185 ..Default::default()
9186 });
9187
9188 let text = "
9189 one
9190 two
9191 three
9192 "
9193 .unindent();
9194
9195 let fs = FakeFs::new(cx.background().clone());
9196 fs.insert_file("/file.rs", text).await;
9197
9198 let project = Project::test(fs, ["/file.rs".as_ref()], cx).await;
9199 project.update(cx, |project, _| project.languages().add(Arc::new(language)));
9200 let buffer = project
9201 .update(cx, |project, cx| project.open_local_buffer("/file.rs", cx))
9202 .await
9203 .unwrap();
9204 let mut fake_server = fake_servers.next().await.unwrap();
9205
9206 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
9207 let (_, editor) = cx.add_window(|cx| build_editor(buffer, cx));
9208
9209 editor.update(cx, |editor, cx| {
9210 editor.project = Some(project);
9211 editor.change_selections(None, cx, |s| {
9212 s.select_ranges([Point::new(0, 3)..Point::new(0, 3)])
9213 });
9214 editor.handle_input(&Input(".".to_string()), cx);
9215 });
9216
9217 handle_completion_request(
9218 &mut fake_server,
9219 "/file.rs",
9220 Point::new(0, 4),
9221 vec![
9222 (Point::new(0, 4)..Point::new(0, 4), "first_completion"),
9223 (Point::new(0, 4)..Point::new(0, 4), "second_completion"),
9224 ],
9225 )
9226 .await;
9227 editor
9228 .condition(&cx, |editor, _| editor.context_menu_visible())
9229 .await;
9230
9231 let apply_additional_edits = editor.update(cx, |editor, cx| {
9232 editor.move_down(&MoveDown, cx);
9233 let apply_additional_edits = editor
9234 .confirm_completion(&ConfirmCompletion::default(), cx)
9235 .unwrap();
9236 assert_eq!(
9237 editor.text(cx),
9238 "
9239 one.second_completion
9240 two
9241 three
9242 "
9243 .unindent()
9244 );
9245 apply_additional_edits
9246 });
9247
9248 handle_resolve_completion_request(
9249 &mut fake_server,
9250 Some((Point::new(2, 5)..Point::new(2, 5), "\nadditional edit")),
9251 )
9252 .await;
9253 apply_additional_edits.await.unwrap();
9254 assert_eq!(
9255 editor.read_with(cx, |editor, cx| editor.text(cx)),
9256 "
9257 one.second_completion
9258 two
9259 three
9260 additional edit
9261 "
9262 .unindent()
9263 );
9264
9265 editor.update(cx, |editor, cx| {
9266 editor.change_selections(None, cx, |s| {
9267 s.select_ranges([
9268 Point::new(1, 3)..Point::new(1, 3),
9269 Point::new(2, 5)..Point::new(2, 5),
9270 ])
9271 });
9272
9273 editor.handle_input(&Input(" ".to_string()), cx);
9274 assert!(editor.context_menu.is_none());
9275 editor.handle_input(&Input("s".to_string()), cx);
9276 assert!(editor.context_menu.is_none());
9277 });
9278
9279 handle_completion_request(
9280 &mut fake_server,
9281 "/file.rs",
9282 Point::new(2, 7),
9283 vec![
9284 (Point::new(2, 6)..Point::new(2, 7), "fourth_completion"),
9285 (Point::new(2, 6)..Point::new(2, 7), "fifth_completion"),
9286 (Point::new(2, 6)..Point::new(2, 7), "sixth_completion"),
9287 ],
9288 )
9289 .await;
9290 editor
9291 .condition(&cx, |editor, _| editor.context_menu_visible())
9292 .await;
9293
9294 editor.update(cx, |editor, cx| {
9295 editor.handle_input(&Input("i".to_string()), cx);
9296 });
9297
9298 handle_completion_request(
9299 &mut fake_server,
9300 "/file.rs",
9301 Point::new(2, 8),
9302 vec![
9303 (Point::new(2, 6)..Point::new(2, 8), "fourth_completion"),
9304 (Point::new(2, 6)..Point::new(2, 8), "fifth_completion"),
9305 (Point::new(2, 6)..Point::new(2, 8), "sixth_completion"),
9306 ],
9307 )
9308 .await;
9309 editor
9310 .condition(&cx, |editor, _| editor.context_menu_visible())
9311 .await;
9312
9313 let apply_additional_edits = editor.update(cx, |editor, cx| {
9314 let apply_additional_edits = editor
9315 .confirm_completion(&ConfirmCompletion::default(), cx)
9316 .unwrap();
9317 assert_eq!(
9318 editor.text(cx),
9319 "
9320 one.second_completion
9321 two sixth_completion
9322 three sixth_completion
9323 additional edit
9324 "
9325 .unindent()
9326 );
9327 apply_additional_edits
9328 });
9329 handle_resolve_completion_request(&mut fake_server, None).await;
9330 apply_additional_edits.await.unwrap();
9331
9332 async fn handle_completion_request(
9333 fake: &mut FakeLanguageServer,
9334 path: &'static str,
9335 position: Point,
9336 completions: Vec<(Range<Point>, &'static str)>,
9337 ) {
9338 fake.handle_request::<lsp::request::Completion, _, _>(move |params, _| {
9339 let completions = completions.clone();
9340 async move {
9341 assert_eq!(
9342 params.text_document_position.text_document.uri,
9343 lsp::Url::from_file_path(path).unwrap()
9344 );
9345 assert_eq!(
9346 params.text_document_position.position,
9347 lsp::Position::new(position.row, position.column)
9348 );
9349 Ok(Some(lsp::CompletionResponse::Array(
9350 completions
9351 .iter()
9352 .map(|(range, new_text)| lsp::CompletionItem {
9353 label: new_text.to_string(),
9354 text_edit: Some(lsp::CompletionTextEdit::Edit(lsp::TextEdit {
9355 range: lsp::Range::new(
9356 lsp::Position::new(range.start.row, range.start.column),
9357 lsp::Position::new(range.start.row, range.start.column),
9358 ),
9359 new_text: new_text.to_string(),
9360 })),
9361 ..Default::default()
9362 })
9363 .collect(),
9364 )))
9365 }
9366 })
9367 .next()
9368 .await;
9369 }
9370
9371 async fn handle_resolve_completion_request(
9372 fake: &mut FakeLanguageServer,
9373 edit: Option<(Range<Point>, &'static str)>,
9374 ) {
9375 fake.handle_request::<lsp::request::ResolveCompletionItem, _, _>(move |_, _| {
9376 let edit = edit.clone();
9377 async move {
9378 Ok(lsp::CompletionItem {
9379 additional_text_edits: edit.map(|(range, new_text)| {
9380 vec![lsp::TextEdit::new(
9381 lsp::Range::new(
9382 lsp::Position::new(range.start.row, range.start.column),
9383 lsp::Position::new(range.end.row, range.end.column),
9384 ),
9385 new_text.to_string(),
9386 )]
9387 }),
9388 ..Default::default()
9389 })
9390 }
9391 })
9392 .next()
9393 .await;
9394 }
9395 }
9396
9397 #[gpui::test]
9398 async fn test_toggle_comment(cx: &mut gpui::TestAppContext) {
9399 cx.update(|cx| cx.set_global(Settings::test(cx)));
9400 let language = Arc::new(Language::new(
9401 LanguageConfig {
9402 line_comment: Some("// ".to_string()),
9403 ..Default::default()
9404 },
9405 Some(tree_sitter_rust::language()),
9406 ));
9407
9408 let text = "
9409 fn a() {
9410 //b();
9411 // c();
9412 // d();
9413 }
9414 "
9415 .unindent();
9416
9417 let buffer = cx.add_model(|cx| Buffer::new(0, text, cx).with_language(language, cx));
9418 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
9419 let (_, view) = cx.add_window(|cx| build_editor(buffer, cx));
9420
9421 view.update(cx, |editor, cx| {
9422 // If multiple selections intersect a line, the line is only
9423 // toggled once.
9424 editor.change_selections(None, cx, |s| {
9425 s.select_display_ranges([
9426 DisplayPoint::new(1, 3)..DisplayPoint::new(2, 3),
9427 DisplayPoint::new(3, 5)..DisplayPoint::new(3, 6),
9428 ])
9429 });
9430 editor.toggle_comments(&ToggleComments, cx);
9431 assert_eq!(
9432 editor.text(cx),
9433 "
9434 fn a() {
9435 b();
9436 c();
9437 d();
9438 }
9439 "
9440 .unindent()
9441 );
9442
9443 // The comment prefix is inserted at the same column for every line
9444 // in a selection.
9445 editor.change_selections(None, cx, |s| {
9446 s.select_display_ranges([DisplayPoint::new(1, 3)..DisplayPoint::new(3, 6)])
9447 });
9448 editor.toggle_comments(&ToggleComments, cx);
9449 assert_eq!(
9450 editor.text(cx),
9451 "
9452 fn a() {
9453 // b();
9454 // c();
9455 // d();
9456 }
9457 "
9458 .unindent()
9459 );
9460
9461 // If a selection ends at the beginning of a line, that line is not toggled.
9462 editor.change_selections(None, cx, |s| {
9463 s.select_display_ranges([DisplayPoint::new(2, 0)..DisplayPoint::new(3, 0)])
9464 });
9465 editor.toggle_comments(&ToggleComments, cx);
9466 assert_eq!(
9467 editor.text(cx),
9468 "
9469 fn a() {
9470 // b();
9471 c();
9472 // d();
9473 }
9474 "
9475 .unindent()
9476 );
9477 });
9478 }
9479
9480 #[gpui::test]
9481 fn test_editing_disjoint_excerpts(cx: &mut gpui::MutableAppContext) {
9482 cx.set_global(Settings::test(cx));
9483 let buffer = cx.add_model(|cx| Buffer::new(0, sample_text(3, 4, 'a'), cx));
9484 let multibuffer = cx.add_model(|cx| {
9485 let mut multibuffer = MultiBuffer::new(0);
9486 multibuffer.push_excerpts(
9487 buffer.clone(),
9488 [
9489 ExcerptRange {
9490 context: Point::new(0, 0)..Point::new(0, 4),
9491 primary: None,
9492 },
9493 ExcerptRange {
9494 context: Point::new(1, 0)..Point::new(1, 4),
9495 primary: None,
9496 },
9497 ],
9498 cx,
9499 );
9500 multibuffer
9501 });
9502
9503 assert_eq!(multibuffer.read(cx).read(cx).text(), "aaaa\nbbbb");
9504
9505 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(multibuffer, cx));
9506 view.update(cx, |view, cx| {
9507 assert_eq!(view.text(cx), "aaaa\nbbbb");
9508 view.change_selections(None, cx, |s| {
9509 s.select_ranges([
9510 Point::new(0, 0)..Point::new(0, 0),
9511 Point::new(1, 0)..Point::new(1, 0),
9512 ])
9513 });
9514
9515 view.handle_input(&Input("X".to_string()), cx);
9516 assert_eq!(view.text(cx), "Xaaaa\nXbbbb");
9517 assert_eq!(
9518 view.selections.ranges(cx),
9519 [
9520 Point::new(0, 1)..Point::new(0, 1),
9521 Point::new(1, 1)..Point::new(1, 1),
9522 ]
9523 )
9524 });
9525 }
9526
9527 #[gpui::test]
9528 fn test_editing_overlapping_excerpts(cx: &mut gpui::MutableAppContext) {
9529 cx.set_global(Settings::test(cx));
9530 let (initial_text, excerpt_ranges) = marked_text_ranges(indoc! {"
9531 [aaaa
9532 (bbbb]
9533 cccc)"});
9534 let excerpt_ranges = excerpt_ranges.into_iter().map(|context| ExcerptRange {
9535 context,
9536 primary: None,
9537 });
9538 let buffer = cx.add_model(|cx| Buffer::new(0, initial_text, cx));
9539 let multibuffer = cx.add_model(|cx| {
9540 let mut multibuffer = MultiBuffer::new(0);
9541 multibuffer.push_excerpts(buffer, excerpt_ranges, cx);
9542 multibuffer
9543 });
9544
9545 let (_, view) = cx.add_window(Default::default(), |cx| build_editor(multibuffer, cx));
9546 view.update(cx, |view, cx| {
9547 let (expected_text, selection_ranges) = marked_text_ranges(indoc! {"
9548 aaaa
9549 b|bbb
9550 b|bb|b
9551 cccc"});
9552 assert_eq!(view.text(cx), expected_text);
9553 view.change_selections(None, cx, |s| s.select_ranges(selection_ranges));
9554
9555 view.handle_input(&Input("X".to_string()), cx);
9556
9557 let (expected_text, expected_selections) = marked_text_ranges(indoc! {"
9558 aaaa
9559 bX|bbXb
9560 bX|bbX|b
9561 cccc"});
9562 assert_eq!(view.text(cx), expected_text);
9563 assert_eq!(view.selections.ranges(cx), expected_selections);
9564
9565 view.newline(&Newline, cx);
9566 let (expected_text, expected_selections) = marked_text_ranges(indoc! {"
9567 aaaa
9568 bX
9569 |bbX
9570 b
9571 bX
9572 |bbX
9573 |b
9574 cccc"});
9575 assert_eq!(view.text(cx), expected_text);
9576 assert_eq!(view.selections.ranges(cx), expected_selections);
9577 });
9578 }
9579
9580 #[gpui::test]
9581 fn test_refresh_selections(cx: &mut gpui::MutableAppContext) {
9582 cx.set_global(Settings::test(cx));
9583 let buffer = cx.add_model(|cx| Buffer::new(0, sample_text(3, 4, 'a'), cx));
9584 let mut excerpt1_id = None;
9585 let multibuffer = cx.add_model(|cx| {
9586 let mut multibuffer = MultiBuffer::new(0);
9587 excerpt1_id = multibuffer
9588 .push_excerpts(
9589 buffer.clone(),
9590 [
9591 ExcerptRange {
9592 context: Point::new(0, 0)..Point::new(1, 4),
9593 primary: None,
9594 },
9595 ExcerptRange {
9596 context: Point::new(1, 0)..Point::new(2, 4),
9597 primary: None,
9598 },
9599 ],
9600 cx,
9601 )
9602 .into_iter()
9603 .next();
9604 multibuffer
9605 });
9606 assert_eq!(
9607 multibuffer.read(cx).read(cx).text(),
9608 "aaaa\nbbbb\nbbbb\ncccc"
9609 );
9610 let (_, editor) = cx.add_window(Default::default(), |cx| {
9611 let mut editor = build_editor(multibuffer.clone(), cx);
9612 let snapshot = editor.snapshot(cx);
9613 editor.change_selections(None, cx, |s| {
9614 s.select_ranges([Point::new(1, 3)..Point::new(1, 3)])
9615 });
9616 editor.begin_selection(Point::new(2, 1).to_display_point(&snapshot), true, 1, cx);
9617 assert_eq!(
9618 editor.selections.ranges(cx),
9619 [
9620 Point::new(1, 3)..Point::new(1, 3),
9621 Point::new(2, 1)..Point::new(2, 1),
9622 ]
9623 );
9624 editor
9625 });
9626
9627 // Refreshing selections is a no-op when excerpts haven't changed.
9628 editor.update(cx, |editor, cx| {
9629 editor.change_selections(None, cx, |s| {
9630 s.refresh();
9631 });
9632 assert_eq!(
9633 editor.selections.ranges(cx),
9634 [
9635 Point::new(1, 3)..Point::new(1, 3),
9636 Point::new(2, 1)..Point::new(2, 1),
9637 ]
9638 );
9639 });
9640
9641 multibuffer.update(cx, |multibuffer, cx| {
9642 multibuffer.remove_excerpts([&excerpt1_id.unwrap()], cx);
9643 });
9644 editor.update(cx, |editor, cx| {
9645 // Removing an excerpt causes the first selection to become degenerate.
9646 assert_eq!(
9647 editor.selections.ranges(cx),
9648 [
9649 Point::new(0, 0)..Point::new(0, 0),
9650 Point::new(0, 1)..Point::new(0, 1)
9651 ]
9652 );
9653
9654 // Refreshing selections will relocate the first selection to the original buffer
9655 // location.
9656 editor.change_selections(None, cx, |s| {
9657 s.refresh();
9658 });
9659 assert_eq!(
9660 editor.selections.ranges(cx),
9661 [
9662 Point::new(0, 1)..Point::new(0, 1),
9663 Point::new(0, 3)..Point::new(0, 3)
9664 ]
9665 );
9666 assert!(editor.selections.pending_anchor().is_some());
9667 });
9668 }
9669
9670 #[gpui::test]
9671 fn test_refresh_selections_while_selecting_with_mouse(cx: &mut gpui::MutableAppContext) {
9672 cx.set_global(Settings::test(cx));
9673 let buffer = cx.add_model(|cx| Buffer::new(0, sample_text(3, 4, 'a'), cx));
9674 let mut excerpt1_id = None;
9675 let multibuffer = cx.add_model(|cx| {
9676 let mut multibuffer = MultiBuffer::new(0);
9677 excerpt1_id = multibuffer
9678 .push_excerpts(
9679 buffer.clone(),
9680 [
9681 ExcerptRange {
9682 context: Point::new(0, 0)..Point::new(1, 4),
9683 primary: None,
9684 },
9685 ExcerptRange {
9686 context: Point::new(1, 0)..Point::new(2, 4),
9687 primary: None,
9688 },
9689 ],
9690 cx,
9691 )
9692 .into_iter()
9693 .next();
9694 multibuffer
9695 });
9696 assert_eq!(
9697 multibuffer.read(cx).read(cx).text(),
9698 "aaaa\nbbbb\nbbbb\ncccc"
9699 );
9700 let (_, editor) = cx.add_window(Default::default(), |cx| {
9701 let mut editor = build_editor(multibuffer.clone(), cx);
9702 let snapshot = editor.snapshot(cx);
9703 editor.begin_selection(Point::new(1, 3).to_display_point(&snapshot), false, 1, cx);
9704 assert_eq!(
9705 editor.selections.ranges(cx),
9706 [Point::new(1, 3)..Point::new(1, 3)]
9707 );
9708 editor
9709 });
9710
9711 multibuffer.update(cx, |multibuffer, cx| {
9712 multibuffer.remove_excerpts([&excerpt1_id.unwrap()], cx);
9713 });
9714 editor.update(cx, |editor, cx| {
9715 assert_eq!(
9716 editor.selections.ranges(cx),
9717 [Point::new(0, 0)..Point::new(0, 0)]
9718 );
9719
9720 // Ensure we don't panic when selections are refreshed and that the pending selection is finalized.
9721 editor.change_selections(None, cx, |s| {
9722 s.refresh();
9723 });
9724 assert_eq!(
9725 editor.selections.ranges(cx),
9726 [Point::new(0, 3)..Point::new(0, 3)]
9727 );
9728 assert!(editor.selections.pending_anchor().is_some());
9729 });
9730 }
9731
9732 #[gpui::test]
9733 async fn test_extra_newline_insertion(cx: &mut gpui::TestAppContext) {
9734 cx.update(|cx| cx.set_global(Settings::test(cx)));
9735 let language = Arc::new(Language::new(
9736 LanguageConfig {
9737 brackets: vec![
9738 BracketPair {
9739 start: "{".to_string(),
9740 end: "}".to_string(),
9741 close: true,
9742 newline: true,
9743 },
9744 BracketPair {
9745 start: "/* ".to_string(),
9746 end: " */".to_string(),
9747 close: true,
9748 newline: true,
9749 },
9750 ],
9751 ..Default::default()
9752 },
9753 Some(tree_sitter_rust::language()),
9754 ));
9755
9756 let text = concat!(
9757 "{ }\n", // Suppress rustfmt
9758 " x\n", //
9759 " /* */\n", //
9760 "x\n", //
9761 "{{} }\n", //
9762 );
9763
9764 let buffer = cx.add_model(|cx| Buffer::new(0, text, cx).with_language(language, cx));
9765 let buffer = cx.add_model(|cx| MultiBuffer::singleton(buffer, cx));
9766 let (_, view) = cx.add_window(|cx| build_editor(buffer, cx));
9767 view.condition(&cx, |view, cx| !view.buffer.read(cx).is_parsing(cx))
9768 .await;
9769
9770 view.update(cx, |view, cx| {
9771 view.change_selections(None, cx, |s| {
9772 s.select_display_ranges([
9773 DisplayPoint::new(0, 2)..DisplayPoint::new(0, 3),
9774 DisplayPoint::new(2, 5)..DisplayPoint::new(2, 5),
9775 DisplayPoint::new(4, 4)..DisplayPoint::new(4, 4),
9776 ])
9777 });
9778 view.newline(&Newline, cx);
9779
9780 assert_eq!(
9781 view.buffer().read(cx).read(cx).text(),
9782 concat!(
9783 "{ \n", // Suppress rustfmt
9784 "\n", //
9785 "}\n", //
9786 " x\n", //
9787 " /* \n", //
9788 " \n", //
9789 " */\n", //
9790 "x\n", //
9791 "{{} \n", //
9792 "}\n", //
9793 )
9794 );
9795 });
9796 }
9797
9798 #[gpui::test]
9799 fn test_highlighted_ranges(cx: &mut gpui::MutableAppContext) {
9800 let buffer = MultiBuffer::build_simple(&sample_text(16, 8, 'a'), cx);
9801
9802 cx.set_global(Settings::test(cx));
9803 let (_, editor) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
9804
9805 editor.update(cx, |editor, cx| {
9806 struct Type1;
9807 struct Type2;
9808
9809 let buffer = buffer.read(cx).snapshot(cx);
9810
9811 let anchor_range = |range: Range<Point>| {
9812 buffer.anchor_after(range.start)..buffer.anchor_after(range.end)
9813 };
9814
9815 editor.highlight_background::<Type1>(
9816 vec![
9817 anchor_range(Point::new(2, 1)..Point::new(2, 3)),
9818 anchor_range(Point::new(4, 2)..Point::new(4, 4)),
9819 anchor_range(Point::new(6, 3)..Point::new(6, 5)),
9820 anchor_range(Point::new(8, 4)..Point::new(8, 6)),
9821 ],
9822 |_| Color::red(),
9823 cx,
9824 );
9825 editor.highlight_background::<Type2>(
9826 vec![
9827 anchor_range(Point::new(3, 2)..Point::new(3, 5)),
9828 anchor_range(Point::new(5, 3)..Point::new(5, 6)),
9829 anchor_range(Point::new(7, 4)..Point::new(7, 7)),
9830 anchor_range(Point::new(9, 5)..Point::new(9, 8)),
9831 ],
9832 |_| Color::green(),
9833 cx,
9834 );
9835
9836 let snapshot = editor.snapshot(cx);
9837 let mut highlighted_ranges = editor.background_highlights_in_range(
9838 anchor_range(Point::new(3, 4)..Point::new(7, 4)),
9839 &snapshot,
9840 cx.global::<Settings>().theme.as_ref(),
9841 );
9842 // Enforce a consistent ordering based on color without relying on the ordering of the
9843 // highlight's `TypeId` which is non-deterministic.
9844 highlighted_ranges.sort_unstable_by_key(|(_, color)| *color);
9845 assert_eq!(
9846 highlighted_ranges,
9847 &[
9848 (
9849 DisplayPoint::new(3, 2)..DisplayPoint::new(3, 5),
9850 Color::green(),
9851 ),
9852 (
9853 DisplayPoint::new(5, 3)..DisplayPoint::new(5, 6),
9854 Color::green(),
9855 ),
9856 (
9857 DisplayPoint::new(4, 2)..DisplayPoint::new(4, 4),
9858 Color::red(),
9859 ),
9860 (
9861 DisplayPoint::new(6, 3)..DisplayPoint::new(6, 5),
9862 Color::red(),
9863 ),
9864 ]
9865 );
9866 assert_eq!(
9867 editor.background_highlights_in_range(
9868 anchor_range(Point::new(5, 6)..Point::new(6, 4)),
9869 &snapshot,
9870 cx.global::<Settings>().theme.as_ref(),
9871 ),
9872 &[(
9873 DisplayPoint::new(6, 3)..DisplayPoint::new(6, 5),
9874 Color::red(),
9875 )]
9876 );
9877 });
9878 }
9879
9880 #[gpui::test]
9881 fn test_following(cx: &mut gpui::MutableAppContext) {
9882 let buffer = MultiBuffer::build_simple(&sample_text(16, 8, 'a'), cx);
9883
9884 cx.set_global(Settings::test(cx));
9885
9886 let (_, leader) = cx.add_window(Default::default(), |cx| build_editor(buffer.clone(), cx));
9887 let (_, follower) = cx.add_window(
9888 WindowOptions {
9889 bounds: WindowBounds::Fixed(RectF::from_points(vec2f(0., 0.), vec2f(10., 80.))),
9890 ..Default::default()
9891 },
9892 |cx| build_editor(buffer.clone(), cx),
9893 );
9894
9895 let pending_update = Rc::new(RefCell::new(None));
9896 follower.update(cx, {
9897 let update = pending_update.clone();
9898 |_, cx| {
9899 cx.subscribe(&leader, move |_, leader, event, cx| {
9900 leader
9901 .read(cx)
9902 .add_event_to_update_proto(event, &mut *update.borrow_mut(), cx);
9903 })
9904 .detach();
9905 }
9906 });
9907
9908 // Update the selections only
9909 leader.update(cx, |leader, cx| {
9910 leader.change_selections(None, cx, |s| s.select_ranges([1..1]));
9911 });
9912 follower.update(cx, |follower, cx| {
9913 follower
9914 .apply_update_proto(pending_update.borrow_mut().take().unwrap(), cx)
9915 .unwrap();
9916 });
9917 assert_eq!(follower.read(cx).selections.ranges(cx), vec![1..1]);
9918
9919 // Update the scroll position only
9920 leader.update(cx, |leader, cx| {
9921 leader.set_scroll_position(vec2f(1.5, 3.5), cx);
9922 });
9923 follower.update(cx, |follower, cx| {
9924 follower
9925 .apply_update_proto(pending_update.borrow_mut().take().unwrap(), cx)
9926 .unwrap();
9927 });
9928 assert_eq!(
9929 follower.update(cx, |follower, cx| follower.scroll_position(cx)),
9930 vec2f(1.5, 3.5)
9931 );
9932
9933 // Update the selections and scroll position
9934 leader.update(cx, |leader, cx| {
9935 leader.change_selections(None, cx, |s| s.select_ranges([0..0]));
9936 leader.request_autoscroll(Autoscroll::Newest, cx);
9937 leader.set_scroll_position(vec2f(1.5, 3.5), cx);
9938 });
9939 follower.update(cx, |follower, cx| {
9940 let initial_scroll_position = follower.scroll_position(cx);
9941 follower
9942 .apply_update_proto(pending_update.borrow_mut().take().unwrap(), cx)
9943 .unwrap();
9944 assert_eq!(follower.scroll_position(cx), initial_scroll_position);
9945 assert!(follower.autoscroll_request.is_some());
9946 });
9947 assert_eq!(follower.read(cx).selections.ranges(cx), vec![0..0]);
9948
9949 // Creating a pending selection that precedes another selection
9950 leader.update(cx, |leader, cx| {
9951 leader.change_selections(None, cx, |s| s.select_ranges([1..1]));
9952 leader.begin_selection(DisplayPoint::new(0, 0), true, 1, cx);
9953 });
9954 follower.update(cx, |follower, cx| {
9955 follower
9956 .apply_update_proto(pending_update.borrow_mut().take().unwrap(), cx)
9957 .unwrap();
9958 });
9959 assert_eq!(follower.read(cx).selections.ranges(cx), vec![0..0, 1..1]);
9960
9961 // Extend the pending selection so that it surrounds another selection
9962 leader.update(cx, |leader, cx| {
9963 leader.extend_selection(DisplayPoint::new(0, 2), 1, cx);
9964 });
9965 follower.update(cx, |follower, cx| {
9966 follower
9967 .apply_update_proto(pending_update.borrow_mut().take().unwrap(), cx)
9968 .unwrap();
9969 });
9970 assert_eq!(follower.read(cx).selections.ranges(cx), vec![0..2]);
9971 }
9972
9973 #[test]
9974 fn test_combine_syntax_and_fuzzy_match_highlights() {
9975 let string = "abcdefghijklmnop";
9976 let syntax_ranges = [
9977 (
9978 0..3,
9979 HighlightStyle {
9980 color: Some(Color::red()),
9981 ..Default::default()
9982 },
9983 ),
9984 (
9985 4..8,
9986 HighlightStyle {
9987 color: Some(Color::green()),
9988 ..Default::default()
9989 },
9990 ),
9991 ];
9992 let match_indices = [4, 6, 7, 8];
9993 assert_eq!(
9994 combine_syntax_and_fuzzy_match_highlights(
9995 &string,
9996 Default::default(),
9997 syntax_ranges.into_iter(),
9998 &match_indices,
9999 ),
10000 &[
10001 (
10002 0..3,
10003 HighlightStyle {
10004 color: Some(Color::red()),
10005 ..Default::default()
10006 },
10007 ),
10008 (
10009 4..5,
10010 HighlightStyle {
10011 color: Some(Color::green()),
10012 weight: Some(fonts::Weight::BOLD),
10013 ..Default::default()
10014 },
10015 ),
10016 (
10017 5..6,
10018 HighlightStyle {
10019 color: Some(Color::green()),
10020 ..Default::default()
10021 },
10022 ),
10023 (
10024 6..8,
10025 HighlightStyle {
10026 color: Some(Color::green()),
10027 weight: Some(fonts::Weight::BOLD),
10028 ..Default::default()
10029 },
10030 ),
10031 (
10032 8..9,
10033 HighlightStyle {
10034 weight: Some(fonts::Weight::BOLD),
10035 ..Default::default()
10036 },
10037 ),
10038 ]
10039 );
10040 }
10041
10042 fn empty_range(row: usize, column: usize) -> Range<DisplayPoint> {
10043 let point = DisplayPoint::new(row as u32, column as u32);
10044 point..point
10045 }
10046
10047 fn assert_selection_ranges(
10048 marked_text: &str,
10049 selection_marker_pairs: Vec<(char, char)>,
10050 view: &mut Editor,
10051 cx: &mut ViewContext<Editor>,
10052 ) {
10053 let snapshot = view.snapshot(cx).display_snapshot;
10054 let mut marker_chars = Vec::new();
10055 for (start, end) in selection_marker_pairs.iter() {
10056 marker_chars.push(*start);
10057 marker_chars.push(*end);
10058 }
10059 let (_, markers) = marked_text_by(marked_text, marker_chars);
10060 let asserted_ranges: Vec<Range<DisplayPoint>> = selection_marker_pairs
10061 .iter()
10062 .map(|(start, end)| {
10063 let start = markers.get(start).unwrap()[0].to_display_point(&snapshot);
10064 let end = markers.get(end).unwrap()[0].to_display_point(&snapshot);
10065 start..end
10066 })
10067 .collect();
10068 assert_eq!(
10069 view.selections.display_ranges(cx),
10070 &asserted_ranges[..],
10071 "Assert selections are {}",
10072 marked_text
10073 );
10074 }
10075}
10076
10077trait RangeExt<T> {
10078 fn sorted(&self) -> Range<T>;
10079 fn to_inclusive(&self) -> RangeInclusive<T>;
10080}
10081
10082impl<T: Ord + Clone> RangeExt<T> for Range<T> {
10083 fn sorted(&self) -> Self {
10084 cmp::min(&self.start, &self.end).clone()..cmp::max(&self.start, &self.end).clone()
10085 }
10086
10087 fn to_inclusive(&self) -> RangeInclusive<T> {
10088 self.start.clone()..=self.end.clone()
10089 }
10090}