pub mod api; pub mod auth; pub mod config; pub mod metadata; pub mod queued_item; pub mod util; pub const SECONDS_TO_SLEEP: u64 = 5; #[tokio::main] async fn main() -> Result<(), Box> { let mut app = config::App { uri: icarus_envy::environment::get_icarus_base_api_url() .await .value, auth_uri: icarus_envy::environment::get_icarus_auth_base_api_url() .await .value, ..Default::default() }; println!("Base URL: {:?}", app.uri); println!("Auth URL: {:?}", app.auth_uri); match auth::get_token(&app).await { Ok(login_result) => { app.token = login_result; } Err(err) => { eprintln!("Error: {err:?}"); std::process::exit(-1); } }; loop { println!("Token: {:?}", app.token); if app.token.token_expired() { println!("Token expired"); app.token = match auth::get_refresh_token(&app).await { Ok(login_result) => login_result, Err(err) => { eprintln!("Error: {err:?}"); continue; } }; println!("Token refreshed"); println!("Refreshed token: {:?}", app.token); } else { println!("Token did not expire"); } match is_queue_empty(&app).await { Ok((empty, song_queue_item)) => { if !empty { println!("Queue is not empty"); println!("SongQueueItem: {song_queue_item:?}"); let song_queue_id = song_queue_item.data[0].id; let user_id = song_queue_item.data[0].user_id; match some_work(&app, &song_queue_id, &user_id).await { Ok((song, coverart, _metadata, queued_song, queued_coverart)) => { match wipe_data_from_queues(&app, &queued_song, &queued_coverart).await { Ok(_) => match cleanup(&song, &coverart).await { Ok(_) => { println!("Successful cleanup"); } Err(err) => { eprintln!("Error: {err:?}"); } }, Err(err) => { eprintln!("Error: {err:?}"); } } } Err(err) => { eprintln!("Error: {err:?}"); } } } else { println!("Queue is empty"); } } Err(err) => { eprintln!("Error checking if queue is empty: {err:?}"); } } println!("Sleeping"); tokio::time::sleep(tokio::time::Duration::from_secs(SECONDS_TO_SLEEP)).await; } } async fn wipe_data_from_queues( app: &config::App, queued_song: &crate::queued_item::QueuedSong, queued_coverart: &crate::queued_item::QueuedCoverArt, ) -> Result<(), std::io::Error> { match api::wipe_data::song_queue::wipe_data(app, queued_song).await { Ok(response) => match response .json::() .await { Ok(_resp) => { match api::wipe_data::coverart_queue::wipe_data(app, queued_coverart).await { Ok(inner_response) => match inner_response .json::() .await { Ok(_inner_resp) => { println!("Wiped data from CoverArt queue"); println!("Resp: {_inner_resp:?}"); Ok(()) } Err(err) => Err(std::io::Error::other(err.to_string())), }, Err(err) => Err(std::io::Error::other(err.to_string())), } } Err(err) => Err(std::io::Error::other(err.to_string())), }, Err(err) => Err(std::io::Error::other(err.to_string())), } } async fn cleanup( song: &icarus_models::song::Song, coverart: &icarus_models::coverart::CoverArt, ) -> Result<(), std::io::Error> { match song.remove_from_filesystem() { Ok(_) => {} Err(err) => { eprintln!("Error: Problem cleaning up SongQueue files {err:?}"); } } match coverart.remove_from_filesystem() { Ok(_) => Ok(()), Err(err) => Err(err), } } async fn is_queue_empty( app: &config::App, ) -> Result<(bool, api::fetch_next_queue_item::response::SongQueueItem), reqwest::Error> { match api::fetch_next_queue_item::fetch_next_queue_item(app).await { Ok(response) => { match response .json::() .await { Ok(response) => { if response.data.is_empty() { Ok((true, response)) } else { Ok((false, response)) } } Err(err) => Err(err), } } Err(err) => Err(err), } } async fn some_work( app: &crate::config::App, song_queue_id: &uuid::Uuid, user_id: &uuid::Uuid, ) -> Result< ( icarus_models::song::Song, icarus_models::coverart::CoverArt, api::get_metadata_queue::response::Metadata, queued_item::QueuedSong, queued_item::QueuedCoverArt, ), std::io::Error, > { match prep_song(app, song_queue_id).await { Ok((queued_song, queued_coverart, metadata)) => { println!("Prepping song"); match metadata::apply_metadata(&queued_song, &queued_coverart, &metadata).await { Ok(_applied) => { match api::update_queued_song::update_queued_song(app, &queued_song).await { Ok(response) => { match response .json::() .await { Ok(_inner_response) => { println!("Updated queued song"); println!("Response: {_inner_response:?}"); let song_type = String::from( icarus_meta::detection::song::constants::FLAC_TYPE, ); match api::create_song::create( app, &metadata, user_id, &song_type, ) .await { Ok(response) => match response .json::() .await { Ok(resp) => { println!("Response: {resp:?}"); let mut song = resp.data[0].clone(); song.directory = queued_song.song.directory.clone(); song.filename = queued_song.song.filename.clone(); match api::create_coverart::create(app, &song, &queued_coverart).await { Ok(response) => match response.json::().await { Ok(resp) => { println!("CoverArt sent and successfully parsed response"); println!("json: {resp:?}"); let mut coverart = resp.data[0].clone(); coverart.directory = queued_coverart.coverart.directory.clone(); coverart.filename = queued_coverart.coverart.filename.clone(); Ok((song.clone(), coverart.clone(), metadata, queued_song.clone(), queued_coverart.clone())) } Err(err) => { Err(std::io::Error::other(err.to_string())) } } Err(err) => { Err(std::io::Error::other(err.to_string())) } } } Err(err) => Err(std::io::Error::other(err.to_string())), }, Err(err) => Err(std::io::Error::other(err.to_string())), } } Err(err) => Err(std::io::Error::other(err.to_string())), } } Err(err) => Err(std::io::Error::other(err.to_string())), } } Err(err) => Err(err), } } Err(err) => Err(std::io::Error::other(err.to_string())), } } async fn prep_song( app: &crate::config::App, song_queue_id: &uuid::Uuid, ) -> Result< ( queued_item::QueuedSong, queued_item::QueuedCoverArt, api::get_metadata_queue::response::Metadata, ), reqwest::Error, > { match api::fetch_song_queue_data::get_data(app, song_queue_id).await { Ok(response) => { // Process data here... match api::parsing::parse_response_into_bytes(response).await { Ok(song_bytes) => { let song = icarus_models::song::Song { directory: icarus_envy::environment::get_root_directory().await.value, filename: icarus_models::song::generate_filename( icarus_models::types::MusicTypes::FlacExtension, true, ), data: song_bytes, ..Default::default() }; let songpath = song.song_path().unwrap_or_default(); let queued_song: crate::queued_item::QueuedSong = match song.save_to_filesystem() { Ok(_) => queued_item::QueuedSong { id: *song_queue_id, song, path: songpath, }, Err(err) => { eprintln!("Error: {err:?}"); queued_item::QueuedSong { ..Default::default() } } }; println!("Saved at: {:?}", queued_song.path); match api::get_metadata_queue::get(app, &queued_song.id).await { Ok(response) => { match response .json::() .await { Ok(response) => { let id = &response.data[0].id; let created_at = &response.data[0].created_at; let metadata = &response.data[0].metadata; println!("Id: {id:?}"); println!("Metadata: {metadata:?}"); println!("Created at: {created_at:?}"); println!("Getting coverart queue"); match api::get_coverart_queue::get(app, &queued_song.id).await { Ok(response) => { match response.json::().await { Ok(response) => { let coverart_queue = &response.data[0]; let coverart_queue_id = coverart_queue.id; println!("Coverart queue Id: {coverart_queue_id:?}"); match api::get_coverart_queue::get_data(app, &coverart_queue_id).await { Ok(response) => match api::parsing::parse_response_into_bytes(response).await { Ok(coverart_queue_bytes) => { let (directory, filename) = generate_coverart_queue_dir_and_filename(&coverart_queue.file_type).await; let coverart = icarus_models::coverart::CoverArt { directory, filename, data: coverart_queue_bytes, ..Default::default() }; coverart.save_to_filesystem().unwrap(); let coverart_queue_fs_path = match coverart.get_path() { Ok(path) => { path } Err(err) => { eprintln!("Error: {err:?}"); std::process::exit(-1); } }; let queued_coverart = queued_item::QueuedCoverArt { id: coverart_queue_id, coverart, path: coverart_queue_fs_path }; println!("Saved coverart queue file at: {:?}", queued_coverart.path); Ok((queued_song, queued_coverart, metadata.clone())) } Err(err) => { Err(err) } } Err(err) => { Err(err) } } } Err(err) => { Err(err) } } } Err(err) => Err(err), } } Err(err) => Err(err), } } Err(err) => Err(err), } } Err(err) => Err(err), } } Err(err) => Err(err), } } // TODO: Consider having something like this in icarus_models pub async fn generate_coverart_queue_dir_and_filename(file_type: &str) -> (String, String) { use rand::Rng; let mut filename: String = String::new(); let filename_len = 10; let some_chars: String = String::from("abcdefghij0123456789"); let mut rng = rand::rng(); for _ in 0..filename_len { let random_number: i32 = rng.random_range(0..=19); let index = random_number as usize; let rando_char = some_chars.chars().nth(index); if let Some(c) = rando_char { filename.push(c); } } filename += if file_type == icarus_meta::detection::coverart::constants::JPEG_TYPE || file_type == icarus_meta::detection::coverart::constants::JPG_TYPE { icarus_models::constants::file_extensions::image::JPEGEXTENSION } else if file_type == icarus_meta::detection::coverart::constants::PNG_TYPE { icarus_models::constants::file_extensions::image::PNGEXTENSION } else { "" }; // TODO: Consider separating song and coverart when saving to the filesystem let directory = icarus_envy::environment::get_root_directory().await.value; (directory, filename) }