darwincode 1.9.104

The open source terminal AI coding agent
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

use crate::app::App;
use anyhow::Result;
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};

pub(crate) fn handle_sessions_key(app: &mut App, key: KeyEvent) -> Result<()> {
    if app
        .core
        .keybindings
        .matches(crate::tui::keybindings::TuiAction::Quit, key)
        || app
            .core
            .keybindings
            .matches(crate::tui::keybindings::TuiAction::Cancel, key)
    {
        app.cancel_sessions();
        return Ok(());
    }
    if app
        .core
        .keybindings
        .matches(crate::tui::keybindings::TuiAction::ScrollUp, key)
    {
        app.ui.sessions.select_previous();
        return Ok(());
    }
    if app
        .core
        .keybindings
        .matches(crate::tui::keybindings::TuiAction::ScrollDown, key)
    {
        app.ui.sessions.select_next();
        return Ok(());
    }
    if app
        .core
        .keybindings
        .matches(crate::tui::keybindings::TuiAction::Submit, key)
    {
        app.apply_selected_session();
        return Ok(());
    }

    match (key.code, key.modifiers) {
        (KeyCode::Backspace, _) => {
            let mut q = app.ui.sessions.query.clone();
            q.pop();
            app.ui.sessions.update_query(q);
        }
        (KeyCode::Char(c), modifiers)
            if !modifiers.contains(KeyModifiers::CONTROL)
                && !modifiers.contains(KeyModifiers::ALT) =>
        {
            let mut q = app.ui.sessions.query.clone();
            q.push(c);
            app.ui.sessions.update_query(q);
        }
        _ => {}
    }
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::config::StoredConfig;
    use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};

    #[test]
    fn test_handle_sessions_key_navigation() {
        let _lock = crate::config::TEST_LOCK.lock().unwrap();
        let mut app = App::new(Some(StoredConfig::default()));

        // Mock a session save to make resume_session work
        app.chat.session_id = "a".to_owned();
        crate::app::session::save_session(&app.chat).unwrap();

        app.ui.sessions.sessions = vec![
            crate::app::session::SessionMeta {
                id: "a".to_owned(),
                snippet: "a".to_owned(),
            },
            crate::app::session::SessionMeta {
                id: "b".to_owned(),
                snippet: "b".to_owned(),
            },
        ];
        app.ui.sessions.selected = 0;

        // Press ScrollDown
        let key_down = KeyEvent::new(KeyCode::Down, KeyModifiers::empty());
        handle_sessions_key(&mut app, key_down).unwrap();
        assert_eq!(app.ui.sessions.selected, 1);

        // Press ScrollUp
        let key_up = KeyEvent::new(KeyCode::Up, KeyModifiers::empty());
        handle_sessions_key(&mut app, key_up).unwrap();
        assert_eq!(app.ui.sessions.selected, 0);

        // Type 'x'
        handle_sessions_key(
            &mut app,
            KeyEvent::new(KeyCode::Char('x'), KeyModifiers::empty()),
        )
        .unwrap();
        assert_eq!(app.ui.sessions.query, "x");

        // Backspace
        handle_sessions_key(
            &mut app,
            KeyEvent::new(KeyCode::Backspace, KeyModifiers::empty()),
        )
        .unwrap();
        assert_eq!(app.ui.sessions.query, "");

        // Cancel
        handle_sessions_key(&mut app, KeyEvent::new(KeyCode::Esc, KeyModifiers::empty())).unwrap();
        assert_eq!(app.ui.screen, crate::app::Screen::Chat);

        // Submit (we need to re-enter sessions screen first)
        app.ui.screen = crate::app::Screen::Sessions;
        app.ui.sessions.selected = 0; // Selects session "a"
        handle_sessions_key(
            &mut app,
            KeyEvent::new(KeyCode::Enter, KeyModifiers::empty()),
        )
        .unwrap();
        assert_eq!(app.ui.screen, crate::app::Screen::Chat);

        // Clean up
        if let Ok(config_path) = crate::config::config_path()
            && let Some(parent) = config_path.parent()
        {
            let sessions_dir = parent.join("sessions");
            let _ = std::fs::remove_file(sessions_dir.join("a.json"));
            let _ = std::fs::remove_file(sessions_dir.join("a.meta"));
        }
    }
}