managarr/src/ui/sonarr_ui/mod.rs

use std::{cmp, iter};

use chrono::{Duration, Utc};
use library::LibraryUi;
use ratatui::{
  layout::{Constraint, Layout, Rect},
  style::Stylize,
  text::Text,
  widgets::Paragraph,
  Frame,
};

use crate::{
  app::App,
  logos::SONARR_LOGO,
  models::{
    servarr_data::sonarr::sonarr_data::SonarrData,
    servarr_models::{DiskSpace, RootFolder},
    sonarr_models::DownloadRecord,
    Route,
  },
  utils::convert_to_gb,
};

use super::{
  draw_tabs,
  styles::ManagarrStyle,
  utils::{
    borderless_block, layout_block, line_gauge_with_label, line_gauge_with_title, title_block,
  },
  widgets::loading_block::LoadingBlock,
  DrawUi,
};

mod library;

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

pub(super) struct SonarrUi;

impl DrawUi for SonarrUi {
  fn accepts(route: Route) -> bool {
    matches!(route, Route::Sonarr(_, _))
  }

  fn draw(f: &mut Frame<'_>, app: &mut App<'_>, area: Rect) {
    let content_area = draw_tabs(f, area, "Series", &app.data.sonarr_data.main_tabs);
    let route = app.get_current_route();

    match route {
      _ if LibraryUi::accepts(route) => LibraryUi::draw(f, app, content_area),
      _ => (),
    }
  }

  fn draw_context_row(f: &mut Frame<'_>, app: &App<'_>, area: Rect) {
    let [main_area, logo_area] =
      Layout::horizontal([Constraint::Fill(0), Constraint::Length(20)]).areas(area);

    let [stats_area, downloads_area] =
      Layout::horizontal([Constraint::Ratio(1, 2), Constraint::Ratio(1, 2)]).areas(main_area);

    draw_stats_context(f, app, stats_area);
    draw_downloads_context(f, app, downloads_area);
    draw_sonarr_logo(f, logo_area);
  }
}

fn draw_stats_context(f: &mut Frame<'_>, app: &App<'_>, area: Rect) {
  let block = title_block("Stats");

  if !app.data.sonarr_data.version.is_empty() {
    f.render_widget(block, area);
    let SonarrData {
      root_folders,
      disk_space_vec,
      start_time,
      ..
    } = &app.data.sonarr_data;

    let mut constraints = vec![
      Constraint::Length(1),
      Constraint::Length(1),
      Constraint::Length(1),
    ];

    constraints.append(
      &mut iter::repeat(Constraint::Length(1))
        .take(disk_space_vec.len() + root_folders.items.len() + 1)
        .collect(),
    );

    let stat_item_areas = Layout::vertical(constraints).margin(1).split(area);

    let version_paragraph = Paragraph::new(Text::from(format!(
      "Sonarr Version:  {}",
      app.data.sonarr_data.version
    )))
    .block(borderless_block())
    .bold();

    let uptime = Utc::now() - start_time.to_owned();
    let days = uptime.num_days();
    let day_difference = uptime - Duration::days(days);
    let hours = day_difference.num_hours();
    let hour_difference = day_difference - Duration::hours(hours);
    let minutes = hour_difference.num_minutes();
    let seconds = (hour_difference - Duration::minutes(minutes)).num_seconds();

    let uptime_paragraph = Paragraph::new(Text::from(format!(
      "Uptime: {days}d {hours:0width$}:{minutes:0width$}:{seconds:0width$}",
      width = 2
    )))
    .block(borderless_block())
    .bold();

    let storage = Paragraph::new(Text::from("Storage:")).block(borderless_block().bold());
    let folders = Paragraph::new(Text::from("Root Folders:")).block(borderless_block().bold());

    f.render_widget(version_paragraph, stat_item_areas[0]);
    f.render_widget(uptime_paragraph, stat_item_areas[1]);
    f.render_widget(storage, stat_item_areas[2]);

    for i in 0..disk_space_vec.len() {
      let DiskSpace {
        free_space,
        total_space,
      } = &disk_space_vec[i];
      let title = format!("Disk {}", i + 1);
      let ratio = if *total_space == 0 {
        0f64
      } else {
        1f64 - (*free_space as f64 / *total_space as f64)
      };

      let space_gauge = line_gauge_with_label(title.as_str(), ratio);

      f.render_widget(space_gauge, stat_item_areas[i + 3]);
    }

    f.render_widget(folders, stat_item_areas[disk_space_vec.len() + 3]);

    for i in 0..root_folders.items.len() {
      let RootFolder {
        path, free_space, ..
      } = &root_folders.items[i];
      let space: f64 = convert_to_gb(*free_space);
      let root_folder_space = Paragraph::new(format!("{path}: {space:.2} GB free"))
        .block(borderless_block())
        .default();

      f.render_widget(
        root_folder_space,
        stat_item_areas[i + disk_space_vec.len() + 4],
      )
    }
  } else {
    f.render_widget(LoadingBlock::new(app.is_loading, block), area);
  }
}

fn draw_downloads_context(f: &mut Frame<'_>, app: &App<'_>, area: Rect) {
  let block = title_block("Downloads");
  let downloads_vec = &app.data.sonarr_data.downloads.items;

  if !downloads_vec.is_empty() {
    f.render_widget(block, area);

    let max_items = ((((area.height as f64 / 2.0).floor() * 2.0) as i64) / 2) - 1;
    let items = cmp::min(downloads_vec.len(), max_items.unsigned_abs() as usize);
    let download_item_areas = Layout::vertical(
      iter::repeat(Constraint::Length(2))
        .take(items)
        .collect::<Vec<Constraint>>(),
    )
    .margin(1)
    .split(area);

    for i in 0..items {
      let DownloadRecord {
        title,
        sizeleft,
        size,
        ..
      } = &downloads_vec[i];
      let percent = if *size == 0.0 {
        0.0
      } else {
        1f64 - (*sizeleft / *size)
      };
      let download_gauge = line_gauge_with_title(title, percent);

      f.render_widget(download_gauge, download_item_areas[i]);
    }
  } else {
    f.render_widget(LoadingBlock::new(app.is_loading, block), area);
  }
}

fn draw_sonarr_logo(f: &mut Frame<'_>, area: Rect) {
  let logo_text = Text::from(SONARR_LOGO);
  let logo = Paragraph::new(logo_text)
    .light_cyan()
    .block(layout_block().default())
    .centered();
  f.render_widget(logo, area);
}