66 lines · 2.8 KB · Rust
RawDownload
use crate::{motion::Motion, object::Object, Vim};use collections::HashMap;use editor::{display_map::ToDisplayPoint, Bias};use gpui::{Context, Window};use language::SelectionGoal; impl Vim {    pub fn toggle_comments_motion(        &mut self,        motion: Motion,        times: Option<usize>,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.stop_recording(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 mut selection_starts: HashMap<_, _> = Default::default();                editor.change_selections(None, window, cx, |s| {                    s.move_with(|map, selection| {                        let anchor = map.display_point_to_anchor(selection.head(), Bias::Right);                        selection_starts.insert(selection.id, anchor);                        motion.expand_selection(map, selection, times, false, &text_layout_details);                    });                });                editor.toggle_comments(&Default::default(), window, cx);                editor.change_selections(None, window, cx, |s| {                    s.move_with(|map, selection| {                        let anchor = selection_starts.remove(&selection.id).unwrap();                        selection.collapse_to(anchor.to_display_point(map), SelectionGoal::None);                    });                });            });        });    }     pub fn toggle_comments_object(        &mut self,        object: Object,        around: bool,        window: &mut Window,        cx: &mut Context<Self>,    ) {        self.stop_recording(cx);        self.update_editor(window, cx, |_, editor, window, cx| {            editor.transact(window, cx, |editor, window, cx| {                let mut original_positions: HashMap<_, _> = Default::default();                editor.change_selections(None, window, cx, |s| {                    s.move_with(|map, selection| {                        let anchor = map.display_point_to_anchor(selection.head(), Bias::Right);                        original_positions.insert(selection.id, anchor);                        object.expand_selection(map, selection, around);                    });                });                editor.toggle_comments(&Default::default(), window, cx);                editor.change_selections(None, window, cx, |s| {                    s.move_with(|map, selection| {                        let anchor = original_positions.remove(&selection.id).unwrap();                        selection.collapse_to(anchor.to_display_point(map), SelectionGoal::None);                    });                });            });        });    }}