darwincode 1.9.89

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

use anyhow::Result;
use crossterm::event::KeyEvent;
use std::sync::mpsc::Sender;

use crate::app::{App, SubmitAction};
use crate::tui::{
    WorkerEvent, handle_function_action, spawn_generation_worker, spawn_models_worker,
};

pub(crate) fn handle_permissions_key(
    app: &mut App,
    sender: &Sender<WorkerEvent>,
    key: KeyEvent,
) -> Result<()> {
    if app
        .core.keybindings
        .matches(crate::tui::keybindings::TuiAction::Quit, key)
    {
        app.cancel_permissions();
        return Ok(());
    }
    if app
        .core.keybindings
        .matches(crate::tui::keybindings::TuiAction::Cancel, key)
    {
        app.cancel_permissions();
        return Ok(());
    }
    if app
        .core.keybindings
        .matches(crate::tui::keybindings::TuiAction::ScrollUp, key)
    {
        app.ui.permissions.select_previous();
        return Ok(());
    }
    if app
        .core.keybindings
        .matches(crate::tui::keybindings::TuiAction::ScrollDown, key)
    {
        app.ui.permissions.select_next();
        return Ok(());
    }
    if app
        .core.keybindings
        .matches(crate::tui::keybindings::TuiAction::Submit, key)
    {
        if let Some(action) = app.apply_permission_level() {
            match action {
                SubmitAction::Generate(request) => {
                    spawn_generation_worker(
                        request.config,
                        request.history,
                        request.cancel_token,
                        request.generation_id,
                        request.dev_mode,
                        sender.clone(),
                    );
                }
                SubmitAction::LoadModels(config) => {
                    spawn_models_worker(config, sender.clone());
                }
                SubmitAction::ExecuteFunction { name, args, config } => {
                    handle_function_action(
                        crate::app::FunctionAction::Execute { name, args, config },
                        sender,
                    );
                }
            }
        }
        return Ok(());
    }
    Ok(())
}

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

    #[test]
    fn test_handle_permissions_key_navigation() {
        let mut app = App::new(Some(StoredConfig::default()));
        let (sender, _receiver) = mpsc::channel();

        // Initially selected = 0 (Safe)
        assert_eq!(app.ui.permissions.selected, 0);

        // Press ScrollDown (Keybindings map Down key to ScrollDown)
        let key_down = KeyEvent::new(KeyCode::Down, KeyModifiers::empty());
        handle_permissions_key(&mut app, &sender, key_down).unwrap();
        assert_eq!(app.ui.permissions.selected, 1);

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

    #[test]
    fn test_handle_permissions_key_submit_and_cancel() {
        let _lock = crate::config::TEST_LOCK.lock().unwrap();
        let mut app = App::new(Some(StoredConfig::default()));
        let (sender, _receiver) = mpsc::channel();

        // Cancel
        app.ui.screen = crate::app::Screen::Permissions;
        let key_cancel = KeyEvent::new(KeyCode::Esc, KeyModifiers::empty());
        handle_permissions_key(&mut app, &sender, key_cancel).unwrap();
        assert_eq!(app.ui.screen, crate::app::Screen::Chat);

        // Submit with Chaos and ExecuteFunction
        app.ui.screen = crate::app::Screen::Permissions;
        app.ui.permissions.selected = 2; // Chaos
        app.proc.pending = Some(crate::app::PendingTask::ConfirmFunction {
            name: "read".to_owned(),
            args: serde_json::json!({}),
        });

        let key_enter = KeyEvent::new(KeyCode::Enter, KeyModifiers::empty());
        handle_permissions_key(&mut app, &sender, key_enter).unwrap();
        assert_eq!(app.ui.screen, crate::app::Screen::Chat);
        assert_eq!(app.chat.config.permission_level, crate::config::PermissionLevel::Chaos);
    }
}