Get queued coverart data (#26)

Reviewed-on: #26
Co-authored-by: kdeng00 <kundeng00@pm.me>
Co-committed-by: kdeng00 <kundeng00@pm.me>

Cargo.lock

@@ -83,6 +83,26 @@ version = "1.0.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
 
+[[package]]
+name = "const_format"
+version = "0.2.34"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "126f97965c8ad46d6d9163268ff28432e8f6a1196a55578867832e3049df63dd"
+dependencies = [
+ "const_format_proc_macros",
+]
+
+[[package]]
+name = "const_format_proc_macros"
+version = "0.2.34"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1d57c2eccfb16dbac1f4e61e206105db5820c9d26c3c472bc17c774259ef7744"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-xid",
+]
+
 [[package]]
 name = "core-foundation"
 version = "0.9.4"
@@ -450,9 +470,10 @@ dependencies = [
 
 [[package]]
 name = "icarus_envy"
-version = "0.2.0"
-source = "git+ssh://git@git.kundeng.us/phoenix/icarus_envy.git?tag=v0.2.0#3ee8c1e573ba9637769aa0538d2c11335e39ed9f"
+version = "0.2.2"
+source = "git+ssh://git@git.kundeng.us/phoenix/icarus_envy.git?tag=v0.2.2-devel-84ea6e4c22-006#84ea6e4c223f19ff96f6b7cae944e831ef4aa831"
 dependencies = [
+ "const_format",
  "dotenvy",
 ]
 
@@ -1148,13 +1169,14 @@ dependencies = [
 name = "songparser"
 version = "0.1.0"
 dependencies = [
- "dotenvy",
  "futures",
  "icarus_envy",
  "icarus_models",
+ "rand",
  "reqwest",
  "serde",
  "serde_json",
+ "time",
  "tokio",
  "uuid",
 ]
@@ -1415,6 +1437,12 @@ version = "1.0.18"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "5a5f39404a5da50712a4c1eecf25e90dd62b613502b7e925fd4e4d19b5c96512"
 
+[[package]]
+name = "unicode-xid"
+version = "0.2.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ebc1c04c71510c7f702b52b7c350734c9ff1295c464a03335b00bb84fc54f853"
+
 [[package]]
 name = "untrusted"
 version = "0.9.0"

Cargo.toml

@@ -10,7 +10,8 @@ futures = { version = "0.3.31" }
 reqwest = { version = "0.12.19", features = ["json", "stream"] }
 serde = { version = "1.0.218", features = ["derive"] }
 serde_json = { version = "1.0.139" }
+time = { version = "0.3.41", features = ["macros", "serde"] }
 uuid = { version = "1.16.0", features = ["v4", "serde"] }
-dotenvy = { version = "0.15.7" }
+rand = { version = "0.9" }
 icarus_models = { git = "ssh://git@git.kundeng.us/phoenix/icarus_models.git", tag = "v0.4.3" }
-icarus_envy = { git = "ssh://git@git.kundeng.us/phoenix/icarus_envy.git", tag = "v0.2.0" }
+icarus_envy = { git = "ssh://git@git.kundeng.us/phoenix/icarus_envy.git", tag = "v0.2.2-devel-84ea6e4c22-006" }

src/main.rs

@@ -4,67 +4,27 @@ pub const SECONDS_TO_SLEEP: u64 = 5;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
-    let app_base_url = get_icarus_url().await;
+    let app_base_url = icarus_envy::environment::get_icarus_base_api_url().await;
 
     loop {
         println!("Base URL: {}", app_base_url);
 
-        match api::fetch_next_queue_item(&app_base_url).await {
-            Ok(response) => {
-                match response
-                    .json::<responses::fetch_next_queue_item::SongQueueItem>()
-                    .await
-                {
-                    Ok(song_queue_item) => {
-                        if !song_queue_item.data.is_empty() {
-                            println!("Song queue item: {:?}", song_queue_item);
+        match is_queue_empty(&app_base_url).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;
 
-                            println!("Fetching song queue data");
-                            match api::fetch_song_queue_data::get_data(
-                                &app_base_url,
-                                &song_queue_item.data[0].id,
-                            )
-                            .await
-                            {
-                                Ok(response) => {
-                                    // Process data here...
-                                    let all_bytes =
-                                        api::fetch_song_queue_data::response::parse_response(
-                                            response,
-                                        )
-                                        .await?;
-
-                                    let (directory, filename) =
-                                        generate_song_queue_dir_and_filename().await;
-                                    let save_path =
-                                        save_song_to_fs(&directory, &filename, &all_bytes).await;
-
-                                    println!("Saved at: {:?}", save_path);
-
-                                    // TODO: Get queued song's metadata
-                                    // TODO: Get queued coverart
-                                    // TODO: Get queued coverart's data
-                                    // TODO: Apply metadata to the queued song
-                                    // TODO: Update the queued song with the updated queued song
-                                    // TODO: Create song
-                                    // TODO: Create coverart
-                                    // TODO: Wipe data from queued song
-                                    // TODO: Wipe data from queued coverart
-                                }
-                                Err(err) => {
-                                    eprintln!("Error fetching song queue data: {:?}", err);
-                                }
-                            }
-                        } else {
-                            println!("No data to fetch");
-                        }
-                    }
-                    Err(err) => {
-                        eprintln!("Error: {:?}", err);
-                    }
+                    // TODO: Do something with the result later
+                    let _ = process_song(&app_base_url, &song_queue_id).await;
+                } else {
+                    println!("Queue is empty");
                 }
             }
-            Err(e) => eprintln!("API call failed: {}", e),
+            Err(err) => {
+                eprintln!("Error checking if queue is empty: {:?}", err);
+            }
         }
 
         println!("Sleeping");
@@ -72,6 +32,117 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     }
 }
 
+async fn is_queue_empty(
+    api_url: &String,
+) -> Result<(bool, responses::fetch_next_queue_item::SongQueueItem), reqwest::Error> {
+    match api::fetch_next_queue_item(api_url).await {
+        Ok(response) => {
+            match response
+                .json::<responses::fetch_next_queue_item::SongQueueItem>()
+                .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 process_song(api_url: &String, song_queue_id: &uuid::Uuid) -> Result<(), reqwest::Error> {
+    match api::fetch_song_queue_data::get_data(api_url, song_queue_id).await {
+        Ok(response) => {
+            // Process data here...
+            match api::parsing::parse_response_into_bytes(response).await {
+                Ok(song_bytes) => {
+                    let (directory, filename) = generate_song_queue_dir_and_filename().await;
+                    let save_path = save_file_to_fs(&directory, &filename, &song_bytes).await;
+
+                    println!("Saved at: {:?}", save_path);
+
+                    match api::get_metadata_queue::get(api_url, song_queue_id).await {
+                        Ok(response) => {
+                            match response
+                                .json::<api::get_metadata_queue::response::Response>()
+                                .await
+                            {
+                                Ok(response) => {
+                                    let id = &response.data[0].id;
+                                    let metadata = &response.data[0].metadata;
+                                    let created_at = &response.data[0].created_at;
+                                    println!("Id: {:?}", id);
+                                    println!("Metadata: {:?}", metadata);
+                                    println!("Created at: {:?}", created_at);
+
+                                    println!("Getting coverart queue");
+                                    match api::get_coverart_queue::get(api_url, song_queue_id).await
+                                    {
+                                        Ok(response) => {
+                                            match response.json::<api::get_coverart_queue::response::Response>().await {
+                                                Ok(response) => {
+                                                    let coverart_queue_id = &response.data[0].id;
+                                                    println!("Coverart queue Id: {:?}", coverart_queue_id);
+
+                                                    match api::get_coverart_queue::get_data(api_url, 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().await;
+                                                                let save_path = save_file_to_fs(&directory, &filename, &coverart_queue_bytes).await;
+
+                                                                println!("Saved coverart queue file at: {:?}", save_path);
+
+                                                                // TODO: Apply metadata to the queued song (modifying file)
+                                                                // TODO: Update the queued song with the updated queued song
+                                                                // TODO: Create song
+                                                                // TODO: Create coverart
+                                                                // TODO: Wipe data from queued song
+                                                                // TODO: Wipe data from queued coverart
+                                                            }
+                                                            Err(err) => {
+                                                                eprintln!("Error: {:?}", err);
+                                                            }
+                                                        }
+                                                        Err(err) => {
+                                                            eprintln!("Error: {:?}", err);
+                                                        }
+                                                    }
+                                                }
+                                                Err(err) => {
+                                                    eprintln!("Error: {:?}", err);
+                                                }
+                                            }
+                                        }
+                                        Err(err) => {
+                                            eprintln!("Error: {:?}", err);
+                                        }
+                                    }
+                                    Ok(())
+                                }
+                                Err(err) => {
+                                    eprintln!("Error: {:?}", err);
+                                    Err(err)
+                                }
+                            }
+                        }
+                        Err(err) => {
+                            eprintln!("Error: {:?}", err);
+                            Err(err)
+                        }
+                    }
+                }
+                Err(err) => Err(err),
+            }
+        }
+        Err(err) => Err(err),
+    }
+}
+
 // TODO: Consider having something like this in icarus_models
 pub async fn generate_song_queue_dir_and_filename() -> (String, String) {
     let mut song = icarus_models::song::Song::default();
@@ -82,8 +153,37 @@ pub async fn generate_song_queue_dir_and_filename() -> (String, String) {
     (song.directory, song.filename)
 }
 
+// TODO: Consider having something like this in icarus_models
+pub async fn generate_coverart_queue_dir_and_filename() -> (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 _i 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);
+        }
+    }
+
+    // TODO: Do not hard code the file extension
+    filename += ".jpeg";
+
+    // TODO: Consider separating song and coverart when saving to the filesystem
+    let directory = icarus_envy::environment::get_root_directory().await;
+
+    (directory, filename)
+}
+
 // TODO: Check to see if this is available in icarus_models
