Get queued coverart (#25)

Reviewed-on: #25
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"
@@ -187,6 +207,21 @@ dependencies = [
  "percent-encoding",
 ]
 
+[[package]]
+name = "futures"
+version = "0.3.31"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "65bc07b1a8bc7c85c5f2e110c476c7389b4554ba72af57d8445ea63a576b0876"
+dependencies = [
+ "futures-channel",
+ "futures-core",
+ "futures-executor",
+ "futures-io",
+ "futures-sink",
+ "futures-task",
+ "futures-util",
+]
+
 [[package]]
 name = "futures-channel"
 version = "0.3.31"
@@ -194,6 +229,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "2dff15bf788c671c1934e366d07e30c1814a8ef514e1af724a602e8a2fbe1b10"
 dependencies = [
  "futures-core",
+ "futures-sink",
 ]
 
 [[package]]
@@ -202,6 +238,34 @@ version = "0.3.31"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "05f29059c0c2090612e8d742178b0580d2dc940c837851ad723096f87af6663e"
 
+[[package]]
+name = "futures-executor"
+version = "0.3.31"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1e28d1d997f585e54aebc3f97d39e72338912123a67330d723fdbb564d646c9f"
+dependencies = [
+ "futures-core",
+ "futures-task",
+ "futures-util",
+]
+
+[[package]]
+name = "futures-io"
+version = "0.3.31"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9e5c1b78ca4aae1ac06c48a526a655760685149f0d465d21f37abfe57ce075c6"
+
+[[package]]
+name = "futures-macro"
+version = "0.3.31"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "162ee34ebcb7c64a8abebc059ce0fee27c2262618d7b60ed8faf72fef13c3650"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
 [[package]]
 name = "futures-sink"
 version = "0.3.31"
@@ -220,10 +284,16 @@ version = "0.3.31"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "9fa08315bb612088cc391249efdc3bc77536f16c91f6cf495e6fbe85b20a4a81"
 dependencies = [
+ "futures-channel",
  "futures-core",
+ "futures-io",
+ "futures-macro",
+ "futures-sink",
  "futures-task",
+ "memchr",
  "pin-project-lite",
  "pin-utils",
+ "slab",
 ]
 
 [[package]]
@@ -398,6 +468,15 @@ dependencies = [
  "windows-registry",
 ]
 
+[[package]]
+name = "icarus_envy"
+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",
+]
+
 [[package]]
 name = "icarus_models"
 version = "0.4.3"
@@ -851,6 +930,7 @@ dependencies = [
  "bytes",
  "encoding_rs",
  "futures-core",
+ "futures-util",
  "h2",
  "http",
  "http-body",
@@ -874,12 +954,14 @@ dependencies = [
  "sync_wrapper",
  "tokio",
  "tokio-native-tls",
+ "tokio-util",
  "tower",
  "tower-http",
  "tower-service",
  "url",
  "wasm-bindgen",
  "wasm-bindgen-futures",
+ "wasm-streams",
  "web-sys",
 ]
 
@@ -1087,10 +1169,15 @@ dependencies = [
 name = "songparser"
 version = "0.1.0"
 dependencies = [
- "dotenvy",
+ "futures",
+ "icarus_envy",
  "icarus_models",
  "reqwest",
+ "serde",
+ "serde_json",
+ "time",
  "tokio",
+ "uuid",
 ]
 
 [[package]]
@@ -1349,6 +1436,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"
@@ -1485,6 +1578,19 @@ dependencies = [
  "unicode-ident",
 ]
 
+[[package]]
+name = "wasm-streams"
+version = "0.4.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "15053d8d85c7eccdbefef60f06769760a563c7f0a9d6902a13d35c7800b0ad65"
+dependencies = [
+ "futures-util",
+ "js-sys",
+ "wasm-bindgen",
+ "wasm-bindgen-futures",
+ "web-sys",
+]
+
 [[package]]
 name = "web-sys"
 version = "0.3.77"

Cargo.toml

@@ -6,6 +6,11 @@ rust-version = "1.86"
 
 [dependencies]
 tokio = { version = "1.44.1", features = ["full"] }
-reqwest = { version = "0.12.19" }
-dotenvy = { version = "0.15.7" }
+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"] }
 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.2-devel-84ea6e4c22-006" }

src/main.rs

@@ -1,49 +1,299 @@
-// use std::error::Error;
-// use tokio::io::AsyncReadExt;
-// use tokio::net::{TcpListener, TcpStream};
-// use tokio::spawn;
+use std::io::Write;
+
+pub const SECONDS_TO_SLEEP: u64 = 5;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
-    let client = reqwest::Client::new();
-    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);
 
-        // TODO: Update the api/v2/song/queue/next endpoint to only retrieve queued song that is
-        // ready to be processed. Make necessary changes to other endpoints
+        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;
 
-        let api_url = format!("{}/api/v2/song/queue/next", app_base_url);
+                    // TODO: Do something with the result later
+                    let _ = process_song(&app_base_url, &song_queue_id).await;
+                } else {
+                    println!("Queue is empty");
+                }
+            }
+            Err(err) => {
+                eprintln!("Error checking if queue is empty: {:?}", err);
+            }
+        }
 
