1#![allow(rustdoc::private_intra_doc_links)]
2//! This is the place where everything editor-related is stored (data-wise) and displayed (ui-wise).
3//! The main point of interest in this crate is [`Editor`] type, which is used in every other Zed part as a user input element.
4//! It comes in different flavors: single line, multiline and a fixed height one.
5//!
6//! Editor contains of multiple large submodules:
7//! * [`element`] — the place where all rendering happens
8//! * [`display_map`] - chunks up text in the editor into the logical blocks, establishes coordinates and mapping between each of them.
9//! Contains all metadata related to text transformations (folds, fake inlay text insertions, soft wraps, tab markup, etc.).
10//! * [`inlay_hint_cache`] - is a storage of inlay hints out of LSP requests, responsible for querying LSP and updating `display_map`'s state accordingly.
11//!
12//! All other submodules and structs are mostly concerned with holding editor data about the way it displays current buffer region(s).
13//!
14//! If you're looking to improve Vim mode, you should check out Vim crate that wraps Editor and overrides its behaviour.
15pub mod actions;
16mod blame_entry_tooltip;
17mod blink_manager;
18pub mod display_map;
19mod editor_settings;
20mod element;
21mod hunk_diff;
22mod inlay_hint_cache;
23
24mod debounced_delay;
25mod git;
26mod highlight_matching_bracket;
27mod hover_links;
28mod hover_popover;
29mod inline_completion_provider;
30pub mod items;
31mod mouse_context_menu;
32pub mod movement;
33mod persistence;
34mod rust_analyzer_ext;
35pub mod scroll;
36mod selections_collection;
37pub mod tasks;
38
39#[cfg(test)]
40mod editor_tests;
41#[cfg(any(test, feature = "test-support"))]
42pub mod test;
43use ::git::diff::{DiffHunk, DiffHunkStatus};
44use ::git::{parse_git_remote_url, BuildPermalinkParams, GitHostingProviderRegistry};
45pub(crate) use actions::*;
46use aho_corasick::AhoCorasick;
47use anyhow::{anyhow, Context as _, Result};
48use blink_manager::BlinkManager;
49use client::{Collaborator, ParticipantIndex};
50use clock::ReplicaId;
51use collections::{hash_map, BTreeMap, Bound, HashMap, HashSet, VecDeque};
52use convert_case::{Case, Casing};
53use debounced_delay::DebouncedDelay;
54pub use display_map::DisplayPoint;
55use display_map::*;
56pub use editor_settings::EditorSettings;
57use element::LineWithInvisibles;
58pub use element::{
59 CursorLayout, EditorElement, HighlightedRange, HighlightedRangeLine, PointForPosition,
60};
61use futures::FutureExt;
62use fuzzy::{StringMatch, StringMatchCandidate};
63use git::blame::GitBlame;
64use git::diff_hunk_to_display;
65use gpui::{
66 div, impl_actions, point, prelude::*, px, relative, size, uniform_list, Action, AnyElement,
67 AppContext, AsyncWindowContext, AvailableSpace, BackgroundExecutor, Bounds, ClipboardItem,
68 Context, DispatchPhase, ElementId, EventEmitter, FocusHandle, FocusableView, FontId, FontStyle,
69 FontWeight, HighlightStyle, Hsla, InteractiveText, KeyContext, Model, MouseButton, PaintQuad,
70 ParentElement, Pixels, Render, SharedString, Size, StrikethroughStyle, Styled, StyledText,
71 Subscription, Task, TextStyle, UnderlineStyle, UniformListScrollHandle, View, ViewContext,
72 ViewInputHandler, VisualContext, WeakView, WhiteSpace, WindowContext,
73};
74use highlight_matching_bracket::refresh_matching_bracket_highlights;
75use hover_popover::{hide_hover, HoverState};
76use hunk_diff::ExpandedHunks;
77pub(crate) use hunk_diff::HunkToExpand;
78use inlay_hint_cache::{InlayHintCache, InlaySplice, InvalidationStrategy};
79pub use inline_completion_provider::*;
80pub use items::MAX_TAB_TITLE_LEN;
81use itertools::Itertools;
82use language::Runnable;
83use language::{
84 char_kind,
85 language_settings::{self, all_language_settings, InlayHintSettings},
86 markdown, point_from_lsp, AutoindentMode, BracketPair, Buffer, Capability, CharKind, CodeLabel,
87 CursorShape, Diagnostic, Documentation, IndentKind, IndentSize, Language, OffsetRangeExt,
88 Point, Selection, SelectionGoal, TransactionId,
89};
90use task::{ResolvedTask, TaskTemplate};
91
92use hover_links::{HoverLink, HoveredLinkState, InlayHighlight};
93use lsp::{DiagnosticSeverity, LanguageServerId};
94use mouse_context_menu::MouseContextMenu;
95use movement::TextLayoutDetails;
96use multi_buffer::ToOffsetUtf16;
97pub use multi_buffer::{
98 Anchor, AnchorRangeExt, ExcerptId, ExcerptRange, MultiBuffer, MultiBufferSnapshot, ToOffset,
99 ToPoint,
100};
101use ordered_float::OrderedFloat;
102use parking_lot::{Mutex, RwLock};
103use project::project_settings::{GitGutterSetting, ProjectSettings};
104use project::{
105 CodeAction, Completion, FormatTrigger, Item, Location, Project, ProjectPath,
106 ProjectTransaction, TaskSourceKind, WorktreeId,
107};
108use rand::prelude::*;
109use rpc::{proto::*, ErrorExt};
110use scroll::{Autoscroll, OngoingScroll, ScrollAnchor, ScrollManager, ScrollbarAutoHide};
111use selections_collection::{resolve_multiple, MutableSelectionsCollection, SelectionsCollection};
112use serde::{Deserialize, Serialize};
113use settings::{Settings, SettingsStore};
114use smallvec::SmallVec;
115use snippet::Snippet;
116use std::ops::Not as _;
117use std::{
118 any::TypeId,
119 borrow::Cow,
120 cmp::{self, Ordering, Reverse},
121 mem,
122 num::NonZeroU32,
123 ops::{ControlFlow, Deref, DerefMut, Range, RangeInclusive},
124 path::Path,
125 sync::Arc,
126 time::{Duration, Instant},
127};
128pub use sum_tree::Bias;
129use sum_tree::TreeMap;
130use text::{BufferId, OffsetUtf16, Rope};
131use theme::{
132 observe_buffer_font_size_adjustment, ActiveTheme, PlayerColor, StatusColors, SyntaxTheme,
133 ThemeColors, ThemeSettings,
134};
135use ui::{
136 h_flex, prelude::*, ButtonSize, ButtonStyle, IconButton, IconName, IconSize, ListItem, Popover,
137 Tooltip,
138};
139use util::{defer, maybe, post_inc, RangeExt, ResultExt, TryFutureExt};
140use workspace::item::{ItemHandle, PreviewTabsSettings};
141use workspace::notifications::{DetachAndPromptErr, NotificationId};
142use workspace::{
143 searchable::SearchEvent, ItemNavHistory, SplitDirection, ViewId, Workspace, WorkspaceId,
144};
145use workspace::{OpenInTerminal, OpenTerminal, Toast};
146
147use crate::hover_links::find_url;
148
149pub const DEFAULT_MULTIBUFFER_CONTEXT: u32 = 2;
150const CURSOR_BLINK_INTERVAL: Duration = Duration::from_millis(500);
151const MAX_LINE_LEN: usize = 1024;
152const MIN_NAVIGATION_HISTORY_ROW_DELTA: i64 = 10;
153const MAX_SELECTION_HISTORY_LEN: usize = 1024;
154pub(crate) const CURSORS_VISIBLE_FOR: Duration = Duration::from_millis(2000);
155#[doc(hidden)]
156pub const CODE_ACTIONS_DEBOUNCE_TIMEOUT: Duration = Duration::from_millis(250);
157#[doc(hidden)]
158pub const DOCUMENT_HIGHLIGHTS_DEBOUNCE_TIMEOUT: Duration = Duration::from_millis(75);
159
160pub(crate) const FORMAT_TIMEOUT: Duration = Duration::from_secs(2);
161
162pub fn render_parsed_markdown(
163 element_id: impl Into<ElementId>,
164 parsed: &language::ParsedMarkdown,
165 editor_style: &EditorStyle,
166 workspace: Option<WeakView<Workspace>>,
167 cx: &mut WindowContext,
168) -> InteractiveText {
169 let code_span_background_color = cx
170 .theme()
171 .colors()
172 .editor_document_highlight_read_background;
173
174 let highlights = gpui::combine_highlights(
175 parsed.highlights.iter().filter_map(|(range, highlight)| {
176 let highlight = highlight.to_highlight_style(&editor_style.syntax)?;
177 Some((range.clone(), highlight))
178 }),
179 parsed
180 .regions
181 .iter()
182 .zip(&parsed.region_ranges)
183 .filter_map(|(region, range)| {
184 if region.code {
185 Some((
186 range.clone(),
187 HighlightStyle {
188 background_color: Some(code_span_background_color),
189 ..Default::default()
190 },
191 ))
192 } else {
193 None
194 }
195 }),
196 );
197
198 let mut links = Vec::new();
199 let mut link_ranges = Vec::new();
200 for (range, region) in parsed.region_ranges.iter().zip(&parsed.regions) {
201 if let Some(link) = region.link.clone() {
202 links.push(link);
203 link_ranges.push(range.clone());
204 }
205 }
206
207 InteractiveText::new(
208 element_id,
209 StyledText::new(parsed.text.clone()).with_highlights(&editor_style.text, highlights),
210 )
211 .on_click(link_ranges, move |clicked_range_ix, cx| {
212 match &links[clicked_range_ix] {
213 markdown::Link::Web { url } => cx.open_url(url),
214 markdown::Link::Path { path } => {
215 if let Some(workspace) = &workspace {
216 _ = workspace.update(cx, |workspace, cx| {
217 workspace.open_abs_path(path.clone(), false, cx).detach();
218 });
219 }
220 }
221 }
222 })
223}
224
225#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
226pub(crate) enum InlayId {
227 Suggestion(usize),
228 Hint(usize),
229}
230
231impl InlayId {
232 fn id(&self) -> usize {
233 match self {
234 Self::Suggestion(id) => *id,
235 Self::Hint(id) => *id,
236 }
237 }
238}
239
240enum DiffRowHighlight {}
241enum DocumentHighlightRead {}
242enum DocumentHighlightWrite {}
243enum InputComposition {}
244
245#[derive(Copy, Clone, PartialEq, Eq)]
246pub enum Direction {
247 Prev,
248 Next,
249}
250
251pub fn init_settings(cx: &mut AppContext) {
252 EditorSettings::register(cx);
253}
254
255pub fn init(cx: &mut AppContext) {
256 init_settings(cx);
257
258 workspace::register_project_item::<Editor>(cx);
259 workspace::register_followable_item::<Editor>(cx);
260 workspace::register_deserializable_item::<Editor>(cx);
261 cx.observe_new_views(
262 |workspace: &mut Workspace, _cx: &mut ViewContext<Workspace>| {
263 workspace.register_action(Editor::new_file);
264 workspace.register_action(Editor::new_file_in_direction);
265 },
266 )
267 .detach();
268
269 cx.on_action(move |_: &workspace::NewFile, cx| {
270 let app_state = workspace::AppState::global(cx);
271 if let Some(app_state) = app_state.upgrade() {
272 workspace::open_new(app_state, cx, |workspace, cx| {
273 Editor::new_file(workspace, &Default::default(), cx)
274 })
275 .detach();
276 }
277 });
278 cx.on_action(move |_: &workspace::NewWindow, cx| {
279 let app_state = workspace::AppState::global(cx);
280 if let Some(app_state) = app_state.upgrade() {
281 workspace::open_new(app_state, cx, |workspace, cx| {
282 Editor::new_file(workspace, &Default::default(), cx)
283 })
284 .detach();
285 }
286 });
287}
288
289pub struct SearchWithinRange;
290
291trait InvalidationRegion {
292 fn ranges(&self) -> &[Range<Anchor>];
293}
294
295#[derive(Clone, Debug, PartialEq)]
296pub enum SelectPhase {
297 Begin {
298 position: DisplayPoint,
299 add: bool,
300 click_count: usize,
301 },
302 BeginColumnar {
303 position: DisplayPoint,
304 goal_column: u32,
305 },
306 Extend {
307 position: DisplayPoint,
308 click_count: usize,
309 },
310 Update {
311 position: DisplayPoint,
312 goal_column: u32,
313 scroll_delta: gpui::Point<f32>,
314 },
315 End,
316}
317
318#[derive(Clone, Debug)]
319pub enum SelectMode {
320 Character,
321 Word(Range<Anchor>),
322 Line(Range<Anchor>),
323 All,
324}
325
326#[derive(Copy, Clone, PartialEq, Eq, Debug)]
327pub enum EditorMode {
328 SingleLine,
329 AutoHeight { max_lines: usize },
330 Full,
331}
332
333#[derive(Clone, Debug)]
334pub enum SoftWrap {
335 None,
336 PreferLine,
337 EditorWidth,
338 Column(u32),
339}
340
341#[derive(Clone)]
342pub struct EditorStyle {
343 pub background: Hsla,
344 pub local_player: PlayerColor,
345 pub text: TextStyle,
346 pub scrollbar_width: Pixels,
347 pub syntax: Arc<SyntaxTheme>,
348 pub status: StatusColors,
349 pub inlay_hints_style: HighlightStyle,
350 pub suggestions_style: HighlightStyle,
351}
352
353impl Default for EditorStyle {
354 fn default() -> Self {
355 Self {
356 background: Hsla::default(),
357 local_player: PlayerColor::default(),
358 text: TextStyle::default(),
359 scrollbar_width: Pixels::default(),
360 syntax: Default::default(),
361 // HACK: Status colors don't have a real default.
362 // We should look into removing the status colors from the editor
363 // style and retrieve them directly from the theme.
364 status: StatusColors::dark(),
365 inlay_hints_style: HighlightStyle::default(),
366 suggestions_style: HighlightStyle::default(),
367 }
368 }
369}
370
371type CompletionId = usize;
372
373// type GetFieldEditorTheme = dyn Fn(&theme::Theme) -> theme::FieldEditor;
374// type OverrideTextStyle = dyn Fn(&EditorStyle) -> Option<HighlightStyle>;
375
376type BackgroundHighlight = (fn(&ThemeColors) -> Hsla, Arc<[Range<Anchor>]>);
377
378struct ScrollbarMarkerState {
379 scrollbar_size: Size<Pixels>,
380 dirty: bool,
381 markers: Arc<[PaintQuad]>,
382 pending_refresh: Option<Task<Result<()>>>,
383}
384
385impl ScrollbarMarkerState {
386 fn should_refresh(&self, scrollbar_size: Size<Pixels>) -> bool {
387 self.pending_refresh.is_none() && (self.scrollbar_size != scrollbar_size || self.dirty)
388 }
389}
390
391impl Default for ScrollbarMarkerState {
392 fn default() -> Self {
393 Self {
394 scrollbar_size: Size::default(),
395 dirty: false,
396 markers: Arc::from([]),
397 pending_refresh: None,
398 }
399 }
400}
401
402#[derive(Clone, Debug)]
403struct RunnableTasks {
404 templates: Vec<(TaskSourceKind, TaskTemplate)>,
405 // We need the column at which the task context evaluation should take place.
406 column: u32,
407}
408
409#[derive(Clone)]
410struct ResolvedTasks {
411 templates: SmallVec<[(TaskSourceKind, ResolvedTask); 1]>,
412 position: text::Point,
413}
414
415/// Zed's primary text input `View`, allowing users to edit a [`MultiBuffer`]
416///
417/// See the [module level documentation](self) for more information.
418pub struct Editor {
419 focus_handle: FocusHandle,
420 /// The text buffer being edited
421 buffer: Model<MultiBuffer>,
422 /// Map of how text in the buffer should be displayed.
423 /// Handles soft wraps, folds, fake inlay text insertions, etc.
424 pub display_map: Model<DisplayMap>,
425 pub selections: SelectionsCollection,
426 pub scroll_manager: ScrollManager,
427 columnar_selection_tail: Option<Anchor>,
428 add_selections_state: Option<AddSelectionsState>,
429 select_next_state: Option<SelectNextState>,
430 select_prev_state: Option<SelectNextState>,
431 selection_history: SelectionHistory,
432 autoclose_regions: Vec<AutocloseRegion>,
433 snippet_stack: InvalidationStack<SnippetState>,
434 select_larger_syntax_node_stack: Vec<Box<[Selection<usize>]>>,
435 ime_transaction: Option<TransactionId>,
436 active_diagnostics: Option<ActiveDiagnosticGroup>,
437 soft_wrap_mode_override: Option<language_settings::SoftWrap>,
438 project: Option<Model<Project>>,
439 completion_provider: Option<Box<dyn CompletionProvider>>,
440 collaboration_hub: Option<Box<dyn CollaborationHub>>,
441 blink_manager: Model<BlinkManager>,
442 show_cursor_names: bool,
443 hovered_cursors: HashMap<HoveredCursor, Task<()>>,
444 pub show_local_selections: bool,
445 mode: EditorMode,
446 show_breadcrumbs: bool,
447 show_gutter: bool,
448 show_wrap_guides: Option<bool>,
449 placeholder_text: Option<Arc<str>>,
450 highlight_order: usize,
451 highlighted_rows: HashMap<TypeId, Vec<(usize, Range<Anchor>, Hsla)>>,
452 background_highlights: TreeMap<TypeId, BackgroundHighlight>,
453 scrollbar_marker_state: ScrollbarMarkerState,
454 nav_history: Option<ItemNavHistory>,
455 context_menu: RwLock<Option<ContextMenu>>,
456 mouse_context_menu: Option<MouseContextMenu>,
457 completion_tasks: Vec<(CompletionId, Task<Option<()>>)>,
458 find_all_references_task_sources: Vec<Anchor>,
459 next_completion_id: CompletionId,
460 completion_documentation_pre_resolve_debounce: DebouncedDelay,
461 available_code_actions: Option<(Model<Buffer>, Arc<[CodeAction]>)>,
462 code_actions_task: Option<Task<()>>,
463 document_highlights_task: Option<Task<()>>,
464 pending_rename: Option<RenameState>,
465 searchable: bool,
466 cursor_shape: CursorShape,
467 collapse_matches: bool,
468 autoindent_mode: Option<AutoindentMode>,
469 workspace: Option<(WeakView<Workspace>, WorkspaceId)>,
470 keymap_context_layers: BTreeMap<TypeId, KeyContext>,
471 input_enabled: bool,
472 use_modal_editing: bool,
473 read_only: bool,
474 leader_peer_id: Option<PeerId>,
475 remote_id: Option<ViewId>,
476 hover_state: HoverState,
477 gutter_hovered: bool,
478 hovered_link_state: Option<HoveredLinkState>,
479 inline_completion_provider: Option<RegisteredInlineCompletionProvider>,
480 active_inline_completion: Option<Inlay>,
481 show_inline_completions: bool,
482 inlay_hint_cache: InlayHintCache,
483 expanded_hunks: ExpandedHunks,
484 next_inlay_id: usize,
485 _subscriptions: Vec<Subscription>,
486 pixel_position_of_newest_cursor: Option<gpui::Point<Pixels>>,
487 gutter_dimensions: GutterDimensions,
488 pub vim_replace_map: HashMap<Range<usize>, String>,
489 style: Option<EditorStyle>,
490 editor_actions: Vec<Box<dyn Fn(&mut ViewContext<Self>)>>,
491 use_autoclose: bool,
492 auto_replace_emoji_shortcode: bool,
493 show_git_blame_gutter: bool,
494 show_git_blame_inline: bool,
495 show_git_blame_inline_delay_task: Option<Task<()>>,
496 git_blame_inline_enabled: bool,
497 blame: Option<Model<GitBlame>>,
498 blame_subscription: Option<Subscription>,
499 custom_context_menu: Option<
500 Box<
501 dyn 'static
502 + Fn(&mut Self, DisplayPoint, &mut ViewContext<Self>) -> Option<View<ui::ContextMenu>>,
503 >,
504 >,
505 last_bounds: Option<Bounds<Pixels>>,
506 expect_bounds_change: Option<Bounds<Pixels>>,
507 tasks: HashMap<u32, RunnableTasks>,
508 tasks_update_task: Option<Task<()>>,
509}
510
511#[derive(Clone)]
512pub struct EditorSnapshot {
513 pub mode: EditorMode,
514 show_gutter: bool,
515 render_git_blame_gutter: bool,
516 pub display_snapshot: DisplaySnapshot,
517 pub placeholder_text: Option<Arc<str>>,
518 is_focused: bool,
519 scroll_anchor: ScrollAnchor,
520 ongoing_scroll: OngoingScroll,
521}
522
523const GIT_BLAME_GUTTER_WIDTH_CHARS: f32 = 53.;
524
525#[derive(Debug, Clone, Copy)]
526pub struct GutterDimensions {
527 pub left_padding: Pixels,
528 pub right_padding: Pixels,
529 pub width: Pixels,
530 pub margin: Pixels,
531 pub git_blame_entries_width: Option<Pixels>,
532}
533
534impl Default for GutterDimensions {
535 fn default() -> Self {
536 Self {
537 left_padding: Pixels::ZERO,
538 right_padding: Pixels::ZERO,
539 width: Pixels::ZERO,
540 margin: Pixels::ZERO,
541 git_blame_entries_width: None,
542 }
543 }
544}
545
546#[derive(Debug)]
547pub struct RemoteSelection {
548 pub replica_id: ReplicaId,
549 pub selection: Selection<Anchor>,
550 pub cursor_shape: CursorShape,
551 pub peer_id: PeerId,
552 pub line_mode: bool,
553 pub participant_index: Option<ParticipantIndex>,
554 pub user_name: Option<SharedString>,
555}
556
557#[derive(Clone, Debug)]
558struct SelectionHistoryEntry {
559 selections: Arc<[Selection<Anchor>]>,
560 select_next_state: Option<SelectNextState>,
561 select_prev_state: Option<SelectNextState>,
562 add_selections_state: Option<AddSelectionsState>,
563}
564
565enum SelectionHistoryMode {
566 Normal,
567 Undoing,
568 Redoing,
569}
570
571#[derive(Clone, PartialEq, Eq, Hash)]
572struct HoveredCursor {
573 replica_id: u16,
574 selection_id: usize,
575}
576
577impl Default for SelectionHistoryMode {
578 fn default() -> Self {
579 Self::Normal
580 }
581}
582
583#[derive(Default)]
584struct SelectionHistory {
585 #[allow(clippy::type_complexity)]
586 selections_by_transaction:
587 HashMap<TransactionId, (Arc<[Selection<Anchor>]>, Option<Arc<[Selection<Anchor>]>>)>,
588 mode: SelectionHistoryMode,
589 undo_stack: VecDeque<SelectionHistoryEntry>,
590 redo_stack: VecDeque<SelectionHistoryEntry>,
591}
592
593impl SelectionHistory {
594 fn insert_transaction(
595 &mut self,
596 transaction_id: TransactionId,
597 selections: Arc<[Selection<Anchor>]>,
598 ) {
599 self.selections_by_transaction
600 .insert(transaction_id, (selections, None));
601 }
602
603 #[allow(clippy::type_complexity)]
604 fn transaction(
605 &self,
606 transaction_id: TransactionId,
607 ) -> Option<&(Arc<[Selection<Anchor>]>, Option<Arc<[Selection<Anchor>]>>)> {
608 self.selections_by_transaction.get(&transaction_id)
609 }
610
611 #[allow(clippy::type_complexity)]
612 fn transaction_mut(
613 &mut self,
614 transaction_id: TransactionId,
615 ) -> Option<&mut (Arc<[Selection<Anchor>]>, Option<Arc<[Selection<Anchor>]>>)> {
616 self.selections_by_transaction.get_mut(&transaction_id)
617 }
618
619 fn push(&mut self, entry: SelectionHistoryEntry) {
620 if !entry.selections.is_empty() {
621 match self.mode {
622 SelectionHistoryMode::Normal => {
623 self.push_undo(entry);
624 self.redo_stack.clear();
625 }
626 SelectionHistoryMode::Undoing => self.push_redo(entry),
627 SelectionHistoryMode::Redoing => self.push_undo(entry),
628 }
629 }
630 }
631
632 fn push_undo(&mut self, entry: SelectionHistoryEntry) {
633 if self
634 .undo_stack
635 .back()
636 .map_or(true, |e| e.selections != entry.selections)
637 {
638 self.undo_stack.push_back(entry);
639 if self.undo_stack.len() > MAX_SELECTION_HISTORY_LEN {
640 self.undo_stack.pop_front();
641 }
642 }
643 }
644
645 fn push_redo(&mut self, entry: SelectionHistoryEntry) {
646 if self
647 .redo_stack
648 .back()
649 .map_or(true, |e| e.selections != entry.selections)
650 {
651 self.redo_stack.push_back(entry);
652 if self.redo_stack.len() > MAX_SELECTION_HISTORY_LEN {
653 self.redo_stack.pop_front();
654 }
655 }
656 }
657}
658
659#[derive(Clone, Debug)]
660struct AddSelectionsState {
661 above: bool,
662 stack: Vec<usize>,
663}
664
665#[derive(Clone)]
666struct SelectNextState {
667 query: AhoCorasick,
668 wordwise: bool,
669 done: bool,
670}
671
672impl std::fmt::Debug for SelectNextState {
673 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
674 f.debug_struct(std::any::type_name::<Self>())
675 .field("wordwise", &self.wordwise)
676 .field("done", &self.done)
677 .finish()
678 }
679}
680
681#[derive(Debug)]
682struct AutocloseRegion {
683 selection_id: usize,
684 range: Range<Anchor>,
685 pair: BracketPair,
686}
687
688#[derive(Debug)]
689struct SnippetState {
690 ranges: Vec<Vec<Range<Anchor>>>,
691 active_index: usize,
692}
693
694#[doc(hidden)]
695pub struct RenameState {
696 pub range: Range<Anchor>,
697 pub old_name: Arc<str>,
698 pub editor: View<Editor>,
699 block_id: BlockId,
700}
701
702struct InvalidationStack<T>(Vec<T>);
703
704struct RegisteredInlineCompletionProvider {
705 provider: Arc<dyn InlineCompletionProviderHandle>,
706 _subscription: Subscription,
707}
708
709enum ContextMenu {
710 Completions(CompletionsMenu),
711 CodeActions(CodeActionsMenu),
712}
713
714impl ContextMenu {
715 fn select_first(
716 &mut self,
717 project: Option<&Model<Project>>,
718 cx: &mut ViewContext<Editor>,
719 ) -> bool {
720 if self.visible() {
721 match self {
722 ContextMenu::Completions(menu) => menu.select_first(project, cx),
723 ContextMenu::CodeActions(menu) => menu.select_first(cx),
724 }
725 true
726 } else {
727 false
728 }
729 }
730
731 fn select_prev(
732 &mut self,
733 project: Option<&Model<Project>>,
734 cx: &mut ViewContext<Editor>,
735 ) -> bool {
736 if self.visible() {
737 match self {
738 ContextMenu::Completions(menu) => menu.select_prev(project, cx),
739 ContextMenu::CodeActions(menu) => menu.select_prev(cx),
740 }
741 true
742 } else {
743 false
744 }
745 }
746
747 fn select_next(
748 &mut self,
749 project: Option<&Model<Project>>,
750 cx: &mut ViewContext<Editor>,
751 ) -> bool {
752 if self.visible() {
753 match self {
754 ContextMenu::Completions(menu) => menu.select_next(project, cx),
755 ContextMenu::CodeActions(menu) => menu.select_next(cx),
756 }
757 true
758 } else {
759 false
760 }
761 }
762
763 fn select_last(
764 &mut self,
765 project: Option<&Model<Project>>,
766 cx: &mut ViewContext<Editor>,
767 ) -> bool {
768 if self.visible() {
769 match self {
770 ContextMenu::Completions(menu) => menu.select_last(project, cx),
771 ContextMenu::CodeActions(menu) => menu.select_last(cx),
772 }
773 true
774 } else {
775 false
776 }
777 }
778
779 fn visible(&self) -> bool {
780 match self {
781 ContextMenu::Completions(menu) => menu.visible(),
782 ContextMenu::CodeActions(menu) => menu.visible(),
783 }
784 }
785
786 fn render(
787 &self,
788 cursor_position: DisplayPoint,
789 style: &EditorStyle,
790 max_height: Pixels,
791 workspace: Option<WeakView<Workspace>>,
792 cx: &mut ViewContext<Editor>,
793 ) -> (ContextMenuOrigin, AnyElement) {
794 match self {
795 ContextMenu::Completions(menu) => (
796 ContextMenuOrigin::EditorPoint(cursor_position),
797 menu.render(style, max_height, workspace, cx),
798 ),
799 ContextMenu::CodeActions(menu) => menu.render(cursor_position, style, max_height, cx),
800 }
801 }
802}
803
804enum ContextMenuOrigin {
805 EditorPoint(DisplayPoint),
806 GutterIndicator(u32),
807}
808
809#[derive(Clone)]
810struct CompletionsMenu {
811 id: CompletionId,
812 initial_position: Anchor,
813 buffer: Model<Buffer>,
814 completions: Arc<RwLock<Box<[Completion]>>>,
815 match_candidates: Arc<[StringMatchCandidate]>,
816 matches: Arc<[StringMatch]>,
817 selected_item: usize,
818 scroll_handle: UniformListScrollHandle,
819 selected_completion_documentation_resolve_debounce: Arc<Mutex<DebouncedDelay>>,
820}
821
822impl CompletionsMenu {
823 fn select_first(&mut self, project: Option<&Model<Project>>, cx: &mut ViewContext<Editor>) {
824 self.selected_item = 0;
825 self.scroll_handle.scroll_to_item(self.selected_item);
826 self.attempt_resolve_selected_completion_documentation(project, cx);
827 cx.notify();
828 }
829
830 fn select_prev(&mut self, project: Option<&Model<Project>>, cx: &mut ViewContext<Editor>) {
831 if self.selected_item > 0 {
832 self.selected_item -= 1;
833 } else {
834 self.selected_item = self.matches.len() - 1;
835 }
836 self.scroll_handle.scroll_to_item(self.selected_item);
837 self.attempt_resolve_selected_completion_documentation(project, cx);
838 cx.notify();
839 }
840
841 fn select_next(&mut self, project: Option<&Model<Project>>, cx: &mut ViewContext<Editor>) {
842 if self.selected_item + 1 < self.matches.len() {
843 self.selected_item += 1;
844 } else {
845 self.selected_item = 0;
846 }
847 self.scroll_handle.scroll_to_item(self.selected_item);
848 self.attempt_resolve_selected_completion_documentation(project, cx);
849 cx.notify();
850 }
851
852 fn select_last(&mut self, project: Option<&Model<Project>>, cx: &mut ViewContext<Editor>) {
853 self.selected_item = self.matches.len() - 1;
854 self.scroll_handle.scroll_to_item(self.selected_item);
855 self.attempt_resolve_selected_completion_documentation(project, cx);
856 cx.notify();
857 }
858
859 fn pre_resolve_completion_documentation(
860 buffer: Model<Buffer>,
861 completions: Arc<RwLock<Box<[Completion]>>>,
862 matches: Arc<[StringMatch]>,
863 editor: &Editor,
864 cx: &mut ViewContext<Editor>,
865 ) -> Task<()> {
866 let settings = EditorSettings::get_global(cx);
867 if !settings.show_completion_documentation {
868 return Task::ready(());
869 }
870
871 let Some(provider) = editor.completion_provider.as_ref() else {
872 return Task::ready(());
873 };
874
875 let resolve_task = provider.resolve_completions(
876 buffer,
877 matches.iter().map(|m| m.candidate_id).collect(),
878 completions.clone(),
879 cx,
880 );
881
882 return cx.spawn(move |this, mut cx| async move {
883 if let Some(true) = resolve_task.await.log_err() {
884 this.update(&mut cx, |_, cx| cx.notify()).ok();
885 }
886 });
887 }
888
889 fn attempt_resolve_selected_completion_documentation(
890 &mut self,
891 project: Option<&Model<Project>>,
892 cx: &mut ViewContext<Editor>,
893 ) {
894 let settings = EditorSettings::get_global(cx);
895 if !settings.show_completion_documentation {
896 return;
897 }
898
899 let completion_index = self.matches[self.selected_item].candidate_id;
900 let Some(project) = project else {
901 return;
902 };
903
904 let resolve_task = project.update(cx, |project, cx| {
905 project.resolve_completions(
906 self.buffer.clone(),
907 vec![completion_index],
908 self.completions.clone(),
909 cx,
910 )
911 });
912
913 let delay_ms =
914 EditorSettings::get_global(cx).completion_documentation_secondary_query_debounce;
915 let delay = Duration::from_millis(delay_ms);
916
917 self.selected_completion_documentation_resolve_debounce
918 .lock()
919 .fire_new(delay, cx, |_, cx| {
920 cx.spawn(move |this, mut cx| async move {
921 if let Some(true) = resolve_task.await.log_err() {
922 this.update(&mut cx, |_, cx| cx.notify()).ok();
923 }
924 })
925 });
926 }
927
928 fn visible(&self) -> bool {
929 !self.matches.is_empty()
930 }
931
932 fn render(
933 &self,
934 style: &EditorStyle,
935 max_height: Pixels,
936 workspace: Option<WeakView<Workspace>>,
937 cx: &mut ViewContext<Editor>,
938 ) -> AnyElement {
939 let settings = EditorSettings::get_global(cx);
940 let show_completion_documentation = settings.show_completion_documentation;
941
942 let widest_completion_ix = self
943 .matches
944 .iter()
945 .enumerate()
946 .max_by_key(|(_, mat)| {
947 let completions = self.completions.read();
948 let completion = &completions[mat.candidate_id];
949 let documentation = &completion.documentation;
950
951 let mut len = completion.label.text.chars().count();
952 if let Some(Documentation::SingleLine(text)) = documentation {
953 if show_completion_documentation {
954 len += text.chars().count();
955 }
956 }
957
958 len
959 })
960 .map(|(ix, _)| ix);
961
962 let completions = self.completions.clone();
963 let matches = self.matches.clone();
964 let selected_item = self.selected_item;
965 let style = style.clone();
966
967 let multiline_docs = if show_completion_documentation {
968 let mat = &self.matches[selected_item];
969 let multiline_docs = match &self.completions.read()[mat.candidate_id].documentation {
970 Some(Documentation::MultiLinePlainText(text)) => {
971 Some(div().child(SharedString::from(text.clone())))
972 }
973 Some(Documentation::MultiLineMarkdown(parsed)) if !parsed.text.is_empty() => {
974 Some(div().child(render_parsed_markdown(
975 "completions_markdown",
976 parsed,
977 &style,
978 workspace,
979 cx,
980 )))
981 }
982 _ => None,
983 };
984 multiline_docs.map(|div| {
985 div.id("multiline_docs")
986 .max_h(max_height)
987 .flex_1()
988 .px_1p5()
989 .py_1()
990 .min_w(px(260.))
991 .max_w(px(640.))
992 .w(px(500.))
993 .overflow_y_scroll()
994 .occlude()
995 })
996 } else {
997 None
998 };
999
1000 let list = uniform_list(
1001 cx.view().clone(),
1002 "completions",
1003 matches.len(),
1004 move |_editor, range, cx| {
1005 let start_ix = range.start;
1006 let completions_guard = completions.read();
1007
1008 matches[range]
1009 .iter()
1010 .enumerate()
1011 .map(|(ix, mat)| {
1012 let item_ix = start_ix + ix;
1013 let candidate_id = mat.candidate_id;
1014 let completion = &completions_guard[candidate_id];
1015
1016 let documentation = if show_completion_documentation {
1017 &completion.documentation
1018 } else {
1019 &None
1020 };
1021
1022 let highlights = gpui::combine_highlights(
1023 mat.ranges().map(|range| (range, FontWeight::BOLD.into())),
1024 styled_runs_for_code_label(&completion.label, &style.syntax).map(
1025 |(range, mut highlight)| {
1026 // Ignore font weight for syntax highlighting, as we'll use it
1027 // for fuzzy matches.
1028 highlight.font_weight = None;
1029
1030 if completion.lsp_completion.deprecated.unwrap_or(false) {
1031 highlight.strikethrough = Some(StrikethroughStyle {
1032 thickness: 1.0.into(),
1033 ..Default::default()
1034 });
1035 highlight.color = Some(cx.theme().colors().text_muted);
1036 }
1037
1038 (range, highlight)
1039 },
1040 ),
1041 );
1042 let completion_label = StyledText::new(completion.label.text.clone())
1043 .with_highlights(&style.text, highlights);
1044 let documentation_label =
1045 if let Some(Documentation::SingleLine(text)) = documentation {
1046 if text.trim().is_empty() {
1047 None
1048 } else {
1049 Some(
1050 h_flex().ml_4().child(
1051 Label::new(text.clone())
1052 .size(LabelSize::Small)
1053 .color(Color::Muted),
1054 ),
1055 )
1056 }
1057 } else {
1058 None
1059 };
1060
1061 div().min_w(px(220.)).max_w(px(540.)).child(
1062 ListItem::new(mat.candidate_id)
1063 .inset(true)
1064 .selected(item_ix == selected_item)
1065 .on_click(cx.listener(move |editor, _event, cx| {
1066 cx.stop_propagation();
1067 if let Some(task) = editor.confirm_completion(
1068 &ConfirmCompletion {
1069 item_ix: Some(item_ix),
1070 },
1071 cx,
1072 ) {
1073 task.detach_and_log_err(cx)
1074 }
1075 }))
1076 .child(h_flex().overflow_hidden().child(completion_label))
1077 .end_slot::<Div>(documentation_label),
1078 )
1079 })
1080 .collect()
1081 },
1082 )
1083 .occlude()
1084 .max_h(max_height)
1085 .track_scroll(self.scroll_handle.clone())
1086 .with_width_from_item(widest_completion_ix);
1087
1088 Popover::new()
1089 .child(list)
1090 .when_some(multiline_docs, |popover, multiline_docs| {
1091 popover.aside(multiline_docs)
1092 })
1093 .into_any_element()
1094 }
1095
1096 pub async fn filter(&mut self, query: Option<&str>, executor: BackgroundExecutor) {
1097 let mut matches = if let Some(query) = query {
1098 fuzzy::match_strings(
1099 &self.match_candidates,
1100 query,
1101 query.chars().any(|c| c.is_uppercase()),
1102 100,
1103 &Default::default(),
1104 executor,
1105 )
1106 .await
1107 } else {
1108 self.match_candidates
1109 .iter()
1110 .enumerate()
1111 .map(|(candidate_id, candidate)| StringMatch {
1112 candidate_id,
1113 score: Default::default(),
1114 positions: Default::default(),
1115 string: candidate.string.clone(),
1116 })
1117 .collect()
1118 };
1119
1120 // Remove all candidates where the query's start does not match the start of any word in the candidate
1121 if let Some(query) = query {
1122 if let Some(query_start) = query.chars().next() {
1123 matches.retain(|string_match| {
1124 split_words(&string_match.string).any(|word| {
1125 // Check that the first codepoint of the word as lowercase matches the first
1126 // codepoint of the query as lowercase
1127 word.chars()
1128 .flat_map(|codepoint| codepoint.to_lowercase())
1129 .zip(query_start.to_lowercase())
1130 .all(|(word_cp, query_cp)| word_cp == query_cp)
1131 })
1132 });
1133 }
1134 }
1135
1136 let completions = self.completions.read();
1137 matches.sort_unstable_by_key(|mat| {
1138 // We do want to strike a balance here between what the language server tells us
1139 // to sort by (the sort_text) and what are "obvious" good matches (i.e. when you type
1140 // `Creat` and there is a local variable called `CreateComponent`).
1141 // So what we do is: we bucket all matches into two buckets
1142 // - Strong matches
1143 // - Weak matches
1144 // Strong matches are the ones with a high fuzzy-matcher score (the "obvious" matches)
1145 // and the Weak matches are the rest.
1146 //
1147 // For the strong matches, we sort by the language-servers score first and for the weak
1148 // matches, we prefer our fuzzy finder first.
1149 //
1150 // The thinking behind that: it's useless to take the sort_text the language-server gives
1151 // us into account when it's obviously a bad match.
1152
1153 #[derive(PartialEq, Eq, PartialOrd, Ord)]
1154 enum MatchScore<'a> {
1155 Strong {
1156 sort_text: Option<&'a str>,
1157 score: Reverse<OrderedFloat<f64>>,
1158 sort_key: (usize, &'a str),
1159 },
1160 Weak {
1161 score: Reverse<OrderedFloat<f64>>,
1162 sort_text: Option<&'a str>,
1163 sort_key: (usize, &'a str),
1164 },
1165 }
1166
1167 let completion = &completions[mat.candidate_id];
1168 let sort_key = completion.sort_key();
1169 let sort_text = completion.lsp_completion.sort_text.as_deref();
1170 let score = Reverse(OrderedFloat(mat.score));
1171
1172 if mat.score >= 0.2 {
1173 MatchScore::Strong {
1174 sort_text,
1175 score,
1176 sort_key,
1177 }
1178 } else {
1179 MatchScore::Weak {
1180 score,
1181 sort_text,
1182 sort_key,
1183 }
1184 }
1185 });
1186
1187 for mat in &mut matches {
1188 let completion = &completions[mat.candidate_id];
1189 mat.string.clone_from(&completion.label.text);
1190 for position in &mut mat.positions {
1191 *position += completion.label.filter_range.start;
1192 }
1193 }
1194 drop(completions);
1195
1196 self.matches = matches.into();
1197 self.selected_item = 0;
1198 }
1199}
1200
1201#[derive(Clone)]
1202struct CodeActionContents {
1203 tasks: Option<Arc<ResolvedTasks>>,
1204 actions: Option<Arc<[CodeAction]>>,
1205}
1206
1207impl CodeActionContents {
1208 fn len(&self) -> usize {
1209 match (&self.tasks, &self.actions) {
1210 (Some(tasks), Some(actions)) => actions.len() + tasks.templates.len(),
1211 (Some(tasks), None) => tasks.templates.len(),
1212 (None, Some(actions)) => actions.len(),
1213 (None, None) => 0,
1214 }
1215 }
1216
1217 fn is_empty(&self) -> bool {
1218 match (&self.tasks, &self.actions) {
1219 (Some(tasks), Some(actions)) => actions.is_empty() && tasks.templates.is_empty(),
1220 (Some(tasks), None) => tasks.templates.is_empty(),
1221 (None, Some(actions)) => actions.is_empty(),
1222 (None, None) => true,
1223 }
1224 }
1225
1226 fn iter(&self) -> impl Iterator<Item = CodeActionsItem> + '_ {
1227 self.tasks
1228 .iter()
1229 .flat_map(|tasks| {
1230 tasks
1231 .templates
1232 .iter()
1233 .map(|(kind, task)| CodeActionsItem::Task(kind.clone(), task.clone()))
1234 })
1235 .chain(self.actions.iter().flat_map(|actions| {
1236 actions
1237 .iter()
1238 .map(|action| CodeActionsItem::CodeAction(action.clone()))
1239 }))
1240 }
1241 fn get(&self, index: usize) -> Option<CodeActionsItem> {
1242 match (&self.tasks, &self.actions) {
1243 (Some(tasks), Some(actions)) => {
1244 if index < tasks.templates.len() {
1245 tasks
1246 .templates
1247 .get(index)
1248 .cloned()
1249 .map(|(kind, task)| CodeActionsItem::Task(kind, task))
1250 } else {
1251 actions
1252 .get(index - tasks.templates.len())
1253 .cloned()
1254 .map(CodeActionsItem::CodeAction)
1255 }
1256 }
1257 (Some(tasks), None) => tasks
1258 .templates
1259 .get(index)
1260 .cloned()
1261 .map(|(kind, task)| CodeActionsItem::Task(kind, task)),
1262 (None, Some(actions)) => actions.get(index).cloned().map(CodeActionsItem::CodeAction),
1263 (None, None) => None,
1264 }
1265 }
1266}
1267
1268#[allow(clippy::large_enum_variant)]
1269#[derive(Clone)]
1270enum CodeActionsItem {
1271 Task(TaskSourceKind, ResolvedTask),
1272 CodeAction(CodeAction),
1273}
1274
1275impl CodeActionsItem {
1276 fn as_task(&self) -> Option<&ResolvedTask> {
1277 let Self::Task(_, task) = self else {
1278 return None;
1279 };
1280 Some(task)
1281 }
1282 fn as_code_action(&self) -> Option<&CodeAction> {
1283 let Self::CodeAction(action) = self else {
1284 return None;
1285 };
1286 Some(action)
1287 }
1288 fn label(&self) -> String {
1289 match self {
1290 Self::CodeAction(action) => action.lsp_action.title.clone(),
1291 Self::Task(_, task) => task.resolved_label.clone(),
1292 }
1293 }
1294}
1295
1296struct CodeActionsMenu {
1297 actions: CodeActionContents,
1298 buffer: Model<Buffer>,
1299 selected_item: usize,
1300 scroll_handle: UniformListScrollHandle,
1301 deployed_from_indicator: Option<u32>,
1302}
1303
1304impl CodeActionsMenu {
1305 fn select_first(&mut self, cx: &mut ViewContext<Editor>) {
1306 self.selected_item = 0;
1307 self.scroll_handle.scroll_to_item(self.selected_item);
1308 cx.notify()
1309 }
1310
1311 fn select_prev(&mut self, cx: &mut ViewContext<Editor>) {
1312 if self.selected_item > 0 {
1313 self.selected_item -= 1;
1314 } else {
1315 self.selected_item = self.actions.len() - 1;
1316 }
1317 self.scroll_handle.scroll_to_item(self.selected_item);
1318 cx.notify();
1319 }
1320
1321 fn select_next(&mut self, cx: &mut ViewContext<Editor>) {
1322 if self.selected_item + 1 < self.actions.len() {
1323 self.selected_item += 1;
1324 } else {
1325 self.selected_item = 0;
1326 }
1327 self.scroll_handle.scroll_to_item(self.selected_item);
1328 cx.notify();
1329 }
1330
1331 fn select_last(&mut self, cx: &mut ViewContext<Editor>) {
1332 self.selected_item = self.actions.len() - 1;
1333 self.scroll_handle.scroll_to_item(self.selected_item);
1334 cx.notify()
1335 }
1336
1337 fn visible(&self) -> bool {
1338 !self.actions.is_empty()
1339 }
1340
1341 fn render(
1342 &self,
1343 cursor_position: DisplayPoint,
1344 _style: &EditorStyle,
1345 max_height: Pixels,
1346 cx: &mut ViewContext<Editor>,
1347 ) -> (ContextMenuOrigin, AnyElement) {
1348 let actions = self.actions.clone();
1349 let selected_item = self.selected_item;
1350 let element = uniform_list(
1351 cx.view().clone(),
1352 "code_actions_menu",
1353 self.actions.len(),
1354 move |_this, range, cx| {
1355 actions
1356 .iter()
1357 .skip(range.start)
1358 .take(range.end - range.start)
1359 .enumerate()
1360 .map(|(ix, action)| {
1361 let item_ix = range.start + ix;
1362 let selected = selected_item == item_ix;
1363 let colors = cx.theme().colors();
1364 div()
1365 .px_2()
1366 .text_color(colors.text)
1367 .when(selected, |style| {
1368 style
1369 .bg(colors.element_active)
1370 .text_color(colors.text_accent)
1371 })
1372 .hover(|style| {
1373 style
1374 .bg(colors.element_hover)
1375 .text_color(colors.text_accent)
1376 })
1377 .whitespace_nowrap()
1378 .when_some(action.as_code_action(), |this, action| {
1379 this.on_mouse_down(
1380 MouseButton::Left,
1381 cx.listener(move |editor, _, cx| {
1382 cx.stop_propagation();
1383 if let Some(task) = editor.confirm_code_action(
1384 &ConfirmCodeAction {
1385 item_ix: Some(item_ix),
1386 },
1387 cx,
1388 ) {
1389 task.detach_and_log_err(cx)
1390 }
1391 }),
1392 )
1393 // TASK: It would be good to make lsp_action.title a SharedString to avoid allocating here.
1394 .child(SharedString::from(action.lsp_action.title.clone()))
1395 })
1396 .when_some(action.as_task(), |this, task| {
1397 this.on_mouse_down(
1398 MouseButton::Left,
1399 cx.listener(move |editor, _, cx| {
1400 cx.stop_propagation();
1401 if let Some(task) = editor.confirm_code_action(
1402 &ConfirmCodeAction {
1403 item_ix: Some(item_ix),
1404 },
1405 cx,
1406 ) {
1407 task.detach_and_log_err(cx)
1408 }
1409 }),
1410 )
1411 .child(SharedString::from(task.resolved_label.clone()))
1412 })
1413 })
1414 .collect()
1415 },
1416 )
1417 .elevation_1(cx)
1418 .px_2()
1419 .py_1()
1420 .max_h(max_height)
1421 .occlude()
1422 .track_scroll(self.scroll_handle.clone())
1423 .with_width_from_item(
1424 self.actions
1425 .iter()
1426 .enumerate()
1427 .max_by_key(|(_, action)| match action {
1428 CodeActionsItem::Task(_, task) => task.resolved_label.chars().count(),
1429 CodeActionsItem::CodeAction(action) => action.lsp_action.title.chars().count(),
1430 })
1431 .map(|(ix, _)| ix),
1432 )
1433 .into_any_element();
1434
1435 let cursor_position = if let Some(row) = self.deployed_from_indicator {
1436 ContextMenuOrigin::GutterIndicator(row)
1437 } else {
1438 ContextMenuOrigin::EditorPoint(cursor_position)
1439 };
1440
1441 (cursor_position, element)
1442 }
1443}
1444
1445#[derive(Debug)]
1446struct ActiveDiagnosticGroup {
1447 primary_range: Range<Anchor>,
1448 primary_message: String,
1449 blocks: HashMap<BlockId, Diagnostic>,
1450 is_valid: bool,
1451}
1452
1453#[derive(Serialize, Deserialize)]
1454pub struct ClipboardSelection {
1455 pub len: usize,
1456 pub is_entire_line: bool,
1457 pub first_line_indent: u32,
1458}
1459
1460#[derive(Debug)]
1461pub(crate) struct NavigationData {
1462 cursor_anchor: Anchor,
1463 cursor_position: Point,
1464 scroll_anchor: ScrollAnchor,
1465 scroll_top_row: u32,
1466}
1467
1468enum GotoDefinitionKind {
1469 Symbol,
1470 Type,
1471 Implementation,
1472}
1473
1474#[derive(Debug, Clone)]
1475enum InlayHintRefreshReason {
1476 Toggle(bool),
1477 SettingsChange(InlayHintSettings),
1478 NewLinesShown,
1479 BufferEdited(HashSet<Arc<Language>>),
1480 RefreshRequested,
1481 ExcerptsRemoved(Vec<ExcerptId>),
1482}
1483
1484impl InlayHintRefreshReason {
1485 fn description(&self) -> &'static str {
1486 match self {
1487 Self::Toggle(_) => "toggle",
1488 Self::SettingsChange(_) => "settings change",
1489 Self::NewLinesShown => "new lines shown",
1490 Self::BufferEdited(_) => "buffer edited",
1491 Self::RefreshRequested => "refresh requested",
1492 Self::ExcerptsRemoved(_) => "excerpts removed",
1493 }
1494 }
1495}
1496
1497impl Editor {
1498 pub fn single_line(cx: &mut ViewContext<Self>) -> Self {
1499 let buffer = cx.new_model(|cx| Buffer::local("", cx));
1500 let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx));
1501 Self::new(EditorMode::SingleLine, buffer, None, cx)
1502 }
1503
1504 pub fn multi_line(cx: &mut ViewContext<Self>) -> Self {
1505 let buffer = cx.new_model(|cx| Buffer::local("", cx));
1506 let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx));
1507 Self::new(EditorMode::Full, buffer, None, cx)
1508 }
1509
1510 pub fn auto_height(max_lines: usize, cx: &mut ViewContext<Self>) -> Self {
1511 let buffer = cx.new_model(|cx| Buffer::local("", cx));
1512 let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx));
1513 Self::new(EditorMode::AutoHeight { max_lines }, buffer, None, cx)
1514 }
1515
1516 pub fn for_buffer(
1517 buffer: Model<Buffer>,
1518 project: Option<Model<Project>>,
1519 cx: &mut ViewContext<Self>,
1520 ) -> Self {
1521 let buffer = cx.new_model(|cx| MultiBuffer::singleton(buffer, cx));
1522 Self::new(EditorMode::Full, buffer, project, cx)
1523 }
1524
1525 pub fn for_multibuffer(
1526 buffer: Model<MultiBuffer>,
1527 project: Option<Model<Project>>,
1528 cx: &mut ViewContext<Self>,
1529 ) -> Self {
1530 Self::new(EditorMode::Full, buffer, project, cx)
1531 }
1532
1533 pub fn clone(&self, cx: &mut ViewContext<Self>) -> Self {
1534 let mut clone = Self::new(self.mode, self.buffer.clone(), self.project.clone(), cx);
1535 self.display_map.update(cx, |display_map, cx| {
1536 let snapshot = display_map.snapshot(cx);
1537 clone.display_map.update(cx, |display_map, cx| {
1538 display_map.set_state(&snapshot, cx);
1539 });
1540 });
1541 clone.selections.clone_state(&self.selections);
1542 clone.scroll_manager.clone_state(&self.scroll_manager);
1543 clone.searchable = self.searchable;
1544 clone
1545 }
1546
1547 fn new(
1548 mode: EditorMode,
1549 buffer: Model<MultiBuffer>,
1550 project: Option<Model<Project>>,
1551 cx: &mut ViewContext<Self>,
1552 ) -> Self {
1553 let style = cx.text_style();
1554 let font_size = style.font_size.to_pixels(cx.rem_size());
1555 let display_map = cx.new_model(|cx| {
1556 DisplayMap::new(buffer.clone(), style.font(), font_size, None, 2, 1, cx)
1557 });
1558
1559 let selections = SelectionsCollection::new(display_map.clone(), buffer.clone());
1560
1561 let blink_manager = cx.new_model(|cx| BlinkManager::new(CURSOR_BLINK_INTERVAL, cx));
1562
1563 let soft_wrap_mode_override =
1564 (mode == EditorMode::SingleLine).then(|| language_settings::SoftWrap::PreferLine);
1565
1566 let mut project_subscriptions = Vec::new();
1567 if mode == EditorMode::Full {
1568 if let Some(project) = project.as_ref() {
1569 if buffer.read(cx).is_singleton() {
1570 project_subscriptions.push(cx.observe(project, |_, _, cx| {
1571 cx.emit(EditorEvent::TitleChanged);
1572 }));
1573 }
1574 project_subscriptions.push(cx.subscribe(project, |editor, _, event, cx| {
1575 if let project::Event::RefreshInlayHints = event {
1576 editor.refresh_inlay_hints(InlayHintRefreshReason::RefreshRequested, cx);
1577 };
1578 }));
1579 }
1580 }
1581
1582 let inlay_hint_settings = inlay_hint_settings(
1583 selections.newest_anchor().head(),
1584 &buffer.read(cx).snapshot(cx),
1585 cx,
1586 );
1587
1588 let focus_handle = cx.focus_handle();
1589 cx.on_focus(&focus_handle, Self::handle_focus).detach();
1590 cx.on_blur(&focus_handle, Self::handle_blur).detach();
1591
1592 let mut this = Self {
1593 focus_handle,
1594 buffer: buffer.clone(),
1595 display_map: display_map.clone(),
1596 selections,
1597 scroll_manager: ScrollManager::new(cx),
1598 columnar_selection_tail: None,
1599 add_selections_state: None,
1600 select_next_state: None,
1601 select_prev_state: None,
1602 selection_history: Default::default(),
1603 autoclose_regions: Default::default(),
1604 snippet_stack: Default::default(),
1605 select_larger_syntax_node_stack: Vec::new(),
1606 ime_transaction: Default::default(),
1607 active_diagnostics: None,
1608 soft_wrap_mode_override,
1609 completion_provider: project.clone().map(|project| Box::new(project) as _),
1610 collaboration_hub: project.clone().map(|project| Box::new(project) as _),
1611 project,
1612 blink_manager: blink_manager.clone(),
1613 show_local_selections: true,
1614 mode,
1615 show_breadcrumbs: EditorSettings::get_global(cx).toolbar.breadcrumbs,
1616 show_gutter: mode == EditorMode::Full,
1617 show_wrap_guides: None,
1618 placeholder_text: None,
1619 highlight_order: 0,
1620 highlighted_rows: HashMap::default(),
1621 background_highlights: Default::default(),
1622 scrollbar_marker_state: ScrollbarMarkerState::default(),
1623 nav_history: None,
1624 context_menu: RwLock::new(None),
1625 mouse_context_menu: None,
1626 completion_tasks: Default::default(),
1627 find_all_references_task_sources: Vec::new(),
1628 next_completion_id: 0,
1629 completion_documentation_pre_resolve_debounce: DebouncedDelay::new(),
1630 next_inlay_id: 0,
1631 available_code_actions: Default::default(),
1632 code_actions_task: Default::default(),
1633 document_highlights_task: Default::default(),
1634 pending_rename: Default::default(),
1635 searchable: true,
1636 cursor_shape: Default::default(),
1637 autoindent_mode: Some(AutoindentMode::EachLine),
1638 collapse_matches: false,
1639 workspace: None,
1640 keymap_context_layers: Default::default(),
1641 input_enabled: true,
1642 use_modal_editing: mode == EditorMode::Full,
1643 read_only: false,
1644 use_autoclose: true,
1645 auto_replace_emoji_shortcode: false,
1646 leader_peer_id: None,
1647 remote_id: None,
1648 hover_state: Default::default(),
1649 hovered_link_state: Default::default(),
1650 inline_completion_provider: None,
1651 active_inline_completion: None,
1652 inlay_hint_cache: InlayHintCache::new(inlay_hint_settings),
1653 expanded_hunks: ExpandedHunks::default(),
1654 gutter_hovered: false,
1655 pixel_position_of_newest_cursor: None,
1656 last_bounds: None,
1657 expect_bounds_change: None,
1658 gutter_dimensions: GutterDimensions::default(),
1659 style: None,
1660 show_cursor_names: false,
1661 hovered_cursors: Default::default(),
1662 editor_actions: Default::default(),
1663 vim_replace_map: Default::default(),
1664 show_inline_completions: mode == EditorMode::Full,
1665 custom_context_menu: None,
1666 show_git_blame_gutter: false,
1667 show_git_blame_inline: false,
1668 show_git_blame_inline_delay_task: None,
1669 git_blame_inline_enabled: ProjectSettings::get_global(cx).git.inline_blame_enabled(),
1670 blame: None,
1671 blame_subscription: None,
1672 tasks: Default::default(),
1673 _subscriptions: vec![
1674 cx.observe(&buffer, Self::on_buffer_changed),
1675 cx.subscribe(&buffer, Self::on_buffer_event),
1676 cx.observe(&display_map, Self::on_display_map_changed),
1677 cx.observe(&blink_manager, |_, _, cx| cx.notify()),
1678 cx.observe_global::<SettingsStore>(Self::settings_changed),
1679 observe_buffer_font_size_adjustment(cx, |_, cx| cx.notify()),
1680 cx.observe_window_activation(|editor, cx| {
1681 let active = cx.is_window_active();
1682 editor.blink_manager.update(cx, |blink_manager, cx| {
1683 if active {
1684 blink_manager.enable(cx);
1685 } else {
1686 blink_manager.show_cursor(cx);
1687 blink_manager.disable(cx);
1688 }
1689 });
1690 }),
1691 ],
1692 tasks_update_task: None,
1693 };
1694 this.tasks_update_task = Some(this.refresh_runnables(cx));
1695 this._subscriptions.extend(project_subscriptions);
1696
1697 this.end_selection(cx);
1698 this.scroll_manager.show_scrollbar(cx);
1699
1700 if mode == EditorMode::Full {
1701 let should_auto_hide_scrollbars = cx.should_auto_hide_scrollbars();
1702 cx.set_global(ScrollbarAutoHide(should_auto_hide_scrollbars));
1703
1704 if this.git_blame_inline_enabled {
1705 this.git_blame_inline_enabled = true;
1706 this.start_git_blame_inline(false, cx);
1707 }
1708 }
1709
1710 this.report_editor_event("open", None, cx);
1711 this
1712 }
1713
1714 fn key_context(&self, cx: &AppContext) -> KeyContext {
1715 let mut key_context = KeyContext::new_with_defaults();
1716 key_context.add("Editor");
1717 let mode = match self.mode {
1718 EditorMode::SingleLine => "single_line",
1719 EditorMode::AutoHeight { .. } => "auto_height",
1720 EditorMode::Full => "full",
1721 };
1722 key_context.set("mode", mode);
1723 if self.pending_rename.is_some() {
1724 key_context.add("renaming");
1725 }
1726 if self.context_menu_visible() {
1727 match self.context_menu.read().as_ref() {
1728 Some(ContextMenu::Completions(_)) => {
1729 key_context.add("menu");
1730 key_context.add("showing_completions")
1731 }
1732 Some(ContextMenu::CodeActions(_)) => {
1733 key_context.add("menu");
1734 key_context.add("showing_code_actions")
1735 }
1736 None => {}
1737 }
1738 }
1739
1740 for layer in self.keymap_context_layers.values() {
1741 key_context.extend(layer);
1742 }
1743
1744 if let Some(extension) = self
1745 .buffer
1746 .read(cx)
1747 .as_singleton()
1748 .and_then(|buffer| buffer.read(cx).file()?.path().extension()?.to_str())
1749 {
1750 key_context.set("extension", extension.to_string());
1751 }
1752
1753 if self.has_active_inline_completion(cx) {
1754 key_context.add("copilot_suggestion");
1755 key_context.add("inline_completion");
1756 }
1757
1758 key_context
1759 }
1760
1761 pub fn new_file(
1762 workspace: &mut Workspace,
1763 _: &workspace::NewFile,
1764 cx: &mut ViewContext<Workspace>,
1765 ) {
1766 let project = workspace.project().clone();
1767 let create = project.update(cx, |project, cx| project.create_buffer(cx));
1768
1769 cx.spawn(|workspace, mut cx| async move {
1770 let buffer = create.await?;
1771 workspace.update(&mut cx, |workspace, cx| {
1772 workspace.add_item_to_active_pane(
1773 Box::new(
1774 cx.new_view(|cx| Editor::for_buffer(buffer, Some(project.clone()), cx)),
1775 ),
1776 None,
1777 cx,
1778 )
1779 })
1780 })
1781 .detach_and_prompt_err("Failed to create buffer", cx, |e, _| match e.error_code() {
1782 ErrorCode::RemoteUpgradeRequired => Some(format!(
1783 "The remote instance of Zed does not support this yet. It must be upgraded to {}",
1784 e.error_tag("required").unwrap_or("the latest version")
1785 )),
1786 _ => None,
1787 });
1788 }
1789
1790 pub fn new_file_in_direction(
1791 workspace: &mut Workspace,
1792 action: &workspace::NewFileInDirection,
1793 cx: &mut ViewContext<Workspace>,
1794 ) {
1795 let project = workspace.project().clone();
1796 let create = project.update(cx, |project, cx| project.create_buffer(cx));
1797 let direction = action.0;
1798
1799 cx.spawn(|workspace, mut cx| async move {
1800 let buffer = create.await?;
1801 workspace.update(&mut cx, move |workspace, cx| {
1802 workspace.split_item(
1803 direction,
1804 Box::new(
1805 cx.new_view(|cx| Editor::for_buffer(buffer, Some(project.clone()), cx)),
1806 ),
1807 cx,
1808 )
1809 })?;
1810 anyhow::Ok(())
1811 })
1812 .detach_and_prompt_err("Failed to create buffer", cx, |e, _| match e.error_code() {
1813 ErrorCode::RemoteUpgradeRequired => Some(format!(
1814 "The remote instance of Zed does not support this yet. It must be upgraded to {}",
1815 e.error_tag("required").unwrap_or("the latest version")
1816 )),
1817 _ => None,
1818 });
1819 }
1820
1821 pub fn replica_id(&self, cx: &AppContext) -> ReplicaId {
1822 self.buffer.read(cx).replica_id()
1823 }
1824
1825 pub fn leader_peer_id(&self) -> Option<PeerId> {
1826 self.leader_peer_id
1827 }
1828
1829 pub fn buffer(&self) -> &Model<MultiBuffer> {
1830 &self.buffer
1831 }
1832
1833 pub fn workspace(&self) -> Option<View<Workspace>> {
1834 self.workspace.as_ref()?.0.upgrade()
1835 }
1836
1837 pub fn title<'a>(&self, cx: &'a AppContext) -> Cow<'a, str> {
1838 self.buffer().read(cx).title(cx)
1839 }
1840
1841 pub fn snapshot(&mut self, cx: &mut WindowContext) -> EditorSnapshot {
1842 EditorSnapshot {
1843 mode: self.mode,
1844 show_gutter: self.show_gutter,
1845 render_git_blame_gutter: self.render_git_blame_gutter(cx),
1846 display_snapshot: self.display_map.update(cx, |map, cx| map.snapshot(cx)),
1847 scroll_anchor: self.scroll_manager.anchor(),
1848 ongoing_scroll: self.scroll_manager.ongoing_scroll(),
1849 placeholder_text: self.placeholder_text.clone(),
1850 is_focused: self.focus_handle.is_focused(cx),
1851 }
1852 }
1853
1854 pub fn language_at<T: ToOffset>(&self, point: T, cx: &AppContext) -> Option<Arc<Language>> {
1855 self.buffer.read(cx).language_at(point, cx)
1856 }
1857
1858 pub fn file_at<T: ToOffset>(
1859 &self,
1860 point: T,
1861 cx: &AppContext,
1862 ) -> Option<Arc<dyn language::File>> {
1863 self.buffer.read(cx).read(cx).file_at(point).cloned()
1864 }
1865
1866 pub fn active_excerpt(
1867 &self,
1868 cx: &AppContext,
1869 ) -> Option<(ExcerptId, Model<Buffer>, Range<text::Anchor>)> {
1870 self.buffer
1871 .read(cx)
1872 .excerpt_containing(self.selections.newest_anchor().head(), cx)
1873 }
1874
1875 pub fn mode(&self) -> EditorMode {
1876 self.mode
1877 }
1878
1879 pub fn collaboration_hub(&self) -> Option<&dyn CollaborationHub> {
1880 self.collaboration_hub.as_deref()
1881 }
1882
1883 pub fn set_collaboration_hub(&mut self, hub: Box<dyn CollaborationHub>) {
1884 self.collaboration_hub = Some(hub);
1885 }
1886
1887 pub fn set_custom_context_menu(
1888 &mut self,
1889 f: impl 'static
1890 + Fn(&mut Self, DisplayPoint, &mut ViewContext<Self>) -> Option<View<ui::ContextMenu>>,
1891 ) {
1892 self.custom_context_menu = Some(Box::new(f))
1893 }
1894
1895 pub fn set_completion_provider(&mut self, hub: Box<dyn CompletionProvider>) {
1896 self.completion_provider = Some(hub);
1897 }
1898
1899 pub fn set_inline_completion_provider<T>(
1900 &mut self,
1901 provider: Option<Model<T>>,
1902 cx: &mut ViewContext<Self>,
1903 ) where
1904 T: InlineCompletionProvider,
1905 {
1906 self.inline_completion_provider =
1907 provider.map(|provider| RegisteredInlineCompletionProvider {
1908 _subscription: cx.observe(&provider, |this, _, cx| {
1909 if this.focus_handle.is_focused(cx) {
1910 this.update_visible_inline_completion(cx);
1911 }
1912 }),
1913 provider: Arc::new(provider),
1914 });
1915 self.refresh_inline_completion(false, cx);
1916 }
1917
1918 pub fn placeholder_text(&self, _cx: &mut WindowContext) -> Option<&str> {
1919 self.placeholder_text.as_deref()
1920 }
1921
1922 pub fn set_placeholder_text(
1923 &mut self,
1924 placeholder_text: impl Into<Arc<str>>,
1925 cx: &mut ViewContext<Self>,
1926 ) {
1927 let placeholder_text = Some(placeholder_text.into());
1928 if self.placeholder_text != placeholder_text {
1929 self.placeholder_text = placeholder_text;
1930 cx.notify();
1931 }
1932 }
1933
1934 pub fn set_cursor_shape(&mut self, cursor_shape: CursorShape, cx: &mut ViewContext<Self>) {
1935 self.cursor_shape = cursor_shape;
1936 cx.notify();
1937 }
1938
1939 pub fn set_collapse_matches(&mut self, collapse_matches: bool) {
1940 self.collapse_matches = collapse_matches;
1941 }
1942
1943 pub fn range_for_match<T: std::marker::Copy>(&self, range: &Range<T>) -> Range<T> {
1944 if self.collapse_matches {
1945 return range.start..range.start;
1946 }
1947 range.clone()
1948 }
1949
1950 pub fn set_clip_at_line_ends(&mut self, clip: bool, cx: &mut ViewContext<Self>) {
1951 if self.display_map.read(cx).clip_at_line_ends != clip {
1952 self.display_map
1953 .update(cx, |map, _| map.clip_at_line_ends = clip);
1954 }
1955 }
1956
1957 pub fn set_keymap_context_layer<Tag: 'static>(
1958 &mut self,
1959 context: KeyContext,
1960 cx: &mut ViewContext<Self>,
1961 ) {
1962 self.keymap_context_layers
1963 .insert(TypeId::of::<Tag>(), context);
1964 cx.notify();
1965 }
1966
1967 pub fn remove_keymap_context_layer<Tag: 'static>(&mut self, cx: &mut ViewContext<Self>) {
1968 self.keymap_context_layers.remove(&TypeId::of::<Tag>());
1969 cx.notify();
1970 }
1971
1972 pub fn set_input_enabled(&mut self, input_enabled: bool) {
1973 self.input_enabled = input_enabled;
1974 }
1975
1976 pub fn set_autoindent(&mut self, autoindent: bool) {
1977 if autoindent {
1978 self.autoindent_mode = Some(AutoindentMode::EachLine);
1979 } else {
1980 self.autoindent_mode = None;
1981 }
1982 }
1983
1984 pub fn read_only(&self, cx: &AppContext) -> bool {
1985 self.read_only || self.buffer.read(cx).read_only()
1986 }
1987
1988 pub fn set_read_only(&mut self, read_only: bool) {
1989 self.read_only = read_only;
1990 }
1991
1992 pub fn set_use_autoclose(&mut self, autoclose: bool) {
1993 self.use_autoclose = autoclose;
1994 }
1995
1996 pub fn set_auto_replace_emoji_shortcode(&mut self, auto_replace: bool) {
1997 self.auto_replace_emoji_shortcode = auto_replace;
1998 }
1999
2000 pub fn set_show_inline_completions(&mut self, show_inline_completions: bool) {
2001 self.show_inline_completions = show_inline_completions;
2002 }
2003
2004 pub fn set_use_modal_editing(&mut self, to: bool) {
2005 self.use_modal_editing = to;
2006 }
2007
2008 pub fn use_modal_editing(&self) -> bool {
2009 self.use_modal_editing
2010 }
2011
2012 fn selections_did_change(
2013 &mut self,
2014 local: bool,
2015 old_cursor_position: &Anchor,
2016 cx: &mut ViewContext<Self>,
2017 ) {
2018 // Copy selections to primary selection buffer
2019 #[cfg(target_os = "linux")]
2020 if local {
2021 let selections = self.selections.all::<usize>(cx);
2022 let buffer_handle = self.buffer.read(cx).read(cx);
2023
2024 let mut text = String::new();
2025 for (index, selection) in selections.iter().enumerate() {
2026 let text_for_selection = buffer_handle
2027 .text_for_range(selection.start..selection.end)
2028 .collect::<String>();
2029
2030 text.push_str(&text_for_selection);
2031 if index != selections.len() - 1 {
2032 text.push('\n');
2033 }
2034 }
2035
2036 if !text.is_empty() {
2037 cx.write_to_primary(ClipboardItem::new(text));
2038 }
2039 }
2040
2041 if self.focus_handle.is_focused(cx) && self.leader_peer_id.is_none() {
2042 self.buffer.update(cx, |buffer, cx| {
2043 buffer.set_active_selections(
2044 &self.selections.disjoint_anchors(),
2045 self.selections.line_mode,
2046 self.cursor_shape,
2047 cx,
2048 )
2049 });
2050 }
2051
2052 let display_map = self
2053 .display_map
2054 .update(cx, |display_map, cx| display_map.snapshot(cx));
2055 let buffer = &display_map.buffer_snapshot;
2056 self.add_selections_state = None;
2057 self.select_next_state = None;
2058 self.select_prev_state = None;
2059 self.select_larger_syntax_node_stack.clear();
2060 self.invalidate_autoclose_regions(&self.selections.disjoint_anchors(), buffer);
2061 self.snippet_stack
2062 .invalidate(&self.selections.disjoint_anchors(), buffer);
2063 self.take_rename(false, cx);
2064
2065 let new_cursor_position = self.selections.newest_anchor().head();
2066
2067 self.push_to_nav_history(
2068 *old_cursor_position,
2069 Some(new_cursor_position.to_point(buffer)),
2070 cx,
2071 );
2072
2073 if local {
2074 let new_cursor_position = self.selections.newest_anchor().head();
2075 let mut context_menu = self.context_menu.write();
2076 let completion_menu = match context_menu.as_ref() {
2077 Some(ContextMenu::Completions(menu)) => Some(menu),
2078
2079 _ => {
2080 *context_menu = None;
2081 None
2082 }
2083 };
2084
2085 if let Some(completion_menu) = completion_menu {
2086 let cursor_position = new_cursor_position.to_offset(buffer);
2087 let (word_range, kind) = buffer.surrounding_word(completion_menu.initial_position);
2088 if kind == Some(CharKind::Word)
2089 && word_range.to_inclusive().contains(&cursor_position)
2090 {
2091 let mut completion_menu = completion_menu.clone();
2092 drop(context_menu);
2093
2094 let query = Self::completion_query(buffer, cursor_position);
2095 cx.spawn(move |this, mut cx| async move {
2096 completion_menu
2097 .filter(query.as_deref(), cx.background_executor().clone())
2098 .await;
2099
2100 this.update(&mut cx, |this, cx| {
2101 let mut context_menu = this.context_menu.write();
2102 let Some(ContextMenu::Completions(menu)) = context_menu.as_ref() else {
2103 return;
2104 };
2105
2106 if menu.id > completion_menu.id {
2107 return;
2108 }
2109
2110 *context_menu = Some(ContextMenu::Completions(completion_menu));
2111 drop(context_menu);
2112 cx.notify();
2113 })
2114 })
2115 .detach();
2116
2117 self.show_completions(&ShowCompletions, cx);
2118 } else {
2119 drop(context_menu);
2120 self.hide_context_menu(cx);
2121 }
2122 } else {
2123 drop(context_menu);
2124 }
2125
2126 hide_hover(self, cx);
2127
2128 if old_cursor_position.to_display_point(&display_map).row()
2129 != new_cursor_position.to_display_point(&display_map).row()
2130 {
2131 self.available_code_actions.take();
2132 }
2133 self.refresh_code_actions(cx);
2134 self.refresh_document_highlights(cx);
2135 refresh_matching_bracket_highlights(self, cx);
2136 self.discard_inline_completion(cx);
2137 if self.git_blame_inline_enabled {
2138 self.start_inline_blame_timer(cx);
2139 }
2140 }
2141
2142 self.blink_manager.update(cx, BlinkManager::pause_blinking);
2143 cx.emit(EditorEvent::SelectionsChanged { local });
2144
2145 if self.selections.disjoint_anchors().len() == 1 {
2146 cx.emit(SearchEvent::ActiveMatchChanged)
2147 }
2148
2149 cx.notify();
2150 }
2151
2152 pub fn change_selections<R>(
2153 &mut self,
2154 autoscroll: Option<Autoscroll>,
2155 cx: &mut ViewContext<Self>,
2156 change: impl FnOnce(&mut MutableSelectionsCollection<'_>) -> R,
2157 ) -> R {
2158 let old_cursor_position = self.selections.newest_anchor().head();
2159 self.push_to_selection_history();
2160
2161 let (changed, result) = self.selections.change_with(cx, change);
2162
2163 if changed {
2164 if let Some(autoscroll) = autoscroll {
2165 self.request_autoscroll(autoscroll, cx);
2166 }
2167 self.selections_did_change(true, &old_cursor_position, cx);
2168 }
2169
2170 result
2171 }
2172
2173 pub fn edit<I, S, T>(&mut self, edits: I, cx: &mut ViewContext<Self>)
2174 where
2175 I: IntoIterator<Item = (Range<S>, T)>,
2176 S: ToOffset,
2177 T: Into<Arc<str>>,
2178 {
2179 if self.read_only(cx) {
2180 return;
2181 }
2182
2183 self.buffer
2184 .update(cx, |buffer, cx| buffer.edit(edits, None, cx));
2185 }
2186
2187 pub fn edit_with_autoindent<I, S, T>(&mut self, edits: I, cx: &mut ViewContext<Self>)
2188 where
2189 I: IntoIterator<Item = (Range<S>, T)>,
2190 S: ToOffset,
2191 T: Into<Arc<str>>,
2192 {
2193 if self.read_only(cx) {
2194 return;
2195 }
2196
2197 self.buffer.update(cx, |buffer, cx| {
2198 buffer.edit(edits, self.autoindent_mode.clone(), cx)
2199 });
2200 }
2201
2202 pub fn edit_with_block_indent<I, S, T>(
2203 &mut self,
2204 edits: I,
2205 original_indent_columns: Vec<u32>,
2206 cx: &mut ViewContext<Self>,
2207 ) where
2208 I: IntoIterator<Item = (Range<S>, T)>,
2209 S: ToOffset,
2210 T: Into<Arc<str>>,
2211 {
2212 if self.read_only(cx) {
2213 return;
2214 }
2215
2216 self.buffer.update(cx, |buffer, cx| {
2217 buffer.edit(
2218 edits,
2219 Some(AutoindentMode::Block {
2220 original_indent_columns,
2221 }),
2222 cx,
2223 )
2224 });
2225 }
2226
2227 fn select(&mut self, phase: SelectPhase, cx: &mut ViewContext<Self>) {
2228 self.hide_context_menu(cx);
2229
2230 match phase {
2231 SelectPhase::Begin {
2232 position,
2233 add,
2234 click_count,
2235 } => self.begin_selection(position, add, click_count, cx),
2236 SelectPhase::BeginColumnar {
2237 position,
2238 goal_column,
2239 } => self.begin_columnar_selection(position, goal_column, cx),
2240 SelectPhase::Extend {
2241 position,
2242 click_count,
2243 } => self.extend_selection(position, click_count, cx),
2244 SelectPhase::Update {
2245 position,
2246 goal_column,
2247 scroll_delta,
2248 } => self.update_selection(position, goal_column, scroll_delta, cx),
2249 SelectPhase::End => self.end_selection(cx),
2250 }
2251 }
2252
2253 fn extend_selection(
2254 &mut self,
2255 position: DisplayPoint,
2256 click_count: usize,
2257 cx: &mut ViewContext<Self>,
2258 ) {
2259 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
2260 let tail = self.selections.newest::<usize>(cx).tail();
2261 self.begin_selection(position, false, click_count, cx);
2262
2263 let position = position.to_offset(&display_map, Bias::Left);
2264 let tail_anchor = display_map.buffer_snapshot.anchor_before(tail);
2265
2266 let mut pending_selection = self
2267 .selections
2268 .pending_anchor()
2269 .expect("extend_selection not called with pending selection");
2270 if position >= tail {
2271 pending_selection.start = tail_anchor;
2272 } else {
2273 pending_selection.end = tail_anchor;
2274 pending_selection.reversed = true;
2275 }
2276
2277 let mut pending_mode = self.selections.pending_mode().unwrap();
2278 match &mut pending_mode {
2279 SelectMode::Word(range) | SelectMode::Line(range) => *range = tail_anchor..tail_anchor,
2280 _ => {}
2281 }
2282
2283 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
2284 s.set_pending(pending_selection, pending_mode)
2285 });
2286 }
2287
2288 fn begin_selection(
2289 &mut self,
2290 position: DisplayPoint,
2291 add: bool,
2292 click_count: usize,
2293 cx: &mut ViewContext<Self>,
2294 ) {
2295 if !self.focus_handle.is_focused(cx) {
2296 cx.focus(&self.focus_handle);
2297 }
2298
2299 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
2300 let buffer = &display_map.buffer_snapshot;
2301 let newest_selection = self.selections.newest_anchor().clone();
2302 let position = display_map.clip_point(position, Bias::Left);
2303
2304 let start;
2305 let end;
2306 let mode;
2307 let auto_scroll;
2308 match click_count {
2309 1 => {
2310 start = buffer.anchor_before(position.to_point(&display_map));
2311 end = start;
2312 mode = SelectMode::Character;
2313 auto_scroll = true;
2314 }
2315 2 => {
2316 let range = movement::surrounding_word(&display_map, position);
2317 start = buffer.anchor_before(range.start.to_point(&display_map));
2318 end = buffer.anchor_before(range.end.to_point(&display_map));
2319 mode = SelectMode::Word(start..end);
2320 auto_scroll = true;
2321 }
2322 3 => {
2323 let position = display_map
2324 .clip_point(position, Bias::Left)
2325 .to_point(&display_map);
2326 let line_start = display_map.prev_line_boundary(position).0;
2327 let next_line_start = buffer.clip_point(
2328 display_map.next_line_boundary(position).0 + Point::new(1, 0),
2329 Bias::Left,
2330 );
2331 start = buffer.anchor_before(line_start);
2332 end = buffer.anchor_before(next_line_start);
2333 mode = SelectMode::Line(start..end);
2334 auto_scroll = true;
2335 }
2336 _ => {
2337 start = buffer.anchor_before(0);
2338 end = buffer.anchor_before(buffer.len());
2339 mode = SelectMode::All;
2340 auto_scroll = false;
2341 }
2342 }
2343
2344 self.change_selections(auto_scroll.then(|| Autoscroll::newest()), cx, |s| {
2345 if !add {
2346 s.clear_disjoint();
2347 } else if click_count > 1 {
2348 s.delete(newest_selection.id)
2349 }
2350
2351 s.set_pending_anchor_range(start..end, mode);
2352 });
2353 }
2354
2355 fn begin_columnar_selection(
2356 &mut self,
2357 position: DisplayPoint,
2358 goal_column: u32,
2359 cx: &mut ViewContext<Self>,
2360 ) {
2361 if !self.focus_handle.is_focused(cx) {
2362 cx.focus(&self.focus_handle);
2363 }
2364
2365 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
2366 let tail = self.selections.newest::<Point>(cx).tail();
2367 self.columnar_selection_tail = Some(display_map.buffer_snapshot.anchor_before(tail));
2368
2369 self.select_columns(
2370 tail.to_display_point(&display_map),
2371 position,
2372 goal_column,
2373 &display_map,
2374 cx,
2375 );
2376 }
2377
2378 fn update_selection(
2379 &mut self,
2380 position: DisplayPoint,
2381 goal_column: u32,
2382 scroll_delta: gpui::Point<f32>,
2383 cx: &mut ViewContext<Self>,
2384 ) {
2385 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
2386
2387 if let Some(tail) = self.columnar_selection_tail.as_ref() {
2388 let tail = tail.to_display_point(&display_map);
2389 self.select_columns(tail, position, goal_column, &display_map, cx);
2390 } else if let Some(mut pending) = self.selections.pending_anchor() {
2391 let buffer = self.buffer.read(cx).snapshot(cx);
2392 let head;
2393 let tail;
2394 let mode = self.selections.pending_mode().unwrap();
2395 match &mode {
2396 SelectMode::Character => {
2397 head = position.to_point(&display_map);
2398 tail = pending.tail().to_point(&buffer);
2399 }
2400 SelectMode::Word(original_range) => {
2401 let original_display_range = original_range.start.to_display_point(&display_map)
2402 ..original_range.end.to_display_point(&display_map);
2403 let original_buffer_range = original_display_range.start.to_point(&display_map)
2404 ..original_display_range.end.to_point(&display_map);
2405 if movement::is_inside_word(&display_map, position)
2406 || original_display_range.contains(&position)
2407 {
2408 let word_range = movement::surrounding_word(&display_map, position);
2409 if word_range.start < original_display_range.start {
2410 head = word_range.start.to_point(&display_map);
2411 } else {
2412 head = word_range.end.to_point(&display_map);
2413 }
2414 } else {
2415 head = position.to_point(&display_map);
2416 }
2417
2418 if head <= original_buffer_range.start {
2419 tail = original_buffer_range.end;
2420 } else {
2421 tail = original_buffer_range.start;
2422 }
2423 }
2424 SelectMode::Line(original_range) => {
2425 let original_range = original_range.to_point(&display_map.buffer_snapshot);
2426
2427 let position = display_map
2428 .clip_point(position, Bias::Left)
2429 .to_point(&display_map);
2430 let line_start = display_map.prev_line_boundary(position).0;
2431 let next_line_start = buffer.clip_point(
2432 display_map.next_line_boundary(position).0 + Point::new(1, 0),
2433 Bias::Left,
2434 );
2435
2436 if line_start < original_range.start {
2437 head = line_start
2438 } else {
2439 head = next_line_start
2440 }
2441
2442 if head <= original_range.start {
2443 tail = original_range.end;
2444 } else {
2445 tail = original_range.start;
2446 }
2447 }
2448 SelectMode::All => {
2449 return;
2450 }
2451 };
2452
2453 if head < tail {
2454 pending.start = buffer.anchor_before(head);
2455 pending.end = buffer.anchor_before(tail);
2456 pending.reversed = true;
2457 } else {
2458 pending.start = buffer.anchor_before(tail);
2459 pending.end = buffer.anchor_before(head);
2460 pending.reversed = false;
2461 }
2462
2463 self.change_selections(None, cx, |s| {
2464 s.set_pending(pending, mode);
2465 });
2466 } else {
2467 log::error!("update_selection dispatched with no pending selection");
2468 return;
2469 }
2470
2471 self.apply_scroll_delta(scroll_delta, cx);
2472 cx.notify();
2473 }
2474
2475 fn end_selection(&mut self, cx: &mut ViewContext<Self>) {
2476 self.columnar_selection_tail.take();
2477 if self.selections.pending_anchor().is_some() {
2478 let selections = self.selections.all::<usize>(cx);
2479 self.change_selections(None, cx, |s| {
2480 s.select(selections);
2481 s.clear_pending();
2482 });
2483 }
2484 }
2485
2486 fn select_columns(
2487 &mut self,
2488 tail: DisplayPoint,
2489 head: DisplayPoint,
2490 goal_column: u32,
2491 display_map: &DisplaySnapshot,
2492 cx: &mut ViewContext<Self>,
2493 ) {
2494 let start_row = cmp::min(tail.row(), head.row());
2495 let end_row = cmp::max(tail.row(), head.row());
2496 let start_column = cmp::min(tail.column(), goal_column);
2497 let end_column = cmp::max(tail.column(), goal_column);
2498 let reversed = start_column < tail.column();
2499
2500 let selection_ranges = (start_row..=end_row)
2501 .filter_map(|row| {
2502 if start_column <= display_map.line_len(row) && !display_map.is_block_line(row) {
2503 let start = display_map
2504 .clip_point(DisplayPoint::new(row, start_column), Bias::Left)
2505 .to_point(display_map);
2506 let end = display_map
2507 .clip_point(DisplayPoint::new(row, end_column), Bias::Right)
2508 .to_point(display_map);
2509 if reversed {
2510 Some(end..start)
2511 } else {
2512 Some(start..end)
2513 }
2514 } else {
2515 None
2516 }
2517 })
2518 .collect::<Vec<_>>();
2519
2520 self.change_selections(None, cx, |s| {
2521 s.select_ranges(selection_ranges);
2522 });
2523 cx.notify();
2524 }
2525
2526 pub fn has_pending_nonempty_selection(&self) -> bool {
2527 let pending_nonempty_selection = match self.selections.pending_anchor() {
2528 Some(Selection { start, end, .. }) => start != end,
2529 None => false,
2530 };
2531 pending_nonempty_selection || self.columnar_selection_tail.is_some()
2532 }
2533
2534 pub fn has_pending_selection(&self) -> bool {
2535 self.selections.pending_anchor().is_some() || self.columnar_selection_tail.is_some()
2536 }
2537
2538 pub fn cancel(&mut self, _: &Cancel, cx: &mut ViewContext<Self>) {
2539 self.clear_expanded_diff_hunks(cx);
2540 if self.dismiss_menus_and_popups(cx) {
2541 return;
2542 }
2543
2544 if self.mode == EditorMode::Full {
2545 if self.change_selections(Some(Autoscroll::fit()), cx, |s| s.try_cancel()) {
2546 return;
2547 }
2548 }
2549
2550 cx.propagate();
2551 }
2552
2553 pub fn dismiss_menus_and_popups(&mut self, cx: &mut ViewContext<Self>) -> bool {
2554 if self.take_rename(false, cx).is_some() {
2555 return true;
2556 }
2557
2558 if hide_hover(self, cx) {
2559 return true;
2560 }
2561
2562 if self.hide_context_menu(cx).is_some() {
2563 return true;
2564 }
2565
2566 if self.discard_inline_completion(cx) {
2567 return true;
2568 }
2569
2570 if self.snippet_stack.pop().is_some() {
2571 return true;
2572 }
2573
2574 if self.mode == EditorMode::Full {
2575 if self.active_diagnostics.is_some() {
2576 self.dismiss_diagnostics(cx);
2577 return true;
2578 }
2579 }
2580
2581 false
2582 }
2583
2584 pub fn handle_input(&mut self, text: &str, cx: &mut ViewContext<Self>) {
2585 let text: Arc<str> = text.into();
2586
2587 if self.read_only(cx) {
2588 return;
2589 }
2590
2591 let selections = self.selections.all_adjusted(cx);
2592 let mut brace_inserted = false;
2593 let mut edits = Vec::new();
2594 let mut new_selections = Vec::with_capacity(selections.len());
2595 let mut new_autoclose_regions = Vec::new();
2596 let snapshot = self.buffer.read(cx).read(cx);
2597
2598 for (selection, autoclose_region) in
2599 self.selections_with_autoclose_regions(selections, &snapshot)
2600 {
2601 if let Some(scope) = snapshot.language_scope_at(selection.head()) {
2602 // Determine if the inserted text matches the opening or closing
2603 // bracket of any of this language's bracket pairs.
2604 let mut bracket_pair = None;
2605 let mut is_bracket_pair_start = false;
2606 let mut is_bracket_pair_end = false;
2607 if !text.is_empty() {
2608 // `text` can be empty when a user is using IME (e.g. Chinese Wubi Simplified)
2609 // and they are removing the character that triggered IME popup.
2610 for (pair, enabled) in scope.brackets() {
2611 if !pair.close {
2612 continue;
2613 }
2614
2615 if enabled && pair.start.ends_with(text.as_ref()) {
2616 bracket_pair = Some(pair.clone());
2617 is_bracket_pair_start = true;
2618 break;
2619 }
2620 if pair.end.as_str() == text.as_ref() {
2621 bracket_pair = Some(pair.clone());
2622 is_bracket_pair_end = true;
2623 break;
2624 }
2625 }
2626 }
2627
2628 if let Some(bracket_pair) = bracket_pair {
2629 if selection.is_empty() {
2630 if is_bracket_pair_start {
2631 let prefix_len = bracket_pair.start.len() - text.len();
2632
2633 // If the inserted text is a suffix of an opening bracket and the
2634 // selection is preceded by the rest of the opening bracket, then
2635 // insert the closing bracket.
2636 let following_text_allows_autoclose = snapshot
2637 .chars_at(selection.start)
2638 .next()
2639 .map_or(true, |c| scope.should_autoclose_before(c));
2640 let preceding_text_matches_prefix = prefix_len == 0
2641 || (selection.start.column >= (prefix_len as u32)
2642 && snapshot.contains_str_at(
2643 Point::new(
2644 selection.start.row,
2645 selection.start.column - (prefix_len as u32),
2646 ),
2647 &bracket_pair.start[..prefix_len],
2648 ));
2649 let autoclose = self.use_autoclose
2650 && snapshot.settings_at(selection.start, cx).use_autoclose;
2651 if autoclose
2652 && following_text_allows_autoclose
2653 && preceding_text_matches_prefix
2654 {
2655 let anchor = snapshot.anchor_before(selection.end);
2656 new_selections.push((selection.map(|_| anchor), text.len()));
2657 new_autoclose_regions.push((
2658 anchor,
2659 text.len(),
2660 selection.id,
2661 bracket_pair.clone(),
2662 ));
2663 edits.push((
2664 selection.range(),
2665 format!("{}{}", text, bracket_pair.end).into(),
2666 ));
2667 brace_inserted = true;
2668 continue;
2669 }
2670 }
2671
2672 if let Some(region) = autoclose_region {
2673 // If the selection is followed by an auto-inserted closing bracket,
2674 // then don't insert that closing bracket again; just move the selection
2675 // past the closing bracket.
2676 let should_skip = selection.end == region.range.end.to_point(&snapshot)
2677 && text.as_ref() == region.pair.end.as_str();
2678 if should_skip {
2679 let anchor = snapshot.anchor_after(selection.end);
2680 new_selections
2681 .push((selection.map(|_| anchor), region.pair.end.len()));
2682 continue;
2683 }
2684 }
2685
2686 let always_treat_brackets_as_autoclosed = snapshot
2687 .settings_at(selection.start, cx)
2688 .always_treat_brackets_as_autoclosed;
2689 if always_treat_brackets_as_autoclosed
2690 && is_bracket_pair_end
2691 && snapshot.contains_str_at(selection.end, text.as_ref())
2692 {
2693 // Otherwise, when `always_treat_brackets_as_autoclosed` is set to `true
2694 // and the inserted text is a closing bracket and the selection is followed
2695 // by the closing bracket then move the selection past the closing bracket.
2696 let anchor = snapshot.anchor_after(selection.end);
2697 new_selections.push((selection.map(|_| anchor), text.len()));
2698 continue;
2699 }
2700 }
2701 // If an opening bracket is 1 character long and is typed while
2702 // text is selected, then surround that text with the bracket pair.
2703 else if is_bracket_pair_start && bracket_pair.start.chars().count() == 1 {
2704 edits.push((selection.start..selection.start, text.clone()));
2705 edits.push((
2706 selection.end..selection.end,
2707 bracket_pair.end.as_str().into(),
2708 ));
2709 brace_inserted = true;
2710 new_selections.push((
2711 Selection {
2712 id: selection.id,
2713 start: snapshot.anchor_after(selection.start),
2714 end: snapshot.anchor_before(selection.end),
2715 reversed: selection.reversed,
2716 goal: selection.goal,
2717 },
2718 0,
2719 ));
2720 continue;
2721 }
2722 }
2723 }
2724
2725 if self.auto_replace_emoji_shortcode
2726 && selection.is_empty()
2727 && text.as_ref().ends_with(':')
2728 {
2729 if let Some(possible_emoji_short_code) =
2730 Self::find_possible_emoji_shortcode_at_position(&snapshot, selection.start)
2731 {
2732 if !possible_emoji_short_code.is_empty() {
2733 if let Some(emoji) = emojis::get_by_shortcode(&possible_emoji_short_code) {
2734 let emoji_shortcode_start = Point::new(
2735 selection.start.row,
2736 selection.start.column - possible_emoji_short_code.len() as u32 - 1,
2737 );
2738
2739 // Remove shortcode from buffer
2740 edits.push((
2741 emoji_shortcode_start..selection.start,
2742 "".to_string().into(),
2743 ));
2744 new_selections.push((
2745 Selection {
2746 id: selection.id,
2747 start: snapshot.anchor_after(emoji_shortcode_start),
2748 end: snapshot.anchor_before(selection.start),
2749 reversed: selection.reversed,
2750 goal: selection.goal,
2751 },
2752 0,
2753 ));
2754
2755 // Insert emoji
2756 let selection_start_anchor = snapshot.anchor_after(selection.start);
2757 new_selections.push((selection.map(|_| selection_start_anchor), 0));
2758 edits.push((selection.start..selection.end, emoji.to_string().into()));
2759
2760 continue;
2761 }
2762 }
2763 }
2764 }
2765
2766 // If not handling any auto-close operation, then just replace the selected
2767 // text with the given input and move the selection to the end of the
2768 // newly inserted text.
2769 let anchor = snapshot.anchor_after(selection.end);
2770 new_selections.push((selection.map(|_| anchor), 0));
2771 edits.push((selection.start..selection.end, text.clone()));
2772 }
2773
2774 drop(snapshot);
2775 self.transact(cx, |this, cx| {
2776 this.buffer.update(cx, |buffer, cx| {
2777 buffer.edit(edits, this.autoindent_mode.clone(), cx);
2778 });
2779
2780 let new_anchor_selections = new_selections.iter().map(|e| &e.0);
2781 let new_selection_deltas = new_selections.iter().map(|e| e.1);
2782 let snapshot = this.buffer.read(cx).read(cx);
2783 let new_selections = resolve_multiple::<usize, _>(new_anchor_selections, &snapshot)
2784 .zip(new_selection_deltas)
2785 .map(|(selection, delta)| Selection {
2786 id: selection.id,
2787 start: selection.start + delta,
2788 end: selection.end + delta,
2789 reversed: selection.reversed,
2790 goal: SelectionGoal::None,
2791 })
2792 .collect::<Vec<_>>();
2793
2794 let mut i = 0;
2795 for (position, delta, selection_id, pair) in new_autoclose_regions {
2796 let position = position.to_offset(&snapshot) + delta;
2797 let start = snapshot.anchor_before(position);
2798 let end = snapshot.anchor_after(position);
2799 while let Some(existing_state) = this.autoclose_regions.get(i) {
2800 match existing_state.range.start.cmp(&start, &snapshot) {
2801 Ordering::Less => i += 1,
2802 Ordering::Greater => break,
2803 Ordering::Equal => match end.cmp(&existing_state.range.end, &snapshot) {
2804 Ordering::Less => i += 1,
2805 Ordering::Equal => break,
2806 Ordering::Greater => break,
2807 },
2808 }
2809 }
2810 this.autoclose_regions.insert(
2811 i,
2812 AutocloseRegion {
2813 selection_id,
2814 range: start..end,
2815 pair,
2816 },
2817 );
2818 }
2819
2820 drop(snapshot);
2821 let had_active_inline_completion = this.has_active_inline_completion(cx);
2822 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(new_selections));
2823
2824 if brace_inserted {
2825 // If we inserted a brace while composing text (i.e. typing `"` on a
2826 // Brazilian keyboard), exit the composing state because most likely
2827 // the user wanted to surround the selection.
2828 this.unmark_text(cx);
2829 } else if EditorSettings::get_global(cx).use_on_type_format {
2830 if let Some(on_type_format_task) =
2831 this.trigger_on_type_formatting(text.to_string(), cx)
2832 {
2833 on_type_format_task.detach_and_log_err(cx);
2834 }
2835 }
2836
2837 let trigger_in_words = !had_active_inline_completion;
2838 this.trigger_completion_on_input(&text, trigger_in_words, cx);
2839 this.refresh_inline_completion(true, cx);
2840 });
2841 }
2842
2843 fn find_possible_emoji_shortcode_at_position(
2844 snapshot: &MultiBufferSnapshot,
2845 position: Point,
2846 ) -> Option<String> {
2847 let mut chars = Vec::new();
2848 let mut found_colon = false;
2849 for char in snapshot.reversed_chars_at(position).take(100) {
2850 // Found a possible emoji shortcode in the middle of the buffer
2851 if found_colon {
2852 if char.is_whitespace() {
2853 chars.reverse();
2854 return Some(chars.iter().collect());
2855 }
2856 // If the previous character is not a whitespace, we are in the middle of a word
2857 // and we only want to complete the shortcode if the word is made up of other emojis
2858 let mut containing_word = String::new();
2859 for ch in snapshot
2860 .reversed_chars_at(position)
2861 .skip(chars.len() + 1)
2862 .take(100)
2863 {
2864 if ch.is_whitespace() {
2865 break;
2866 }
2867 containing_word.push(ch);
2868 }
2869 let containing_word = containing_word.chars().rev().collect::<String>();
2870 if util::word_consists_of_emojis(containing_word.as_str()) {
2871 chars.reverse();
2872 return Some(chars.iter().collect());
2873 }
2874 }
2875
2876 if char.is_whitespace() || !char.is_ascii() {
2877 return None;
2878 }
2879 if char == ':' {
2880 found_colon = true;
2881 } else {
2882 chars.push(char);
2883 }
2884 }
2885 // Found a possible emoji shortcode at the beginning of the buffer
2886 chars.reverse();
2887 Some(chars.iter().collect())
2888 }
2889
2890 pub fn newline(&mut self, _: &Newline, cx: &mut ViewContext<Self>) {
2891 self.transact(cx, |this, cx| {
2892 let (edits, selection_fixup_info): (Vec<_>, Vec<_>) = {
2893 let selections = this.selections.all::<usize>(cx);
2894 let multi_buffer = this.buffer.read(cx);
2895 let buffer = multi_buffer.snapshot(cx);
2896 selections
2897 .iter()
2898 .map(|selection| {
2899 let start_point = selection.start.to_point(&buffer);
2900 let mut indent = buffer.indent_size_for_line(start_point.row);
2901 indent.len = cmp::min(indent.len, start_point.column);
2902 let start = selection.start;
2903 let end = selection.end;
2904 let selection_is_empty = start == end;
2905 let language_scope = buffer.language_scope_at(start);
2906 let (comment_delimiter, insert_extra_newline) = if let Some(language) =
2907 &language_scope
2908 {
2909 let leading_whitespace_len = buffer
2910 .reversed_chars_at(start)
2911 .take_while(|c| c.is_whitespace() && *c != '\n')
2912 .map(|c| c.len_utf8())
2913 .sum::<usize>();
2914
2915 let trailing_whitespace_len = buffer
2916 .chars_at(end)
2917 .take_while(|c| c.is_whitespace() && *c != '\n')
2918 .map(|c| c.len_utf8())
2919 .sum::<usize>();
2920
2921 let insert_extra_newline =
2922 language.brackets().any(|(pair, enabled)| {
2923 let pair_start = pair.start.trim_end();
2924 let pair_end = pair.end.trim_start();
2925
2926 enabled
2927 && pair.newline
2928 && buffer.contains_str_at(
2929 end + trailing_whitespace_len,
2930 pair_end,
2931 )
2932 && buffer.contains_str_at(
2933 (start - leading_whitespace_len)
2934 .saturating_sub(pair_start.len()),
2935 pair_start,
2936 )
2937 });
2938
2939 // Comment extension on newline is allowed only for cursor selections
2940 let comment_delimiter = maybe!({
2941 if !selection_is_empty {
2942 return None;
2943 }
2944
2945 if !multi_buffer.settings_at(0, cx).extend_comment_on_newline {
2946 return None;
2947 }
2948
2949 let delimiters = language.line_comment_prefixes();
2950 let max_len_of_delimiter =
2951 delimiters.iter().map(|delimiter| delimiter.len()).max()?;
2952 let (snapshot, range) =
2953 buffer.buffer_line_for_row(start_point.row)?;
2954
2955 let mut index_of_first_non_whitespace = 0;
2956 let comment_candidate = snapshot
2957 .chars_for_range(range)
2958 .skip_while(|c| {
2959 let should_skip = c.is_whitespace();
2960 if should_skip {
2961 index_of_first_non_whitespace += 1;
2962 }
2963 should_skip
2964 })
2965 .take(max_len_of_delimiter)
2966 .collect::<String>();
2967 let comment_prefix = delimiters.iter().find(|comment_prefix| {
2968 comment_candidate.starts_with(comment_prefix.as_ref())
2969 })?;
2970 let cursor_is_placed_after_comment_marker =
2971 index_of_first_non_whitespace + comment_prefix.len()
2972 <= start_point.column as usize;
2973 if cursor_is_placed_after_comment_marker {
2974 Some(comment_prefix.clone())
2975 } else {
2976 None
2977 }
2978 });
2979 (comment_delimiter, insert_extra_newline)
2980 } else {
2981 (None, false)
2982 };
2983
2984 let capacity_for_delimiter = comment_delimiter
2985 .as_deref()
2986 .map(str::len)
2987 .unwrap_or_default();
2988 let mut new_text =
2989 String::with_capacity(1 + capacity_for_delimiter + indent.len as usize);
2990 new_text.push_str("\n");
2991 new_text.extend(indent.chars());
2992 if let Some(delimiter) = &comment_delimiter {
2993 new_text.push_str(&delimiter);
2994 }
2995 if insert_extra_newline {
2996 new_text = new_text.repeat(2);
2997 }
2998
2999 let anchor = buffer.anchor_after(end);
3000 let new_selection = selection.map(|_| anchor);
3001 (
3002 (start..end, new_text),
3003 (insert_extra_newline, new_selection),
3004 )
3005 })
3006 .unzip()
3007 };
3008
3009 this.edit_with_autoindent(edits, cx);
3010 let buffer = this.buffer.read(cx).snapshot(cx);
3011 let new_selections = selection_fixup_info
3012 .into_iter()
3013 .map(|(extra_newline_inserted, new_selection)| {
3014 let mut cursor = new_selection.end.to_point(&buffer);
3015 if extra_newline_inserted {
3016 cursor.row -= 1;
3017 cursor.column = buffer.line_len(cursor.row);
3018 }
3019 new_selection.map(|_| cursor)
3020 })
3021 .collect();
3022
3023 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(new_selections));
3024 this.refresh_inline_completion(true, cx);
3025 });
3026 }
3027
3028 pub fn newline_above(&mut self, _: &NewlineAbove, cx: &mut ViewContext<Self>) {
3029 let buffer = self.buffer.read(cx);
3030 let snapshot = buffer.snapshot(cx);
3031
3032 let mut edits = Vec::new();
3033 let mut rows = Vec::new();
3034
3035 for (rows_inserted, selection) in self.selections.all_adjusted(cx).into_iter().enumerate() {
3036 let cursor = selection.head();
3037 let row = cursor.row;
3038
3039 let start_of_line = snapshot.clip_point(Point::new(row, 0), Bias::Left);
3040
3041 let newline = "\n".to_string();
3042 edits.push((start_of_line..start_of_line, newline));
3043
3044 rows.push(row + rows_inserted as u32);
3045 }
3046
3047 self.transact(cx, |editor, cx| {
3048 editor.edit(edits, cx);
3049
3050 editor.change_selections(Some(Autoscroll::fit()), cx, |s| {
3051 let mut index = 0;
3052 s.move_cursors_with(|map, _, _| {
3053 let row = rows[index];
3054 index += 1;
3055
3056 let point = Point::new(row, 0);
3057 let boundary = map.next_line_boundary(point).1;
3058 let clipped = map.clip_point(boundary, Bias::Left);
3059
3060 (clipped, SelectionGoal::None)
3061 });
3062 });
3063
3064 let mut indent_edits = Vec::new();
3065 let multibuffer_snapshot = editor.buffer.read(cx).snapshot(cx);
3066 for row in rows {
3067 let indents = multibuffer_snapshot.suggested_indents(row..row + 1, cx);
3068 for (row, indent) in indents {
3069 if indent.len == 0 {
3070 continue;
3071 }
3072
3073 let text = match indent.kind {
3074 IndentKind::Space => " ".repeat(indent.len as usize),
3075 IndentKind::Tab => "\t".repeat(indent.len as usize),
3076 };
3077 let point = Point::new(row, 0);
3078 indent_edits.push((point..point, text));
3079 }
3080 }
3081 editor.edit(indent_edits, cx);
3082 });
3083 }
3084
3085 pub fn newline_below(&mut self, _: &NewlineBelow, cx: &mut ViewContext<Self>) {
3086 let buffer = self.buffer.read(cx);
3087 let snapshot = buffer.snapshot(cx);
3088
3089 let mut edits = Vec::new();
3090 let mut rows = Vec::new();
3091 let mut rows_inserted = 0;
3092
3093 for selection in self.selections.all_adjusted(cx) {
3094 let cursor = selection.head();
3095 let row = cursor.row;
3096
3097 let point = Point::new(row + 1, 0);
3098 let start_of_line = snapshot.clip_point(point, Bias::Left);
3099
3100 let newline = "\n".to_string();
3101 edits.push((start_of_line..start_of_line, newline));
3102
3103 rows_inserted += 1;
3104 rows.push(row + rows_inserted);
3105 }
3106
3107 self.transact(cx, |editor, cx| {
3108 editor.edit(edits, cx);
3109
3110 editor.change_selections(Some(Autoscroll::fit()), cx, |s| {
3111 let mut index = 0;
3112 s.move_cursors_with(|map, _, _| {
3113 let row = rows[index];
3114 index += 1;
3115
3116 let point = Point::new(row, 0);
3117 let boundary = map.next_line_boundary(point).1;
3118 let clipped = map.clip_point(boundary, Bias::Left);
3119
3120 (clipped, SelectionGoal::None)
3121 });
3122 });
3123
3124 let mut indent_edits = Vec::new();
3125 let multibuffer_snapshot = editor.buffer.read(cx).snapshot(cx);
3126 for row in rows {
3127 let indents = multibuffer_snapshot.suggested_indents(row..row + 1, cx);
3128 for (row, indent) in indents {
3129 if indent.len == 0 {
3130 continue;
3131 }
3132
3133 let text = match indent.kind {
3134 IndentKind::Space => " ".repeat(indent.len as usize),
3135 IndentKind::Tab => "\t".repeat(indent.len as usize),
3136 };
3137 let point = Point::new(row, 0);
3138 indent_edits.push((point..point, text));
3139 }
3140 }
3141 editor.edit(indent_edits, cx);
3142 });
3143 }
3144
3145 pub fn insert(&mut self, text: &str, cx: &mut ViewContext<Self>) {
3146 let autoindent = text.is_empty().not().then(|| AutoindentMode::Block {
3147 original_indent_columns: Vec::new(),
3148 });
3149 self.insert_with_autoindent_mode(text, autoindent, cx);
3150 }
3151
3152 fn insert_with_autoindent_mode(
3153 &mut self,
3154 text: &str,
3155 autoindent_mode: Option<AutoindentMode>,
3156 cx: &mut ViewContext<Self>,
3157 ) {
3158 if self.read_only(cx) {
3159 return;
3160 }
3161
3162 let text: Arc<str> = text.into();
3163 self.transact(cx, |this, cx| {
3164 let old_selections = this.selections.all_adjusted(cx);
3165 let selection_anchors = this.buffer.update(cx, |buffer, cx| {
3166 let anchors = {
3167 let snapshot = buffer.read(cx);
3168 old_selections
3169 .iter()
3170 .map(|s| {
3171 let anchor = snapshot.anchor_after(s.head());
3172 s.map(|_| anchor)
3173 })
3174 .collect::<Vec<_>>()
3175 };
3176 buffer.edit(
3177 old_selections
3178 .iter()
3179 .map(|s| (s.start..s.end, text.clone())),
3180 autoindent_mode,
3181 cx,
3182 );
3183 anchors
3184 });
3185
3186 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
3187 s.select_anchors(selection_anchors);
3188 })
3189 });
3190 }
3191
3192 fn trigger_completion_on_input(
3193 &mut self,
3194 text: &str,
3195 trigger_in_words: bool,
3196 cx: &mut ViewContext<Self>,
3197 ) {
3198 if !EditorSettings::get_global(cx).show_completions_on_input {
3199 return;
3200 }
3201
3202 let selection = self.selections.newest_anchor();
3203 if self
3204 .buffer
3205 .read(cx)
3206 .is_completion_trigger(selection.head(), text, trigger_in_words, cx)
3207 {
3208 self.show_completions(&ShowCompletions, cx);
3209 } else {
3210 self.hide_context_menu(cx);
3211 }
3212 }
3213
3214 /// If any empty selections is touching the start of its innermost containing autoclose
3215 /// region, expand it to select the brackets.
3216 fn select_autoclose_pair(&mut self, cx: &mut ViewContext<Self>) {
3217 let selections = self.selections.all::<usize>(cx);
3218 let buffer = self.buffer.read(cx).read(cx);
3219 let new_selections = self
3220 .selections_with_autoclose_regions(selections, &buffer)
3221 .map(|(mut selection, region)| {
3222 if !selection.is_empty() {
3223 return selection;
3224 }
3225
3226 if let Some(region) = region {
3227 let mut range = region.range.to_offset(&buffer);
3228 if selection.start == range.start && range.start >= region.pair.start.len() {
3229 range.start -= region.pair.start.len();
3230 if buffer.contains_str_at(range.start, ®ion.pair.start)
3231 && buffer.contains_str_at(range.end, ®ion.pair.end)
3232 {
3233 range.end += region.pair.end.len();
3234 selection.start = range.start;
3235 selection.end = range.end;
3236
3237 return selection;
3238 }
3239 }
3240 }
3241
3242 let always_treat_brackets_as_autoclosed = buffer
3243 .settings_at(selection.start, cx)
3244 .always_treat_brackets_as_autoclosed;
3245
3246 if !always_treat_brackets_as_autoclosed {
3247 return selection;
3248 }
3249
3250 if let Some(scope) = buffer.language_scope_at(selection.start) {
3251 for (pair, enabled) in scope.brackets() {
3252 if !enabled || !pair.close {
3253 continue;
3254 }
3255
3256 if buffer.contains_str_at(selection.start, &pair.end) {
3257 let pair_start_len = pair.start.len();
3258 if buffer.contains_str_at(selection.start - pair_start_len, &pair.start)
3259 {
3260 selection.start -= pair_start_len;
3261 selection.end += pair.end.len();
3262
3263 return selection;
3264 }
3265 }
3266 }
3267 }
3268
3269 selection
3270 })
3271 .collect();
3272
3273 drop(buffer);
3274 self.change_selections(None, cx, |selections| selections.select(new_selections));
3275 }
3276
3277 /// Iterate the given selections, and for each one, find the smallest surrounding
3278 /// autoclose region. This uses the ordering of the selections and the autoclose
3279 /// regions to avoid repeated comparisons.
3280 fn selections_with_autoclose_regions<'a, D: ToOffset + Clone>(
3281 &'a self,
3282 selections: impl IntoIterator<Item = Selection<D>>,
3283 buffer: &'a MultiBufferSnapshot,
3284 ) -> impl Iterator<Item = (Selection<D>, Option<&'a AutocloseRegion>)> {
3285 let mut i = 0;
3286 let mut regions = self.autoclose_regions.as_slice();
3287 selections.into_iter().map(move |selection| {
3288 let range = selection.start.to_offset(buffer)..selection.end.to_offset(buffer);
3289
3290 let mut enclosing = None;
3291 while let Some(pair_state) = regions.get(i) {
3292 if pair_state.range.end.to_offset(buffer) < range.start {
3293 regions = ®ions[i + 1..];
3294 i = 0;
3295 } else if pair_state.range.start.to_offset(buffer) > range.end {
3296 break;
3297 } else {
3298 if pair_state.selection_id == selection.id {
3299 enclosing = Some(pair_state);
3300 }
3301 i += 1;
3302 }
3303 }
3304
3305 (selection.clone(), enclosing)
3306 })
3307 }
3308
3309 /// Remove any autoclose regions that no longer contain their selection.
3310 fn invalidate_autoclose_regions(
3311 &mut self,
3312 mut selections: &[Selection<Anchor>],
3313 buffer: &MultiBufferSnapshot,
3314 ) {
3315 self.autoclose_regions.retain(|state| {
3316 let mut i = 0;
3317 while let Some(selection) = selections.get(i) {
3318 if selection.end.cmp(&state.range.start, buffer).is_lt() {
3319 selections = &selections[1..];
3320 continue;
3321 }
3322 if selection.start.cmp(&state.range.end, buffer).is_gt() {
3323 break;
3324 }
3325 if selection.id == state.selection_id {
3326 return true;
3327 } else {
3328 i += 1;
3329 }
3330 }
3331 false
3332 });
3333 }
3334
3335 fn completion_query(buffer: &MultiBufferSnapshot, position: impl ToOffset) -> Option<String> {
3336 let offset = position.to_offset(buffer);
3337 let (word_range, kind) = buffer.surrounding_word(offset);
3338 if offset > word_range.start && kind == Some(CharKind::Word) {
3339 Some(
3340 buffer
3341 .text_for_range(word_range.start..offset)
3342 .collect::<String>(),
3343 )
3344 } else {
3345 None
3346 }
3347 }
3348
3349 pub fn toggle_inlay_hints(&mut self, _: &ToggleInlayHints, cx: &mut ViewContext<Self>) {
3350 self.refresh_inlay_hints(
3351 InlayHintRefreshReason::Toggle(!self.inlay_hint_cache.enabled),
3352 cx,
3353 );
3354 }
3355
3356 pub fn inlay_hints_enabled(&self) -> bool {
3357 self.inlay_hint_cache.enabled
3358 }
3359
3360 fn refresh_inlay_hints(&mut self, reason: InlayHintRefreshReason, cx: &mut ViewContext<Self>) {
3361 if self.project.is_none() || self.mode != EditorMode::Full {
3362 return;
3363 }
3364
3365 let reason_description = reason.description();
3366 let ignore_debounce = matches!(
3367 reason,
3368 InlayHintRefreshReason::SettingsChange(_)
3369 | InlayHintRefreshReason::Toggle(_)
3370 | InlayHintRefreshReason::ExcerptsRemoved(_)
3371 );
3372 let (invalidate_cache, required_languages) = match reason {
3373 InlayHintRefreshReason::Toggle(enabled) => {
3374 self.inlay_hint_cache.enabled = enabled;
3375 if enabled {
3376 (InvalidationStrategy::RefreshRequested, None)
3377 } else {
3378 self.inlay_hint_cache.clear();
3379 self.splice_inlays(
3380 self.visible_inlay_hints(cx)
3381 .iter()
3382 .map(|inlay| inlay.id)
3383 .collect(),
3384 Vec::new(),
3385 cx,
3386 );
3387 return;
3388 }
3389 }
3390 InlayHintRefreshReason::SettingsChange(new_settings) => {
3391 match self.inlay_hint_cache.update_settings(
3392 &self.buffer,
3393 new_settings,
3394 self.visible_inlay_hints(cx),
3395 cx,
3396 ) {
3397 ControlFlow::Break(Some(InlaySplice {
3398 to_remove,
3399 to_insert,
3400 })) => {
3401 self.splice_inlays(to_remove, to_insert, cx);
3402 return;
3403 }
3404 ControlFlow::Break(None) => return,
3405 ControlFlow::Continue(()) => (InvalidationStrategy::RefreshRequested, None),
3406 }
3407 }
3408 InlayHintRefreshReason::ExcerptsRemoved(excerpts_removed) => {
3409 if let Some(InlaySplice {
3410 to_remove,
3411 to_insert,
3412 }) = self.inlay_hint_cache.remove_excerpts(excerpts_removed)
3413 {
3414 self.splice_inlays(to_remove, to_insert, cx);
3415 }
3416 return;
3417 }
3418 InlayHintRefreshReason::NewLinesShown => (InvalidationStrategy::None, None),
3419 InlayHintRefreshReason::BufferEdited(buffer_languages) => {
3420 (InvalidationStrategy::BufferEdited, Some(buffer_languages))
3421 }
3422 InlayHintRefreshReason::RefreshRequested => {
3423 (InvalidationStrategy::RefreshRequested, None)
3424 }
3425 };
3426
3427 if let Some(InlaySplice {
3428 to_remove,
3429 to_insert,
3430 }) = self.inlay_hint_cache.spawn_hint_refresh(
3431 reason_description,
3432 self.excerpts_for_inlay_hints_query(required_languages.as_ref(), cx),
3433 invalidate_cache,
3434 ignore_debounce,
3435 cx,
3436 ) {
3437 self.splice_inlays(to_remove, to_insert, cx);
3438 }
3439 }
3440
3441 fn visible_inlay_hints(&self, cx: &ViewContext<'_, Editor>) -> Vec<Inlay> {
3442 self.display_map
3443 .read(cx)
3444 .current_inlays()
3445 .filter(move |inlay| matches!(inlay.id, InlayId::Hint(_)))
3446 .cloned()
3447 .collect()
3448 }
3449
3450 pub fn excerpts_for_inlay_hints_query(
3451 &self,
3452 restrict_to_languages: Option<&HashSet<Arc<Language>>>,
3453 cx: &mut ViewContext<Editor>,
3454 ) -> HashMap<ExcerptId, (Model<Buffer>, clock::Global, Range<usize>)> {
3455 let Some(project) = self.project.as_ref() else {
3456 return HashMap::default();
3457 };
3458 let project = project.read(cx);
3459 let multi_buffer = self.buffer().read(cx);
3460 let multi_buffer_snapshot = multi_buffer.snapshot(cx);
3461 let multi_buffer_visible_start = self
3462 .scroll_manager
3463 .anchor()
3464 .anchor
3465 .to_point(&multi_buffer_snapshot);
3466 let multi_buffer_visible_end = multi_buffer_snapshot.clip_point(
3467 multi_buffer_visible_start
3468 + Point::new(self.visible_line_count().unwrap_or(0.).ceil() as u32, 0),
3469 Bias::Left,
3470 );
3471 let multi_buffer_visible_range = multi_buffer_visible_start..multi_buffer_visible_end;
3472 multi_buffer
3473 .range_to_buffer_ranges(multi_buffer_visible_range, cx)
3474 .into_iter()
3475 .filter(|(_, excerpt_visible_range, _)| !excerpt_visible_range.is_empty())
3476 .filter_map(|(buffer_handle, excerpt_visible_range, excerpt_id)| {
3477 let buffer = buffer_handle.read(cx);
3478 let buffer_file = project::File::from_dyn(buffer.file())?;
3479 let buffer_worktree = project.worktree_for_id(buffer_file.worktree_id(cx), cx)?;
3480 let worktree_entry = buffer_worktree
3481 .read(cx)
3482 .entry_for_id(buffer_file.project_entry_id(cx)?)?;
3483 if worktree_entry.is_ignored {
3484 return None;
3485 }
3486
3487 let language = buffer.language()?;
3488 if let Some(restrict_to_languages) = restrict_to_languages {
3489 if !restrict_to_languages.contains(language) {
3490 return None;
3491 }
3492 }
3493 Some((
3494 excerpt_id,
3495 (
3496 buffer_handle,
3497 buffer.version().clone(),
3498 excerpt_visible_range,
3499 ),
3500 ))
3501 })
3502 .collect()
3503 }
3504
3505 pub fn text_layout_details(&self, cx: &WindowContext) -> TextLayoutDetails {
3506 TextLayoutDetails {
3507 text_system: cx.text_system().clone(),
3508 editor_style: self.style.clone().unwrap(),
3509 rem_size: cx.rem_size(),
3510 scroll_anchor: self.scroll_manager.anchor(),
3511 visible_rows: self.visible_line_count(),
3512 vertical_scroll_margin: self.scroll_manager.vertical_scroll_margin,
3513 }
3514 }
3515
3516 fn splice_inlays(
3517 &self,
3518 to_remove: Vec<InlayId>,
3519 to_insert: Vec<Inlay>,
3520 cx: &mut ViewContext<Self>,
3521 ) {
3522 self.display_map.update(cx, |display_map, cx| {
3523 display_map.splice_inlays(to_remove, to_insert, cx);
3524 });
3525 cx.notify();
3526 }
3527
3528 fn trigger_on_type_formatting(
3529 &self,
3530 input: String,
3531 cx: &mut ViewContext<Self>,
3532 ) -> Option<Task<Result<()>>> {
3533 if input.len() != 1 {
3534 return None;
3535 }
3536
3537 let project = self.project.as_ref()?;
3538 let position = self.selections.newest_anchor().head();
3539 let (buffer, buffer_position) = self
3540 .buffer
3541 .read(cx)
3542 .text_anchor_for_position(position, cx)?;
3543
3544 // OnTypeFormatting returns a list of edits, no need to pass them between Zed instances,
3545 // hence we do LSP request & edit on host side only — add formats to host's history.
3546 let push_to_lsp_host_history = true;
3547 // If this is not the host, append its history with new edits.
3548 let push_to_client_history = project.read(cx).is_remote();
3549
3550 let on_type_formatting = project.update(cx, |project, cx| {
3551 project.on_type_format(
3552 buffer.clone(),
3553 buffer_position,
3554 input,
3555 push_to_lsp_host_history,
3556 cx,
3557 )
3558 });
3559 Some(cx.spawn(|editor, mut cx| async move {
3560 if let Some(transaction) = on_type_formatting.await? {
3561 if push_to_client_history {
3562 buffer
3563 .update(&mut cx, |buffer, _| {
3564 buffer.push_transaction(transaction, Instant::now());
3565 })
3566 .ok();
3567 }
3568 editor.update(&mut cx, |editor, cx| {
3569 editor.refresh_document_highlights(cx);
3570 })?;
3571 }
3572 Ok(())
3573 }))
3574 }
3575
3576 fn show_completions(&mut self, _: &ShowCompletions, cx: &mut ViewContext<Self>) {
3577 if self.pending_rename.is_some() {
3578 return;
3579 }
3580
3581 let Some(provider) = self.completion_provider.as_ref() else {
3582 return;
3583 };
3584
3585 let position = self.selections.newest_anchor().head();
3586 let (buffer, buffer_position) =
3587 if let Some(output) = self.buffer.read(cx).text_anchor_for_position(position, cx) {
3588 output
3589 } else {
3590 return;
3591 };
3592
3593 let query = Self::completion_query(&self.buffer.read(cx).read(cx), position);
3594 let completions = provider.completions(&buffer, buffer_position, cx);
3595
3596 let id = post_inc(&mut self.next_completion_id);
3597 let task = cx.spawn(|this, mut cx| {
3598 async move {
3599 let completions = completions.await.log_err();
3600 let menu = if let Some(completions) = completions {
3601 let mut menu = CompletionsMenu {
3602 id,
3603 initial_position: position,
3604 match_candidates: completions
3605 .iter()
3606 .enumerate()
3607 .map(|(id, completion)| {
3608 StringMatchCandidate::new(
3609 id,
3610 completion.label.text[completion.label.filter_range.clone()]
3611 .into(),
3612 )
3613 })
3614 .collect(),
3615 buffer: buffer.clone(),
3616 completions: Arc::new(RwLock::new(completions.into())),
3617 matches: Vec::new().into(),
3618 selected_item: 0,
3619 scroll_handle: UniformListScrollHandle::new(),
3620 selected_completion_documentation_resolve_debounce: Arc::new(Mutex::new(
3621 DebouncedDelay::new(),
3622 )),
3623 };
3624 menu.filter(query.as_deref(), cx.background_executor().clone())
3625 .await;
3626
3627 if menu.matches.is_empty() {
3628 None
3629 } else {
3630 this.update(&mut cx, |editor, cx| {
3631 let completions = menu.completions.clone();
3632 let matches = menu.matches.clone();
3633
3634 let delay_ms = EditorSettings::get_global(cx)
3635 .completion_documentation_secondary_query_debounce;
3636 let delay = Duration::from_millis(delay_ms);
3637
3638 editor
3639 .completion_documentation_pre_resolve_debounce
3640 .fire_new(delay, cx, |editor, cx| {
3641 CompletionsMenu::pre_resolve_completion_documentation(
3642 buffer,
3643 completions,
3644 matches,
3645 editor,
3646 cx,
3647 )
3648 });
3649 })
3650 .ok();
3651 Some(menu)
3652 }
3653 } else {
3654 None
3655 };
3656
3657 this.update(&mut cx, |this, cx| {
3658 this.completion_tasks.retain(|(task_id, _)| *task_id >= id);
3659
3660 let mut context_menu = this.context_menu.write();
3661 match context_menu.as_ref() {
3662 None => {}
3663
3664 Some(ContextMenu::Completions(prev_menu)) => {
3665 if prev_menu.id > id {
3666 return;
3667 }
3668 }
3669
3670 _ => return,
3671 }
3672
3673 if this.focus_handle.is_focused(cx) && menu.is_some() {
3674 let menu = menu.unwrap();
3675 *context_menu = Some(ContextMenu::Completions(menu));
3676 drop(context_menu);
3677 this.discard_inline_completion(cx);
3678 cx.notify();
3679 } else if this.completion_tasks.len() <= 1 {
3680 // If there are no more completion tasks and the last menu was
3681 // empty, we should hide it. If it was already hidden, we should
3682 // also show the copilot completion when available.
3683 drop(context_menu);
3684 if this.hide_context_menu(cx).is_none() {
3685 this.update_visible_inline_completion(cx);
3686 }
3687 }
3688 })?;
3689
3690 Ok::<_, anyhow::Error>(())
3691 }
3692 .log_err()
3693 });
3694
3695 self.completion_tasks.push((id, task));
3696 }
3697
3698 pub fn confirm_completion(
3699 &mut self,
3700 action: &ConfirmCompletion,
3701 cx: &mut ViewContext<Self>,
3702 ) -> Option<Task<Result<()>>> {
3703 use language::ToOffset as _;
3704
3705 let completions_menu = if let ContextMenu::Completions(menu) = self.hide_context_menu(cx)? {
3706 menu
3707 } else {
3708 return None;
3709 };
3710
3711 let mat = completions_menu
3712 .matches
3713 .get(action.item_ix.unwrap_or(completions_menu.selected_item))?;
3714 let buffer_handle = completions_menu.buffer;
3715 let completions = completions_menu.completions.read();
3716 let completion = completions.get(mat.candidate_id)?;
3717 cx.stop_propagation();
3718
3719 let snippet;
3720 let text;
3721 if completion.is_snippet() {
3722 snippet = Some(Snippet::parse(&completion.new_text).log_err()?);
3723 text = snippet.as_ref().unwrap().text.clone();
3724 } else {
3725 snippet = None;
3726 text = completion.new_text.clone();
3727 };
3728 let selections = self.selections.all::<usize>(cx);
3729 let buffer = buffer_handle.read(cx);
3730 let old_range = completion.old_range.to_offset(buffer);
3731 let old_text = buffer.text_for_range(old_range.clone()).collect::<String>();
3732
3733 let newest_selection = self.selections.newest_anchor();
3734 if newest_selection.start.buffer_id != Some(buffer_handle.read(cx).remote_id()) {
3735 return None;
3736 }
3737
3738 let lookbehind = newest_selection
3739 .start
3740 .text_anchor
3741 .to_offset(buffer)
3742 .saturating_sub(old_range.start);
3743 let lookahead = old_range
3744 .end
3745 .saturating_sub(newest_selection.end.text_anchor.to_offset(buffer));
3746 let mut common_prefix_len = old_text
3747 .bytes()
3748 .zip(text.bytes())
3749 .take_while(|(a, b)| a == b)
3750 .count();
3751
3752 let snapshot = self.buffer.read(cx).snapshot(cx);
3753 let mut range_to_replace: Option<Range<isize>> = None;
3754 let mut ranges = Vec::new();
3755 for selection in &selections {
3756 if snapshot.contains_str_at(selection.start.saturating_sub(lookbehind), &old_text) {
3757 let start = selection.start.saturating_sub(lookbehind);
3758 let end = selection.end + lookahead;
3759 if selection.id == newest_selection.id {
3760 range_to_replace = Some(
3761 ((start + common_prefix_len) as isize - selection.start as isize)
3762 ..(end as isize - selection.start as isize),
3763 );
3764 }
3765 ranges.push(start + common_prefix_len..end);
3766 } else {
3767 common_prefix_len = 0;
3768 ranges.clear();
3769 ranges.extend(selections.iter().map(|s| {
3770 if s.id == newest_selection.id {
3771 range_to_replace = Some(
3772 old_range.start.to_offset_utf16(&snapshot).0 as isize
3773 - selection.start as isize
3774 ..old_range.end.to_offset_utf16(&snapshot).0 as isize
3775 - selection.start as isize,
3776 );
3777 old_range.clone()
3778 } else {
3779 s.start..s.end
3780 }
3781 }));
3782 break;
3783 }
3784 }
3785 let text = &text[common_prefix_len..];
3786
3787 cx.emit(EditorEvent::InputHandled {
3788 utf16_range_to_replace: range_to_replace,
3789 text: text.into(),
3790 });
3791
3792 self.transact(cx, |this, cx| {
3793 if let Some(mut snippet) = snippet {
3794 snippet.text = text.to_string();
3795 for tabstop in snippet.tabstops.iter_mut().flatten() {
3796 tabstop.start -= common_prefix_len as isize;
3797 tabstop.end -= common_prefix_len as isize;
3798 }
3799
3800 this.insert_snippet(&ranges, snippet, cx).log_err();
3801 } else {
3802 this.buffer.update(cx, |buffer, cx| {
3803 buffer.edit(
3804 ranges.iter().map(|range| (range.clone(), text)),
3805 this.autoindent_mode.clone(),
3806 cx,
3807 );
3808 });
3809 }
3810
3811 this.refresh_inline_completion(true, cx);
3812 });
3813
3814 let provider = self.completion_provider.as_ref()?;
3815 let apply_edits = provider.apply_additional_edits_for_completion(
3816 buffer_handle,
3817 completion.clone(),
3818 true,
3819 cx,
3820 );
3821 Some(cx.foreground_executor().spawn(async move {
3822 apply_edits.await?;
3823 Ok(())
3824 }))
3825 }
3826
3827 pub fn toggle_code_actions(&mut self, action: &ToggleCodeActions, cx: &mut ViewContext<Self>) {
3828 let mut context_menu = self.context_menu.write();
3829 if let Some(ContextMenu::CodeActions(code_actions)) = context_menu.as_ref() {
3830 if code_actions.deployed_from_indicator == action.deployed_from_indicator {
3831 // Toggle if we're selecting the same one
3832 *context_menu = None;
3833 cx.notify();
3834 return;
3835 } else {
3836 // Otherwise, clear it and start a new one
3837 *context_menu = None;
3838 cx.notify();
3839 }
3840 }
3841 drop(context_menu);
3842
3843 let deployed_from_indicator = action.deployed_from_indicator;
3844 let mut task = self.code_actions_task.take();
3845 let action = action.clone();
3846 cx.spawn(|this, mut cx| async move {
3847 while let Some(prev_task) = task {
3848 prev_task.await;
3849 task = this.update(&mut cx, |this, _| this.code_actions_task.take())?;
3850 }
3851
3852 let spawned_test_task = this.update(&mut cx, |this, cx| {
3853 if this.focus_handle.is_focused(cx) {
3854 let snapshot = this.snapshot(cx);
3855 let display_row = action.deployed_from_indicator.unwrap_or_else(|| {
3856 this.selections
3857 .newest::<Point>(cx)
3858 .head()
3859 .to_display_point(&snapshot.display_snapshot)
3860 .row()
3861 });
3862
3863 let buffer_point =
3864 DisplayPoint::new(display_row, 0).to_point(&snapshot.display_snapshot);
3865 let buffer_row = snapshot
3866 .buffer_snapshot
3867 .buffer_line_for_row(buffer_point.row)
3868 .map(|(_, Range { start, .. })| start);
3869 let tasks = this.tasks.get(&display_row).map(|t| Arc::new(t.to_owned()));
3870 let (buffer, code_actions) = this.available_code_actions.clone().unzip();
3871 if tasks.is_none() && code_actions.is_none() {
3872 return None;
3873 }
3874 let buffer = buffer.or_else(|| {
3875 let snapshot = this.snapshot(cx);
3876 let (buffer_snapshot, _) =
3877 snapshot.buffer_snapshot.buffer_line_for_row(display_row)?;
3878 let buffer_id = buffer_snapshot.remote_id();
3879 this.buffer().read(cx).buffer(buffer_id)
3880 });
3881 let Some(buffer) = buffer else {
3882 return None;
3883 };
3884 this.completion_tasks.clear();
3885 this.discard_inline_completion(cx);
3886 let task_context = tasks.as_ref().zip(this.workspace.clone()).and_then(
3887 |(tasks, (workspace, _))| {
3888 if let Some(buffer_point) = buffer_row {
3889 let position = Point::new(buffer_point.row, tasks.column);
3890 let range_start = buffer.read(cx).anchor_at(position, Bias::Right);
3891 let location = Location {
3892 buffer: buffer.clone(),
3893 range: range_start..range_start,
3894 };
3895 workspace
3896 .update(cx, |workspace, cx| {
3897 tasks::task_context_for_location(workspace, location, cx)
3898 })
3899 .ok()
3900 .flatten()
3901 } else {
3902 None
3903 }
3904 },
3905 );
3906 let tasks = tasks
3907 .zip(task_context.as_ref())
3908 .map(|(tasks, task_context)| {
3909 Arc::new(ResolvedTasks {
3910 templates: tasks
3911 .templates
3912 .iter()
3913 .filter_map(|(kind, template)| {
3914 template
3915 .resolve_task(&kind.to_id_base(), &task_context)
3916 .map(|task| (kind.clone(), task))
3917 })
3918 .collect(),
3919 position: Point::new(display_row, tasks.column),
3920 })
3921 });
3922 let spawn_straight_away = tasks
3923 .as_ref()
3924 .map_or(false, |tasks| tasks.templates.len() == 1)
3925 && code_actions
3926 .as_ref()
3927 .map_or(true, |actions| actions.is_empty());
3928 *this.context_menu.write() = Some(ContextMenu::CodeActions(CodeActionsMenu {
3929 buffer,
3930 actions: CodeActionContents {
3931 tasks,
3932 actions: code_actions,
3933 },
3934 selected_item: Default::default(),
3935 scroll_handle: UniformListScrollHandle::default(),
3936 deployed_from_indicator,
3937 }));
3938 if spawn_straight_away {
3939 if let Some(task) =
3940 this.confirm_code_action(&ConfirmCodeAction { item_ix: Some(0) }, cx)
3941 {
3942 cx.notify();
3943 return Some(task);
3944 }
3945 }
3946 cx.notify();
3947 }
3948 Some(Task::ready(Ok(())))
3949 })?;
3950 if let Some(task) = spawned_test_task {
3951 task.await?;
3952 }
3953
3954 Ok::<_, anyhow::Error>(())
3955 })
3956 .detach_and_log_err(cx);
3957 }
3958
3959 pub fn confirm_code_action(
3960 &mut self,
3961 action: &ConfirmCodeAction,
3962 cx: &mut ViewContext<Self>,
3963 ) -> Option<Task<Result<()>>> {
3964 let actions_menu = if let ContextMenu::CodeActions(menu) = self.hide_context_menu(cx)? {
3965 menu
3966 } else {
3967 return None;
3968 };
3969 let action_ix = action.item_ix.unwrap_or(actions_menu.selected_item);
3970 let action = actions_menu.actions.get(action_ix)?;
3971 let title = action.label();
3972 let buffer = actions_menu.buffer;
3973 let workspace = self.workspace()?;
3974
3975 match action {
3976 CodeActionsItem::Task(task_source_kind, resolved_task) => {
3977 workspace.update(cx, |workspace, cx| {
3978 workspace::tasks::schedule_resolved_task(
3979 workspace,
3980 task_source_kind,
3981 resolved_task,
3982 false,
3983 cx,
3984 );
3985
3986 None
3987 })
3988 }
3989 CodeActionsItem::CodeAction(action) => {
3990 let apply_code_actions = workspace
3991 .read(cx)
3992 .project()
3993 .clone()
3994 .update(cx, |project, cx| {
3995 project.apply_code_action(buffer, action, true, cx)
3996 });
3997 let workspace = workspace.downgrade();
3998 Some(cx.spawn(|editor, cx| async move {
3999 let project_transaction = apply_code_actions.await?;
4000 Self::open_project_transaction(
4001 &editor,
4002 workspace,
4003 project_transaction,
4004 title,
4005 cx,
4006 )
4007 .await
4008 }))
4009 }
4010 }
4011 }
4012
4013 async fn open_project_transaction(
4014 this: &WeakView<Editor>,
4015 workspace: WeakView<Workspace>,
4016 transaction: ProjectTransaction,
4017 title: String,
4018 mut cx: AsyncWindowContext,
4019 ) -> Result<()> {
4020 let replica_id = this.update(&mut cx, |this, cx| this.replica_id(cx))?;
4021
4022 let mut entries = transaction.0.into_iter().collect::<Vec<_>>();
4023 cx.update(|cx| {
4024 entries.sort_unstable_by_key(|(buffer, _)| {
4025 buffer.read(cx).file().map(|f| f.path().clone())
4026 });
4027 })?;
4028
4029 // If the project transaction's edits are all contained within this editor, then
4030 // avoid opening a new editor to display them.
4031
4032 if let Some((buffer, transaction)) = entries.first() {
4033 if entries.len() == 1 {
4034 let excerpt = this.update(&mut cx, |editor, cx| {
4035 editor
4036 .buffer()
4037 .read(cx)
4038 .excerpt_containing(editor.selections.newest_anchor().head(), cx)
4039 })?;
4040 if let Some((_, excerpted_buffer, excerpt_range)) = excerpt {
4041 if excerpted_buffer == *buffer {
4042 let all_edits_within_excerpt = buffer.read_with(&cx, |buffer, _| {
4043 let excerpt_range = excerpt_range.to_offset(buffer);
4044 buffer
4045 .edited_ranges_for_transaction::<usize>(transaction)
4046 .all(|range| {
4047 excerpt_range.start <= range.start
4048 && excerpt_range.end >= range.end
4049 })
4050 })?;
4051
4052 if all_edits_within_excerpt {
4053 return Ok(());
4054 }
4055 }
4056 }
4057 }
4058 } else {
4059 return Ok(());
4060 }
4061
4062 let mut ranges_to_highlight = Vec::new();
4063 let excerpt_buffer = cx.new_model(|cx| {
4064 let mut multibuffer =
4065 MultiBuffer::new(replica_id, Capability::ReadWrite).with_title(title);
4066 for (buffer_handle, transaction) in &entries {
4067 let buffer = buffer_handle.read(cx);
4068 ranges_to_highlight.extend(
4069 multibuffer.push_excerpts_with_context_lines(
4070 buffer_handle.clone(),
4071 buffer
4072 .edited_ranges_for_transaction::<usize>(transaction)
4073 .collect(),
4074 DEFAULT_MULTIBUFFER_CONTEXT,
4075 cx,
4076 ),
4077 );
4078 }
4079 multibuffer.push_transaction(entries.iter().map(|(b, t)| (b, t)), cx);
4080 multibuffer
4081 })?;
4082
4083 workspace.update(&mut cx, |workspace, cx| {
4084 let project = workspace.project().clone();
4085 let editor =
4086 cx.new_view(|cx| Editor::for_multibuffer(excerpt_buffer, Some(project), cx));
4087 workspace.add_item_to_active_pane(Box::new(editor.clone()), None, cx);
4088 editor.update(cx, |editor, cx| {
4089 editor.highlight_background::<Self>(
4090 &ranges_to_highlight,
4091 |theme| theme.editor_highlighted_line_background,
4092 cx,
4093 );
4094 });
4095 })?;
4096
4097 Ok(())
4098 }
4099
4100 fn refresh_code_actions(&mut self, cx: &mut ViewContext<Self>) -> Option<()> {
4101 let project = self.project.clone()?;
4102 let buffer = self.buffer.read(cx);
4103 let newest_selection = self.selections.newest_anchor().clone();
4104 let (start_buffer, start) = buffer.text_anchor_for_position(newest_selection.start, cx)?;
4105 let (end_buffer, end) = buffer.text_anchor_for_position(newest_selection.end, cx)?;
4106 if start_buffer != end_buffer {
4107 return None;
4108 }
4109
4110 self.code_actions_task = Some(cx.spawn(|this, mut cx| async move {
4111 cx.background_executor()
4112 .timer(CODE_ACTIONS_DEBOUNCE_TIMEOUT)
4113 .await;
4114
4115 let actions = if let Ok(code_actions) = project.update(&mut cx, |project, cx| {
4116 project.code_actions(&start_buffer, start..end, cx)
4117 }) {
4118 code_actions.await
4119 } else {
4120 Vec::new()
4121 };
4122
4123 this.update(&mut cx, |this, cx| {
4124 this.available_code_actions = if actions.is_empty() {
4125 None
4126 } else {
4127 Some((start_buffer, actions.into()))
4128 };
4129 cx.notify();
4130 })
4131 .log_err();
4132 }));
4133 None
4134 }
4135
4136 fn start_inline_blame_timer(&mut self, cx: &mut ViewContext<Self>) {
4137 if let Some(delay) = ProjectSettings::get_global(cx).git.inline_blame_delay() {
4138 self.show_git_blame_inline = false;
4139
4140 self.show_git_blame_inline_delay_task = Some(cx.spawn(|this, mut cx| async move {
4141 cx.background_executor().timer(delay).await;
4142
4143 this.update(&mut cx, |this, cx| {
4144 this.show_git_blame_inline = true;
4145 cx.notify();
4146 })
4147 .log_err();
4148 }));
4149 }
4150 }
4151
4152 fn refresh_document_highlights(&mut self, cx: &mut ViewContext<Self>) -> Option<()> {
4153 if self.pending_rename.is_some() {
4154 return None;
4155 }
4156
4157 let project = self.project.clone()?;
4158 let buffer = self.buffer.read(cx);
4159 let newest_selection = self.selections.newest_anchor().clone();
4160 let cursor_position = newest_selection.head();
4161 let (cursor_buffer, cursor_buffer_position) =
4162 buffer.text_anchor_for_position(cursor_position, cx)?;
4163 let (tail_buffer, _) = buffer.text_anchor_for_position(newest_selection.tail(), cx)?;
4164 if cursor_buffer != tail_buffer {
4165 return None;
4166 }
4167
4168 self.document_highlights_task = Some(cx.spawn(|this, mut cx| async move {
4169 cx.background_executor()
4170 .timer(DOCUMENT_HIGHLIGHTS_DEBOUNCE_TIMEOUT)
4171 .await;
4172
4173 let highlights = if let Some(highlights) = project
4174 .update(&mut cx, |project, cx| {
4175 project.document_highlights(&cursor_buffer, cursor_buffer_position, cx)
4176 })
4177 .log_err()
4178 {
4179 highlights.await.log_err()
4180 } else {
4181 None
4182 };
4183
4184 if let Some(highlights) = highlights {
4185 this.update(&mut cx, |this, cx| {
4186 if this.pending_rename.is_some() {
4187 return;
4188 }
4189
4190 let buffer_id = cursor_position.buffer_id;
4191 let buffer = this.buffer.read(cx);
4192 if !buffer
4193 .text_anchor_for_position(cursor_position, cx)
4194 .map_or(false, |(buffer, _)| buffer == cursor_buffer)
4195 {
4196 return;
4197 }
4198
4199 let cursor_buffer_snapshot = cursor_buffer.read(cx);
4200 let mut write_ranges = Vec::new();
4201 let mut read_ranges = Vec::new();
4202 for highlight in highlights {
4203 for (excerpt_id, excerpt_range) in
4204 buffer.excerpts_for_buffer(&cursor_buffer, cx)
4205 {
4206 let start = highlight
4207 .range
4208 .start
4209 .max(&excerpt_range.context.start, cursor_buffer_snapshot);
4210 let end = highlight
4211 .range
4212 .end
4213 .min(&excerpt_range.context.end, cursor_buffer_snapshot);
4214 if start.cmp(&end, cursor_buffer_snapshot).is_ge() {
4215 continue;
4216 }
4217
4218 let range = Anchor {
4219 buffer_id,
4220 excerpt_id: excerpt_id,
4221 text_anchor: start,
4222 }..Anchor {
4223 buffer_id,
4224 excerpt_id,
4225 text_anchor: end,
4226 };
4227 if highlight.kind == lsp::DocumentHighlightKind::WRITE {
4228 write_ranges.push(range);
4229 } else {
4230 read_ranges.push(range);
4231 }
4232 }
4233 }
4234
4235 this.highlight_background::<DocumentHighlightRead>(
4236 &read_ranges,
4237 |theme| theme.editor_document_highlight_read_background,
4238 cx,
4239 );
4240 this.highlight_background::<DocumentHighlightWrite>(
4241 &write_ranges,
4242 |theme| theme.editor_document_highlight_write_background,
4243 cx,
4244 );
4245 cx.notify();
4246 })
4247 .log_err();
4248 }
4249 }));
4250 None
4251 }
4252
4253 fn refresh_inline_completion(
4254 &mut self,
4255 debounce: bool,
4256 cx: &mut ViewContext<Self>,
4257 ) -> Option<()> {
4258 let provider = self.inline_completion_provider()?;
4259 let cursor = self.selections.newest_anchor().head();
4260 let (buffer, cursor_buffer_position) =
4261 self.buffer.read(cx).text_anchor_for_position(cursor, cx)?;
4262 if !self.show_inline_completions
4263 || !provider.is_enabled(&buffer, cursor_buffer_position, cx)
4264 {
4265 self.discard_inline_completion(cx);
4266 return None;
4267 }
4268
4269 self.update_visible_inline_completion(cx);
4270 provider.refresh(buffer, cursor_buffer_position, debounce, cx);
4271 Some(())
4272 }
4273
4274 fn cycle_inline_completion(
4275 &mut self,
4276 direction: Direction,
4277 cx: &mut ViewContext<Self>,
4278 ) -> Option<()> {
4279 let provider = self.inline_completion_provider()?;
4280 let cursor = self.selections.newest_anchor().head();
4281 let (buffer, cursor_buffer_position) =
4282 self.buffer.read(cx).text_anchor_for_position(cursor, cx)?;
4283 if !self.show_inline_completions
4284 || !provider.is_enabled(&buffer, cursor_buffer_position, cx)
4285 {
4286 return None;
4287 }
4288
4289 provider.cycle(buffer, cursor_buffer_position, direction, cx);
4290 self.update_visible_inline_completion(cx);
4291
4292 Some(())
4293 }
4294
4295 pub fn show_inline_completion(&mut self, _: &ShowInlineCompletion, cx: &mut ViewContext<Self>) {
4296 if !self.has_active_inline_completion(cx) {
4297 self.refresh_inline_completion(false, cx);
4298 return;
4299 }
4300
4301 self.update_visible_inline_completion(cx);
4302 }
4303
4304 pub fn display_cursor_names(&mut self, _: &DisplayCursorNames, cx: &mut ViewContext<Self>) {
4305 self.show_cursor_names(cx);
4306 }
4307
4308 fn show_cursor_names(&mut self, cx: &mut ViewContext<Self>) {
4309 self.show_cursor_names = true;
4310 cx.notify();
4311 cx.spawn(|this, mut cx| async move {
4312 cx.background_executor().timer(CURSORS_VISIBLE_FOR).await;
4313 this.update(&mut cx, |this, cx| {
4314 this.show_cursor_names = false;
4315 cx.notify()
4316 })
4317 .ok()
4318 })
4319 .detach();
4320 }
4321
4322 pub fn next_inline_completion(&mut self, _: &NextInlineCompletion, cx: &mut ViewContext<Self>) {
4323 if self.has_active_inline_completion(cx) {
4324 self.cycle_inline_completion(Direction::Next, cx);
4325 } else {
4326 let is_copilot_disabled = self.refresh_inline_completion(false, cx).is_none();
4327 if is_copilot_disabled {
4328 cx.propagate();
4329 }
4330 }
4331 }
4332
4333 pub fn previous_inline_completion(
4334 &mut self,
4335 _: &PreviousInlineCompletion,
4336 cx: &mut ViewContext<Self>,
4337 ) {
4338 if self.has_active_inline_completion(cx) {
4339 self.cycle_inline_completion(Direction::Prev, cx);
4340 } else {
4341 let is_copilot_disabled = self.refresh_inline_completion(false, cx).is_none();
4342 if is_copilot_disabled {
4343 cx.propagate();
4344 }
4345 }
4346 }
4347
4348 fn accept_inline_completion(&mut self, cx: &mut ViewContext<Self>) -> bool {
4349 if let Some(completion) = self.take_active_inline_completion(cx) {
4350 if let Some(provider) = self.inline_completion_provider() {
4351 provider.accept(cx);
4352 }
4353
4354 cx.emit(EditorEvent::InputHandled {
4355 utf16_range_to_replace: None,
4356 text: completion.text.to_string().into(),
4357 });
4358 self.insert_with_autoindent_mode(&completion.text.to_string(), None, cx);
4359 self.refresh_inline_completion(true, cx);
4360 cx.notify();
4361 true
4362 } else {
4363 false
4364 }
4365 }
4366
4367 pub fn accept_partial_inline_completion(
4368 &mut self,
4369 _: &AcceptPartialInlineCompletion,
4370 cx: &mut ViewContext<Self>,
4371 ) {
4372 if self.selections.count() == 1 && self.has_active_inline_completion(cx) {
4373 if let Some(completion) = self.take_active_inline_completion(cx) {
4374 let mut partial_completion = completion
4375 .text
4376 .chars()
4377 .by_ref()
4378 .take_while(|c| c.is_alphabetic())
4379 .collect::<String>();
4380 if partial_completion.is_empty() {
4381 partial_completion = completion
4382 .text
4383 .chars()
4384 .by_ref()
4385 .take_while(|c| c.is_whitespace() || !c.is_alphabetic())
4386 .collect::<String>();
4387 }
4388
4389 cx.emit(EditorEvent::InputHandled {
4390 utf16_range_to_replace: None,
4391 text: partial_completion.clone().into(),
4392 });
4393 self.insert_with_autoindent_mode(&partial_completion, None, cx);
4394 self.refresh_inline_completion(true, cx);
4395 cx.notify();
4396 }
4397 }
4398 }
4399
4400 fn discard_inline_completion(&mut self, cx: &mut ViewContext<Self>) -> bool {
4401 if let Some(provider) = self.inline_completion_provider() {
4402 provider.discard(cx);
4403 }
4404
4405 self.take_active_inline_completion(cx).is_some()
4406 }
4407
4408 pub fn has_active_inline_completion(&self, cx: &AppContext) -> bool {
4409 if let Some(completion) = self.active_inline_completion.as_ref() {
4410 let buffer = self.buffer.read(cx).read(cx);
4411 completion.position.is_valid(&buffer)
4412 } else {
4413 false
4414 }
4415 }
4416
4417 fn take_active_inline_completion(&mut self, cx: &mut ViewContext<Self>) -> Option<Inlay> {
4418 let completion = self.active_inline_completion.take()?;
4419 self.display_map.update(cx, |map, cx| {
4420 map.splice_inlays(vec![completion.id], Default::default(), cx);
4421 });
4422 let buffer = self.buffer.read(cx).read(cx);
4423
4424 if completion.position.is_valid(&buffer) {
4425 Some(completion)
4426 } else {
4427 None
4428 }
4429 }
4430
4431 fn update_visible_inline_completion(&mut self, cx: &mut ViewContext<Self>) {
4432 let selection = self.selections.newest_anchor();
4433 let cursor = selection.head();
4434
4435 if self.context_menu.read().is_none()
4436 && self.completion_tasks.is_empty()
4437 && selection.start == selection.end
4438 {
4439 if let Some(provider) = self.inline_completion_provider() {
4440 if let Some((buffer, cursor_buffer_position)) =
4441 self.buffer.read(cx).text_anchor_for_position(cursor, cx)
4442 {
4443 if let Some(text) =
4444 provider.active_completion_text(&buffer, cursor_buffer_position, cx)
4445 {
4446 let text = Rope::from(text);
4447 let mut to_remove = Vec::new();
4448 if let Some(completion) = self.active_inline_completion.take() {
4449 to_remove.push(completion.id);
4450 }
4451
4452 let completion_inlay =
4453 Inlay::suggestion(post_inc(&mut self.next_inlay_id), cursor, text);
4454 self.active_inline_completion = Some(completion_inlay.clone());
4455 self.display_map.update(cx, move |map, cx| {
4456 map.splice_inlays(to_remove, vec![completion_inlay], cx)
4457 });
4458 cx.notify();
4459 return;
4460 }
4461 }
4462 }
4463 }
4464
4465 self.discard_inline_completion(cx);
4466 }
4467
4468 fn inline_completion_provider(&self) -> Option<Arc<dyn InlineCompletionProviderHandle>> {
4469 Some(self.inline_completion_provider.as_ref()?.provider.clone())
4470 }
4471
4472 fn render_code_actions_indicator(
4473 &self,
4474 _style: &EditorStyle,
4475 row: u32,
4476 is_active: bool,
4477 cx: &mut ViewContext<Self>,
4478 ) -> Option<IconButton> {
4479 if self.available_code_actions.is_some() {
4480 Some(
4481 IconButton::new("code_actions_indicator", ui::IconName::Bolt)
4482 .icon_size(IconSize::XSmall)
4483 .size(ui::ButtonSize::None)
4484 .icon_color(Color::Muted)
4485 .selected(is_active)
4486 .on_click(cx.listener(move |editor, _e, cx| {
4487 editor.toggle_code_actions(
4488 &ToggleCodeActions {
4489 deployed_from_indicator: Some(row),
4490 },
4491 cx,
4492 );
4493 })),
4494 )
4495 } else {
4496 None
4497 }
4498 }
4499
4500 fn clear_tasks(&mut self) {
4501 self.tasks.clear()
4502 }
4503
4504 fn insert_tasks(&mut self, row: u32, tasks: RunnableTasks) {
4505 if let Some(_) = self.tasks.insert(row, tasks) {
4506 // This case should hopefully be rare, but just in case...
4507 log::error!("multiple different run targets found on a single line, only the last target will be rendered")
4508 }
4509 }
4510
4511 fn render_run_indicator(
4512 &self,
4513 _style: &EditorStyle,
4514 is_active: bool,
4515 row: u32,
4516 cx: &mut ViewContext<Self>,
4517 ) -> IconButton {
4518 IconButton::new("code_actions_indicator", ui::IconName::Play)
4519 .icon_size(IconSize::XSmall)
4520 .size(ui::ButtonSize::None)
4521 .icon_color(Color::Muted)
4522 .selected(is_active)
4523 .on_click(cx.listener(move |editor, _e, cx| {
4524 editor.toggle_code_actions(
4525 &ToggleCodeActions {
4526 deployed_from_indicator: Some(row),
4527 },
4528 cx,
4529 );
4530 }))
4531 }
4532
4533 pub fn render_fold_indicators(
4534 &mut self,
4535 fold_data: Vec<Option<(FoldStatus, u32, bool)>>,
4536 _style: &EditorStyle,
4537 gutter_hovered: bool,
4538 _line_height: Pixels,
4539 _gutter_margin: Pixels,
4540 cx: &mut ViewContext<Self>,
4541 ) -> Vec<Option<AnyElement>> {
4542 fold_data
4543 .iter()
4544 .enumerate()
4545 .map(|(ix, fold_data)| {
4546 fold_data
4547 .map(|(fold_status, buffer_row, active)| {
4548 (active || gutter_hovered || fold_status == FoldStatus::Folded).then(|| {
4549 IconButton::new(ix, ui::IconName::ChevronDown)
4550 .on_click(cx.listener(move |this, _e, cx| match fold_status {
4551 FoldStatus::Folded => {
4552 this.unfold_at(&UnfoldAt { buffer_row }, cx);
4553 }
4554 FoldStatus::Foldable => {
4555 this.fold_at(&FoldAt { buffer_row }, cx);
4556 }
4557 }))
4558 .icon_color(ui::Color::Muted)
4559 .icon_size(ui::IconSize::Small)
4560 .selected(fold_status == FoldStatus::Folded)
4561 .selected_icon(ui::IconName::ChevronRight)
4562 .size(ui::ButtonSize::None)
4563 .into_any_element()
4564 })
4565 })
4566 .flatten()
4567 })
4568 .collect()
4569 }
4570
4571 pub fn context_menu_visible(&self) -> bool {
4572 self.context_menu
4573 .read()
4574 .as_ref()
4575 .map_or(false, |menu| menu.visible())
4576 }
4577
4578 fn render_context_menu(
4579 &self,
4580 cursor_position: DisplayPoint,
4581 style: &EditorStyle,
4582 max_height: Pixels,
4583 cx: &mut ViewContext<Editor>,
4584 ) -> Option<(ContextMenuOrigin, AnyElement)> {
4585 self.context_menu.read().as_ref().map(|menu| {
4586 menu.render(
4587 cursor_position,
4588 style,
4589 max_height,
4590 self.workspace.as_ref().map(|(w, _)| w.clone()),
4591 cx,
4592 )
4593 })
4594 }
4595
4596 fn hide_context_menu(&mut self, cx: &mut ViewContext<Self>) -> Option<ContextMenu> {
4597 cx.notify();
4598 self.completion_tasks.clear();
4599 let context_menu = self.context_menu.write().take();
4600 if context_menu.is_some() {
4601 self.update_visible_inline_completion(cx);
4602 }
4603 context_menu
4604 }
4605
4606 pub fn insert_snippet(
4607 &mut self,
4608 insertion_ranges: &[Range<usize>],
4609 snippet: Snippet,
4610 cx: &mut ViewContext<Self>,
4611 ) -> Result<()> {
4612 let tabstops = self.buffer.update(cx, |buffer, cx| {
4613 let snippet_text: Arc<str> = snippet.text.clone().into();
4614 buffer.edit(
4615 insertion_ranges
4616 .iter()
4617 .cloned()
4618 .map(|range| (range, snippet_text.clone())),
4619 Some(AutoindentMode::EachLine),
4620 cx,
4621 );
4622
4623 let snapshot = &*buffer.read(cx);
4624 let snippet = &snippet;
4625 snippet
4626 .tabstops
4627 .iter()
4628 .map(|tabstop| {
4629 let mut tabstop_ranges = tabstop
4630 .iter()
4631 .flat_map(|tabstop_range| {
4632 let mut delta = 0_isize;
4633 insertion_ranges.iter().map(move |insertion_range| {
4634 let insertion_start = insertion_range.start as isize + delta;
4635 delta +=
4636 snippet.text.len() as isize - insertion_range.len() as isize;
4637
4638 let start = snapshot.anchor_before(
4639 (insertion_start + tabstop_range.start) as usize,
4640 );
4641 let end = snapshot
4642 .anchor_after((insertion_start + tabstop_range.end) as usize);
4643 start..end
4644 })
4645 })
4646 .collect::<Vec<_>>();
4647 tabstop_ranges.sort_unstable_by(|a, b| a.start.cmp(&b.start, snapshot));
4648 tabstop_ranges
4649 })
4650 .collect::<Vec<_>>()
4651 });
4652
4653 if let Some(tabstop) = tabstops.first() {
4654 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
4655 s.select_ranges(tabstop.iter().cloned());
4656 });
4657 self.snippet_stack.push(SnippetState {
4658 active_index: 0,
4659 ranges: tabstops,
4660 });
4661
4662 // Check whether the just-entered snippet ends with an auto-closable bracket.
4663 if self.autoclose_regions.is_empty() {
4664 let snapshot = self.buffer.read(cx).snapshot(cx);
4665 for selection in &mut self.selections.all::<Point>(cx) {
4666 let selection_head = selection.head();
4667 let Some(scope) = snapshot.language_scope_at(selection_head) else {
4668 continue;
4669 };
4670
4671 let mut bracket_pair = None;
4672 let next_chars = snapshot.chars_at(selection_head).collect::<String>();
4673 let prev_chars = snapshot
4674 .reversed_chars_at(selection_head)
4675 .collect::<String>();
4676 for (pair, enabled) in scope.brackets() {
4677 if enabled
4678 && pair.close
4679 && prev_chars.starts_with(pair.start.as_str())
4680 && next_chars.starts_with(pair.end.as_str())
4681 {
4682 bracket_pair = Some(pair.clone());
4683 break;
4684 }
4685 }
4686 if let Some(pair) = bracket_pair {
4687 let start = snapshot.anchor_after(selection_head);
4688 let end = snapshot.anchor_after(selection_head);
4689 self.autoclose_regions.push(AutocloseRegion {
4690 selection_id: selection.id,
4691 range: start..end,
4692 pair,
4693 });
4694 }
4695 }
4696 }
4697 }
4698 Ok(())
4699 }
4700
4701 pub fn move_to_next_snippet_tabstop(&mut self, cx: &mut ViewContext<Self>) -> bool {
4702 self.move_to_snippet_tabstop(Bias::Right, cx)
4703 }
4704
4705 pub fn move_to_prev_snippet_tabstop(&mut self, cx: &mut ViewContext<Self>) -> bool {
4706 self.move_to_snippet_tabstop(Bias::Left, cx)
4707 }
4708
4709 pub fn move_to_snippet_tabstop(&mut self, bias: Bias, cx: &mut ViewContext<Self>) -> bool {
4710 if let Some(mut snippet) = self.snippet_stack.pop() {
4711 match bias {
4712 Bias::Left => {
4713 if snippet.active_index > 0 {
4714 snippet.active_index -= 1;
4715 } else {
4716 self.snippet_stack.push(snippet);
4717 return false;
4718 }
4719 }
4720 Bias::Right => {
4721 if snippet.active_index + 1 < snippet.ranges.len() {
4722 snippet.active_index += 1;
4723 } else {
4724 self.snippet_stack.push(snippet);
4725 return false;
4726 }
4727 }
4728 }
4729 if let Some(current_ranges) = snippet.ranges.get(snippet.active_index) {
4730 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
4731 s.select_anchor_ranges(current_ranges.iter().cloned())
4732 });
4733 // If snippet state is not at the last tabstop, push it back on the stack
4734 if snippet.active_index + 1 < snippet.ranges.len() {
4735 self.snippet_stack.push(snippet);
4736 }
4737 return true;
4738 }
4739 }
4740
4741 false
4742 }
4743
4744 pub fn clear(&mut self, cx: &mut ViewContext<Self>) {
4745 self.transact(cx, |this, cx| {
4746 this.select_all(&SelectAll, cx);
4747 this.insert("", cx);
4748 });
4749 }
4750
4751 pub fn backspace(&mut self, _: &Backspace, cx: &mut ViewContext<Self>) {
4752 self.transact(cx, |this, cx| {
4753 this.select_autoclose_pair(cx);
4754 let mut selections = this.selections.all::<Point>(cx);
4755 if !this.selections.line_mode {
4756 let display_map = this.display_map.update(cx, |map, cx| map.snapshot(cx));
4757 for selection in &mut selections {
4758 if selection.is_empty() {
4759 let old_head = selection.head();
4760 let mut new_head =
4761 movement::left(&display_map, old_head.to_display_point(&display_map))
4762 .to_point(&display_map);
4763 if let Some((buffer, line_buffer_range)) = display_map
4764 .buffer_snapshot
4765 .buffer_line_for_row(old_head.row)
4766 {
4767 let indent_size =
4768 buffer.indent_size_for_line(line_buffer_range.start.row);
4769 let indent_len = match indent_size.kind {
4770 IndentKind::Space => {
4771 buffer.settings_at(line_buffer_range.start, cx).tab_size
4772 }
4773 IndentKind::Tab => NonZeroU32::new(1).unwrap(),
4774 };
4775 if old_head.column <= indent_size.len && old_head.column > 0 {
4776 let indent_len = indent_len.get();
4777 new_head = cmp::min(
4778 new_head,
4779 Point::new(
4780 old_head.row,
4781 ((old_head.column - 1) / indent_len) * indent_len,
4782 ),
4783 );
4784 }
4785 }
4786
4787 selection.set_head(new_head, SelectionGoal::None);
4788 }
4789 }
4790 }
4791
4792 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(selections));
4793 this.insert("", cx);
4794 this.refresh_inline_completion(true, cx);
4795 });
4796 }
4797
4798 pub fn delete(&mut self, _: &Delete, cx: &mut ViewContext<Self>) {
4799 self.transact(cx, |this, cx| {
4800 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
4801 let line_mode = s.line_mode;
4802 s.move_with(|map, selection| {
4803 if selection.is_empty() && !line_mode {
4804 let cursor = movement::right(map, selection.head());
4805 selection.end = cursor;
4806 selection.reversed = true;
4807 selection.goal = SelectionGoal::None;
4808 }
4809 })
4810 });
4811 this.insert("", cx);
4812 this.refresh_inline_completion(true, cx);
4813 });
4814 }
4815
4816 pub fn tab_prev(&mut self, _: &TabPrev, cx: &mut ViewContext<Self>) {
4817 if self.move_to_prev_snippet_tabstop(cx) {
4818 return;
4819 }
4820
4821 self.outdent(&Outdent, cx);
4822 }
4823
4824 pub fn tab(&mut self, _: &Tab, cx: &mut ViewContext<Self>) {
4825 if self.move_to_next_snippet_tabstop(cx) || self.read_only(cx) {
4826 return;
4827 }
4828
4829 let mut selections = self.selections.all_adjusted(cx);
4830 let buffer = self.buffer.read(cx);
4831 let snapshot = buffer.snapshot(cx);
4832 let rows_iter = selections.iter().map(|s| s.head().row);
4833 let suggested_indents = snapshot.suggested_indents(rows_iter, cx);
4834
4835 let mut edits = Vec::new();
4836 let mut prev_edited_row = 0;
4837 let mut row_delta = 0;
4838 for selection in &mut selections {
4839 if selection.start.row != prev_edited_row {
4840 row_delta = 0;
4841 }
4842 prev_edited_row = selection.end.row;
4843
4844 // If the selection is non-empty, then increase the indentation of the selected lines.
4845 if !selection.is_empty() {
4846 row_delta =
4847 Self::indent_selection(buffer, &snapshot, selection, &mut edits, row_delta, cx);
4848 continue;
4849 }
4850
4851 // If the selection is empty and the cursor is in the leading whitespace before the
4852 // suggested indentation, then auto-indent the line.
4853 let cursor = selection.head();
4854 let current_indent = snapshot.indent_size_for_line(cursor.row);
4855 if let Some(suggested_indent) = suggested_indents.get(&cursor.row).copied() {
4856 if cursor.column < suggested_indent.len
4857 && cursor.column <= current_indent.len
4858 && current_indent.len <= suggested_indent.len
4859 {
4860 selection.start = Point::new(cursor.row, suggested_indent.len);
4861 selection.end = selection.start;
4862 if row_delta == 0 {
4863 edits.extend(Buffer::edit_for_indent_size_adjustment(
4864 cursor.row,
4865 current_indent,
4866 suggested_indent,
4867 ));
4868 row_delta = suggested_indent.len - current_indent.len;
4869 }
4870 continue;
4871 }
4872 }
4873
4874 // Accept copilot completion if there is only one selection and the cursor is not
4875 // in the leading whitespace.
4876 if self.selections.count() == 1
4877 && cursor.column >= current_indent.len
4878 && self.has_active_inline_completion(cx)
4879 {
4880 self.accept_inline_completion(cx);
4881 return;
4882 }
4883
4884 // Otherwise, insert a hard or soft tab.
4885 let settings = buffer.settings_at(cursor, cx);
4886 let tab_size = if settings.hard_tabs {
4887 IndentSize::tab()
4888 } else {
4889 let tab_size = settings.tab_size.get();
4890 let char_column = snapshot
4891 .text_for_range(Point::new(cursor.row, 0)..cursor)
4892 .flat_map(str::chars)
4893 .count()
4894 + row_delta as usize;
4895 let chars_to_next_tab_stop = tab_size - (char_column as u32 % tab_size);
4896 IndentSize::spaces(chars_to_next_tab_stop)
4897 };
4898 selection.start = Point::new(cursor.row, cursor.column + row_delta + tab_size.len);
4899 selection.end = selection.start;
4900 edits.push((cursor..cursor, tab_size.chars().collect::<String>()));
4901 row_delta += tab_size.len;
4902 }
4903
4904 self.transact(cx, |this, cx| {
4905 this.buffer.update(cx, |b, cx| b.edit(edits, None, cx));
4906 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(selections));
4907 this.refresh_inline_completion(true, cx);
4908 });
4909 }
4910
4911 pub fn indent(&mut self, _: &Indent, cx: &mut ViewContext<Self>) {
4912 if self.read_only(cx) {
4913 return;
4914 }
4915 let mut selections = self.selections.all::<Point>(cx);
4916 let mut prev_edited_row = 0;
4917 let mut row_delta = 0;
4918 let mut edits = Vec::new();
4919 let buffer = self.buffer.read(cx);
4920 let snapshot = buffer.snapshot(cx);
4921 for selection in &mut selections {
4922 if selection.start.row != prev_edited_row {
4923 row_delta = 0;
4924 }
4925 prev_edited_row = selection.end.row;
4926
4927 row_delta =
4928 Self::indent_selection(buffer, &snapshot, selection, &mut edits, row_delta, cx);
4929 }
4930
4931 self.transact(cx, |this, cx| {
4932 this.buffer.update(cx, |b, cx| b.edit(edits, None, cx));
4933 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(selections));
4934 });
4935 }
4936
4937 fn indent_selection(
4938 buffer: &MultiBuffer,
4939 snapshot: &MultiBufferSnapshot,
4940 selection: &mut Selection<Point>,
4941 edits: &mut Vec<(Range<Point>, String)>,
4942 delta_for_start_row: u32,
4943 cx: &AppContext,
4944 ) -> u32 {
4945 let settings = buffer.settings_at(selection.start, cx);
4946 let tab_size = settings.tab_size.get();
4947 let indent_kind = if settings.hard_tabs {
4948 IndentKind::Tab
4949 } else {
4950 IndentKind::Space
4951 };
4952 let mut start_row = selection.start.row;
4953 let mut end_row = selection.end.row + 1;
4954
4955 // If a selection ends at the beginning of a line, don't indent
4956 // that last line.
4957 if selection.end.column == 0 && selection.end.row > selection.start.row {
4958 end_row -= 1;
4959 }
4960
4961 // Avoid re-indenting a row that has already been indented by a
4962 // previous selection, but still update this selection's column
4963 // to reflect that indentation.
4964 if delta_for_start_row > 0 {
4965 start_row += 1;
4966 selection.start.column += delta_for_start_row;
4967 if selection.end.row == selection.start.row {
4968 selection.end.column += delta_for_start_row;
4969 }
4970 }
4971
4972 let mut delta_for_end_row = 0;
4973 let has_multiple_rows = start_row + 1 != end_row;
4974 for row in start_row..end_row {
4975 let current_indent = snapshot.indent_size_for_line(row);
4976 let indent_delta = match (current_indent.kind, indent_kind) {
4977 (IndentKind::Space, IndentKind::Space) => {
4978 let columns_to_next_tab_stop = tab_size - (current_indent.len % tab_size);
4979 IndentSize::spaces(columns_to_next_tab_stop)
4980 }
4981 (IndentKind::Tab, IndentKind::Space) => IndentSize::spaces(tab_size),
4982 (_, IndentKind::Tab) => IndentSize::tab(),
4983 };
4984
4985 let start = if has_multiple_rows || current_indent.len < selection.start.column {
4986 0
4987 } else {
4988 selection.start.column
4989 };
4990 let row_start = Point::new(row, start);
4991 edits.push((
4992 row_start..row_start,
4993 indent_delta.chars().collect::<String>(),
4994 ));
4995
4996 // Update this selection's endpoints to reflect the indentation.
4997 if row == selection.start.row {
4998 selection.start.column += indent_delta.len;
4999 }
5000 if row == selection.end.row {
5001 selection.end.column += indent_delta.len;
5002 delta_for_end_row = indent_delta.len;
5003 }
5004 }
5005
5006 if selection.start.row == selection.end.row {
5007 delta_for_start_row + delta_for_end_row
5008 } else {
5009 delta_for_end_row
5010 }
5011 }
5012
5013 pub fn outdent(&mut self, _: &Outdent, cx: &mut ViewContext<Self>) {
5014 if self.read_only(cx) {
5015 return;
5016 }
5017 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5018 let selections = self.selections.all::<Point>(cx);
5019 let mut deletion_ranges = Vec::new();
5020 let mut last_outdent = None;
5021 {
5022 let buffer = self.buffer.read(cx);
5023 let snapshot = buffer.snapshot(cx);
5024 for selection in &selections {
5025 let settings = buffer.settings_at(selection.start, cx);
5026 let tab_size = settings.tab_size.get();
5027 let mut rows = selection.spanned_rows(false, &display_map);
5028
5029 // Avoid re-outdenting a row that has already been outdented by a
5030 // previous selection.
5031 if let Some(last_row) = last_outdent {
5032 if last_row == rows.start {
5033 rows.start += 1;
5034 }
5035 }
5036 let has_multiple_rows = rows.len() > 1;
5037 for row in rows {
5038 let indent_size = snapshot.indent_size_for_line(row);
5039 if indent_size.len > 0 {
5040 let deletion_len = match indent_size.kind {
5041 IndentKind::Space => {
5042 let columns_to_prev_tab_stop = indent_size.len % tab_size;
5043 if columns_to_prev_tab_stop == 0 {
5044 tab_size
5045 } else {
5046 columns_to_prev_tab_stop
5047 }
5048 }
5049 IndentKind::Tab => 1,
5050 };
5051 let start = if has_multiple_rows
5052 || deletion_len > selection.start.column
5053 || indent_size.len < selection.start.column
5054 {
5055 0
5056 } else {
5057 selection.start.column - deletion_len
5058 };
5059 deletion_ranges
5060 .push(Point::new(row, start)..Point::new(row, start + deletion_len));
5061 last_outdent = Some(row);
5062 }
5063 }
5064 }
5065 }
5066
5067 self.transact(cx, |this, cx| {
5068 this.buffer.update(cx, |buffer, cx| {
5069 let empty_str: Arc<str> = "".into();
5070 buffer.edit(
5071 deletion_ranges
5072 .into_iter()
5073 .map(|range| (range, empty_str.clone())),
5074 None,
5075 cx,
5076 );
5077 });
5078 let selections = this.selections.all::<usize>(cx);
5079 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(selections));
5080 });
5081 }
5082
5083 pub fn delete_line(&mut self, _: &DeleteLine, cx: &mut ViewContext<Self>) {
5084 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5085 let selections = self.selections.all::<Point>(cx);
5086
5087 let mut new_cursors = Vec::new();
5088 let mut edit_ranges = Vec::new();
5089 let mut selections = selections.iter().peekable();
5090 while let Some(selection) = selections.next() {
5091 let mut rows = selection.spanned_rows(false, &display_map);
5092 let goal_display_column = selection.head().to_display_point(&display_map).column();
5093
5094 // Accumulate contiguous regions of rows that we want to delete.
5095 while let Some(next_selection) = selections.peek() {
5096 let next_rows = next_selection.spanned_rows(false, &display_map);
5097 if next_rows.start <= rows.end {
5098 rows.end = next_rows.end;
5099 selections.next().unwrap();
5100 } else {
5101 break;
5102 }
5103 }
5104
5105 let buffer = &display_map.buffer_snapshot;
5106 let mut edit_start = Point::new(rows.start, 0).to_offset(buffer);
5107 let edit_end;
5108 let cursor_buffer_row;
5109 if buffer.max_point().row >= rows.end {
5110 // If there's a line after the range, delete the \n from the end of the row range
5111 // and position the cursor on the next line.
5112 edit_end = Point::new(rows.end, 0).to_offset(buffer);
5113 cursor_buffer_row = rows.end;
5114 } else {
5115 // If there isn't a line after the range, delete the \n from the line before the
5116 // start of the row range and position the cursor there.
5117 edit_start = edit_start.saturating_sub(1);
5118 edit_end = buffer.len();
5119 cursor_buffer_row = rows.start.saturating_sub(1);
5120 }
5121
5122 let mut cursor = Point::new(cursor_buffer_row, 0).to_display_point(&display_map);
5123 *cursor.column_mut() =
5124 cmp::min(goal_display_column, display_map.line_len(cursor.row()));
5125
5126 new_cursors.push((
5127 selection.id,
5128 buffer.anchor_after(cursor.to_point(&display_map)),
5129 ));
5130 edit_ranges.push(edit_start..edit_end);
5131 }
5132
5133 self.transact(cx, |this, cx| {
5134 let buffer = this.buffer.update(cx, |buffer, cx| {
5135 let empty_str: Arc<str> = "".into();
5136 buffer.edit(
5137 edit_ranges
5138 .into_iter()
5139 .map(|range| (range, empty_str.clone())),
5140 None,
5141 cx,
5142 );
5143 buffer.snapshot(cx)
5144 });
5145 let new_selections = new_cursors
5146 .into_iter()
5147 .map(|(id, cursor)| {
5148 let cursor = cursor.to_point(&buffer);
5149 Selection {
5150 id,
5151 start: cursor,
5152 end: cursor,
5153 reversed: false,
5154 goal: SelectionGoal::None,
5155 }
5156 })
5157 .collect();
5158
5159 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5160 s.select(new_selections);
5161 });
5162 });
5163 }
5164
5165 pub fn join_lines(&mut self, _: &JoinLines, cx: &mut ViewContext<Self>) {
5166 if self.read_only(cx) {
5167 return;
5168 }
5169 let mut row_ranges = Vec::<Range<u32>>::new();
5170 for selection in self.selections.all::<Point>(cx) {
5171 let start = selection.start.row;
5172 let end = if selection.start.row == selection.end.row {
5173 selection.start.row + 1
5174 } else {
5175 selection.end.row
5176 };
5177
5178 if let Some(last_row_range) = row_ranges.last_mut() {
5179 if start <= last_row_range.end {
5180 last_row_range.end = end;
5181 continue;
5182 }
5183 }
5184 row_ranges.push(start..end);
5185 }
5186
5187 let snapshot = self.buffer.read(cx).snapshot(cx);
5188 let mut cursor_positions = Vec::new();
5189 for row_range in &row_ranges {
5190 let anchor = snapshot.anchor_before(Point::new(
5191 row_range.end - 1,
5192 snapshot.line_len(row_range.end - 1),
5193 ));
5194 cursor_positions.push(anchor..anchor);
5195 }
5196
5197 self.transact(cx, |this, cx| {
5198 for row_range in row_ranges.into_iter().rev() {
5199 for row in row_range.rev() {
5200 let end_of_line = Point::new(row, snapshot.line_len(row));
5201 let indent = snapshot.indent_size_for_line(row + 1);
5202 let start_of_next_line = Point::new(row + 1, indent.len);
5203
5204 let replace = if snapshot.line_len(row + 1) > indent.len {
5205 " "
5206 } else {
5207 ""
5208 };
5209
5210 this.buffer.update(cx, |buffer, cx| {
5211 buffer.edit([(end_of_line..start_of_next_line, replace)], None, cx)
5212 });
5213 }
5214 }
5215
5216 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5217 s.select_anchor_ranges(cursor_positions)
5218 });
5219 });
5220 }
5221
5222 pub fn sort_lines_case_sensitive(
5223 &mut self,
5224 _: &SortLinesCaseSensitive,
5225 cx: &mut ViewContext<Self>,
5226 ) {
5227 self.manipulate_lines(cx, |lines| lines.sort())
5228 }
5229
5230 pub fn sort_lines_case_insensitive(
5231 &mut self,
5232 _: &SortLinesCaseInsensitive,
5233 cx: &mut ViewContext<Self>,
5234 ) {
5235 self.manipulate_lines(cx, |lines| lines.sort_by_key(|line| line.to_lowercase()))
5236 }
5237
5238 pub fn unique_lines_case_insensitive(
5239 &mut self,
5240 _: &UniqueLinesCaseInsensitive,
5241 cx: &mut ViewContext<Self>,
5242 ) {
5243 self.manipulate_lines(cx, |lines| {
5244 let mut seen = HashSet::default();
5245 lines.retain(|line| seen.insert(line.to_lowercase()));
5246 })
5247 }
5248
5249 pub fn unique_lines_case_sensitive(
5250 &mut self,
5251 _: &UniqueLinesCaseSensitive,
5252 cx: &mut ViewContext<Self>,
5253 ) {
5254 self.manipulate_lines(cx, |lines| {
5255 let mut seen = HashSet::default();
5256 lines.retain(|line| seen.insert(*line));
5257 })
5258 }
5259
5260 pub fn revert_selected_hunks(&mut self, _: &RevertSelectedHunks, cx: &mut ViewContext<Self>) {
5261 let revert_changes = self.gather_revert_changes(&self.selections.disjoint_anchors(), cx);
5262 if !revert_changes.is_empty() {
5263 self.transact(cx, |editor, cx| {
5264 editor.buffer().update(cx, |multi_buffer, cx| {
5265 for (buffer_id, changes) in revert_changes {
5266 if let Some(buffer) = multi_buffer.buffer(buffer_id) {
5267 buffer.update(cx, |buffer, cx| {
5268 buffer.edit(
5269 changes.into_iter().map(|(range, text)| {
5270 (range, text.to_string().map(Arc::<str>::from))
5271 }),
5272 None,
5273 cx,
5274 );
5275 });
5276 }
5277 }
5278 });
5279 editor.change_selections(None, cx, |selections| selections.refresh());
5280 });
5281 }
5282 }
5283
5284 pub fn open_active_item_in_terminal(&mut self, _: &OpenInTerminal, cx: &mut ViewContext<Self>) {
5285 if let Some(working_directory) = self.active_excerpt(cx).and_then(|(_, buffer, _)| {
5286 let project_path = buffer.read(cx).project_path(cx)?;
5287 let project = self.project.as_ref()?.read(cx);
5288 let entry = project.entry_for_path(&project_path, cx)?;
5289 let abs_path = project.absolute_path(&project_path, cx)?;
5290 let parent = if entry.is_symlink {
5291 abs_path.canonicalize().ok()?
5292 } else {
5293 abs_path
5294 }
5295 .parent()?
5296 .to_path_buf();
5297 Some(parent)
5298 }) {
5299 cx.dispatch_action(OpenTerminal { working_directory }.boxed_clone());
5300 }
5301 }
5302
5303 fn gather_revert_changes(
5304 &mut self,
5305 selections: &[Selection<Anchor>],
5306 cx: &mut ViewContext<'_, Editor>,
5307 ) -> HashMap<BufferId, Vec<(Range<text::Anchor>, Rope)>> {
5308 let mut revert_changes = HashMap::default();
5309 self.buffer.update(cx, |multi_buffer, cx| {
5310 let multi_buffer_snapshot = multi_buffer.snapshot(cx);
5311 for hunk in hunks_for_selections(&multi_buffer_snapshot, selections) {
5312 Self::prepare_revert_change(&mut revert_changes, &multi_buffer, &hunk, cx);
5313 }
5314 });
5315 revert_changes
5316 }
5317
5318 fn prepare_revert_change(
5319 revert_changes: &mut HashMap<BufferId, Vec<(Range<text::Anchor>, Rope)>>,
5320 multi_buffer: &MultiBuffer,
5321 hunk: &DiffHunk<u32>,
5322 cx: &mut AppContext,
5323 ) -> Option<()> {
5324 let buffer = multi_buffer.buffer(hunk.buffer_id)?;
5325 let buffer = buffer.read(cx);
5326 let original_text = buffer.diff_base()?.slice(hunk.diff_base_byte_range.clone());
5327 let buffer_snapshot = buffer.snapshot();
5328 let buffer_revert_changes = revert_changes.entry(buffer.remote_id()).or_default();
5329 if let Err(i) = buffer_revert_changes.binary_search_by(|probe| {
5330 probe
5331 .0
5332 .start
5333 .cmp(&hunk.buffer_range.start, &buffer_snapshot)
5334 .then(probe.0.end.cmp(&hunk.buffer_range.end, &buffer_snapshot))
5335 }) {
5336 buffer_revert_changes.insert(i, (hunk.buffer_range.clone(), original_text));
5337 Some(())
5338 } else {
5339 None
5340 }
5341 }
5342
5343 pub fn reverse_lines(&mut self, _: &ReverseLines, cx: &mut ViewContext<Self>) {
5344 self.manipulate_lines(cx, |lines| lines.reverse())
5345 }
5346
5347 pub fn shuffle_lines(&mut self, _: &ShuffleLines, cx: &mut ViewContext<Self>) {
5348 self.manipulate_lines(cx, |lines| lines.shuffle(&mut thread_rng()))
5349 }
5350
5351 fn manipulate_lines<Fn>(&mut self, cx: &mut ViewContext<Self>, mut callback: Fn)
5352 where
5353 Fn: FnMut(&mut Vec<&str>),
5354 {
5355 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5356 let buffer = self.buffer.read(cx).snapshot(cx);
5357
5358 let mut edits = Vec::new();
5359
5360 let selections = self.selections.all::<Point>(cx);
5361 let mut selections = selections.iter().peekable();
5362 let mut contiguous_row_selections = Vec::new();
5363 let mut new_selections = Vec::new();
5364 let mut added_lines = 0;
5365 let mut removed_lines = 0;
5366
5367 while let Some(selection) = selections.next() {
5368 let (start_row, end_row) = consume_contiguous_rows(
5369 &mut contiguous_row_selections,
5370 selection,
5371 &display_map,
5372 &mut selections,
5373 );
5374
5375 let start_point = Point::new(start_row, 0);
5376 let end_point = Point::new(end_row - 1, buffer.line_len(end_row - 1));
5377 let text = buffer
5378 .text_for_range(start_point..end_point)
5379 .collect::<String>();
5380
5381 let mut lines = text.split('\n').collect_vec();
5382
5383 let lines_before = lines.len();
5384 callback(&mut lines);
5385 let lines_after = lines.len();
5386
5387 edits.push((start_point..end_point, lines.join("\n")));
5388
5389 // Selections must change based on added and removed line count
5390 let start_row = start_point.row + added_lines as u32 - removed_lines as u32;
5391 let end_row = start_row + lines_after.saturating_sub(1) as u32;
5392 new_selections.push(Selection {
5393 id: selection.id,
5394 start: start_row,
5395 end: end_row,
5396 goal: SelectionGoal::None,
5397 reversed: selection.reversed,
5398 });
5399
5400 if lines_after > lines_before {
5401 added_lines += lines_after - lines_before;
5402 } else if lines_before > lines_after {
5403 removed_lines += lines_before - lines_after;
5404 }
5405 }
5406
5407 self.transact(cx, |this, cx| {
5408 let buffer = this.buffer.update(cx, |buffer, cx| {
5409 buffer.edit(edits, None, cx);
5410 buffer.snapshot(cx)
5411 });
5412
5413 // Recalculate offsets on newly edited buffer
5414 let new_selections = new_selections
5415 .iter()
5416 .map(|s| {
5417 let start_point = Point::new(s.start, 0);
5418 let end_point = Point::new(s.end, buffer.line_len(s.end));
5419 Selection {
5420 id: s.id,
5421 start: buffer.point_to_offset(start_point),
5422 end: buffer.point_to_offset(end_point),
5423 goal: s.goal,
5424 reversed: s.reversed,
5425 }
5426 })
5427 .collect();
5428
5429 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5430 s.select(new_selections);
5431 });
5432
5433 this.request_autoscroll(Autoscroll::fit(), cx);
5434 });
5435 }
5436
5437 pub fn convert_to_upper_case(&mut self, _: &ConvertToUpperCase, cx: &mut ViewContext<Self>) {
5438 self.manipulate_text(cx, |text| text.to_uppercase())
5439 }
5440
5441 pub fn convert_to_lower_case(&mut self, _: &ConvertToLowerCase, cx: &mut ViewContext<Self>) {
5442 self.manipulate_text(cx, |text| text.to_lowercase())
5443 }
5444
5445 pub fn convert_to_title_case(&mut self, _: &ConvertToTitleCase, cx: &mut ViewContext<Self>) {
5446 self.manipulate_text(cx, |text| {
5447 // Hack to get around the fact that to_case crate doesn't support '\n' as a word boundary
5448 // https://github.com/rutrum/convert-case/issues/16
5449 text.split('\n')
5450 .map(|line| line.to_case(Case::Title))
5451 .join("\n")
5452 })
5453 }
5454
5455 pub fn convert_to_snake_case(&mut self, _: &ConvertToSnakeCase, cx: &mut ViewContext<Self>) {
5456 self.manipulate_text(cx, |text| text.to_case(Case::Snake))
5457 }
5458
5459 pub fn convert_to_kebab_case(&mut self, _: &ConvertToKebabCase, cx: &mut ViewContext<Self>) {
5460 self.manipulate_text(cx, |text| text.to_case(Case::Kebab))
5461 }
5462
5463 pub fn convert_to_upper_camel_case(
5464 &mut self,
5465 _: &ConvertToUpperCamelCase,
5466 cx: &mut ViewContext<Self>,
5467 ) {
5468 self.manipulate_text(cx, |text| {
5469 // Hack to get around the fact that to_case crate doesn't support '\n' as a word boundary
5470 // https://github.com/rutrum/convert-case/issues/16
5471 text.split('\n')
5472 .map(|line| line.to_case(Case::UpperCamel))
5473 .join("\n")
5474 })
5475 }
5476
5477 pub fn convert_to_lower_camel_case(
5478 &mut self,
5479 _: &ConvertToLowerCamelCase,
5480 cx: &mut ViewContext<Self>,
5481 ) {
5482 self.manipulate_text(cx, |text| text.to_case(Case::Camel))
5483 }
5484
5485 pub fn convert_to_opposite_case(
5486 &mut self,
5487 _: &ConvertToOppositeCase,
5488 cx: &mut ViewContext<Self>,
5489 ) {
5490 self.manipulate_text(cx, |text| {
5491 text.chars()
5492 .fold(String::with_capacity(text.len()), |mut t, c| {
5493 if c.is_uppercase() {
5494 t.extend(c.to_lowercase());
5495 } else {
5496 t.extend(c.to_uppercase());
5497 }
5498 t
5499 })
5500 })
5501 }
5502
5503 fn manipulate_text<Fn>(&mut self, cx: &mut ViewContext<Self>, mut callback: Fn)
5504 where
5505 Fn: FnMut(&str) -> String,
5506 {
5507 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5508 let buffer = self.buffer.read(cx).snapshot(cx);
5509
5510 let mut new_selections = Vec::new();
5511 let mut edits = Vec::new();
5512 let mut selection_adjustment = 0i32;
5513
5514 for selection in self.selections.all::<usize>(cx) {
5515 let selection_is_empty = selection.is_empty();
5516
5517 let (start, end) = if selection_is_empty {
5518 let word_range = movement::surrounding_word(
5519 &display_map,
5520 selection.start.to_display_point(&display_map),
5521 );
5522 let start = word_range.start.to_offset(&display_map, Bias::Left);
5523 let end = word_range.end.to_offset(&display_map, Bias::Left);
5524 (start, end)
5525 } else {
5526 (selection.start, selection.end)
5527 };
5528
5529 let text = buffer.text_for_range(start..end).collect::<String>();
5530 let old_length = text.len() as i32;
5531 let text = callback(&text);
5532
5533 new_selections.push(Selection {
5534 start: (start as i32 - selection_adjustment) as usize,
5535 end: ((start + text.len()) as i32 - selection_adjustment) as usize,
5536 goal: SelectionGoal::None,
5537 ..selection
5538 });
5539
5540 selection_adjustment += old_length - text.len() as i32;
5541
5542 edits.push((start..end, text));
5543 }
5544
5545 self.transact(cx, |this, cx| {
5546 this.buffer.update(cx, |buffer, cx| {
5547 buffer.edit(edits, None, cx);
5548 });
5549
5550 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5551 s.select(new_selections);
5552 });
5553
5554 this.request_autoscroll(Autoscroll::fit(), cx);
5555 });
5556 }
5557
5558 pub fn duplicate_line(&mut self, upwards: bool, cx: &mut ViewContext<Self>) {
5559 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5560 let buffer = &display_map.buffer_snapshot;
5561 let selections = self.selections.all::<Point>(cx);
5562
5563 let mut edits = Vec::new();
5564 let mut selections_iter = selections.iter().peekable();
5565 while let Some(selection) = selections_iter.next() {
5566 // Avoid duplicating the same lines twice.
5567 let mut rows = selection.spanned_rows(false, &display_map);
5568
5569 while let Some(next_selection) = selections_iter.peek() {
5570 let next_rows = next_selection.spanned_rows(false, &display_map);
5571 if next_rows.start < rows.end {
5572 rows.end = next_rows.end;
5573 selections_iter.next().unwrap();
5574 } else {
5575 break;
5576 }
5577 }
5578
5579 // Copy the text from the selected row region and splice it either at the start
5580 // or end of the region.
5581 let start = Point::new(rows.start, 0);
5582 let end = Point::new(rows.end - 1, buffer.line_len(rows.end - 1));
5583 let text = buffer
5584 .text_for_range(start..end)
5585 .chain(Some("\n"))
5586 .collect::<String>();
5587 let insert_location = if upwards {
5588 Point::new(rows.end, 0)
5589 } else {
5590 start
5591 };
5592 edits.push((insert_location..insert_location, text));
5593 }
5594
5595 self.transact(cx, |this, cx| {
5596 this.buffer.update(cx, |buffer, cx| {
5597 buffer.edit(edits, None, cx);
5598 });
5599
5600 this.request_autoscroll(Autoscroll::fit(), cx);
5601 });
5602 }
5603
5604 pub fn duplicate_line_up(&mut self, _: &DuplicateLineUp, cx: &mut ViewContext<Self>) {
5605 self.duplicate_line(true, cx);
5606 }
5607
5608 pub fn duplicate_line_down(&mut self, _: &DuplicateLineDown, cx: &mut ViewContext<Self>) {
5609 self.duplicate_line(false, cx);
5610 }
5611
5612 pub fn move_line_up(&mut self, _: &MoveLineUp, cx: &mut ViewContext<Self>) {
5613 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5614 let buffer = self.buffer.read(cx).snapshot(cx);
5615
5616 let mut edits = Vec::new();
5617 let mut unfold_ranges = Vec::new();
5618 let mut refold_ranges = Vec::new();
5619
5620 let selections = self.selections.all::<Point>(cx);
5621 let mut selections = selections.iter().peekable();
5622 let mut contiguous_row_selections = Vec::new();
5623 let mut new_selections = Vec::new();
5624
5625 while let Some(selection) = selections.next() {
5626 // Find all the selections that span a contiguous row range
5627 let (start_row, end_row) = consume_contiguous_rows(
5628 &mut contiguous_row_selections,
5629 selection,
5630 &display_map,
5631 &mut selections,
5632 );
5633
5634 // Move the text spanned by the row range to be before the line preceding the row range
5635 if start_row > 0 {
5636 let range_to_move = Point::new(start_row - 1, buffer.line_len(start_row - 1))
5637 ..Point::new(end_row - 1, buffer.line_len(end_row - 1));
5638 let insertion_point = display_map
5639 .prev_line_boundary(Point::new(start_row - 1, 0))
5640 .0;
5641
5642 // Don't move lines across excerpts
5643 if buffer
5644 .excerpt_boundaries_in_range((
5645 Bound::Excluded(insertion_point),
5646 Bound::Included(range_to_move.end),
5647 ))
5648 .next()
5649 .is_none()
5650 {
5651 let text = buffer
5652 .text_for_range(range_to_move.clone())
5653 .flat_map(|s| s.chars())
5654 .skip(1)
5655 .chain(['\n'])
5656 .collect::<String>();
5657
5658 edits.push((
5659 buffer.anchor_after(range_to_move.start)
5660 ..buffer.anchor_before(range_to_move.end),
5661 String::new(),
5662 ));
5663 let insertion_anchor = buffer.anchor_after(insertion_point);
5664 edits.push((insertion_anchor..insertion_anchor, text));
5665
5666 let row_delta = range_to_move.start.row - insertion_point.row + 1;
5667
5668 // Move selections up
5669 new_selections.extend(contiguous_row_selections.drain(..).map(
5670 |mut selection| {
5671 selection.start.row -= row_delta;
5672 selection.end.row -= row_delta;
5673 selection
5674 },
5675 ));
5676
5677 // Move folds up
5678 unfold_ranges.push(range_to_move.clone());
5679 for fold in display_map.folds_in_range(
5680 buffer.anchor_before(range_to_move.start)
5681 ..buffer.anchor_after(range_to_move.end),
5682 ) {
5683 let mut start = fold.range.start.to_point(&buffer);
5684 let mut end = fold.range.end.to_point(&buffer);
5685 start.row -= row_delta;
5686 end.row -= row_delta;
5687 refold_ranges.push(start..end);
5688 }
5689 }
5690 }
5691
5692 // If we didn't move line(s), preserve the existing selections
5693 new_selections.append(&mut contiguous_row_selections);
5694 }
5695
5696 self.transact(cx, |this, cx| {
5697 this.unfold_ranges(unfold_ranges, true, true, cx);
5698 this.buffer.update(cx, |buffer, cx| {
5699 for (range, text) in edits {
5700 buffer.edit([(range, text)], None, cx);
5701 }
5702 });
5703 this.fold_ranges(refold_ranges, true, cx);
5704 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5705 s.select(new_selections);
5706 })
5707 });
5708 }
5709
5710 pub fn move_line_down(&mut self, _: &MoveLineDown, cx: &mut ViewContext<Self>) {
5711 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
5712 let buffer = self.buffer.read(cx).snapshot(cx);
5713
5714 let mut edits = Vec::new();
5715 let mut unfold_ranges = Vec::new();
5716 let mut refold_ranges = Vec::new();
5717
5718 let selections = self.selections.all::<Point>(cx);
5719 let mut selections = selections.iter().peekable();
5720 let mut contiguous_row_selections = Vec::new();
5721 let mut new_selections = Vec::new();
5722
5723 while let Some(selection) = selections.next() {
5724 // Find all the selections that span a contiguous row range
5725 let (start_row, end_row) = consume_contiguous_rows(
5726 &mut contiguous_row_selections,
5727 selection,
5728 &display_map,
5729 &mut selections,
5730 );
5731
5732 // Move the text spanned by the row range to be after the last line of the row range
5733 if end_row <= buffer.max_point().row {
5734 let range_to_move = Point::new(start_row, 0)..Point::new(end_row, 0);
5735 let insertion_point = display_map.next_line_boundary(Point::new(end_row, 0)).0;
5736
5737 // Don't move lines across excerpt boundaries
5738 if buffer
5739 .excerpt_boundaries_in_range((
5740 Bound::Excluded(range_to_move.start),
5741 Bound::Included(insertion_point),
5742 ))
5743 .next()
5744 .is_none()
5745 {
5746 let mut text = String::from("\n");
5747 text.extend(buffer.text_for_range(range_to_move.clone()));
5748 text.pop(); // Drop trailing newline
5749 edits.push((
5750 buffer.anchor_after(range_to_move.start)
5751 ..buffer.anchor_before(range_to_move.end),
5752 String::new(),
5753 ));
5754 let insertion_anchor = buffer.anchor_after(insertion_point);
5755 edits.push((insertion_anchor..insertion_anchor, text));
5756
5757 let row_delta = insertion_point.row - range_to_move.end.row + 1;
5758
5759 // Move selections down
5760 new_selections.extend(contiguous_row_selections.drain(..).map(
5761 |mut selection| {
5762 selection.start.row += row_delta;
5763 selection.end.row += row_delta;
5764 selection
5765 },
5766 ));
5767
5768 // Move folds down
5769 unfold_ranges.push(range_to_move.clone());
5770 for fold in display_map.folds_in_range(
5771 buffer.anchor_before(range_to_move.start)
5772 ..buffer.anchor_after(range_to_move.end),
5773 ) {
5774 let mut start = fold.range.start.to_point(&buffer);
5775 let mut end = fold.range.end.to_point(&buffer);
5776 start.row += row_delta;
5777 end.row += row_delta;
5778 refold_ranges.push(start..end);
5779 }
5780 }
5781 }
5782
5783 // If we didn't move line(s), preserve the existing selections
5784 new_selections.append(&mut contiguous_row_selections);
5785 }
5786
5787 self.transact(cx, |this, cx| {
5788 this.unfold_ranges(unfold_ranges, true, true, cx);
5789 this.buffer.update(cx, |buffer, cx| {
5790 for (range, text) in edits {
5791 buffer.edit([(range, text)], None, cx);
5792 }
5793 });
5794 this.fold_ranges(refold_ranges, true, cx);
5795 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(new_selections));
5796 });
5797 }
5798
5799 pub fn transpose(&mut self, _: &Transpose, cx: &mut ViewContext<Self>) {
5800 let text_layout_details = &self.text_layout_details(cx);
5801 self.transact(cx, |this, cx| {
5802 let edits = this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5803 let mut edits: Vec<(Range<usize>, String)> = Default::default();
5804 let line_mode = s.line_mode;
5805 s.move_with(|display_map, selection| {
5806 if !selection.is_empty() || line_mode {
5807 return;
5808 }
5809
5810 let mut head = selection.head();
5811 let mut transpose_offset = head.to_offset(display_map, Bias::Right);
5812 if head.column() == display_map.line_len(head.row()) {
5813 transpose_offset = display_map
5814 .buffer_snapshot
5815 .clip_offset(transpose_offset.saturating_sub(1), Bias::Left);
5816 }
5817
5818 if transpose_offset == 0 {
5819 return;
5820 }
5821
5822 *head.column_mut() += 1;
5823 head = display_map.clip_point(head, Bias::Right);
5824 let goal = SelectionGoal::HorizontalPosition(
5825 display_map
5826 .x_for_display_point(head, &text_layout_details)
5827 .into(),
5828 );
5829 selection.collapse_to(head, goal);
5830
5831 let transpose_start = display_map
5832 .buffer_snapshot
5833 .clip_offset(transpose_offset.saturating_sub(1), Bias::Left);
5834 if edits.last().map_or(true, |e| e.0.end <= transpose_start) {
5835 let transpose_end = display_map
5836 .buffer_snapshot
5837 .clip_offset(transpose_offset + 1, Bias::Right);
5838 if let Some(ch) =
5839 display_map.buffer_snapshot.chars_at(transpose_start).next()
5840 {
5841 edits.push((transpose_start..transpose_offset, String::new()));
5842 edits.push((transpose_end..transpose_end, ch.to_string()));
5843 }
5844 }
5845 });
5846 edits
5847 });
5848 this.buffer
5849 .update(cx, |buffer, cx| buffer.edit(edits, None, cx));
5850 let selections = this.selections.all::<usize>(cx);
5851 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5852 s.select(selections);
5853 });
5854 });
5855 }
5856
5857 pub fn cut(&mut self, _: &Cut, cx: &mut ViewContext<Self>) {
5858 let mut text = String::new();
5859 let buffer = self.buffer.read(cx).snapshot(cx);
5860 let mut selections = self.selections.all::<Point>(cx);
5861 let mut clipboard_selections = Vec::with_capacity(selections.len());
5862 {
5863 let max_point = buffer.max_point();
5864 let mut is_first = true;
5865 for selection in &mut selections {
5866 let is_entire_line = selection.is_empty() || self.selections.line_mode;
5867 if is_entire_line {
5868 selection.start = Point::new(selection.start.row, 0);
5869 selection.end = cmp::min(max_point, Point::new(selection.end.row + 1, 0));
5870 selection.goal = SelectionGoal::None;
5871 }
5872 if is_first {
5873 is_first = false;
5874 } else {
5875 text += "\n";
5876 }
5877 let mut len = 0;
5878 for chunk in buffer.text_for_range(selection.start..selection.end) {
5879 text.push_str(chunk);
5880 len += chunk.len();
5881 }
5882 clipboard_selections.push(ClipboardSelection {
5883 len,
5884 is_entire_line,
5885 first_line_indent: buffer.indent_size_for_line(selection.start.row).len,
5886 });
5887 }
5888 }
5889
5890 self.transact(cx, |this, cx| {
5891 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
5892 s.select(selections);
5893 });
5894 this.insert("", cx);
5895 cx.write_to_clipboard(ClipboardItem::new(text).with_metadata(clipboard_selections));
5896 });
5897 }
5898
5899 pub fn copy(&mut self, _: &Copy, cx: &mut ViewContext<Self>) {
5900 let selections = self.selections.all::<Point>(cx);
5901 let buffer = self.buffer.read(cx).read(cx);
5902 let mut text = String::new();
5903
5904 let mut clipboard_selections = Vec::with_capacity(selections.len());
5905 {
5906 let max_point = buffer.max_point();
5907 let mut is_first = true;
5908 for selection in selections.iter() {
5909 let mut start = selection.start;
5910 let mut end = selection.end;
5911 let is_entire_line = selection.is_empty() || self.selections.line_mode;
5912 if is_entire_line {
5913 start = Point::new(start.row, 0);
5914 end = cmp::min(max_point, Point::new(end.row + 1, 0));
5915 }
5916 if is_first {
5917 is_first = false;
5918 } else {
5919 text += "\n";
5920 }
5921 let mut len = 0;
5922 for chunk in buffer.text_for_range(start..end) {
5923 text.push_str(chunk);
5924 len += chunk.len();
5925 }
5926 clipboard_selections.push(ClipboardSelection {
5927 len,
5928 is_entire_line,
5929 first_line_indent: buffer.indent_size_for_line(start.row).len,
5930 });
5931 }
5932 }
5933
5934 cx.write_to_clipboard(ClipboardItem::new(text).with_metadata(clipboard_selections));
5935 }
5936
5937 pub fn paste(&mut self, _: &Paste, cx: &mut ViewContext<Self>) {
5938 if self.read_only(cx) {
5939 return;
5940 }
5941
5942 self.transact(cx, |this, cx| {
5943 if let Some(item) = cx.read_from_clipboard() {
5944 let clipboard_text = Cow::Borrowed(item.text());
5945 if let Some(mut clipboard_selections) = item.metadata::<Vec<ClipboardSelection>>() {
5946 let old_selections = this.selections.all::<usize>(cx);
5947 let all_selections_were_entire_line =
5948 clipboard_selections.iter().all(|s| s.is_entire_line);
5949 let first_selection_indent_column =
5950 clipboard_selections.first().map(|s| s.first_line_indent);
5951 if clipboard_selections.len() != old_selections.len() {
5952 clipboard_selections.drain(..);
5953 }
5954
5955 this.buffer.update(cx, |buffer, cx| {
5956 let snapshot = buffer.read(cx);
5957 let mut start_offset = 0;
5958 let mut edits = Vec::new();
5959 let mut original_indent_columns = Vec::new();
5960 let line_mode = this.selections.line_mode;
5961 for (ix, selection) in old_selections.iter().enumerate() {
5962 let to_insert;
5963 let entire_line;
5964 let original_indent_column;
5965 if let Some(clipboard_selection) = clipboard_selections.get(ix) {
5966 let end_offset = start_offset + clipboard_selection.len;
5967 to_insert = &clipboard_text[start_offset..end_offset];
5968 entire_line = clipboard_selection.is_entire_line;
5969 start_offset = end_offset + 1;
5970 original_indent_column =
5971 Some(clipboard_selection.first_line_indent);
5972 } else {
5973 to_insert = clipboard_text.as_str();
5974 entire_line = all_selections_were_entire_line;
5975 original_indent_column = first_selection_indent_column
5976 }
5977
5978 // If the corresponding selection was empty when this slice of the
5979 // clipboard text was written, then the entire line containing the
5980 // selection was copied. If this selection is also currently empty,
5981 // then paste the line before the current line of the buffer.
5982 let range = if selection.is_empty() && !line_mode && entire_line {
5983 let column = selection.start.to_point(&snapshot).column as usize;
5984 let line_start = selection.start - column;
5985 line_start..line_start
5986 } else {
5987 selection.range()
5988 };
5989
5990 edits.push((range, to_insert));
5991 original_indent_columns.extend(original_indent_column);
5992 }
5993 drop(snapshot);
5994
5995 buffer.edit(
5996 edits,
5997 Some(AutoindentMode::Block {
5998 original_indent_columns,
5999 }),
6000 cx,
6001 );
6002 });
6003
6004 let selections = this.selections.all::<usize>(cx);
6005 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(selections));
6006 } else {
6007 this.insert(&clipboard_text, cx);
6008 }
6009 }
6010 });
6011 }
6012
6013 pub fn undo(&mut self, _: &Undo, cx: &mut ViewContext<Self>) {
6014 if self.read_only(cx) {
6015 return;
6016 }
6017
6018 if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.undo(cx)) {
6019 if let Some((selections, _)) = self.selection_history.transaction(tx_id).cloned() {
6020 self.change_selections(None, cx, |s| {
6021 s.select_anchors(selections.to_vec());
6022 });
6023 }
6024 self.request_autoscroll(Autoscroll::fit(), cx);
6025 self.unmark_text(cx);
6026 self.refresh_inline_completion(true, cx);
6027 cx.emit(EditorEvent::Edited);
6028 cx.emit(EditorEvent::TransactionUndone {
6029 transaction_id: tx_id,
6030 });
6031 }
6032 }
6033
6034 pub fn redo(&mut self, _: &Redo, cx: &mut ViewContext<Self>) {
6035 if self.read_only(cx) {
6036 return;
6037 }
6038
6039 if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.redo(cx)) {
6040 if let Some((_, Some(selections))) = self.selection_history.transaction(tx_id).cloned()
6041 {
6042 self.change_selections(None, cx, |s| {
6043 s.select_anchors(selections.to_vec());
6044 });
6045 }
6046 self.request_autoscroll(Autoscroll::fit(), cx);
6047 self.unmark_text(cx);
6048 self.refresh_inline_completion(true, cx);
6049 cx.emit(EditorEvent::Edited);
6050 }
6051 }
6052
6053 pub fn finalize_last_transaction(&mut self, cx: &mut ViewContext<Self>) {
6054 self.buffer
6055 .update(cx, |buffer, cx| buffer.finalize_last_transaction(cx));
6056 }
6057
6058 pub fn group_until_transaction(&mut self, tx_id: TransactionId, cx: &mut ViewContext<Self>) {
6059 self.buffer
6060 .update(cx, |buffer, cx| buffer.group_until_transaction(tx_id, cx));
6061 }
6062
6063 pub fn move_left(&mut self, _: &MoveLeft, cx: &mut ViewContext<Self>) {
6064 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6065 let line_mode = s.line_mode;
6066 s.move_with(|map, selection| {
6067 let cursor = if selection.is_empty() && !line_mode {
6068 movement::left(map, selection.start)
6069 } else {
6070 selection.start
6071 };
6072 selection.collapse_to(cursor, SelectionGoal::None);
6073 });
6074 })
6075 }
6076
6077 pub fn select_left(&mut self, _: &SelectLeft, cx: &mut ViewContext<Self>) {
6078 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6079 s.move_heads_with(|map, head, _| (movement::left(map, head), SelectionGoal::None));
6080 })
6081 }
6082
6083 pub fn move_right(&mut self, _: &MoveRight, cx: &mut ViewContext<Self>) {
6084 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6085 let line_mode = s.line_mode;
6086 s.move_with(|map, selection| {
6087 let cursor = if selection.is_empty() && !line_mode {
6088 movement::right(map, selection.end)
6089 } else {
6090 selection.end
6091 };
6092 selection.collapse_to(cursor, SelectionGoal::None)
6093 });
6094 })
6095 }
6096
6097 pub fn select_right(&mut self, _: &SelectRight, cx: &mut ViewContext<Self>) {
6098 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6099 s.move_heads_with(|map, head, _| (movement::right(map, head), SelectionGoal::None));
6100 })
6101 }
6102
6103 pub fn move_up(&mut self, _: &MoveUp, cx: &mut ViewContext<Self>) {
6104 if self.take_rename(true, cx).is_some() {
6105 return;
6106 }
6107
6108 if matches!(self.mode, EditorMode::SingleLine) {
6109 cx.propagate();
6110 return;
6111 }
6112
6113 let text_layout_details = &self.text_layout_details(cx);
6114
6115 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6116 let line_mode = s.line_mode;
6117 s.move_with(|map, selection| {
6118 if !selection.is_empty() && !line_mode {
6119 selection.goal = SelectionGoal::None;
6120 }
6121 let (cursor, goal) = movement::up(
6122 map,
6123 selection.start,
6124 selection.goal,
6125 false,
6126 &text_layout_details,
6127 );
6128 selection.collapse_to(cursor, goal);
6129 });
6130 })
6131 }
6132
6133 pub fn move_up_by_lines(&mut self, action: &MoveUpByLines, cx: &mut ViewContext<Self>) {
6134 if self.take_rename(true, cx).is_some() {
6135 return;
6136 }
6137
6138 if matches!(self.mode, EditorMode::SingleLine) {
6139 cx.propagate();
6140 return;
6141 }
6142
6143 let text_layout_details = &self.text_layout_details(cx);
6144
6145 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6146 let line_mode = s.line_mode;
6147 s.move_with(|map, selection| {
6148 if !selection.is_empty() && !line_mode {
6149 selection.goal = SelectionGoal::None;
6150 }
6151 let (cursor, goal) = movement::up_by_rows(
6152 map,
6153 selection.start,
6154 action.lines,
6155 selection.goal,
6156 false,
6157 &text_layout_details,
6158 );
6159 selection.collapse_to(cursor, goal);
6160 });
6161 })
6162 }
6163
6164 pub fn move_down_by_lines(&mut self, action: &MoveDownByLines, cx: &mut ViewContext<Self>) {
6165 if self.take_rename(true, cx).is_some() {
6166 return;
6167 }
6168
6169 if matches!(self.mode, EditorMode::SingleLine) {
6170 cx.propagate();
6171 return;
6172 }
6173
6174 let text_layout_details = &self.text_layout_details(cx);
6175
6176 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6177 let line_mode = s.line_mode;
6178 s.move_with(|map, selection| {
6179 if !selection.is_empty() && !line_mode {
6180 selection.goal = SelectionGoal::None;
6181 }
6182 let (cursor, goal) = movement::down_by_rows(
6183 map,
6184 selection.start,
6185 action.lines,
6186 selection.goal,
6187 false,
6188 &text_layout_details,
6189 );
6190 selection.collapse_to(cursor, goal);
6191 });
6192 })
6193 }
6194
6195 pub fn select_down_by_lines(&mut self, action: &SelectDownByLines, cx: &mut ViewContext<Self>) {
6196 let text_layout_details = &self.text_layout_details(cx);
6197 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6198 s.move_heads_with(|map, head, goal| {
6199 movement::down_by_rows(map, head, action.lines, goal, false, &text_layout_details)
6200 })
6201 })
6202 }
6203
6204 pub fn select_up_by_lines(&mut self, action: &SelectUpByLines, cx: &mut ViewContext<Self>) {
6205 let text_layout_details = &self.text_layout_details(cx);
6206 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6207 s.move_heads_with(|map, head, goal| {
6208 movement::up_by_rows(map, head, action.lines, goal, false, &text_layout_details)
6209 })
6210 })
6211 }
6212
6213 pub fn move_page_up(&mut self, action: &MovePageUp, cx: &mut ViewContext<Self>) {
6214 if self.take_rename(true, cx).is_some() {
6215 return;
6216 }
6217
6218 if matches!(self.mode, EditorMode::SingleLine) {
6219 cx.propagate();
6220 return;
6221 }
6222
6223 let row_count = if let Some(row_count) = self.visible_line_count() {
6224 row_count as u32 - 1
6225 } else {
6226 return;
6227 };
6228
6229 let autoscroll = if action.center_cursor {
6230 Autoscroll::center()
6231 } else {
6232 Autoscroll::fit()
6233 };
6234
6235 let text_layout_details = &self.text_layout_details(cx);
6236
6237 self.change_selections(Some(autoscroll), cx, |s| {
6238 let line_mode = s.line_mode;
6239 s.move_with(|map, selection| {
6240 if !selection.is_empty() && !line_mode {
6241 selection.goal = SelectionGoal::None;
6242 }
6243 let (cursor, goal) = movement::up_by_rows(
6244 map,
6245 selection.end,
6246 row_count,
6247 selection.goal,
6248 false,
6249 &text_layout_details,
6250 );
6251 selection.collapse_to(cursor, goal);
6252 });
6253 });
6254 }
6255
6256 pub fn select_up(&mut self, _: &SelectUp, cx: &mut ViewContext<Self>) {
6257 let text_layout_details = &self.text_layout_details(cx);
6258 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6259 s.move_heads_with(|map, head, goal| {
6260 movement::up(map, head, goal, false, &text_layout_details)
6261 })
6262 })
6263 }
6264
6265 pub fn move_down(&mut self, _: &MoveDown, cx: &mut ViewContext<Self>) {
6266 self.take_rename(true, cx);
6267
6268 if self.mode == EditorMode::SingleLine {
6269 cx.propagate();
6270 return;
6271 }
6272
6273 let text_layout_details = &self.text_layout_details(cx);
6274 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6275 let line_mode = s.line_mode;
6276 s.move_with(|map, selection| {
6277 if !selection.is_empty() && !line_mode {
6278 selection.goal = SelectionGoal::None;
6279 }
6280 let (cursor, goal) = movement::down(
6281 map,
6282 selection.end,
6283 selection.goal,
6284 false,
6285 &text_layout_details,
6286 );
6287 selection.collapse_to(cursor, goal);
6288 });
6289 });
6290 }
6291
6292 pub fn move_page_down(&mut self, action: &MovePageDown, cx: &mut ViewContext<Self>) {
6293 if self.take_rename(true, cx).is_some() {
6294 return;
6295 }
6296
6297 if self
6298 .context_menu
6299 .write()
6300 .as_mut()
6301 .map(|menu| menu.select_last(self.project.as_ref(), cx))
6302 .unwrap_or(false)
6303 {
6304 return;
6305 }
6306
6307 if matches!(self.mode, EditorMode::SingleLine) {
6308 cx.propagate();
6309 return;
6310 }
6311
6312 let row_count = if let Some(row_count) = self.visible_line_count() {
6313 row_count as u32 - 1
6314 } else {
6315 return;
6316 };
6317
6318 let autoscroll = if action.center_cursor {
6319 Autoscroll::center()
6320 } else {
6321 Autoscroll::fit()
6322 };
6323
6324 let text_layout_details = &self.text_layout_details(cx);
6325 self.change_selections(Some(autoscroll), cx, |s| {
6326 let line_mode = s.line_mode;
6327 s.move_with(|map, selection| {
6328 if !selection.is_empty() && !line_mode {
6329 selection.goal = SelectionGoal::None;
6330 }
6331 let (cursor, goal) = movement::down_by_rows(
6332 map,
6333 selection.end,
6334 row_count,
6335 selection.goal,
6336 false,
6337 &text_layout_details,
6338 );
6339 selection.collapse_to(cursor, goal);
6340 });
6341 });
6342 }
6343
6344 pub fn select_down(&mut self, _: &SelectDown, cx: &mut ViewContext<Self>) {
6345 let text_layout_details = &self.text_layout_details(cx);
6346 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6347 s.move_heads_with(|map, head, goal| {
6348 movement::down(map, head, goal, false, &text_layout_details)
6349 })
6350 });
6351 }
6352
6353 pub fn context_menu_first(&mut self, _: &ContextMenuFirst, cx: &mut ViewContext<Self>) {
6354 if let Some(context_menu) = self.context_menu.write().as_mut() {
6355 context_menu.select_first(self.project.as_ref(), cx);
6356 }
6357 }
6358
6359 pub fn context_menu_prev(&mut self, _: &ContextMenuPrev, cx: &mut ViewContext<Self>) {
6360 if let Some(context_menu) = self.context_menu.write().as_mut() {
6361 context_menu.select_prev(self.project.as_ref(), cx);
6362 }
6363 }
6364
6365 pub fn context_menu_next(&mut self, _: &ContextMenuNext, cx: &mut ViewContext<Self>) {
6366 if let Some(context_menu) = self.context_menu.write().as_mut() {
6367 context_menu.select_next(self.project.as_ref(), cx);
6368 }
6369 }
6370
6371 pub fn context_menu_last(&mut self, _: &ContextMenuLast, cx: &mut ViewContext<Self>) {
6372 if let Some(context_menu) = self.context_menu.write().as_mut() {
6373 context_menu.select_last(self.project.as_ref(), cx);
6374 }
6375 }
6376
6377 pub fn move_to_previous_word_start(
6378 &mut self,
6379 _: &MoveToPreviousWordStart,
6380 cx: &mut ViewContext<Self>,
6381 ) {
6382 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6383 s.move_cursors_with(|map, head, _| {
6384 (
6385 movement::previous_word_start(map, head),
6386 SelectionGoal::None,
6387 )
6388 });
6389 })
6390 }
6391
6392 pub fn move_to_previous_subword_start(
6393 &mut self,
6394 _: &MoveToPreviousSubwordStart,
6395 cx: &mut ViewContext<Self>,
6396 ) {
6397 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6398 s.move_cursors_with(|map, head, _| {
6399 (
6400 movement::previous_subword_start(map, head),
6401 SelectionGoal::None,
6402 )
6403 });
6404 })
6405 }
6406
6407 pub fn select_to_previous_word_start(
6408 &mut self,
6409 _: &SelectToPreviousWordStart,
6410 cx: &mut ViewContext<Self>,
6411 ) {
6412 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6413 s.move_heads_with(|map, head, _| {
6414 (
6415 movement::previous_word_start(map, head),
6416 SelectionGoal::None,
6417 )
6418 });
6419 })
6420 }
6421
6422 pub fn select_to_previous_subword_start(
6423 &mut self,
6424 _: &SelectToPreviousSubwordStart,
6425 cx: &mut ViewContext<Self>,
6426 ) {
6427 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6428 s.move_heads_with(|map, head, _| {
6429 (
6430 movement::previous_subword_start(map, head),
6431 SelectionGoal::None,
6432 )
6433 });
6434 })
6435 }
6436
6437 pub fn delete_to_previous_word_start(
6438 &mut self,
6439 _: &DeleteToPreviousWordStart,
6440 cx: &mut ViewContext<Self>,
6441 ) {
6442 self.transact(cx, |this, cx| {
6443 this.select_autoclose_pair(cx);
6444 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
6445 let line_mode = s.line_mode;
6446 s.move_with(|map, selection| {
6447 if selection.is_empty() && !line_mode {
6448 let cursor = movement::previous_word_start(map, selection.head());
6449 selection.set_head(cursor, SelectionGoal::None);
6450 }
6451 });
6452 });
6453 this.insert("", cx);
6454 });
6455 }
6456
6457 pub fn delete_to_previous_subword_start(
6458 &mut self,
6459 _: &DeleteToPreviousSubwordStart,
6460 cx: &mut ViewContext<Self>,
6461 ) {
6462 self.transact(cx, |this, cx| {
6463 this.select_autoclose_pair(cx);
6464 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
6465 let line_mode = s.line_mode;
6466 s.move_with(|map, selection| {
6467 if selection.is_empty() && !line_mode {
6468 let cursor = movement::previous_subword_start(map, selection.head());
6469 selection.set_head(cursor, SelectionGoal::None);
6470 }
6471 });
6472 });
6473 this.insert("", cx);
6474 });
6475 }
6476
6477 pub fn move_to_next_word_end(&mut self, _: &MoveToNextWordEnd, cx: &mut ViewContext<Self>) {
6478 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6479 s.move_cursors_with(|map, head, _| {
6480 (movement::next_word_end(map, head), SelectionGoal::None)
6481 });
6482 })
6483 }
6484
6485 pub fn move_to_next_subword_end(
6486 &mut self,
6487 _: &MoveToNextSubwordEnd,
6488 cx: &mut ViewContext<Self>,
6489 ) {
6490 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6491 s.move_cursors_with(|map, head, _| {
6492 (movement::next_subword_end(map, head), SelectionGoal::None)
6493 });
6494 })
6495 }
6496
6497 pub fn select_to_next_word_end(&mut self, _: &SelectToNextWordEnd, cx: &mut ViewContext<Self>) {
6498 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6499 s.move_heads_with(|map, head, _| {
6500 (movement::next_word_end(map, head), SelectionGoal::None)
6501 });
6502 })
6503 }
6504
6505 pub fn select_to_next_subword_end(
6506 &mut self,
6507 _: &SelectToNextSubwordEnd,
6508 cx: &mut ViewContext<Self>,
6509 ) {
6510 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6511 s.move_heads_with(|map, head, _| {
6512 (movement::next_subword_end(map, head), SelectionGoal::None)
6513 });
6514 })
6515 }
6516
6517 pub fn delete_to_next_word_end(&mut self, _: &DeleteToNextWordEnd, cx: &mut ViewContext<Self>) {
6518 self.transact(cx, |this, cx| {
6519 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
6520 let line_mode = s.line_mode;
6521 s.move_with(|map, selection| {
6522 if selection.is_empty() && !line_mode {
6523 let cursor = movement::next_word_end(map, selection.head());
6524 selection.set_head(cursor, SelectionGoal::None);
6525 }
6526 });
6527 });
6528 this.insert("", cx);
6529 });
6530 }
6531
6532 pub fn delete_to_next_subword_end(
6533 &mut self,
6534 _: &DeleteToNextSubwordEnd,
6535 cx: &mut ViewContext<Self>,
6536 ) {
6537 self.transact(cx, |this, cx| {
6538 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
6539 s.move_with(|map, selection| {
6540 if selection.is_empty() {
6541 let cursor = movement::next_subword_end(map, selection.head());
6542 selection.set_head(cursor, SelectionGoal::None);
6543 }
6544 });
6545 });
6546 this.insert("", cx);
6547 });
6548 }
6549
6550 pub fn move_to_beginning_of_line(
6551 &mut self,
6552 action: &MoveToBeginningOfLine,
6553 cx: &mut ViewContext<Self>,
6554 ) {
6555 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6556 s.move_cursors_with(|map, head, _| {
6557 (
6558 movement::indented_line_beginning(map, head, action.stop_at_soft_wraps),
6559 SelectionGoal::None,
6560 )
6561 });
6562 })
6563 }
6564
6565 pub fn select_to_beginning_of_line(
6566 &mut self,
6567 action: &SelectToBeginningOfLine,
6568 cx: &mut ViewContext<Self>,
6569 ) {
6570 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6571 s.move_heads_with(|map, head, _| {
6572 (
6573 movement::indented_line_beginning(map, head, action.stop_at_soft_wraps),
6574 SelectionGoal::None,
6575 )
6576 });
6577 });
6578 }
6579
6580 pub fn delete_to_beginning_of_line(
6581 &mut self,
6582 _: &DeleteToBeginningOfLine,
6583 cx: &mut ViewContext<Self>,
6584 ) {
6585 self.transact(cx, |this, cx| {
6586 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
6587 s.move_with(|_, selection| {
6588 selection.reversed = true;
6589 });
6590 });
6591
6592 this.select_to_beginning_of_line(
6593 &SelectToBeginningOfLine {
6594 stop_at_soft_wraps: false,
6595 },
6596 cx,
6597 );
6598 this.backspace(&Backspace, cx);
6599 });
6600 }
6601
6602 pub fn move_to_end_of_line(&mut self, action: &MoveToEndOfLine, cx: &mut ViewContext<Self>) {
6603 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6604 s.move_cursors_with(|map, head, _| {
6605 (
6606 movement::line_end(map, head, action.stop_at_soft_wraps),
6607 SelectionGoal::None,
6608 )
6609 });
6610 })
6611 }
6612
6613 pub fn select_to_end_of_line(
6614 &mut self,
6615 action: &SelectToEndOfLine,
6616 cx: &mut ViewContext<Self>,
6617 ) {
6618 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6619 s.move_heads_with(|map, head, _| {
6620 (
6621 movement::line_end(map, head, action.stop_at_soft_wraps),
6622 SelectionGoal::None,
6623 )
6624 });
6625 })
6626 }
6627
6628 pub fn delete_to_end_of_line(&mut self, _: &DeleteToEndOfLine, cx: &mut ViewContext<Self>) {
6629 self.transact(cx, |this, cx| {
6630 this.select_to_end_of_line(
6631 &SelectToEndOfLine {
6632 stop_at_soft_wraps: false,
6633 },
6634 cx,
6635 );
6636 this.delete(&Delete, cx);
6637 });
6638 }
6639
6640 pub fn cut_to_end_of_line(&mut self, _: &CutToEndOfLine, cx: &mut ViewContext<Self>) {
6641 self.transact(cx, |this, cx| {
6642 this.select_to_end_of_line(
6643 &SelectToEndOfLine {
6644 stop_at_soft_wraps: false,
6645 },
6646 cx,
6647 );
6648 this.cut(&Cut, cx);
6649 });
6650 }
6651
6652 pub fn move_to_start_of_paragraph(
6653 &mut self,
6654 _: &MoveToStartOfParagraph,
6655 cx: &mut ViewContext<Self>,
6656 ) {
6657 if matches!(self.mode, EditorMode::SingleLine) {
6658 cx.propagate();
6659 return;
6660 }
6661
6662 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6663 s.move_with(|map, selection| {
6664 selection.collapse_to(
6665 movement::start_of_paragraph(map, selection.head(), 1),
6666 SelectionGoal::None,
6667 )
6668 });
6669 })
6670 }
6671
6672 pub fn move_to_end_of_paragraph(
6673 &mut self,
6674 _: &MoveToEndOfParagraph,
6675 cx: &mut ViewContext<Self>,
6676 ) {
6677 if matches!(self.mode, EditorMode::SingleLine) {
6678 cx.propagate();
6679 return;
6680 }
6681
6682 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6683 s.move_with(|map, selection| {
6684 selection.collapse_to(
6685 movement::end_of_paragraph(map, selection.head(), 1),
6686 SelectionGoal::None,
6687 )
6688 });
6689 })
6690 }
6691
6692 pub fn select_to_start_of_paragraph(
6693 &mut self,
6694 _: &SelectToStartOfParagraph,
6695 cx: &mut ViewContext<Self>,
6696 ) {
6697 if matches!(self.mode, EditorMode::SingleLine) {
6698 cx.propagate();
6699 return;
6700 }
6701
6702 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6703 s.move_heads_with(|map, head, _| {
6704 (
6705 movement::start_of_paragraph(map, head, 1),
6706 SelectionGoal::None,
6707 )
6708 });
6709 })
6710 }
6711
6712 pub fn select_to_end_of_paragraph(
6713 &mut self,
6714 _: &SelectToEndOfParagraph,
6715 cx: &mut ViewContext<Self>,
6716 ) {
6717 if matches!(self.mode, EditorMode::SingleLine) {
6718 cx.propagate();
6719 return;
6720 }
6721
6722 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6723 s.move_heads_with(|map, head, _| {
6724 (
6725 movement::end_of_paragraph(map, head, 1),
6726 SelectionGoal::None,
6727 )
6728 });
6729 })
6730 }
6731
6732 pub fn move_to_beginning(&mut self, _: &MoveToBeginning, cx: &mut ViewContext<Self>) {
6733 if matches!(self.mode, EditorMode::SingleLine) {
6734 cx.propagate();
6735 return;
6736 }
6737
6738 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6739 s.select_ranges(vec![0..0]);
6740 });
6741 }
6742
6743 pub fn select_to_beginning(&mut self, _: &SelectToBeginning, cx: &mut ViewContext<Self>) {
6744 let mut selection = self.selections.last::<Point>(cx);
6745 selection.set_head(Point::zero(), SelectionGoal::None);
6746
6747 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6748 s.select(vec![selection]);
6749 });
6750 }
6751
6752 pub fn move_to_end(&mut self, _: &MoveToEnd, cx: &mut ViewContext<Self>) {
6753 if matches!(self.mode, EditorMode::SingleLine) {
6754 cx.propagate();
6755 return;
6756 }
6757
6758 let cursor = self.buffer.read(cx).read(cx).len();
6759 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6760 s.select_ranges(vec![cursor..cursor])
6761 });
6762 }
6763
6764 pub fn set_nav_history(&mut self, nav_history: Option<ItemNavHistory>) {
6765 self.nav_history = nav_history;
6766 }
6767
6768 pub fn nav_history(&self) -> Option<&ItemNavHistory> {
6769 self.nav_history.as_ref()
6770 }
6771
6772 fn push_to_nav_history(
6773 &mut self,
6774 cursor_anchor: Anchor,
6775 new_position: Option<Point>,
6776 cx: &mut ViewContext<Self>,
6777 ) {
6778 if let Some(nav_history) = self.nav_history.as_mut() {
6779 let buffer = self.buffer.read(cx).read(cx);
6780 let cursor_position = cursor_anchor.to_point(&buffer);
6781 let scroll_state = self.scroll_manager.anchor();
6782 let scroll_top_row = scroll_state.top_row(&buffer);
6783 drop(buffer);
6784
6785 if let Some(new_position) = new_position {
6786 let row_delta = (new_position.row as i64 - cursor_position.row as i64).abs();
6787 if row_delta < MIN_NAVIGATION_HISTORY_ROW_DELTA {
6788 return;
6789 }
6790 }
6791
6792 nav_history.push(
6793 Some(NavigationData {
6794 cursor_anchor,
6795 cursor_position,
6796 scroll_anchor: scroll_state,
6797 scroll_top_row,
6798 }),
6799 cx,
6800 );
6801 }
6802 }
6803
6804 pub fn select_to_end(&mut self, _: &SelectToEnd, cx: &mut ViewContext<Self>) {
6805 let buffer = self.buffer.read(cx).snapshot(cx);
6806 let mut selection = self.selections.first::<usize>(cx);
6807 selection.set_head(buffer.len(), SelectionGoal::None);
6808 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6809 s.select(vec![selection]);
6810 });
6811 }
6812
6813 pub fn select_all(&mut self, _: &SelectAll, cx: &mut ViewContext<Self>) {
6814 let end = self.buffer.read(cx).read(cx).len();
6815 self.change_selections(None, cx, |s| {
6816 s.select_ranges(vec![0..end]);
6817 });
6818 }
6819
6820 pub fn select_line(&mut self, _: &SelectLine, cx: &mut ViewContext<Self>) {
6821 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
6822 let mut selections = self.selections.all::<Point>(cx);
6823 let max_point = display_map.buffer_snapshot.max_point();
6824 for selection in &mut selections {
6825 let rows = selection.spanned_rows(true, &display_map);
6826 selection.start = Point::new(rows.start, 0);
6827 selection.end = cmp::min(max_point, Point::new(rows.end, 0));
6828 selection.reversed = false;
6829 }
6830 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6831 s.select(selections);
6832 });
6833 }
6834
6835 pub fn split_selection_into_lines(
6836 &mut self,
6837 _: &SplitSelectionIntoLines,
6838 cx: &mut ViewContext<Self>,
6839 ) {
6840 let mut to_unfold = Vec::new();
6841 let mut new_selection_ranges = Vec::new();
6842 {
6843 let selections = self.selections.all::<Point>(cx);
6844 let buffer = self.buffer.read(cx).read(cx);
6845 for selection in selections {
6846 for row in selection.start.row..selection.end.row {
6847 let cursor = Point::new(row, buffer.line_len(row));
6848 new_selection_ranges.push(cursor..cursor);
6849 }
6850 new_selection_ranges.push(selection.end..selection.end);
6851 to_unfold.push(selection.start..selection.end);
6852 }
6853 }
6854 self.unfold_ranges(to_unfold, true, true, cx);
6855 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6856 s.select_ranges(new_selection_ranges);
6857 });
6858 }
6859
6860 pub fn add_selection_above(&mut self, _: &AddSelectionAbove, cx: &mut ViewContext<Self>) {
6861 self.add_selection(true, cx);
6862 }
6863
6864 pub fn add_selection_below(&mut self, _: &AddSelectionBelow, cx: &mut ViewContext<Self>) {
6865 self.add_selection(false, cx);
6866 }
6867
6868 fn add_selection(&mut self, above: bool, cx: &mut ViewContext<Self>) {
6869 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
6870 let mut selections = self.selections.all::<Point>(cx);
6871 let text_layout_details = self.text_layout_details(cx);
6872 let mut state = self.add_selections_state.take().unwrap_or_else(|| {
6873 let oldest_selection = selections.iter().min_by_key(|s| s.id).unwrap().clone();
6874 let range = oldest_selection.display_range(&display_map).sorted();
6875
6876 let start_x = display_map.x_for_display_point(range.start, &text_layout_details);
6877 let end_x = display_map.x_for_display_point(range.end, &text_layout_details);
6878 let positions = start_x.min(end_x)..start_x.max(end_x);
6879
6880 selections.clear();
6881 let mut stack = Vec::new();
6882 for row in range.start.row()..=range.end.row() {
6883 if let Some(selection) = self.selections.build_columnar_selection(
6884 &display_map,
6885 row,
6886 &positions,
6887 oldest_selection.reversed,
6888 &text_layout_details,
6889 ) {
6890 stack.push(selection.id);
6891 selections.push(selection);
6892 }
6893 }
6894
6895 if above {
6896 stack.reverse();
6897 }
6898
6899 AddSelectionsState { above, stack }
6900 });
6901
6902 let last_added_selection = *state.stack.last().unwrap();
6903 let mut new_selections = Vec::new();
6904 if above == state.above {
6905 let end_row = if above {
6906 0
6907 } else {
6908 display_map.max_point().row()
6909 };
6910
6911 'outer: for selection in selections {
6912 if selection.id == last_added_selection {
6913 let range = selection.display_range(&display_map).sorted();
6914 debug_assert_eq!(range.start.row(), range.end.row());
6915 let mut row = range.start.row();
6916 let positions =
6917 if let SelectionGoal::HorizontalRange { start, end } = selection.goal {
6918 px(start)..px(end)
6919 } else {
6920 let start_x =
6921 display_map.x_for_display_point(range.start, &text_layout_details);
6922 let end_x =
6923 display_map.x_for_display_point(range.end, &text_layout_details);
6924 start_x.min(end_x)..start_x.max(end_x)
6925 };
6926
6927 while row != end_row {
6928 if above {
6929 row -= 1;
6930 } else {
6931 row += 1;
6932 }
6933
6934 if let Some(new_selection) = self.selections.build_columnar_selection(
6935 &display_map,
6936 row,
6937 &positions,
6938 selection.reversed,
6939 &text_layout_details,
6940 ) {
6941 state.stack.push(new_selection.id);
6942 if above {
6943 new_selections.push(new_selection);
6944 new_selections.push(selection);
6945 } else {
6946 new_selections.push(selection);
6947 new_selections.push(new_selection);
6948 }
6949
6950 continue 'outer;
6951 }
6952 }
6953 }
6954
6955 new_selections.push(selection);
6956 }
6957 } else {
6958 new_selections = selections;
6959 new_selections.retain(|s| s.id != last_added_selection);
6960 state.stack.pop();
6961 }
6962
6963 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
6964 s.select(new_selections);
6965 });
6966 if state.stack.len() > 1 {
6967 self.add_selections_state = Some(state);
6968 }
6969 }
6970
6971 pub fn select_next_match_internal(
6972 &mut self,
6973 display_map: &DisplaySnapshot,
6974 replace_newest: bool,
6975 autoscroll: Option<Autoscroll>,
6976 cx: &mut ViewContext<Self>,
6977 ) -> Result<()> {
6978 fn select_next_match_ranges(
6979 this: &mut Editor,
6980 range: Range<usize>,
6981 replace_newest: bool,
6982 auto_scroll: Option<Autoscroll>,
6983 cx: &mut ViewContext<Editor>,
6984 ) {
6985 this.unfold_ranges([range.clone()], false, true, cx);
6986 this.change_selections(auto_scroll, cx, |s| {
6987 if replace_newest {
6988 s.delete(s.newest_anchor().id);
6989 }
6990 s.insert_range(range.clone());
6991 });
6992 }
6993
6994 let buffer = &display_map.buffer_snapshot;
6995 let mut selections = self.selections.all::<usize>(cx);
6996 if let Some(mut select_next_state) = self.select_next_state.take() {
6997 let query = &select_next_state.query;
6998 if !select_next_state.done {
6999 let first_selection = selections.iter().min_by_key(|s| s.id).unwrap();
7000 let last_selection = selections.iter().max_by_key(|s| s.id).unwrap();
7001 let mut next_selected_range = None;
7002
7003 let bytes_after_last_selection =
7004 buffer.bytes_in_range(last_selection.end..buffer.len());
7005 let bytes_before_first_selection = buffer.bytes_in_range(0..first_selection.start);
7006 let query_matches = query
7007 .stream_find_iter(bytes_after_last_selection)
7008 .map(|result| (last_selection.end, result))
7009 .chain(
7010 query
7011 .stream_find_iter(bytes_before_first_selection)
7012 .map(|result| (0, result)),
7013 );
7014
7015 for (start_offset, query_match) in query_matches {
7016 let query_match = query_match.unwrap(); // can only fail due to I/O
7017 let offset_range =
7018 start_offset + query_match.start()..start_offset + query_match.end();
7019 let display_range = offset_range.start.to_display_point(&display_map)
7020 ..offset_range.end.to_display_point(&display_map);
7021
7022 if !select_next_state.wordwise
7023 || (!movement::is_inside_word(&display_map, display_range.start)
7024 && !movement::is_inside_word(&display_map, display_range.end))
7025 {
7026 // TODO: This is n^2, because we might check all the selections
7027 if !selections
7028 .iter()
7029 .any(|selection| selection.range().overlaps(&offset_range))
7030 {
7031 next_selected_range = Some(offset_range);
7032 break;
7033 }
7034 }
7035 }
7036
7037 if let Some(next_selected_range) = next_selected_range {
7038 select_next_match_ranges(
7039 self,
7040 next_selected_range,
7041 replace_newest,
7042 autoscroll,
7043 cx,
7044 );
7045 } else {
7046 select_next_state.done = true;
7047 }
7048 }
7049
7050 self.select_next_state = Some(select_next_state);
7051 } else {
7052 let mut only_carets = true;
7053 let mut same_text_selected = true;
7054 let mut selected_text = None;
7055
7056 let mut selections_iter = selections.iter().peekable();
7057 while let Some(selection) = selections_iter.next() {
7058 if selection.start != selection.end {
7059 only_carets = false;
7060 }
7061
7062 if same_text_selected {
7063 if selected_text.is_none() {
7064 selected_text =
7065 Some(buffer.text_for_range(selection.range()).collect::<String>());
7066 }
7067
7068 if let Some(next_selection) = selections_iter.peek() {
7069 if next_selection.range().len() == selection.range().len() {
7070 let next_selected_text = buffer
7071 .text_for_range(next_selection.range())
7072 .collect::<String>();
7073 if Some(next_selected_text) != selected_text {
7074 same_text_selected = false;
7075 selected_text = None;
7076 }
7077 } else {
7078 same_text_selected = false;
7079 selected_text = None;
7080 }
7081 }
7082 }
7083 }
7084
7085 if only_carets {
7086 for selection in &mut selections {
7087 let word_range = movement::surrounding_word(
7088 &display_map,
7089 selection.start.to_display_point(&display_map),
7090 );
7091 selection.start = word_range.start.to_offset(&display_map, Bias::Left);
7092 selection.end = word_range.end.to_offset(&display_map, Bias::Left);
7093 selection.goal = SelectionGoal::None;
7094 selection.reversed = false;
7095 select_next_match_ranges(
7096 self,
7097 selection.start..selection.end,
7098 replace_newest,
7099 autoscroll,
7100 cx,
7101 );
7102 }
7103
7104 if selections.len() == 1 {
7105 let selection = selections
7106 .last()
7107 .expect("ensured that there's only one selection");
7108 let query = buffer
7109 .text_for_range(selection.start..selection.end)
7110 .collect::<String>();
7111 let is_empty = query.is_empty();
7112 let select_state = SelectNextState {
7113 query: AhoCorasick::new(&[query])?,
7114 wordwise: true,
7115 done: is_empty,
7116 };
7117 self.select_next_state = Some(select_state);
7118 } else {
7119 self.select_next_state = None;
7120 }
7121 } else if let Some(selected_text) = selected_text {
7122 self.select_next_state = Some(SelectNextState {
7123 query: AhoCorasick::new(&[selected_text])?,
7124 wordwise: false,
7125 done: false,
7126 });
7127 self.select_next_match_internal(display_map, replace_newest, autoscroll, cx)?;
7128 }
7129 }
7130 Ok(())
7131 }
7132
7133 pub fn select_all_matches(
7134 &mut self,
7135 _action: &SelectAllMatches,
7136 cx: &mut ViewContext<Self>,
7137 ) -> Result<()> {
7138 self.push_to_selection_history();
7139 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
7140
7141 self.select_next_match_internal(&display_map, false, None, cx)?;
7142 let Some(select_next_state) = self.select_next_state.as_mut() else {
7143 return Ok(());
7144 };
7145 if select_next_state.done {
7146 return Ok(());
7147 }
7148
7149 let mut new_selections = self.selections.all::<usize>(cx);
7150
7151 let buffer = &display_map.buffer_snapshot;
7152 let query_matches = select_next_state
7153 .query
7154 .stream_find_iter(buffer.bytes_in_range(0..buffer.len()));
7155
7156 for query_match in query_matches {
7157 let query_match = query_match.unwrap(); // can only fail due to I/O
7158 let offset_range = query_match.start()..query_match.end();
7159 let display_range = offset_range.start.to_display_point(&display_map)
7160 ..offset_range.end.to_display_point(&display_map);
7161
7162 if !select_next_state.wordwise
7163 || (!movement::is_inside_word(&display_map, display_range.start)
7164 && !movement::is_inside_word(&display_map, display_range.end))
7165 {
7166 self.selections.change_with(cx, |selections| {
7167 new_selections.push(Selection {
7168 id: selections.new_selection_id(),
7169 start: offset_range.start,
7170 end: offset_range.end,
7171 reversed: false,
7172 goal: SelectionGoal::None,
7173 });
7174 });
7175 }
7176 }
7177
7178 new_selections.sort_by_key(|selection| selection.start);
7179 let mut ix = 0;
7180 while ix + 1 < new_selections.len() {
7181 let current_selection = &new_selections[ix];
7182 let next_selection = &new_selections[ix + 1];
7183 if current_selection.range().overlaps(&next_selection.range()) {
7184 if current_selection.id < next_selection.id {
7185 new_selections.remove(ix + 1);
7186 } else {
7187 new_selections.remove(ix);
7188 }
7189 } else {
7190 ix += 1;
7191 }
7192 }
7193
7194 select_next_state.done = true;
7195 self.unfold_ranges(
7196 new_selections.iter().map(|selection| selection.range()),
7197 false,
7198 false,
7199 cx,
7200 );
7201 self.change_selections(Some(Autoscroll::fit()), cx, |selections| {
7202 selections.select(new_selections)
7203 });
7204
7205 Ok(())
7206 }
7207
7208 pub fn select_next(&mut self, action: &SelectNext, cx: &mut ViewContext<Self>) -> Result<()> {
7209 self.push_to_selection_history();
7210 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
7211 self.select_next_match_internal(
7212 &display_map,
7213 action.replace_newest,
7214 Some(Autoscroll::newest()),
7215 cx,
7216 )?;
7217 Ok(())
7218 }
7219
7220 pub fn select_previous(
7221 &mut self,
7222 action: &SelectPrevious,
7223 cx: &mut ViewContext<Self>,
7224 ) -> Result<()> {
7225 self.push_to_selection_history();
7226 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
7227 let buffer = &display_map.buffer_snapshot;
7228 let mut selections = self.selections.all::<usize>(cx);
7229 if let Some(mut select_prev_state) = self.select_prev_state.take() {
7230 let query = &select_prev_state.query;
7231 if !select_prev_state.done {
7232 let first_selection = selections.iter().min_by_key(|s| s.id).unwrap();
7233 let last_selection = selections.iter().max_by_key(|s| s.id).unwrap();
7234 let mut next_selected_range = None;
7235 // When we're iterating matches backwards, the oldest match will actually be the furthest one in the buffer.
7236 let bytes_before_last_selection =
7237 buffer.reversed_bytes_in_range(0..last_selection.start);
7238 let bytes_after_first_selection =
7239 buffer.reversed_bytes_in_range(first_selection.end..buffer.len());
7240 let query_matches = query
7241 .stream_find_iter(bytes_before_last_selection)
7242 .map(|result| (last_selection.start, result))
7243 .chain(
7244 query
7245 .stream_find_iter(bytes_after_first_selection)
7246 .map(|result| (buffer.len(), result)),
7247 );
7248 for (end_offset, query_match) in query_matches {
7249 let query_match = query_match.unwrap(); // can only fail due to I/O
7250 let offset_range =
7251 end_offset - query_match.end()..end_offset - query_match.start();
7252 let display_range = offset_range.start.to_display_point(&display_map)
7253 ..offset_range.end.to_display_point(&display_map);
7254
7255 if !select_prev_state.wordwise
7256 || (!movement::is_inside_word(&display_map, display_range.start)
7257 && !movement::is_inside_word(&display_map, display_range.end))
7258 {
7259 next_selected_range = Some(offset_range);
7260 break;
7261 }
7262 }
7263
7264 if let Some(next_selected_range) = next_selected_range {
7265 self.unfold_ranges([next_selected_range.clone()], false, true, cx);
7266 self.change_selections(Some(Autoscroll::newest()), cx, |s| {
7267 if action.replace_newest {
7268 s.delete(s.newest_anchor().id);
7269 }
7270 s.insert_range(next_selected_range);
7271 });
7272 } else {
7273 select_prev_state.done = true;
7274 }
7275 }
7276
7277 self.select_prev_state = Some(select_prev_state);
7278 } else {
7279 let mut only_carets = true;
7280 let mut same_text_selected = true;
7281 let mut selected_text = None;
7282
7283 let mut selections_iter = selections.iter().peekable();
7284 while let Some(selection) = selections_iter.next() {
7285 if selection.start != selection.end {
7286 only_carets = false;
7287 }
7288
7289 if same_text_selected {
7290 if selected_text.is_none() {
7291 selected_text =
7292 Some(buffer.text_for_range(selection.range()).collect::<String>());
7293 }
7294
7295 if let Some(next_selection) = selections_iter.peek() {
7296 if next_selection.range().len() == selection.range().len() {
7297 let next_selected_text = buffer
7298 .text_for_range(next_selection.range())
7299 .collect::<String>();
7300 if Some(next_selected_text) != selected_text {
7301 same_text_selected = false;
7302 selected_text = None;
7303 }
7304 } else {
7305 same_text_selected = false;
7306 selected_text = None;
7307 }
7308 }
7309 }
7310 }
7311
7312 if only_carets {
7313 for selection in &mut selections {
7314 let word_range = movement::surrounding_word(
7315 &display_map,
7316 selection.start.to_display_point(&display_map),
7317 );
7318 selection.start = word_range.start.to_offset(&display_map, Bias::Left);
7319 selection.end = word_range.end.to_offset(&display_map, Bias::Left);
7320 selection.goal = SelectionGoal::None;
7321 selection.reversed = false;
7322 }
7323 if selections.len() == 1 {
7324 let selection = selections
7325 .last()
7326 .expect("ensured that there's only one selection");
7327 let query = buffer
7328 .text_for_range(selection.start..selection.end)
7329 .collect::<String>();
7330 let is_empty = query.is_empty();
7331 let select_state = SelectNextState {
7332 query: AhoCorasick::new(&[query.chars().rev().collect::<String>()])?,
7333 wordwise: true,
7334 done: is_empty,
7335 };
7336 self.select_prev_state = Some(select_state);
7337 } else {
7338 self.select_prev_state = None;
7339 }
7340
7341 self.unfold_ranges(
7342 selections.iter().map(|s| s.range()).collect::<Vec<_>>(),
7343 false,
7344 true,
7345 cx,
7346 );
7347 self.change_selections(Some(Autoscroll::newest()), cx, |s| {
7348 s.select(selections);
7349 });
7350 } else if let Some(selected_text) = selected_text {
7351 self.select_prev_state = Some(SelectNextState {
7352 query: AhoCorasick::new(&[selected_text.chars().rev().collect::<String>()])?,
7353 wordwise: false,
7354 done: false,
7355 });
7356 self.select_previous(action, cx)?;
7357 }
7358 }
7359 Ok(())
7360 }
7361
7362 pub fn toggle_comments(&mut self, action: &ToggleComments, cx: &mut ViewContext<Self>) {
7363 let text_layout_details = &self.text_layout_details(cx);
7364 self.transact(cx, |this, cx| {
7365 let mut selections = this.selections.all::<Point>(cx);
7366 let mut edits = Vec::new();
7367 let mut selection_edit_ranges = Vec::new();
7368 let mut last_toggled_row = None;
7369 let snapshot = this.buffer.read(cx).read(cx);
7370 let empty_str: Arc<str> = "".into();
7371 let mut suffixes_inserted = Vec::new();
7372
7373 fn comment_prefix_range(
7374 snapshot: &MultiBufferSnapshot,
7375 row: u32,
7376 comment_prefix: &str,
7377 comment_prefix_whitespace: &str,
7378 ) -> Range<Point> {
7379 let start = Point::new(row, snapshot.indent_size_for_line(row).len);
7380
7381 let mut line_bytes = snapshot
7382 .bytes_in_range(start..snapshot.max_point())
7383 .flatten()
7384 .copied();
7385
7386 // If this line currently begins with the line comment prefix, then record
7387 // the range containing the prefix.
7388 if line_bytes
7389 .by_ref()
7390 .take(comment_prefix.len())
7391 .eq(comment_prefix.bytes())
7392 {
7393 // Include any whitespace that matches the comment prefix.
7394 let matching_whitespace_len = line_bytes
7395 .zip(comment_prefix_whitespace.bytes())
7396 .take_while(|(a, b)| a == b)
7397 .count() as u32;
7398 let end = Point::new(
7399 start.row,
7400 start.column + comment_prefix.len() as u32 + matching_whitespace_len,
7401 );
7402 start..end
7403 } else {
7404 start..start
7405 }
7406 }
7407
7408 fn comment_suffix_range(
7409 snapshot: &MultiBufferSnapshot,
7410 row: u32,
7411 comment_suffix: &str,
7412 comment_suffix_has_leading_space: bool,
7413 ) -> Range<Point> {
7414 let end = Point::new(row, snapshot.line_len(row));
7415 let suffix_start_column = end.column.saturating_sub(comment_suffix.len() as u32);
7416
7417 let mut line_end_bytes = snapshot
7418 .bytes_in_range(Point::new(end.row, suffix_start_column.saturating_sub(1))..end)
7419 .flatten()
7420 .copied();
7421
7422 let leading_space_len = if suffix_start_column > 0
7423 && line_end_bytes.next() == Some(b' ')
7424 && comment_suffix_has_leading_space
7425 {
7426 1
7427 } else {
7428 0
7429 };
7430
7431 // If this line currently begins with the line comment prefix, then record
7432 // the range containing the prefix.
7433 if line_end_bytes.by_ref().eq(comment_suffix.bytes()) {
7434 let start = Point::new(end.row, suffix_start_column - leading_space_len);
7435 start..end
7436 } else {
7437 end..end
7438 }
7439 }
7440
7441 // TODO: Handle selections that cross excerpts
7442 for selection in &mut selections {
7443 let start_column = snapshot.indent_size_for_line(selection.start.row).len;
7444 let language = if let Some(language) =
7445 snapshot.language_scope_at(Point::new(selection.start.row, start_column))
7446 {
7447 language
7448 } else {
7449 continue;
7450 };
7451
7452 selection_edit_ranges.clear();
7453
7454 // If multiple selections contain a given row, avoid processing that
7455 // row more than once.
7456 let mut start_row = selection.start.row;
7457 if last_toggled_row == Some(start_row) {
7458 start_row += 1;
7459 }
7460 let end_row =
7461 if selection.end.row > selection.start.row && selection.end.column == 0 {
7462 selection.end.row - 1
7463 } else {
7464 selection.end.row
7465 };
7466 last_toggled_row = Some(end_row);
7467
7468 if start_row > end_row {
7469 continue;
7470 }
7471
7472 // If the language has line comments, toggle those.
7473 let full_comment_prefixes = language.line_comment_prefixes();
7474 if !full_comment_prefixes.is_empty() {
7475 let first_prefix = full_comment_prefixes
7476 .first()
7477 .expect("prefixes is non-empty");
7478 let prefix_trimmed_lengths = full_comment_prefixes
7479 .iter()
7480 .map(|p| p.trim_end_matches(' ').len())
7481 .collect::<SmallVec<[usize; 4]>>();
7482
7483 let mut all_selection_lines_are_comments = true;
7484
7485 for row in start_row..=end_row {
7486 if start_row < end_row && snapshot.is_line_blank(row) {
7487 continue;
7488 }
7489
7490 let prefix_range = full_comment_prefixes
7491 .iter()
7492 .zip(prefix_trimmed_lengths.iter().copied())
7493 .map(|(prefix, trimmed_prefix_len)| {
7494 comment_prefix_range(
7495 snapshot.deref(),
7496 row,
7497 &prefix[..trimmed_prefix_len],
7498 &prefix[trimmed_prefix_len..],
7499 )
7500 })
7501 .max_by_key(|range| range.end.column - range.start.column)
7502 .expect("prefixes is non-empty");
7503
7504 if prefix_range.is_empty() {
7505 all_selection_lines_are_comments = false;
7506 }
7507
7508 selection_edit_ranges.push(prefix_range);
7509 }
7510
7511 if all_selection_lines_are_comments {
7512 edits.extend(
7513 selection_edit_ranges
7514 .iter()
7515 .cloned()
7516 .map(|range| (range, empty_str.clone())),
7517 );
7518 } else {
7519 let min_column = selection_edit_ranges
7520 .iter()
7521 .map(|range| range.start.column)
7522 .min()
7523 .unwrap_or(0);
7524 edits.extend(selection_edit_ranges.iter().map(|range| {
7525 let position = Point::new(range.start.row, min_column);
7526 (position..position, first_prefix.clone())
7527 }));
7528 }
7529 } else if let Some((full_comment_prefix, comment_suffix)) =
7530 language.block_comment_delimiters()
7531 {
7532 let comment_prefix = full_comment_prefix.trim_end_matches(' ');
7533 let comment_prefix_whitespace = &full_comment_prefix[comment_prefix.len()..];
7534 let prefix_range = comment_prefix_range(
7535 snapshot.deref(),
7536 start_row,
7537 comment_prefix,
7538 comment_prefix_whitespace,
7539 );
7540 let suffix_range = comment_suffix_range(
7541 snapshot.deref(),
7542 end_row,
7543 comment_suffix.trim_start_matches(' '),
7544 comment_suffix.starts_with(' '),
7545 );
7546
7547 if prefix_range.is_empty() || suffix_range.is_empty() {
7548 edits.push((
7549 prefix_range.start..prefix_range.start,
7550 full_comment_prefix.clone(),
7551 ));
7552 edits.push((suffix_range.end..suffix_range.end, comment_suffix.clone()));
7553 suffixes_inserted.push((end_row, comment_suffix.len()));
7554 } else {
7555 edits.push((prefix_range, empty_str.clone()));
7556 edits.push((suffix_range, empty_str.clone()));
7557 }
7558 } else {
7559 continue;
7560 }
7561 }
7562
7563 drop(snapshot);
7564 this.buffer.update(cx, |buffer, cx| {
7565 buffer.edit(edits, None, cx);
7566 });
7567
7568 // Adjust selections so that they end before any comment suffixes that
7569 // were inserted.
7570 let mut suffixes_inserted = suffixes_inserted.into_iter().peekable();
7571 let mut selections = this.selections.all::<Point>(cx);
7572 let snapshot = this.buffer.read(cx).read(cx);
7573 for selection in &mut selections {
7574 while let Some((row, suffix_len)) = suffixes_inserted.peek().copied() {
7575 match row.cmp(&selection.end.row) {
7576 Ordering::Less => {
7577 suffixes_inserted.next();
7578 continue;
7579 }
7580 Ordering::Greater => break,
7581 Ordering::Equal => {
7582 if selection.end.column == snapshot.line_len(row) {
7583 if selection.is_empty() {
7584 selection.start.column -= suffix_len as u32;
7585 }
7586 selection.end.column -= suffix_len as u32;
7587 }
7588 break;
7589 }
7590 }
7591 }
7592 }
7593
7594 drop(snapshot);
7595 this.change_selections(Some(Autoscroll::fit()), cx, |s| s.select(selections));
7596
7597 let selections = this.selections.all::<Point>(cx);
7598 let selections_on_single_row = selections.windows(2).all(|selections| {
7599 selections[0].start.row == selections[1].start.row
7600 && selections[0].end.row == selections[1].end.row
7601 && selections[0].start.row == selections[0].end.row
7602 });
7603 let selections_selecting = selections
7604 .iter()
7605 .any(|selection| selection.start != selection.end);
7606 let advance_downwards = action.advance_downwards
7607 && selections_on_single_row
7608 && !selections_selecting
7609 && this.mode != EditorMode::SingleLine;
7610
7611 if advance_downwards {
7612 let snapshot = this.buffer.read(cx).snapshot(cx);
7613
7614 this.change_selections(Some(Autoscroll::fit()), cx, |s| {
7615 s.move_cursors_with(|display_snapshot, display_point, _| {
7616 let mut point = display_point.to_point(display_snapshot);
7617 point.row += 1;
7618 point = snapshot.clip_point(point, Bias::Left);
7619 let display_point = point.to_display_point(display_snapshot);
7620 let goal = SelectionGoal::HorizontalPosition(
7621 display_snapshot
7622 .x_for_display_point(display_point, &text_layout_details)
7623 .into(),
7624 );
7625 (display_point, goal)
7626 })
7627 });
7628 }
7629 });
7630 }
7631
7632 pub fn select_larger_syntax_node(
7633 &mut self,
7634 _: &SelectLargerSyntaxNode,
7635 cx: &mut ViewContext<Self>,
7636 ) {
7637 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
7638 let buffer = self.buffer.read(cx).snapshot(cx);
7639 let old_selections = self.selections.all::<usize>(cx).into_boxed_slice();
7640
7641 let mut stack = mem::take(&mut self.select_larger_syntax_node_stack);
7642 let mut selected_larger_node = false;
7643 let new_selections = old_selections
7644 .iter()
7645 .map(|selection| {
7646 let old_range = selection.start..selection.end;
7647 let mut new_range = old_range.clone();
7648 while let Some(containing_range) =
7649 buffer.range_for_syntax_ancestor(new_range.clone())
7650 {
7651 new_range = containing_range;
7652 if !display_map.intersects_fold(new_range.start)
7653 && !display_map.intersects_fold(new_range.end)
7654 {
7655 break;
7656 }
7657 }
7658
7659 selected_larger_node |= new_range != old_range;
7660 Selection {
7661 id: selection.id,
7662 start: new_range.start,
7663 end: new_range.end,
7664 goal: SelectionGoal::None,
7665 reversed: selection.reversed,
7666 }
7667 })
7668 .collect::<Vec<_>>();
7669
7670 if selected_larger_node {
7671 stack.push(old_selections);
7672 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
7673 s.select(new_selections);
7674 });
7675 }
7676 self.select_larger_syntax_node_stack = stack;
7677 }
7678
7679 pub fn select_smaller_syntax_node(
7680 &mut self,
7681 _: &SelectSmallerSyntaxNode,
7682 cx: &mut ViewContext<Self>,
7683 ) {
7684 let mut stack = mem::take(&mut self.select_larger_syntax_node_stack);
7685 if let Some(selections) = stack.pop() {
7686 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
7687 s.select(selections.to_vec());
7688 });
7689 }
7690 self.select_larger_syntax_node_stack = stack;
7691 }
7692
7693 fn refresh_runnables(&mut self, cx: &mut ViewContext<Self>) -> Task<()> {
7694 let project = self.project.clone();
7695 cx.spawn(|this, mut cx| async move {
7696 let Ok(display_snapshot) = this.update(&mut cx, |this, cx| {
7697 this.display_map.update(cx, |map, cx| map.snapshot(cx))
7698 }) else {
7699 return;
7700 };
7701
7702 let Some(project) = project else {
7703 return;
7704 };
7705 if project
7706 .update(&mut cx, |this, _| this.is_remote())
7707 .unwrap_or(true)
7708 {
7709 // Do not display any test indicators in remote projects.
7710 return;
7711 }
7712 let new_rows =
7713 cx.background_executor()
7714 .spawn({
7715 let snapshot = display_snapshot.clone();
7716 async move {
7717 Self::fetch_runnable_ranges(&snapshot, Anchor::min()..Anchor::max())
7718 }
7719 })
7720 .await;
7721 let rows = Self::refresh_runnable_display_rows(
7722 project,
7723 display_snapshot,
7724 new_rows,
7725 cx.clone(),
7726 );
7727
7728 this.update(&mut cx, |this, _| {
7729 this.clear_tasks();
7730 for (row, tasks) in rows {
7731 this.insert_tasks(row, tasks);
7732 }
7733 })
7734 .ok();
7735 })
7736 }
7737 fn fetch_runnable_ranges(
7738 snapshot: &DisplaySnapshot,
7739 range: Range<Anchor>,
7740 ) -> Vec<(Range<usize>, Runnable)> {
7741 snapshot.buffer_snapshot.runnable_ranges(range).collect()
7742 }
7743 fn refresh_runnable_display_rows(
7744 project: Model<Project>,
7745 snapshot: DisplaySnapshot,
7746 runnable_ranges: Vec<(Range<usize>, Runnable)>,
7747 mut cx: AsyncWindowContext,
7748 ) -> Vec<(u32, RunnableTasks)> {
7749 runnable_ranges
7750 .into_iter()
7751 .filter_map(|(multi_buffer_range, mut runnable)| {
7752 let (tasks, _) = cx
7753 .update(|cx| Self::resolve_runnable(project.clone(), &mut runnable, cx))
7754 .ok()?;
7755 if tasks.is_empty() {
7756 return None;
7757 }
7758 let point = multi_buffer_range.start.to_display_point(&snapshot);
7759 Some((
7760 point.row(),
7761 RunnableTasks {
7762 templates: tasks,
7763 column: point.column(),
7764 },
7765 ))
7766 })
7767 .collect()
7768 }
7769
7770 fn resolve_runnable(
7771 project: Model<Project>,
7772 runnable: &mut Runnable,
7773 cx: &WindowContext<'_>,
7774 ) -> (Vec<(TaskSourceKind, TaskTemplate)>, Option<WorktreeId>) {
7775 let (inventory, worktree_id) = project.read_with(cx, |project, cx| {
7776 let worktree_id = project
7777 .buffer_for_id(runnable.buffer)
7778 .and_then(|buffer| buffer.read(cx).file())
7779 .map(|file| WorktreeId::from_usize(file.worktree_id()));
7780
7781 (project.task_inventory().clone(), worktree_id)
7782 });
7783
7784 let inventory = inventory.read(cx);
7785 let tags = mem::take(&mut runnable.tags);
7786 let mut tags: Vec<_> = tags
7787 .into_iter()
7788 .flat_map(|tag| {
7789 let tag = tag.0.clone();
7790 inventory
7791 .list_tasks(Some(runnable.language.clone()), worktree_id)
7792 .into_iter()
7793 .filter(move |(_, template)| {
7794 template.tags.iter().any(|source_tag| source_tag == &tag)
7795 })
7796 })
7797 .sorted_by_key(|(kind, _)| kind.to_owned())
7798 .collect();
7799 if let Some((leading_tag_source, _)) = tags.first() {
7800 // Strongest source wins; if we have worktree tag binding, prefer that to
7801 // global and language bindings;
7802 // if we have a global binding, prefer that to language binding.
7803 let first_mismatch = tags
7804 .iter()
7805 .position(|(tag_source, _)| tag_source != leading_tag_source);
7806 if let Some(index) = first_mismatch {
7807 tags.truncate(index);
7808 }
7809 }
7810
7811 (tags, worktree_id)
7812 }
7813
7814 pub fn move_to_enclosing_bracket(
7815 &mut self,
7816 _: &MoveToEnclosingBracket,
7817 cx: &mut ViewContext<Self>,
7818 ) {
7819 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
7820 s.move_offsets_with(|snapshot, selection| {
7821 let Some(enclosing_bracket_ranges) =
7822 snapshot.enclosing_bracket_ranges(selection.start..selection.end)
7823 else {
7824 return;
7825 };
7826
7827 let mut best_length = usize::MAX;
7828 let mut best_inside = false;
7829 let mut best_in_bracket_range = false;
7830 let mut best_destination = None;
7831 for (open, close) in enclosing_bracket_ranges {
7832 let close = close.to_inclusive();
7833 let length = close.end() - open.start;
7834 let inside = selection.start >= open.end && selection.end <= *close.start();
7835 let in_bracket_range = open.to_inclusive().contains(&selection.head())
7836 || close.contains(&selection.head());
7837
7838 // If best is next to a bracket and current isn't, skip
7839 if !in_bracket_range && best_in_bracket_range {
7840 continue;
7841 }
7842
7843 // Prefer smaller lengths unless best is inside and current isn't
7844 if length > best_length && (best_inside || !inside) {
7845 continue;
7846 }
7847
7848 best_length = length;
7849 best_inside = inside;
7850 best_in_bracket_range = in_bracket_range;
7851 best_destination = Some(
7852 if close.contains(&selection.start) && close.contains(&selection.end) {
7853 if inside {
7854 open.end
7855 } else {
7856 open.start
7857 }
7858 } else {
7859 if inside {
7860 *close.start()
7861 } else {
7862 *close.end()
7863 }
7864 },
7865 );
7866 }
7867
7868 if let Some(destination) = best_destination {
7869 selection.collapse_to(destination, SelectionGoal::None);
7870 }
7871 })
7872 });
7873 }
7874
7875 pub fn undo_selection(&mut self, _: &UndoSelection, cx: &mut ViewContext<Self>) {
7876 self.end_selection(cx);
7877 self.selection_history.mode = SelectionHistoryMode::Undoing;
7878 if let Some(entry) = self.selection_history.undo_stack.pop_back() {
7879 self.change_selections(None, cx, |s| s.select_anchors(entry.selections.to_vec()));
7880 self.select_next_state = entry.select_next_state;
7881 self.select_prev_state = entry.select_prev_state;
7882 self.add_selections_state = entry.add_selections_state;
7883 self.request_autoscroll(Autoscroll::newest(), cx);
7884 }
7885 self.selection_history.mode = SelectionHistoryMode::Normal;
7886 }
7887
7888 pub fn redo_selection(&mut self, _: &RedoSelection, cx: &mut ViewContext<Self>) {
7889 self.end_selection(cx);
7890 self.selection_history.mode = SelectionHistoryMode::Redoing;
7891 if let Some(entry) = self.selection_history.redo_stack.pop_back() {
7892 self.change_selections(None, cx, |s| s.select_anchors(entry.selections.to_vec()));
7893 self.select_next_state = entry.select_next_state;
7894 self.select_prev_state = entry.select_prev_state;
7895 self.add_selections_state = entry.add_selections_state;
7896 self.request_autoscroll(Autoscroll::newest(), cx);
7897 }
7898 self.selection_history.mode = SelectionHistoryMode::Normal;
7899 }
7900
7901 pub fn expand_excerpts(&mut self, action: &ExpandExcerpts, cx: &mut ViewContext<Self>) {
7902 let selections = self.selections.disjoint_anchors();
7903
7904 let lines = if action.lines == 0 { 3 } else { action.lines };
7905
7906 self.buffer.update(cx, |buffer, cx| {
7907 buffer.expand_excerpts(
7908 selections
7909 .into_iter()
7910 .map(|selection| selection.head().excerpt_id)
7911 .dedup(),
7912 lines,
7913 cx,
7914 )
7915 })
7916 }
7917
7918 pub fn expand_excerpt(&mut self, excerpt: ExcerptId, cx: &mut ViewContext<Self>) {
7919 self.buffer
7920 .update(cx, |buffer, cx| buffer.expand_excerpts([excerpt], 3, cx))
7921 }
7922
7923 fn go_to_diagnostic(&mut self, _: &GoToDiagnostic, cx: &mut ViewContext<Self>) {
7924 self.go_to_diagnostic_impl(Direction::Next, cx)
7925 }
7926
7927 fn go_to_prev_diagnostic(&mut self, _: &GoToPrevDiagnostic, cx: &mut ViewContext<Self>) {
7928 self.go_to_diagnostic_impl(Direction::Prev, cx)
7929 }
7930
7931 pub fn go_to_diagnostic_impl(&mut self, direction: Direction, cx: &mut ViewContext<Self>) {
7932 let buffer = self.buffer.read(cx).snapshot(cx);
7933 let selection = self.selections.newest::<usize>(cx);
7934
7935 // If there is an active Diagnostic Popover jump to its diagnostic instead.
7936 if direction == Direction::Next {
7937 if let Some(popover) = self.hover_state.diagnostic_popover.as_ref() {
7938 let (group_id, jump_to) = popover.activation_info();
7939 if self.activate_diagnostics(group_id, cx) {
7940 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
7941 let mut new_selection = s.newest_anchor().clone();
7942 new_selection.collapse_to(jump_to, SelectionGoal::None);
7943 s.select_anchors(vec![new_selection.clone()]);
7944 });
7945 }
7946 return;
7947 }
7948 }
7949
7950 let mut active_primary_range = self.active_diagnostics.as_ref().map(|active_diagnostics| {
7951 active_diagnostics
7952 .primary_range
7953 .to_offset(&buffer)
7954 .to_inclusive()
7955 });
7956 let mut search_start = if let Some(active_primary_range) = active_primary_range.as_ref() {
7957 if active_primary_range.contains(&selection.head()) {
7958 *active_primary_range.end()
7959 } else {
7960 selection.head()
7961 }
7962 } else {
7963 selection.head()
7964 };
7965 let snapshot = self.snapshot(cx);
7966 loop {
7967 let mut diagnostics = if direction == Direction::Prev {
7968 buffer.diagnostics_in_range::<_, usize>(0..search_start, true)
7969 } else {
7970 buffer.diagnostics_in_range::<_, usize>(search_start..buffer.len(), false)
7971 }
7972 .filter(|diagnostic| !snapshot.intersects_fold(diagnostic.range.start));
7973 let group = diagnostics.find_map(|entry| {
7974 if entry.diagnostic.is_primary
7975 && entry.diagnostic.severity <= DiagnosticSeverity::WARNING
7976 && !entry.range.is_empty()
7977 && Some(entry.range.end) != active_primary_range.as_ref().map(|r| *r.end())
7978 && !entry.range.contains(&search_start)
7979 {
7980 Some((entry.range, entry.diagnostic.group_id))
7981 } else {
7982 None
7983 }
7984 });
7985
7986 if let Some((primary_range, group_id)) = group {
7987 if self.activate_diagnostics(group_id, cx) {
7988 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
7989 s.select(vec![Selection {
7990 id: selection.id,
7991 start: primary_range.start,
7992 end: primary_range.start,
7993 reversed: false,
7994 goal: SelectionGoal::None,
7995 }]);
7996 });
7997 }
7998 break;
7999 } else {
8000 // Cycle around to the start of the buffer, potentially moving back to the start of
8001 // the currently active diagnostic.
8002 active_primary_range.take();
8003 if direction == Direction::Prev {
8004 if search_start == buffer.len() {
8005 break;
8006 } else {
8007 search_start = buffer.len();
8008 }
8009 } else if search_start == 0 {
8010 break;
8011 } else {
8012 search_start = 0;
8013 }
8014 }
8015 }
8016 }
8017
8018 fn go_to_hunk(&mut self, _: &GoToHunk, cx: &mut ViewContext<Self>) {
8019 let snapshot = self
8020 .display_map
8021 .update(cx, |display_map, cx| display_map.snapshot(cx));
8022 let selection = self.selections.newest::<Point>(cx);
8023
8024 if !self.seek_in_direction(
8025 &snapshot,
8026 selection.head(),
8027 false,
8028 snapshot
8029 .buffer_snapshot
8030 .git_diff_hunks_in_range((selection.head().row + 1)..u32::MAX),
8031 cx,
8032 ) {
8033 let wrapped_point = Point::zero();
8034 self.seek_in_direction(
8035 &snapshot,
8036 wrapped_point,
8037 true,
8038 snapshot
8039 .buffer_snapshot
8040 .git_diff_hunks_in_range((wrapped_point.row + 1)..u32::MAX),
8041 cx,
8042 );
8043 }
8044 }
8045
8046 fn go_to_prev_hunk(&mut self, _: &GoToPrevHunk, cx: &mut ViewContext<Self>) {
8047 let snapshot = self
8048 .display_map
8049 .update(cx, |display_map, cx| display_map.snapshot(cx));
8050 let selection = self.selections.newest::<Point>(cx);
8051
8052 if !self.seek_in_direction(
8053 &snapshot,
8054 selection.head(),
8055 false,
8056 snapshot
8057 .buffer_snapshot
8058 .git_diff_hunks_in_range_rev(0..selection.head().row),
8059 cx,
8060 ) {
8061 let wrapped_point = snapshot.buffer_snapshot.max_point();
8062 self.seek_in_direction(
8063 &snapshot,
8064 wrapped_point,
8065 true,
8066 snapshot
8067 .buffer_snapshot
8068 .git_diff_hunks_in_range_rev(0..wrapped_point.row),
8069 cx,
8070 );
8071 }
8072 }
8073
8074 fn seek_in_direction(
8075 &mut self,
8076 snapshot: &DisplaySnapshot,
8077 initial_point: Point,
8078 is_wrapped: bool,
8079 hunks: impl Iterator<Item = DiffHunk<u32>>,
8080 cx: &mut ViewContext<Editor>,
8081 ) -> bool {
8082 let display_point = initial_point.to_display_point(snapshot);
8083 let mut hunks = hunks
8084 .map(|hunk| diff_hunk_to_display(&hunk, &snapshot))
8085 .filter(|hunk| {
8086 if is_wrapped {
8087 true
8088 } else {
8089 !hunk.contains_display_row(display_point.row())
8090 }
8091 })
8092 .dedup();
8093
8094 if let Some(hunk) = hunks.next() {
8095 self.change_selections(Some(Autoscroll::fit()), cx, |s| {
8096 let row = hunk.start_display_row();
8097 let point = DisplayPoint::new(row, 0);
8098 s.select_display_ranges([point..point]);
8099 });
8100
8101 true
8102 } else {
8103 false
8104 }
8105 }
8106
8107 pub fn go_to_definition(
8108 &mut self,
8109 _: &GoToDefinition,
8110 cx: &mut ViewContext<Self>,
8111 ) -> Task<Result<bool>> {
8112 self.go_to_definition_of_kind(GotoDefinitionKind::Symbol, false, cx)
8113 }
8114
8115 pub fn go_to_implementation(
8116 &mut self,
8117 _: &GoToImplementation,
8118 cx: &mut ViewContext<Self>,
8119 ) -> Task<Result<bool>> {
8120 self.go_to_definition_of_kind(GotoDefinitionKind::Implementation, false, cx)
8121 }
8122
8123 pub fn go_to_implementation_split(
8124 &mut self,
8125 _: &GoToImplementationSplit,
8126 cx: &mut ViewContext<Self>,
8127 ) -> Task<Result<bool>> {
8128 self.go_to_definition_of_kind(GotoDefinitionKind::Implementation, true, cx)
8129 }
8130
8131 pub fn go_to_type_definition(
8132 &mut self,
8133 _: &GoToTypeDefinition,
8134 cx: &mut ViewContext<Self>,
8135 ) -> Task<Result<bool>> {
8136 self.go_to_definition_of_kind(GotoDefinitionKind::Type, false, cx)
8137 }
8138
8139 pub fn go_to_definition_split(
8140 &mut self,
8141 _: &GoToDefinitionSplit,
8142 cx: &mut ViewContext<Self>,
8143 ) -> Task<Result<bool>> {
8144 self.go_to_definition_of_kind(GotoDefinitionKind::Symbol, true, cx)
8145 }
8146
8147 pub fn go_to_type_definition_split(
8148 &mut self,
8149 _: &GoToTypeDefinitionSplit,
8150 cx: &mut ViewContext<Self>,
8151 ) -> Task<Result<bool>> {
8152 self.go_to_definition_of_kind(GotoDefinitionKind::Type, true, cx)
8153 }
8154
8155 fn go_to_definition_of_kind(
8156 &mut self,
8157 kind: GotoDefinitionKind,
8158 split: bool,
8159 cx: &mut ViewContext<Self>,
8160 ) -> Task<Result<bool>> {
8161 let Some(workspace) = self.workspace() else {
8162 return Task::ready(Ok(false));
8163 };
8164 let buffer = self.buffer.read(cx);
8165 let head = self.selections.newest::<usize>(cx).head();
8166 let (buffer, head) = if let Some(text_anchor) = buffer.text_anchor_for_position(head, cx) {
8167 text_anchor
8168 } else {
8169 return Task::ready(Ok(false));
8170 };
8171
8172 let project = workspace.read(cx).project().clone();
8173 let definitions = project.update(cx, |project, cx| match kind {
8174 GotoDefinitionKind::Symbol => project.definition(&buffer, head, cx),
8175 GotoDefinitionKind::Type => project.type_definition(&buffer, head, cx),
8176 GotoDefinitionKind::Implementation => project.implementation(&buffer, head, cx),
8177 });
8178
8179 cx.spawn(|editor, mut cx| async move {
8180 let definitions = definitions.await?;
8181 let navigated = editor
8182 .update(&mut cx, |editor, cx| {
8183 editor.navigate_to_hover_links(
8184 Some(kind),
8185 definitions
8186 .into_iter()
8187 .filter(|location| {
8188 hover_links::exclude_link_to_position(&buffer, &head, location, cx)
8189 })
8190 .map(HoverLink::Text)
8191 .collect::<Vec<_>>(),
8192 split,
8193 cx,
8194 )
8195 })?
8196 .await?;
8197 anyhow::Ok(navigated)
8198 })
8199 }
8200
8201 pub fn open_url(&mut self, _: &OpenUrl, cx: &mut ViewContext<Self>) {
8202 let position = self.selections.newest_anchor().head();
8203 let Some((buffer, buffer_position)) =
8204 self.buffer.read(cx).text_anchor_for_position(position, cx)
8205 else {
8206 return;
8207 };
8208
8209 cx.spawn(|editor, mut cx| async move {
8210 if let Some((_, url)) = find_url(&buffer, buffer_position, cx.clone()) {
8211 editor.update(&mut cx, |_, cx| {
8212 cx.open_url(&url);
8213 })
8214 } else {
8215 Ok(())
8216 }
8217 })
8218 .detach();
8219 }
8220
8221 pub(crate) fn navigate_to_hover_links(
8222 &mut self,
8223 kind: Option<GotoDefinitionKind>,
8224 mut definitions: Vec<HoverLink>,
8225 split: bool,
8226 cx: &mut ViewContext<Editor>,
8227 ) -> Task<Result<bool>> {
8228 // If there is one definition, just open it directly
8229 if definitions.len() == 1 {
8230 let definition = definitions.pop().unwrap();
8231 let target_task = match definition {
8232 HoverLink::Text(link) => Task::Ready(Some(Ok(Some(link.target)))),
8233 HoverLink::InlayHint(lsp_location, server_id) => {
8234 self.compute_target_location(lsp_location, server_id, cx)
8235 }
8236 HoverLink::Url(url) => {
8237 cx.open_url(&url);
8238 Task::ready(Ok(None))
8239 }
8240 };
8241 cx.spawn(|editor, mut cx| async move {
8242 let target = target_task.await.context("target resolution task")?;
8243 if let Some(target) = target {
8244 editor.update(&mut cx, |editor, cx| {
8245 let Some(workspace) = editor.workspace() else {
8246 return false;
8247 };
8248 let pane = workspace.read(cx).active_pane().clone();
8249
8250 let range = target.range.to_offset(target.buffer.read(cx));
8251 let range = editor.range_for_match(&range);
8252 if Some(&target.buffer) == editor.buffer.read(cx).as_singleton().as_ref() {
8253 editor.change_selections(Some(Autoscroll::focused()), cx, |s| {
8254 s.select_ranges([range]);
8255 });
8256 } else {
8257 cx.window_context().defer(move |cx| {
8258 let target_editor: View<Self> =
8259 workspace.update(cx, |workspace, cx| {
8260 let pane = if split {
8261 workspace.adjacent_pane(cx)
8262 } else {
8263 workspace.active_pane().clone()
8264 };
8265
8266 workspace.open_project_item(pane, target.buffer.clone(), cx)
8267 });
8268 target_editor.update(cx, |target_editor, cx| {
8269 // When selecting a definition in a different buffer, disable the nav history
8270 // to avoid creating a history entry at the previous cursor location.
8271 pane.update(cx, |pane, _| pane.disable_history());
8272 target_editor.change_selections(
8273 Some(Autoscroll::focused()),
8274 cx,
8275 |s| {
8276 s.select_ranges([range]);
8277 },
8278 );
8279 pane.update(cx, |pane, _| pane.enable_history());
8280 });
8281 });
8282 }
8283 true
8284 })
8285 } else {
8286 Ok(false)
8287 }
8288 })
8289 } else if !definitions.is_empty() {
8290 let replica_id = self.replica_id(cx);
8291 cx.spawn(|editor, mut cx| async move {
8292 let (title, location_tasks, workspace) = editor
8293 .update(&mut cx, |editor, cx| {
8294 let tab_kind = match kind {
8295 Some(GotoDefinitionKind::Implementation) => "Implementations",
8296 _ => "Definitions",
8297 };
8298 let title = definitions
8299 .iter()
8300 .find_map(|definition| match definition {
8301 HoverLink::Text(link) => link.origin.as_ref().map(|origin| {
8302 let buffer = origin.buffer.read(cx);
8303 format!(
8304 "{} for {}",
8305 tab_kind,
8306 buffer
8307 .text_for_range(origin.range.clone())
8308 .collect::<String>()
8309 )
8310 }),
8311 HoverLink::InlayHint(_, _) => None,
8312 HoverLink::Url(_) => None,
8313 })
8314 .unwrap_or(tab_kind.to_string());
8315 let location_tasks = definitions
8316 .into_iter()
8317 .map(|definition| match definition {
8318 HoverLink::Text(link) => Task::Ready(Some(Ok(Some(link.target)))),
8319 HoverLink::InlayHint(lsp_location, server_id) => {
8320 editor.compute_target_location(lsp_location, server_id, cx)
8321 }
8322 HoverLink::Url(_) => Task::ready(Ok(None)),
8323 })
8324 .collect::<Vec<_>>();
8325 (title, location_tasks, editor.workspace().clone())
8326 })
8327 .context("location tasks preparation")?;
8328
8329 let locations = futures::future::join_all(location_tasks)
8330 .await
8331 .into_iter()
8332 .filter_map(|location| location.transpose())
8333 .collect::<Result<_>>()
8334 .context("location tasks")?;
8335
8336 let Some(workspace) = workspace else {
8337 return Ok(false);
8338 };
8339 let opened = workspace
8340 .update(&mut cx, |workspace, cx| {
8341 Self::open_locations_in_multibuffer(
8342 workspace, locations, replica_id, title, split, cx,
8343 )
8344 })
8345 .ok();
8346
8347 anyhow::Ok(opened.is_some())
8348 })
8349 } else {
8350 Task::ready(Ok(false))
8351 }
8352 }
8353
8354 fn compute_target_location(
8355 &self,
8356 lsp_location: lsp::Location,
8357 server_id: LanguageServerId,
8358 cx: &mut ViewContext<Editor>,
8359 ) -> Task<anyhow::Result<Option<Location>>> {
8360 let Some(project) = self.project.clone() else {
8361 return Task::Ready(Some(Ok(None)));
8362 };
8363
8364 cx.spawn(move |editor, mut cx| async move {
8365 let location_task = editor.update(&mut cx, |editor, cx| {
8366 project.update(cx, |project, cx| {
8367 let language_server_name =
8368 editor.buffer.read(cx).as_singleton().and_then(|buffer| {
8369 project
8370 .language_server_for_buffer(buffer.read(cx), server_id, cx)
8371 .map(|(lsp_adapter, _)| lsp_adapter.name.clone())
8372 });
8373 language_server_name.map(|language_server_name| {
8374 project.open_local_buffer_via_lsp(
8375 lsp_location.uri.clone(),
8376 server_id,
8377 language_server_name,
8378 cx,
8379 )
8380 })
8381 })
8382 })?;
8383 let location = match location_task {
8384 Some(task) => Some({
8385 let target_buffer_handle = task.await.context("open local buffer")?;
8386 let range = target_buffer_handle.update(&mut cx, |target_buffer, _| {
8387 let target_start = target_buffer
8388 .clip_point_utf16(point_from_lsp(lsp_location.range.start), Bias::Left);
8389 let target_end = target_buffer
8390 .clip_point_utf16(point_from_lsp(lsp_location.range.end), Bias::Left);
8391 target_buffer.anchor_after(target_start)
8392 ..target_buffer.anchor_before(target_end)
8393 })?;
8394 Location {
8395 buffer: target_buffer_handle,
8396 range,
8397 }
8398 }),
8399 None => None,
8400 };
8401 Ok(location)
8402 })
8403 }
8404
8405 pub fn find_all_references(
8406 &mut self,
8407 _: &FindAllReferences,
8408 cx: &mut ViewContext<Self>,
8409 ) -> Option<Task<Result<()>>> {
8410 let multi_buffer = self.buffer.read(cx);
8411 let selection = self.selections.newest::<usize>(cx);
8412 let head = selection.head();
8413
8414 let multi_buffer_snapshot = multi_buffer.snapshot(cx);
8415 let head_anchor = multi_buffer_snapshot.anchor_at(
8416 head,
8417 if head < selection.tail() {
8418 Bias::Right
8419 } else {
8420 Bias::Left
8421 },
8422 );
8423
8424 match self
8425 .find_all_references_task_sources
8426 .binary_search_by(|anchor| anchor.cmp(&head_anchor, &multi_buffer_snapshot))
8427 {
8428 Ok(_) => {
8429 log::info!(
8430 "Ignoring repeated FindAllReferences invocation with the position of already running task"
8431 );
8432 return None;
8433 }
8434 Err(i) => {
8435 self.find_all_references_task_sources.insert(i, head_anchor);
8436 }
8437 }
8438
8439 let (buffer, head) = multi_buffer.text_anchor_for_position(head, cx)?;
8440 let replica_id = self.replica_id(cx);
8441 let workspace = self.workspace()?;
8442 let project = workspace.read(cx).project().clone();
8443 let references = project.update(cx, |project, cx| project.references(&buffer, head, cx));
8444 Some(cx.spawn(|editor, mut cx| async move {
8445 let _cleanup = defer({
8446 let mut cx = cx.clone();
8447 move || {
8448 let _ = editor.update(&mut cx, |editor, _| {
8449 if let Ok(i) =
8450 editor
8451 .find_all_references_task_sources
8452 .binary_search_by(|anchor| {
8453 anchor.cmp(&head_anchor, &multi_buffer_snapshot)
8454 })
8455 {
8456 editor.find_all_references_task_sources.remove(i);
8457 }
8458 });
8459 }
8460 });
8461
8462 let locations = references.await?;
8463 if locations.is_empty() {
8464 return anyhow::Ok(());
8465 }
8466
8467 workspace.update(&mut cx, |workspace, cx| {
8468 let title = locations
8469 .first()
8470 .as_ref()
8471 .map(|location| {
8472 let buffer = location.buffer.read(cx);
8473 format!(
8474 "References to `{}`",
8475 buffer
8476 .text_for_range(location.range.clone())
8477 .collect::<String>()
8478 )
8479 })
8480 .unwrap();
8481 Self::open_locations_in_multibuffer(
8482 workspace, locations, replica_id, title, false, cx,
8483 );
8484 })
8485 }))
8486 }
8487
8488 /// Opens a multibuffer with the given project locations in it
8489 pub fn open_locations_in_multibuffer(
8490 workspace: &mut Workspace,
8491 mut locations: Vec<Location>,
8492 replica_id: ReplicaId,
8493 title: String,
8494 split: bool,
8495 cx: &mut ViewContext<Workspace>,
8496 ) {
8497 // If there are multiple definitions, open them in a multibuffer
8498 locations.sort_by_key(|location| location.buffer.read(cx).remote_id());
8499 let mut locations = locations.into_iter().peekable();
8500 let mut ranges_to_highlight = Vec::new();
8501 let capability = workspace.project().read(cx).capability();
8502
8503 let excerpt_buffer = cx.new_model(|cx| {
8504 let mut multibuffer = MultiBuffer::new(replica_id, capability);
8505 while let Some(location) = locations.next() {
8506 let buffer = location.buffer.read(cx);
8507 let mut ranges_for_buffer = Vec::new();
8508 let range = location.range.to_offset(buffer);
8509 ranges_for_buffer.push(range.clone());
8510
8511 while let Some(next_location) = locations.peek() {
8512 if next_location.buffer == location.buffer {
8513 ranges_for_buffer.push(next_location.range.to_offset(buffer));
8514 locations.next();
8515 } else {
8516 break;
8517 }
8518 }
8519
8520 ranges_for_buffer.sort_by_key(|range| (range.start, Reverse(range.end)));
8521 ranges_to_highlight.extend(multibuffer.push_excerpts_with_context_lines(
8522 location.buffer.clone(),
8523 ranges_for_buffer,
8524 DEFAULT_MULTIBUFFER_CONTEXT,
8525 cx,
8526 ))
8527 }
8528
8529 multibuffer.with_title(title)
8530 });
8531
8532 let editor = cx.new_view(|cx| {
8533 Editor::for_multibuffer(excerpt_buffer, Some(workspace.project().clone()), cx)
8534 });
8535 editor.update(cx, |editor, cx| {
8536 editor.highlight_background::<Self>(
8537 &ranges_to_highlight,
8538 |theme| theme.editor_highlighted_line_background,
8539 cx,
8540 );
8541 });
8542
8543 let item = Box::new(editor);
8544 let item_id = item.item_id();
8545
8546 if split {
8547 workspace.split_item(SplitDirection::Right, item.clone(), cx);
8548 } else {
8549 let destination_index = workspace.active_pane().update(cx, |pane, cx| {
8550 if PreviewTabsSettings::get_global(cx).enable_preview_from_code_navigation {
8551 pane.close_current_preview_item(cx)
8552 } else {
8553 None
8554 }
8555 });
8556 workspace.add_item_to_active_pane(item.clone(), destination_index, cx);
8557 }
8558 workspace.active_pane().update(cx, |pane, cx| {
8559 pane.set_preview_item_id(Some(item_id), cx);
8560 });
8561 }
8562
8563 pub fn rename(&mut self, _: &Rename, cx: &mut ViewContext<Self>) -> Option<Task<Result<()>>> {
8564 use language::ToOffset as _;
8565
8566 let project = self.project.clone()?;
8567 let selection = self.selections.newest_anchor().clone();
8568 let (cursor_buffer, cursor_buffer_position) = self
8569 .buffer
8570 .read(cx)
8571 .text_anchor_for_position(selection.head(), cx)?;
8572 let (tail_buffer, cursor_buffer_position_end) = self
8573 .buffer
8574 .read(cx)
8575 .text_anchor_for_position(selection.tail(), cx)?;
8576 if tail_buffer != cursor_buffer {
8577 return None;
8578 }
8579
8580 let snapshot = cursor_buffer.read(cx).snapshot();
8581 let cursor_buffer_offset = cursor_buffer_position.to_offset(&snapshot);
8582 let cursor_buffer_offset_end = cursor_buffer_position_end.to_offset(&snapshot);
8583 let prepare_rename = project.update(cx, |project, cx| {
8584 project.prepare_rename(cursor_buffer.clone(), cursor_buffer_offset, cx)
8585 });
8586 drop(snapshot);
8587
8588 Some(cx.spawn(|this, mut cx| async move {
8589 let rename_range = if let Some(range) = prepare_rename.await? {
8590 Some(range)
8591 } else {
8592 this.update(&mut cx, |this, cx| {
8593 let buffer = this.buffer.read(cx).snapshot(cx);
8594 let mut buffer_highlights = this
8595 .document_highlights_for_position(selection.head(), &buffer)
8596 .filter(|highlight| {
8597 highlight.start.excerpt_id == selection.head().excerpt_id
8598 && highlight.end.excerpt_id == selection.head().excerpt_id
8599 });
8600 buffer_highlights
8601 .next()
8602 .map(|highlight| highlight.start.text_anchor..highlight.end.text_anchor)
8603 })?
8604 };
8605 if let Some(rename_range) = rename_range {
8606 this.update(&mut cx, |this, cx| {
8607 let snapshot = cursor_buffer.read(cx).snapshot();
8608 let rename_buffer_range = rename_range.to_offset(&snapshot);
8609 let cursor_offset_in_rename_range =
8610 cursor_buffer_offset.saturating_sub(rename_buffer_range.start);
8611 let cursor_offset_in_rename_range_end =
8612 cursor_buffer_offset_end.saturating_sub(rename_buffer_range.start);
8613
8614 this.take_rename(false, cx);
8615 let buffer = this.buffer.read(cx).read(cx);
8616 let cursor_offset = selection.head().to_offset(&buffer);
8617 let rename_start = cursor_offset.saturating_sub(cursor_offset_in_rename_range);
8618 let rename_end = rename_start + rename_buffer_range.len();
8619 let range = buffer.anchor_before(rename_start)..buffer.anchor_after(rename_end);
8620 let mut old_highlight_id = None;
8621 let old_name: Arc<str> = buffer
8622 .chunks(rename_start..rename_end, true)
8623 .map(|chunk| {
8624 if old_highlight_id.is_none() {
8625 old_highlight_id = chunk.syntax_highlight_id;
8626 }
8627 chunk.text
8628 })
8629 .collect::<String>()
8630 .into();
8631
8632 drop(buffer);
8633
8634 // Position the selection in the rename editor so that it matches the current selection.
8635 this.show_local_selections = false;
8636 let rename_editor = cx.new_view(|cx| {
8637 let mut editor = Editor::single_line(cx);
8638 editor.buffer.update(cx, |buffer, cx| {
8639 buffer.edit([(0..0, old_name.clone())], None, cx)
8640 });
8641 let rename_selection_range = match cursor_offset_in_rename_range
8642 .cmp(&cursor_offset_in_rename_range_end)
8643 {
8644 Ordering::Equal => {
8645 editor.select_all(&SelectAll, cx);
8646 return editor;
8647 }
8648 Ordering::Less => {
8649 cursor_offset_in_rename_range..cursor_offset_in_rename_range_end
8650 }
8651 Ordering::Greater => {
8652 cursor_offset_in_rename_range_end..cursor_offset_in_rename_range
8653 }
8654 };
8655 if rename_selection_range.end > old_name.len() {
8656 editor.select_all(&SelectAll, cx);
8657 } else {
8658 editor.change_selections(Some(Autoscroll::fit()), cx, |s| {
8659 s.select_ranges([rename_selection_range]);
8660 });
8661 }
8662 editor
8663 });
8664
8665 let write_highlights =
8666 this.clear_background_highlights::<DocumentHighlightWrite>(cx);
8667 let read_highlights =
8668 this.clear_background_highlights::<DocumentHighlightRead>(cx);
8669 let ranges = write_highlights
8670 .iter()
8671 .flat_map(|(_, ranges)| ranges.iter())
8672 .chain(read_highlights.iter().flat_map(|(_, ranges)| ranges.iter()))
8673 .cloned()
8674 .collect();
8675
8676 this.highlight_text::<Rename>(
8677 ranges,
8678 HighlightStyle {
8679 fade_out: Some(0.6),
8680 ..Default::default()
8681 },
8682 cx,
8683 );
8684 let rename_focus_handle = rename_editor.focus_handle(cx);
8685 cx.focus(&rename_focus_handle);
8686 let block_id = this.insert_blocks(
8687 [BlockProperties {
8688 style: BlockStyle::Flex,
8689 position: range.start,
8690 height: 1,
8691 render: Box::new({
8692 let rename_editor = rename_editor.clone();
8693 move |cx: &mut BlockContext| {
8694 let mut text_style = cx.editor_style.text.clone();
8695 if let Some(highlight_style) = old_highlight_id
8696 .and_then(|h| h.style(&cx.editor_style.syntax))
8697 {
8698 text_style = text_style.highlight(highlight_style);
8699 }
8700 div()
8701 .pl(cx.anchor_x)
8702 .child(EditorElement::new(
8703 &rename_editor,
8704 EditorStyle {
8705 background: cx.theme().system().transparent,
8706 local_player: cx.editor_style.local_player,
8707 text: text_style,
8708 scrollbar_width: cx.editor_style.scrollbar_width,
8709 syntax: cx.editor_style.syntax.clone(),
8710 status: cx.editor_style.status.clone(),
8711 inlay_hints_style: HighlightStyle {
8712 color: Some(cx.theme().status().hint),
8713 font_weight: Some(FontWeight::BOLD),
8714 ..HighlightStyle::default()
8715 },
8716 suggestions_style: HighlightStyle {
8717 color: Some(cx.theme().status().predictive),
8718 ..HighlightStyle::default()
8719 },
8720 },
8721 ))
8722 .into_any_element()
8723 }
8724 }),
8725 disposition: BlockDisposition::Below,
8726 }],
8727 Some(Autoscroll::fit()),
8728 cx,
8729 )[0];
8730 this.pending_rename = Some(RenameState {
8731 range,
8732 old_name,
8733 editor: rename_editor,
8734 block_id,
8735 });
8736 })?;
8737 }
8738
8739 Ok(())
8740 }))
8741 }
8742
8743 pub fn confirm_rename(
8744 &mut self,
8745 _: &ConfirmRename,
8746 cx: &mut ViewContext<Self>,
8747 ) -> Option<Task<Result<()>>> {
8748 let rename = self.take_rename(false, cx)?;
8749 let workspace = self.workspace()?;
8750 let (start_buffer, start) = self
8751 .buffer
8752 .read(cx)
8753 .text_anchor_for_position(rename.range.start, cx)?;
8754 let (end_buffer, end) = self
8755 .buffer
8756 .read(cx)
8757 .text_anchor_for_position(rename.range.end, cx)?;
8758 if start_buffer != end_buffer {
8759 return None;
8760 }
8761
8762 let buffer = start_buffer;
8763 let range = start..end;
8764 let old_name = rename.old_name;
8765 let new_name = rename.editor.read(cx).text(cx);
8766
8767 let rename = workspace
8768 .read(cx)
8769 .project()
8770 .clone()
8771 .update(cx, |project, cx| {
8772 project.perform_rename(buffer.clone(), range.start, new_name.clone(), true, cx)
8773 });
8774 let workspace = workspace.downgrade();
8775
8776 Some(cx.spawn(|editor, mut cx| async move {
8777 let project_transaction = rename.await?;
8778 Self::open_project_transaction(
8779 &editor,
8780 workspace,
8781 project_transaction,
8782 format!("Rename: {} → {}", old_name, new_name),
8783 cx.clone(),
8784 )
8785 .await?;
8786
8787 editor.update(&mut cx, |editor, cx| {
8788 editor.refresh_document_highlights(cx);
8789 })?;
8790 Ok(())
8791 }))
8792 }
8793
8794 fn take_rename(
8795 &mut self,
8796 moving_cursor: bool,
8797 cx: &mut ViewContext<Self>,
8798 ) -> Option<RenameState> {
8799 let rename = self.pending_rename.take()?;
8800 if rename.editor.focus_handle(cx).is_focused(cx) {
8801 cx.focus(&self.focus_handle);
8802 }
8803
8804 self.remove_blocks(
8805 [rename.block_id].into_iter().collect(),
8806 Some(Autoscroll::fit()),
8807 cx,
8808 );
8809 self.clear_highlights::<Rename>(cx);
8810 self.show_local_selections = true;
8811
8812 if moving_cursor {
8813 let rename_editor = rename.editor.read(cx);
8814 let cursor_in_rename_editor = rename_editor.selections.newest::<usize>(cx).head();
8815
8816 // Update the selection to match the position of the selection inside
8817 // the rename editor.
8818 let snapshot = self.buffer.read(cx).read(cx);
8819 let rename_range = rename.range.to_offset(&snapshot);
8820 let cursor_in_editor = snapshot
8821 .clip_offset(rename_range.start + cursor_in_rename_editor, Bias::Left)
8822 .min(rename_range.end);
8823 drop(snapshot);
8824
8825 self.change_selections(None, cx, |s| {
8826 s.select_ranges(vec![cursor_in_editor..cursor_in_editor])
8827 });
8828 } else {
8829 self.refresh_document_highlights(cx);
8830 }
8831
8832 Some(rename)
8833 }
8834
8835 pub fn pending_rename(&self) -> Option<&RenameState> {
8836 self.pending_rename.as_ref()
8837 }
8838
8839 fn format(&mut self, _: &Format, cx: &mut ViewContext<Self>) -> Option<Task<Result<()>>> {
8840 let project = match &self.project {
8841 Some(project) => project.clone(),
8842 None => return None,
8843 };
8844
8845 Some(self.perform_format(project, FormatTrigger::Manual, cx))
8846 }
8847
8848 fn perform_format(
8849 &mut self,
8850 project: Model<Project>,
8851 trigger: FormatTrigger,
8852 cx: &mut ViewContext<Self>,
8853 ) -> Task<Result<()>> {
8854 let buffer = self.buffer().clone();
8855 let mut buffers = buffer.read(cx).all_buffers();
8856 if trigger == FormatTrigger::Save {
8857 buffers.retain(|buffer| buffer.read(cx).is_dirty());
8858 }
8859
8860 let mut timeout = cx.background_executor().timer(FORMAT_TIMEOUT).fuse();
8861 let format = project.update(cx, |project, cx| project.format(buffers, true, trigger, cx));
8862
8863 cx.spawn(|_, mut cx| async move {
8864 let transaction = futures::select_biased! {
8865 () = timeout => {
8866 log::warn!("timed out waiting for formatting");
8867 None
8868 }
8869 transaction = format.log_err().fuse() => transaction,
8870 };
8871
8872 buffer
8873 .update(&mut cx, |buffer, cx| {
8874 if let Some(transaction) = transaction {
8875 if !buffer.is_singleton() {
8876 buffer.push_transaction(&transaction.0, cx);
8877 }
8878 }
8879
8880 cx.notify();
8881 })
8882 .ok();
8883
8884 Ok(())
8885 })
8886 }
8887
8888 fn restart_language_server(&mut self, _: &RestartLanguageServer, cx: &mut ViewContext<Self>) {
8889 if let Some(project) = self.project.clone() {
8890 self.buffer.update(cx, |multi_buffer, cx| {
8891 project.update(cx, |project, cx| {
8892 project.restart_language_servers_for_buffers(multi_buffer.all_buffers(), cx);
8893 });
8894 })
8895 }
8896 }
8897
8898 fn show_character_palette(&mut self, _: &ShowCharacterPalette, cx: &mut ViewContext<Self>) {
8899 cx.show_character_palette();
8900 }
8901
8902 fn refresh_active_diagnostics(&mut self, cx: &mut ViewContext<Editor>) {
8903 if let Some(active_diagnostics) = self.active_diagnostics.as_mut() {
8904 let buffer = self.buffer.read(cx).snapshot(cx);
8905 let primary_range_start = active_diagnostics.primary_range.start.to_offset(&buffer);
8906 let is_valid = buffer
8907 .diagnostics_in_range::<_, usize>(active_diagnostics.primary_range.clone(), false)
8908 .any(|entry| {
8909 entry.diagnostic.is_primary
8910 && !entry.range.is_empty()
8911 && entry.range.start == primary_range_start
8912 && entry.diagnostic.message == active_diagnostics.primary_message
8913 });
8914
8915 if is_valid != active_diagnostics.is_valid {
8916 active_diagnostics.is_valid = is_valid;
8917 let mut new_styles = HashMap::default();
8918 for (block_id, diagnostic) in &active_diagnostics.blocks {
8919 new_styles.insert(
8920 *block_id,
8921 diagnostic_block_renderer(diagnostic.clone(), is_valid),
8922 );
8923 }
8924 self.display_map
8925 .update(cx, |display_map, _| display_map.replace_blocks(new_styles));
8926 }
8927 }
8928 }
8929
8930 fn activate_diagnostics(&mut self, group_id: usize, cx: &mut ViewContext<Self>) -> bool {
8931 self.dismiss_diagnostics(cx);
8932 let snapshot = self.snapshot(cx);
8933 self.active_diagnostics = self.display_map.update(cx, |display_map, cx| {
8934 let buffer = self.buffer.read(cx).snapshot(cx);
8935
8936 let mut primary_range = None;
8937 let mut primary_message = None;
8938 let mut group_end = Point::zero();
8939 let diagnostic_group = buffer
8940 .diagnostic_group::<Point>(group_id)
8941 .filter_map(|entry| {
8942 if snapshot.is_line_folded(entry.range.start.row)
8943 && (entry.range.start.row == entry.range.end.row
8944 || snapshot.is_line_folded(entry.range.end.row))
8945 {
8946 return None;
8947 }
8948 if entry.range.end > group_end {
8949 group_end = entry.range.end;
8950 }
8951 if entry.diagnostic.is_primary {
8952 primary_range = Some(entry.range.clone());
8953 primary_message = Some(entry.diagnostic.message.clone());
8954 }
8955 Some(entry)
8956 })
8957 .collect::<Vec<_>>();
8958 let primary_range = primary_range?;
8959 let primary_message = primary_message?;
8960 let primary_range =
8961 buffer.anchor_after(primary_range.start)..buffer.anchor_before(primary_range.end);
8962
8963 let blocks = display_map
8964 .insert_blocks(
8965 diagnostic_group.iter().map(|entry| {
8966 let diagnostic = entry.diagnostic.clone();
8967 let message_height = diagnostic.message.matches('\n').count() as u8 + 1;
8968 BlockProperties {
8969 style: BlockStyle::Fixed,
8970 position: buffer.anchor_after(entry.range.start),
8971 height: message_height,
8972 render: diagnostic_block_renderer(diagnostic, true),
8973 disposition: BlockDisposition::Below,
8974 }
8975 }),
8976 cx,
8977 )
8978 .into_iter()
8979 .zip(diagnostic_group.into_iter().map(|entry| entry.diagnostic))
8980 .collect();
8981
8982 Some(ActiveDiagnosticGroup {
8983 primary_range,
8984 primary_message,
8985 blocks,
8986 is_valid: true,
8987 })
8988 });
8989 self.active_diagnostics.is_some()
8990 }
8991
8992 fn dismiss_diagnostics(&mut self, cx: &mut ViewContext<Self>) {
8993 if let Some(active_diagnostic_group) = self.active_diagnostics.take() {
8994 self.display_map.update(cx, |display_map, cx| {
8995 display_map.remove_blocks(active_diagnostic_group.blocks.into_keys().collect(), cx);
8996 });
8997 cx.notify();
8998 }
8999 }
9000
9001 pub fn set_selections_from_remote(
9002 &mut self,
9003 selections: Vec<Selection<Anchor>>,
9004 pending_selection: Option<Selection<Anchor>>,
9005 cx: &mut ViewContext<Self>,
9006 ) {
9007 let old_cursor_position = self.selections.newest_anchor().head();
9008 self.selections.change_with(cx, |s| {
9009 s.select_anchors(selections);
9010 if let Some(pending_selection) = pending_selection {
9011 s.set_pending(pending_selection, SelectMode::Character);
9012 } else {
9013 s.clear_pending();
9014 }
9015 });
9016 self.selections_did_change(false, &old_cursor_position, cx);
9017 }
9018
9019 fn push_to_selection_history(&mut self) {
9020 self.selection_history.push(SelectionHistoryEntry {
9021 selections: self.selections.disjoint_anchors(),
9022 select_next_state: self.select_next_state.clone(),
9023 select_prev_state: self.select_prev_state.clone(),
9024 add_selections_state: self.add_selections_state.clone(),
9025 });
9026 }
9027
9028 pub fn transact(
9029 &mut self,
9030 cx: &mut ViewContext<Self>,
9031 update: impl FnOnce(&mut Self, &mut ViewContext<Self>),
9032 ) -> Option<TransactionId> {
9033 self.start_transaction_at(Instant::now(), cx);
9034 update(self, cx);
9035 self.end_transaction_at(Instant::now(), cx)
9036 }
9037
9038 fn start_transaction_at(&mut self, now: Instant, cx: &mut ViewContext<Self>) {
9039 self.end_selection(cx);
9040 if let Some(tx_id) = self
9041 .buffer
9042 .update(cx, |buffer, cx| buffer.start_transaction_at(now, cx))
9043 {
9044 self.selection_history
9045 .insert_transaction(tx_id, self.selections.disjoint_anchors());
9046 cx.emit(EditorEvent::TransactionBegun {
9047 transaction_id: tx_id,
9048 })
9049 }
9050 }
9051
9052 fn end_transaction_at(
9053 &mut self,
9054 now: Instant,
9055 cx: &mut ViewContext<Self>,
9056 ) -> Option<TransactionId> {
9057 if let Some(tx_id) = self
9058 .buffer
9059 .update(cx, |buffer, cx| buffer.end_transaction_at(now, cx))
9060 {
9061 if let Some((_, end_selections)) = self.selection_history.transaction_mut(tx_id) {
9062 *end_selections = Some(self.selections.disjoint_anchors());
9063 } else {
9064 log::error!("unexpectedly ended a transaction that wasn't started by this editor");
9065 }
9066
9067 cx.emit(EditorEvent::Edited);
9068 Some(tx_id)
9069 } else {
9070 None
9071 }
9072 }
9073
9074 pub fn fold(&mut self, _: &actions::Fold, cx: &mut ViewContext<Self>) {
9075 let mut fold_ranges = Vec::new();
9076
9077 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
9078
9079 let selections = self.selections.all_adjusted(cx);
9080 for selection in selections {
9081 let range = selection.range().sorted();
9082 let buffer_start_row = range.start.row;
9083
9084 for row in (0..=range.end.row).rev() {
9085 let fold_range = display_map.foldable_range(row);
9086
9087 if let Some(fold_range) = fold_range {
9088 if fold_range.end.row >= buffer_start_row {
9089 fold_ranges.push(fold_range);
9090 if row <= range.start.row {
9091 break;
9092 }
9093 }
9094 }
9095 }
9096 }
9097
9098 self.fold_ranges(fold_ranges, true, cx);
9099 }
9100
9101 pub fn fold_at(&mut self, fold_at: &FoldAt, cx: &mut ViewContext<Self>) {
9102 let buffer_row = fold_at.buffer_row;
9103 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
9104
9105 if let Some(fold_range) = display_map.foldable_range(buffer_row) {
9106 let autoscroll = self
9107 .selections
9108 .all::<Point>(cx)
9109 .iter()
9110 .any(|selection| fold_range.overlaps(&selection.range()));
9111
9112 self.fold_ranges(std::iter::once(fold_range), autoscroll, cx);
9113 }
9114 }
9115
9116 pub fn unfold_lines(&mut self, _: &UnfoldLines, cx: &mut ViewContext<Self>) {
9117 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
9118 let buffer = &display_map.buffer_snapshot;
9119 let selections = self.selections.all::<Point>(cx);
9120 let ranges = selections
9121 .iter()
9122 .map(|s| {
9123 let range = s.display_range(&display_map).sorted();
9124 let mut start = range.start.to_point(&display_map);
9125 let mut end = range.end.to_point(&display_map);
9126 start.column = 0;
9127 end.column = buffer.line_len(end.row);
9128 start..end
9129 })
9130 .collect::<Vec<_>>();
9131
9132 self.unfold_ranges(ranges, true, true, cx);
9133 }
9134
9135 pub fn unfold_at(&mut self, unfold_at: &UnfoldAt, cx: &mut ViewContext<Self>) {
9136 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
9137
9138 let intersection_range = Point::new(unfold_at.buffer_row, 0)
9139 ..Point::new(
9140 unfold_at.buffer_row,
9141 display_map.buffer_snapshot.line_len(unfold_at.buffer_row),
9142 );
9143
9144 let autoscroll = self
9145 .selections
9146 .all::<Point>(cx)
9147 .iter()
9148 .any(|selection| selection.range().overlaps(&intersection_range));
9149
9150 self.unfold_ranges(std::iter::once(intersection_range), true, autoscroll, cx)
9151 }
9152
9153 pub fn fold_selected_ranges(&mut self, _: &FoldSelectedRanges, cx: &mut ViewContext<Self>) {
9154 let selections = self.selections.all::<Point>(cx);
9155 let display_map = self.display_map.update(cx, |map, cx| map.snapshot(cx));
9156 let line_mode = self.selections.line_mode;
9157 let ranges = selections.into_iter().map(|s| {
9158 if line_mode {
9159 let start = Point::new(s.start.row, 0);
9160 let end = Point::new(s.end.row, display_map.buffer_snapshot.line_len(s.end.row));
9161 start..end
9162 } else {
9163 s.start..s.end
9164 }
9165 });
9166 self.fold_ranges(ranges, true, cx);
9167 }
9168
9169 pub fn fold_ranges<T: ToOffset + Clone>(
9170 &mut self,
9171 ranges: impl IntoIterator<Item = Range<T>>,
9172 auto_scroll: bool,
9173 cx: &mut ViewContext<Self>,
9174 ) {
9175 let mut fold_ranges = Vec::new();
9176 let mut buffers_affected = HashMap::default();
9177 let multi_buffer = self.buffer().read(cx);
9178 for range in ranges {
9179 if let Some((_, buffer, _)) = multi_buffer.excerpt_containing(range.start.clone(), cx) {
9180 buffers_affected.insert(buffer.read(cx).remote_id(), buffer);
9181 };
9182 fold_ranges.push(range);
9183 }
9184
9185 let mut ranges = fold_ranges.into_iter().peekable();
9186 if ranges.peek().is_some() {
9187 self.display_map.update(cx, |map, cx| map.fold(ranges, cx));
9188
9189 if auto_scroll {
9190 self.request_autoscroll(Autoscroll::fit(), cx);
9191 }
9192
9193 for buffer in buffers_affected.into_values() {
9194 self.sync_expanded_diff_hunks(buffer, cx);
9195 }
9196
9197 cx.notify();
9198
9199 if let Some(active_diagnostics) = self.active_diagnostics.take() {
9200 // Clear diagnostics block when folding a range that contains it.
9201 let snapshot = self.snapshot(cx);
9202 if snapshot.intersects_fold(active_diagnostics.primary_range.start) {
9203 drop(snapshot);
9204 self.active_diagnostics = Some(active_diagnostics);
9205 self.dismiss_diagnostics(cx);
9206 } else {
9207 self.active_diagnostics = Some(active_diagnostics);
9208 }
9209 }
9210 }
9211 }
9212
9213 pub fn unfold_ranges<T: ToOffset + Clone>(
9214 &mut self,
9215 ranges: impl IntoIterator<Item = Range<T>>,
9216 inclusive: bool,
9217 auto_scroll: bool,
9218 cx: &mut ViewContext<Self>,
9219 ) {
9220 let mut unfold_ranges = Vec::new();
9221 let mut buffers_affected = HashMap::default();
9222 let multi_buffer = self.buffer().read(cx);
9223 for range in ranges {
9224 if let Some((_, buffer, _)) = multi_buffer.excerpt_containing(range.start.clone(), cx) {
9225 buffers_affected.insert(buffer.read(cx).remote_id(), buffer);
9226 };
9227 unfold_ranges.push(range);
9228 }
9229
9230 let mut ranges = unfold_ranges.into_iter().peekable();
9231 if ranges.peek().is_some() {
9232 self.display_map
9233 .update(cx, |map, cx| map.unfold(ranges, inclusive, cx));
9234 if auto_scroll {
9235 self.request_autoscroll(Autoscroll::fit(), cx);
9236 }
9237
9238 for buffer in buffers_affected.into_values() {
9239 self.sync_expanded_diff_hunks(buffer, cx);
9240 }
9241
9242 cx.notify();
9243 }
9244 }
9245
9246 pub fn set_gutter_hovered(&mut self, hovered: bool, cx: &mut ViewContext<Self>) {
9247 if hovered != self.gutter_hovered {
9248 self.gutter_hovered = hovered;
9249 cx.notify();
9250 }
9251 }
9252
9253 pub fn insert_blocks(
9254 &mut self,
9255 blocks: impl IntoIterator<Item = BlockProperties<Anchor>>,
9256 autoscroll: Option<Autoscroll>,
9257 cx: &mut ViewContext<Self>,
9258 ) -> Vec<BlockId> {
9259 let blocks = self
9260 .display_map
9261 .update(cx, |display_map, cx| display_map.insert_blocks(blocks, cx));
9262 if let Some(autoscroll) = autoscroll {
9263 self.request_autoscroll(autoscroll, cx);
9264 }
9265 blocks
9266 }
9267
9268 pub fn replace_blocks(
9269 &mut self,
9270 blocks: HashMap<BlockId, RenderBlock>,
9271 autoscroll: Option<Autoscroll>,
9272 cx: &mut ViewContext<Self>,
9273 ) {
9274 self.display_map
9275 .update(cx, |display_map, _| display_map.replace_blocks(blocks));
9276 if let Some(autoscroll) = autoscroll {
9277 self.request_autoscroll(autoscroll, cx);
9278 }
9279 }
9280
9281 pub fn remove_blocks(
9282 &mut self,
9283 block_ids: HashSet<BlockId>,
9284 autoscroll: Option<Autoscroll>,
9285 cx: &mut ViewContext<Self>,
9286 ) {
9287 self.display_map.update(cx, |display_map, cx| {
9288 display_map.remove_blocks(block_ids, cx)
9289 });
9290 if let Some(autoscroll) = autoscroll {
9291 self.request_autoscroll(autoscroll, cx);
9292 }
9293 }
9294
9295 pub fn longest_row(&self, cx: &mut AppContext) -> u32 {
9296 self.display_map
9297 .update(cx, |map, cx| map.snapshot(cx))
9298 .longest_row()
9299 }
9300
9301 pub fn max_point(&self, cx: &mut AppContext) -> DisplayPoint {
9302 self.display_map
9303 .update(cx, |map, cx| map.snapshot(cx))
9304 .max_point()
9305 }
9306
9307 pub fn text(&self, cx: &AppContext) -> String {
9308 self.buffer.read(cx).read(cx).text()
9309 }
9310
9311 pub fn text_option(&self, cx: &AppContext) -> Option<String> {
9312 let text = self.text(cx);
9313 let text = text.trim();
9314
9315 if text.is_empty() {
9316 return None;
9317 }
9318
9319 Some(text.to_string())
9320 }
9321
9322 pub fn set_text(&mut self, text: impl Into<Arc<str>>, cx: &mut ViewContext<Self>) {
9323 self.transact(cx, |this, cx| {
9324 this.buffer
9325 .read(cx)
9326 .as_singleton()
9327 .expect("you can only call set_text on editors for singleton buffers")
9328 .update(cx, |buffer, cx| buffer.set_text(text, cx));
9329 });
9330 }
9331
9332 pub fn display_text(&self, cx: &mut AppContext) -> String {
9333 self.display_map
9334 .update(cx, |map, cx| map.snapshot(cx))
9335 .text()
9336 }
9337
9338 pub fn wrap_guides(&self, cx: &AppContext) -> SmallVec<[(usize, bool); 2]> {
9339 let mut wrap_guides = smallvec::smallvec![];
9340
9341 if self.show_wrap_guides == Some(false) {
9342 return wrap_guides;
9343 }
9344
9345 let settings = self.buffer.read(cx).settings_at(0, cx);
9346 if settings.show_wrap_guides {
9347 if let SoftWrap::Column(soft_wrap) = self.soft_wrap_mode(cx) {
9348 wrap_guides.push((soft_wrap as usize, true));
9349 }
9350 wrap_guides.extend(settings.wrap_guides.iter().map(|guide| (*guide, false)))
9351 }
9352
9353 wrap_guides
9354 }
9355
9356 pub fn soft_wrap_mode(&self, cx: &AppContext) -> SoftWrap {
9357 let settings = self.buffer.read(cx).settings_at(0, cx);
9358 let mode = self
9359 .soft_wrap_mode_override
9360 .unwrap_or_else(|| settings.soft_wrap);
9361 match mode {
9362 language_settings::SoftWrap::None => SoftWrap::None,
9363 language_settings::SoftWrap::PreferLine => SoftWrap::PreferLine,
9364 language_settings::SoftWrap::EditorWidth => SoftWrap::EditorWidth,
9365 language_settings::SoftWrap::PreferredLineLength => {
9366 SoftWrap::Column(settings.preferred_line_length)
9367 }
9368 }
9369 }
9370
9371 pub fn set_soft_wrap_mode(
9372 &mut self,
9373 mode: language_settings::SoftWrap,
9374 cx: &mut ViewContext<Self>,
9375 ) {
9376 self.soft_wrap_mode_override = Some(mode);
9377 cx.notify();
9378 }
9379
9380 pub fn set_style(&mut self, style: EditorStyle, cx: &mut ViewContext<Self>) {
9381 let rem_size = cx.rem_size();
9382 self.display_map.update(cx, |map, cx| {
9383 map.set_font(
9384 style.text.font(),
9385 style.text.font_size.to_pixels(rem_size),
9386 cx,
9387 )
9388 });
9389 self.style = Some(style);
9390 }
9391
9392 pub fn style(&self) -> Option<&EditorStyle> {
9393 self.style.as_ref()
9394 }
9395
9396 // Called by the element. This method is not designed to be called outside of the editor
9397 // element's layout code because it does not notify when rewrapping is computed synchronously.
9398 pub(crate) fn set_wrap_width(&self, width: Option<Pixels>, cx: &mut AppContext) -> bool {
9399 self.display_map
9400 .update(cx, |map, cx| map.set_wrap_width(width, cx))
9401 }
9402
9403 pub fn toggle_soft_wrap(&mut self, _: &ToggleSoftWrap, cx: &mut ViewContext<Self>) {
9404 if self.soft_wrap_mode_override.is_some() {
9405 self.soft_wrap_mode_override.take();
9406 } else {
9407 let soft_wrap = match self.soft_wrap_mode(cx) {
9408 SoftWrap::None | SoftWrap::PreferLine => language_settings::SoftWrap::EditorWidth,
9409 SoftWrap::EditorWidth | SoftWrap::Column(_) => {
9410 language_settings::SoftWrap::PreferLine
9411 }
9412 };
9413 self.soft_wrap_mode_override = Some(soft_wrap);
9414 }
9415 cx.notify();
9416 }
9417
9418 pub fn toggle_line_numbers(&mut self, _: &ToggleLineNumbers, cx: &mut ViewContext<Self>) {
9419 let mut editor_settings = EditorSettings::get_global(cx).clone();
9420 editor_settings.gutter.line_numbers = !editor_settings.gutter.line_numbers;
9421 EditorSettings::override_global(editor_settings, cx);
9422 }
9423
9424 pub fn set_show_gutter(&mut self, show_gutter: bool, cx: &mut ViewContext<Self>) {
9425 self.show_gutter = show_gutter;
9426 cx.notify();
9427 }
9428
9429 pub fn set_show_wrap_guides(&mut self, show_gutter: bool, cx: &mut ViewContext<Self>) {
9430 self.show_wrap_guides = Some(show_gutter);
9431 cx.notify();
9432 }
9433
9434 pub fn reveal_in_finder(&mut self, _: &RevealInFinder, cx: &mut ViewContext<Self>) {
9435 if let Some(buffer) = self.buffer().read(cx).as_singleton() {
9436 if let Some(file) = buffer.read(cx).file().and_then(|f| f.as_local()) {
9437 cx.reveal_path(&file.abs_path(cx));
9438 }
9439 }
9440 }
9441
9442 pub fn copy_path(&mut self, _: &CopyPath, cx: &mut ViewContext<Self>) {
9443 if let Some(buffer) = self.buffer().read(cx).as_singleton() {
9444 if let Some(file) = buffer.read(cx).file().and_then(|f| f.as_local()) {
9445 if let Some(path) = file.abs_path(cx).to_str() {
9446 cx.write_to_clipboard(ClipboardItem::new(path.to_string()));
9447 }
9448 }
9449 }
9450 }
9451
9452 pub fn copy_relative_path(&mut self, _: &CopyRelativePath, cx: &mut ViewContext<Self>) {
9453 if let Some(buffer) = self.buffer().read(cx).as_singleton() {
9454 if let Some(file) = buffer.read(cx).file().and_then(|f| f.as_local()) {
9455 if let Some(path) = file.path().to_str() {
9456 cx.write_to_clipboard(ClipboardItem::new(path.to_string()));
9457 }
9458 }
9459 }
9460 }
9461
9462 pub fn toggle_git_blame(&mut self, _: &ToggleGitBlame, cx: &mut ViewContext<Self>) {
9463 self.show_git_blame_gutter = !self.show_git_blame_gutter;
9464
9465 if self.show_git_blame_gutter && !self.has_blame_entries(cx) {
9466 self.start_git_blame(true, cx);
9467 }
9468
9469 cx.notify();
9470 }
9471
9472 pub fn toggle_git_blame_inline(
9473 &mut self,
9474 _: &ToggleGitBlameInline,
9475 cx: &mut ViewContext<Self>,
9476 ) {
9477 self.toggle_git_blame_inline_internal(true, cx);
9478 cx.notify();
9479 }
9480
9481 pub fn git_blame_inline_enabled(&self) -> bool {
9482 self.git_blame_inline_enabled
9483 }
9484
9485 fn start_git_blame(&mut self, user_triggered: bool, cx: &mut ViewContext<Self>) {
9486 if let Some(project) = self.project.as_ref() {
9487 let Some(buffer) = self.buffer().read(cx).as_singleton() else {
9488 return;
9489 };
9490
9491 if buffer.read(cx).file().is_none() {
9492 return;
9493 }
9494
9495 let focused = self.focus_handle(cx).contains_focused(cx);
9496
9497 let project = project.clone();
9498 let blame =
9499 cx.new_model(|cx| GitBlame::new(buffer, project, user_triggered, focused, cx));
9500 self.blame_subscription = Some(cx.observe(&blame, |_, _, cx| cx.notify()));
9501 self.blame = Some(blame);
9502 }
9503 }
9504
9505 fn toggle_git_blame_inline_internal(
9506 &mut self,
9507 user_triggered: bool,
9508 cx: &mut ViewContext<Self>,
9509 ) {
9510 if self.git_blame_inline_enabled {
9511 self.git_blame_inline_enabled = false;
9512 self.show_git_blame_inline = false;
9513 self.show_git_blame_inline_delay_task.take();
9514 } else {
9515 self.git_blame_inline_enabled = true;
9516 self.start_git_blame_inline(user_triggered, cx);
9517 }
9518
9519 cx.notify();
9520 }
9521
9522 fn start_git_blame_inline(&mut self, user_triggered: bool, cx: &mut ViewContext<Self>) {
9523 self.start_git_blame(user_triggered, cx);
9524
9525 if ProjectSettings::get_global(cx)
9526 .git
9527 .inline_blame_delay()
9528 .is_some()
9529 {
9530 self.start_inline_blame_timer(cx);
9531 } else {
9532 self.show_git_blame_inline = true
9533 }
9534 }
9535
9536 pub fn blame(&self) -> Option<&Model<GitBlame>> {
9537 self.blame.as_ref()
9538 }
9539
9540 pub fn render_git_blame_gutter(&mut self, cx: &mut WindowContext) -> bool {
9541 self.show_git_blame_gutter && self.has_blame_entries(cx)
9542 }
9543
9544 pub fn render_git_blame_inline(&mut self, cx: &mut WindowContext) -> bool {
9545 self.show_git_blame_inline
9546 && self.focus_handle.is_focused(cx)
9547 && !self.newest_selection_head_on_empty_line(cx)
9548 && self.has_blame_entries(cx)
9549 }
9550
9551 fn has_blame_entries(&self, cx: &mut WindowContext) -> bool {
9552 self.blame()
9553 .map_or(false, |blame| blame.read(cx).has_generated_entries())
9554 }
9555
9556 fn newest_selection_head_on_empty_line(&mut self, cx: &mut WindowContext) -> bool {
9557 let cursor_anchor = self.selections.newest_anchor().head();
9558
9559 let snapshot = self.buffer.read(cx).snapshot(cx);
9560 let buffer_row = cursor_anchor.to_point(&snapshot).row;
9561
9562 snapshot.line_len(buffer_row) == 0
9563 }
9564
9565 fn get_permalink_to_line(&mut self, cx: &mut ViewContext<Self>) -> Result<url::Url> {
9566 let (path, repo) = maybe!({
9567 let project_handle = self.project.as_ref()?.clone();
9568 let project = project_handle.read(cx);
9569 let buffer = self.buffer().read(cx).as_singleton()?;
9570 let path = buffer
9571 .read(cx)
9572 .file()?
9573 .as_local()?
9574 .path()
9575 .to_str()?
9576 .to_string();
9577 let repo = project.get_repo(&buffer.read(cx).project_path(cx)?, cx)?;
9578 Some((path, repo))
9579 })
9580 .ok_or_else(|| anyhow!("unable to open git repository"))?;
9581
9582 const REMOTE_NAME: &str = "origin";
9583 let origin_url = repo
9584 .lock()
9585 .remote_url(REMOTE_NAME)
9586 .ok_or_else(|| anyhow!("remote \"{REMOTE_NAME}\" not found"))?;
9587 let sha = repo
9588 .lock()
9589 .head_sha()
9590 .ok_or_else(|| anyhow!("failed to read HEAD SHA"))?;
9591 let selections = self.selections.all::<Point>(cx);
9592 let selection = selections.iter().peekable().next();
9593
9594 let (provider, remote) =
9595 parse_git_remote_url(GitHostingProviderRegistry::default_global(cx), &origin_url)
9596 .ok_or_else(|| anyhow!("failed to parse Git remote URL"))?;
9597
9598 Ok(provider.build_permalink(
9599 remote,
9600 BuildPermalinkParams {
9601 sha: &sha,
9602 path: &path,
9603 selection: selection.map(|selection| {
9604 let range = selection.range();
9605 let start = range.start.row;
9606 let end = range.end.row;
9607 start..end
9608 }),
9609 },
9610 ))
9611 }
9612
9613 pub fn copy_permalink_to_line(&mut self, _: &CopyPermalinkToLine, cx: &mut ViewContext<Self>) {
9614 let permalink = self.get_permalink_to_line(cx);
9615
9616 match permalink {
9617 Ok(permalink) => {
9618 cx.write_to_clipboard(ClipboardItem::new(permalink.to_string()));
9619 }
9620 Err(err) => {
9621 let message = format!("Failed to copy permalink: {err}");
9622
9623 Err::<(), anyhow::Error>(err).log_err();
9624
9625 if let Some(workspace) = self.workspace() {
9626 workspace.update(cx, |workspace, cx| {
9627 struct CopyPermalinkToLine;
9628
9629 workspace.show_toast(
9630 Toast::new(NotificationId::unique::<CopyPermalinkToLine>(), message),
9631 cx,
9632 )
9633 })
9634 }
9635 }
9636 }
9637 }
9638
9639 pub fn open_permalink_to_line(&mut self, _: &OpenPermalinkToLine, cx: &mut ViewContext<Self>) {
9640 let permalink = self.get_permalink_to_line(cx);
9641
9642 match permalink {
9643 Ok(permalink) => {
9644 cx.open_url(permalink.as_ref());
9645 }
9646 Err(err) => {
9647 let message = format!("Failed to open permalink: {err}");
9648
9649 Err::<(), anyhow::Error>(err).log_err();
9650
9651 if let Some(workspace) = self.workspace() {
9652 workspace.update(cx, |workspace, cx| {
9653 struct OpenPermalinkToLine;
9654
9655 workspace.show_toast(
9656 Toast::new(NotificationId::unique::<OpenPermalinkToLine>(), message),
9657 cx,
9658 )
9659 })
9660 }
9661 }
9662 }
9663 }
9664
9665 /// Adds or removes (on `None` color) a highlight for the rows corresponding to the anchor range given.
9666 /// On matching anchor range, replaces the old highlight; does not clear the other existing highlights.
9667 /// If multiple anchor ranges will produce highlights for the same row, the last range added will be used.
9668 pub fn highlight_rows<T: 'static>(
9669 &mut self,
9670 rows: Range<Anchor>,
9671 color: Option<Hsla>,
9672 cx: &mut ViewContext<Self>,
9673 ) {
9674 let multi_buffer_snapshot = self.buffer().read(cx).snapshot(cx);
9675 match self.highlighted_rows.entry(TypeId::of::<T>()) {
9676 hash_map::Entry::Occupied(o) => {
9677 let row_highlights = o.into_mut();
9678 let existing_highlight_index =
9679 row_highlights.binary_search_by(|(_, highlight_range, _)| {
9680 highlight_range
9681 .start
9682 .cmp(&rows.start, &multi_buffer_snapshot)
9683 .then(highlight_range.end.cmp(&rows.end, &multi_buffer_snapshot))
9684 });
9685 match color {
9686 Some(color) => {
9687 let insert_index = match existing_highlight_index {
9688 Ok(i) => i,
9689 Err(i) => i,
9690 };
9691 row_highlights.insert(
9692 insert_index,
9693 (post_inc(&mut self.highlight_order), rows, color),
9694 );
9695 }
9696 None => {
9697 if let Ok(i) = existing_highlight_index {
9698 row_highlights.remove(i);
9699 }
9700 }
9701 }
9702 }
9703 hash_map::Entry::Vacant(v) => {
9704 if let Some(color) = color {
9705 v.insert(vec![(post_inc(&mut self.highlight_order), rows, color)]);
9706 }
9707 }
9708 }
9709 }
9710
9711 /// Clear all anchor ranges for a certain highlight context type, so no corresponding rows will be highlighted.
9712 pub fn clear_row_highlights<T: 'static>(&mut self) {
9713 self.highlighted_rows.remove(&TypeId::of::<T>());
9714 }
9715
9716 /// For a highlight given context type, gets all anchor ranges that will be used for row highlighting.
9717 pub fn highlighted_rows<T: 'static>(
9718 &self,
9719 ) -> Option<impl Iterator<Item = (&Range<Anchor>, &Hsla)>> {
9720 Some(
9721 self.highlighted_rows
9722 .get(&TypeId::of::<T>())?
9723 .iter()
9724 .map(|(_, range, color)| (range, color)),
9725 )
9726 }
9727
9728 /// Merges all anchor ranges for all context types ever set, picking the last highlight added in case of a row conflict.
9729 /// Rerturns a map of display rows that are highlighted and their corresponding highlight color.
9730 /// Allows to ignore certain kinds of highlights.
9731 pub fn highlighted_display_rows(
9732 &mut self,
9733 exclude_highlights: HashSet<TypeId>,
9734 cx: &mut WindowContext,
9735 ) -> BTreeMap<u32, Hsla> {
9736 let snapshot = self.snapshot(cx);
9737 let mut used_highlight_orders = HashMap::default();
9738 self.highlighted_rows
9739 .iter()
9740 .filter(|(type_id, _)| !exclude_highlights.contains(type_id))
9741 .flat_map(|(_, highlighted_rows)| highlighted_rows.iter())
9742 .fold(
9743 BTreeMap::<u32, Hsla>::new(),
9744 |mut unique_rows, (highlight_order, anchor_range, hsla)| {
9745 let start_row = anchor_range.start.to_display_point(&snapshot).row();
9746 let end_row = anchor_range.end.to_display_point(&snapshot).row();
9747 for row in start_row..=end_row {
9748 let used_index =
9749 used_highlight_orders.entry(row).or_insert(*highlight_order);
9750 if highlight_order >= used_index {
9751 *used_index = *highlight_order;
9752 unique_rows.insert(row, *hsla);
9753 }
9754 }
9755 unique_rows
9756 },
9757 )
9758 }
9759
9760 pub fn set_search_within_ranges(
9761 &mut self,
9762 ranges: &[Range<Anchor>],
9763 cx: &mut ViewContext<Self>,
9764 ) {
9765 self.highlight_background::<SearchWithinRange>(
9766 ranges,
9767 |colors| colors.editor_document_highlight_read_background,
9768 cx,
9769 )
9770 }
9771
9772 pub fn highlight_background<T: 'static>(
9773 &mut self,
9774 ranges: &[Range<Anchor>],
9775 color_fetcher: fn(&ThemeColors) -> Hsla,
9776 cx: &mut ViewContext<Self>,
9777 ) {
9778 let snapshot = self.snapshot(cx);
9779 // this is to try and catch a panic sooner
9780 for range in ranges {
9781 snapshot
9782 .buffer_snapshot
9783 .summary_for_anchor::<usize>(&range.start);
9784 snapshot
9785 .buffer_snapshot
9786 .summary_for_anchor::<usize>(&range.end);
9787 }
9788
9789 self.background_highlights
9790 .insert(TypeId::of::<T>(), (color_fetcher, Arc::from(ranges)));
9791 self.scrollbar_marker_state.dirty = true;
9792 cx.notify();
9793 }
9794
9795 pub fn clear_background_highlights<T: 'static>(
9796 &mut self,
9797 cx: &mut ViewContext<Self>,
9798 ) -> Option<BackgroundHighlight> {
9799 let text_highlights = self.background_highlights.remove(&TypeId::of::<T>())?;
9800 if !text_highlights.1.is_empty() {
9801 self.scrollbar_marker_state.dirty = true;
9802 cx.notify();
9803 }
9804 Some(text_highlights)
9805 }
9806
9807 #[cfg(feature = "test-support")]
9808 pub fn all_text_background_highlights(
9809 &mut self,
9810 cx: &mut ViewContext<Self>,
9811 ) -> Vec<(Range<DisplayPoint>, Hsla)> {
9812 let snapshot = self.snapshot(cx);
9813 let buffer = &snapshot.buffer_snapshot;
9814 let start = buffer.anchor_before(0);
9815 let end = buffer.anchor_after(buffer.len());
9816 let theme = cx.theme().colors();
9817 self.background_highlights_in_range(start..end, &snapshot, theme)
9818 }
9819
9820 fn document_highlights_for_position<'a>(
9821 &'a self,
9822 position: Anchor,
9823 buffer: &'a MultiBufferSnapshot,
9824 ) -> impl 'a + Iterator<Item = &Range<Anchor>> {
9825 let read_highlights = self
9826 .background_highlights
9827 .get(&TypeId::of::<DocumentHighlightRead>())
9828 .map(|h| &h.1);
9829 let write_highlights = self
9830 .background_highlights
9831 .get(&TypeId::of::<DocumentHighlightWrite>())
9832 .map(|h| &h.1);
9833 let left_position = position.bias_left(buffer);
9834 let right_position = position.bias_right(buffer);
9835 read_highlights
9836 .into_iter()
9837 .chain(write_highlights)
9838 .flat_map(move |ranges| {
9839 let start_ix = match ranges.binary_search_by(|probe| {
9840 let cmp = probe.end.cmp(&left_position, buffer);
9841 if cmp.is_ge() {
9842 Ordering::Greater
9843 } else {
9844 Ordering::Less
9845 }
9846 }) {
9847 Ok(i) | Err(i) => i,
9848 };
9849
9850 ranges[start_ix..]
9851 .iter()
9852 .take_while(move |range| range.start.cmp(&right_position, buffer).is_le())
9853 })
9854 }
9855
9856 pub fn has_background_highlights<T: 'static>(&self) -> bool {
9857 self.background_highlights
9858 .get(&TypeId::of::<T>())
9859 .map_or(false, |(_, highlights)| !highlights.is_empty())
9860 }
9861
9862 pub fn background_highlights_in_range(
9863 &self,
9864 search_range: Range<Anchor>,
9865 display_snapshot: &DisplaySnapshot,
9866 theme: &ThemeColors,
9867 ) -> Vec<(Range<DisplayPoint>, Hsla)> {
9868 let mut results = Vec::new();
9869 for (color_fetcher, ranges) in self.background_highlights.values() {
9870 let color = color_fetcher(theme);
9871 let start_ix = match ranges.binary_search_by(|probe| {
9872 let cmp = probe
9873 .end
9874 .cmp(&search_range.start, &display_snapshot.buffer_snapshot);
9875 if cmp.is_gt() {
9876 Ordering::Greater
9877 } else {
9878 Ordering::Less
9879 }
9880 }) {
9881 Ok(i) | Err(i) => i,
9882 };
9883 for range in &ranges[start_ix..] {
9884 if range
9885 .start
9886 .cmp(&search_range.end, &display_snapshot.buffer_snapshot)
9887 .is_ge()
9888 {
9889 break;
9890 }
9891
9892 let start = range.start.to_display_point(&display_snapshot);
9893 let end = range.end.to_display_point(&display_snapshot);
9894 results.push((start..end, color))
9895 }
9896 }
9897 results
9898 }
9899
9900 pub fn background_highlight_row_ranges<T: 'static>(
9901 &self,
9902 search_range: Range<Anchor>,
9903 display_snapshot: &DisplaySnapshot,
9904 count: usize,
9905 ) -> Vec<RangeInclusive<DisplayPoint>> {
9906 let mut results = Vec::new();
9907 let Some((_, ranges)) = self.background_highlights.get(&TypeId::of::<T>()) else {
9908 return vec![];
9909 };
9910
9911 let start_ix = match ranges.binary_search_by(|probe| {
9912 let cmp = probe
9913 .end
9914 .cmp(&search_range.start, &display_snapshot.buffer_snapshot);
9915 if cmp.is_gt() {
9916 Ordering::Greater
9917 } else {
9918 Ordering::Less
9919 }
9920 }) {
9921 Ok(i) | Err(i) => i,
9922 };
9923 let mut push_region = |start: Option<Point>, end: Option<Point>| {
9924 if let (Some(start_display), Some(end_display)) = (start, end) {
9925 results.push(
9926 start_display.to_display_point(display_snapshot)
9927 ..=end_display.to_display_point(display_snapshot),
9928 );
9929 }
9930 };
9931 let mut start_row: Option<Point> = None;
9932 let mut end_row: Option<Point> = None;
9933 if ranges.len() > count {
9934 return Vec::new();
9935 }
9936 for range in &ranges[start_ix..] {
9937 if range
9938 .start
9939 .cmp(&search_range.end, &display_snapshot.buffer_snapshot)
9940 .is_ge()
9941 {
9942 break;
9943 }
9944 let end = range.end.to_point(&display_snapshot.buffer_snapshot);
9945 if let Some(current_row) = &end_row {
9946 if end.row == current_row.row {
9947 continue;
9948 }
9949 }
9950 let start = range.start.to_point(&display_snapshot.buffer_snapshot);
9951 if start_row.is_none() {
9952 assert_eq!(end_row, None);
9953 start_row = Some(start);
9954 end_row = Some(end);
9955 continue;
9956 }
9957 if let Some(current_end) = end_row.as_mut() {
9958 if start.row > current_end.row + 1 {
9959 push_region(start_row, end_row);
9960 start_row = Some(start);
9961 end_row = Some(end);
9962 } else {
9963 // Merge two hunks.
9964 *current_end = end;
9965 }
9966 } else {
9967 unreachable!();
9968 }
9969 }
9970 // We might still have a hunk that was not rendered (if there was a search hit on the last line)
9971 push_region(start_row, end_row);
9972 results
9973 }
9974
9975 /// Get the text ranges corresponding to the redaction query
9976 pub fn redacted_ranges(
9977 &self,
9978 search_range: Range<Anchor>,
9979 display_snapshot: &DisplaySnapshot,
9980 cx: &WindowContext,
9981 ) -> Vec<Range<DisplayPoint>> {
9982 display_snapshot
9983 .buffer_snapshot
9984 .redacted_ranges(search_range, |file| {
9985 if let Some(file) = file {
9986 file.is_private()
9987 && EditorSettings::get(Some(file.as_ref().into()), cx).redact_private_values
9988 } else {
9989 false
9990 }
9991 })
9992 .map(|range| {
9993 range.start.to_display_point(display_snapshot)
9994 ..range.end.to_display_point(display_snapshot)
9995 })
9996 .collect()
9997 }
9998
9999 pub fn highlight_text<T: 'static>(
10000 &mut self,
10001 ranges: Vec<Range<Anchor>>,
10002 style: HighlightStyle,
10003 cx: &mut ViewContext<Self>,
10004 ) {
10005 self.display_map.update(cx, |map, _| {
10006 map.highlight_text(TypeId::of::<T>(), ranges, style)
10007 });
10008 cx.notify();
10009 }
10010
10011 pub(crate) fn highlight_inlays<T: 'static>(
10012 &mut self,
10013 highlights: Vec<InlayHighlight>,
10014 style: HighlightStyle,
10015 cx: &mut ViewContext<Self>,
10016 ) {
10017 self.display_map.update(cx, |map, _| {
10018 map.highlight_inlays(TypeId::of::<T>(), highlights, style)
10019 });
10020 cx.notify();
10021 }
10022
10023 pub fn text_highlights<'a, T: 'static>(
10024 &'a self,
10025 cx: &'a AppContext,
10026 ) -> Option<(HighlightStyle, &'a [Range<Anchor>])> {
10027 self.display_map.read(cx).text_highlights(TypeId::of::<T>())
10028 }
10029
10030 pub fn clear_highlights<T: 'static>(&mut self, cx: &mut ViewContext<Self>) {
10031 let cleared = self
10032 .display_map
10033 .update(cx, |map, _| map.clear_highlights(TypeId::of::<T>()));
10034 if cleared {
10035 cx.notify();
10036 }
10037 }
10038
10039 pub fn show_local_cursors(&self, cx: &WindowContext) -> bool {
10040 (self.read_only(cx) || self.blink_manager.read(cx).visible())
10041 && self.focus_handle.is_focused(cx)
10042 }
10043
10044 fn on_buffer_changed(&mut self, _: Model<MultiBuffer>, cx: &mut ViewContext<Self>) {
10045 cx.notify();
10046 }
10047
10048 fn on_buffer_event(
10049 &mut self,
10050 multibuffer: Model<MultiBuffer>,
10051 event: &multi_buffer::Event,
10052 cx: &mut ViewContext<Self>,
10053 ) {
10054 match event {
10055 multi_buffer::Event::Edited {
10056 singleton_buffer_edited,
10057 } => {
10058 self.scrollbar_marker_state.dirty = true;
10059 self.refresh_active_diagnostics(cx);
10060 self.refresh_code_actions(cx);
10061 if self.has_active_inline_completion(cx) {
10062 self.update_visible_inline_completion(cx);
10063 }
10064 cx.emit(EditorEvent::BufferEdited);
10065 cx.emit(SearchEvent::MatchesInvalidated);
10066
10067 if *singleton_buffer_edited {
10068 if let Some(project) = &self.project {
10069 let project = project.read(cx);
10070 let languages_affected = multibuffer
10071 .read(cx)
10072 .all_buffers()
10073 .into_iter()
10074 .filter_map(|buffer| {
10075 let buffer = buffer.read(cx);
10076 let language = buffer.language()?;
10077 if project.is_local()
10078 && project.language_servers_for_buffer(buffer, cx).count() == 0
10079 {
10080 None
10081 } else {
10082 Some(language)
10083 }
10084 })
10085 .cloned()
10086 .collect::<HashSet<_>>();
10087 if !languages_affected.is_empty() {
10088 self.refresh_inlay_hints(
10089 InlayHintRefreshReason::BufferEdited(languages_affected),
10090 cx,
10091 );
10092 }
10093 }
10094 }
10095
10096 let Some(project) = &self.project else { return };
10097 let telemetry = project.read(cx).client().telemetry().clone();
10098 telemetry.log_edit_event("editor");
10099 }
10100 multi_buffer::Event::ExcerptsAdded {
10101 buffer,
10102 predecessor,
10103 excerpts,
10104 } => {
10105 cx.emit(EditorEvent::ExcerptsAdded {
10106 buffer: buffer.clone(),
10107 predecessor: *predecessor,
10108 excerpts: excerpts.clone(),
10109 });
10110 self.refresh_inlay_hints(InlayHintRefreshReason::NewLinesShown, cx);
10111 }
10112 multi_buffer::Event::ExcerptsRemoved { ids } => {
10113 self.refresh_inlay_hints(InlayHintRefreshReason::ExcerptsRemoved(ids.clone()), cx);
10114 cx.emit(EditorEvent::ExcerptsRemoved { ids: ids.clone() })
10115 }
10116 multi_buffer::Event::Reparsed => {
10117 self.tasks_update_task = Some(self.refresh_runnables(cx));
10118
10119 cx.emit(EditorEvent::Reparsed);
10120 }
10121 multi_buffer::Event::LanguageChanged => {
10122 cx.emit(EditorEvent::Reparsed);
10123 cx.notify();
10124 }
10125 multi_buffer::Event::DirtyChanged => cx.emit(EditorEvent::DirtyChanged),
10126 multi_buffer::Event::Saved => cx.emit(EditorEvent::Saved),
10127 multi_buffer::Event::FileHandleChanged | multi_buffer::Event::Reloaded => {
10128 cx.emit(EditorEvent::TitleChanged)
10129 }
10130 multi_buffer::Event::DiffBaseChanged => {
10131 self.scrollbar_marker_state.dirty = true;
10132 cx.emit(EditorEvent::DiffBaseChanged);
10133 cx.notify();
10134 }
10135 multi_buffer::Event::DiffUpdated { buffer } => {
10136 self.sync_expanded_diff_hunks(buffer.clone(), cx);
10137 cx.notify();
10138 }
10139 multi_buffer::Event::Closed => cx.emit(EditorEvent::Closed),
10140 multi_buffer::Event::DiagnosticsUpdated => {
10141 self.refresh_active_diagnostics(cx);
10142 self.scrollbar_marker_state.dirty = true;
10143 cx.notify();
10144 }
10145 _ => {}
10146 };
10147 }
10148
10149 fn on_display_map_changed(&mut self, _: Model<DisplayMap>, cx: &mut ViewContext<Self>) {
10150 cx.notify();
10151 }
10152
10153 fn settings_changed(&mut self, cx: &mut ViewContext<Self>) {
10154 self.refresh_inline_completion(true, cx);
10155 self.refresh_inlay_hints(
10156 InlayHintRefreshReason::SettingsChange(inlay_hint_settings(
10157 self.selections.newest_anchor().head(),
10158 &self.buffer.read(cx).snapshot(cx),
10159 cx,
10160 )),
10161 cx,
10162 );
10163 let editor_settings = EditorSettings::get_global(cx);
10164 self.scroll_manager.vertical_scroll_margin = editor_settings.vertical_scroll_margin;
10165 self.show_breadcrumbs = editor_settings.toolbar.breadcrumbs;
10166
10167 if self.mode == EditorMode::Full {
10168 let inline_blame_enabled = ProjectSettings::get_global(cx).git.inline_blame_enabled();
10169 if self.git_blame_inline_enabled != inline_blame_enabled {
10170 self.toggle_git_blame_inline_internal(false, cx);
10171 }
10172 }
10173
10174 cx.notify();
10175 }
10176
10177 pub fn set_searchable(&mut self, searchable: bool) {
10178 self.searchable = searchable;
10179 }
10180
10181 pub fn searchable(&self) -> bool {
10182 self.searchable
10183 }
10184
10185 fn open_excerpts_in_split(&mut self, _: &OpenExcerptsSplit, cx: &mut ViewContext<Self>) {
10186 self.open_excerpts_common(true, cx)
10187 }
10188
10189 fn open_excerpts(&mut self, _: &OpenExcerpts, cx: &mut ViewContext<Self>) {
10190 self.open_excerpts_common(false, cx)
10191 }
10192
10193 fn open_excerpts_common(&mut self, split: bool, cx: &mut ViewContext<Self>) {
10194 let buffer = self.buffer.read(cx);
10195 if buffer.is_singleton() {
10196 cx.propagate();
10197 return;
10198 }
10199
10200 let Some(workspace) = self.workspace() else {
10201 cx.propagate();
10202 return;
10203 };
10204
10205 let mut new_selections_by_buffer = HashMap::default();
10206 for selection in self.selections.all::<usize>(cx) {
10207 for (buffer, mut range, _) in
10208 buffer.range_to_buffer_ranges(selection.start..selection.end, cx)
10209 {
10210 if selection.reversed {
10211 mem::swap(&mut range.start, &mut range.end);
10212 }
10213 new_selections_by_buffer
10214 .entry(buffer)
10215 .or_insert(Vec::new())
10216 .push(range)
10217 }
10218 }
10219
10220 // We defer the pane interaction because we ourselves are a workspace item
10221 // and activating a new item causes the pane to call a method on us reentrantly,
10222 // which panics if we're on the stack.
10223 cx.window_context().defer(move |cx| {
10224 workspace.update(cx, |workspace, cx| {
10225 let pane = if split {
10226 workspace.adjacent_pane(cx)
10227 } else {
10228 workspace.active_pane().clone()
10229 };
10230
10231 for (buffer, ranges) in new_selections_by_buffer {
10232 let editor = workspace.open_project_item::<Self>(pane.clone(), buffer, cx);
10233 editor.update(cx, |editor, cx| {
10234 editor.change_selections(Some(Autoscroll::newest()), cx, |s| {
10235 s.select_ranges(ranges);
10236 });
10237 });
10238 }
10239 })
10240 });
10241 }
10242
10243 fn jump(
10244 &mut self,
10245 path: ProjectPath,
10246 position: Point,
10247 anchor: language::Anchor,
10248 offset_from_top: u32,
10249 cx: &mut ViewContext<Self>,
10250 ) {
10251 let workspace = self.workspace();
10252 cx.spawn(|_, mut cx| async move {
10253 let workspace = workspace.ok_or_else(|| anyhow!("cannot jump without workspace"))?;
10254 let editor = workspace.update(&mut cx, |workspace, cx| {
10255 // Reset the preview item id before opening the new item
10256 workspace.active_pane().update(cx, |pane, cx| {
10257 pane.set_preview_item_id(None, cx);
10258 });
10259 workspace.open_path_preview(path, None, true, true, cx)
10260 })?;
10261 let editor = editor
10262 .await?
10263 .downcast::<Editor>()
10264 .ok_or_else(|| anyhow!("opened item was not an editor"))?
10265 .downgrade();
10266 editor.update(&mut cx, |editor, cx| {
10267 let buffer = editor
10268 .buffer()
10269 .read(cx)
10270 .as_singleton()
10271 .ok_or_else(|| anyhow!("cannot jump in a multi-buffer"))?;
10272 let buffer = buffer.read(cx);
10273 let cursor = if buffer.can_resolve(&anchor) {
10274 language::ToPoint::to_point(&anchor, buffer)
10275 } else {
10276 buffer.clip_point(position, Bias::Left)
10277 };
10278
10279 let nav_history = editor.nav_history.take();
10280 editor.change_selections(
10281 Some(Autoscroll::top_relative(offset_from_top as usize)),
10282 cx,
10283 |s| {
10284 s.select_ranges([cursor..cursor]);
10285 },
10286 );
10287 editor.nav_history = nav_history;
10288
10289 anyhow::Ok(())
10290 })??;
10291
10292 anyhow::Ok(())
10293 })
10294 .detach_and_log_err(cx);
10295 }
10296
10297 fn marked_text_ranges(&self, cx: &AppContext) -> Option<Vec<Range<OffsetUtf16>>> {
10298 let snapshot = self.buffer.read(cx).read(cx);
10299 let (_, ranges) = self.text_highlights::<InputComposition>(cx)?;
10300 Some(
10301 ranges
10302 .iter()
10303 .map(move |range| {
10304 range.start.to_offset_utf16(&snapshot)..range.end.to_offset_utf16(&snapshot)
10305 })
10306 .collect(),
10307 )
10308 }
10309
10310 fn selection_replacement_ranges(
10311 &self,
10312 range: Range<OffsetUtf16>,
10313 cx: &AppContext,
10314 ) -> Vec<Range<OffsetUtf16>> {
10315 let selections = self.selections.all::<OffsetUtf16>(cx);
10316 let newest_selection = selections
10317 .iter()
10318 .max_by_key(|selection| selection.id)
10319 .unwrap();
10320 let start_delta = range.start.0 as isize - newest_selection.start.0 as isize;
10321 let end_delta = range.end.0 as isize - newest_selection.end.0 as isize;
10322 let snapshot = self.buffer.read(cx).read(cx);
10323 selections
10324 .into_iter()
10325 .map(|mut selection| {
10326 selection.start.0 =
10327 (selection.start.0 as isize).saturating_add(start_delta) as usize;
10328 selection.end.0 = (selection.end.0 as isize).saturating_add(end_delta) as usize;
10329 snapshot.clip_offset_utf16(selection.start, Bias::Left)
10330 ..snapshot.clip_offset_utf16(selection.end, Bias::Right)
10331 })
10332 .collect()
10333 }
10334
10335 fn report_editor_event(
10336 &self,
10337 operation: &'static str,
10338 file_extension: Option<String>,
10339 cx: &AppContext,
10340 ) {
10341 if cfg!(any(test, feature = "test-support")) {
10342 return;
10343 }
10344
10345 let Some(project) = &self.project else { return };
10346
10347 // If None, we are in a file without an extension
10348 let file = self
10349 .buffer
10350 .read(cx)
10351 .as_singleton()
10352 .and_then(|b| b.read(cx).file());
10353 let file_extension = file_extension.or(file
10354 .as_ref()
10355 .and_then(|file| Path::new(file.file_name(cx)).extension())
10356 .and_then(|e| e.to_str())
10357 .map(|a| a.to_string()));
10358
10359 let vim_mode = cx
10360 .global::<SettingsStore>()
10361 .raw_user_settings()
10362 .get("vim_mode")
10363 == Some(&serde_json::Value::Bool(true));
10364
10365 let copilot_enabled = all_language_settings(file, cx).inline_completions.provider
10366 == language::language_settings::InlineCompletionProvider::Copilot;
10367 let copilot_enabled_for_language = self
10368 .buffer
10369 .read(cx)
10370 .settings_at(0, cx)
10371 .show_inline_completions;
10372
10373 let telemetry = project.read(cx).client().telemetry().clone();
10374 telemetry.report_editor_event(
10375 file_extension,
10376 vim_mode,
10377 operation,
10378 copilot_enabled,
10379 copilot_enabled_for_language,
10380 )
10381 }
10382
10383 /// Copy the highlighted chunks to the clipboard as JSON. The format is an array of lines,
10384 /// with each line being an array of {text, highlight} objects.
10385 fn copy_highlight_json(&mut self, _: &CopyHighlightJson, cx: &mut ViewContext<Self>) {
10386 let Some(buffer) = self.buffer.read(cx).as_singleton() else {
10387 return;
10388 };
10389
10390 #[derive(Serialize)]
10391 struct Chunk<'a> {
10392 text: String,
10393 highlight: Option<&'a str>,
10394 }
10395
10396 let snapshot = buffer.read(cx).snapshot();
10397 let range = self
10398 .selected_text_range(cx)
10399 .and_then(|selected_range| {
10400 if selected_range.is_empty() {
10401 None
10402 } else {
10403 Some(selected_range)
10404 }
10405 })
10406 .unwrap_or_else(|| 0..snapshot.len());
10407
10408 let chunks = snapshot.chunks(range, true);
10409 let mut lines = Vec::new();
10410 let mut line: VecDeque<Chunk> = VecDeque::new();
10411
10412 let Some(style) = self.style.as_ref() else {
10413 return;
10414 };
10415
10416 for chunk in chunks {
10417 let highlight = chunk
10418 .syntax_highlight_id
10419 .and_then(|id| id.name(&style.syntax));
10420 let mut chunk_lines = chunk.text.split('\n').peekable();
10421 while let Some(text) = chunk_lines.next() {
10422 let mut merged_with_last_token = false;
10423 if let Some(last_token) = line.back_mut() {
10424 if last_token.highlight == highlight {
10425 last_token.text.push_str(text);
10426 merged_with_last_token = true;
10427 }
10428 }
10429
10430 if !merged_with_last_token {
10431 line.push_back(Chunk {
10432 text: text.into(),
10433 highlight,
10434 });
10435 }
10436
10437 if chunk_lines.peek().is_some() {
10438 if line.len() > 1 && line.front().unwrap().text.is_empty() {
10439 line.pop_front();
10440 }
10441 if line.len() > 1 && line.back().unwrap().text.is_empty() {
10442 line.pop_back();
10443 }
10444
10445 lines.push(mem::take(&mut line));
10446 }
10447 }
10448 }
10449
10450 let Some(lines) = serde_json::to_string_pretty(&lines).log_err() else {
10451 return;
10452 };
10453 cx.write_to_clipboard(ClipboardItem::new(lines));
10454 }
10455
10456 pub fn inlay_hint_cache(&self) -> &InlayHintCache {
10457 &self.inlay_hint_cache
10458 }
10459
10460 pub fn replay_insert_event(
10461 &mut self,
10462 text: &str,
10463 relative_utf16_range: Option<Range<isize>>,
10464 cx: &mut ViewContext<Self>,
10465 ) {
10466 if !self.input_enabled {
10467 cx.emit(EditorEvent::InputIgnored { text: text.into() });
10468 return;
10469 }
10470 if let Some(relative_utf16_range) = relative_utf16_range {
10471 let selections = self.selections.all::<OffsetUtf16>(cx);
10472 self.change_selections(None, cx, |s| {
10473 let new_ranges = selections.into_iter().map(|range| {
10474 let start = OffsetUtf16(
10475 range
10476 .head()
10477 .0
10478 .saturating_add_signed(relative_utf16_range.start),
10479 );
10480 let end = OffsetUtf16(
10481 range
10482 .head()
10483 .0
10484 .saturating_add_signed(relative_utf16_range.end),
10485 );
10486 start..end
10487 });
10488 s.select_ranges(new_ranges);
10489 });
10490 }
10491
10492 self.handle_input(text, cx);
10493 }
10494
10495 pub fn supports_inlay_hints(&self, cx: &AppContext) -> bool {
10496 let Some(project) = self.project.as_ref() else {
10497 return false;
10498 };
10499 let project = project.read(cx);
10500
10501 let mut supports = false;
10502 self.buffer().read(cx).for_each_buffer(|buffer| {
10503 if !supports {
10504 supports = project
10505 .language_servers_for_buffer(buffer.read(cx), cx)
10506 .any(
10507 |(_, server)| match server.capabilities().inlay_hint_provider {
10508 Some(lsp::OneOf::Left(enabled)) => enabled,
10509 Some(lsp::OneOf::Right(_)) => true,
10510 None => false,
10511 },
10512 )
10513 }
10514 });
10515 supports
10516 }
10517
10518 pub fn focus(&self, cx: &mut WindowContext) {
10519 cx.focus(&self.focus_handle)
10520 }
10521
10522 pub fn is_focused(&self, cx: &WindowContext) -> bool {
10523 self.focus_handle.is_focused(cx)
10524 }
10525
10526 fn handle_focus(&mut self, cx: &mut ViewContext<Self>) {
10527 cx.emit(EditorEvent::Focused);
10528
10529 if let Some(rename) = self.pending_rename.as_ref() {
10530 let rename_editor_focus_handle = rename.editor.read(cx).focus_handle.clone();
10531 cx.focus(&rename_editor_focus_handle);
10532 } else {
10533 if let Some(blame) = self.blame.as_ref() {
10534 blame.update(cx, GitBlame::focus)
10535 }
10536
10537 self.blink_manager.update(cx, BlinkManager::enable);
10538 self.show_cursor_names(cx);
10539 self.buffer.update(cx, |buffer, cx| {
10540 buffer.finalize_last_transaction(cx);
10541 if self.leader_peer_id.is_none() {
10542 buffer.set_active_selections(
10543 &self.selections.disjoint_anchors(),
10544 self.selections.line_mode,
10545 self.cursor_shape,
10546 cx,
10547 );
10548 }
10549 });
10550 }
10551 }
10552
10553 pub fn handle_blur(&mut self, cx: &mut ViewContext<Self>) {
10554 self.blink_manager.update(cx, BlinkManager::disable);
10555 self.buffer
10556 .update(cx, |buffer, cx| buffer.remove_active_selections(cx));
10557
10558 if let Some(blame) = self.blame.as_ref() {
10559 blame.update(cx, GitBlame::blur)
10560 }
10561 self.hide_context_menu(cx);
10562 hide_hover(self, cx);
10563 cx.emit(EditorEvent::Blurred);
10564 cx.notify();
10565 }
10566
10567 pub fn register_action<A: Action>(
10568 &mut self,
10569 listener: impl Fn(&A, &mut WindowContext) + 'static,
10570 ) -> &mut Self {
10571 let listener = Arc::new(listener);
10572
10573 self.editor_actions.push(Box::new(move |cx| {
10574 let _view = cx.view().clone();
10575 let cx = cx.window_context();
10576 let listener = listener.clone();
10577 cx.on_action(TypeId::of::<A>(), move |action, phase, cx| {
10578 let action = action.downcast_ref().unwrap();
10579 if phase == DispatchPhase::Bubble {
10580 listener(action, cx)
10581 }
10582 })
10583 }));
10584 self
10585 }
10586}
10587
10588fn hunks_for_selections(
10589 multi_buffer_snapshot: &MultiBufferSnapshot,
10590 selections: &[Selection<Anchor>],
10591) -> Vec<DiffHunk<u32>> {
10592 let mut hunks = Vec::with_capacity(selections.len());
10593 let mut processed_buffer_rows: HashMap<BufferId, HashSet<Range<text::Anchor>>> =
10594 HashMap::default();
10595 let display_rows_for_selections = selections.iter().map(|selection| {
10596 let head = selection.head();
10597 let tail = selection.tail();
10598 let start = tail.to_point(&multi_buffer_snapshot).row;
10599 let end = head.to_point(&multi_buffer_snapshot).row;
10600 if start > end {
10601 end..start
10602 } else {
10603 start..end
10604 }
10605 });
10606
10607 for selected_multi_buffer_rows in display_rows_for_selections {
10608 let query_rows = selected_multi_buffer_rows.start..selected_multi_buffer_rows.end + 1;
10609 for hunk in multi_buffer_snapshot.git_diff_hunks_in_range(query_rows.clone()) {
10610 // Deleted hunk is an empty row range, no caret can be placed there and Zed allows to revert it
10611 // when the caret is just above or just below the deleted hunk.
10612 let allow_adjacent = hunk.status() == DiffHunkStatus::Removed;
10613 let related_to_selection = if allow_adjacent {
10614 hunk.associated_range.overlaps(&query_rows)
10615 || hunk.associated_range.start == query_rows.end
10616 || hunk.associated_range.end == query_rows.start
10617 } else {
10618 // `selected_multi_buffer_rows` are inclusive (e.g. [2..2] means 2nd row is selected)
10619 // `hunk.associated_range` is exclusive (e.g. [2..3] means 2nd row is selected)
10620 hunk.associated_range.overlaps(&selected_multi_buffer_rows)
10621 || selected_multi_buffer_rows.end == hunk.associated_range.start
10622 };
10623 if related_to_selection {
10624 if !processed_buffer_rows
10625 .entry(hunk.buffer_id)
10626 .or_default()
10627 .insert(hunk.buffer_range.start..hunk.buffer_range.end)
10628 {
10629 continue;
10630 }
10631 hunks.push(hunk);
10632 }
10633 }
10634 }
10635
10636 hunks
10637}
10638
10639pub trait CollaborationHub {
10640 fn collaborators<'a>(&self, cx: &'a AppContext) -> &'a HashMap<PeerId, Collaborator>;
10641 fn user_participant_indices<'a>(
10642 &self,
10643 cx: &'a AppContext,
10644 ) -> &'a HashMap<u64, ParticipantIndex>;
10645 fn user_names(&self, cx: &AppContext) -> HashMap<u64, SharedString>;
10646}
10647
10648impl CollaborationHub for Model<Project> {
10649 fn collaborators<'a>(&self, cx: &'a AppContext) -> &'a HashMap<PeerId, Collaborator> {
10650 self.read(cx).collaborators()
10651 }
10652
10653 fn user_participant_indices<'a>(
10654 &self,
10655 cx: &'a AppContext,
10656 ) -> &'a HashMap<u64, ParticipantIndex> {
10657 self.read(cx).user_store().read(cx).participant_indices()
10658 }
10659
10660 fn user_names(&self, cx: &AppContext) -> HashMap<u64, SharedString> {
10661 let this = self.read(cx);
10662 let user_ids = this.collaborators().values().map(|c| c.user_id);
10663 this.user_store().read_with(cx, |user_store, cx| {
10664 user_store.participant_names(user_ids, cx)
10665 })
10666 }
10667}
10668
10669pub trait CompletionProvider {
10670 fn completions(
10671 &self,
10672 buffer: &Model<Buffer>,
10673 buffer_position: text::Anchor,
10674 cx: &mut ViewContext<Editor>,
10675 ) -> Task<Result<Vec<Completion>>>;
10676
10677 fn resolve_completions(
10678 &self,
10679 buffer: Model<Buffer>,
10680 completion_indices: Vec<usize>,
10681 completions: Arc<RwLock<Box<[Completion]>>>,
10682 cx: &mut ViewContext<Editor>,
10683 ) -> Task<Result<bool>>;
10684
10685 fn apply_additional_edits_for_completion(
10686 &self,
10687 buffer: Model<Buffer>,
10688 completion: Completion,
10689 push_to_history: bool,
10690 cx: &mut ViewContext<Editor>,
10691 ) -> Task<Result<Option<language::Transaction>>>;
10692}
10693
10694impl CompletionProvider for Model<Project> {
10695 fn completions(
10696 &self,
10697 buffer: &Model<Buffer>,
10698 buffer_position: text::Anchor,
10699 cx: &mut ViewContext<Editor>,
10700 ) -> Task<Result<Vec<Completion>>> {
10701 self.update(cx, |project, cx| {
10702 project.completions(&buffer, buffer_position, cx)
10703 })
10704 }
10705
10706 fn resolve_completions(
10707 &self,
10708 buffer: Model<Buffer>,
10709 completion_indices: Vec<usize>,
10710 completions: Arc<RwLock<Box<[Completion]>>>,
10711 cx: &mut ViewContext<Editor>,
10712 ) -> Task<Result<bool>> {
10713 self.update(cx, |project, cx| {
10714 project.resolve_completions(buffer, completion_indices, completions, cx)
10715 })
10716 }
10717
10718 fn apply_additional_edits_for_completion(
10719 &self,
10720 buffer: Model<Buffer>,
10721 completion: Completion,
10722 push_to_history: bool,
10723 cx: &mut ViewContext<Editor>,
10724 ) -> Task<Result<Option<language::Transaction>>> {
10725 self.update(cx, |project, cx| {
10726 project.apply_additional_edits_for_completion(buffer, completion, push_to_history, cx)
10727 })
10728 }
10729}
10730
10731fn inlay_hint_settings(
10732 location: Anchor,
10733 snapshot: &MultiBufferSnapshot,
10734 cx: &mut ViewContext<'_, Editor>,
10735) -> InlayHintSettings {
10736 let file = snapshot.file_at(location);
10737 let language = snapshot.language_at(location);
10738 let settings = all_language_settings(file, cx);
10739 settings
10740 .language(language.map(|l| l.name()).as_deref())
10741 .inlay_hints
10742}
10743
10744fn consume_contiguous_rows(
10745 contiguous_row_selections: &mut Vec<Selection<Point>>,
10746 selection: &Selection<Point>,
10747 display_map: &DisplaySnapshot,
10748 selections: &mut std::iter::Peekable<std::slice::Iter<Selection<Point>>>,
10749) -> (u32, u32) {
10750 contiguous_row_selections.push(selection.clone());
10751 let start_row = selection.start.row;
10752 let mut end_row = ending_row(selection, display_map);
10753
10754 while let Some(next_selection) = selections.peek() {
10755 if next_selection.start.row <= end_row {
10756 end_row = ending_row(next_selection, display_map);
10757 contiguous_row_selections.push(selections.next().unwrap().clone());
10758 } else {
10759 break;
10760 }
10761 }
10762 (start_row, end_row)
10763}
10764
10765fn ending_row(next_selection: &Selection<Point>, display_map: &DisplaySnapshot) -> u32 {
10766 if next_selection.end.column > 0 || next_selection.is_empty() {
10767 display_map.next_line_boundary(next_selection.end).0.row + 1
10768 } else {
10769 next_selection.end.row
10770 }
10771}
10772
10773impl EditorSnapshot {
10774 pub fn remote_selections_in_range<'a>(
10775 &'a self,
10776 range: &'a Range<Anchor>,
10777 collaboration_hub: &dyn CollaborationHub,
10778 cx: &'a AppContext,
10779 ) -> impl 'a + Iterator<Item = RemoteSelection> {
10780 let participant_names = collaboration_hub.user_names(cx);
10781 let participant_indices = collaboration_hub.user_participant_indices(cx);
10782 let collaborators_by_peer_id = collaboration_hub.collaborators(cx);
10783 let collaborators_by_replica_id = collaborators_by_peer_id
10784 .iter()
10785 .map(|(_, collaborator)| (collaborator.replica_id, collaborator))
10786 .collect::<HashMap<_, _>>();
10787 self.buffer_snapshot
10788 .remote_selections_in_range(range)
10789 .filter_map(move |(replica_id, line_mode, cursor_shape, selection)| {
10790 let collaborator = collaborators_by_replica_id.get(&replica_id)?;
10791 let participant_index = participant_indices.get(&collaborator.user_id).copied();
10792 let user_name = participant_names.get(&collaborator.user_id).cloned();
10793 Some(RemoteSelection {
10794 replica_id,
10795 selection,
10796 cursor_shape,
10797 line_mode,
10798 participant_index,
10799 peer_id: collaborator.peer_id,
10800 user_name,
10801 })
10802 })
10803 }
10804
10805 pub fn language_at<T: ToOffset>(&self, position: T) -> Option<&Arc<Language>> {
10806 self.display_snapshot.buffer_snapshot.language_at(position)
10807 }
10808
10809 pub fn is_focused(&self) -> bool {
10810 self.is_focused
10811 }
10812
10813 pub fn placeholder_text(&self) -> Option<&Arc<str>> {
10814 self.placeholder_text.as_ref()
10815 }
10816
10817 pub fn scroll_position(&self) -> gpui::Point<f32> {
10818 self.scroll_anchor.scroll_position(&self.display_snapshot)
10819 }
10820
10821 pub fn gutter_dimensions(
10822 &self,
10823 font_id: FontId,
10824 font_size: Pixels,
10825 em_width: Pixels,
10826 max_line_number_width: Pixels,
10827 cx: &AppContext,
10828 ) -> GutterDimensions {
10829 if !self.show_gutter {
10830 return GutterDimensions::default();
10831 }
10832 let descent = cx.text_system().descent(font_id, font_size);
10833
10834 let show_git_gutter = matches!(
10835 ProjectSettings::get_global(cx).git.git_gutter,
10836 Some(GitGutterSetting::TrackedFiles)
10837 );
10838 let gutter_settings = EditorSettings::get_global(cx).gutter;
10839 let gutter_lines_enabled = gutter_settings.line_numbers;
10840 let line_gutter_width = if gutter_lines_enabled {
10841 // Avoid flicker-like gutter resizes when the line number gains another digit and only resize the gutter on files with N*10^5 lines.
10842 let min_width_for_number_on_gutter = em_width * 4.0;
10843 max_line_number_width.max(min_width_for_number_on_gutter)
10844 } else {
10845 0.0.into()
10846 };
10847
10848 let git_blame_entries_width = self
10849 .render_git_blame_gutter
10850 .then_some(em_width * GIT_BLAME_GUTTER_WIDTH_CHARS);
10851
10852 let mut left_padding = git_blame_entries_width.unwrap_or(Pixels::ZERO);
10853 left_padding += if gutter_settings.code_actions {
10854 em_width * 3.0
10855 } else if show_git_gutter && gutter_lines_enabled {
10856 em_width * 2.0
10857 } else if show_git_gutter || gutter_lines_enabled {
10858 em_width
10859 } else {
10860 px(0.)
10861 };
10862
10863 let right_padding = if gutter_settings.folds && gutter_lines_enabled {
10864 em_width * 4.0
10865 } else if gutter_settings.folds {
10866 em_width * 3.0
10867 } else if gutter_lines_enabled {
10868 em_width
10869 } else {
10870 px(0.)
10871 };
10872
10873 GutterDimensions {
10874 left_padding,
10875 right_padding,
10876 width: line_gutter_width + left_padding + right_padding,
10877 margin: -descent,
10878 git_blame_entries_width,
10879 }
10880 }
10881}
10882
10883impl Deref for EditorSnapshot {
10884 type Target = DisplaySnapshot;
10885
10886 fn deref(&self) -> &Self::Target {
10887 &self.display_snapshot
10888 }
10889}
10890
10891#[derive(Clone, Debug, PartialEq, Eq)]
10892pub enum EditorEvent {
10893 InputIgnored {
10894 text: Arc<str>,
10895 },
10896 InputHandled {
10897 utf16_range_to_replace: Option<Range<isize>>,
10898 text: Arc<str>,
10899 },
10900 ExcerptsAdded {
10901 buffer: Model<Buffer>,
10902 predecessor: ExcerptId,
10903 excerpts: Vec<(ExcerptId, ExcerptRange<language::Anchor>)>,
10904 },
10905 ExcerptsRemoved {
10906 ids: Vec<ExcerptId>,
10907 },
10908 BufferEdited,
10909 Edited,
10910 Reparsed,
10911 Focused,
10912 Blurred,
10913 DirtyChanged,
10914 Saved,
10915 TitleChanged,
10916 DiffBaseChanged,
10917 SelectionsChanged {
10918 local: bool,
10919 },
10920 ScrollPositionChanged {
10921 local: bool,
10922 autoscroll: bool,
10923 },
10924 Closed,
10925 TransactionUndone {
10926 transaction_id: clock::Lamport,
10927 },
10928 TransactionBegun {
10929 transaction_id: clock::Lamport,
10930 },
10931}
10932
10933impl EventEmitter<EditorEvent> for Editor {}
10934
10935impl FocusableView for Editor {
10936 fn focus_handle(&self, _cx: &AppContext) -> FocusHandle {
10937 self.focus_handle.clone()
10938 }
10939}
10940
10941impl Render for Editor {
10942 fn render<'a>(&mut self, cx: &mut ViewContext<'a, Self>) -> impl IntoElement {
10943 let settings = ThemeSettings::get_global(cx);
10944
10945 let text_style = match self.mode {
10946 EditorMode::SingleLine | EditorMode::AutoHeight { .. } => TextStyle {
10947 color: cx.theme().colors().editor_foreground,
10948 font_family: settings.ui_font.family.clone(),
10949 font_features: settings.ui_font.features.clone(),
10950 font_size: rems(0.875).into(),
10951 font_weight: FontWeight::NORMAL,
10952 font_style: FontStyle::Normal,
10953 line_height: relative(settings.buffer_line_height.value()),
10954 background_color: None,
10955 underline: None,
10956 strikethrough: None,
10957 white_space: WhiteSpace::Normal,
10958 },
10959 EditorMode::Full => TextStyle {
10960 color: cx.theme().colors().editor_foreground,
10961 font_family: settings.buffer_font.family.clone(),
10962 font_features: settings.buffer_font.features.clone(),
10963 font_size: settings.buffer_font_size(cx).into(),
10964 font_weight: FontWeight::NORMAL,
10965 font_style: FontStyle::Normal,
10966 line_height: relative(settings.buffer_line_height.value()),
10967 background_color: None,
10968 underline: None,
10969 strikethrough: None,
10970 white_space: WhiteSpace::Normal,
10971 },
10972 };
10973
10974 let background = match self.mode {
10975 EditorMode::SingleLine => cx.theme().system().transparent,
10976 EditorMode::AutoHeight { max_lines: _ } => cx.theme().system().transparent,
10977 EditorMode::Full => cx.theme().colors().editor_background,
10978 };
10979
10980 EditorElement::new(
10981 cx.view(),
10982 EditorStyle {
10983 background,
10984 local_player: cx.theme().players().local(),
10985 text: text_style,
10986 scrollbar_width: px(13.),
10987 syntax: cx.theme().syntax().clone(),
10988 status: cx.theme().status().clone(),
10989 inlay_hints_style: HighlightStyle {
10990 color: Some(cx.theme().status().hint),
10991 ..HighlightStyle::default()
10992 },
10993 suggestions_style: HighlightStyle {
10994 color: Some(cx.theme().status().predictive),
10995 ..HighlightStyle::default()
10996 },
10997 },
10998 )
10999 }
11000}
11001
11002impl ViewInputHandler for Editor {
11003 fn text_for_range(
11004 &mut self,
11005 range_utf16: Range<usize>,
11006 cx: &mut ViewContext<Self>,
11007 ) -> Option<String> {
11008 Some(
11009 self.buffer
11010 .read(cx)
11011 .read(cx)
11012 .text_for_range(OffsetUtf16(range_utf16.start)..OffsetUtf16(range_utf16.end))
11013 .collect(),
11014 )
11015 }
11016
11017 fn selected_text_range(&mut self, cx: &mut ViewContext<Self>) -> Option<Range<usize>> {
11018 // Prevent the IME menu from appearing when holding down an alphabetic key
11019 // while input is disabled.
11020 if !self.input_enabled {
11021 return None;
11022 }
11023
11024 let range = self.selections.newest::<OffsetUtf16>(cx).range();
11025 Some(range.start.0..range.end.0)
11026 }
11027
11028 fn marked_text_range(&self, cx: &mut ViewContext<Self>) -> Option<Range<usize>> {
11029 let snapshot = self.buffer.read(cx).read(cx);
11030 let range = self.text_highlights::<InputComposition>(cx)?.1.get(0)?;
11031 Some(range.start.to_offset_utf16(&snapshot).0..range.end.to_offset_utf16(&snapshot).0)
11032 }
11033
11034 fn unmark_text(&mut self, cx: &mut ViewContext<Self>) {
11035 self.clear_highlights::<InputComposition>(cx);
11036 self.ime_transaction.take();
11037 }
11038
11039 fn replace_text_in_range(
11040 &mut self,
11041 range_utf16: Option<Range<usize>>,
11042 text: &str,
11043 cx: &mut ViewContext<Self>,
11044 ) {
11045 if !self.input_enabled {
11046 cx.emit(EditorEvent::InputIgnored { text: text.into() });
11047 return;
11048 }
11049
11050 self.transact(cx, |this, cx| {
11051 let new_selected_ranges = if let Some(range_utf16) = range_utf16 {
11052 let range_utf16 = OffsetUtf16(range_utf16.start)..OffsetUtf16(range_utf16.end);
11053 Some(this.selection_replacement_ranges(range_utf16, cx))
11054 } else {
11055 this.marked_text_ranges(cx)
11056 };
11057
11058 let range_to_replace = new_selected_ranges.as_ref().and_then(|ranges_to_replace| {
11059 let newest_selection_id = this.selections.newest_anchor().id;
11060 this.selections
11061 .all::<OffsetUtf16>(cx)
11062 .iter()
11063 .zip(ranges_to_replace.iter())
11064 .find_map(|(selection, range)| {
11065 if selection.id == newest_selection_id {
11066 Some(
11067 (range.start.0 as isize - selection.head().0 as isize)
11068 ..(range.end.0 as isize - selection.head().0 as isize),
11069 )
11070 } else {
11071 None
11072 }
11073 })
11074 });
11075
11076 cx.emit(EditorEvent::InputHandled {
11077 utf16_range_to_replace: range_to_replace,
11078 text: text.into(),
11079 });
11080
11081 if let Some(new_selected_ranges) = new_selected_ranges {
11082 this.change_selections(None, cx, |selections| {
11083 selections.select_ranges(new_selected_ranges)
11084 });
11085 this.backspace(&Default::default(), cx);
11086 }
11087
11088 this.handle_input(text, cx);
11089 });
11090
11091 if let Some(transaction) = self.ime_transaction {
11092 self.buffer.update(cx, |buffer, cx| {
11093 buffer.group_until_transaction(transaction, cx);
11094 });
11095 }
11096
11097 self.unmark_text(cx);
11098 }
11099
11100 fn replace_and_mark_text_in_range(
11101 &mut self,
11102 range_utf16: Option<Range<usize>>,
11103 text: &str,
11104 new_selected_range_utf16: Option<Range<usize>>,
11105 cx: &mut ViewContext<Self>,
11106 ) {
11107 if !self.input_enabled {
11108 cx.emit(EditorEvent::InputIgnored { text: text.into() });
11109 return;
11110 }
11111
11112 let transaction = self.transact(cx, |this, cx| {
11113 let ranges_to_replace = if let Some(mut marked_ranges) = this.marked_text_ranges(cx) {
11114 let snapshot = this.buffer.read(cx).read(cx);
11115 if let Some(relative_range_utf16) = range_utf16.as_ref() {
11116 for marked_range in &mut marked_ranges {
11117 marked_range.end.0 = marked_range.start.0 + relative_range_utf16.end;
11118 marked_range.start.0 += relative_range_utf16.start;
11119 marked_range.start =
11120 snapshot.clip_offset_utf16(marked_range.start, Bias::Left);
11121 marked_range.end =
11122 snapshot.clip_offset_utf16(marked_range.end, Bias::Right);
11123 }
11124 }
11125 Some(marked_ranges)
11126 } else if let Some(range_utf16) = range_utf16 {
11127 let range_utf16 = OffsetUtf16(range_utf16.start)..OffsetUtf16(range_utf16.end);
11128 Some(this.selection_replacement_ranges(range_utf16, cx))
11129 } else {
11130 None
11131 };
11132
11133 let range_to_replace = ranges_to_replace.as_ref().and_then(|ranges_to_replace| {
11134 let newest_selection_id = this.selections.newest_anchor().id;
11135 this.selections
11136 .all::<OffsetUtf16>(cx)
11137 .iter()
11138 .zip(ranges_to_replace.iter())
11139 .find_map(|(selection, range)| {
11140 if selection.id == newest_selection_id {
11141 Some(
11142 (range.start.0 as isize - selection.head().0 as isize)
11143 ..(range.end.0 as isize - selection.head().0 as isize),
11144 )
11145 } else {
11146 None
11147 }
11148 })
11149 });
11150
11151 cx.emit(EditorEvent::InputHandled {
11152 utf16_range_to_replace: range_to_replace,
11153 text: text.into(),
11154 });
11155
11156 if let Some(ranges) = ranges_to_replace {
11157 this.change_selections(None, cx, |s| s.select_ranges(ranges));
11158 }
11159
11160 let marked_ranges = {
11161 let snapshot = this.buffer.read(cx).read(cx);
11162 this.selections
11163 .disjoint_anchors()
11164 .iter()
11165 .map(|selection| {
11166 selection.start.bias_left(&snapshot)..selection.end.bias_right(&snapshot)
11167 })
11168 .collect::<Vec<_>>()
11169 };
11170
11171 if text.is_empty() {
11172 this.unmark_text(cx);
11173 } else {
11174 this.highlight_text::<InputComposition>(
11175 marked_ranges.clone(),
11176 HighlightStyle {
11177 underline: Some(UnderlineStyle {
11178 thickness: px(1.),
11179 color: None,
11180 wavy: false,
11181 }),
11182 ..Default::default()
11183 },
11184 cx,
11185 );
11186 }
11187
11188 // Disable auto-closing when composing text (i.e. typing a `"` on a Brazilian keyboard)
11189 let use_autoclose = this.use_autoclose;
11190 this.set_use_autoclose(false);
11191 this.handle_input(text, cx);
11192 this.set_use_autoclose(use_autoclose);
11193
11194 if let Some(new_selected_range) = new_selected_range_utf16 {
11195 let snapshot = this.buffer.read(cx).read(cx);
11196 let new_selected_ranges = marked_ranges
11197 .into_iter()
11198 .map(|marked_range| {
11199 let insertion_start = marked_range.start.to_offset_utf16(&snapshot).0;
11200 let new_start = OffsetUtf16(new_selected_range.start + insertion_start);
11201 let new_end = OffsetUtf16(new_selected_range.end + insertion_start);
11202 snapshot.clip_offset_utf16(new_start, Bias::Left)
11203 ..snapshot.clip_offset_utf16(new_end, Bias::Right)
11204 })
11205 .collect::<Vec<_>>();
11206
11207 drop(snapshot);
11208 this.change_selections(None, cx, |selections| {
11209 selections.select_ranges(new_selected_ranges)
11210 });
11211 }
11212 });
11213
11214 self.ime_transaction = self.ime_transaction.or(transaction);
11215 if let Some(transaction) = self.ime_transaction {
11216 self.buffer.update(cx, |buffer, cx| {
11217 buffer.group_until_transaction(transaction, cx);
11218 });
11219 }
11220
11221 if self.text_highlights::<InputComposition>(cx).is_none() {
11222 self.ime_transaction.take();
11223 }
11224 }
11225
11226 fn bounds_for_range(
11227 &mut self,
11228 range_utf16: Range<usize>,
11229 element_bounds: gpui::Bounds<Pixels>,
11230 cx: &mut ViewContext<Self>,
11231 ) -> Option<gpui::Bounds<Pixels>> {
11232 let text_layout_details = self.text_layout_details(cx);
11233 let style = &text_layout_details.editor_style;
11234 let font_id = cx.text_system().resolve_font(&style.text.font());
11235 let font_size = style.text.font_size.to_pixels(cx.rem_size());
11236 let line_height = style.text.line_height_in_pixels(cx.rem_size());
11237 let em_width = cx
11238 .text_system()
11239 .typographic_bounds(font_id, font_size, 'm')
11240 .unwrap()
11241 .size
11242 .width;
11243
11244 let snapshot = self.snapshot(cx);
11245 let scroll_position = snapshot.scroll_position();
11246 let scroll_left = scroll_position.x * em_width;
11247
11248 let start = OffsetUtf16(range_utf16.start).to_display_point(&snapshot);
11249 let x = snapshot.x_for_display_point(start, &text_layout_details) - scroll_left
11250 + self.gutter_dimensions.width;
11251 let y = line_height * (start.row() as f32 - scroll_position.y);
11252
11253 Some(Bounds {
11254 origin: element_bounds.origin + point(x, y),
11255 size: size(em_width, line_height),
11256 })
11257 }
11258}
11259
11260trait SelectionExt {
11261 fn display_range(&self, map: &DisplaySnapshot) -> Range<DisplayPoint>;
11262 fn spanned_rows(&self, include_end_if_at_line_start: bool, map: &DisplaySnapshot)
11263 -> Range<u32>;
11264}
11265
11266impl<T: ToPoint + ToOffset> SelectionExt for Selection<T> {
11267 fn display_range(&self, map: &DisplaySnapshot) -> Range<DisplayPoint> {
11268 let start = self
11269 .start
11270 .to_point(&map.buffer_snapshot)
11271 .to_display_point(map);
11272 let end = self
11273 .end
11274 .to_point(&map.buffer_snapshot)
11275 .to_display_point(map);
11276 if self.reversed {
11277 end..start
11278 } else {
11279 start..end
11280 }
11281 }
11282
11283 fn spanned_rows(
11284 &self,
11285 include_end_if_at_line_start: bool,
11286 map: &DisplaySnapshot,
11287 ) -> Range<u32> {
11288 let start = self.start.to_point(&map.buffer_snapshot);
11289 let mut end = self.end.to_point(&map.buffer_snapshot);
11290 if !include_end_if_at_line_start && start.row != end.row && end.column == 0 {
11291 end.row -= 1;
11292 }
11293
11294 let buffer_start = map.prev_line_boundary(start).0;
11295 let buffer_end = map.next_line_boundary(end).0;
11296 buffer_start.row..buffer_end.row + 1
11297 }
11298}
11299
11300impl<T: InvalidationRegion> InvalidationStack<T> {
11301 fn invalidate<S>(&mut self, selections: &[Selection<S>], buffer: &MultiBufferSnapshot)
11302 where
11303 S: Clone + ToOffset,
11304 {
11305 while let Some(region) = self.last() {
11306 let all_selections_inside_invalidation_ranges =
11307 if selections.len() == region.ranges().len() {
11308 selections
11309 .iter()
11310 .zip(region.ranges().iter().map(|r| r.to_offset(buffer)))
11311 .all(|(selection, invalidation_range)| {
11312 let head = selection.head().to_offset(buffer);
11313 invalidation_range.start <= head && invalidation_range.end >= head
11314 })
11315 } else {
11316 false
11317 };
11318
11319 if all_selections_inside_invalidation_ranges {
11320 break;
11321 } else {
11322 self.pop();
11323 }
11324 }
11325 }
11326}
11327
11328impl<T> Default for InvalidationStack<T> {
11329 fn default() -> Self {
11330 Self(Default::default())
11331 }
11332}
11333
11334impl<T> Deref for InvalidationStack<T> {
11335 type Target = Vec<T>;
11336
11337 fn deref(&self) -> &Self::Target {
11338 &self.0
11339 }
11340}
11341
11342impl<T> DerefMut for InvalidationStack<T> {
11343 fn deref_mut(&mut self) -> &mut Self::Target {
11344 &mut self.0
11345 }
11346}
11347
11348impl InvalidationRegion for SnippetState {
11349 fn ranges(&self) -> &[Range<Anchor>] {
11350 &self.ranges[self.active_index]
11351 }
11352}
11353
11354pub fn diagnostic_block_renderer(diagnostic: Diagnostic, _is_valid: bool) -> RenderBlock {
11355 let (text_without_backticks, code_ranges) = highlight_diagnostic_message(&diagnostic);
11356
11357 Box::new(move |cx: &mut BlockContext| {
11358 let group_id: SharedString = cx.block_id.to_string().into();
11359
11360 let mut text_style = cx.text_style().clone();
11361 text_style.color = diagnostic_style(diagnostic.severity, true, cx.theme().status());
11362 let theme_settings = ThemeSettings::get_global(cx);
11363 text_style.font_family = theme_settings.buffer_font.family.clone();
11364 text_style.font_style = theme_settings.buffer_font.style;
11365 text_style.font_features = theme_settings.buffer_font.features.clone();
11366 text_style.font_weight = theme_settings.buffer_font.weight;
11367
11368 let multi_line_diagnostic = diagnostic.message.contains('\n');
11369
11370 let buttons = |diagnostic: &Diagnostic, block_id: usize| {
11371 if multi_line_diagnostic {
11372 v_flex()
11373 } else {
11374 h_flex()
11375 }
11376 .children(diagnostic.is_primary.then(|| {
11377 IconButton::new(("close-block", block_id), IconName::XCircle)
11378 .icon_color(Color::Muted)
11379 .size(ButtonSize::Compact)
11380 .style(ButtonStyle::Transparent)
11381 .visible_on_hover(group_id.clone())
11382 .on_click(move |_click, cx| cx.dispatch_action(Box::new(Cancel)))
11383 .tooltip(|cx| Tooltip::for_action("Close Diagnostics", &Cancel, cx))
11384 }))
11385 .child(
11386 IconButton::new(("copy-block", block_id), IconName::Copy)
11387 .icon_color(Color::Muted)
11388 .size(ButtonSize::Compact)
11389 .style(ButtonStyle::Transparent)
11390 .visible_on_hover(group_id.clone())
11391 .on_click({
11392 let message = diagnostic.message.clone();
11393 move |_click, cx| cx.write_to_clipboard(ClipboardItem::new(message.clone()))
11394 })
11395 .tooltip(|cx| Tooltip::text("Copy diagnostic message", cx)),
11396 )
11397 };
11398
11399 let icon_size = buttons(&diagnostic, cx.block_id)
11400 .into_any_element()
11401 .layout_as_root(AvailableSpace::min_size(), cx);
11402
11403 h_flex()
11404 .id(cx.block_id)
11405 .group(group_id.clone())
11406 .relative()
11407 .size_full()
11408 .pl(cx.gutter_dimensions.width)
11409 .w(cx.max_width + cx.gutter_dimensions.width)
11410 .child(
11411 div()
11412 .flex()
11413 .w(cx.anchor_x - cx.gutter_dimensions.width - icon_size.width)
11414 .flex_shrink(),
11415 )
11416 .child(buttons(&diagnostic, cx.block_id))
11417 .child(div().flex().flex_shrink_0().child(
11418 StyledText::new(text_without_backticks.clone()).with_highlights(
11419 &text_style,
11420 code_ranges.iter().map(|range| {
11421 (
11422 range.clone(),
11423 HighlightStyle {
11424 font_weight: Some(FontWeight::BOLD),
11425 ..Default::default()
11426 },
11427 )
11428 }),
11429 ),
11430 ))
11431 .into_any_element()
11432 })
11433}
11434
11435pub fn highlight_diagnostic_message(diagnostic: &Diagnostic) -> (SharedString, Vec<Range<usize>>) {
11436 let mut text_without_backticks = String::new();
11437 let mut code_ranges = Vec::new();
11438
11439 if let Some(source) = &diagnostic.source {
11440 text_without_backticks.push_str(&source);
11441 code_ranges.push(0..source.len());
11442 text_without_backticks.push_str(": ");
11443 }
11444
11445 let mut prev_offset = 0;
11446 let mut in_code_block = false;
11447 for (ix, _) in diagnostic
11448 .message
11449 .match_indices('`')
11450 .chain([(diagnostic.message.len(), "")])
11451 {
11452 let prev_len = text_without_backticks.len();
11453 text_without_backticks.push_str(&diagnostic.message[prev_offset..ix]);
11454 prev_offset = ix + 1;
11455 if in_code_block {
11456 code_ranges.push(prev_len..text_without_backticks.len());
11457 in_code_block = false;
11458 } else {
11459 in_code_block = true;
11460 }
11461 }
11462
11463 (text_without_backticks.into(), code_ranges)
11464}
11465
11466fn diagnostic_style(severity: DiagnosticSeverity, valid: bool, colors: &StatusColors) -> Hsla {
11467 match (severity, valid) {
11468 (DiagnosticSeverity::ERROR, true) => colors.error,
11469 (DiagnosticSeverity::ERROR, false) => colors.error,
11470 (DiagnosticSeverity::WARNING, true) => colors.warning,
11471 (DiagnosticSeverity::WARNING, false) => colors.warning,
11472 (DiagnosticSeverity::INFORMATION, true) => colors.info,
11473 (DiagnosticSeverity::INFORMATION, false) => colors.info,
11474 (DiagnosticSeverity::HINT, true) => colors.info,
11475 (DiagnosticSeverity::HINT, false) => colors.info,
11476 _ => colors.ignored,
11477 }
11478}
11479
11480pub fn styled_runs_for_code_label<'a>(
11481 label: &'a CodeLabel,
11482 syntax_theme: &'a theme::SyntaxTheme,
11483) -> impl 'a + Iterator<Item = (Range<usize>, HighlightStyle)> {
11484 let fade_out = HighlightStyle {
11485 fade_out: Some(0.35),
11486 ..Default::default()
11487 };
11488
11489 let mut prev_end = label.filter_range.end;
11490 label
11491 .runs
11492 .iter()
11493 .enumerate()
11494 .flat_map(move |(ix, (range, highlight_id))| {
11495 let style = if let Some(style) = highlight_id.style(syntax_theme) {
11496 style
11497 } else {
11498 return Default::default();
11499 };
11500 let mut muted_style = style;
11501 muted_style.highlight(fade_out);
11502
11503 let mut runs = SmallVec::<[(Range<usize>, HighlightStyle); 3]>::new();
11504 if range.start >= label.filter_range.end {
11505 if range.start > prev_end {
11506 runs.push((prev_end..range.start, fade_out));
11507 }
11508 runs.push((range.clone(), muted_style));
11509 } else if range.end <= label.filter_range.end {
11510 runs.push((range.clone(), style));
11511 } else {
11512 runs.push((range.start..label.filter_range.end, style));
11513 runs.push((label.filter_range.end..range.end, muted_style));
11514 }
11515 prev_end = cmp::max(prev_end, range.end);
11516
11517 if ix + 1 == label.runs.len() && label.text.len() > prev_end {
11518 runs.push((prev_end..label.text.len(), fade_out));
11519 }
11520
11521 runs
11522 })
11523}
11524
11525pub(crate) fn split_words(text: &str) -> impl std::iter::Iterator<Item = &str> + '_ {
11526 let mut prev_index = 0;
11527 let mut prev_codepoint: Option<char> = None;
11528 text.char_indices()
11529 .chain([(text.len(), '\0')])
11530 .filter_map(move |(index, codepoint)| {
11531 let prev_codepoint = prev_codepoint.replace(codepoint)?;
11532 let is_boundary = index == text.len()
11533 || !prev_codepoint.is_uppercase() && codepoint.is_uppercase()
11534 || !prev_codepoint.is_alphanumeric() && codepoint.is_alphanumeric();
11535 if is_boundary {
11536 let chunk = &text[prev_index..index];
11537 prev_index = index;
11538 Some(chunk)
11539 } else {
11540 None
11541 }
11542 })
11543}
11544
11545trait RangeToAnchorExt {
11546 fn to_anchors(self, snapshot: &MultiBufferSnapshot) -> Range<Anchor>;
11547}
11548
11549impl<T: ToOffset> RangeToAnchorExt for Range<T> {
11550 fn to_anchors(self, snapshot: &MultiBufferSnapshot) -> Range<Anchor> {
11551 snapshot.anchor_after(self.start)..snapshot.anchor_before(self.end)
11552 }
11553}