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();
                            }
                        }
                        _ => {}
                    }
                }
            }
        }
    }
}