-        match client.get(api_url).send().await {
+        println!("Sleeping");
+        tokio::time::sleep(tokio::time::Duration::from_secs(SECONDS_TO_SLEEP)).await;
+    }
+}
+
+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) => {
-                let body = response.text().await?;
-                println!("API response: {}", body);
-                // Process data here...
+            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),
+    }
+}
 
-                // TODO: Parse the response body to a struct
-                // TODO: Get queued song data
-                // TODO: Get queued song's metadata
-                // TODO: Get queued coverart
+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::fetch_song_queue_data::response::parse_response(response).await {
+                Ok(all_bytes) => {
+                    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);
+
+                    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);
                                                     // TODO: Get queued coverart's data
-                // TODO: Apply metadata to the queued song
+                                                    // 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(e) => eprintln!("API call failed: {}", e),
+                                                Err(err) => {
+                                                    eprintln!("Error: {:?}", err);
                                                 }
-
-        println!("Sleeping");
-        tokio::time::sleep(tokio::time::Duration::from_secs(5)).await;
                                             }
-    // Ok(())
+                                        }
+                                        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),
+    }
 }
 
-async fn get_icarus_url() -> String {
-    dotenvy::dotenv().ok();
-    std::env::var("ICARUS_BASE_API_URL").expect("Could not find url")
+// 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();
+    song.filename = song.generate_filename(icarus_models::types::MusicTypes::FlacExtension, true);
+
+    song.directory = icarus_envy::environment::get_root_directory().await;
+
+    (song.directory, song.filename)
+}
+
+// TODO: Check to see if this is available in icarus_models
+pub async fn save_song_to_fs(
+    directory: &String,
+    filename: &String,
+    data: &[u8],
+) -> std::path::PathBuf {
+    // TODO: Add function to save bytes to a file in icarus_models
+    // repo
+    let dir = std::path::Path::new(directory);
+    let save_path = dir.join(filename);
+
+    let mut file = std::fs::File::create(&save_path).unwrap();
+    file.write_all(data).unwrap();
+
+    save_path
+}
+
+mod responses {
+    pub mod fetch_next_queue_item {
+        use serde::{Deserialize, Serialize};
+
+        #[derive(Debug, Deserialize, Serialize)]
+        pub struct QueueItem {
+            pub id: uuid::Uuid,
+            pub filename: String,
+            pub status: String,
+        }
+
+        #[derive(Debug, Deserialize, Serialize)]
+        pub struct SongQueueItem {
+            pub message: String,
+            pub data: Vec<QueueItem>,
+        }
+    }
+}
+
+mod api {
+    pub async fn fetch_next_queue_item(
+        base_url: &String,
+    ) -> Result<reqwest::Response, reqwest::Error> {
+        let client = reqwest::Client::new();
+        let fetch_endpoint = String::from("api/v2/song/queue/next");
+        let api_url = format!("{}/{}", base_url, fetch_endpoint);
+        client.get(api_url).send().await
+    }
+
+    pub mod fetch_song_queue_data {
+        pub async fn get_data(
+            base_url: &String,
+            id: &uuid::Uuid,
+        ) -> Result<reqwest::Response, reqwest::Error> {
+            let client = reqwest::Client::new();
+            let endpoint = String::from("api/v2/song/queue");
+            let api_url = format!("{}/{}/{}", base_url, endpoint, id);
+            client.get(api_url).send().await
+        }
+
+        pub mod response {
+            use futures::StreamExt;
+
+            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();
+
+                while let Some(chunk) = byte_stream.next().await {
+                    let chunk = chunk?;
+                    all_bytes.extend_from_slice(&chunk);
+                }
+
+                Ok(all_bytes)
+            }
+        }
+    }
+
+    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 serde::{Deserialize, Serialize};
+
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct Metadata {
+                pub 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,
+            }
+
+            #[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,
+            }
+
+            #[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 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>,
+            }
+        }
+    }
 }