Apply metadata to queued song (#27)

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

Cargo.lock

@@ -62,6 +62,12 @@ version = "3.17.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "1628fb46dfa0b37568d12e5edd512553eccf6a22a78e8bde00bb4aed84d5bdbf"
 
+[[package]]
+name = "byteorder"
+version = "1.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b"
+
 [[package]]
 name = "bytes"
 version = "1.10.1"
@@ -119,6 +125,21 @@ version = "0.8.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "773648b94d0e5d620f64f280777445740e61fe701025087ec8b57f45c791888b"
 
+[[package]]
+name = "crc32fast"
+version = "1.4.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a97769d94ddab943e4510d138150169a2758b5ef3eb191a9ee688de3e23ef7b3"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "data-encoding"
+version = "2.9.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2a2330da5de22e8a3cb63252ce2abb30116bf5265e89c0e01bc17015ce30a476"
+
 [[package]]
 name = "deranged"
 version = "0.4.0"
@@ -177,6 +198,16 @@ version = "2.3.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "37909eebbb50d72f9059c3b6d82c0463f2ff062c9e95845c43a6c9c0355411be"
 
+[[package]]
+name = "flate2"
+version = "1.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4a3d7db9596fecd151c5f638c0ee5d5bd487b6e0ea232e5dc96d5250f6f94b1d"
+dependencies = [
+ "crc32fast",
+ "miniz_oxide",
+]
+
 [[package]]
 name = "fnv"
 version = "1.0.7"
@@ -477,6 +508,14 @@ dependencies = [
  "dotenvy",
 ]
 
+[[package]]
+name = "icarus_meta"
+version = "0.2.2"
+source = "git+ssh://git@git.kundeng.us/phoenix/icarus_meta.git?tag=v0.2.2-devel-565d361b64-680#565d361b64f23c2a0100b33fceeebacfad3331c4"
+dependencies = [
+ "lofty",
+]
+
 [[package]]
 name = "icarus_models"
 version = "0.4.3"
@@ -666,6 +705,32 @@ dependencies = [
  "scopeguard",
 ]
 
+[[package]]
+name = "lofty"
+version = "0.22.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ca260c51a9c71f823fbfd2e6fbc8eb2ee09834b98c00763d877ca8bfa85cde3e"
+dependencies = [
+ "byteorder",
+ "data-encoding",
+ "flate2",
+ "lofty_attr",
+ "log",
+ "ogg_pager",
+ "paste",
+]
+
+[[package]]
+name = "lofty_attr"
+version = "0.11.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ed9983e64b2358522f745c1251924e3ab7252d55637e80f6a0a3de642d6a9efc"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
 [[package]]
 name = "log"
 version = "0.4.27"
@@ -736,6 +801,15 @@ dependencies = [
  "memchr",
 ]
 
+[[package]]
+name = "ogg_pager"
+version = "0.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e034c10fb5c1c012c1b327b85df89fb0ef98ae66ec28af30f0d1eed804a40c19"
+dependencies = [
+ "byteorder",
+]
+
 [[package]]
 name = "once_cell"
 version = "1.21.3"
@@ -809,6 +883,12 @@ dependencies = [
  "windows-targets 0.52.6",
 ]
 
+[[package]]
+name = "paste"
+version = "1.0.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
+
 [[package]]
 name = "percent-encoding"
 version = "2.3.1"
@@ -1171,7 +1251,9 @@ version = "0.1.0"
 dependencies = [
  "futures",
  "icarus_envy",
+ "icarus_meta",
  "icarus_models",
+ "rand",
  "reqwest",
  "serde",
  "serde_json",

Cargo.toml

@@ -12,5 +12,7 @@ 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"] }
+rand = { version = "0.9" }
+icarus_meta = { git = "ssh://git@git.kundeng.us/phoenix/icarus_meta.git", tag = "v0.2.2-devel-565d361b64-680" }
 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

@@ -59,12 +59,12 @@ async fn process_song(api_url: &String, song_queue_id: &uuid::Uuid) -> Result<()
     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) => {
+            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_song_to_fs(&directory, &filename, &all_bytes).await;
+                    let song_queue_path = save_file_to_fs(&directory, &filename, &song_bytes).await;
 
-                    println!("Saved at: {:?}", save_path);
+                    println!("Saved at: {:?}", song_queue_path);
 
                     match api::get_metadata_queue::get(api_url, song_queue_id).await {
                         Ok(response) => {
@@ -88,13 +88,36 @@ async fn process_song(api_url: &String, song_queue_id: &uuid::Uuid) -> Result<()
                                                 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 (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
+
+                                                    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 coverart_queue_path = save_file_to_fs(&directory, &filename, &coverart_queue_bytes).await;
+
+                                                                println!("Saved coverart queue file at: {:?}", coverart_queue_path);
+
+                                                                match apply_metadata(song_queue_path, coverart_queue_path, metadata).await {
+                                                                    Ok(_) => {
+                                                                        // 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);
@@ -136,8 +159,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],
@@ -153,6 +205,161 @@ pub async fn save_song_to_fs(
     save_path
 }
 
+pub async fn apply_metadata(
+    song_queue_path: std::path::PathBuf,
+    coverart_queue_path: std::path::PathBuf,
+    metadata: &api::get_metadata_queue::response::Metadata,
+) -> Result<bool, std::io::Error> {
+    // Apply metadata fields
+    let s_path = match song_queue_path.to_str() {
+        Some(val) => String::from(val),
+        None => String::new(),
+    };
+
+    if s_path.is_empty() {
+        println!("Song queue path is empty");
+        return Ok(false);
+    }
+
+    let types = icarus_meta::types::all_metadata_types();
+
+    for t in types {
+        match t {
+            icarus_meta::types::Type::Album => {
+                let meta_type =
+                    icarus_meta::types::MetadataType::from_string(metadata.album.clone());
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::AlbumArtist => {
+                let meta_type =
+                    icarus_meta::types::MetadataType::from_string(metadata.album_artist.clone());
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::Artist => {
+                let meta_type =
+                    icarus_meta::types::MetadataType::from_string(metadata.artist.clone());
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::Date => {
+                // TODO: Do something about this discrepancy
+                let meta_type =
+                    icarus_meta::types::MetadataType::from_string(metadata.year.to_string());
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::Disc => {
+                let meta_type = icarus_meta::types::MetadataType::from_int(metadata.disc);
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::Genre => {
+                let meta_type =
+                    icarus_meta::types::MetadataType::from_string(metadata.genre.clone());
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::Title => {
+                let meta_type =
+                    icarus_meta::types::MetadataType::from_string(metadata.title.clone());
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::Track => {
+                let meta_type = icarus_meta::types::MetadataType::from_int(metadata.track);
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::TrackCount => {
+                let meta_type = icarus_meta::types::MetadataType::from_int(metadata.track_count);
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+            icarus_meta::types::Type::DiscCount => {
+                let meta_type = icarus_meta::types::MetadataType::from_int(metadata.disc_count);
+                match icarus_meta::meta::metadata::set_meta_value(t, &s_path, meta_type) {
+                    Ok(_) => {}
+                    Err(_err) => {
+                        return Err(_err);
+                    }
+                }
+            }
+        }
+    }
+
+    // Apply coverart
+    let c_path: String = match coverart_queue_path.to_str() {
+        Some(val) => String::from(val),
+        None => String::new(),
+    };
+
+    match icarus_meta::meta::coverart::contains_coverart(&s_path) {
+        Ok((exists, size)) => {
+            if exists {
+                println!("Coverart exists: {:?} size", size);
+                match icarus_meta::meta::coverart::remove_coverart(&s_path) {
+                    Ok(_data) => {}
+                    Err(err) => {
+                        return Err(err);
+                    }
+                }
+            }
+
+            match icarus_meta::meta::coverart::set_coverart(&s_path, &c_path) {
+                Ok(_data) => {
+                    if _data.is_empty() {
+                        println!("There was an issue");
+                        Ok(false)
+                    } else {
+                        println!("Success in applying coverart to song");
+                        Ok(true)
+                    }
+                }
+                Err(err) => Err(err),
+            }
+        }
+        Err(err) => Err(err),
+    }
+}
+
 mod responses {
     pub mod fetch_next_queue_item {
         use serde::{Deserialize, Serialize};
@@ -182,6 +389,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,
@@ -192,26 +419,6 @@ mod api {
             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 {
@@ -234,7 +441,7 @@ mod api {
 
             #[derive(Debug, Deserialize, Serialize)]
             pub struct Metadata {
-                pub id: uuid::Uuid,
+                pub song_queue_id: uuid::Uuid,
                 pub album: String,
                 pub album_artist: String,
                 pub artist: String,
@@ -280,6 +487,16 @@ mod api {
                 .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};