use std::ops::Range; use crate::{    motion::right,    state::{Mode, Operator},    Vim,};use editor::{    display_map::{DisplaySnapshot, ToDisplayPoint},    movement::{self, FindRange},    Bias, DisplayPoint, Editor, ToOffset,};use gpui::{actions, impl_actions, Window};use itertools::Itertools;use language::{BufferSnapshot, CharKind, Point, Selection, TextObject, TreeSitterOptions};use multi_buffer::MultiBufferRow;use schemars::JsonSchema;use serde::Deserialize;use ui::Context; #[derive(Copy, Clone, Debug, PartialEq, Eq, Deserialize, JsonSchema)]#[serde(rename_all = "snake_case")]pub enum Object {    Word { ignore_punctuation: bool },    Subword { ignore_punctuation: bool },    Sentence,    Paragraph,    Quotes,    BackQuotes,    AnyQuotes,    DoubleQuotes,    VerticalBars,    AnyBrackets,    Parentheses,    SquareBrackets,    CurlyBrackets,    AngleBrackets,    Argument,    IndentObj { include_below: bool },    Tag,    Method,    Class,    Comment,    EntireFile,} #[derive(Clone, Deserialize, JsonSchema, PartialEq)]#[serde(deny_unknown_fields)]struct Word {    #[serde(default)]    ignore_punctuation: bool,} #[derive(Clone, Deserialize, JsonSchema, PartialEq)]#[serde(deny_unknown_fields)]struct Subword {    #[serde(default)]    ignore_punctuation: bool,}#[derive(Clone, Deserialize, JsonSchema, PartialEq)]#[serde(deny_unknown_fields)]struct IndentObj {    #[serde(default)]    include_below: bool,} #[derive(Debug, Clone)]pub struct CandidateRange {    pub start: DisplayPoint,    pub end: DisplayPoint,} #[derive(Debug, Clone)]pub struct CandidateWithRanges {    candidate: CandidateRange,    open_range: Range<usize>,    close_range: Range<usize>,} fn cover_or_next<I: Iterator<Item = (Range<usize>, Range<usize>)>>(    candidates: Option<I>,    caret: DisplayPoint,    map: &DisplaySnapshot,    range_filter: Option<&dyn Fn(Range<usize>, Range<usize>) -> bool>,) -> Option<CandidateWithRanges> {    let caret_offset = caret.to_offset(map, Bias::Left);    let mut covering = vec![];    let mut next_ones = vec![];    let snapshot = &map.buffer_snapshot;     if let Some(ranges) = candidates {        for (open_range, close_range) in ranges {            let start_off = open_range.start;            let end_off = close_range.end;            if let Some(range_filter) = range_filter {                if !range_filter(open_range.clone(), close_range.clone()) {                    continue;                }            }            let candidate = CandidateWithRanges {                candidate: CandidateRange {                    start: start_off.to_display_point(map),                    end: end_off.to_display_point(map),                },                open_range: open_range.clone(),                close_range: close_range.clone(),            };             if open_range                .start                .to_offset(snapshot)                .to_display_point(map)                .row()                == caret_offset.to_display_point(map).row()            {                if start_off <= caret_offset && caret_offset < end_off {                    covering.push(candidate);                } else if start_off >= caret_offset {                    next_ones.push(candidate);                }            }        }    }     // 1) covering -> smallest width    if !covering.is_empty() {        return covering.into_iter().min_by_key(|r| {            r.candidate.end.to_offset(map, Bias::Right)                - r.candidate.start.to_offset(map, Bias::Left)        });    }     // 2) next -> closest by start    if !next_ones.is_empty() {        return next_ones.into_iter().min_by_key(|r| {            let start = r.candidate.start.to_offset(map, Bias::Left);            (start as isize - caret_offset as isize).abs()        });    }     None} type DelimiterPredicate = dyn Fn(&BufferSnapshot, usize, usize) -> bool; struct DelimiterRange {    open: Range<usize>,    close: Range<usize>,} impl DelimiterRange {    fn to_display_range(&self, map: &DisplaySnapshot, around: bool) -> Range<DisplayPoint> {        if around {            self.open.start.to_display_point(map)..self.close.end.to_display_point(map)        } else {            self.open.end.to_display_point(map)..self.close.start.to_display_point(map)        }    }} fn find_any_delimiters(    map: &DisplaySnapshot,    display_point: DisplayPoint,    around: bool,    is_valid_delimiter: &DelimiterPredicate,) -> Option<Range<DisplayPoint>> {    let point = map.clip_at_line_end(display_point).to_point(map);    let offset = point.to_offset(&map.buffer_snapshot);     let line_range = get_line_range(map, point);    let visible_line_range = get_visible_line_range(&line_range);     let snapshot = &map.buffer_snapshot;    let excerpt = snapshot.excerpt_containing(offset..offset)?;    let buffer = excerpt.buffer();     let bracket_filter = |open: Range<usize>, close: Range<usize>| {        is_valid_delimiter(buffer, open.start, close.start)    };     // Try to find delimiters in visible range first    let ranges = map        .buffer_snapshot        .bracket_ranges(visible_line_range.clone());    if let Some(candidate) = cover_or_next(ranges, display_point, map, Some(&bracket_filter)) {        return Some(            DelimiterRange {                open: candidate.open_range,                close: candidate.close_range,            }            .to_display_range(map, around),        );    }     // Fall back to innermost enclosing brackets    let (open_bracket, close_bracket) =        buffer.innermost_enclosing_bracket_ranges(offset..offset, Some(&bracket_filter))?;     Some(        DelimiterRange {            open: open_bracket,            close: close_bracket,        }        .to_display_range(map, around),    )} fn get_line_range(map: &DisplaySnapshot, point: Point) -> Range<Point> {    let (start, mut end) = (        map.prev_line_boundary(point).0,        map.next_line_boundary(point).0,    );     if end == point {        end = map.max_point().to_point(map);    }     start..end} fn get_visible_line_range(line_range: &Range<Point>) -> Range<Point> {    let end_column = line_range.end.column.saturating_sub(1);    line_range.start..Point::new(line_range.end.row, end_column)} fn is_quote_delimiter(buffer: &BufferSnapshot, _start: usize, end: usize) -> bool {    matches!(buffer.chars_at(end).next(), Some('\'' | '"' | '`'))} fn is_bracket_delimiter(buffer: &BufferSnapshot, start: usize, _end: usize) -> bool {    matches!(        buffer.chars_at(start).next(),        Some('(' | '[' | '{' | '<' | '|')    )} fn find_any_quotes(    map: &DisplaySnapshot,    display_point: DisplayPoint,    around: bool,) -> Option<Range<DisplayPoint>> {    find_any_delimiters(map, display_point, around, &is_quote_delimiter)} fn find_any_brackets(    map: &DisplaySnapshot,    display_point: DisplayPoint,    around: bool,) -> Option<Range<DisplayPoint>> {    find_any_delimiters(map, display_point, around, &is_bracket_delimiter)} impl_actions!(vim, [Word, Subword, IndentObj]); actions!(    vim,    [        Sentence,        Paragraph,        Quotes,        BackQuotes,        AnyQuotes,        DoubleQuotes,        VerticalBars,        Parentheses,        AnyBrackets,        SquareBrackets,        CurlyBrackets,        AngleBrackets,        Argument,        Tag,        Method,        Class,        Comment,        EntireFile    ]); pub fn register(editor: &mut Editor, cx: &mut Context<Vim>) {    Vim::action(        editor,        cx,        |vim, &Word { ignore_punctuation }: &Word, window, cx| {            vim.object(Object::Word { ignore_punctuation }, window, cx)        },    );    Vim::action(        editor,        cx,        |vim, &Subword { ignore_punctuation }: &Subword, window, cx| {            vim.object(Object::Subword { ignore_punctuation }, window, cx)        },    );    Vim::action(editor, cx, |vim, _: &Tag, window, cx| {        vim.object(Object::Tag, window, cx)    });    Vim::action(editor, cx, |vim, _: &Sentence, window, cx| {        vim.object(Object::Sentence, window, cx)    });    Vim::action(editor, cx, |vim, _: &Paragraph, window, cx| {        vim.object(Object::Paragraph, window, cx)    });    Vim::action(editor, cx, |vim, _: &Quotes, window, cx| {        vim.object(Object::Quotes, window, cx)    });    Vim::action(editor, cx, |vim, _: &BackQuotes, window, cx| {        vim.object(Object::BackQuotes, window, cx)    });    Vim::action(editor, cx, |vim, _: &AnyQuotes, window, cx| {        vim.object(Object::AnyQuotes, window, cx)    });    Vim::action(editor, cx, |vim, _: &AnyBrackets, window, cx| {        vim.object(Object::AnyBrackets, window, cx)    });    Vim::action(editor, cx, |vim, _: &BackQuotes, window, cx| {        vim.object(Object::BackQuotes, window, cx)    });    Vim::action(editor, cx, |vim, _: &DoubleQuotes, window, cx| {        vim.object(Object::DoubleQuotes, window, cx)    });    Vim::action(editor, cx, |vim, _: &Parentheses, window, cx| {        vim.object(Object::Parentheses, window, cx)    });    Vim::action(editor, cx, |vim, _: &SquareBrackets, window, cx| {        vim.object(Object::SquareBrackets, window, cx)    });    Vim::action(editor, cx, |vim, _: &CurlyBrackets, window, cx| {        vim.object(Object::CurlyBrackets, window, cx)    });    Vim::action(editor, cx, |vim, _: &AngleBrackets, window, cx| {        vim.object(Object::AngleBrackets, window, cx)    });    Vim::action(editor, cx, |vim, _: &VerticalBars, window, cx| {        vim.object(Object::VerticalBars, window, cx)    });    Vim::action(editor, cx, |vim, _: &Argument, window, cx| {        vim.object(Object::Argument, window, cx)    });    Vim::action(editor, cx, |vim, _: &Method, window, cx| {        vim.object(Object::Method, window, cx)    });    Vim::action(editor, cx, |vim, _: &Class, window, cx| {        vim.object(Object::Class, window, cx)    });    Vim::action(editor, cx, |vim, _: &EntireFile, window, cx| {        vim.object(Object::EntireFile, window, cx)    });    Vim::action(editor, cx, |vim, _: &Comment, window, cx| {        if !matches!(vim.active_operator(), Some(Operator::Object { .. })) {            vim.push_operator(Operator::Object { around: true }, window, cx);        }        vim.object(Object::Comment, window, cx)    });    Vim::action(        editor,        cx,        |vim, &IndentObj { include_below }: &IndentObj, window, cx| {            vim.object(Object::IndentObj { include_below }, window, cx)        },    );} impl Vim {    fn object(&mut self, object: Object, window: &mut Window, cx: &mut Context<Self>) {        match self.mode {            Mode::Normal => self.normal_object(object, window, cx),            Mode::Visual | Mode::VisualLine | Mode::VisualBlock => {                self.visual_object(object, window, cx)            }            Mode::Insert | Mode::Replace | Mode::HelixNormal => {                // Shouldn't execute a text object in insert mode. Ignoring            }        }    }} impl Object {    pub fn is_multiline(self) -> bool {        match self {            Object::Word { .. }            | Object::Subword { .. }            | Object::Quotes            | Object::BackQuotes            | Object::AnyQuotes            | Object::VerticalBars            | Object::DoubleQuotes => false,            Object::Sentence            | Object::Paragraph            | Object::AnyBrackets            | Object::Parentheses            | Object::Tag            | Object::AngleBrackets            | Object::CurlyBrackets            | Object::SquareBrackets            | Object::Argument            | Object::Method            | Object::Class            | Object::EntireFile            | Object::Comment            | Object::IndentObj { .. } => true,        }    }     pub fn always_expands_both_ways(self) -> bool {        match self {            Object::Word { .. }            | Object::Subword { .. }            | Object::Sentence            | Object::Paragraph            | Object::Argument            | Object::IndentObj { .. } => false,            Object::Quotes            | Object::BackQuotes            | Object::AnyQuotes            | Object::DoubleQuotes            | Object::VerticalBars            | Object::AnyBrackets            | Object::Parentheses            | Object::SquareBrackets            | Object::Tag            | Object::Method            | Object::Class            | Object::Comment            | Object::EntireFile            | Object::CurlyBrackets            | Object::AngleBrackets => true,        }    }     pub fn target_visual_mode(self, current_mode: Mode, around: bool) -> Mode {        match self {            Object::Word { .. }            | Object::Subword { .. }            | Object::Sentence            | Object::Quotes            | Object::AnyQuotes            | Object::BackQuotes            | Object::DoubleQuotes => {                if current_mode == Mode::VisualBlock {                    Mode::VisualBlock                } else {                    Mode::Visual                }            }            Object::Parentheses            | Object::AnyBrackets            | Object::SquareBrackets            | Object::CurlyBrackets            | Object::AngleBrackets            | Object::VerticalBars            | Object::Tag            | Object::Comment            | Object::Argument            | Object::IndentObj { .. } => Mode::Visual,            Object::Method | Object::Class => {                if around {                    Mode::VisualLine                } else {                    Mode::Visual                }            }            Object::Paragraph | Object::EntireFile => Mode::VisualLine,        }    }     pub fn range(        self,        map: &DisplaySnapshot,        selection: Selection<DisplayPoint>,        around: bool,    ) -> Option<Range<DisplayPoint>> {        let relative_to = selection.head();        match self {            Object::Word { ignore_punctuation } => {                if around {                    around_word(map, relative_to, ignore_punctuation)                } else {                    in_word(map, relative_to, ignore_punctuation)                }            }            Object::Subword { ignore_punctuation } => {                if around {                    around_subword(map, relative_to, ignore_punctuation)                } else {                    in_subword(map, relative_to, ignore_punctuation)                }            }            Object::Sentence => sentence(map, relative_to, around),            Object::Paragraph => paragraph(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::AnyQuotes => find_any_quotes(map, relative_to, around),            Object::DoubleQuotes => {                surrounding_markers(map, relative_to, around, self.is_multiline(), '"', '"')            }            Object::VerticalBars => {

Lines 1–500 of 2698. Load more