feat(network): Support for fetching all Sonarr root folders

This commit is contained in:
2024-11-21 16:37:23 -07:00
parent f4c647342b
commit 16538a3158
18 changed files with 138 additions and 45 deletions
+23 -1
View File
@@ -10,7 +10,8 @@ use crate::{
sonarr_data::ActiveSonarrBlock,
},
servarr_models::{
HostConfig, Indexer, LogResponse, QualityProfile, QueueEvent, Release, SecurityConfig,
HostConfig, Indexer, LogResponse, QualityProfile, QueueEvent, Release, RootFolder,
SecurityConfig,
},
sonarr_models::{
BlocklistResponse, DownloadRecord, DownloadsResponse, Episode, IndexerSettings, Series,
@@ -46,6 +47,7 @@ pub enum SonarrEvent {
GetLogs(Option<u64>),
GetQualityProfiles,
GetQueuedEvents,
GetRootFolders,
GetEpisodeReleases(Option<i64>),
GetSeasonReleases(Option<(i64, i64)>),
GetSecurityConfig,
@@ -71,6 +73,7 @@ impl NetworkResource for SonarrEvent {
SonarrEvent::GetLogs(_) => "/log",
SonarrEvent::GetQualityProfiles => "/qualityprofile",
SonarrEvent::GetQueuedEvents => "/command",
SonarrEvent::GetRootFolders => "/rootfolder",
SonarrEvent::GetSeasonReleases(_) | SonarrEvent::GetEpisodeReleases(_) => "/release",
SonarrEvent::GetSeriesHistory(_) => "/history/series",
SonarrEvent::GetStatus => "/system/status",
@@ -147,6 +150,10 @@ impl<'a, 'b> Network<'a, 'b> {
.get_queued_sonarr_events()
.await
.map(SonarrSerdeable::from),
SonarrEvent::GetRootFolders => self
.get_sonarr_root_folders()
.await
.map(SonarrSerdeable::from),
SonarrEvent::GetEpisodeReleases(params) => self
.get_episode_releases(params)
.await
@@ -746,6 +753,21 @@ impl<'a, 'b> Network<'a, 'b> {
.await
}
async fn get_sonarr_root_folders(&mut self) -> Result<Vec<RootFolder>> {
info!("Fetching Sonarr root folders");
let event = SonarrEvent::GetRootFolders;
let request_props = self
.request_props_from(event, RequestMethod::Get, None::<()>, None, None)
.await;
self
.handle_request::<(), Vec<RootFolder>>(request_props, |root_folders, mut app| {
app.data.sonarr_data.root_folders.set_items(root_folders);
})
.await
}
async fn get_episode_releases(&mut self, episode_id: Option<i64>) -> Result<Vec<Release>> {
let event = SonarrEvent::GetEpisodeReleases(None);
let id = self.extract_episode_id(episode_id).await;