phoenix/icarus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 33 Wiki Activity

Formatting

Browse Source
This commit is contained in:
kdeng00
2025-06-17 21:33:32 -04:00
parent c22f522bad
commit 60930ca10c
2 changed files with 37 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/callers/coverart.rs
+11 -8
View File
@@ -450,10 +450,7 @@ pub mod endpoint {
pub async fn fetch_coverart_with_data(
axum::Extension(pool): axum::Extension<sqlx::PgPool>,
axum::extract::Path(id): axum::extract::Path<uuid::Uuid>,
) -> (
axum::http::StatusCode,
axum::response::Response,
) {
) -> (axum::http::StatusCode, axum::response::Response) {
match super::db::get_coverart_queue_data_with_id(&pool, &id).await {
Ok(data) => {
let bytes = axum::body::Bytes::from(data);
@@ -462,13 +459,19 @@ pub mod endpoint {
// TODO: Address this hard coding for the coverart content type
headers.insert(axum::http::header::CONTENT_TYPE, "image".parse().unwrap());
// TODO: Make the conent disposition more dynamic
headers.insert(axum::http::header::CONTENT_DISPOSITION, format!("attachment; filename=\"{}.jpg\"", id).parse().unwrap());
headers.insert(
axum::http::header::CONTENT_DISPOSITION,
format!("attachment; filename=\"{}.jpg\"", id)
.parse()
.unwrap(),
);
(axum::http::StatusCode::OK, response)
}
Err(_err) => {
(axum::http::StatusCode::BAD_REQUEST, axum::response::Response::default())
}
Err(_err) => (
axum::http::StatusCode::BAD_REQUEST,
axum::response::Response::default(),
),
}
}
src/main.rs
+26 -16
View File
@@ -1258,9 +1258,14 @@ mod tests {
"Should not be empty"
);
let raw_uri = String::from(crate::callers::endpoints::QUEUECOVERARTDATA);
let raw_uri =
String::from(crate::callers::endpoints::QUEUECOVERARTDATA);
let end_index = raw_uri.len() - 5;
let uri = format!("{}/{}", (&raw_uri[..end_index]).to_string(), resp_coverart_id);
let uri = format!(
"{}/{}",
(&raw_uri[..end_index]).to_string(),
resp_coverart_id
);
println!("Uri: {:?}", uri);
match app
@@ -1275,22 +1280,27 @@ mod tests {
)
.await
{
Ok(response) => {
match resp_to_bytes(response).await {
Ok(bytes) => {
assert_eq!(false, bytes.is_empty(), "Downloaded coverart data should not be empty");
let temp_file = tempfile::tempdir().expect("Could not create test directory");
let test_dir = String::from(temp_file.path().to_str().unwrap());
let new_file = format!("{}/new_image.jpeg", test_dir);
Ok(response) => match resp_to_bytes(response).await {
Ok(bytes) => {
assert_eq!(
false,
bytes.is_empty(),
"Downloaded coverart data should not be empty"
);
let temp_file = tempfile::tempdir()
.expect("Could not create test directory");
let test_dir =
String::from(temp_file.path().to_str().unwrap());
let new_file = format!("{}/new_image.jpeg", test_dir);
let mut file = std::fs::File::create(&new_file).unwrap();
file.write_all(&bytes).unwrap();
}
Err(err) => {
assert!(false, "Error: {:?}", err);
}
let mut file =
std::fs::File::create(&new_file).unwrap();
file.write_all(&bytes).unwrap();
}
}
Err(err) => {
assert!(false, "Error: {:?}", err);
}
},
Err(err) => {
assert!(false, "Error: {:?}", err);
}