joecalsend/src/frontend/mod.rs

use std::{collections::BTreeMap, io, net::SocketAddr, sync::OnceLock, time::Duration};

use crossterm::event::{Event, EventStream, KeyCode, KeyEvent, KeyEventKind};
use futures::{FutureExt, StreamExt};
use joecalsend::{Config, JoecalState, Listeners, models::Device};
use ratatui::{
    DefaultTerminal,
    buffer::Buffer,
    layout::Rect,
    style::Stylize,
    symbols::border,
    text::{Line, Text},
    widgets::{Block, Paragraph, Widget},
};
use tokio::task::JoinSet;

pub mod ui;

pub type Peers = BTreeMap<SocketAddr, (String, String)>;

pub struct App {
    pub state: OnceLock<JoecalState>,
    pub screen: Vec<CurrentScreen>,
    pub events: EventStream,
    // addr -> (alias, fingerprint)
    pub peers: Peers,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum CurrentScreen {
    Main,
    Sending,
    Receiving,
    Stopping,
}

impl App {
    pub fn new() -> Self {
        App {
            state: Default::default(),
            screen: vec![CurrentScreen::Main],
            peers: Default::default(),
            events: Default::default(),
        }
    }

    #[tokio::main]
    pub async fn start_and_run(
        &mut self,
        terminal: &mut DefaultTerminal,
        config: Config,
        device: Device,
    ) -> io::Result<()> {
        let state = JoecalState::new(device)
            .await
            .expect("Could not create JoecalState");

        let mut handles = JoinSet::new();
        state.start(&config, &mut handles).await;
        self.state.get_or_init(|| state);
        loop {
            terminal.draw(|frame| self.draw(frame))?;
            self.handle_events().await?;

            if let Some(&top) = self.screen.last()
                && top == CurrentScreen::Stopping
            {
                self.state.get().unwrap().stop().await;
                break;
            }

            let peers = self.state.get().unwrap().peers.lock().await;
            self.peers.clear();
            peers.iter().for_each(|(fingerprint, (addr, device))| {
                let alias = device.alias.clone();
                self.peers
                    .insert(addr.to_owned(), (alias, fingerprint.to_owned()));
            });
        }

        shutdown(&mut handles).await;

        Ok(())
    }

    async fn handle_events(&mut self) -> io::Result<()> {
        tokio::select! {
            event = self.events.next().fuse() => {
                if let Some(Ok(evt)) = event {
                    match evt {
                        Event::Key(key)
                            if key.kind == KeyEventKind::Press
                                => self.handle_key_event(key),
                        Event::Mouse(_) => {}
                        Event::Resize(_, _) => {}
                        _ => {}
                    }
                }
            }
            _ = tokio::time::sleep(Duration::from_millis(200)) => {}
        }

        Ok(())
    }

    fn handle_key_event(&mut self, key_event: KeyEvent) {
        match key_event.code {
            KeyCode::Char('q') => self.exit(),
            KeyCode::Char('s') => self.send(),
            KeyCode::Char('r') => self.recv(),
            KeyCode::Esc => self.pop(),
            _ => {}
        }
    }

    fn exit(&mut self) {
        self.screen.push(CurrentScreen::Stopping);
    }

    fn send(&mut self) {
        let last = self.screen.last();
        match last {
            Some(CurrentScreen::Sending) => {}
            _ => self.screen.push(CurrentScreen::Sending),
        }
    }

    fn recv(&mut self) {
        let last = self.screen.last();
        match last {
            Some(CurrentScreen::Receiving) => {}
            _ => self.screen.push(CurrentScreen::Receiving),
        }
    }

    fn pop(&mut self) {
        self.screen.pop();
        if self.screen.last().is_none() {
            self.screen.push(CurrentScreen::Main);
        }
    }
}

impl Widget for &App {
    fn render(self, area: Rect, buf: &mut Buffer) {
        let title = Line::from(" Joecalsend ".bold());
        let instructions = Line::from(vec![
            " Send ".into(),
            "<S>".blue().bold(),
            " Receive ".into(),
            "<R>".blue().bold(),
            " Discover ".into(),
            "<D>".blue().bold(),
            " Quit ".into(),
            "<Q> ".blue().bold(),
        ]);
        let block = Block::bordered()
            .title(title.centered())
            .title_bottom(instructions.centered())
            .border_set(border::THICK);

        let current_screen = format!(
            "{:?}",
            self.screen.last().copied().unwrap_or(CurrentScreen::Main)
        );
        let text = Text::from(Line::from(current_screen.yellow()));

        Paragraph::new(text)
            .centered()
            .block(block)
            .render(area, buf);
    }
}

async fn shutdown(handles: &mut JoinSet<Listeners>) {
    let mut alarm = tokio::time::interval(tokio::time::Duration::from_secs(5));
    alarm.tick().await;
    loop {
        tokio::select! {
            join_result = handles.join_next() => {
                match join_result {
                    Some(handle) => match handle {
                        Ok(h) => println!("Stopped {h:?}"),
                        Err(e) => println!("Got error {e:?}"),
                    }
                    None => break,
                }
            }
            _ = alarm.tick() => {
                println!("Exit timeout reached, aborting all unjoined tasks");
                handles.abort_all();
                break;
            },
        }
    }
}