Dark-Alex-17/managarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Almost kinda functional description box

Browse Source
This commit is contained in:
Dark-Alex-17
2023-08-08 10:50:04 -06:00
parent d39acb0683
commit b24e0cdccd
16 changed files with 710 additions and 288 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/network/radarr_network.rs
+206 -50
View File
@@ -7,8 +7,11 @@ use serde::Deserialize;
use serde_json::Number;
use tokio::sync::MutexGuard;
use crate::app::models::ScrollableText;
use crate::app::{App, RadarrConfig};
use crate::network::{Network, NetworkEvent, utils};
use crate::network::utils::get_movie_status;
use crate::network::{utils, Network, NetworkEvent};
use crate::utils::{convert_runtime, convert_to_gb};
#[derive(Debug, Eq, PartialEq)]
pub enum RadarrEvent {
@@ -17,6 +20,7 @@ pub enum RadarrEvent {
GetOverview,
GetStatus,
GetMovies,
GetMovieDetails,
GetQualityProfiles,
}
@@ -26,9 +30,9 @@ impl RadarrEvent {
RadarrEvent::HealthCheck => "/health",
RadarrEvent::GetOverview => "/diskspace",
RadarrEvent::GetStatus => "/system/status",
RadarrEvent::GetMovies => "/movie",
RadarrEvent::GetMovies | RadarrEvent::GetMovieDetails => "/movie",
RadarrEvent::GetDownloads => "/queue",
RadarrEvent::GetQualityProfiles => "/qualityprofile"
RadarrEvent::GetQualityProfiles => "/qualityprofile",
}
}
}
@@ -65,6 +69,10 @@ pub struct Movie {
#[derivative(Default(value = "Number::from(0)"))]
pub size_on_disk: Number,
pub status: String,
pub overview: String,
pub path: String,
pub studio: String,
pub genres: Vec<String>,
#[derivative(Default(value = "Number::from(0)"))]
pub year: Number,
pub monitored: bool,
@@ -73,13 +81,29 @@ pub struct Movie {
pub runtime: Number,
#[derivative(Default(value = "Number::from(0)"))]
pub quality_profile_id: Number,
pub ratings: RatingsList,
}
#[derive(Default, Deserialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct RatingsList {
pub imdb: Option<Rating>,
pub tmdb: Option<Rating>,
pub rotten_tomatoes: Option<Rating>,
}
#[derive(Derivative, Deserialize, Debug)]
#[derivative(Default)]
pub struct Rating {
#[derivative(Default(value = "Number::from(0)"))]
pub value: Number,
}
#[derive(Derivative, Deserialize, Debug)]
#[derivative(Default)]
#[serde(rename_all = "camelCase")]
pub struct DownloadsResponse {
pub records: Vec<DownloadRecord>
pub records: Vec<DownloadRecord>,
}
#[derive(Derivative, Deserialize, Debug)]
@@ -96,7 +120,7 @@ pub struct DownloadRecord {
pub sizeleft: Number,
pub output_path: String,
pub indexer: String,
pub download_client: String
pub download_client: String,
}
#[derive(Derivative, Deserialize, Debug)]
@@ -105,22 +129,43 @@ pub struct DownloadRecord {
struct QualityProfile {
#[derivative(Default(value = "Number::from(0)"))]
pub id: Number,
pub name: String
pub name: String,
}
impl<'a> Network<'a> {
pub async fn handle_radarr_event(&self, radarr_event: RadarrEvent) {
match radarr_event {
RadarrEvent::HealthCheck => self.get_healthcheck(RadarrEvent::HealthCheck.resource()).await,
RadarrEvent::GetOverview => self.get_diskspace(RadarrEvent::GetOverview.resource()).await,
RadarrEvent::HealthCheck => {
self
.get_healthcheck(RadarrEvent::HealthCheck.resource())
.await
}
RadarrEvent::GetOverview => {
self
.get_diskspace(RadarrEvent::GetOverview.resource())
.await
}
RadarrEvent::GetStatus => self.get_status(RadarrEvent::GetStatus.resource()).await,
RadarrEvent::GetMovies => self.get_movies(RadarrEvent::GetMovies.resource()).await,
RadarrEvent::GetDownloads => self.get_downloads(RadarrEvent::GetDownloads.resource()).await,
RadarrEvent::GetQualityProfiles => self.get_quality_profiles(RadarrEvent::GetQualityProfiles.resource()).await
RadarrEvent::GetMovieDetails => {
debug!("TEST received GetMovieDetails event");
self
.get_movie_details(RadarrEvent::GetMovieDetails.resource())
.await
}
RadarrEvent::GetDownloads => {
self
.get_downloads(RadarrEvent::GetDownloads.resource())
.await
}
RadarrEvent::GetQualityProfiles => {
self
.get_quality_profiles(RadarrEvent::GetQualityProfiles.resource())
.await
}
}
let mut app = self.app.lock().await;
app.reset_tick_count();
}
async fn get_healthcheck(&self, resource: &str) {
@@ -130,43 +175,155 @@ impl<'a> Network<'a> {
}
async fn get_diskspace(&self, resource: &str) {
self.handle_get_request::<Vec<DiskSpace>>(resource, | disk_space_vec, mut app | {
let DiskSpace {
free_space,
total_space,
..
} = &disk_space_vec[0];
self
.handle_get_request::<Vec<DiskSpace>>(resource, |disk_space_vec, mut app| {
let DiskSpace {
free_space,
total_space,
..
} = &disk_space_vec[0];
app.data.radarr_data.free_space = free_space.as_u64().unwrap();
app.data.radarr_data.total_space = total_space.as_u64().unwrap();
}).await;
app.data.radarr_data.free_space = free_space.as_u64().unwrap();
app.data.radarr_data.total_space = total_space.as_u64().unwrap();
})
.await;
}
async fn get_status(&self, resource: &str) {
self.handle_get_request::<SystemStatus>(resource, | system_status, mut app | {
app.data.radarr_data.version = system_status.version;
app.data.radarr_data.start_time = system_status.start_time;
}).await;
self
.handle_get_request::<SystemStatus>(resource, |system_status, mut app| {
app.data.radarr_data.version = system_status.version;
app.data.radarr_data.start_time = system_status.start_time;
})
.await;
}
async fn get_movies(&self, resource: &str) {
self.handle_get_request::<Vec<Movie>>(resource, |movie_vec, mut app| {
app.data.radarr_data.movies.set_items(movie_vec);
}).await;
self
.handle_get_request::<Vec<Movie>>(resource, |movie_vec, mut app| {
app.data.radarr_data.movies.set_items(movie_vec);
})
.await;
}
async fn get_movie_details(&self, resource: &str) {
debug!("TEST handling get_movie_details");
let movie_id = self
.app
.lock()
.await
.data
.radarr_data
.movies
.current_selection()
.id
.clone()
.as_u64()
.unwrap();
let mut url = resource.to_owned();
url.push('/');
url.push_str(movie_id.to_string().as_str());
debug!("TEST sending request{}", url.as_str());
self
.handle_get_request::<Movie>(url.as_str(), |movie_response, mut app| {
let Movie {
id,
title,
year,
overview,
path,
studio,
has_file,
quality_profile_id,
size_on_disk,
genres,
runtime,
ratings,
..
} = movie_response;
let (hours, minutes) = convert_runtime(runtime.as_u64().unwrap());
let size = convert_to_gb(size_on_disk.as_u64().unwrap());
let quality_profile = app
.data
.radarr_data
.quality_profile_map
.get(&quality_profile_id.as_u64().unwrap())
.unwrap()
.to_owned();
let imdb_rating = if let Some(rating) = ratings.imdb {
rating.value.as_f64().unwrap()
} else {
0f64
};
let tmdb_rating = if let Some(rating) = ratings.tmdb {
rating.value.as_u64().unwrap()
} else {
0u64
};
let rotten_tomatoes_rating = if let Some(rating) = ratings.rotten_tomatoes {
rating.value.as_u64().unwrap()
} else {
0u64
};
let status = get_movie_status(has_file, &app.data.radarr_data.downloads.items, id);
app.data.radarr_data.movie_details = ScrollableText::with_string(format!(
"Title: {}\n
Year: {}\n
Runtime: {}h {}m\n
Status: {}\n
Description: {}\n
TMDB: {}%\n
IMDB: {:.1}\n
Rotten Tomatoes: {}%\n
Quality Profile: {}\n
Size: {:.2} GB\n
Path: {}\n
Studio: {}\n
Genres: {}",
title,
year,
hours,
minutes,
status,
overview,
tmdb_rating,
imdb_rating,
rotten_tomatoes_rating,
quality_profile,
size,
path,
studio,
genres.join(", ")
))
})
.await;
}
async fn get_downloads(&self, resource: &str) {
self.handle_get_request::<DownloadsResponse>(resource, |queue_response, mut app | {
app.data.radarr_data.downloads.set_items(queue_response.records);
}).await
self
.handle_get_request::<DownloadsResponse>(resource, |queue_response, mut app| {
app
.data
.radarr_data
.downloads
.set_items(queue_response.records);
})
.await
}
async fn get_quality_profiles(&self, resource: &str) {
self.handle_get_request::<Vec<QualityProfile>>(resource, | quality_profiles, mut app | {
app.data.radarr_data.quality_profile_map = quality_profiles.into_iter()
.map(| profile | (profile.id.as_u64().unwrap(), profile.name))
self
.handle_get_request::<Vec<QualityProfile>>(resource, |quality_profiles, mut app| {
app.data.radarr_data.quality_profile_map = quality_profiles
.into_iter()
.map(|profile| (profile.id.as_u64().unwrap(), profile.name))
.collect();
}).await;
})
.await;
}
async fn call_radarr_api(&self, resource: &str) -> RequestBuilder {
@@ -189,23 +346,22 @@ impl<'a> Network<'a> {
.header("X-Api-Key", api_token)
}
async fn handle_get_request<T>(&self, resource: &str, mut app_update_fn: impl FnMut(T, MutexGuard<App>))
where
T: DeserializeOwned {
match self.call_radarr_api(resource)
.await
.send()
.await {
Ok(response) => {
match utils::parse_response::<T>(response).await {
Ok(value) => {
let app = self.app.lock().await;
app_update_fn(value, app);
}
Err(e) => error!("Failed to parse response! {:?}", e)
async fn handle_get_request<T>(
&self,
resource: &str,
mut app_update_fn: impl FnMut(T, MutexGuard<App>),
) where
T: DeserializeOwned,
{
match self.call_radarr_api(resource).await.send().await {
Ok(response) => match utils::parse_response::<T>(response).await {
Ok(value) => {
let app = self.app.lock().await;
app_update_fn(value, app);
}
}
Err(e) => error!("Failed to fetch resource. {:?}", e)
Err(e) => error!("Failed to parse response! {:?}", e),
},
Err(e) => error!("Failed to fetch resource. {:?}", e),
}
}
}