1638 lines · 51.0 KB · Rust
RawDownload
mod case;mod change;mod delete;mod increment;pub(crate) mod mark;mod paste;pub(crate) mod repeat;mod scroll;pub(crate) mod search;pub mod substitute;mod toggle_comments;pub(crate) mod yank; use std::collections::HashMap;use std::sync::Arc; use crate::{    indent::IndentDirection,    motion::{self, first_non_whitespace, next_line_end, right, Motion},    object::Object,    state::{Mode, Operator},    surrounds::SurroundsType,    Vim,};use case::CaseTarget;use collections::BTreeSet;use editor::scroll::Autoscroll;use editor::Anchor;use editor::Bias;use editor::Editor;use editor::{display_map::ToDisplayPoint, movement};use gpui::{actions, Context, Window};use language::{Point, SelectionGoal, ToPoint};use log::error;use multi_buffer::MultiBufferRow; actions!(    vim,    [        InsertAfter,        InsertBefore,        InsertFirstNonWhitespace,        InsertEndOfLine,        InsertLineAbove,        InsertLineBelow,        InsertAtPrevious,        JoinLines,        JoinLinesNoWhitespace,        DeleteLeft,        DeleteRight,        ChangeToEndOfLine,        DeleteToEndOfLine,        Yank,        YankLine,        ChangeCase,        ConvertToUpperCase,        ConvertToLowerCase,        ToggleComments,        ShowLocation,        Undo,        Redo,    ]); pub(crate) fn register(editor: &mut Editor, cx: &mut Context<Vim>) {    Vim::action(editor, cx, Vim::insert_after);    Vim::action(editor, cx, Vim::insert_before);    Vim::action(editor, cx, Vim::insert_first_non_whitespace);    Vim::action(editor, cx, Vim::insert_end_of_line);    Vim::action(editor, cx, Vim::insert_line_above);    Vim::action(editor, cx, Vim::insert_line_below);    Vim::action(editor, cx, Vim::insert_at_previous);    Vim::action(editor, cx, Vim::change_case);    Vim::action(editor, cx, Vim::convert_to_upper_case);    Vim::action(editor, cx, Vim::convert_to_lower_case);    Vim::action(editor, cx, Vim::yank_line);    Vim::action(editor, cx, Vim::toggle_comments);    Vim::action(editor, cx, Vim::paste);    Vim::action(editor, cx, Vim::show_location);     Vim::action(editor, cx, |vim, _: &DeleteLeft, window, cx| {        vim.record_current_action(cx);        let times = Vim::take_count(cx);        vim.delete_motion(Motion::Left, times, window, cx);    });    Vim::action(editor, cx, |vim, _: &DeleteRight, window, cx| {        vim.record_current_action(cx);        let times = Vim::take_count(cx);        vim.delete_motion(Motion::Right, times, window, cx);    });    Vim::action(editor, cx, |vim, _: &ChangeToEndOfLine, window, cx| {        vim.start_recording(cx);        let times = Vim::take_count(cx);        vim.change_motion(            Motion::EndOfLine {                display_lines: false,            },            times,            window,            cx,        );    });    Vim::action(editor, cx, |vim, _: &DeleteToEndOfLine, window, cx| {        vim.record_current_action(cx);        let times = Vim::take_count(cx);        vim.delete_motion(            Motion::EndOfLine {                display_lines: false,            },            times,            window,            cx,        );    });    Vim::action(editor, cx, |vim, _: &JoinLines, window, cx| {        vim.join_lines_impl(true, window, cx);    });     Vim::action(editor, cx, |vim, _: &JoinLinesNoWhitespace, window, cx| {        vim.join_lines_impl(false, window, cx);    });     Vim::action(editor, cx, |vim, _: &Undo, window, cx| {        let times = Vim::take_count(cx);        vim.update_editor(window, cx, |_, editor, window, cx| {            for _ in 0..times.unwrap_or(1) {                editor.undo(&editor::actions::Undo, window, cx);            }        });    });    Vim::action(editor, cx, |vim, _: &Redo, window, cx| {        let times = Vim::take_count(cx);        vim.update_editor(window, cx, |_, editor, window, cx| {            for _ in 0..times.unwrap_or(1) {                editor.redo(&editor::actions::Redo, window, cx);            }        });    });     repeat::register(editor, cx);    scroll::register(editor, cx);    search::register(editor, cx);    substitute::register(editor, cx);    increment::register(editor, cx);} impl Vim {    pub fn normal_motion(        &mut self,        motion: Motion,        operator: Option<Operator>,        times: Option<usize>,        window: &mut Window,        cx: &mut Context<Self>,    ) {        match operator {            None => self.move_cursor(motion, times, window, cx),            Some(Operator::Change) => self.change_motion(motion, times, window, cx),            Some(Operator::Delete) => self.delete_motion(motion, times, window, cx),            Some(Operator::Yank) => self.yank_motion(motion, times, window, cx),            Some(Operator::AddSurrounds { target: None }) => {}            Some(Operator::Indent) => {                self.indent_motion(motion, times, IndentDirection::In, window, cx)            }            Some(Operator::Rewrap) => self.rewrap_motion(motion, times, window, cx),            Some(Operator::Outdent) => {                self.indent_motion(motion, times, IndentDirection::Out, window, cx)            }            Some(Operator::AutoIndent) => {                self.indent_motion(motion, times, IndentDirection::Auto, window, cx)            }            Some(Operator::ShellCommand) => self.shell_command_motion(motion, times, window, cx),            Some(Operator::Lowercase) => {                self.change_case_motion(motion, times, CaseTarget::Lowercase, window, cx)            }            Some(Operator::Uppercase) => {                self.change_case_motion(motion, times, CaseTarget::Uppercase, window, cx)            }            Some(Operator::OppositeCase) => {                self.change_case_motion(motion, times, CaseTarget::OppositeCase, window, cx)            }            Some(Operator::ToggleComments) => {                self.toggle_comments_motion(motion, times, window, cx)            }            Some(Operator::ReplaceWithRegister) => {                self.replace_with_register_motion(motion, times, window, cx)            }            Some(Operator::Exchange) => self.exchange_motion(motion, times, window, cx),            Some(operator) => {                // Can't do anything for text objects, Ignoring                error!("Unexpected normal mode motion operator: {:?}", operator)            }        }        // Exit temporary normal mode (if active).        self.exit_temporary_normal(window, cx);    }     pub fn normal_object(&mut self, object: Object, window: &mut Window, cx: &mut Context<Self>) {        let mut waiting_operator: Option<Operator> = None;        match self.maybe_pop_operator() {            Some(Operator::Object { around }) => match self.maybe_pop_operator() {                Some(Operator::Change) => self.change_object(object, around, window, cx),                Some(Operator::Delete) => self.delete_object(object, around, window, cx),                Some(Operator::Yank) => self.yank_object(object, around, window, cx),                Some(Operator::Indent) => {                    self.indent_object(object, around, IndentDirection::In, window, cx)                }                Some(Operator::Outdent) => {                    self.indent_object(object, around, IndentDirection::Out, window, cx)                }                Some(Operator::AutoIndent) => {                    self.indent_object(object, around, IndentDirection::Auto, window, cx)                }                Some(Operator::ShellCommand) => {                    self.shell_command_object(object, around, window, cx);                }                Some(Operator::Rewrap) => self.rewrap_object(object, around, window, cx),                Some(Operator::Lowercase) => {                    self.change_case_object(object, around, CaseTarget::Lowercase, window, cx)                }                Some(Operator::Uppercase) => {                    self.change_case_object(object, around, CaseTarget::Uppercase, window, cx)                }                Some(Operator::OppositeCase) => {                    self.change_case_object(object, around, CaseTarget::OppositeCase, window, cx)                }                Some(Operator::AddSurrounds { target: None }) => {                    waiting_operator = Some(Operator::AddSurrounds {                        target: Some(SurroundsType::Object(object, around)),                    });                }                Some(Operator::ToggleComments) => {                    self.toggle_comments_object(object, around, window, cx)                }                Some(Operator::ReplaceWithRegister) => {                    self.replace_with_register_object(object, around, window, cx)                }                Some(Operator::Exchange) => self.exchange_object(object, around, window, cx),                _ => {                    // Can't do anything for namespace operators. Ignoring                }            },            Some(Operator::DeleteSurrounds) => {                waiting_operator = Some(Operator::DeleteSurrounds);            }            Some(Operator::ChangeSurrounds { target: None }) => {                if self.check_and_move_to_valid_bracket_pair(object, window, cx) {                    waiting_operator = Some(Operator::ChangeSurrounds {                        target: Some(object),                    });                }            }            _ => {                // Can't do anything with change/delete/yank/surrounds and text objects. Ignoring            }        }        self.clear_operator(window, cx);        if let Some(operator) = waiting_operator {            self.push_operator(operator, window, cx);        }    }     pub(crate) fn move_cursor(        &mut self,        motion: Motion,        times: Option<usize>,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.update_editor(window, cx, |_, editor, window, cx| {            let text_layout_details = editor.text_layout_details(window);            editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                s.move_cursors_with(|map, cursor, goal| {                    motion                        .move_point(map, cursor, goal, times, &text_layout_details)                        .unwrap_or((cursor, goal))                })            })        });    }     fn insert_after(&mut self, _: &InsertAfter, window: &mut Window, cx: &mut Context<Self>) {        self.start_recording(cx);        self.switch_mode(Mode::Insert, false, window, cx);        self.update_editor(window, cx, |_, editor, window, cx| {            editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                s.move_cursors_with(|map, cursor, _| (right(map, cursor, 1), SelectionGoal::None));            });        });    }     fn insert_before(&mut self, _: &InsertBefore, window: &mut Window, cx: &mut Context<Self>) {        self.start_recording(cx);        if self.mode.is_visual() {            let current_mode = self.mode;            self.update_editor(window, cx, |_, editor, window, cx| {                editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                    s.move_with(|map, selection| {                        if current_mode == Mode::VisualLine {                            let start_of_line = motion::start_of_line(map, false, selection.start);                            selection.collapse_to(start_of_line, SelectionGoal::None)                        } else {                            selection.collapse_to(selection.start, SelectionGoal::None)                        }                    });                });            });        }        self.switch_mode(Mode::Insert, false, window, cx);    }     fn insert_first_non_whitespace(        &mut self,        _: &InsertFirstNonWhitespace,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.start_recording(cx);        self.switch_mode(Mode::Insert, false, window, cx);        self.update_editor(window, cx, |_, editor, window, cx| {            editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                s.move_cursors_with(|map, cursor, _| {                    (                        first_non_whitespace(map, false, cursor),                        SelectionGoal::None,                    )                });            });        });    }     fn insert_end_of_line(        &mut self,        _: &InsertEndOfLine,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.start_recording(cx);        self.switch_mode(Mode::Insert, false, window, cx);        self.update_editor(window, cx, |_, editor, window, cx| {            editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                s.move_cursors_with(|map, cursor, _| {                    (next_line_end(map, cursor, 1), SelectionGoal::None)                });            });        });    }     fn insert_at_previous(        &mut self,        _: &InsertAtPrevious,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.start_recording(cx);        self.switch_mode(Mode::Insert, false, window, cx);        self.update_editor(window, cx, |vim, editor, window, cx| {            if let Some(marks) = vim.marks.get("^") {                editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                    s.select_anchor_ranges(marks.iter().map(|mark| *mark..*mark))                });            }        });    }     fn insert_line_above(        &mut self,        _: &InsertLineAbove,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.start_recording(cx);        self.switch_mode(Mode::Insert, false, window, cx);        self.update_editor(window, cx, |_, editor, window, cx| {            editor.transact(window, cx, |editor, window, cx| {                let selections = editor.selections.all::<Point>(cx);                let snapshot = editor.buffer().read(cx).snapshot(cx);                 let selection_start_rows: BTreeSet<u32> = selections                    .into_iter()                    .map(|selection| selection.start.row)                    .collect();                let edits = selection_start_rows                    .into_iter()                    .map(|row| {                        let indent = snapshot                            .indent_and_comment_for_line(MultiBufferRow(row), cx)                            .chars()                            .collect::<String>();                         let start_of_line = Point::new(row, 0);                        (start_of_line..start_of_line, indent + "\n")                    })                    .collect::<Vec<_>>();                editor.edit_with_autoindent(edits, cx);                editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                    s.move_cursors_with(|map, cursor, _| {                        let previous_line = motion::start_of_relative_buffer_row(map, cursor, -1);                        let insert_point = motion::end_of_line(map, false, previous_line, 1);                        (insert_point, SelectionGoal::None)                    });                });            });        });    }     fn insert_line_below(        &mut self,        _: &InsertLineBelow,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.start_recording(cx);        self.switch_mode(Mode::Insert, false, window, cx);        self.update_editor(window, cx, |_, editor, window, cx| {            let text_layout_details = editor.text_layout_details(window);            editor.transact(window, cx, |editor, window, cx| {                let selections = editor.selections.all::<Point>(cx);                let snapshot = editor.buffer().read(cx).snapshot(cx);                 let selection_end_rows: BTreeSet<u32> = selections                    .into_iter()                    .map(|selection| selection.end.row)                    .collect();                let edits = selection_end_rows                    .into_iter()                    .map(|row| {                        let indent = snapshot                            .indent_and_comment_for_line(MultiBufferRow(row), cx)                            .chars()                            .collect::<String>();                         let end_of_line = Point::new(row, snapshot.line_len(MultiBufferRow(row)));                        (end_of_line..end_of_line, "\n".to_string() + &indent)                    })                    .collect::<Vec<_>>();                editor.change_selections(Some(Autoscroll::fit()), window, cx, |s| {                    s.maybe_move_cursors_with(|map, cursor, goal| {                        Motion::CurrentLine.move_point(                            map,                            cursor,                            goal,                            None,                            &text_layout_details,                        )                    });                });                editor.edit_with_autoindent(edits, cx);            });        });    }     fn join_lines_impl(        &mut self,        insert_whitespace: bool,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.record_current_action(cx);        let mut times = Vim::take_count(cx).unwrap_or(1);        if self.mode.is_visual() {            times = 1;        } else if times > 1 {            // 2J joins two lines together (same as J or 1J)            times -= 1;        }         self.update_editor(window, cx, |_, editor, window, cx| {            editor.transact(window, cx, |editor, window, cx| {                for _ in 0..times {                    editor.join_lines_impl(insert_whitespace, window, cx)                }            })        });        if self.mode.is_visual() {            self.switch_mode(Mode::Normal, true, window, cx)        }    }     fn yank_line(&mut self, _: &YankLine, window: &mut Window, cx: &mut Context<Self>) {        let count = Vim::take_count(cx);        self.yank_motion(motion::Motion::CurrentLine, count, window, cx)    }     fn show_location(&mut self, _: &ShowLocation, window: &mut Window, cx: &mut Context<Self>) {        let count = Vim::take_count(cx);        self.update_editor(window, cx, |vim, editor, _window, cx| {            let selection = editor.selections.newest_anchor();            if let Some((_, buffer, _)) = editor.active_excerpt(cx) {                let filename = if let Some(file) = buffer.read(cx).file() {                    if count.is_some() {                        if let Some(local) = file.as_local() {                            local.abs_path(cx).to_string_lossy().to_string()                        } else {                            file.full_path(cx).to_string_lossy().to_string()                        }                    } else {                        file.path().to_string_lossy().to_string()                    }                } else {

Lines 1–500 of 1638.