purple_ssh/
handler.rs

1use std::sync::atomic::Ordering;
2use std::sync::mpsc;
3
4use anyhow::Result;
5use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
6
7use crate::app::{App, Screen};
8use crate::event::AppEvent;
9
10mod bulk_tag_editor;
11mod confirm;
12mod container_exec_prompt;
13pub(crate) mod container_host_picker;
14pub(crate) mod container_logs;
15mod containers;
16pub(crate) mod containers_overview;
17mod ctx;
18pub(crate) mod event_loop;
19mod file_browser;
20mod help;
21mod host_detail;
22mod host_form;
23mod host_list;
24mod jump;
25pub(crate) mod key_push_picker;
26mod keys_overview;
27mod picker;
28mod ping;
29mod provider;
30mod snippet;
31pub(crate) mod snippet_host_picker;
32mod snippets_overview;
33mod sync;
34mod tag_picker;
35mod theme_picker;
36mod tunnel;
37pub(crate) mod tunnel_host_picker;
38mod tunnels_overview;
39mod welcome;
40mod whats_new;
41
42pub use confirm::{ConfirmAction, route_confirm_key};
43pub(crate) use provider::zone_data_for;
44pub use sync::spawn_provider_sync;
45
46/// Handle a key event based on the current screen.
47pub fn handle_key_event(
48    app: &mut App,
49    key: KeyEvent,
50    events_tx: &mpsc::Sender<AppEvent>,
51) -> Result<()> {
52    // Global Ctrl+C handler — screen-conditional for SnippetOutput
53    if key.modifiers.contains(KeyModifiers::CONTROL) && key.code == KeyCode::Char('c') {
54        if matches!(app.screen, Screen::SnippetOutput) {
55            if let Some(state) = app.snippets.output() {
56                if !state.all_done {
57                    if state.cancel.load(Ordering::Relaxed) {
58                        // Second Ctrl+C: cancel already pending, force close
59                    } else {
60                        // First Ctrl+C: request cancellation
61                        state.cancel.store(true, Ordering::Relaxed);
62                        return Ok(());
63                    }
64                }
65            }
66            app.snippets.set_output(None);
67            app.set_screen(Screen::HostList);
68            return Ok(());
69        }
70        if let Some(cancel) = app.vault.signing_cancel() {
71            cancel.store(true, std::sync::atomic::Ordering::Relaxed);
72        }
73        app.running = false;
74        return Ok(());
75    }
76
77    // Jump intercept
78    if app.jump.is_some() {
79        jump::handle_key(app, key, events_tx);
80        return Ok(());
81    }
82
83    match &app.screen {
84        Screen::HostList => host_list::handle_key(app, key, events_tx),
85        Screen::AddHost | Screen::EditHost { .. } => host_form::handle_key(app, key),
86        Screen::ConfirmDelete { .. } => confirm::handle_delete_key(app, key),
87        Screen::Help { .. } => help::handle_key(app, key),
88        Screen::KeyList => help::handle_key_list_key(app, key),
89        Screen::KeyDetail { .. } => help::handle_key_detail_key(app, key),
90        Screen::KeyPushPicker { .. } => key_push_picker::handle_key(app, key),
91        Screen::ConfirmKeyPush { .. } => confirm::handle_key_push_key(app, key, events_tx),
92        Screen::HostDetail { .. } => host_detail::handle_key(app, key),
93        Screen::TagPicker => tag_picker::handle_key(app, key),
94        Screen::BulkTagEditor => bulk_tag_editor::handle_key(app, key),
95        Screen::ThemePicker => theme_picker::handle_key(app, key),
96        Screen::Providers => provider::handle_provider_list_key(app, key, events_tx),
97        Screen::ProviderForm { .. } => provider::handle_provider_form_key(app, key, events_tx),
98        Screen::ProviderLabelMigration { .. } => {
99            provider::handle_label_migration_key(app, key, events_tx)
100        }
101        Screen::TunnelList { .. } => tunnel::handle_tunnel_list_key(app, key),
102        Screen::TunnelForm { .. } => tunnel::handle_tunnel_form_key(app, key),
103        Screen::TunnelHostPicker => tunnel_host_picker::handle_key(app, key),
104        Screen::ContainerHostPicker => container_host_picker::handle_key(app, key, events_tx),
105        Screen::SnippetPicker => snippet::handle_picker_key(app, key, events_tx),
106        Screen::SnippetForm => snippet::handle_form_key(app, key),
107        Screen::SnippetOutput => snippet::handle_output_key(app, key),
108        Screen::SnippetParamForm => snippet::handle_param_form_key(app, key, events_tx),
109        Screen::SnippetHostPicker => snippet_host_picker::handle_key(app, key),
110        Screen::ConfirmRunSnippet => confirm::handle_run_snippet_confirm_key(app, key, events_tx),
111        Screen::ConfirmHostKeyReset { .. } => confirm::handle_host_key_reset_key(app, key),
112        Screen::ConfirmVaultSign => confirm::handle_vault_sign_key(app, key, events_tx),
113        Screen::ConfirmImport { .. } => confirm::handle_import_key(app, key),
114        Screen::ConfirmPurgeStale => confirm::handle_purge_stale_key(app, key),
115        Screen::FileBrowser { .. } => file_browser::handle_key(app, key, events_tx),
116        Screen::Containers { .. } => containers::handle_key(app, key, events_tx)?,
117        Screen::ContainerLogs => container_logs::handle_key(app, key, events_tx),
118        Screen::ConfirmContainerRestart { .. } => confirm::handle_container_restart_key(app, key),
119        Screen::ConfirmContainerStop { .. } => confirm::handle_container_stop_key(app, key),
120        Screen::ContainerExecPrompt { .. } => container_exec_prompt::handle_key(app, key),
121        Screen::ConfirmStackRestart => confirm::handle_stack_restart_key(app, key),
122        Screen::ConfirmHostRestartAll => confirm::handle_host_restart_all_key(app, key),
123        Screen::ConfirmHostStopAll => confirm::handle_host_stop_all_key(app, key),
124        Screen::Welcome { .. } => welcome::handle_key(app, key),
125        Screen::WhatsNew(_) => whats_new::handle_key(app, key),
126    }
127    Ok(())
128}
129
130#[cfg(test)]
131mod tests;
purple_ssh/handler.rs

purple_ssh/
handler.rs
