kraal 0.1.1

A terminal workspace manager for SSH fleets and coding agents
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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164

//! Scrolling a pane's history and searching it.

use alacritty_terminal::selection::SelectionType;

use crate::input::{InputEvent, Key, KeyCode};
use crate::layout::PaneId;
use crate::server::commands::NO_MARKS;
use crate::server::ui::Mode;
use crate::server::{ClientId, Server};

impl Server {
    pub(super) fn on_scroll(&mut self, id: ClientId, ev: InputEvent) -> Mode {
        let Some(p) = self.client_focus(id) else { return Mode::Normal };
        match ev {
            InputEvent::Key(k) => {
                let Some(pane) = self.panes.get_mut(&p) else { return Mode::Normal };
                if !pane.is_terminal() {
                    return self.scroll_other(id, p, &k);
                }
                pane.copy_mode(true);
                let page = pane.rows.max(2) as i32 - 1;
                let half = (pane.rows / 2).max(1) as i32;
                use alacritty_terminal::vi_mode::ViMotion as V;
                match (k.code, k.mods.ctrl) {
                    (KeyCode::Up, _) | (KeyCode::Char('k'), false) => pane.vi_motion(V::Up),
                    (KeyCode::Down, _) | (KeyCode::Char('j'), false) => pane.vi_motion(V::Down),
                    (KeyCode::Left, _) | (KeyCode::Char('h'), false) => pane.vi_motion(V::Left),
                    (KeyCode::Right, _) | (KeyCode::Char('l'), false) => pane.vi_motion(V::Right),
                    (KeyCode::Char('w'), false) => pane.vi_motion(V::SemanticRight),
                    (KeyCode::Char('b'), false) => pane.vi_motion(V::SemanticLeft),
                    (KeyCode::Char('e'), false) => pane.vi_motion(V::SemanticRightEnd),
                    (KeyCode::Char('W'), false) => pane.vi_motion(V::WordRight),
                    (KeyCode::Char('B'), false) => pane.vi_motion(V::WordLeft),
                    (KeyCode::Char('E'), false) => pane.vi_motion(V::WordRightEnd),
                    (KeyCode::Char('0'), false) | (KeyCode::Home, _) => pane.vi_motion(V::First),
                    (KeyCode::Char('^'), false) => pane.vi_motion(V::FirstOccupied),
                    (KeyCode::Char('$'), false) | (KeyCode::End, _) => pane.vi_motion(V::Last),
                    (KeyCode::Char('H'), false) => pane.vi_motion(V::High),
                    (KeyCode::Char('M'), false) => pane.vi_motion(V::Middle),
                    (KeyCode::Char('L'), false) => pane.vi_motion(V::Low),
                    (KeyCode::Char('{'), false) => pane.vi_motion(V::ParagraphUp),
                    (KeyCode::Char('}'), false) => pane.vi_motion(V::ParagraphDown),
                    (KeyCode::Char('%'), false) => pane.vi_motion(V::Bracket),
                    (KeyCode::Char('g'), false) => pane.vi_edge(true),
                    (KeyCode::Char('G'), false) => pane.vi_edge(false),
                    (KeyCode::PageUp, _) | (KeyCode::Char('b'), true) => pane.vi_scroll(page),
                    (KeyCode::PageDown, _) | (KeyCode::Char('f'), true) | (KeyCode::Char(' '), false) => {
                        pane.vi_scroll(-page)
                    }
                    (KeyCode::Char('u'), true) => pane.vi_scroll(half),
                    (KeyCode::Char('d'), true) => pane.vi_scroll(-half),
                    (KeyCode::Char('y'), true) => pane.scroll(1),
                    (KeyCode::Char('e'), true) => pane.scroll(-1),
                    (KeyCode::Char('v'), false) => pane.vi_select(SelectionType::Simple),
                    (KeyCode::Char('V'), false) => pane.vi_select(SelectionType::Lines),
                    (KeyCode::Char('v'), true) => pane.vi_select(SelectionType::Block),
                    (KeyCode::Char('y'), false) | (KeyCode::Enter, _) => {
                        if let Some(text) = pane.selection_text() {
                            self.copy_to_client(id, &text);
                        }
                        let pane = self.panes.get_mut(&p).unwrap();
                        pane.end_search();
                        pane.copy_mode(false);
                        return Mode::Normal;
                    }
                    (KeyCode::Char('/'), false) => {
                        return Mode::Search { query: String::new(), found: false };
                    }
                    (KeyCode::Char('n'), false) => {
                        pane.search_next(true);
                    }
                    (KeyCode::Char('N'), false) => {
                        pane.search_next(false);
                    }
                    // The commands the shell marked: p the prompt above, P the one below, o a
                    // command's output, O the whole command with its prompt.
                    (KeyCode::Char(c @ ('p' | 'P' | 'o' | 'O')), false) => {
                        if !pane.has_prompts() {
                            self.toast(id, NO_MARKS);
                        } else if !match c {
                            'p' | 'P' => pane.jump_to_prompt(c == 'p'),
                            _ => pane.select_command(c == 'O'),
                        } {
                            let what = if c == 'p' || c == 'P' { "no prompt that way" } else { "no command here" };
                            self.toast(id, what);
                        }
                    }
                    (KeyCode::Esc, _) if pane.has_selection() => pane.clear_selection(),
                    (KeyCode::Esc, _) | (KeyCode::Char('q'), false) | (KeyCode::Char('c'), true) => {
                        pane.end_search();
                        pane.copy_mode(false);
                        return Mode::Normal;
                    }
                    _ => {}
                }
                Mode::Scroll
            }
            InputEvent::Mouse(m) => self.on_mouse(id, m, true).unwrap_or(Mode::Scroll),
            _ => Mode::Scroll,
        }
    }

