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

purple_ssh/
handler.rs
