use crate::{ models::servarr_data::sonarr::sonarr_data::ActiveSonarrBlock, network::sonarr_network::SonarrEvent, }; use super::App; pub mod sonarr_context_clues; #[cfg(test)] #[path = "sonarr_tests.rs"] mod sonarr_tests; impl<'a> App<'a> { pub(super) async fn dispatch_by_sonarr_block(&mut self, active_sonarr_block: &ActiveSonarrBlock) { match active_sonarr_block { ActiveSonarrBlock::Series => { self .dispatch_network_event(SonarrEvent::GetQualityProfiles.into()) .await; self .dispatch_network_event(SonarrEvent::GetLanguageProfiles.into()) .await; self .dispatch_network_event(SonarrEvent::GetTags.into()) .await; self .dispatch_network_event(SonarrEvent::ListSeries.into()) .await; } ActiveSonarrBlock::SeriesDetails => { self .dispatch_network_event(SonarrEvent::ListSeries.into()) .await; self.is_loading = true; self.populate_seasons_table().await; self.is_loading = false; } ActiveSonarrBlock::SeriesHistory => { self .dispatch_network_event(SonarrEvent::GetSeriesHistory(None).into()) .await; } ActiveSonarrBlock::SeasonDetails => { self .dispatch_network_event(SonarrEvent::GetEpisodes(None).into()) .await; self .dispatch_network_event(SonarrEvent::GetEpisodeFiles(None).into()) .await; self .dispatch_network_event(SonarrEvent::GetDownloads.into()) .await; } ActiveSonarrBlock::SeasonHistory => { self .dispatch_network_event(SonarrEvent::GetSeasonHistory(None).into()) .await; } ActiveSonarrBlock::ManualSeasonSearch => { match self.data.sonarr_data.season_details_modal.as_ref() { Some(season_details_modal) if season_details_modal.season_releases.is_empty() => { self .dispatch_network_event(SonarrEvent::GetSeasonReleases(None).into()) .await; } _ => (), } } ActiveSonarrBlock::EpisodeDetails | ActiveSonarrBlock::EpisodeFile => { self .dispatch_network_event(SonarrEvent::GetEpisodeDetails(None).into()) .await; } ActiveSonarrBlock::EpisodeHistory => { self .dispatch_network_event(SonarrEvent::GetEpisodeHistory(None).into()) .await; } ActiveSonarrBlock::ManualEpisodeSearch => { if let Some(season_details_modal) = self.data.sonarr_data.season_details_modal.as_ref() { if let Some(episode_details_modal) = season_details_modal.episode_details_modal.as_ref() { if episode_details_modal.episode_releases.is_empty() { self .dispatch_network_event(SonarrEvent::GetEpisodeReleases(None).into()) .await; } } } } ActiveSonarrBlock::Downloads => { self .dispatch_network_event(SonarrEvent::GetDownloads.into()) .await; } ActiveSonarrBlock::Blocklist => { self .dispatch_network_event(SonarrEvent::ListSeries.into()) .await; self .dispatch_network_event(SonarrEvent::GetBlocklist.into()) .await; } ActiveSonarrBlock::History => { self .dispatch_network_event(SonarrEvent::GetHistory(None).into()) .await; } ActiveSonarrBlock::RootFolders => { self .dispatch_network_event(SonarrEvent::GetRootFolders.into()) .await; } ActiveSonarrBlock::Indexers => { self .dispatch_network_event(SonarrEvent::GetTags.into()) .await; self .dispatch_network_event(SonarrEvent::GetIndexers.into()) .await; } ActiveSonarrBlock::AllIndexerSettingsPrompt => { self .dispatch_network_event(SonarrEvent::GetAllIndexerSettings.into()) .await; } ActiveSonarrBlock::TestIndexer => { self .dispatch_network_event(SonarrEvent::TestIndexer(None).into()) .await; } ActiveSonarrBlock::TestAllIndexers => { self .dispatch_network_event(SonarrEvent::TestAllIndexers.into()) .await; } ActiveSonarrBlock::System => { self .dispatch_network_event(SonarrEvent::GetTasks.into()) .await; self .dispatch_network_event(SonarrEvent::GetQueuedEvents.into()) .await; self .dispatch_network_event(SonarrEvent::GetLogs(None).into()) .await; } ActiveSonarrBlock::AddSeriesSearchResults => { self .dispatch_network_event(SonarrEvent::SearchNewSeries(None).into()) .await; } ActiveSonarrBlock::SystemUpdates => { self .dispatch_network_event(SonarrEvent::GetUpdates.into()) .await; } _ => (), } self.check_for_sonarr_prompt_action().await; self.reset_tick_count(); } async fn check_for_sonarr_prompt_action(&mut self) { if self.data.sonarr_data.prompt_confirm { self.data.sonarr_data.prompt_confirm = false; if let Some(sonarr_event) = &self.data.sonarr_data.prompt_confirm_action { self .dispatch_network_event(sonarr_event.clone().into()) .await; self.should_refresh = true; self.data.sonarr_data.prompt_confirm_action = None; } } } pub(super) async fn sonarr_on_tick(&mut self, active_sonarr_block: ActiveSonarrBlock) { if self.is_first_render { self.refresh_sonarr_metadata().await; self.dispatch_by_sonarr_block(&active_sonarr_block).await; self.is_first_render = false; return; } if self.should_refresh { self.dispatch_by_sonarr_block(&active_sonarr_block).await; self.refresh_sonarr_metadata().await; } if self.is_routing { if !self.should_refresh { self.cancellation_token.cancel(); } else { self.dispatch_by_sonarr_block(&active_sonarr_block).await; } } if self.tick_count % self.tick_until_poll == 0 { self.refresh_sonarr_metadata().await; } } async fn refresh_sonarr_metadata(&mut self) { self .dispatch_network_event(SonarrEvent::GetQualityProfiles.into()) .await; self .dispatch_network_event(SonarrEvent::GetLanguageProfiles.into()) .await; self .dispatch_network_event(SonarrEvent::GetTags.into()) .await; self .dispatch_network_event(SonarrEvent::GetRootFolders.into()) .await; self .dispatch_network_event(SonarrEvent::GetDownloads.into()) .await; self .dispatch_network_event(SonarrEvent::GetDiskSpace.into()) .await; self .dispatch_network_event(SonarrEvent::GetStatus.into()) .await; } async fn populate_seasons_table(&mut self) { let seasons = self .data .sonarr_data .series .current_selection() .clone() .seasons .unwrap_or_default() .into_iter() .map(|mut season| { season.title = Some(format!("Season {}", season.season_number)); season }) .collect(); self.data.sonarr_data.seasons.set_items(seasons); } }