834 lines · 24.7 KB · Rust
RawDownload
mod case;mod change;mod delete;mod paste;mod scroll;mod search;pub mod substitute;mod yank; use std::sync::Arc; use crate::{    motion::{self, Motion},    object::Object,    state::{Mode, Operator},    Vim,};use collections::HashSet;use editor::scroll::autoscroll::Autoscroll;use editor::{Bias, DisplayPoint};use gpui::{actions, AppContext, ViewContext, WindowContext};use language::SelectionGoal;use log::error;use workspace::Workspace; use self::{    case::change_case,    change::{change_motion, change_object},    delete::{delete_motion, delete_object},    substitute::substitute,    yank::{yank_motion, yank_object},}; actions!(    vim,    [        InsertAfter,        InsertFirstNonWhitespace,        InsertEndOfLine,        InsertLineAbove,        InsertLineBelow,        DeleteLeft,        DeleteRight,        ChangeToEndOfLine,        DeleteToEndOfLine,        Yank,        Substitute,        ChangeCase,    ]); pub fn init(cx: &mut AppContext) {    cx.add_action(insert_after);    cx.add_action(insert_first_non_whitespace);    cx.add_action(insert_end_of_line);    cx.add_action(insert_line_above);    cx.add_action(insert_line_below);    cx.add_action(change_case);    search::init(cx);    cx.add_action(|_: &mut Workspace, _: &Substitute, cx| {        Vim::update(cx, |vim, cx| {            let times = vim.pop_number_operator(cx);            substitute(vim, times, cx);        })    });    cx.add_action(|_: &mut Workspace, _: &DeleteLeft, cx| {        Vim::update(cx, |vim, cx| {            let times = vim.pop_number_operator(cx);            delete_motion(vim, Motion::Left, times, cx);        })    });    cx.add_action(|_: &mut Workspace, _: &DeleteRight, cx| {        Vim::update(cx, |vim, cx| {            let times = vim.pop_number_operator(cx);            delete_motion(vim, Motion::Right, times, cx);        })    });    cx.add_action(|_: &mut Workspace, _: &ChangeToEndOfLine, cx| {        Vim::update(cx, |vim, cx| {            let times = vim.pop_number_operator(cx);            change_motion(                vim,                Motion::EndOfLine {                    display_lines: false,                },                times,                cx,            );        })    });    cx.add_action(|_: &mut Workspace, _: &DeleteToEndOfLine, cx| {        Vim::update(cx, |vim, cx| {            let times = vim.pop_number_operator(cx);            delete_motion(                vim,                Motion::EndOfLine {                    display_lines: false,                },                times,                cx,            );        })    });    scroll::init(cx);    paste::init(cx);} pub fn normal_motion(    motion: Motion,    operator: Option<Operator>,    times: Option<usize>,    cx: &mut WindowContext,) {    Vim::update(cx, |vim, cx| {        match operator {            None => move_cursor(vim, motion, times, cx),            Some(Operator::Change) => change_motion(vim, motion, times, cx),            Some(Operator::Delete) => delete_motion(vim, motion, times, cx),            Some(Operator::Yank) => yank_motion(vim, motion, times, cx),            Some(operator) => {                // Can't do anything for text objects, Ignoring                error!("Unexpected normal mode motion operator: {:?}", operator)            }        }    });} pub fn normal_object(object: Object, cx: &mut WindowContext) {    Vim::update(cx, |vim, cx| {        match vim.maybe_pop_operator() {            Some(Operator::Object { around }) => match vim.maybe_pop_operator() {                Some(Operator::Change) => change_object(vim, object, around, cx),                Some(Operator::Delete) => delete_object(vim, object, around, cx),                Some(Operator::Yank) => yank_object(vim, object, around, cx),                _ => {                    // Can't do anything for namespace operators. Ignoring                }            },            _ => {                // Can't do anything with change/delete/yank and text objects. Ignoring            }        }        vim.clear_operator(cx);    })} fn move_cursor(vim: &mut Vim, motion: Motion, times: Option<usize>, cx: &mut WindowContext) {    vim.update_active_editor(cx, |editor, cx| {        editor.change_selections(Some(Autoscroll::fit()), cx, |s| {            s.move_cursors_with(|map, cursor, goal| {                motion                    .move_point(map, cursor, goal, times)                    .unwrap_or((cursor, goal))            })        })    });} fn insert_after(_: &mut Workspace, _: &InsertAfter, cx: &mut ViewContext<Workspace>) {    Vim::update(cx, |vim, cx| {        vim.switch_mode(Mode::Insert, false, cx);        vim.update_active_editor(cx, |editor, cx| {            editor.change_selections(Some(Autoscroll::fit()), cx, |s| {                s.maybe_move_cursors_with(|map, cursor, goal| {                    Motion::Right.move_point(map, cursor, goal, None)                });            });        });    });} fn insert_first_non_whitespace(    _: &mut Workspace,    _: &InsertFirstNonWhitespace,    cx: &mut ViewContext<Workspace>,) {    Vim::update(cx, |vim, cx| {        vim.switch_mode(Mode::Insert, false, cx);        vim.update_active_editor(cx, |editor, cx| {            editor.change_selections(Some(Autoscroll::fit()), cx, |s| {                s.maybe_move_cursors_with(|map, cursor, goal| {                    Motion::FirstNonWhitespace {                        display_lines: false,                    }                    .move_point(map, cursor, goal, None)                });            });        });    });} fn insert_end_of_line(_: &mut Workspace, _: &InsertEndOfLine, cx: &mut ViewContext<Workspace>) {    Vim::update(cx, |vim, cx| {        vim.switch_mode(Mode::Insert, false, cx);        vim.update_active_editor(cx, |editor, cx| {            editor.change_selections(Some(Autoscroll::fit()), cx, |s| {                s.maybe_move_cursors_with(|map, cursor, goal| {                    Motion::CurrentLine.move_point(map, cursor, goal, None)                });            });        });    });} fn insert_line_above(_: &mut Workspace, _: &InsertLineAbove, cx: &mut ViewContext<Workspace>) {    Vim::update(cx, |vim, cx| {        vim.switch_mode(Mode::Insert, false, cx);        vim.update_active_editor(cx, |editor, cx| {            editor.transact(cx, |editor, cx| {                let (map, old_selections) = editor.selections.all_display(cx);                let selection_start_rows: HashSet<u32> = old_selections                    .into_iter()                    .map(|selection| selection.start.row())                    .collect();                let edits = selection_start_rows.into_iter().map(|row| {                    let (indent, _) = map.line_indent(row);                    let start_of_line =                        motion::start_of_line(&map, false, DisplayPoint::new(row, 0))                            .to_point(&map);                    let mut new_text = " ".repeat(indent as usize);                    new_text.push('\n');                    (start_of_line..start_of_line, new_text)                });                editor.edit_with_autoindent(edits, cx);                editor.change_selections(Some(Autoscroll::fit()), cx, |s| {                    s.move_cursors_with(|map, cursor, _| {                        let previous_line = motion::up(map, cursor, SelectionGoal::None, 1).0;                        let insert_point = motion::end_of_line(map, false, previous_line);                        (insert_point, SelectionGoal::None)                    });                });            });        });    });} fn insert_line_below(_: &mut Workspace, _: &InsertLineBelow, cx: &mut ViewContext<Workspace>) {    Vim::update(cx, |vim, cx| {        vim.switch_mode(Mode::Insert, false, cx);        vim.update_active_editor(cx, |editor, cx| {            editor.transact(cx, |editor, cx| {                let (map, old_selections) = editor.selections.all_display(cx);                 let selection_end_rows: HashSet<u32> = old_selections                    .into_iter()                    .map(|selection| selection.end.row())                    .collect();                let edits = selection_end_rows.into_iter().map(|row| {                    let (indent, _) = map.line_indent(row);                    let end_of_line =                        motion::end_of_line(&map, false, DisplayPoint::new(row, 0)).to_point(&map);                     let mut new_text = "\n".to_string();                    new_text.push_str(&" ".repeat(indent as usize));                    (end_of_line..end_of_line, new_text)                });                editor.change_selections(Some(Autoscroll::fit()), cx, |s| {                    s.maybe_move_cursors_with(|map, cursor, goal| {                        Motion::CurrentLine.move_point(map, cursor, goal, None)                    });                });                editor.edit_with_autoindent(edits, cx);            });        });    });} pub(crate) fn normal_replace(text: Arc<str>, cx: &mut WindowContext) {    Vim::update(cx, |vim, cx| {        vim.update_active_editor(cx, |editor, cx| {            editor.transact(cx, |editor, cx| {                editor.set_clip_at_line_ends(false, cx);                let (map, display_selections) = editor.selections.all_display(cx);                // Selections are biased right at the start. So we need to store                // anchors that are biased left so that we can restore the selections                // after the change                let stable_anchors = editor                    .selections                    .disjoint_anchors()                    .into_iter()                    .map(|selection| {                        let start = selection.start.bias_left(&map.buffer_snapshot);                        start..start                    })                    .collect::<Vec<_>>();                 let edits = display_selections                    .into_iter()                    .map(|selection| {                        let mut range = selection.range();                        *range.end.column_mut() += 1;                        range.end = map.clip_point(range.end, Bias::Right);                         (                            range.start.to_offset(&map, Bias::Left)                                ..range.end.to_offset(&map, Bias::Left),                            text.clone(),                        )                    })                    .collect::<Vec<_>>();                 editor.buffer().update(cx, |buffer, cx| {                    buffer.edit(edits, None, cx);                });                editor.set_clip_at_line_ends(true, cx);                editor.change_selections(None, cx, |s| {                    s.select_anchor_ranges(stable_anchors);                });            });        });        vim.pop_operator(cx)    });} #[cfg(test)]mod test {    use gpui::TestAppContext;    use indoc::indoc;     use crate::{        state::Mode::{self},        test::{ExemptionFeatures, NeovimBackedTestContext},    };     #[gpui::test]    async fn test_h(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["h"]);        cx.assert_all(indoc! {"            ˇThe qˇuick            ˇbrown"        })        .await;    }     #[gpui::test]    async fn test_backspace(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx)            .await            .binding(["backspace"]);        cx.assert_all(indoc! {"            ˇThe qˇuick            ˇbrown"        })        .await;    }     #[gpui::test]    async fn test_j(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["j"]);        cx.assert_all(indoc! {"            ˇThe qˇuick broˇwn            ˇfox jumps"        })        .await;    }     #[gpui::test]    async fn test_enter(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["enter"]);        cx.assert_all(indoc! {"            ˇThe qˇuick broˇwn            ˇfox jumps"        })        .await;    }     #[gpui::test]    async fn test_k(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["k"]);        cx.assert_all(indoc! {"            ˇThe qˇuick            ˇbrown fˇox jumˇps"        })        .await;    }     #[gpui::test]    async fn test_l(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["l"]);        cx.assert_all(indoc! {"            ˇThe qˇuicˇk            ˇbrowˇn"})            .await;    }     #[gpui::test]    async fn test_jump_to_line_boundaries(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await;        cx.assert_binding_matches_all(            ["$"],            indoc! {"            ˇThe qˇuicˇk            ˇbrowˇn"},        )        .await;        cx.assert_binding_matches_all(            ["0"],            indoc! {"                ˇThe qˇuicˇk                ˇbrowˇn"},        )        .await;    }     #[gpui::test]    async fn test_jump_to_end(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["shift-g"]);         cx.assert_all(indoc! {"                The ˇquick                 brown fox jumps                overˇ the lazy doˇg"})            .await;        cx.assert(indoc! {"            The quiˇck             brown"})            .await;        cx.assert(indoc! {"            The quiˇck             "})            .await;    }     #[gpui::test]    async fn test_w(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["w"]);        cx.assert_all(indoc! {"            The ˇquickˇ-ˇbrown            ˇ            ˇ            ˇfox_jumps ˇover            ˇthˇe"})            .await;        let mut cx = cx.binding(["shift-w"]);        cx.assert_all(indoc! {"            The ˇquickˇ-ˇbrown            ˇ            ˇ            ˇfox_jumps ˇover            ˇthˇe"})            .await;    }     #[gpui::test]    async fn test_e(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["e"]);        cx.assert_all(indoc! {"            Thˇe quicˇkˇ-browˇn              fox_jumpˇs oveˇr            thˇe"})            .await;        let mut cx = cx.binding(["shift-e"]);        cx.assert_all(indoc! {"            Thˇe quicˇkˇ-browˇn              fox_jumpˇs oveˇr            thˇe"})            .await;    }     #[gpui::test]    async fn test_b(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await.binding(["b"]);        cx.assert_all(indoc! {"            ˇThe ˇquickˇ-ˇbrown            ˇ            ˇ            ˇfox_jumps ˇover            ˇthe"})            .await;        let mut cx = cx.binding(["shift-b"]);        cx.assert_all(indoc! {"            ˇThe ˇquickˇ-ˇbrown            ˇ            ˇ            ˇfox_jumps ˇover            ˇthe"})            .await;    }     #[gpui::test]    async fn test_gg(cx: &mut gpui::TestAppContext) {        let mut cx = NeovimBackedTestContext::new(cx).await;        cx.assert_binding_matches_all(            ["g", "g"],            indoc! {"                The qˇuick                 brown fox jumps                over ˇthe laˇzy dog"},        )        .await;        cx.assert_binding_matches(            ["g", "g"],

Lines 1–500 of 834.