-pub async fn save_song_to_fs(
+pub async fn save_file_to_fs(
     directory: &String,
     filename: &String,
     data: &[u8],
@@ -128,6 +228,26 @@ mod api {
         client.get(api_url).send().await
     }
 
+    pub mod parsing {
+        use futures::StreamExt;
+
+        pub async fn parse_response_into_bytes(
+            response: reqwest::Response,
+        ) -> Result<Vec<u8>, reqwest::Error> {
+            // TODO: At some point, handle the flow if the size is small or
+            // large
+            let mut byte_stream = response.bytes_stream();
+            let mut all_bytes = Vec::new();
+
+            while let Some(chunk) = byte_stream.next().await {
+                let chunk = chunk?;
+                all_bytes.extend_from_slice(&chunk);
+            }
+
+            Ok(all_bytes)
+        }
+    }
+
     pub mod fetch_song_queue_data {
         pub async fn get_data(
             base_url: &String,
@@ -138,30 +258,98 @@ mod api {
             let api_url = format!("{}/{}/{}", base_url, endpoint, id);
             client.get(api_url).send().await
         }
+    }
+
+    pub mod get_metadata_queue {
+        pub async fn get(
+            base_url: &String,
+            song_queue_id: &uuid::Uuid,
+        ) -> Result<reqwest::Response, reqwest::Error> {
+            let client = reqwest::Client::new();
+            let endpoint = String::from("api/v2/song/metadata/queue");
+            let api_url = format!("{}/{}", base_url, endpoint);
+            client
+                .get(api_url)
+                .query(&[("song_queue_id", song_queue_id)])
+                .send()
+                .await
+        }
 
         pub mod response {
-            use futures::StreamExt;
+            use serde::{Deserialize, Serialize};
 
-            pub async fn parse_response(
-                response: reqwest::Response,
-            ) -> Result<Vec<u8>, reqwest::Error> {
-                // TODO: At some point, handle the flow if the size is small or
-                // large
-                let mut byte_stream = response.bytes_stream();
-                let mut all_bytes = Vec::new();
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct Metadata {
+                pub song_queue_id: uuid::Uuid,
+                pub album: String,
+                pub album_artist: String,
+                pub artist: String,
+                pub disc: i32,
+                pub disc_count: i32,
+                pub duration: i64,
+                pub genre: String,
+                pub title: String,
+                pub track: i32,
+                pub track_count: i32,
+                pub year: i32,
+            }
 
-                while let Some(chunk) = byte_stream.next().await {
-                    let chunk = chunk?;
-                    all_bytes.extend_from_slice(&chunk);
-                }
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct QueueItem {
+                pub id: uuid::Uuid,
+                pub metadata: Metadata,
+                #[serde(with = "time::serde::rfc3339")]
+                pub created_at: time::OffsetDateTime,
+                pub song_queue_id: uuid::Uuid,
+            }
 
-                Ok(all_bytes)
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct Response {
+                pub message: String,
+                pub data: Vec<QueueItem>,
+            }
+        }
+    }
+
+    pub mod get_coverart_queue {
+        pub async fn get(
+            base_url: &String,
+            song_queue_id: &uuid::Uuid,
+        ) -> Result<reqwest::Response, reqwest::Error> {
+            let client = reqwest::Client::new();
+            let endpoint = String::from("api/v2/coverart/queue");
+            let api_url = format!("{}/{}", base_url, endpoint);
+            client
+                .get(api_url)
+                .query(&[("song_queue_id", song_queue_id)])
+                .send()
+                .await
+        }
+
+        pub async fn get_data(
+            base_url: &String,
+            coverart_queue_id: &uuid::Uuid,
+        ) -> Result<reqwest::Response, reqwest::Error> {
+            let client = reqwest::Client::new();
+            let endpoint = String::from("api/v2/coverart/queue/data");
+            let api_url = format!("{}/{}/{}", base_url, endpoint, coverart_queue_id);
+            client.get(api_url).send().await
+        }
+
+        pub mod response {
+            use serde::{Deserialize, Serialize};
+
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct CoverArtQueue {
+                pub id: uuid::Uuid,
+                pub song_queue_id: uuid::Uuid,
+            }
+
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct Response {
+                pub message: String,
+                pub data: Vec<CoverArtQueue>,
             }
         }
     }
 }
-
-async fn get_icarus_url() -> String {
-    dotenvy::dotenv().ok();
-    std::env::var("ICARUS_BASE_API_URL").expect("Could not find url")
-}