managarr/src/handlers/mod.rs

use radarr_handlers::RadarrHandler;
use sonarr_handlers::SonarrHandler;

use crate::app::key_binding::DEFAULT_KEYBINDINGS;
use crate::app::App;
use crate::event::Key;
use crate::models::{HorizontallyScrollableText, Route};

mod radarr_handlers;
mod sonarr_handlers;

#[cfg(test)]
#[path = "handlers_tests.rs"]
mod handlers_tests;

#[cfg(test)]
#[path = "handler_test_utils.rs"]
pub mod handler_test_utils;
mod table_handler;

pub trait KeyEventHandler<'a, 'b, T: Into<Route> + Copy> {
  fn handle_key_event(&mut self) {
    let key = self.get_key();
    match key {
      _ if key == DEFAULT_KEYBINDINGS.up.key => {
        if self.is_ready() {
          self.handle_scroll_up();
        }
      }
      _ if key == DEFAULT_KEYBINDINGS.down.key => {
        if self.is_ready() {
          self.handle_scroll_down();
        }
      }
      _ if key == DEFAULT_KEYBINDINGS.home.key => {
        if self.is_ready() {
          self.handle_home();
        }
      }
      _ if key == DEFAULT_KEYBINDINGS.end.key => {
        if self.is_ready() {
          self.handle_end();
        }
      }
      _ if key == DEFAULT_KEYBINDINGS.delete.key => {
        if self.is_ready() {
          self.handle_delete();
        }
      }
      _ if key == DEFAULT_KEYBINDINGS.left.key || key == DEFAULT_KEYBINDINGS.right.key => {
        self.handle_left_right_action()
      }
      _ if key == DEFAULT_KEYBINDINGS.submit.key => {
        if self.is_ready() {
          self.handle_submit();
        }
      }
      _ if key == DEFAULT_KEYBINDINGS.esc.key => self.handle_esc(),
      _ => {
        if self.is_ready() {
          self.handle_char_key_event();
        }
      }
    }
  }

  fn handle(&mut self) {
    self.handle_key_event();
  }

  fn accepts(active_block: T) -> bool;
  fn with(key: Key, app: &'a mut App<'b>, active_block: T, context: Option<T>) -> Self;
  fn get_key(&self) -> Key;
  fn is_ready(&self) -> bool;
  fn handle_scroll_up(&mut self);
  fn handle_scroll_down(&mut self);
  fn handle_home(&mut self);
  fn handle_end(&mut self);
  fn handle_delete(&mut self);
  fn handle_left_right_action(&mut self);
  fn handle_submit(&mut self);
  fn handle_esc(&mut self);
  fn handle_char_key_event(&mut self);
}

pub fn handle_events(key: Key, app: &mut App<'_>) {
  if key == DEFAULT_KEYBINDINGS.next_servarr.key {
    app.reset();
    app.server_tabs.next();
    app.pop_and_push_navigation_stack(app.server_tabs.get_active_route());
  } else if key == DEFAULT_KEYBINDINGS.previous_servarr.key {
    app.reset();
    app.server_tabs.previous();
    app.pop_and_push_navigation_stack(app.server_tabs.get_active_route());
  } else {
    match app.get_current_route() {
      Route::Radarr(active_radarr_block, context) => {
        RadarrHandler::with(key, app, active_radarr_block, context).handle()
      }
      Route::Sonarr(active_sonarr_block, context) => {
        SonarrHandler::with(key, app, active_sonarr_block, context).handle()
      }
      _ => (),
    }
  }
}

fn handle_clear_errors(app: &mut App<'_>) {
  if !app.error.text.is_empty() {
    app.error = HorizontallyScrollableText::default();
  }
}

fn handle_prompt_toggle(app: &mut App<'_>, key: Key) {
  match key {
    _ if key == DEFAULT_KEYBINDINGS.left.key || key == DEFAULT_KEYBINDINGS.right.key => {
      match app.get_current_route() {
        Route::Radarr(_, _) => {
          app.data.radarr_data.prompt_confirm = !app.data.radarr_data.prompt_confirm
        }
        Route::Sonarr(_, _) => {
          app.data.sonarr_data.prompt_confirm = !app.data.sonarr_data.prompt_confirm
        }
        _ => (),
      }
    }
    _ => (),
  }
}

#[macro_export]
macro_rules! handle_text_box_left_right_keys {
  ($self:expr, $key:expr, $input:expr) => {
    match $self.key {
      _ if $key == DEFAULT_KEYBINDINGS.left.key => {
        $input.scroll_left();
      }
      _ if $key == DEFAULT_KEYBINDINGS.right.key => {
        $input.scroll_right();
      }
      _ => (),
    }
  };
}

#[macro_export]
macro_rules! handle_text_box_keys {
  ($self:expr, $key:expr, $input:expr) => {
    match $self.key {
      _ if $key == DEFAULT_KEYBINDINGS.backspace.key => {
        $input.pop();
      }
      Key::Char(character) => {
        $input.push(character);
      }
      _ => (),
    }
  };
}

#[macro_export]
macro_rules! handle_prompt_left_right_keys {
  ($self:expr, $confirm_prompt:expr, $data:ident) => {
    if $self.app.data.$data.selected_block.get_active_block() == $confirm_prompt {
      handle_prompt_toggle($self.app, $self.key);
    } else if $self.key == DEFAULT_KEYBINDINGS.left.key {
      $self.app.data.$data.selected_block.left();
    } else {
      $self.app.data.$data.selected_block.right();
    }
  };
}