    /// Scrolling a preview or file manager pane (no cursor there).
    pub(super) fn scroll_other(&mut self, _id: ClientId, p: PaneId, k: &Key) -> Mode {
        let Some(pane) = self.panes.get_mut(&p) else { return Mode::Normal };
        let half = (pane.rows / 2).max(1) as i32;
        match (k.code, k.mods.ctrl) {
            (KeyCode::Up, _) | (KeyCode::Char('k'), false) => pane.scroll(1),
            (KeyCode::Down, _) | (KeyCode::Char('j'), false) => pane.scroll(-1),
            (KeyCode::PageUp, _) | (KeyCode::Char('b'), true) => pane.scroll_page(true),
            (KeyCode::PageDown, _) | (KeyCode::Char(' '), false) | (KeyCode::Char('f'), true) => {
                pane.scroll_page(false)
            }
            (KeyCode::Char('u'), true) => pane.scroll(half),
            (KeyCode::Char('d'), true) => pane.scroll(-half),
            (KeyCode::Char('g'), false) | (KeyCode::Home, _) => pane.scroll_to_top(),
            (KeyCode::Char('G'), false) | (KeyCode::End, _) => pane.scroll_to_bottom(),
            (KeyCode::Esc, _) | (KeyCode::Char('q'), false) => return Mode::Normal,
            _ => {}
        }
        Mode::Scroll
    }

    pub(super) fn on_search(&mut self, id: ClientId, mut query: String, mut found: bool, ev: InputEvent) -> Mode {
        let Some(p) = self.client_focus(id) else { return Mode::Normal };
        let Some(pane) = self.panes.get_mut(&p) else { return Mode::Normal };
        match ev {
            InputEvent::Key(k) => match (k.code, k.mods.ctrl) {
                (KeyCode::Esc, _) | (KeyCode::Char('c'), true) | (KeyCode::Char('g'), true) => {
                    pane.end_search();
                    pane.clear_selection();
                    return Mode::Scroll;
                }
                (KeyCode::Enter, _) => return Mode::Scroll,
                (KeyCode::Char('n'), true) | (KeyCode::Up, _) => found = pane.search_next(true),
                (KeyCode::Char('p'), true) | (KeyCode::Down, _) => found = pane.search_next(false),
                (KeyCode::Backspace, _) => {
                    query.pop();
                    found = pane.search(&query);
                }
                (KeyCode::Char('u'), true) => {
                    query.clear();
                    found = pane.search(&query);
                }
                (KeyCode::Char(ch), false) if !k.mods.alt => {
                    query.push(ch);
                    found = pane.search(&query);
                }
                _ => {}
            },
            InputEvent::Paste(data) => {
                query.push_str(String::from_utf8_lossy(&data).lines().next().unwrap_or(""));
                found = pane.search(&query);
            }
            InputEvent::Mouse(m) => {
                if let Some(next) = self.on_mouse(id, m, true) {
                    return next;
                }
            }
            _ => {}
        }
        Mode::Search { query, found }
    }
}