499 lines
16 KiB
Rust
499 lines
16 KiB
Rust
use super::{Bias, DisplayPoint, DisplaySnapshot, SelectionGoal, ToDisplayPoint};
|
||
use crate::{char_kind, CharKind, ToPoint};
|
||
use anyhow::Result;
|
||
use language::Point;
|
||
use std::ops::Range;
|
||
|
||
pub fn left(map: &DisplaySnapshot, mut point: DisplayPoint) -> Result<DisplayPoint> {
|
||
if point.column() > 0 {
|
||
*point.column_mut() -= 1;
|
||
} else if point.row() > 0 {
|
||
*point.row_mut() -= 1;
|
||
*point.column_mut() = map.line_len(point.row());
|
||
}
|
||
Ok(map.clip_point(point, Bias::Left))
|
||
}
|
||
|
||
pub fn right(map: &DisplaySnapshot, mut point: DisplayPoint) -> Result<DisplayPoint> {
|
||
let max_column = map.line_len(point.row());
|
||
if point.column() < max_column {
|
||
*point.column_mut() += 1;
|
||
} else if point.row() < map.max_point().row() {
|
||
*point.row_mut() += 1;
|
||
*point.column_mut() = 0;
|
||
}
|
||
Ok(map.clip_point(point, Bias::Right))
|
||
}
|
||
|
||
pub fn up(
|
||
map: &DisplaySnapshot,
|
||
start: DisplayPoint,
|
||
goal: SelectionGoal,
|
||
) -> Result<(DisplayPoint, SelectionGoal)> {
|
||
let mut goal_column = if let SelectionGoal::Column(column) = goal {
|
||
column
|
||
} else {
|
||
map.column_to_chars(start.row(), start.column())
|
||
};
|
||
|
||
let prev_row = start.row().saturating_sub(1);
|
||
let mut point = map.clip_point(
|
||
DisplayPoint::new(prev_row, map.line_len(prev_row)),
|
||
Bias::Left,
|
||
);
|
||
if point.row() < start.row() {
|
||
*point.column_mut() = map.column_from_chars(point.row(), goal_column);
|
||
} else {
|
||
point = DisplayPoint::new(0, 0);
|
||
goal_column = 0;
|
||
}
|
||
|
||
let clip_bias = if point.column() == map.line_len(point.row()) {
|
||
Bias::Left
|
||
} else {
|
||
Bias::Right
|
||
};
|
||
|
||
Ok((
|
||
map.clip_point(point, clip_bias),
|
||
SelectionGoal::Column(goal_column),
|
||
))
|
||
}
|
||
|
||
pub fn down(
|
||
map: &DisplaySnapshot,
|
||
start: DisplayPoint,
|
||
goal: SelectionGoal,
|
||
) -> Result<(DisplayPoint, SelectionGoal)> {
|
||
let mut goal_column = if let SelectionGoal::Column(column) = goal {
|
||
column
|
||
} else {
|
||
map.column_to_chars(start.row(), start.column())
|
||
};
|
||
|
||
let next_row = start.row() + 1;
|
||
let mut point = map.clip_point(DisplayPoint::new(next_row, 0), Bias::Right);
|
||
if point.row() > start.row() {
|
||
*point.column_mut() = map.column_from_chars(point.row(), goal_column);
|
||
} else {
|
||
point = map.max_point();
|
||
goal_column = map.column_to_chars(point.row(), point.column())
|
||
}
|
||
|
||
let clip_bias = if point.column() == map.line_len(point.row()) {
|
||
Bias::Left
|
||
} else {
|
||
Bias::Right
|
||
};
|
||
|
||
Ok((
|
||
map.clip_point(point, clip_bias),
|
||
SelectionGoal::Column(goal_column),
|
||
))
|
||
}
|
||
|
||
pub fn line_beginning(
|
||
map: &DisplaySnapshot,
|
||
display_point: DisplayPoint,
|
||
stop_at_soft_boundaries: bool,
|
||
) -> DisplayPoint {
|
||
let point = display_point.to_point(map);
|
||
let soft_line_start = map.clip_point(DisplayPoint::new(display_point.row(), 0), Bias::Right);
|
||
let indent_start = Point::new(
|
||
point.row,
|
||
map.buffer_snapshot.indent_column_for_line(point.row),
|
||
)
|
||
.to_display_point(map);
|
||
let line_start = map.prev_line_boundary(point).1;
|
||
|
||
if stop_at_soft_boundaries && soft_line_start > indent_start && display_point != soft_line_start
|
||
{
|
||
soft_line_start
|
||
} else if stop_at_soft_boundaries && display_point != indent_start {
|
||
indent_start
|
||
} else {
|
||
line_start
|
||
}
|
||
}
|
||
|
||
pub fn line_end(
|
||
map: &DisplaySnapshot,
|
||
display_point: DisplayPoint,
|
||
stop_at_soft_boundaries: bool,
|
||
) -> DisplayPoint {
|
||
let soft_line_end = map.clip_point(
|
||
DisplayPoint::new(display_point.row(), map.line_len(display_point.row())),
|
||
Bias::Left,
|
||
);
|
||
if stop_at_soft_boundaries && display_point != soft_line_end {
|
||
soft_line_end
|
||
} else {
|
||
map.next_line_boundary(display_point.to_point(map)).1
|
||
}
|
||
}
|
||
|
||
pub fn previous_word_start(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
|
||
find_boundary_reversed(map, point, |left, right| {
|
||
char_kind(left) != char_kind(right) && !right.is_whitespace()
|
||
})
|
||
}
|
||
|
||
pub fn previous_subword_start(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
|
||
find_boundary_reversed(map, point, |left, right| {
|
||
(char_kind(left) != char_kind(right)
|
||
|| left == '_' && right != '_'
|
||
|| left.is_lowercase() && right.is_uppercase())
|
||
&& !right.is_whitespace()
|
||
})
|
||
}
|
||
|
||
pub fn next_word_end(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
|
||
find_boundary(map, point, |left, right| {
|
||
char_kind(left) != char_kind(right) && !left.is_whitespace()
|
||
})
|
||
}
|
||
|
||
pub fn next_subword_end(map: &DisplaySnapshot, point: DisplayPoint) -> DisplayPoint {
|
||
find_boundary(map, point, |left, right| {
|
||
(char_kind(left) != char_kind(right)
|
||
|| left != '_' && right == '_'
|
||
|| left.is_lowercase() && right.is_uppercase())
|
||
&& !left.is_whitespace()
|
||
})
|
||
}
|
||
|
||
pub fn find_boundary_reversed(
|
||
map: &DisplaySnapshot,
|
||
mut start: DisplayPoint,
|
||
is_boundary: impl Fn(char, char) -> bool,
|
||
) -> DisplayPoint {
|
||
let mut line_start = 0;
|
||
if start.row() > 0 {
|
||
if let Some(indent) = map.soft_wrap_indent(start.row() - 1) {
|
||
line_start = indent;
|
||
}
|
||
}
|
||
|
||
if start.column() == line_start {
|
||
if start.row() == 0 {
|
||
return DisplayPoint::new(0, 0);
|
||
} else {
|
||
let row = start.row() - 1;
|
||
start = map.clip_point(DisplayPoint::new(row, map.line_len(row)), Bias::Left);
|
||
}
|
||
}
|
||
|
||
let mut boundary = DisplayPoint::new(start.row(), 0);
|
||
let mut column = 0;
|
||
let mut prev_ch = None;
|
||
for ch in map.chars_at(DisplayPoint::new(start.row(), 0)) {
|
||
if column >= start.column() {
|
||
break;
|
||
}
|
||
|
||
if let Some(prev_ch) = prev_ch {
|
||
if is_boundary(prev_ch, ch) {
|
||
*boundary.column_mut() = column;
|
||
}
|
||
}
|
||
|
||
prev_ch = Some(ch);
|
||
column += ch.len_utf8() as u32;
|
||
}
|
||
boundary
|
||
}
|
||
|
||
pub fn find_boundary(
|
||
map: &DisplaySnapshot,
|
||
mut start: DisplayPoint,
|
||
is_boundary: impl Fn(char, char) -> bool,
|
||
) -> DisplayPoint {
|
||
let mut prev_ch = None;
|
||
for ch in map.chars_at(start) {
|
||
if let Some(prev_ch) = prev_ch {
|
||
if ch == '\n' {
|
||
break;
|
||
}
|
||
if is_boundary(prev_ch, ch) {
|
||
break;
|
||
}
|
||
}
|
||
|
||
if ch == '\n' {
|
||
*start.row_mut() += 1;
|
||
*start.column_mut() = 0;
|
||
} else {
|
||
*start.column_mut() += ch.len_utf8() as u32;
|
||
}
|
||
prev_ch = Some(ch);
|
||
}
|
||
map.clip_point(start, Bias::Right)
|
||
}
|
||
|
||
pub fn is_inside_word(map: &DisplaySnapshot, point: DisplayPoint) -> bool {
|
||
let ix = map.clip_point(point, Bias::Left).to_offset(map, Bias::Left);
|
||
let text = &map.buffer_snapshot;
|
||
let next_char_kind = text.chars_at(ix).next().map(char_kind);
|
||
let prev_char_kind = text.reversed_chars_at(ix).next().map(char_kind);
|
||
prev_char_kind.zip(next_char_kind) == Some((CharKind::Word, CharKind::Word))
|
||
}
|
||
|
||
pub fn surrounding_word(map: &DisplaySnapshot, position: DisplayPoint) -> Range<DisplayPoint> {
|
||
let position = map
|
||
.clip_point(position, Bias::Left)
|
||
.to_offset(map, Bias::Left);
|
||
let (range, _) = map.buffer_snapshot.surrounding_word(position);
|
||
let start = range
|
||
.start
|
||
.to_point(&map.buffer_snapshot)
|
||
.to_display_point(map);
|
||
let end = range
|
||
.end
|
||
.to_point(&map.buffer_snapshot)
|
||
.to_display_point(map);
|
||
start..end
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
use crate::{Buffer, DisplayMap, MultiBuffer};
|
||
use language::Point;
|
||
|
||
#[gpui::test]
|
||
fn test_previous_word_start(cx: &mut gpui::MutableAppContext) {
|
||
fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
|
||
let (snapshot, display_points) = marked_snapshot(marked_text, cx);
|
||
assert_eq!(
|
||
previous_word_start(&snapshot, display_points[1]),
|
||
display_points[0]
|
||
);
|
||
}
|
||
|
||
assert("\n| |lorem", cx);
|
||
assert(" |lorem|", cx);
|
||
assert(" |lor|em", cx);
|
||
assert("\nlorem\n| |ipsum", cx);
|
||
assert("\n\n|\n|", cx);
|
||
assert(" |lorem |ipsum", cx);
|
||
assert("lorem|-|ipsum", cx);
|
||
assert("lorem|-#$@|ipsum", cx);
|
||
assert("|lorem_|ipsum", cx);
|
||
assert(" |defγ|", cx);
|
||
assert(" |bcΔ|", cx);
|
||
assert(" ab|——|cd", cx);
|
||
}
|
||
|
||
#[gpui::test]
|
||
fn test_previous_subword_start(cx: &mut gpui::MutableAppContext) {
|
||
fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
|
||
let (snapshot, display_points) = marked_snapshot(marked_text, cx);
|
||
assert_eq!(
|
||
previous_subword_start(&snapshot, display_points[1]),
|
||
display_points[0]
|
||
);
|
||
}
|
||
|
||
// Subword boundaries are respected
|
||
assert("lorem_|ip|sum", cx);
|
||
assert("lorem_|ipsum|", cx);
|
||
assert("|lorem_|ipsum", cx);
|
||
assert("lorem_|ipsum_|dolor", cx);
|
||
assert("lorem|Ip|sum", cx);
|
||
assert("lorem|Ipsum|", cx);
|
||
|
||
// Word boundaries are still respected
|
||
assert("\n| |lorem", cx);
|
||
assert(" |lorem|", cx);
|
||
assert(" |lor|em", cx);
|
||
assert("\nlorem\n| |ipsum", cx);
|
||
assert("\n\n|\n|", cx);
|
||
assert(" |lorem |ipsum", cx);
|
||
assert("lorem|-|ipsum", cx);
|
||
assert("lorem|-#$@|ipsum", cx);
|
||
assert(" |defγ|", cx);
|
||
assert(" bc|Δ|", cx);
|
||
assert(" |bcδ|", cx);
|
||
assert(" ab|——|cd", cx);
|
||
}
|
||
|
||
#[gpui::test]
|
||
fn test_next_word_end(cx: &mut gpui::MutableAppContext) {
|
||
fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
|
||
let (snapshot, display_points) = marked_snapshot(marked_text, cx);
|
||
assert_eq!(
|
||
next_word_end(&snapshot, display_points[0]),
|
||
display_points[1]
|
||
);
|
||
}
|
||
|
||
assert("\n| lorem|", cx);
|
||
assert(" |lorem|", cx);
|
||
assert(" lor|em|", cx);
|
||
assert(" lorem| |\nipsum\n", cx);
|
||
assert("\n|\n|\n\n", cx);
|
||
assert("lorem| ipsum| ", cx);
|
||
assert("lorem|-|ipsum", cx);
|
||
assert("lorem|#$@-|ipsum", cx);
|
||
assert("lorem|_ipsum|", cx);
|
||
assert(" |bcΔ|", cx);
|
||
assert(" ab|——|cd", cx);
|
||
}
|
||
|
||
#[gpui::test]
|
||
fn test_next_subword_end(cx: &mut gpui::MutableAppContext) {
|
||
fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
|
||
let (snapshot, display_points) = marked_snapshot(marked_text, cx);
|
||
assert_eq!(
|
||
next_subword_end(&snapshot, display_points[0]),
|
||
display_points[1]
|
||
);
|
||
}
|
||
|
||
// Subword boundaries are respected
|
||
assert("lo|rem|_ipsum", cx);
|
||
assert("|lorem|_ipsum", cx);
|
||
assert("lorem|_ipsum|", cx);
|
||
assert("lorem|_ipsum|_dolor", cx);
|
||
assert("lo|rem|Ipsum", cx);
|
||
assert("lorem|Ipsum|Dolor", cx);
|
||
|
||
// Word boundaries are still respected
|
||
assert("\n| lorem|", cx);
|
||
assert(" |lorem|", cx);
|
||
assert(" lor|em|", cx);
|
||
assert(" lorem| |\nipsum\n", cx);
|
||
assert("\n|\n|\n\n", cx);
|
||
assert("lorem| ipsum| ", cx);
|
||
assert("lorem|-|ipsum", cx);
|
||
assert("lorem|#$@-|ipsum", cx);
|
||
assert("lorem|_ipsum|", cx);
|
||
assert(" |bc|Δ", cx);
|
||
assert(" ab|——|cd", cx);
|
||
}
|
||
|
||
#[gpui::test]
|
||
fn test_surrounding_word(cx: &mut gpui::MutableAppContext) {
|
||
fn assert(marked_text: &str, cx: &mut gpui::MutableAppContext) {
|
||
let (snapshot, display_points) = marked_snapshot(marked_text, cx);
|
||
assert_eq!(
|
||
surrounding_word(&snapshot, display_points[1]),
|
||
display_points[0]..display_points[2]
|
||
);
|
||
}
|
||
|
||
assert("||lorem| ipsum", cx);
|
||
assert("|lo|rem| ipsum", cx);
|
||
assert("|lorem|| ipsum", cx);
|
||
assert("lorem| | |ipsum", cx);
|
||
assert("lorem\n|||\nipsum", cx);
|
||
assert("lorem\n||ipsum|", cx);
|
||
assert("lorem,|| |ipsum", cx);
|
||
assert("|lorem||, ipsum", cx);
|
||
}
|
||
|
||
#[gpui::test]
|
||
fn test_move_up_and_down_with_excerpts(cx: &mut gpui::MutableAppContext) {
|
||
let family_id = cx.font_cache().load_family(&["Helvetica"]).unwrap();
|
||
let font_id = cx
|
||
.font_cache()
|
||
.select_font(family_id, &Default::default())
|
||
.unwrap();
|
||
|
||
let buffer = cx.add_model(|cx| Buffer::new(0, "abc\ndefg\nhijkl\nmn", cx));
|
||
let multibuffer = cx.add_model(|cx| {
|
||
let mut multibuffer = MultiBuffer::new(0);
|
||
multibuffer.push_excerpts(
|
||
buffer.clone(),
|
||
[
|
||
Point::new(0, 0)..Point::new(1, 4),
|
||
Point::new(2, 0)..Point::new(3, 2),
|
||
],
|
||
cx,
|
||
);
|
||
multibuffer
|
||
});
|
||
|
||
let display_map =
|
||
cx.add_model(|cx| DisplayMap::new(multibuffer, 2, font_id, 14.0, None, 2, 2, cx));
|
||
|
||
let snapshot = display_map.update(cx, |map, cx| map.snapshot(cx));
|
||
assert_eq!(snapshot.text(), "\n\nabc\ndefg\n\n\nhijkl\nmn");
|
||
|
||
// Can't move up into the first excerpt's header
|
||
assert_eq!(
|
||
up(&snapshot, DisplayPoint::new(2, 2), SelectionGoal::Column(2)).unwrap(),
|
||
(DisplayPoint::new(2, 0), SelectionGoal::Column(0)),
|
||
);
|
||
assert_eq!(
|
||
up(&snapshot, DisplayPoint::new(2, 0), SelectionGoal::None).unwrap(),
|
||
(DisplayPoint::new(2, 0), SelectionGoal::Column(0)),
|
||
);
|
||
|
||
// Move up and down within first excerpt
|
||
assert_eq!(
|
||
up(&snapshot, DisplayPoint::new(3, 4), SelectionGoal::Column(4)).unwrap(),
|
||
(DisplayPoint::new(2, 3), SelectionGoal::Column(4)),
|
||
);
|
||
assert_eq!(
|
||
down(&snapshot, DisplayPoint::new(2, 3), SelectionGoal::Column(4)).unwrap(),
|
||
(DisplayPoint::new(3, 4), SelectionGoal::Column(4)),
|
||
);
|
||
|
||
// Move up and down across second excerpt's header
|
||
assert_eq!(
|
||
up(&snapshot, DisplayPoint::new(6, 5), SelectionGoal::Column(5)).unwrap(),
|
||
(DisplayPoint::new(3, 4), SelectionGoal::Column(5)),
|
||
);
|
||
assert_eq!(
|
||
down(&snapshot, DisplayPoint::new(3, 4), SelectionGoal::Column(5)).unwrap(),
|
||
(DisplayPoint::new(6, 5), SelectionGoal::Column(5)),
|
||
);
|
||
|
||
// Can't move down off the end
|
||
assert_eq!(
|
||
down(&snapshot, DisplayPoint::new(7, 0), SelectionGoal::Column(0)).unwrap(),
|
||
(DisplayPoint::new(7, 2), SelectionGoal::Column(2)),
|
||
);
|
||
assert_eq!(
|
||
down(&snapshot, DisplayPoint::new(7, 2), SelectionGoal::Column(2)).unwrap(),
|
||
(DisplayPoint::new(7, 2), SelectionGoal::Column(2)),
|
||
);
|
||
}
|
||
|
||
// Returns a snapshot from text containing '|' character markers with the markers removed, and DisplayPoints for each one.
|
||
fn marked_snapshot(
|
||
text: &str,
|
||
cx: &mut gpui::MutableAppContext,
|
||
) -> (DisplaySnapshot, Vec<DisplayPoint>) {
|
||
let mut marked_offsets = Vec::new();
|
||
let chunks = text.split('|');
|
||
let mut text = String::new();
|
||
|
||
for chunk in chunks {
|
||
text.push_str(chunk);
|
||
marked_offsets.push(text.len());
|
||
}
|
||
marked_offsets.pop();
|
||
|
||
let tab_size = 4;
|
||
let family_id = cx.font_cache().load_family(&["Helvetica"]).unwrap();
|
||
let font_id = cx
|
||
.font_cache()
|
||
.select_font(family_id, &Default::default())
|
||
.unwrap();
|
||
let font_size = 14.0;
|
||
|
||
let buffer = MultiBuffer::build_simple(&text, cx);
|
||
let display_map = cx
|
||
.add_model(|cx| DisplayMap::new(buffer, tab_size, font_id, font_size, None, 1, 1, cx));
|
||
let snapshot = display_map.update(cx, |map, cx| map.snapshot(cx));
|
||
let marked_display_points = marked_offsets
|
||
.into_iter()
|
||
.map(|offset| offset.to_display_point(&snapshot))
|
||
.collect();
|
||
|
||
(snapshot, marked_display_points)
|
||
}
|
||
}
|