omnyssh 1.0.3

TUI SSH dashboard & server manager
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

// Suppress dead_code warnings for stub/future-stage modules.
#![allow(dead_code)]

use clap::Parser;

mod app;
mod cli;
mod config;
mod event;
mod ssh;
mod ui;
mod update;
mod utils;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let cli = cli::Cli::parse();

    // Initialise logging. Output goes to a log file instead of stderr to prevent
    // log messages from bleeding through the TUI interface.
    // Set RUST_LOG or pass -v to enable debug output.
    let log_level = if cli.verbose { "debug" } else { "warn" };

    // Create log directory if it doesn't exist
    let log_dir = dirs::config_dir()
        .map(|d| d.join("omnyssh"))
        .unwrap_or_else(|| std::path::PathBuf::from("."));

    if let Err(e) = std::fs::create_dir_all(&log_dir) {
        eprintln!("Warning: Failed to create log directory: {}", e);
    }

    // Use daily rolling file appender
    // IMPORTANT: _guard must live for the entire duration of the program.
    // If it's dropped, logs will stop being written to the file.
    let file_appender = tracing_appender::rolling::daily(&log_dir, "omnyssh.log");
    let (non_blocking, _guard) = tracing_appender::non_blocking(file_appender);

    tracing_subscriber::fmt()
        .with_env_filter(
            tracing_subscriber::EnvFilter::try_from_default_env()
                .unwrap_or_else(|_| tracing_subscriber::EnvFilter::new(log_level)),
        )
        .with_writer(non_blocking)
        .init();

    // Restore the terminal if a panic occurs so the user is not left with a
    // broken shell.
    let default_hook = std::panic::take_hook();
    std::panic::set_hook(Box::new(move |info| {
        // Best-effort terminal restore — ignore errors here.
        let _ = crossterm::terminal::disable_raw_mode();
        let _ = crossterm::execute!(
            std::io::stdout(),
            crossterm::terminal::LeaveAlternateScreen,
            utils::mouse::DisableMinimalMouseCapture,
        );
        default_hook(info);
    }));

    // Load application config.
    // A missing file is silently ignored; a malformed file is reported and
    // we fall back to defaults so the app always starts.
    let mut app_config = match config::app_config::load_app_config(cli.config.as_deref()) {
        Ok(cfg) => {
            if let Some(ref path) = cli.config {
                tracing::info!("Loaded config from: {}", path.display());
            }
            cfg
        }
        Err(e) => {
            tracing::warn!("Config load error (using defaults): {}", e);
            config::app_config::AppConfig::default()
        }
    };

    // Apply CLI theme override if provided and save it to config.
    if let Some(ref theme) = cli.theme {
        if config::app_config::UiConfig::is_valid_theme(theme) {
            tracing::info!("Applying theme '{}' and saving to config", theme);
            app_config.ui.theme = theme.clone();

            // Save theme to config file for persistence
            if let Err(e) = config::app_config::save_theme_to_config(theme) {
                eprintln!("Warning: Failed to save theme to config: {}", e);
                eprintln!("Theme will be applied for this session only.");
            } else {
                eprintln!("✓ Theme '{}' saved to config", theme);
            }
        } else {
            eprintln!(
                "Error: Unknown theme '{}'. Available themes: {}",
                theme,
                config::app_config::UiConfig::available_themes().join(", ")
            );
            eprintln!(
                "Falling back to theme from config: '{}'",
                app_config.ui.theme
            );
        }
    }

    let mut app = app::App::new(app_config);
    let result = app.run().await;

    // Keep the guard alive until the very end to ensure all logs are flushed
    drop(_guard);

    result
}