Pass the pending selection to update_selections and set_selections

This helps us preserve our invariant of always having at least a pending selection in set_selections when comparing old and new cursor positions.

Co-Authored-By: Max Brunsfeld <maxbrunsfeld@gmail.com>
This commit is contained in:
Nathan Sobo 2022-02-11 10:07:42 -07:00
parent ad9a0e2d4f
commit fa194c8e14
2 changed files with 80 additions and 71 deletions

View file

@ -490,7 +490,7 @@ impl ProjectDiagnosticsEditor {
} }
} }
} }
editor.update_selections(selections, None, cx); editor.update_selections(selections, None, None, cx);
Some(()) Some(())
}); });

View file

@ -357,7 +357,7 @@ pub enum SelectPhase {
} }
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
enum SelectMode { pub enum SelectMode {
Character, Character,
Word(Range<Anchor>), Word(Range<Anchor>),
Line(Range<Anchor>), Line(Range<Anchor>),
@ -443,7 +443,7 @@ pub struct EditorSnapshot {
scroll_top_anchor: Option<Anchor>, scroll_top_anchor: Option<Anchor>,
} }
struct PendingSelection { pub struct PendingSelection {
selection: Selection<Anchor>, selection: Selection<Anchor>,
mode: SelectMode, mode: SelectMode,
} }
@ -1242,16 +1242,20 @@ impl Editor {
goal: SelectionGoal::None, goal: SelectionGoal::None,
}; };
if !add { let mut selections = Vec::new();
self.update_selections::<usize>(Vec::new(), None, cx); if add {
} else if click_count > 1 { selections = self.local_selections(cx);
// Remove the newest selection since it was only added as part of this multi-click. // Remove the newest selection since it was only added as part of this multi-click.
let mut selections = self.local_selections(cx); if click_count > 1 {
selections.retain(|selection| selection.id != newest_selection.id); selections.retain(|selection| selection.id != newest_selection.id);
self.update_selections::<usize>(selections, None, cx) }
} }
self.update_selections::<usize>(
self.pending_selection = Some(PendingSelection { selection, mode }); selections,
Some(PendingSelection { selection, mode }),
None,
cx,
);
cx.notify(); cx.notify();
} }
@ -1378,7 +1382,7 @@ impl Editor {
self.columnar_selection_tail.take(); self.columnar_selection_tail.take();
if self.pending_selection.is_some() { if self.pending_selection.is_some() {
let selections = self.local_selections::<usize>(cx); let selections = self.local_selections::<usize>(cx);
self.update_selections(selections, None, cx); self.update_selections(selections, None, None, cx);
} }
} }
@ -1418,7 +1422,7 @@ impl Editor {
}) })
.collect::<Vec<_>>(); .collect::<Vec<_>>();
self.update_selections(selections, None, cx); self.update_selections(selections, None, None, cx);
cx.notify(); cx.notify();
} }
@ -1452,7 +1456,7 @@ impl Editor {
goal: selection.goal, goal: selection.goal,
}; };
if self.local_selections::<Point>(cx).is_empty() { if self.local_selections::<Point>(cx).is_empty() {
self.update_selections(vec![selection], Some(Autoscroll::Fit), cx); self.update_selections(vec![selection], None, Some(Autoscroll::Fit), cx);
} }
} else { } else {
let buffer = self.buffer.read(cx).snapshot(cx); let buffer = self.buffer.read(cx).snapshot(cx);
@ -1466,7 +1470,7 @@ impl Editor {
oldest_selection.start = oldest_selection.head().clone(); oldest_selection.start = oldest_selection.head().clone();
oldest_selection.end = oldest_selection.head().clone(); oldest_selection.end = oldest_selection.head().clone();
} }
self.update_selections(vec![oldest_selection], Some(Autoscroll::Fit), cx); self.update_selections(vec![oldest_selection], None, Some(Autoscroll::Fit), cx);
} }
} }
@ -1539,7 +1543,7 @@ impl Editor {
} }
}) })
.collect::<Vec<_>>(); .collect::<Vec<_>>();
self.update_selections(selections, autoscroll, cx); self.update_selections(selections, None, autoscroll, cx);
} }
#[cfg(any(test, feature = "test-support"))] #[cfg(any(test, feature = "test-support"))]
@ -1568,7 +1572,7 @@ impl Editor {
} }
}) })
.collect(); .collect();
self.update_selections(selections, None, cx); self.update_selections(selections, None, None, cx);
} }
pub fn handle_input(&mut self, action: &Input, cx: &mut ViewContext<Self>) { pub fn handle_input(&mut self, action: &Input, cx: &mut ViewContext<Self>) {
@ -1742,7 +1746,7 @@ impl Editor {
}) })
.collect() .collect()
}; };
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -1835,7 +1839,7 @@ impl Editor {
}); });
if let Some(new_selections) = new_selections { if let Some(new_selections) = new_selections {
self.update_selections(new_selections, None, cx); self.update_selections(new_selections, None, None, cx);
} }
if let Some(bracket_pair_state) = bracket_pair_state { if let Some(bracket_pair_state) = bracket_pair_state {
self.autoclose_stack.push(bracket_pair_state); self.autoclose_stack.push(bracket_pair_state);
@ -1878,7 +1882,7 @@ impl Editor {
}) })
.collect(); .collect();
self.autoclose_stack.pop(); self.autoclose_stack.pop();
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
true true
} else { } else {
false false
@ -2394,7 +2398,7 @@ impl Editor {
self.snippet_stack.pop(); self.snippet_stack.pop();
} }
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
return true; return true;
} }
self.snippet_stack.pop(); self.snippet_stack.pop();
@ -2424,7 +2428,7 @@ impl Editor {
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
self.insert("", cx); self.insert("", cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -2443,7 +2447,7 @@ impl Editor {
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
self.insert(&"", cx); self.insert(&"", cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -2520,7 +2524,7 @@ impl Editor {
} }
}); });
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -2568,6 +2572,7 @@ impl Editor {
self.update_selections( self.update_selections(
self.local_selections::<usize>(cx), self.local_selections::<usize>(cx),
None,
Some(Autoscroll::Fit), Some(Autoscroll::Fit),
cx, cx,
); );
@ -2643,7 +2648,7 @@ impl Editor {
} }
}) })
.collect(); .collect();
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -2796,7 +2801,7 @@ impl Editor {
} }
}); });
self.fold_ranges(refold_ranges, cx); self.fold_ranges(refold_ranges, cx);
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -2899,7 +2904,7 @@ impl Editor {
} }
}); });
self.fold_ranges(refold_ranges, cx); self.fold_ranges(refold_ranges, cx);
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -2929,7 +2934,7 @@ impl Editor {
}); });
} }
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
self.insert("", cx); self.insert("", cx);
self.end_transaction(cx); self.end_transaction(cx);
@ -3016,7 +3021,7 @@ impl Editor {
selection.end = selection.start; selection.end = selection.start;
}); });
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} else { } else {
self.insert(clipboard_text, cx); self.insert(clipboard_text, cx);
} }
@ -3026,7 +3031,7 @@ impl Editor {
pub fn undo(&mut self, _: &Undo, cx: &mut ViewContext<Self>) { pub fn undo(&mut self, _: &Undo, cx: &mut ViewContext<Self>) {
if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.undo(cx)) { if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.undo(cx)) {
if let Some((selections, _)) = self.selection_history.get(&tx_id).cloned() { if let Some((selections, _)) = self.selection_history.get(&tx_id).cloned() {
self.set_selections(selections, cx); self.set_selections(selections, None, cx);
} }
self.request_autoscroll(Autoscroll::Fit, cx); self.request_autoscroll(Autoscroll::Fit, cx);
} }
@ -3035,7 +3040,7 @@ impl Editor {
pub fn redo(&mut self, _: &Redo, cx: &mut ViewContext<Self>) { pub fn redo(&mut self, _: &Redo, cx: &mut ViewContext<Self>) {
if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.redo(cx)) { if let Some(tx_id) = self.buffer.update(cx, |buffer, cx| buffer.redo(cx)) {
if let Some((_, Some(selections))) = self.selection_history.get(&tx_id).cloned() { if let Some((_, Some(selections))) = self.selection_history.get(&tx_id).cloned() {
self.set_selections(selections, cx); self.set_selections(selections, None, cx);
} }
self.request_autoscroll(Autoscroll::Fit, cx); self.request_autoscroll(Autoscroll::Fit, cx);
} }
@ -3060,7 +3065,7 @@ impl Editor {
selection.reversed = false; selection.reversed = false;
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_left(&mut self, _: &SelectLeft, cx: &mut ViewContext<Self>) { pub fn select_left(&mut self, _: &SelectLeft, cx: &mut ViewContext<Self>) {
@ -3074,7 +3079,7 @@ impl Editor {
selection.set_head(cursor); selection.set_head(cursor);
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn move_right(&mut self, _: &MoveRight, cx: &mut ViewContext<Self>) { pub fn move_right(&mut self, _: &MoveRight, cx: &mut ViewContext<Self>) {
@ -3096,7 +3101,7 @@ impl Editor {
selection.reversed = false; selection.reversed = false;
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_right(&mut self, _: &SelectRight, cx: &mut ViewContext<Self>) { pub fn select_right(&mut self, _: &SelectRight, cx: &mut ViewContext<Self>) {
@ -3110,7 +3115,7 @@ impl Editor {
selection.set_head(cursor); selection.set_head(cursor);
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn move_up(&mut self, _: &MoveUp, cx: &mut ViewContext<Self>) { pub fn move_up(&mut self, _: &MoveUp, cx: &mut ViewContext<Self>) {
@ -3141,7 +3146,7 @@ impl Editor {
selection.goal = goal; selection.goal = goal;
selection.reversed = false; selection.reversed = false;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_up(&mut self, _: &SelectUp, cx: &mut ViewContext<Self>) { pub fn select_up(&mut self, _: &SelectUp, cx: &mut ViewContext<Self>) {
@ -3154,7 +3159,7 @@ impl Editor {
selection.set_head(cursor); selection.set_head(cursor);
selection.goal = goal; selection.goal = goal;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn move_down(&mut self, _: &MoveDown, cx: &mut ViewContext<Self>) { pub fn move_down(&mut self, _: &MoveDown, cx: &mut ViewContext<Self>) {
@ -3185,7 +3190,7 @@ impl Editor {
selection.goal = goal; selection.goal = goal;
selection.reversed = false; selection.reversed = false;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_down(&mut self, _: &SelectDown, cx: &mut ViewContext<Self>) { pub fn select_down(&mut self, _: &SelectDown, cx: &mut ViewContext<Self>) {
@ -3198,7 +3203,7 @@ impl Editor {
selection.set_head(cursor); selection.set_head(cursor);
selection.goal = goal; selection.goal = goal;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn move_to_previous_word_boundary( pub fn move_to_previous_word_boundary(
@ -3216,7 +3221,7 @@ impl Editor {
selection.reversed = false; selection.reversed = false;
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_to_previous_word_boundary( pub fn select_to_previous_word_boundary(
@ -3232,7 +3237,7 @@ impl Editor {
selection.set_head(cursor); selection.set_head(cursor);
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn delete_to_previous_word_boundary( pub fn delete_to_previous_word_boundary(
@ -3252,7 +3257,7 @@ impl Editor {
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
self.insert("", cx); self.insert("", cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -3272,7 +3277,7 @@ impl Editor {
selection.reversed = false; selection.reversed = false;
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_to_next_word_boundary( pub fn select_to_next_word_boundary(
@ -3288,7 +3293,7 @@ impl Editor {
selection.set_head(cursor); selection.set_head(cursor);
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn delete_to_next_word_boundary( pub fn delete_to_next_word_boundary(
@ -3308,7 +3313,7 @@ impl Editor {
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
self.insert("", cx); self.insert("", cx);
self.end_transaction(cx); self.end_transaction(cx);
} }
@ -3329,7 +3334,7 @@ impl Editor {
selection.reversed = false; selection.reversed = false;
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_to_beginning_of_line( pub fn select_to_beginning_of_line(
@ -3345,7 +3350,7 @@ impl Editor {
selection.set_head(new_head.to_point(&display_map)); selection.set_head(new_head.to_point(&display_map));
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn delete_to_beginning_of_line( pub fn delete_to_beginning_of_line(
@ -3373,7 +3378,7 @@ impl Editor {
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn select_to_end_of_line( pub fn select_to_end_of_line(
@ -3389,7 +3394,7 @@ impl Editor {
selection.set_head(new_head.to_point(&display_map)); selection.set_head(new_head.to_point(&display_map));
selection.goal = SelectionGoal::None; selection.goal = SelectionGoal::None;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn delete_to_end_of_line(&mut self, _: &DeleteToEndOfLine, cx: &mut ViewContext<Self>) { pub fn delete_to_end_of_line(&mut self, _: &DeleteToEndOfLine, cx: &mut ViewContext<Self>) {
@ -3419,13 +3424,13 @@ impl Editor {
reversed: false, reversed: false,
goal: SelectionGoal::None, goal: SelectionGoal::None,
}; };
self.update_selections(vec![selection], Some(Autoscroll::Fit), cx); self.update_selections(vec![selection], None, Some(Autoscroll::Fit), cx);
} }
pub fn select_to_beginning(&mut self, _: &SelectToBeginning, cx: &mut ViewContext<Self>) { pub fn select_to_beginning(&mut self, _: &SelectToBeginning, cx: &mut ViewContext<Self>) {
let mut selection = self.local_selections::<Point>(cx).last().unwrap().clone(); let mut selection = self.local_selections::<Point>(cx).last().unwrap().clone();
selection.set_head(Point::zero()); selection.set_head(Point::zero());
self.update_selections(vec![selection], Some(Autoscroll::Fit), cx); self.update_selections(vec![selection], None, Some(Autoscroll::Fit), cx);
} }
pub fn move_to_end(&mut self, _: &MoveToEnd, cx: &mut ViewContext<Self>) { pub fn move_to_end(&mut self, _: &MoveToEnd, cx: &mut ViewContext<Self>) {
@ -3442,7 +3447,7 @@ impl Editor {
reversed: false, reversed: false,
goal: SelectionGoal::None, goal: SelectionGoal::None,
}; };
self.update_selections(vec![selection], Some(Autoscroll::Fit), cx); self.update_selections(vec![selection], None, Some(Autoscroll::Fit), cx);
} }
pub fn set_nav_history(&mut self, nav_history: Option<ItemNavHistory>) { pub fn set_nav_history(&mut self, nav_history: Option<ItemNavHistory>) {
@ -3482,7 +3487,7 @@ impl Editor {
pub fn select_to_end(&mut self, _: &SelectToEnd, cx: &mut ViewContext<Self>) { pub fn select_to_end(&mut self, _: &SelectToEnd, cx: &mut ViewContext<Self>) {
let mut selection = self.local_selections::<usize>(cx).first().unwrap().clone(); let mut selection = self.local_selections::<usize>(cx).first().unwrap().clone();
selection.set_head(self.buffer.read(cx).read(cx).len()); selection.set_head(self.buffer.read(cx).read(cx).len());
self.update_selections(vec![selection], Some(Autoscroll::Fit), cx); self.update_selections(vec![selection], None, Some(Autoscroll::Fit), cx);
} }
pub fn select_all(&mut self, _: &SelectAll, cx: &mut ViewContext<Self>) { pub fn select_all(&mut self, _: &SelectAll, cx: &mut ViewContext<Self>) {
@ -3493,7 +3498,7 @@ impl Editor {
reversed: false, reversed: false,
goal: SelectionGoal::None, goal: SelectionGoal::None,
}; };
self.update_selections(vec![selection], None, cx); self.update_selections(vec![selection], None, None, cx);
} }
pub fn select_line(&mut self, _: &SelectLine, cx: &mut ViewContext<Self>) { pub fn select_line(&mut self, _: &SelectLine, cx: &mut ViewContext<Self>) {
@ -3506,7 +3511,7 @@ impl Editor {
selection.end = cmp::min(max_point, Point::new(rows.end, 0)); selection.end = cmp::min(max_point, Point::new(rows.end, 0));
selection.reversed = false; selection.reversed = false;
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn split_selection_into_lines( pub fn split_selection_into_lines(
@ -3541,7 +3546,7 @@ impl Editor {
} }
} }
self.unfold_ranges(to_unfold, cx); self.unfold_ranges(to_unfold, cx);
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn add_selection_above(&mut self, _: &AddSelectionAbove, cx: &mut ViewContext<Self>) { pub fn add_selection_above(&mut self, _: &AddSelectionAbove, cx: &mut ViewContext<Self>) {
@ -3639,7 +3644,7 @@ impl Editor {
state.stack.pop(); state.stack.pop();
} }
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
if state.stack.len() > 1 { if state.stack.len() > 1 {
self.add_selections_state = Some(state); self.add_selections_state = Some(state);
} }
@ -3699,7 +3704,7 @@ impl Editor {
reversed: false, reversed: false,
goal: SelectionGoal::None, goal: SelectionGoal::None,
}); });
self.update_selections(selections, Some(Autoscroll::Newest), cx); self.update_selections(selections, None, Some(Autoscroll::Newest), cx);
} else { } else {
select_next_state.done = true; select_next_state.done = true;
} }
@ -3726,7 +3731,7 @@ impl Editor {
wordwise: true, wordwise: true,
done: false, done: false,
}; };
self.update_selections(selections, Some(Autoscroll::Newest), cx); self.update_selections(selections, None, Some(Autoscroll::Newest), cx);
self.select_next_state = Some(select_state); self.select_next_state = Some(select_state);
} else { } else {
let query = buffer let query = buffer
@ -3834,6 +3839,7 @@ impl Editor {
self.update_selections( self.update_selections(
self.local_selections::<usize>(cx), self.local_selections::<usize>(cx),
None,
Some(Autoscroll::Fit), Some(Autoscroll::Fit),
cx, cx,
); );
@ -3880,7 +3886,7 @@ impl Editor {
if selected_larger_node { if selected_larger_node {
stack.push(old_selections); stack.push(old_selections);
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
} }
self.select_larger_syntax_node_stack = stack; self.select_larger_syntax_node_stack = stack;
} }
@ -3892,7 +3898,7 @@ impl Editor {
) { ) {
let mut stack = mem::take(&mut self.select_larger_syntax_node_stack); let mut stack = mem::take(&mut self.select_larger_syntax_node_stack);
if let Some(selections) = stack.pop() { if let Some(selections) = stack.pop() {
self.update_selections(selections.to_vec(), Some(Autoscroll::Fit), cx); self.update_selections(selections.to_vec(), None, Some(Autoscroll::Fit), cx);
} }
self.select_larger_syntax_node_stack = stack; self.select_larger_syntax_node_stack = stack;
} }
@ -3921,7 +3927,7 @@ impl Editor {
} }
} }
self.update_selections(selections, Some(Autoscroll::Fit), cx); self.update_selections(selections, None, Some(Autoscroll::Fit), cx);
} }
pub fn show_next_diagnostic(&mut self, _: &ShowNextDiagnostic, cx: &mut ViewContext<Self>) { pub fn show_next_diagnostic(&mut self, _: &ShowNextDiagnostic, cx: &mut ViewContext<Self>) {
@ -3967,6 +3973,7 @@ impl Editor {
reversed: false, reversed: false,
goal: SelectionGoal::None, goal: SelectionGoal::None,
}], }],
None,
Some(Autoscroll::Center), Some(Autoscroll::Center),
cx, cx,
); );
@ -4327,6 +4334,7 @@ impl Editor {
pub fn update_selections<T>( pub fn update_selections<T>(
&mut self, &mut self,
mut selections: Vec<Selection<T>>, mut selections: Vec<Selection<T>>,
pending_selection: Option<PendingSelection>,
autoscroll: Option<Autoscroll>, autoscroll: Option<Autoscroll>,
cx: &mut ViewContext<Self>, cx: &mut ViewContext<Self>,
) where ) where
@ -4370,6 +4378,7 @@ impl Editor {
goal: selection.goal, goal: selection.goal,
} }
})), })),
pending_selection,
cx, cx,
); );
} }
@ -4419,14 +4428,20 @@ impl Editor {
}) })
.collect(); .collect();
drop(snapshot); drop(snapshot);
self.update_selections(new_selections, Some(Autoscroll::Fit), cx); self.update_selections(new_selections, None, Some(Autoscroll::Fit), cx);
selections_with_lost_position selections_with_lost_position
} }
fn set_selections(&mut self, selections: Arc<[Selection<Anchor>]>, cx: &mut ViewContext<Self>) { fn set_selections(
&mut self,
selections: Arc<[Selection<Anchor>]>,
pending_selection: Option<PendingSelection>,
cx: &mut ViewContext<Self>,
) {
let old_cursor_position = self.newest_anchor_selection().head(); let old_cursor_position = self.newest_anchor_selection().head();
self.selections = selections; self.selections = selections;
self.pending_selection = pending_selection;
if self.focused { if self.focused {
self.buffer.update(cx, |buffer, cx| { self.buffer.update(cx, |buffer, cx| {
buffer.set_active_selections(&self.selections, cx) buffer.set_active_selections(&self.selections, cx)
@ -4437,19 +4452,13 @@ impl Editor {
.display_map .display_map
.update(cx, |display_map, cx| display_map.snapshot(cx)); .update(cx, |display_map, cx| display_map.snapshot(cx));
let buffer = &display_map.buffer_snapshot; let buffer = &display_map.buffer_snapshot;
self.pending_selection = None;
self.add_selections_state = None; self.add_selections_state = None;
self.select_next_state = None; self.select_next_state = None;
self.select_larger_syntax_node_stack.clear(); self.select_larger_syntax_node_stack.clear();
self.autoclose_stack.invalidate(&self.selections, &buffer); self.autoclose_stack.invalidate(&self.selections, &buffer);
self.snippet_stack.invalidate(&self.selections, &buffer); self.snippet_stack.invalidate(&self.selections, &buffer);
let new_cursor_position = self let new_cursor_position = self.newest_anchor_selection().head();
.selections
.iter()
.max_by_key(|s| s.id)
.map(|s| s.head())
.unwrap();
self.push_to_nav_history( self.push_to_nav_history(
old_cursor_position.clone(), old_cursor_position.clone(),