src/main.rs


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

use std::collections::HashMap;

use clap::Parser;

use sshr::{Commands, Config, History, KnownHosts, Session, SessionSource, Tmux};

fn main() -> anyhow::Result<()> {
    let mut config = Config::parse();

    tracing_subscriber::fmt::fmt()
        .with_max_level(&config.verbosity)
        .without_time()
        .init();

    config.history_file = config.history_file.or_else(History::default_path);

    match config.command.to_owned() {
        Commands::List => list_sessions(config),
        Commands::Switch { name } => switch(config, name),
    }
}

fn list_sessions(config: Config) -> anyhow::Result<()> {
    let mut sessions = HashMap::new();

    sessions = KnownHosts::default().update(sessions)?;
    sessions = Tmux::new(config.socket_name).update(sessions)?;

    if let Some(history) = config.history_file.map(History::new) {
        if history.exists() {
            sessions = history.update(sessions)?;
        }
    }

    let mut sessions: Vec<Session> = sessions.into_values().collect();

    sessions.sort();

    for session in sessions {
        println!("{session}");
    }

    Ok(())
}

fn switch(config: Config, name: String) -> anyhow::Result<()> {
    let tmux = Tmux::new(config.socket_name);

    if let Some(history) = config.history_file.map(History::new) {
        if tmux.switch(&name)?.success() {
            let session = tmux.show(&name)?;
            history.update_session(session)?;
        }
    }

    Ok(())
}