1use crate::{Vim, VimEvent};
 2use editor::{EditorBlurred, EditorFocused, EditorReleased};
 3use gpui::AppContext;
 4
 5pub fn init(cx: &mut AppContext) {
 6    cx.subscribe_global(focused).detach();
 7    cx.subscribe_global(blurred).detach();
 8    cx.subscribe_global(released).detach();
 9}
10
11fn focused(EditorFocused(editor): &EditorFocused, cx: &mut AppContext) {
12    if let Some(previously_active_editor) = Vim::read(cx).active_editor.clone() {
13        previously_active_editor.window().update(cx, |cx| {
14            Vim::update(cx, |vim, cx| {
15                vim.update_active_editor(cx, |previously_active_editor, cx| {
16                    vim.unhook_vim_settings(previously_active_editor, cx)
17                });
18            });
19        });
20    }
21
22    editor.window().update(cx, |cx| {
23        Vim::update(cx, |vim, cx| {
24            vim.set_active_editor(editor.clone(), cx);
25            if vim.enabled {
26                cx.emit_global(VimEvent::ModeChanged {
27                    mode: vim.state().mode,
28                });
29            }
30        });
31    });
32}
33
34fn blurred(EditorBlurred(editor): &EditorBlurred, cx: &mut AppContext) {
35    editor.window().update(cx, |cx| {
36        Vim::update(cx, |vim, cx| {
37            vim.workspace_state.recording = false;
38            vim.workspace_state.recorded_actions.clear();
39            if let Some(previous_editor) = vim.active_editor.clone() {
40                if previous_editor == editor.clone() {
41                    vim.clear_operator(cx);
42                    vim.active_editor = None;
43                    vim.editor_subscription = None;
44                }
45            }
46
47            editor.update(cx, |editor, cx| vim.unhook_vim_settings(editor, cx))
48        });
49    });
50}
51
52fn released(EditorReleased(editor): &EditorReleased, cx: &mut AppContext) {
53    editor.window().update(cx, |cx| {
54        Vim::update(cx, |vim, _| {
55            if let Some(previous_editor) = vim.active_editor.clone() {
56                if previous_editor == editor.clone() {
57                    vim.active_editor = None;
58                    vim.editor_subscription = None;
59                }
60            }
61            vim.editor_states.remove(&editor.id())
62        });
63    });
64}
65
66#[cfg(test)]
67mod test {
68    use crate::{test::VimTestContext, Vim};
69    use editor::Editor;
70    use gpui::View;
71    use language::Buffer;
72
73    // regression test for blur called with a different active editor
74    #[gpui::test]
75    async fn test_blur_focus(cx: &mut gpui::TestAppContext) {
76        let mut cx = VimTestContext::new(cx, true).await;
77
78        let buffer = cx.add_model(|_| Buffer::new(0, 0, "a = 1\nb = 2\n"));
79        let window2 = cx.add_window(|cx| Editor::for_buffer(buffer, None, cx));
80        let editor2 = cx.read(|cx| window2.root(cx)).unwrap();
81
82        cx.update(|cx| {
83            let vim = Vim::read(cx);
84            assert_eq!(vim.active_editor.unwrap().id(), editor2.id())
85        });
86
87        // no panic when blurring an editor in a different window.
88        cx.update_editor(|editor1, cx| {
89            editor1.focus_out(cx.handle().into_any(), cx);
90        });
91    }
92}