From 2b824a268b7a96bb9664cf5b1f22074401a8d93b Mon Sep 17 00:00:00 2001
From: phoenix <kundeng00@pm.me>
Date: Tue, 21 Oct 2025 13:14:31 -0400
Subject: [PATCH] Using file_type passed when getting the CoverArtQueue

---
 src/api.rs  |  1 +
 src/main.rs | 20 +++++++++++++-------
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/src/api.rs b/src/api.rs
index 7beaec0..ae4d756 100644
--- a/src/api.rs
+++ b/src/api.rs
@@ -158,6 +158,7 @@ pub mod get_coverart_queue {
         #[derive(Debug, Deserialize, Serialize)]
         pub struct CoverArtQueue {
             pub id: uuid::Uuid,
+            pub file_type: String,
             pub song_queue_id: uuid::Uuid,
         }
 
diff --git a/src/main.rs b/src/main.rs
index 5190565..5bb917b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -341,13 +341,14 @@ async fn prep_song(
                                         Ok(response) => {
                                             match response.json::<api::get_coverart_queue::response::Response>().await {
                                                 Ok(response) => {
-                                                    let coverart_queue_id = &response.data[0].id;
+                                                    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 {
+                                                    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().await;
+                                                                let (directory, filename) = generate_coverart_queue_dir_and_filename(&coverart_queue.file_type).await;
                                                                 let coverart = icarus_models::coverart::CoverArt {
                                                                     directory,
                                                                     filename,
@@ -367,7 +368,7 @@ async fn prep_song(
                                                                 let coverart_queue_path = std::path::Path::new(&coverart_queue_fs_path);
                                                                 println!("Saved coverart queue file at: {coverart_queue_path:?}");
 
-                                                                Ok(((song.directory, song.filename), (coverart.directory, coverart.filename), metadata.clone(), *coverart_queue_id))
+                                                                Ok(((song.directory, song.filename), (coverart.directory, coverart.filename), metadata.clone(), coverart_queue_id))
                                                             }
                                                             Err(err) => {
                                                                 Err(err)
@@ -413,7 +414,7 @@ pub async fn generate_song_queue_dir_and_filename() -> (String, String) {
 }
 
 // TODO: Consider having something like this in icarus_models
-pub async fn generate_coverart_queue_dir_and_filename() -> (String, String) {
+pub async fn generate_coverart_queue_dir_and_filename(file_type: &str) -> (String, String) {
     use rand::Rng;
 
     let mut filename: String = String::new();
@@ -432,8 +433,13 @@ pub async fn generate_coverart_queue_dir_and_filename() -> (String, String) {
         }
     }
 
-    // TODO: Do not hard code the file extension
-    filename += ".jpeg";
+    filename += if file_type == "jpeg" || file_type == "jpg" {
+        ".jpeg"
+    } else if file_type == "png" {
+        ".png"
+    } else {
+        ""
+    };
 
     // TODO: Consider separating song and coverart when saving to the filesystem
     let directory = icarus_envy::environment::get_root_directory().await.value;