1//! Movement module contains helper functions for calculating intended position
2//! in editor given a given motion (e.g. it handles converting a "move left" command into coordinates in editor). It is exposed mostly for use by vim crate.
3
4use super::{Bias, DisplayPoint, DisplaySnapshot, SelectionGoal, ToDisplayPoint};
5use crate::{
6 DisplayRow, EditorStyle, ToOffset, ToPoint,
7 scroll::{ScrollAnchor, ScrollOffset},
8};
9use gpui::{Pixels, WindowTextSystem};
10use language::{CharClassifier, Point};
11use multi_buffer::{MultiBufferRow, MultiBufferSnapshot};
12use serde::Deserialize;
13use workspace::searchable::Direction;
14
15use std::{ops::Range, sync::Arc};
16
17/// Defines search strategy for items in `movement` module.
18/// `FindRange::SingeLine` only looks for a match on a single line at a time, whereas
19/// `FindRange::MultiLine` keeps going until the end of a string.
20#[derive(Clone, Copy, Debug, PartialEq, Eq, Deserialize)]
21pub enum FindRange {
22 SingleLine,
23 MultiLine,
24}
25
26/// TextLayoutDetails encompasses everything we need to move vertically
27/// taking into account variable width characters.
28pub struct TextLayoutDetails {
29 pub(crate) text_system: Arc<WindowTextSystem>,
30 pub(crate) editor_style: EditorStyle,
31 pub(crate) rem_size: Pixels,
32 pub scroll_anchor: ScrollAnchor,
33 pub visible_rows: Option<f64>,
34 pub vertical_scroll_margin: ScrollOffset,
35}
36
37/// Returns a column to the left of the current point, wrapping
38/// to the previous line if that point is at the start of line.
39pub fn left(map: &DisplaySnapshot, mut point: DisplayPoint) -> DisplayPoint {
40 if point.column() > 0 {
41 *point.column_mut() -= 1;
42 } else if point.row().0 > 0 {
43 *point.row_mut() -= 1;
44 *point.column_mut() = map.line_len(point.row());
45 }
46 map.clip_point(point, Bias::Left)
47}
48
49/// Returns a column to the left of the current point, doing nothing if
50/// that point is already at the start of line.
51pub fn saturating_left(map: &DisplaySnapshot, mut point: DisplayPoint) -> DisplayPoint {
52 if point.column() > 0 {
53 *point.column_mut() -= 1;
54 } else if point.column() == 0 {
55 // If the current sofr_wrap mode is used, the column corresponding to the display is 0,
56 // which does not necessarily mean that the actual beginning of a paragraph
57 if map.display_point_to_fold_point(point, Bias::Left).column() > 0 {
58 return left(map, point);
59 }
60 }
61 map.clip_point(point, Bias::Left)
62}
63
64/// Returns a column to the right of the current point, wrapping
65/// to the next line if that point is at the end of line.
66pub fn right(map: &DisplaySnapshot, mut point: DisplayPoint) -> DisplayPoint {
67 if point.column() < map.line_len(point.row()) {
68 *point.column_mut() += 1;
69 } else if point.row() < map.max_point().row() {
70 *point.row_mut() += 1;
71 *point.column_mut() = 0;
72 }
73 map.clip_point(point, Bias::Right)
74}
75
76/// Returns a column to the right of the current point, not performing any wrapping
77/// if that point is already at the end of line.
78pub fn saturating_right(map: &DisplaySnapshot, mut point: DisplayPoint) -> DisplayPoint {
79 *point.column_mut() += 1;
80 map.clip_point(point, Bias::Right)
81}
82
83/// Returns a display point for the preceding displayed line (which might be a soft-wrapped line).
84pub fn up(
85 map: &DisplaySnapshot,
86 start: DisplayPoint,
87 goal: SelectionGoal,
88 preserve_column_at_start: bool,
89 text_layout_details: &TextLayoutDetails,
90) -> (DisplayPoint, SelectionGoal) {
91 up_by_rows(
92 map,
93 start,
94 1,
95 goal,
96 preserve_column_at_start,
97 text_layout_details,
98 )
99}
100
101/// Returns a display point for the next displayed line (which might be a soft-wrapped line).
102pub fn down(
103 map: &DisplaySnapshot,
104 start: DisplayPoint,
105 goal: SelectionGoal,
106 preserve_column_at_end: bool,
107 text_layout_details: &TextLayoutDetails,
108) -> (DisplayPoint, SelectionGoal) {
109 down_by_rows(
110 map,
111 start,
112 1,
113 goal,
114 preserve_column_at_end,
115 text_layout_details,
116 )
117}
118
119pub(crate) fn up_by_rows(
120 map: &DisplaySnapshot,
121 start: DisplayPoint,
122 row_count: u32,
123 goal: SelectionGoal,
124 preserve_column_at_start: bool,
125 text_layout_details: &TextLayoutDetails,
126) -> (DisplayPoint, SelectionGoal) {
127 let goal_x = match goal {
128 SelectionGoal::HorizontalPosition(x) => x.into(),
129 SelectionGoal::WrappedHorizontalPosition((_, x)) => x.into(),
130 SelectionGoal::HorizontalRange { end, .. } => end.into(),
131 _ => map.x_for_display_point(start, text_layout_details),
132 };
133
134 let prev_row = DisplayRow(start.row().0.saturating_sub(row_count));
135 let mut point = map.clip_point(
136 DisplayPoint::new(prev_row, map.line_len(prev_row)),
137 Bias::Left,
138 );
139 if point.row() < start.row() {
140 *point.column_mut() = map.display_column_for_x(point.row(), goal_x, text_layout_details)
141 } else if preserve_column_at_start {
142 return (start, goal);
143 } else {
144 point = DisplayPoint::new(DisplayRow(0), 0);
145 }
146
147 let mut clipped_point = map.clip_point(point, Bias::Left);
148 if clipped_point.row() < point.row() {
149 clipped_point = map.clip_point(point, Bias::Right);
150 }
151 (
152 clipped_point,
153 SelectionGoal::HorizontalPosition(goal_x.into()),
154 )
155}
156
157pub(crate) fn down_by_rows(
158 map: &DisplaySnapshot,
159 start: DisplayPoint,
160 row_count: u32,
161 goal: SelectionGoal,
162 preserve_column_at_end: bool,
163 text_layout_details: &TextLayoutDetails,
164) -> (DisplayPoint, SelectionGoal) {
165 let goal_x = match goal {
166 SelectionGoal::HorizontalPosition(x) => x.into(),
167 SelectionGoal::WrappedHorizontalPosition((_, x)) => x.into(),
168 SelectionGoal::HorizontalRange { end, .. } => end.into(),
169 _ => map.x_for_display_point(start, text_layout_details),
170 };
171
172 let new_row = DisplayRow(start.row().0 + row_count);
173 let mut point = map.clip_point(DisplayPoint::new(new_row, 0), Bias::Right);
174 if point.row() > start.row() {
175 *point.column_mut() = map.display_column_for_x(point.row(), goal_x, text_layout_details)
176 } else if preserve_column_at_end {
177 return (start, goal);
178 } else {
179 point = map.max_point();
180 }
181
182 let mut clipped_point = map.clip_point(point, Bias::Right);
183 if clipped_point.row() > point.row() {
184 clipped_point = map.clip_point(point, Bias::Left);
185 }
186 (
187 clipped_point,
188 SelectionGoal::HorizontalPosition(goal_x.into()),
189 )
190}
191
192/// Returns a position of the start of line.
193/// If `stop_at_soft_boundaries` is true, the returned position is that of the
194/// displayed line (e.g. it could actually be in the middle of a text line if that line is soft-wrapped).
195/// Otherwise it's always going to be the start of a logical line.
196pub fn line_beginning(
197 map: &DisplaySnapshot,
198 display_point: DisplayPoint,
199 stop_at_soft_boundaries: bool,
200) -> DisplayPoint {
201 let point = display_point.to_point(map);
202 let soft_line_start = map.clip_point(DisplayPoint::new(display_point.row(), 0), Bias::Right);
203 let line_start = map.prev_line_boundary(point).1;
204
205 if stop_at_soft_boundaries && display_point != soft_line_start {
206 soft_line_start
207 } else {
208 line_start
209 }
210}
211
212/// Returns the last indented position on a given line.
213/// If `stop_at_soft_boundaries` is true, the returned [`DisplayPoint`] is that of a
214/// displayed line (e.g. if there's soft wrap it's gonna be returned),
215/// otherwise it's always going to be a start of a logical line.
216pub fn indented_line_beginning(
217 map: &DisplaySnapshot,
218 display_point: DisplayPoint,
219 stop_at_soft_boundaries: bool,
220 stop_at_indent: bool,
221) -> DisplayPoint {
222 let point = display_point.to_point(map);
223 let soft_line_start = map.clip_point(DisplayPoint::new(display_point.row(), 0), Bias::Right);
224 let indent_start = Point::new(
225 point.row,
226 map.buffer_snapshot()
227 .indent_size_for_line(MultiBufferRow(point.row))
228 .len,
229 )
230 .to_display_point(map);
231 let line_start = map.prev_line_boundary(point).1;
232
233 if stop_at_soft_boundaries && soft_line_start > indent_start && display_point != soft_line_start
234 {
235 soft_line_start
236 } else if stop_at_indent && (display_point > indent_start || display_point == line_start) {
237 indent_start
238 } else {
239 line_start
240 }
241}
242
243/// Returns a position of the end of line.
244///
245/// If `stop_at_soft_boundaries` is true, the returned position is that of the
246/// displayed line (e.g. it could actually be in the middle of a text line if that line is soft-wrapped).
247/// Otherwise it's always going to be the end of a logical line.
248pub fn line_end(
249 map: &DisplaySnapshot,
250 display_point: DisplayPoint,
251 stop_at_soft_boundaries: bool,
252) -> DisplayPoint {
253 let soft_line_end = map.clip_point(
254 DisplayPoint::new(display_point.row(), map.line_len(display_point.row())),
255 Bias::Left,
256 );
257 if stop_at_soft_boundaries && display_point != soft_line_end {
258 soft_line_end
259 } else {
260 map.next_line_boundary(display_point.to_point(map)).1
261 }
262}
263
264/// Returns a position of the previous word boundary, where a word character is defined as either
265/// uppercase letter, lowercase letter, '_' character or language-specific word character (like '-' in CSS).
266pub fn previous_word_start(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
267 let raw_point = point.to_point(map);
268 let classifier = map.buffer_snapshot().char_classifier_at(raw_point);
269
270 let mut is_first_iteration = true;
271 find_preceding_boundary_display_point(map, point, FindRange::MultiLine, |left, right| {
272 // Make alt-left skip punctuation to respect VSCode behaviour. For example: hello.| goes to |hello.
273 if is_first_iteration
274 && classifier.is_punctuation(right)
275 && !classifier.is_punctuation(left)
276 && left != '\n'
277 {
278 is_first_iteration = false;
279 return false;
280 }
281 is_first_iteration = false;
282
283 (classifier.kind(left) != classifier.kind(right) && !classifier.is_whitespace(right))
284 || left == '\n'
285 })
286}
287
288/// Returns a position of the previous word boundary, where a word character is defined as either
289/// uppercase letter, lowercase letter, '_' character, language-specific word character (like '-' in CSS) or newline.
290pub fn previous_word_start_or_newline(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
291 let raw_point = point.to_point(map);
292 let classifier = map.buffer_snapshot().char_classifier_at(raw_point);
293
294 find_preceding_boundary_display_point(map, point, FindRange::MultiLine, |left, right| {
295 (classifier.kind(left) != classifier.kind(right) && !classifier.is_whitespace(right))
296 || left == '\n'
297 || right == '\n'
298 })
299}
300
301/// Text movements are too greedy, making deletions too greedy too.
302/// Makes deletions more ergonomic by potentially reducing the deletion range based on its text contents:
303/// * whitespace sequences with length >= 2 stop the deletion after removal (despite movement jumping over the word behind the whitespaces)
304/// * brackets stop the deletion after removal (despite movement currently not accounting for these and jumping over)
305pub fn adjust_greedy_deletion(
306 map: &DisplaySnapshot,
307 delete_from: DisplayPoint,
308 delete_until: DisplayPoint,
309 ignore_brackets: bool,
310) -> DisplayPoint {
311 if delete_from == delete_until {
312 return delete_until;
313 }
314 let is_backward = delete_from > delete_until;
315 let delete_range = if is_backward {
316 map.display_point_to_point(delete_until, Bias::Left)
317 .to_offset(map.buffer_snapshot())
318 ..map
319 .display_point_to_point(delete_from, Bias::Right)
320 .to_offset(map.buffer_snapshot())
321 } else {
322 map.display_point_to_point(delete_from, Bias::Left)
323 .to_offset(map.buffer_snapshot())
324 ..map
325 .display_point_to_point(delete_until, Bias::Right)
326 .to_offset(map.buffer_snapshot())
327 };
328
329 let trimmed_delete_range = if ignore_brackets {
330 delete_range
331 } else {
332 let brackets_in_delete_range = map
333 .buffer_snapshot()
334 .bracket_ranges(delete_range.clone())
335 .into_iter()
336 .flatten()
337 .flat_map(|(left_bracket, right_bracket)| {
338 [
339 left_bracket.start,
340 left_bracket.end,
341 right_bracket.start,
342 right_bracket.end,
343 ]
344 })
345 .filter(|&bracket| delete_range.start < bracket && bracket < delete_range.end);
346 let closest_bracket = if is_backward {
347 brackets_in_delete_range.max()
348 } else {
349 brackets_in_delete_range.min()
350 };
351
352 if is_backward {
353 closest_bracket.unwrap_or(delete_range.start)..delete_range.end
354 } else {
355 delete_range.start..closest_bracket.unwrap_or(delete_range.end)
356 }
357 };
358
359 let mut whitespace_sequences = Vec::new();
360 let mut current_offset = trimmed_delete_range.start;
361 let mut whitespace_sequence_length = 0;
362 let mut whitespace_sequence_start = 0;
363 for ch in map
364 .buffer_snapshot()
365 .text_for_range(trimmed_delete_range.clone())
366 .flat_map(str::chars)
367 {
368 if ch.is_whitespace() {
369 if whitespace_sequence_length == 0 {
370 whitespace_sequence_start = current_offset;
371 }
372 whitespace_sequence_length += 1;
373 } else {
374 if whitespace_sequence_length >= 2 {
375 whitespace_sequences.push((whitespace_sequence_start, current_offset));
376 }
377 whitespace_sequence_start = 0;
378 whitespace_sequence_length = 0;
379 }
380 current_offset += ch.len_utf8();
381 }
382 if whitespace_sequence_length >= 2 {
383 whitespace_sequences.push((whitespace_sequence_start, current_offset));
384 }
385
386 let closest_whitespace_end = if is_backward {
387 whitespace_sequences.last().map(|&(start, _)| start)
388 } else {
389 whitespace_sequences.first().map(|&(_, end)| end)
390 };
391
392 closest_whitespace_end
393 .unwrap_or_else(|| {
394 if is_backward {
395 trimmed_delete_range.start
396 } else {
397 trimmed_delete_range.end
398 }
399 })
400 .to_display_point(map)
401}
402
403/// Returns a position of the previous subword boundary, where a subword is defined as a run of
404/// word characters of the same "subkind" - where subcharacter kinds are '_' character,
405/// lowerspace characters and uppercase characters.
406pub fn previous_subword_start(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
407 let raw_point = point.to_point(map);
408 let classifier = map.buffer_snapshot().char_classifier_at(raw_point);
409
410 find_preceding_boundary_display_point(map, point, FindRange::MultiLine, |left, right| {
411 is_subword_start(left, right, &classifier) || left == '\n'
412 })
413}
414
415pub fn is_subword_start(left: char, right: char, classifier: &CharClassifier) -> bool {
416 let is_word_start = classifier.kind(left) != classifier.kind(right) && !right.is_whitespace();
417 let is_subword_start = classifier.is_word('-') && left == '-' && right != '-'
418 || left == '_' && right != '_'
419 || left.is_lowercase() && right.is_uppercase();
420 is_word_start || is_subword_start
421}
422
423/// Returns a position of the next word boundary, where a word character is defined as either
424/// uppercase letter, lowercase letter, '_' character or language-specific word character (like '-' in CSS).
425pub fn next_word_end(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
426 let raw_point = point.to_point(map);
427 let classifier = map.buffer_snapshot().char_classifier_at(raw_point);
428 let mut is_first_iteration = true;
429 find_boundary(map, point, FindRange::MultiLine, |left, right| {
430 // Make alt-right skip punctuation to respect VSCode behaviour. For example: |.hello goes to .hello|
431 if is_first_iteration
432 && classifier.is_punctuation(left)
433 && !classifier.is_punctuation(right)
434 && right != '\n'
435 {
436 is_first_iteration = false;
437 return false;
438 }
439 is_first_iteration = false;
440
441 (classifier.kind(left) != classifier.kind(right) && !classifier.is_whitespace(left))
442 || right == '\n'
443 })
444}
445
446/// Returns a position of the next word boundary, where a word character is defined as either
447/// uppercase letter, lowercase letter, '_' character, language-specific word character (like '-' in CSS) or newline.
448pub fn next_word_end_or_newline(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
449 let raw_point = point.to_point(map);
450 let classifier = map.buffer_snapshot().char_classifier_at(raw_point);
451
452 let mut on_starting_row = true;
453 find_boundary(map, point, FindRange::MultiLine, |left, right| {
454 if left == '\n' {
455 on_starting_row = false;
456 }
457 (classifier.kind(left) != classifier.kind(right)
458 && ((on_starting_row && !left.is_whitespace())
459 || (!on_starting_row && !right.is_whitespace())))
460 || right == '\n'
461 })
462}
463
464/// Returns a position of the next subword boundary, where a subword is defined as a run of
465/// word characters of the same "subkind" - where subcharacter kinds are '_' character,
466/// lowerspace characters and uppercase characters.
467pub fn next_subword_end(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
468 let raw_point = point.to_point(map);
469 let classifier = map.buffer_snapshot().char_classifier_at(raw_point);
470
471 find_boundary(map, point, FindRange::MultiLine, |left, right| {
472 is_subword_end(left, right, &classifier) || right == '\n'
473 })
474}
475
476pub fn is_subword_end(left: char, right: char, classifier: &CharClassifier) -> bool {
477 let is_word_end =
478 (classifier.kind(left) != classifier.kind(right)) && !classifier.is_whitespace(left);
479 let is_subword_end = classifier.is_word('-') && left != '-' && right == '-'
480 || left != '_' && right == '_'
481 || left.is_lowercase() && right.is_uppercase();
482 is_word_end || is_subword_end
483}
484
485/// Returns a position of the start of the current paragraph, where a paragraph
486/// is defined as a run of non-blank lines.
487pub fn start_of_paragraph(
488 map: &DisplaySnapshot,
489 display_point: DisplayPoint,
490 mut count: usize,
491) -> DisplayPoint {
492 let point = display_point.to_point(map);
493 if point.row == 0 {
494 return DisplayPoint::zero();
495 }
496
497 let mut found_non_blank_line = false;
498 for row in (0..point.row + 1).rev() {
499 let blank = map.buffer_snapshot().is_line_blank(MultiBufferRow(row));
500 if found_non_blank_line && blank {
501 if count <= 1 {
502 return Point::new(row, 0).to_display_point(map);
503 }
504 count -= 1;
505 found_non_blank_line = false;
506 }
507
508 found_non_blank_line |= !blank;
509 }
510
511 DisplayPoint::zero()
512}
513
514/// Returns a position of the end of the current paragraph, where a paragraph
515/// is defined as a run of non-blank lines.
516pub fn end_of_paragraph(
517 map: &DisplaySnapshot,
518 display_point: DisplayPoint,
519 mut count: usize,
520) -> DisplayPoint {
521 let point = display_point.to_point(map);
522 if point.row == map.buffer_snapshot().max_row().0 {
523 return map.max_point();
524 }
525
526 let mut found_non_blank_line = false;
527 for row in point.row..=map.buffer_snapshot().max_row().0 {
528 let blank = map.buffer_snapshot().is_line_blank(MultiBufferRow(row));
529 if found_non_blank_line && blank {
530 if count <= 1 {
531 return Point::new(row, 0).to_display_point(map);
532 }
533 count -= 1;
534 found_non_blank_line = false;
535 }
536
537 found_non_blank_line |= !blank;
538 }
539
540 map.max_point()
541}
542
543pub fn start_of_excerpt(
544 map: &DisplaySnapshot,
545 display_point: DisplayPoint,
546 direction: Direction,
547) -> DisplayPoint {
548 let point = map.display_point_to_point(display_point, Bias::Left);
549 let Some(excerpt) = map.buffer_snapshot().excerpt_containing(point..point) else {
550 return display_point;
551 };
552 match direction {
553 Direction::Prev => {
554 let mut start = excerpt.start_anchor().to_display_point(map);
555 if start >= display_point && start.row() > DisplayRow(0) {
556 let Some(excerpt) = map.buffer_snapshot().excerpt_before(excerpt.id()) else {
557 return display_point;
558 };
559 start = excerpt.start_anchor().to_display_point(map);
560 }
561 start
562 }
563 Direction::Next => {
564 let mut end = excerpt.end_anchor().to_display_point(map);
565 *end.row_mut() += 1;
566 map.clip_point(end, Bias::Right)
567 }
568 }
569}
570
571pub fn end_of_excerpt(
572 map: &DisplaySnapshot,
573 display_point: DisplayPoint,
574 direction: Direction,
575) -> DisplayPoint {
576 let point = map.display_point_to_point(display_point, Bias::Left);
577 let Some(excerpt) = map.buffer_snapshot().excerpt_containing(point..point) else {
578 return display_point;
579 };
580 match direction {
581 Direction::Prev => {
582 let mut start = excerpt.start_anchor().to_display_point(map);
583 if start.row() > DisplayRow(0) {
584 *start.row_mut() -= 1;
585 }
586 start = map.clip_point(start, Bias::Left);
587 *start.column_mut() = 0;
588 start
589 }
590 Direction::Next => {
591 let mut end = excerpt.end_anchor().to_display_point(map);
592 *end.column_mut() = 0;
593 if end <= display_point {
594 *end.row_mut() += 1;
595 let point_end = map.display_point_to_point(end, Bias::Right);
596 let Some(excerpt) = map
597 .buffer_snapshot()
598 .excerpt_containing(point_end..point_end)
599 else {
600 return display_point;
601 };
602 end = excerpt.end_anchor().to_display_point(map);
603 *end.column_mut() = 0;
604 }
605 end
606 }
607 }
608}
609
610/// Scans for a boundary preceding the given start point `from` until a boundary is found,
611/// indicated by the given predicate returning true.
612/// The predicate is called with the character to the left and right of the candidate boundary location.
613/// If FindRange::SingleLine is specified and no boundary is found before the start of the current line, the start of the current line will be returned.
614pub fn find_preceding_boundary_point(
615 buffer_snapshot: &MultiBufferSnapshot,
616 from: Point,
617 find_range: FindRange,
618 mut is_boundary: impl FnMut(char, char) -> bool,
619) -> Point {
620 let mut prev_ch = None;
621 let mut offset = from.to_offset(buffer_snapshot);
622
623 for ch in buffer_snapshot.reversed_chars_at(offset) {
624 if find_range == FindRange::SingleLine && ch == '\n' {
625 break;
626 }
627 if let Some(prev_ch) = prev_ch
628 && is_boundary(ch, prev_ch)
629 {
630 break;
631 }
632
633 offset -= ch.len_utf8();
634 prev_ch = Some(ch);
635 }
636
637 offset.to_point(buffer_snapshot)
638}
639
640/// Scans for a boundary preceding the given start point `from` until a boundary is found,
641/// indicated by the given predicate returning true.
642/// The predicate is called with the character to the left and right of the candidate boundary location.
643/// If FindRange::SingleLine is specified and no boundary is found before the start of the current line, the start of the current line will be returned.
644pub fn find_preceding_boundary_display_point(
645 map: &DisplaySnapshot,
646 from: DisplayPoint,
647 find_range: FindRange,
648 is_boundary: impl FnMut(char, char) -> bool,
649) -> DisplayPoint {
650 let result = find_preceding_boundary_point(
651 map.buffer_snapshot(),
652 from.to_point(map),
653 find_range,
654 is_boundary,
655 );
656 map.clip_point(result.to_display_point(map), Bias::Left)
657}
658
659/// Scans for a boundary following the given start point until a boundary is found, indicated by the
660/// given predicate returning true. The predicate is called with the character to the left and right
661/// of the candidate boundary location, and will be called with `\n` characters indicating the start
662/// or end of a line. The function supports optionally returning the point just before the boundary
663/// is found via return_point_before_boundary.
664pub fn find_boundary_point(
665 map: &DisplaySnapshot,
666 from: DisplayPoint,
667 find_range: FindRange,
668 mut is_boundary: impl FnMut(char, char) -> bool,
669 return_point_before_boundary: bool,
670) -> DisplayPoint {
671 let mut offset = from.to_offset(map, Bias::Right);
672 let mut prev_offset = offset;
673 let mut prev_ch = None;
674
675 for ch in map.buffer_snapshot().chars_at(offset) {
676 if find_range == FindRange::SingleLine && ch == '\n' {
677 break;
678 }
679 if let Some(prev_ch) = prev_ch
680 && is_boundary(prev_ch, ch)
681 {
682 if return_point_before_boundary {
683 return map.clip_point(prev_offset.to_display_point(map), Bias::Right);
684 } else {
685 break;
686 }
687 }
688 prev_offset = offset;
689 offset += ch.len_utf8();
690 prev_ch = Some(ch);
691 }
692 map.clip_point(offset.to_display_point(map), Bias::Right)
693}
694
695pub fn find_preceding_boundary_trail(
696 map: &DisplaySnapshot,
697 head: DisplayPoint,
698 mut is_boundary: impl FnMut(char, char) -> bool,
699) -> (Option<DisplayPoint>, DisplayPoint) {
700 let mut offset = head.to_offset(map, Bias::Left);
701 let mut trail_offset = None;
702
703 let mut prev_ch = map.buffer_snapshot().chars_at(offset).next();
704 let mut forward = map.buffer_snapshot().reversed_chars_at(offset).peekable();
705
706 // Skip newlines
707 while let Some(&ch) = forward.peek() {
708 if ch == '\n' {
709 prev_ch = forward.next();
710 offset -= ch.len_utf8();
711 trail_offset = Some(offset);
712 } else {
713 break;
714 }
715 }
716
717 // Find the boundary
718 let start_offset = offset;
719 for ch in forward {
720 if let Some(prev_ch) = prev_ch
721 && is_boundary(prev_ch, ch)
722 {
723 if start_offset == offset {
724 trail_offset = Some(offset);
725 } else {
726 break;
727 }
728 }
729 offset -= ch.len_utf8();
730 prev_ch = Some(ch);
731 }
732
733 let trail = trail_offset
734 .map(|trail_offset: usize| map.clip_point(trail_offset.to_display_point(map), Bias::Left));
735
736 (
737 trail,
738 map.clip_point(offset.to_display_point(map), Bias::Left),
739 )
740}
741
742/// Finds the location of a boundary
743pub fn find_boundary_trail(
744 map: &DisplaySnapshot,
745 head: DisplayPoint,
746 mut is_boundary: impl FnMut(char, char) -> bool,
747) -> (Option<DisplayPoint>, DisplayPoint) {
748 let mut offset = head.to_offset(map, Bias::Right);
749 let mut trail_offset = None;
750
751 let mut prev_ch = map.buffer_snapshot().reversed_chars_at(offset).next();
752 let mut forward = map.buffer_snapshot().chars_at(offset).peekable();
753
754 // Skip newlines
755 while let Some(&ch) = forward.peek() {
756 if ch == '\n' {
757 prev_ch = forward.next();
758 offset += ch.len_utf8();
759 trail_offset = Some(offset);
760 } else {
761 break;
762 }
763 }
764
765 // Find the boundary
766 let start_offset = offset;
767 for ch in forward {
768 if let Some(prev_ch) = prev_ch
769 && is_boundary(prev_ch, ch)
770 {
771 if start_offset == offset {
772 trail_offset = Some(offset);
773 } else {
774 break;
775 }
776 }
777 offset += ch.len_utf8();
778 prev_ch = Some(ch);
779 }
780
781 let trail = trail_offset
782 .map(|trail_offset: usize| map.clip_point(trail_offset.to_display_point(map), Bias::Right));
783
784 (
785 trail,
786 map.clip_point(offset.to_display_point(map), Bias::Right),
787 )
788}
789
790pub fn find_boundary(
791 map: &DisplaySnapshot,
792 from: DisplayPoint,
793 find_range: FindRange,
794 is_boundary: impl FnMut(char, char) -> bool,
795) -> DisplayPoint {
796 find_boundary_point(map, from, find_range, is_boundary, false)
797}
798
799pub fn find_boundary_exclusive(
800 map: &DisplaySnapshot,
801 from: DisplayPoint,
802 find_range: FindRange,
803 is_boundary: impl FnMut(char, char) -> bool,
804) -> DisplayPoint {
805 find_boundary_point(map, from, find_range, is_boundary, true)
806}
807
808/// Returns an iterator over the characters following a given offset in the [`DisplaySnapshot`].
809/// The returned value also contains a range of the start/end of a returned character in
810/// the [`DisplaySnapshot`]. The offsets are relative to the start of a buffer.
811pub fn chars_after(
812 map: &DisplaySnapshot,
813 mut offset: usize,
814) -> impl Iterator<Item = (char, Range<usize>)> + '_ {
815 map.buffer_snapshot().chars_at(offset).map(move |ch| {
816 let before = offset;
817 offset += ch.len_utf8();
818 (ch, before..offset)
819 })
820}
821
822/// Returns a reverse iterator over the characters following a given offset in the [`DisplaySnapshot`].
823/// The returned value also contains a range of the start/end of a returned character in
824/// the [`DisplaySnapshot`]. The offsets are relative to the start of a buffer.
825pub fn chars_before(
826 map: &DisplaySnapshot,
827 mut offset: usize,
828) -> impl Iterator<Item = (char, Range<usize>)> + '_ {
829 map.buffer_snapshot()
830 .reversed_chars_at(offset)
831 .map(move |ch| {
832 let after = offset;
833 offset -= ch.len_utf8();
834 (ch, offset..after)
835 })
836}
837
838/// Returns a list of lines (represented as a [`DisplayPoint`] range) contained
839/// within a passed range.
840///
841/// The line ranges are **always* going to be in bounds of a requested range, which means that
842/// the first and the last lines might not necessarily represent the
843/// full range of a logical line (as their `.start`/`.end` values are clipped to those of a passed in range).
844pub fn split_display_range_by_lines(
845 map: &DisplaySnapshot,
846 range: Range<DisplayPoint>,
847) -> Vec<Range<DisplayPoint>> {
848 let mut result = Vec::new();
849
850 let mut start = range.start;
851 // Loop over all the covered rows until the one containing the range end
852 for row in range.start.row().0..range.end.row().0 {
853 let row_end_column = map.line_len(DisplayRow(row));
854 let end = map.clip_point(
855 DisplayPoint::new(DisplayRow(row), row_end_column),
856 Bias::Left,
857 );
858 if start != end {
859 result.push(start..end);
860 }
861 start = map.clip_point(DisplayPoint::new(DisplayRow(row + 1), 0), Bias::Left);
862 }
863
864 // Add the final range from the start of the last end to the original range end.
865 result.push(start..range.end);
866
867 result
868}
869
870#[cfg(test)]
871mod tests {
872 use super::*;
873 use crate::{
874 Buffer, DisplayMap, DisplayRow, ExcerptRange, FoldPlaceholder, MultiBuffer,
875 display_map::Inlay,
876 test::{editor_test_context::EditorTestContext, marked_display_snapshot},
877 };
878 use gpui::{AppContext as _, font, px};
879 use language::Capability;
880 use project::{Project, project_settings::DiagnosticSeverity};
881 use settings::SettingsStore;
882 use util::post_inc;
883
884 #[gpui::test]
885 fn test_previous_word_start(cx: &mut gpui::App) {
886 init_test(cx);
887
888 fn assert(marked_text: &str, cx: &mut gpui::App) {
889 let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
890 let actual = previous_word_start(&snapshot, display_points[1]);
891 let expected = display_points[0];
892 if actual != expected {
893 eprintln!(
894 "previous_word_start mismatch for '{}': actual={:?}, expected={:?}",
895 marked_text, actual, expected
896 );
897 }
898 assert_eq!(actual, expected);
899 }
900
901 assert("\nˇ ˇlorem", cx);
902 assert("ˇ\nˇ lorem", cx);
903 assert(" ˇloremˇ", cx);
904 assert("ˇ ˇlorem", cx);
905 assert(" ˇlorˇem", cx);
906 assert("\nlorem\nˇ ˇipsum", cx);
907 assert("\n\nˇ\nˇ", cx);
908 assert(" ˇlorem ˇipsum", cx);
909 assert("ˇlorem-ˇipsum", cx);
910 assert("loremˇ-#$@ˇipsum", cx);
911 assert("ˇlorem_ˇipsum", cx);
912 assert(" ˇdefγˇ", cx);
913 assert(" ˇbcΔˇ", cx);
914 // Test punctuation skipping behavior
915 assert("ˇhello.ˇ", cx);
916 assert("helloˇ...ˇ", cx);
917 assert("helloˇ.---..ˇtest", cx);
918 assert("test ˇ.--ˇtest", cx);
919 assert("oneˇ,;:!?ˇtwo", cx);
920 }
921
922 #[gpui::test]
923 fn test_previous_subword_start(cx: &mut gpui::App) {
924 init_test(cx);
925
926 fn assert(marked_text: &str, cx: &mut gpui::App) {
927 let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
928 assert_eq!(
929 previous_subword_start(&snapshot, display_points[1]),
930 display_points[0]
931 );
932 }
933
934 // Subword boundaries are respected
935 assert("lorem_ˇipˇsum", cx);
936 assert("lorem_ˇipsumˇ", cx);
937 assert("ˇlorem_ˇipsum", cx);
938 assert("lorem_ˇipsum_ˇdolor", cx);
939 assert("loremˇIpˇsum", cx);
940 assert("loremˇIpsumˇ", cx);
941
942 // Word boundaries are still respected
943 assert("\nˇ ˇlorem", cx);
944 assert(" ˇloremˇ", cx);
945 assert(" ˇlorˇem", cx);
946 assert("\nlorem\nˇ ˇipsum", cx);
947 assert("\n\nˇ\nˇ", cx);
948 assert(" ˇlorem ˇipsum", cx);
949 assert("loremˇ-ˇipsum", cx);
950 assert("loremˇ-#$@ˇipsum", cx);
951 assert(" ˇdefγˇ", cx);
952 assert(" bcˇΔˇ", cx);
953 assert(" ˇbcδˇ", cx);
954 assert(" abˇ——ˇcd", cx);
955 }
956
957 #[gpui::test]
958 fn test_find_preceding_boundary(cx: &mut gpui::App) {
959 init_test(cx);
960
961 fn assert(
962 marked_text: &str,
963 cx: &mut gpui::App,
964 is_boundary: impl FnMut(char, char) -> bool,
965 ) {
966 let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
967 assert_eq!(
968 find_preceding_boundary_display_point(
969 &snapshot,
970 display_points[1],
971 FindRange::MultiLine,
972 is_boundary
973 ),
974 display_points[0]
975 );
976 }
977
978 assert("abcˇdef\ngh\nijˇk", cx, |left, right| {
979 left == 'c' && right == 'd'
980 });
981 assert("abcdef\nˇgh\nijˇk", cx, |left, right| {
982 left == '\n' && right == 'g'
983 });
984 let mut line_count = 0;
985 assert("abcdef\nˇgh\nijˇk", cx, |left, _| {
986 if left == '\n' {
987 line_count += 1;
988 line_count == 2
989 } else {
990 false
991 }
992 });
993 }
994
995 #[gpui::test]
996 fn test_find_preceding_boundary_with_inlays(cx: &mut gpui::App) {
997 init_test(cx);
998
999 let input_text = "abcdefghijklmnopqrstuvwxys";
1000 let font = font("Helvetica");
1001 let font_size = px(14.0);
1002 let buffer = MultiBuffer::build_simple(input_text, cx);
1003 let buffer_snapshot = buffer.read(cx).snapshot(cx);
1004
1005 let display_map = cx.new(|cx| {
1006 DisplayMap::new(
1007 buffer,
1008 font,
1009 font_size,
1010 None,
1011 1,
1012 1,
1013 FoldPlaceholder::test(),
1014 DiagnosticSeverity::Warning,
1015 cx,
1016 )
1017 });
1018
1019 // add all kinds of inlays between two word boundaries: we should be able to cross them all, when looking for another boundary
1020 let mut id = 0;
1021 let inlays = (0..buffer_snapshot.len())
1022 .flat_map(|offset| {
1023 [
1024 Inlay::edit_prediction(
1025 post_inc(&mut id),
1026 buffer_snapshot.anchor_before(offset),
1027 "test",
1028 ),
1029 Inlay::edit_prediction(
1030 post_inc(&mut id),
1031 buffer_snapshot.anchor_after(offset),
1032 "test",
1033 ),
1034 Inlay::mock_hint(
1035 post_inc(&mut id),
1036 buffer_snapshot.anchor_before(offset),
1037 "test",
1038 ),
1039 Inlay::mock_hint(
1040 post_inc(&mut id),
1041 buffer_snapshot.anchor_after(offset),
1042 "test",
1043 ),
1044 ]
1045 })
1046 .collect();
1047 let snapshot = display_map.update(cx, |map, cx| {
1048 map.splice_inlays(&[], inlays, cx);
1049 map.snapshot(cx)
1050 });
1051
1052 assert_eq!(
1053 find_preceding_boundary_display_point(
1054 &snapshot,
1055 buffer_snapshot.len().to_display_point(&snapshot),
1056 FindRange::MultiLine,
1057 |left, _| left == 'e',
1058 ),
1059 snapshot
1060 .buffer_snapshot()
1061 .offset_to_point(5)
1062 .to_display_point(&snapshot),
1063 "Should not stop at inlays when looking for boundaries"
1064 );
1065 }
1066
1067 #[gpui::test]
1068 fn test_next_word_end(cx: &mut gpui::App) {
1069 init_test(cx);
1070
1071 fn assert(marked_text: &str, cx: &mut gpui::App) {
1072 let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
1073 let actual = next_word_end(&snapshot, display_points[0]);
1074 let expected = display_points[1];
1075 if actual != expected {
1076 eprintln!(
1077 "next_word_end mismatch for '{}': actual={:?}, expected={:?}",
1078 marked_text, actual, expected
1079 );
1080 }
1081 assert_eq!(actual, expected);
1082 }
1083
1084 assert("\nˇ loremˇ", cx);
1085 assert(" ˇloremˇ", cx);
1086 assert(" lorˇemˇ", cx);
1087 assert(" loremˇ ˇ\nipsum\n", cx);
1088 assert("\nˇ\nˇ\n\n", cx);
1089 assert("loremˇ ipsumˇ ", cx);
1090 assert("loremˇ-ipsumˇ", cx);
1091 assert("loremˇ#$@-ˇipsum", cx);
1092 assert("loremˇ_ipsumˇ", cx);
1093 assert(" ˇbcΔˇ", cx);
1094 assert(" abˇ——ˇcd", cx);
1095 // Test punctuation skipping behavior
1096 assert("ˇ.helloˇ", cx);
1097 assert("display_pointsˇ[0ˇ]", cx);
1098 assert("ˇ...ˇhello", cx);
1099 assert("helloˇ.---..ˇtest", cx);
1100 assert("testˇ.--ˇ test", cx);
1101 assert("oneˇ,;:!?ˇtwo", cx);
1102 }
1103
1104 #[gpui::test]
1105 fn test_next_subword_end(cx: &mut gpui::App) {
1106 init_test(cx);
1107
1108 fn assert(marked_text: &str, cx: &mut gpui::App) {
1109 let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
1110 assert_eq!(
1111 next_subword_end(&snapshot, display_points[0]),
1112 display_points[1]
1113 );
1114 }
1115
1116 // Subword boundaries are respected
1117 assert("loˇremˇ_ipsum", cx);
1118 assert("ˇloremˇ_ipsum", cx);
1119 assert("loremˇ_ipsumˇ", cx);
1120 assert("loremˇ_ipsumˇ_dolor", cx);
1121 assert("loˇremˇIpsum", cx);
1122 assert("loremˇIpsumˇDolor", cx);
1123
1124 // Word boundaries are still respected
1125 assert("\nˇ loremˇ", cx);
1126 assert(" ˇloremˇ", cx);
1127 assert(" lorˇemˇ", cx);
1128 assert(" loremˇ ˇ\nipsum\n", cx);
1129 assert("\nˇ\nˇ\n\n", cx);
1130 assert("loremˇ ipsumˇ ", cx);
1131 assert("loremˇ-ˇipsum", cx);
1132 assert("loremˇ#$@-ˇipsum", cx);
1133 assert("loremˇ_ipsumˇ", cx);
1134 assert(" ˇbcˇΔ", cx);
1135 assert(" abˇ——ˇcd", cx);
1136 }
1137
1138 #[gpui::test]
1139 fn test_find_boundary(cx: &mut gpui::App) {
1140 init_test(cx);
1141
1142 fn assert(
1143 marked_text: &str,
1144 cx: &mut gpui::App,
1145 is_boundary: impl FnMut(char, char) -> bool,
1146 ) {
1147 let (snapshot, display_points) = marked_display_snapshot(marked_text, cx);
1148 assert_eq!(
1149 find_boundary(
1150 &snapshot,
1151 display_points[0],
1152 FindRange::MultiLine,
1153 is_boundary,
1154 ),
1155 display_points[1]
1156 );
1157 }
1158
1159 assert("abcˇdef\ngh\nijˇk", cx, |left, right| {
1160 left == 'j' && right == 'k'
1161 });
1162 assert("abˇcdef\ngh\nˇijk", cx, |left, right| {
1163 left == '\n' && right == 'i'
1164 });
1165 let mut line_count = 0;
1166 assert("abcˇdef\ngh\nˇijk", cx, |left, _| {
1167 if left == '\n' {
1168 line_count += 1;
1169 line_count == 2
1170 } else {
1171 false
1172 }
1173 });
1174 }
1175
1176 #[gpui::test]
1177 async fn test_move_up_and_down_with_excerpts(cx: &mut gpui::TestAppContext) {
1178 cx.update(|cx| {
1179 init_test(cx);
1180 });
1181
1182 let mut cx = EditorTestContext::new(cx).await;
1183 let editor = cx.editor.clone();
1184 let window = cx.window;
1185 _ = cx.update_window(window, |_, window, cx| {
1186 let text_layout_details = editor.read(cx).text_layout_details(window);
1187
1188 let font = font("Helvetica");
1189
1190 let buffer = cx.new(|cx| Buffer::local("abc\ndefg\nhijkl\nmn", cx));
1191 let multibuffer = cx.new(|cx| {
1192 let mut multibuffer = MultiBuffer::new(Capability::ReadWrite);
1193 multibuffer.push_excerpts(
1194 buffer.clone(),
1195 [
1196 ExcerptRange::new(Point::new(0, 0)..Point::new(1, 4)),
1197 ExcerptRange::new(Point::new(2, 0)..Point::new(3, 2)),
1198 ],
1199 cx,
1200 );
1201 multibuffer
1202 });
1203 let display_map = cx.new(|cx| {
1204 DisplayMap::new(
1205 multibuffer,
1206 font,
1207 px(14.0),
1208 None,
1209 0,
1210 1,
1211 FoldPlaceholder::test(),
1212 DiagnosticSeverity::Warning,
1213 cx,
1214 )
1215 });
1216 let snapshot = display_map.update(cx, |map, cx| map.snapshot(cx));
1217
1218 assert_eq!(snapshot.text(), "abc\ndefg\n\nhijkl\nmn");
1219
1220 let col_2_x = snapshot
1221 .x_for_display_point(DisplayPoint::new(DisplayRow(0), 2), &text_layout_details);
1222
1223 // Can't move up into the first excerpt's header
1224 assert_eq!(
1225 up(
1226 &snapshot,
1227 DisplayPoint::new(DisplayRow(0), 2),
1228 SelectionGoal::HorizontalPosition(f64::from(col_2_x)),
1229 false,
1230 &text_layout_details
1231 ),
1232 (
1233 DisplayPoint::new(DisplayRow(0), 0),
1234 SelectionGoal::HorizontalPosition(f64::from(col_2_x)),
1235 ),
1236 );
1237 assert_eq!(
1238 up(
1239 &snapshot,
1240 DisplayPoint::new(DisplayRow(0), 0),
1241 SelectionGoal::None,
1242 false,
1243 &text_layout_details
1244 ),
1245 (
1246 DisplayPoint::new(DisplayRow(0), 0),
1247 SelectionGoal::HorizontalPosition(0.0),
1248 ),
1249 );
1250
1251 let col_4_x = snapshot
1252 .x_for_display_point(DisplayPoint::new(DisplayRow(1), 4), &text_layout_details);
1253
1254 // Move up and down within first excerpt
1255 assert_eq!(
1256 up(
1257 &snapshot,
1258 DisplayPoint::new(DisplayRow(1), 4),
1259 SelectionGoal::HorizontalPosition(col_4_x.into()),
1260 false,
1261 &text_layout_details
1262 ),
1263 (
1264 DisplayPoint::new(DisplayRow(0), 3),
1265 SelectionGoal::HorizontalPosition(col_4_x.into())
1266 ),
1267 );
1268 assert_eq!(
1269 down(
1270 &snapshot,
1271 DisplayPoint::new(DisplayRow(0), 3),
1272 SelectionGoal::HorizontalPosition(col_4_x.into()),
1273 false,
1274 &text_layout_details
1275 ),
1276 (
1277 DisplayPoint::new(DisplayRow(1), 4),
1278 SelectionGoal::HorizontalPosition(col_4_x.into())
1279 ),
1280 );
1281
1282 let col_5_x = snapshot
1283 .x_for_display_point(DisplayPoint::new(DisplayRow(3), 5), &text_layout_details);
1284
1285 // Move up and down across second excerpt's header
1286 assert_eq!(
1287 up(
1288 &snapshot,
1289 DisplayPoint::new(DisplayRow(3), 5),
1290 SelectionGoal::HorizontalPosition(col_5_x.into()),
1291 false,
1292 &text_layout_details
1293 ),
1294 (
1295 DisplayPoint::new(DisplayRow(1), 4),
1296 SelectionGoal::HorizontalPosition(col_5_x.into())
1297 ),
1298 );
1299 assert_eq!(
1300 down(
1301 &snapshot,
1302 DisplayPoint::new(DisplayRow(1), 4),
1303 SelectionGoal::HorizontalPosition(col_5_x.into()),
1304 false,
1305 &text_layout_details
1306 ),
1307 (
1308 DisplayPoint::new(DisplayRow(3), 5),
1309 SelectionGoal::HorizontalPosition(col_5_x.into())
1310 ),
1311 );
1312
1313 let max_point_x = snapshot
1314 .x_for_display_point(DisplayPoint::new(DisplayRow(4), 2), &text_layout_details);
1315
1316 // Can't move down off the end, and attempting to do so leaves the selection goal unchanged
1317 assert_eq!(
1318 down(
1319 &snapshot,
1320 DisplayPoint::new(DisplayRow(4), 0),
1321 SelectionGoal::HorizontalPosition(0.0),
1322 false,
1323 &text_layout_details
1324 ),
1325 (
1326 DisplayPoint::new(DisplayRow(4), 2),
1327 SelectionGoal::HorizontalPosition(0.0)
1328 ),
1329 );
1330 assert_eq!(
1331 down(
1332 &snapshot,
1333 DisplayPoint::new(DisplayRow(4), 2),
1334 SelectionGoal::HorizontalPosition(max_point_x.into()),
1335 false,
1336 &text_layout_details
1337 ),
1338 (
1339 DisplayPoint::new(DisplayRow(4), 2),
1340 SelectionGoal::HorizontalPosition(max_point_x.into())
1341 ),
1342 );
1343 });
1344 }
1345
1346 fn init_test(cx: &mut gpui::App) {
1347 let settings_store = SettingsStore::test(cx);
1348 cx.set_global(settings_store);
1349 workspace::init_settings(cx);
1350 theme::init(theme::LoadThemes::JustBase, cx);
1351 language::init(cx);
1352 crate::init(cx);
1353 Project::init_settings(cx);
1354 }
1355}