655585d9ef
- Box large enum variants in EventData to reduce memory footprint - Add Default trait implementations for types with new() methods - Replace or_insert_with(Vec::new) with or_default() - Convert vec init+push patterns to vec! macro - Fix field reassignment with default initialization - Convert match to if for simple equality checks - Remove unused Backend type parameters from TUI draw functions - Apply rustfmt formatting All tests passing (24 total). Zero clippy warnings. Ready for CI/CD.
121 lines
3.7 KiB
Rust
121 lines
3.7 KiB
Rust
use anyhow::Result;
|
|
use crossterm::{
|
|
event::{self, DisableMouseCapture, EnableMouseCapture, Event, KeyCode, KeyEventKind},
|
|
execute,
|
|
terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
|
|
};
|
|
use ratatui::{
|
|
backend::{Backend, CrosstermBackend},
|
|
Terminal,
|
|
};
|
|
use std::{io, sync::Arc, time::Duration};
|
|
use tokio::{sync::Mutex, time};
|
|
|
|
mod app;
|
|
mod events;
|
|
mod ui;
|
|
|
|
use app::App;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<()> {
|
|
// Setup terminal
|
|
enable_raw_mode()?;
|
|
let mut stdout = io::stdout();
|
|
execute!(stdout, EnterAlternateScreen, EnableMouseCapture)?;
|
|
let backend = CrosstermBackend::new(stdout);
|
|
let mut terminal = Terminal::new(backend)?;
|
|
|
|
// Create app state
|
|
let app = Arc::new(Mutex::new(App::new().await?));
|
|
|
|
// Clone for background task
|
|
let app_clone = Arc::clone(&app);
|
|
|
|
// Start background scanning task
|
|
tokio::spawn(async move {
|
|
let mut interval = time::interval(Duration::from_secs(2));
|
|
loop {
|
|
interval.tick().await;
|
|
if let Ok(mut app) = app_clone.try_lock() {
|
|
if let Err(e) = app.update_scan_data().await {
|
|
app.add_log_message(format!("Scan error: {}", e));
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
// Main event loop
|
|
let res = run_app(&mut terminal, app).await;
|
|
|
|
// Restore terminal
|
|
disable_raw_mode()?;
|
|
execute!(
|
|
terminal.backend_mut(),
|
|
LeaveAlternateScreen,
|
|
DisableMouseCapture
|
|
)?;
|
|
terminal.show_cursor()?;
|
|
|
|
if let Err(err) = res {
|
|
println!("{:?}", err);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn run_app<B: Backend>(terminal: &mut Terminal<B>, app: Arc<Mutex<App>>) -> Result<()> {
|
|
loop {
|
|
// Draw the UI
|
|
terminal.draw(|f| {
|
|
if let Ok(app) = app.try_lock() {
|
|
ui::draw(f, &app);
|
|
}
|
|
})?;
|
|
|
|
// Handle events
|
|
if event::poll(Duration::from_millis(100))? {
|
|
if let Event::Key(key) = event::read()? {
|
|
if key.kind == KeyEventKind::Press {
|
|
match key.code {
|
|
KeyCode::Char('q') => return Ok(()),
|
|
KeyCode::Tab => {
|
|
if let Ok(mut app) = app.try_lock() {
|
|
app.next_tab();
|
|
}
|
|
}
|
|
KeyCode::Up => {
|
|
if let Ok(mut app) = app.try_lock() {
|
|
app.scroll_up();
|
|
}
|
|
}
|
|
KeyCode::Down => {
|
|
if let Ok(mut app) = app.try_lock() {
|
|
app.scroll_down();
|
|
}
|
|
}
|
|
KeyCode::Enter => {
|
|
if let Ok(mut app) = app.try_lock() {
|
|
app.select_item();
|
|
}
|
|
}
|
|
KeyCode::Char('r') => {
|
|
if let Ok(mut app) = app.try_lock() {
|
|
if let Err(e) = app.force_refresh().await {
|
|
app.add_log_message(format!("Refresh error: {}", e));
|
|
}
|
|
}
|
|
}
|
|
KeyCode::Char('c') => {
|
|
if let Ok(mut app) = app.try_lock() {
|
|
app.clear_detections();
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|