1029 lines · 31.8 KB · Rust
RawDownload
use std::ops::Range; use editor::{    display_map::{DisplaySnapshot, ToDisplayPoint},    movement::{self, FindRange},    Bias, DisplayPoint,};use gpui::{actions, impl_actions, ViewContext, WindowContext};use language::{char_kind, CharKind, Selection};use serde::Deserialize;use workspace::Workspace; use crate::{    motion::right, normal::normal_object, state::Mode, utils::coerce_punctuation,    visual::visual_object, Vim,}; #[derive(Copy, Clone, Debug, PartialEq)]pub enum Object {    Word { ignore_punctuation: bool },    Sentence,    Quotes,    BackQuotes,    DoubleQuotes,    VerticalBars,    Parentheses,    SquareBrackets,    CurlyBrackets,    AngleBrackets,} #[derive(Clone, Deserialize, PartialEq)]#[serde(rename_all = "camelCase")]struct Word {    #[serde(default)]    ignore_punctuation: bool,} impl_actions!(vim, [Word]); actions!(    vim,    [        Sentence,        Quotes,        BackQuotes,        DoubleQuotes,        VerticalBars,        Parentheses,        SquareBrackets,        CurlyBrackets,        AngleBrackets    ]); pub fn register(workspace: &mut Workspace, _: &mut ViewContext<Workspace>) {    workspace.register_action(        |_: &mut Workspace, &Word { ignore_punctuation }: &Word, cx: _| {            object(Object::Word { ignore_punctuation }, cx)        },    );    workspace        .register_action(|_: &mut Workspace, _: &Sentence, cx: _| object(Object::Sentence, cx));    workspace.register_action(|_: &mut Workspace, _: &Quotes, cx: _| object(Object::Quotes, cx));    workspace        .register_action(|_: &mut Workspace, _: &BackQuotes, cx: _| object(Object::BackQuotes, cx));    workspace.register_action(|_: &mut Workspace, _: &DoubleQuotes, cx: _| {        object(Object::DoubleQuotes, cx)    });    workspace.register_action(|_: &mut Workspace, _: &Parentheses, cx: _| {        object(Object::Parentheses, cx)    });    workspace.register_action(|_: &mut Workspace, _: &SquareBrackets, cx: _| {        object(Object::SquareBrackets, cx)    });    workspace.register_action(|_: &mut Workspace, _: &CurlyBrackets, cx: _| {        object(Object::CurlyBrackets, cx)    });    workspace.register_action(|_: &mut Workspace, _: &AngleBrackets, cx: _| {        object(Object::AngleBrackets, cx)    });    workspace.register_action(|_: &mut Workspace, _: &VerticalBars, cx: _| {        object(Object::VerticalBars, cx)    });} fn object(object: Object, cx: &mut WindowContext) {    match Vim::read(cx).state().mode {        Mode::Normal => normal_object(object, cx),        Mode::Visual | Mode::VisualLine | Mode::VisualBlock => visual_object(object, cx),        Mode::Insert => {            // Shouldn't execute a text object in insert mode. Ignoring        }    }} impl Object {    pub fn is_multiline(self) -> bool {        match self {            Object::Word { .. }            | Object::Quotes            | Object::BackQuotes            | Object::VerticalBars            | Object::DoubleQuotes => false,            Object::Sentence            | Object::Parentheses            | Object::AngleBrackets            | Object::CurlyBrackets            | Object::SquareBrackets => true,        }    }     pub fn always_expands_both_ways(self) -> bool {        match self {            Object::Word { .. } | Object::Sentence => false,            Object::Quotes            | Object::BackQuotes            | Object::DoubleQuotes            | Object::VerticalBars            | Object::Parentheses            | Object::SquareBrackets            | Object::CurlyBrackets            | Object::AngleBrackets => true,        }    }     pub fn target_visual_mode(self, current_mode: Mode) -> Mode {        match self {            Object::Word { .. } if current_mode == Mode::VisualLine => Mode::Visual,            Object::Word { .. } => current_mode,            Object::Sentence            | Object::Quotes            | Object::BackQuotes            | Object::DoubleQuotes            | Object::VerticalBars            | Object::Parentheses            | Object::SquareBrackets            | Object::CurlyBrackets            | Object::AngleBrackets => Mode::Visual,        }    }     pub fn range(        self,        map: &DisplaySnapshot,        relative_to: DisplayPoint,        around: bool,    ) -> Option<Range<DisplayPoint>> {        match self {            Object::Word { ignore_punctuation } => {                if around {                    around_word(map, relative_to, ignore_punctuation)                } else {                    in_word(map, relative_to, ignore_punctuation)                }            }            Object::Sentence => sentence(map, relative_to, around),            Object::Quotes => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '\'', '\'')            }            Object::BackQuotes => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '`', '`')            }            Object::DoubleQuotes => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '"', '"')            }            Object::VerticalBars => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '|', '|')            }            Object::Parentheses => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '(', ')')            }            Object::SquareBrackets => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '[', ']')            }            Object::CurlyBrackets => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '{', '}')            }            Object::AngleBrackets => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '<', '>')            }        }    }     pub fn expand_selection(        self,        map: &DisplaySnapshot,        selection: &mut Selection<DisplayPoint>,        around: bool,    ) -> bool {        if let Some(range) = self.range(map, selection.head(), around) {            selection.start = range.start;            selection.end = range.end;            true        } else {            false        }    }} /// Returns a range that surrounds the word `relative_to` is in.////// If `relative_to` is at the start of a word, return the word./// If `relative_to` is between words, return the space between.fn in_word(    map: &DisplaySnapshot,    relative_to: DisplayPoint,    ignore_punctuation: bool,) -> Option<Range<DisplayPoint>> {    // Use motion::right so that we consider the character under the cursor when looking for the start    let scope = map        .buffer_snapshot        .language_scope_at(relative_to.to_point(map));    let start = movement::find_preceding_boundary_display_point(        map,        right(map, relative_to, 1),        movement::FindRange::SingleLine,        |left, right| {            coerce_punctuation(char_kind(&scope, left), ignore_punctuation)                != coerce_punctuation(char_kind(&scope, right), ignore_punctuation)        },    );     let end = movement::find_boundary(map, relative_to, FindRange::SingleLine, |left, right| {        coerce_punctuation(char_kind(&scope, left), ignore_punctuation)            != coerce_punctuation(char_kind(&scope, right), ignore_punctuation)    });     Some(start..end)} /// Returns a range that surrounds the word and following whitespace/// relative_to is in.////// If `relative_to` is at the start of a word, return the word and following whitespace./// If `relative_to` is between words, return the whitespace back and the following word.////// if in word///   delete that word///   if there is whitespace following the word, delete that as well///   otherwise, delete any preceding whitespace/// otherwise///   delete whitespace around cursor///   delete word following the cursorfn around_word(    map: &DisplaySnapshot,    relative_to: DisplayPoint,    ignore_punctuation: bool,) -> Option<Range<DisplayPoint>> {    let scope = map        .buffer_snapshot        .language_scope_at(relative_to.to_point(map));    let in_word = map        .chars_at(relative_to)        .next()        .map(|(c, _)| char_kind(&scope, c) != CharKind::Whitespace)        .unwrap_or(false);     if in_word {        around_containing_word(map, relative_to, ignore_punctuation)    } else {        around_next_word(map, relative_to, ignore_punctuation)    }} fn around_containing_word(    map: &DisplaySnapshot,    relative_to: DisplayPoint,    ignore_punctuation: bool,) -> Option<Range<DisplayPoint>> {    in_word(map, relative_to, ignore_punctuation)        .map(|range| expand_to_include_whitespace(map, range, true))} fn around_next_word(    map: &DisplaySnapshot,    relative_to: DisplayPoint,    ignore_punctuation: bool,) -> Option<Range<DisplayPoint>> {    let scope = map        .buffer_snapshot        .language_scope_at(relative_to.to_point(map));    // Get the start of the word    let start = movement::find_preceding_boundary_display_point(        map,        right(map, relative_to, 1),        FindRange::SingleLine,        |left, right| {            coerce_punctuation(char_kind(&scope, left), ignore_punctuation)                != coerce_punctuation(char_kind(&scope, right), ignore_punctuation)        },    );     let mut word_found = false;    let end = movement::find_boundary(map, relative_to, FindRange::MultiLine, |left, right| {        let left_kind = coerce_punctuation(char_kind(&scope, left), ignore_punctuation);        let right_kind = coerce_punctuation(char_kind(&scope, right), ignore_punctuation);         let found = (word_found && left_kind != right_kind) || right == '\n' && left == '\n';         if right_kind != CharKind::Whitespace {            word_found = true;        }         found    });     Some(start..end)} fn sentence(    map: &DisplaySnapshot,    relative_to: DisplayPoint,    around: bool,) -> Option<Range<DisplayPoint>> {    let mut start = None;    let mut previous_end = relative_to;     let mut chars = map.chars_at(relative_to).peekable();     // Search backwards for the previous sentence end or current sentence start. Include the character under relative_to    for (char, point) in chars        .peek()        .cloned()        .into_iter()        .chain(map.reverse_chars_at(relative_to))    {        if is_sentence_end(map, point) {            break;        }         if is_possible_sentence_start(char) {            start = Some(point);        }         previous_end = point;    }     // Search forward for the end of the current sentence or if we are between sentences, the start of the next one    let mut end = relative_to;    for (char, point) in chars {        if start.is_none() && is_possible_sentence_start(char) {            if around {                start = Some(point);                continue;            } else {                end = point;                break;            }        }         end = point;        *end.column_mut() += char.len_utf8() as u32;        end = map.clip_point(end, Bias::Left);         if is_sentence_end(map, end) {            break;        }    }     let mut range = start.unwrap_or(previous_end)..end;    if around {        range = expand_to_include_whitespace(map, range, false);    }     Some(range)} fn is_possible_sentence_start(character: char) -> bool {    !character.is_whitespace() && character != '.'} const SENTENCE_END_PUNCTUATION: &[char] = &['.', '!', '?'];const SENTENCE_END_FILLERS: &[char] = &[')', ']', '"', '\''];const SENTENCE_END_WHITESPACE: &[char] = &[' ', '\t', '\n'];fn is_sentence_end(map: &DisplaySnapshot, point: DisplayPoint) -> bool {    let mut next_chars = map.chars_at(point).peekable();    if let Some((char, _)) = next_chars.next() {        // We are at a double newline. This position is a sentence end.        if char == '\n' && next_chars.peek().map(|(c, _)| c == &'\n').unwrap_or(false) {            return true;        }         // The next text is not a valid whitespace. This is not a sentence end        if !SENTENCE_END_WHITESPACE.contains(&char) {            return false;        }    }     for (char, _) in map.reverse_chars_at(point) {        if SENTENCE_END_PUNCTUATION.contains(&char) {            return true;        }         if !SENTENCE_END_FILLERS.contains(&char) {            return false;        }    }     return false;} /// Expands the passed range to include whitespace on one side or the other in a line. Attempts to add the/// whitespace to the end first and falls back to the start if there was none.fn expand_to_include_whitespace(    map: &DisplaySnapshot,    mut range: Range<DisplayPoint>,    stop_at_newline: bool,) -> Range<DisplayPoint> {    let mut whitespace_included = false;     let mut chars = map.chars_at(range.end).peekable();    while let Some((char, point)) = chars.next() {        if char == '\n' && stop_at_newline {            break;        }         if char.is_whitespace() {            // Set end to the next display_point or the character position after the current display_point            range.end = chars.peek().map(|(_, point)| *point).unwrap_or_else(|| {                let mut end = point;                *end.column_mut() += char.len_utf8() as u32;                map.clip_point(end, Bias::Left)            });             if char != '\n' {                whitespace_included = true;            }        } else {            // Found non whitespace. Quit out.            break;        }    }     if !whitespace_included {        for (char, point) in map.reverse_chars_at(range.start) {            if char == '\n' && stop_at_newline {                break;            }             if !char.is_whitespace() {                break;            }             range.start = point;        }    }     range} fn surrounding_markers(    map: &DisplaySnapshot,    relative_to: DisplayPoint,    around: bool,    search_across_lines: bool,    open_marker: char,    close_marker: char,) -> Option<Range<DisplayPoint>> {    let point = relative_to.to_offset(map, Bias::Left);     let mut matched_closes = 0;    let mut opening = None;     if let Some((ch, range)) = movement::chars_after(map, point).next() {        if ch == open_marker {            if open_marker == close_marker {                let mut total = 0;                for (ch, _) in movement::chars_before(map, point) {                    if ch == '\n' {                        break;                    }                    if ch == open_marker {                        total += 1;                    }                }                if total % 2 == 0 {                    opening = Some(range)                }            } else {                opening = Some(range)            }        }    }     if opening.is_none() {        for (ch, range) in movement::chars_before(map, point) {            if ch == '\n' && !search_across_lines {                break;            }             if ch == open_marker {                if matched_closes == 0 {                    opening = Some(range);                    break;                }                matched_closes -= 1;            } else if ch == close_marker {                matched_closes += 1            }

Lines 1–500 of 1029.