susshi 0.25.0

Terminal SSH manager with YAML inventories, multi-hop jump hosts, Wallix bastion, tunnels, SCP, and Catppuccin TUI
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

use super::*;

impl App {
    /// Lance une commande SSH non-interactive dans un thread dedie, selon le
    /// mode de connexion courant (Direct, Jump ou Wallix).
    /// Stocke le resultat via `cmd_rx`.
    pub fn start_cmd(&mut self, server: &ResolvedServer, cmd: String) {
        if self.cmd_history.last().map(|s| s.as_str()) != Some(cmd.as_str()) {
            self.cmd_history.push(cmd.clone());
        }
        self.cmd_history_cursor = None;

        let server = server.clone();
        let mode = self.connection_mode;

        let (tx, rx) = mpsc::channel();
        self.cmd_state = CmdState::Running(cmd.clone());
        self.cmd_rx = Some(rx);

        std::thread::spawn(move || {
            let result = match crate::ssh::exec::run_remote(
                &server,
                mode,
                &cmd,
                crate::ssh::exec::COMMAND_TIMEOUT,
            ) {
                Ok(out) => (format_cmd_output(&out), out.success),
                Err(e) => (e.to_string(), false),
            };
            let _ = tx.send(result);
        });
    }

    /// Verifie si le thread de commande a produit un resultat et met a jour `cmd_state`.
    pub fn poll_cmd(&mut self) {
        let done = if let Some(rx) = &self.cmd_rx {
            rx.try_recv().ok()
        } else {
            None
        };
        if let Some((output, exit_ok)) = done {
            let cmd = match &self.cmd_state {
                CmdState::Running(c) => c.clone(),
                _ => String::new(),
            };
            self.cmd_state = CmdState::Done {
                cmd,
                output,
                exit_ok,
            };
            self.cmd_rx = None;
        }
    }

    /// Reinitialise l'etat de la commande ad-hoc.
    pub fn reset_cmd(&mut self) {
        self.cmd_state = CmdState::Idle;
        self.cmd_rx = None;
        self.cmd_history_cursor = None;
    }
}

/// Fusionne stdout et stderr pour l'affichage, separes par `---` si les deux
/// sont presents.
pub(crate) fn format_cmd_output(out: &crate::ssh::exec::ExecOutput) -> String {
    if out.stderr.is_empty() {
        out.stdout.clone()
    } else if out.stdout.is_empty() {
        out.stderr.clone()
    } else {
        format!("{}\n---\n{}", out.stdout, out.stderr)
    }
}