615 lines
19 KiB
Rust
615 lines
19 KiB
Rust
use crate::app::App;
|
|
use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, SEASON_DETAILS_BLOCKS};
|
|
use crate::models::sonarr_models::{
|
|
DownloadRecord, DownloadStatus, Episode, SonarrHistoryEventType, SonarrHistoryItem, SonarrRelease,
|
|
};
|
|
use crate::models::Route;
|
|
use crate::ui::sonarr_ui::library::episode_details_ui::EpisodeDetailsUi;
|
|
use crate::ui::sonarr_ui::sonarr_ui_utils::{
|
|
create_download_failed_history_event_details,
|
|
create_download_folder_imported_history_event_details,
|
|
create_episode_file_deleted_history_event_details,
|
|
create_episode_file_renamed_history_event_details, create_grabbed_history_event_details,
|
|
create_no_data_history_event_details,
|
|
};
|
|
use crate::ui::styles::ManagarrStyle;
|
|
use crate::ui::utils::{
|
|
borderless_block, decorate_peer_style, get_width_from_percentage, layout_block_top_border,
|
|
};
|
|
use crate::ui::widgets::confirmation_prompt::ConfirmationPrompt;
|
|
use crate::ui::widgets::loading_block::LoadingBlock;
|
|
use crate::ui::widgets::managarr_table::ManagarrTable;
|
|
use crate::ui::widgets::message::Message;
|
|
use crate::ui::widgets::popup::{Popup, Size};
|
|
use crate::ui::{draw_popup, draw_tabs, DrawUi};
|
|
use crate::utils::convert_to_gb;
|
|
use chrono::Utc;
|
|
use ratatui::layout::{Alignment, Constraint, Rect};
|
|
use ratatui::prelude::{Line, Style, Stylize, Text};
|
|
use ratatui::widgets::{Cell, Paragraph, Row, Wrap};
|
|
use ratatui::Frame;
|
|
use serde_json::Number;
|
|
|
|
#[cfg(test)]
|
|
#[path = "season_details_ui_tests.rs"]
|
|
mod season_details_ui_tests;
|
|
|
|
pub(super) struct SeasonDetailsUi;
|
|
|
|
impl DrawUi for SeasonDetailsUi {
|
|
fn accepts(route: Route) -> bool {
|
|
if let Route::Sonarr(active_sonarr_block, _) = route {
|
|
return EpisodeDetailsUi::accepts(route)
|
|
|| SEASON_DETAILS_BLOCKS.contains(&active_sonarr_block);
|
|
}
|
|
|
|
false
|
|
}
|
|
|
|
fn draw(f: &mut Frame<'_>, app: &mut App<'_>, _area: Rect) {
|
|
let route = app.get_current_route();
|
|
if app.data.sonarr_data.season_details_modal.is_some() {
|
|
if let Route::Sonarr(active_sonarr_block, _) = app.get_current_route() {
|
|
let draw_season_details_popup = |f: &mut Frame<'_>, app: &mut App<'_>, popup_area: Rect| {
|
|
let content_area = draw_tabs(
|
|
f,
|
|
popup_area,
|
|
&format!(
|
|
"Season {} Details",
|
|
app
|
|
.data
|
|
.sonarr_data
|
|
.seasons
|
|
.current_selection()
|
|
.season_number
|
|
),
|
|
&app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_ref()
|
|
.unwrap()
|
|
.season_details_tabs,
|
|
);
|
|
draw_season_details(f, app, content_area);
|
|
|
|
match active_sonarr_block {
|
|
ActiveSonarrBlock::AutomaticallySearchSeasonPrompt => {
|
|
let prompt = format!(
|
|
"Do you want to trigger an automatic search of your indexers for season packs for: {}",
|
|
app.data.sonarr_data.seasons.current_selection().title.as_ref().unwrap()
|
|
);
|
|
let confirmation_prompt = ConfirmationPrompt::new()
|
|
.title("Automatic Season Search")
|
|
.prompt(&prompt)
|
|
.yes_no_value(app.data.sonarr_data.prompt_confirm);
|
|
|
|
f.render_widget(
|
|
Popup::new(confirmation_prompt).size(Size::MediumPrompt),
|
|
f.area(),
|
|
);
|
|
}
|
|
ActiveSonarrBlock::DeleteEpisodeFilePrompt => {
|
|
let prompt = format!(
|
|
"Do you really want to delete this episode: \n{}?",
|
|
app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_ref()
|
|
.unwrap()
|
|
.episodes
|
|
.current_selection()
|
|
.title
|
|
);
|
|
let confirmation_prompt = ConfirmationPrompt::new()
|
|
.title("Delete Episode")
|
|
.prompt(&prompt)
|
|
.yes_no_value(app.data.sonarr_data.prompt_confirm);
|
|
|
|
f.render_widget(
|
|
Popup::new(confirmation_prompt).size(Size::MediumPrompt),
|
|
f.area(),
|
|
);
|
|
}
|
|
ActiveSonarrBlock::ManualSeasonSearchConfirmPrompt => {
|
|
draw_manual_season_search_confirm_prompt(f, app);
|
|
}
|
|
ActiveSonarrBlock::SeasonHistoryDetails => {
|
|
draw_history_item_details_popup(f, app, popup_area);
|
|
}
|
|
_ => (),
|
|
}
|
|
};
|
|
|
|
draw_popup(f, app, draw_season_details_popup, Size::XLarge);
|
|
|
|
if EpisodeDetailsUi::accepts(route) {
|
|
EpisodeDetailsUi::draw(f, app, _area);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn draw_season_details(f: &mut Frame<'_>, app: &mut App<'_>, area: Rect) {
|
|
if let Some(season_details_modal) = app.data.sonarr_data.season_details_modal.as_ref() {
|
|
if let Route::Sonarr(active_sonarr_block, _) =
|
|
season_details_modal.season_details_tabs.get_active_route()
|
|
{
|
|
match active_sonarr_block {
|
|
ActiveSonarrBlock::SeasonDetails => draw_episodes_table(f, app, area),
|
|
ActiveSonarrBlock::SeasonHistory => draw_season_history_table(f, app, area),
|
|
ActiveSonarrBlock::ManualSeasonSearch => draw_season_releases(f, app, area),
|
|
_ => (),
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn draw_episodes_table(f: &mut Frame<'_>, app: &mut App<'_>, area: Rect) {
|
|
if let Route::Sonarr(active_sonarr_block, _) = app.get_current_route() {
|
|
let help_footer = app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_ref()
|
|
.expect("Season details modal is unpopulated")
|
|
.season_details_tabs
|
|
.get_active_tab_contextual_help();
|
|
let episode_files = app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_ref()
|
|
.expect("Season details modal is unpopulated")
|
|
.episode_files
|
|
.items
|
|
.clone();
|
|
let content = Some(
|
|
&mut app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_mut()
|
|
.expect("Season details modal is unpopulated")
|
|
.episodes,
|
|
);
|
|
let downloads_vec = &app.data.sonarr_data.downloads.items;
|
|
|
|
let episode_row_mapping = |episode: &Episode| {
|
|
let Episode {
|
|
episode_number,
|
|
title,
|
|
air_date_utc,
|
|
episode_file_id,
|
|
..
|
|
} = episode;
|
|
let episode_file = episode_files
|
|
.iter()
|
|
.find(|episode_file| episode_file.id == *episode_file_id);
|
|
let (quality_profile, size_on_disk) = if let Some(episode_file) = episode_file {
|
|
(
|
|
episode_file.quality.quality.name.to_owned(),
|
|
episode_file.size,
|
|
)
|
|
} else {
|
|
(String::new(), 0)
|
|
};
|
|
|
|
let episode_monitored = if episode.monitored { "🏷" } else { "" };
|
|
let size = convert_to_gb(size_on_disk);
|
|
let air_date = if let Some(air_date) = air_date_utc.as_ref() {
|
|
air_date.to_string()
|
|
} else {
|
|
String::new()
|
|
};
|
|
|
|
decorate_with_row_style(
|
|
downloads_vec,
|
|
episode,
|
|
Row::new(vec![
|
|
Cell::from(episode_monitored.to_owned()),
|
|
Cell::from(episode_number.to_string()),
|
|
Cell::from(title.clone()),
|
|
Cell::from(air_date),
|
|
Cell::from(format!("{size:.2} GB")),
|
|
Cell::from(quality_profile),
|
|
]),
|
|
)
|
|
};
|
|
let is_searching = active_sonarr_block == ActiveSonarrBlock::SearchEpisodes;
|
|
let season_table = ManagarrTable::new(content, episode_row_mapping)
|
|
.block(layout_block_top_border())
|
|
.loading(app.is_loading)
|
|
.footer(help_footer)
|
|
.searching(is_searching)
|
|
.search_produced_empty_results(active_sonarr_block == ActiveSonarrBlock::SearchEpisodesError)
|
|
.headers([
|
|
"🏷",
|
|
"#",
|
|
"Title",
|
|
"Air Date",
|
|
"Size on Disk",
|
|
"Quality Profile",
|
|
])
|
|
.constraints([
|
|
Constraint::Percentage(4),
|
|
Constraint::Percentage(4),
|
|
Constraint::Percentage(50),
|
|
Constraint::Percentage(19),
|
|
Constraint::Percentage(10),
|
|
Constraint::Percentage(12),
|
|
]);
|
|
|
|
if is_searching {
|
|
season_table.show_cursor(f, area);
|
|
}
|
|
|
|
f.render_widget(season_table, area);
|
|
}
|
|
}
|
|
|
|
fn draw_season_history_table(f: &mut Frame<'_>, app: &mut App<'_>, area: Rect) {
|
|
match app.data.sonarr_data.season_details_modal.as_ref() {
|
|
Some(season_details_modal) if !app.is_loading => {
|
|
let current_selection = if season_details_modal.season_history.is_empty() {
|
|
SonarrHistoryItem::default()
|
|
} else {
|
|
season_details_modal
|
|
.season_history
|
|
.current_selection()
|
|
.clone()
|
|
};
|
|
let season_history_table_footer = season_details_modal
|
|
.season_details_tabs
|
|
.get_active_tab_contextual_help();
|
|
|
|
if let Route::Sonarr(active_sonarr_block, _) = app.get_current_route() {
|
|
let history_row_mapping = |history_item: &SonarrHistoryItem| {
|
|
let SonarrHistoryItem {
|
|
source_title,
|
|
languages,
|
|
quality,
|
|
event_type,
|
|
date,
|
|
..
|
|
} = history_item;
|
|
|
|
source_title.scroll_left_or_reset(
|
|
get_width_from_percentage(area, 40),
|
|
current_selection == *history_item,
|
|
app.tick_count % app.ticks_until_scroll == 0,
|
|
);
|
|
|
|
Row::new(vec![
|
|
Cell::from(source_title.to_string()),
|
|
Cell::from(event_type.to_string()),
|
|
Cell::from(
|
|
languages
|
|
.iter()
|
|
.map(|language| language.name.to_owned())
|
|
.collect::<Vec<String>>()
|
|
.join(","),
|
|
),
|
|
Cell::from(quality.quality.name.to_owned()),
|
|
Cell::from(date.to_string()),
|
|
])
|
|
.primary()
|
|
};
|
|
let mut season_history_table = &mut app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_mut()
|
|
.unwrap()
|
|
.season_history;
|
|
let history_table =
|
|
ManagarrTable::new(Some(&mut season_history_table), history_row_mapping)
|
|
.block(layout_block_top_border())
|
|
.loading(app.is_loading)
|
|
.footer(season_history_table_footer)
|
|
.sorting(active_sonarr_block == ActiveSonarrBlock::SeasonHistorySortPrompt)
|
|
.searching(active_sonarr_block == ActiveSonarrBlock::SearchSeasonHistory)
|
|
.search_produced_empty_results(
|
|
active_sonarr_block == ActiveSonarrBlock::SearchSeasonHistoryError,
|
|
)
|
|
.filtering(active_sonarr_block == ActiveSonarrBlock::FilterSeasonHistory)
|
|
.filter_produced_empty_results(
|
|
active_sonarr_block == ActiveSonarrBlock::FilterSeasonHistoryError,
|
|
)
|
|
.headers(["Source Title", "Event Type", "Language", "Quality", "Date"])
|
|
.constraints([
|
|
Constraint::Percentage(40),
|
|
Constraint::Percentage(15),
|
|
Constraint::Percentage(12),
|
|
Constraint::Percentage(13),
|
|
Constraint::Percentage(20),
|
|
]);
|
|
|
|
if [
|
|
ActiveSonarrBlock::SearchSeriesHistory,
|
|
ActiveSonarrBlock::FilterSeriesHistory,
|
|
]
|
|
.contains(&active_sonarr_block)
|
|
{
|
|
history_table.show_cursor(f, area);
|
|
}
|
|
|
|
f.render_widget(history_table, area);
|
|
}
|
|
}
|
|
_ => f.render_widget(
|
|
LoadingBlock::new(
|
|
app.is_loading || app.data.sonarr_data.season_details_modal.is_none(),
|
|
layout_block_top_border(),
|
|
),
|
|
area,
|
|
),
|
|
}
|
|
}
|
|
|
|
fn draw_season_releases(f: &mut Frame<'_>, app: &mut App<'_>, area: Rect) {
|
|
match app.data.sonarr_data.season_details_modal.as_ref() {
|
|
Some(season_details_modal) if !app.is_loading => {
|
|
let (current_selection, is_empty) = if season_details_modal.season_releases.is_empty() {
|
|
(SonarrRelease::default(), true)
|
|
} else {
|
|
(
|
|
season_details_modal
|
|
.season_releases
|
|
.current_selection()
|
|
.clone(),
|
|
season_details_modal.season_releases.is_empty(),
|
|
)
|
|
};
|
|
let season_release_table_footer = season_details_modal
|
|
.season_details_tabs
|
|
.get_active_tab_contextual_help();
|
|
|
|
if let Route::Sonarr(active_sonarr_block, _) = app.get_current_route() {
|
|
let season_release_row_mapping = |release: &SonarrRelease| {
|
|
let SonarrRelease {
|
|
protocol,
|
|
age,
|
|
title,
|
|
indexer,
|
|
size,
|
|
rejected,
|
|
seeders,
|
|
leechers,
|
|
languages,
|
|
quality,
|
|
..
|
|
} = release;
|
|
|
|
let age = format!("{age} days");
|
|
title.scroll_left_or_reset(
|
|
get_width_from_percentage(area, 30),
|
|
current_selection == *release
|
|
&& active_sonarr_block != ActiveSonarrBlock::ManualSeasonSearchConfirmPrompt,
|
|
app.tick_count % app.ticks_until_scroll == 0,
|
|
);
|
|
let size = convert_to_gb(*size);
|
|
let rejected_str = if *rejected { "⛔" } else { "" };
|
|
let peers = if seeders.is_none() || leechers.is_none() {
|
|
Text::from("")
|
|
} else {
|
|
let seeders = seeders.clone().unwrap().as_u64().unwrap();
|
|
let leechers = leechers.clone().unwrap().as_u64().unwrap();
|
|
|
|
decorate_peer_style(
|
|
seeders,
|
|
leechers,
|
|
Text::from(format!("{seeders} / {leechers}")),
|
|
)
|
|
};
|
|
|
|
let language = if languages.is_some() {
|
|
languages.clone().unwrap()[0].name.clone()
|
|
} else {
|
|
String::new()
|
|
};
|
|
let quality = quality.quality.name.clone();
|
|
|
|
Row::new(vec![
|
|
Cell::from(protocol.clone()),
|
|
Cell::from(age),
|
|
Cell::from(rejected_str),
|
|
Cell::from(title.to_string()),
|
|
Cell::from(indexer.clone()),
|
|
Cell::from(format!("{size:.1} GB")),
|
|
Cell::from(peers),
|
|
Cell::from(language),
|
|
Cell::from(quality),
|
|
])
|
|
.primary()
|
|
};
|
|
let mut season_release_table = &mut app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_mut()
|
|
.unwrap()
|
|
.season_releases;
|
|
let release_table =
|
|
ManagarrTable::new(Some(&mut season_release_table), season_release_row_mapping)
|
|
.block(layout_block_top_border())
|
|
.loading(app.is_loading || is_empty)
|
|
.footer(season_release_table_footer)
|
|
.sorting(active_sonarr_block == ActiveSonarrBlock::ManualSeasonSearchSortPrompt)
|
|
.headers([
|
|
"Source", "Age", "⛔", "Title", "Indexer", "Size", "Peers", "Language", "Quality",
|
|
])
|
|
.constraints([
|
|
Constraint::Length(9),
|
|
Constraint::Length(10),
|
|
Constraint::Length(5),
|
|
Constraint::Percentage(30),
|
|
Constraint::Percentage(18),
|
|
Constraint::Length(12),
|
|
Constraint::Length(12),
|
|
Constraint::Percentage(7),
|
|
Constraint::Percentage(10),
|
|
]);
|
|
|
|
f.render_widget(release_table, area);
|
|
}
|
|
}
|
|
_ => f.render_widget(
|
|
LoadingBlock::new(
|
|
app.is_loading || app.data.sonarr_data.season_details_modal.is_none(),
|
|
layout_block_top_border(),
|
|
),
|
|
area,
|
|
),
|
|
}
|
|
}
|
|
|
|
fn draw_manual_season_search_confirm_prompt(f: &mut Frame<'_>, app: &mut App<'_>) {
|
|
let current_selection = app
|
|
.data
|
|
.sonarr_data
|
|
.season_details_modal
|
|
.as_ref()
|
|
.unwrap()
|
|
.season_releases
|
|
.current_selection();
|
|
let title = if current_selection.rejected {
|
|
"Download Rejected Release"
|
|
} else {
|
|
"Download Release"
|
|
};
|
|
let prompt = if current_selection.rejected {
|
|
format!(
|
|
"Do you really want to download the rejected release: {}?",
|
|
¤t_selection.title.text
|
|
)
|
|
} else {
|
|
format!(
|
|
"Do you want to download the release: {}?",
|
|
¤t_selection.title.text
|
|
)
|
|
};
|
|
|
|
if current_selection.rejected {
|
|
let mut lines_vec = vec![Line::from("Rejection reasons: ".primary().bold())];
|
|
let mut rejections_spans = current_selection
|
|
.rejections
|
|
.clone()
|
|
.unwrap_or_default()
|
|
.iter()
|
|
.map(|item| Line::from(format!("• {item}").primary().bold()))
|
|
.collect::<Vec<Line<'_>>>();
|
|
lines_vec.append(&mut rejections_spans);
|
|
|
|
let content_paragraph = Paragraph::new(lines_vec)
|
|
.block(borderless_block())
|
|
.wrap(Wrap { trim: false })
|
|
.left_aligned();
|
|
let confirmation_prompt = ConfirmationPrompt::new()
|
|
.title(title)
|
|
.prompt(&prompt)
|
|
.content(content_paragraph)
|
|
.yes_no_value(app.data.sonarr_data.prompt_confirm);
|
|
|
|
f.render_widget(Popup::new(confirmation_prompt).size(Size::Small), f.area());
|
|
} else {
|
|
let confirmation_prompt = ConfirmationPrompt::new()
|
|
.title(title)
|
|
.prompt(&prompt)
|
|
.yes_no_value(app.data.sonarr_data.prompt_confirm);
|
|
|
|
f.render_widget(
|
|
Popup::new(confirmation_prompt).size(Size::MediumPrompt),
|
|
f.area(),
|
|
);
|
|
}
|
|
}
|
|
|
|
fn draw_history_item_details_popup(f: &mut Frame<'_>, app: &mut App<'_>, area: Rect) {
|
|
let current_selection =
|
|
if let Some(season_details_modal) = app.data.sonarr_data.season_details_modal.as_ref() {
|
|
if season_details_modal.season_history.is_empty() {
|
|
SonarrHistoryItem::default()
|
|
} else {
|
|
season_details_modal
|
|
.season_history
|
|
.current_selection()
|
|
.clone()
|
|
}
|
|
} else {
|
|
SonarrHistoryItem::default()
|
|
};
|
|
|
|
let line_vec = match current_selection.event_type {
|
|
SonarrHistoryEventType::Grabbed => create_grabbed_history_event_details(current_selection),
|
|
SonarrHistoryEventType::DownloadFolderImported => {
|
|
create_download_folder_imported_history_event_details(current_selection)
|
|
}
|
|
SonarrHistoryEventType::DownloadFailed => {
|
|
create_download_failed_history_event_details(current_selection)
|
|
}
|
|
SonarrHistoryEventType::EpisodeFileDeleted => {
|
|
create_episode_file_deleted_history_event_details(current_selection)
|
|
}
|
|
SonarrHistoryEventType::EpisodeFileRenamed => {
|
|
create_episode_file_renamed_history_event_details(current_selection)
|
|
}
|
|
_ => create_no_data_history_event_details(current_selection),
|
|
};
|
|
let text = Text::from(line_vec);
|
|
|
|
let message = Message::new(text)
|
|
.title("Details")
|
|
.style(Style::new().secondary())
|
|
.alignment(Alignment::Left);
|
|
|
|
f.render_widget(Popup::new(message).size(Size::NarrowMessage), area);
|
|
}
|
|
|
|
fn decorate_with_row_style<'a>(
|
|
downloads_vec: &[DownloadRecord],
|
|
episode: &Episode,
|
|
row: Row<'a>,
|
|
) -> Row<'a> {
|
|
if !episode.has_file {
|
|
let default_episode_id = Number::from(-1i64);
|
|
if let Some(download) = downloads_vec.iter().find(|&download| {
|
|
download
|
|
.episode_id
|
|
.as_ref()
|
|
.unwrap_or(&default_episode_id)
|
|
.as_i64()
|
|
.unwrap()
|
|
== episode.id
|
|
}) {
|
|
if download.status == DownloadStatus::Downloading {
|
|
return row.downloading();
|
|
}
|
|
|
|
if download.status == DownloadStatus::Completed {
|
|
return row.awaiting_import();
|
|
}
|
|
}
|
|
|
|
if !episode.monitored {
|
|
return row.unmonitored_missing();
|
|
}
|
|
|
|
if let Some(air_date) = episode.air_date_utc.as_ref() {
|
|
if air_date > &Utc::now() {
|
|
return row.unreleased();
|
|
}
|
|
}
|
|
|
|
return row.missing();
|
|
}
|
|
|
|
if !episode.monitored {
|
|
row.unmonitored()
|
|
} else {
|
|
row.downloaded()
|
|
}
|
|
}
|