Skip to content

Commit

Permalink
Merge pull request #12 from tcheronneau/alpha
Browse files Browse the repository at this point in the history 
Alpha
  • Loading branch information
@tcheronneau
tcheronneau authored Nov 19, 2024
2 parents ea0bf7c + 0a05593 commit f6c377a
Show file tree
Hide file tree
Showing 5 changed files with 267 additions and 188 deletions.
147 changes: 88 additions & 59 deletions src/http_server.rs
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
use anyhow::Result;
use futures::future::try_join_all;
use log::{error, info};
use rocket::http::{Accept, ContentType, Status};
use rocket::tokio::task;
Expand Down Expand Up @@ -63,72 +64,100 @@ async fn metrics(
) -> Result<MetricsResponse, MetricsError> {
Ok(serve_metrics(Format::Prometheus, unscheduled_tasks).await)
}
async fn process_task(task: Task) -> Result<TaskResult, JoinError> {
info!("Requesting data for {:?}", &task,);
match task {
Task::SonarrToday(sonarr) => {
let name = &sonarr.name;
let result = sonarr.get_today_shows().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::SonarrToday(result))
}
Task::SonarrMissing(sonarr) => {
let name = &sonarr.name;
let result = sonarr.get_last_week_missing_shows().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::SonarrMissing(result))
}
Task::TautulliSession(tautulli) => {
let result = tautulli.get_session_summary().await;
Ok(TaskResult::TautulliSession(result))
}
Task::TautulliLibrary(tautulli) => {
let result = tautulli.get_libraries().await;
Ok(TaskResult::TautulliLibrary(result))
}
Task::Radarr(radarr) => {
let name = &radarr.name;
let result = radarr.get_radarr_movies().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::Radarr(result))
}
Task::Overseerr(overseerr) => {
let result = overseerr.get_overseerr_requests().await;
Ok(TaskResult::Overseerr(result))
}
Task::PlexSession(plex) => {
let name = &plex.name;
let result = plex.get_current_sessions().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::PlexSession(result))
}
Task::PlexLibrary(plex) => {
let name = &plex.name;
let result = plex.get_all_library_size().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::PlexLibrary(result))
}
Task::Default => Ok(TaskResult::Default),
}
async fn process_tasks(tasks: Vec<Task>) -> Result<Vec<TaskResult>, JoinError> {
let task_futures: Vec<_> = tasks
.into_iter()
.map(|task| async {
info!("Requesting data for {:?}", &task,);
match task {
Task::SonarrToday(sonarr) => {
let name = &sonarr.name;
let result = sonarr.get_today_shows().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::SonarrToday(result))
}
Task::SonarrMissing(sonarr) => {
let name = &sonarr.name;
let result = sonarr.get_last_week_missing_shows().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::SonarrMissing(result))
}
Task::TautulliSession(tautulli) => {
let result = tautulli.get_session_summary().await;
Ok(TaskResult::TautulliSession(result))
}
Task::TautulliLibrary(tautulli) => {
let result = tautulli.get_libraries().await;
Ok(TaskResult::TautulliLibrary(result))
}
Task::Radarr(radarr) => {
let name = &radarr.name;
let result = radarr.get_radarr_movies().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::Radarr(result))
}
Task::Overseerr(overseerr) => {
let result = overseerr.get_overseerr_requests().await;
Ok(TaskResult::Overseerr(result))
}
Task::PlexSession(plex) => {
let name = &plex.name;
let result = plex.get_current_sessions().await;
let result = HashMap::from([(name.to_string(), result)]);
let users = plex.get_users().await;
Ok(TaskResult::PlexSession(result, users))
}
Task::PlexLibrary(plex) => {
let name = &plex.name;
let result = plex.get_all_library_size().await;
let result = HashMap::from([(name.to_string(), result)]);
Ok(TaskResult::PlexLibrary(result))
}
Task::Default => Ok(TaskResult::Default),
}
})
.collect();
try_join_all(task_futures).await
}

async fn serve_metrics(format: Format, unscheduled_tasks: &State<Vec<Task>>) -> MetricsResponse {
let mut join_set = JoinSet::new();
for task in unscheduled_tasks.iter().cloned() {
join_set.spawn(process_task(task));
}

wait_for_metrics(format, join_set).await.map_or_else(
|e| {
error!("General error while fetching providers data: {e}");
match process_tasks(unscheduled_tasks.inner().clone()).await {
Ok(task_results) => match format_metrics(task_results) {
Ok(metrics) => MetricsResponse::new(Status::Ok, format, metrics),
Err(e) => {
error!("Error formatting metrics: {e}");
MetricsResponse::new(
Status::InternalServerError,
format,
"Error formatting metrics. Check the logs.".into(),
)
}
},
Err(e) => {
error!("Error while processing tasks: {e}");
MetricsResponse::new(
Status::InternalServerError,
format,
"Error while fetching providers data. Check the logs".into(),
"Error while fetching provider data. Check the logs.".into(),
)
},
|metrics| MetricsResponse::new(Status::Ok, format, metrics),
)
}
}
//let mut join_set = JoinSet::new();
//for task in unscheduled_tasks.iter().cloned() {
// join_set.spawn(process_task(task));
//}

//wait_for_metrics(format, join_set).await.map_or_else(
// |e| {
// error!("General error while fetching providers data: {e}");
// MetricsResponse::new(
// Status::InternalServerError,
// format,
// "Error while fetching providers data. Check the logs".into(),
// )
// },
// |metrics| MetricsResponse::new(Status::Ok, format, metrics),
//)
}

async fn wait_for_metrics(
Expand Down
23 changes: 14 additions & 9 deletions src/main.rs.test
View file
Original file line number Diff line number Diff line change
Expand Up @@ -82,16 +82,21 @@ async fn main() -> Result<()> {
.log_level()
.expect("Log level cannot be not available");
let config = config::read(args.config.clone(), log_level)?;
let jelly_conf = config.jellyfin.expect("Jellyfin config not found");
for (name, j) in jelly_conf {
let jellyfin = Jellyfin::new(&name, &j.address, &j.api_key)?;
let session = jellyfin.get_sessions().await?;
println!("{:?}", session);
}
//let jelly_conf = config.jellyfin.expect("Jellyfin config not found");
//for (name, j) in jelly_conf {
// let jellyfin = Jellyfin::new(&name, &j.address, &j.api_key)?;
// let session = jellyfin.get_sessions().await?;
// println!("{:?}", session);
//}

//let plex = config.plex.expect("plex config not found");
//for (name, p) in plex {
// let plex = providers::plex::Plex::new(&name, &p.address, &p.token)?;
let plex = config.plex.expect("plex config not found");
for (name, p) in plex {
let plex = providers::plex::Plex::new(&name, &p.address, &p.token)?;
let users = plex.get_users().await;
for user in users {
println!("{:?}", user);
}
}
// let history = match plex.get_history().await {
// Ok(history) => history,
// Err(e) => {
Expand Down
Loading

0 comments on commit f6c377a

Please sign in to comment.