From 9f9abf4d9fff8d6f34dfdf3753e772c24aab94ba Mon Sep 17 00:00:00 2001
From: kdeng00 <kundeng00@pm.me>
Date: Fri, 31 Oct 2025 12:24:55 -0400
Subject: [PATCH] tsk-201: Making content disposition more unique

---
 src/callers/queue/coverart.rs | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/callers/queue/coverart.rs b/src/callers/queue/coverart.rs
index 2ca3748..80bd30d 100644
--- a/src/callers/queue/coverart.rs
+++ b/src/callers/queue/coverart.rs
@@ -328,9 +328,28 @@ pub mod endpoint {
                     file_type.mime.parse().unwrap(),
                 );
                 // TODO: Make the conent disposition more dynamic
+                let coverart_type = if file_type.file_type
+                    == icarus_meta::detection::coverart::constants::JPEG_TYPE
+                {
+                    icarus_models::types::CoverArtType::JpegExtension
+                } else if file_type.file_type
+                    == icarus_meta::detection::coverart::constants::JPG_TYPE
+                {
+                    icarus_models::types::CoverArtType::JpgExtension
+                } else if file_type.file_type
+                    == icarus_meta::detection::coverart::constants::PNG_TYPE
+                {
+                    icarus_models::types::CoverArtType::PngExtension
+                } else {
+                    return (
+                        axum::http::StatusCode::INTERNAL_SERVER_ERROR,
+                        axum::response::Response::default()
+                    );
+                };
+                let filename = icarus_models::coverart::generate_filename(coverart_type, true).unwrap();
                 headers.insert(
                     axum::http::header::CONTENT_DISPOSITION,
-                    format!("attachment; filename=\"{id}.jpg\"")
+                    format!("attachment; filename=\"{filename}\"")
                         .parse()
                         .unwrap(),
                 );