pinakes/crates/pinakes-tui/src/event.rs

use std::time::Duration;

use crossterm::event::{self, Event as CrosstermEvent, KeyEvent};
use tokio::sync::mpsc;

#[derive(Debug)]
pub enum AppEvent {
  Key(KeyEvent),
  Tick,
  ApiResult(ApiResult),
}

#[derive(Debug)]
pub enum ApiResult {
  MediaList(Vec<crate::client::MediaResponse>),
  SearchResults(crate::client::SearchResponse),
  AllTags(Vec<crate::client::TagResponse>),
  Collections(Vec<crate::client::CollectionResponse>),
  ImportDone(crate::client::ImportResponse),
  ScanDone(Vec<crate::client::ScanResponse>),
  AuditLog(Vec<crate::client::AuditEntryResponse>),
  Duplicates(Vec<crate::client::DuplicateGroupResponse>),
  DatabaseStats(crate::client::DatabaseStatsResponse),
  Statistics(crate::client::LibraryStatisticsResponse),
  ScheduledTasks(Vec<crate::client::ScheduledTaskResponse>),
  BooksList(Vec<crate::client::MediaResponse>),
  BookSeries(Vec<crate::client::SeriesSummary>),
  BookAuthors(Vec<crate::client::AuthorSummary>),
  MediaUpdated,
  ReadingProgressUpdated,
  // Playlists
  PlaylistsLoaded(Vec<crate::client::PlaylistResponse>),
  PlaylistItemsLoaded(Vec<crate::client::MediaResponse>),
  // Social
  CommentsLoaded(Vec<crate::client::CommentResponse>),
  RatingSet(u8),
  FavoriteToggled,
  // Subtitles
  SubtitlesLoaded(crate::client::SubtitleListResponse),
  // Enrichment / transcode
  EnrichmentTriggered,
  TranscodeStarted,
  // Admin
  UsersLoaded(Vec<crate::client::UserResponse>),
  SyncDevicesLoaded(Vec<crate::client::DeviceResponse>),
  WebhooksLoaded(Vec<crate::client::WebhookInfo>),
  TranscodesLoaded(Vec<crate::client::TranscodeSessionResponse>),
  Error(String),
}

pub struct EventHandler {
  tx: mpsc::UnboundedSender<AppEvent>,
  rx: mpsc::UnboundedReceiver<AppEvent>,
}

impl EventHandler {
  pub fn new(tick_rate: Duration) -> Self {
    let (tx, rx) = mpsc::unbounded_channel();
    let event_tx = tx.clone();

    std::thread::spawn(move || {
      loop {
        match event::poll(tick_rate) {
          Ok(true) => {
            if let Ok(CrosstermEvent::Key(key)) = event::read()
              && event_tx.send(AppEvent::Key(key)).is_err()
            {
              break;
            }
          },
          Ok(false) => {
            if event_tx.send(AppEvent::Tick).is_err() {
              break;
            }
          },
          Err(e) => {
            tracing::warn!(error = %e, "event poll failed");
          },
        }
      }
    });

    Self { tx, rx }
  }

  pub fn sender(&self) -> mpsc::UnboundedSender<AppEvent> {
    self.tx.clone()
  }

  pub async fn next(&mut self) -> Option<AppEvent> {
    self.rx.recv().await
  }
}