sshell/src/ui/app.rs

use crate::app::{App, Mode};
use crate::config::ConnectionType;
use anyhow::Result;
use crossterm::{
    cursor::{Hide, Show},
    event::{self, Event, KeyCode, KeyEvent, KeyEventKind, KeyModifiers},
    execute,
    terminal::{EnterAlternateScreen, LeaveAlternateScreen, disable_raw_mode, enable_raw_mode},
};
use ratatui::{Terminal, backend::CrosstermBackend};
use std::collections::HashSet;
use std::io;
use std::sync::atomic::{AtomicBool, Ordering};
use std::time::Duration;

use super::view::{
    ActionMenuView, CredFormView, CredListView, DeleteConfirmView, FormView, HomeListView,
    ImportView, QuickSelectView, SearchView, SettingsView, View,
};

static TERMINAL_RESTORED: AtomicBool = AtomicBool::new(false);

pub fn run() -> Result<()> {
    let _guard = TuiGuard::init()?;
    let backend = CrosstermBackend::new(io::stdout());
    let mut terminal = Terminal::new(backend)?;
    let mut app = App::load()?;

    spawn_latency_probes(&app);

    loop {
        terminal.draw(|frame| super::draw(frame, &mut app))?;
        if app.session.should_quit {
            break;
        }
        if event::poll(Duration::from_millis(200))?
            && let Event::Key(key) = event::read()?
            && key.kind == KeyEventKind::Press
        {
            handle_key(&mut app, key)?;
            spawn_latency_probes(&app);
        }
    }

    Ok(())
}

struct TuiGuard;

impl TuiGuard {
    fn init() -> Result<Self> {
        enable_raw_mode()?;
        execute!(io::stdout(), EnterAlternateScreen, Hide)?;
        Ok(Self)
    }
}

impl Drop for TuiGuard {
    fn drop(&mut self) {
        let _ = restore_terminal();
    }
}

pub fn restore_terminal() -> Result<()> {
    if TERMINAL_RESTORED.swap(true, Ordering::SeqCst) {
        return Ok(());
    }
    let _ = disable_raw_mode();
    execute!(io::stdout(), Show, LeaveAlternateScreen)?;
    Ok(())
}

fn handle_key(app: &mut App, key: KeyEvent) -> Result<()> {
    if key.modifiers.contains(KeyModifiers::CONTROL) && key.code == KeyCode::Char('c') {
        app.session.should_quit = true;
        return Ok(());
    }
    match app.session.mode {
        Mode::Home => HomeListView.handle_key(app, key),
        Mode::ActionMenu => ActionMenuView.handle_key(app, key),
        Mode::Search => SearchView.handle_key(app, key),
        Mode::QuickSelect => QuickSelectView.handle_key(app, key),
        Mode::DeleteConfirm => DeleteConfirmView.handle_key(app, key),
        Mode::Form => FormView.handle_key(app, key),
        Mode::ImportSelector => ImportView.handle_key(app, key),
        Mode::Credentials => CredListView.handle_key(app, key),
        Mode::CredForm => CredFormView.handle_key(app, key),
        Mode::Settings => SettingsView.handle_key(app, key),
    }
}

fn spawn_latency_probes(app: &App) {
    let cache = app.session.latency.lock().unwrap();
    let existing: HashSet<String> = cache.keys().cloned().collect();
    drop(cache);

    for (_, profile) in app.entries() {
        if let ConnectionType::Ssh { host, port, .. } = &profile.kind {
            let key = format!("{host}:{port}");
            if existing.contains(&key) {
                continue;
            }
            let cache_clone = app.session.latency.clone();
            let host = host.clone();
            let port = *port;
            std::thread::spawn(move || {
                let status = crate::app::latency::probe(&host, port);
                if let Ok(mut cache) = cache_clone.lock() {
                    cache.insert(key, status);
                }
            });
        }
    }
}