tsk-213: Adding song check when queueing songs

This commit is contained in:
kdeng00
2025-10-29 14:36:19 -04:00
parent 1871ebe40b
commit 1bf0ed7e19
+29 -1
View File
@@ -96,6 +96,21 @@ pub mod response {
} }
} }
pub async fn is_song_valid(data: &[u8]) -> Result<bool, std::io::Error> {
match icarus_meta::detection::song::file_type_from_data(data) {
Ok(file_type) => {
if file_type.file_type == icarus_meta::detection::song::constants::FLAC_TYPE {
Ok(true)
} else {
Ok(false)
}
}
Err(err) => {
Err(err)
}
}
}
pub mod endpoint { pub mod endpoint {
use axum::response::IntoResponse; use axum::response::IntoResponse;
@@ -139,6 +154,9 @@ pub mod endpoint {
); );
let raw_data: Vec<u8> = data.to_vec(); let raw_data: Vec<u8> = data.to_vec();
match super::is_song_valid(&raw_data).await {
Ok(valid) => {
if valid {
let queue_repo = repo::song::insert( let queue_repo = repo::song::insert(
&pool, &pool,
&raw_data, &raw_data,
@@ -148,13 +166,23 @@ pub mod endpoint {
.await .await
.unwrap(); .unwrap();
results.push(queue_repo); results.push(queue_repo);
} else {
response.message = String::from("Invalid song type");
return (axum::http::StatusCode::INTERNAL_SERVER_ERROR, axum::Json(response))
}
}
Err(err) => {
response.message = err.to_string();
return (axum::http::StatusCode::INTERNAL_SERVER_ERROR, axum::Json(response));
}
}
} }
response.data = results; response.data = results;
response.message = if response.data.is_empty() { response.message = if response.data.is_empty() {
String::from("Error") String::from("Error")
} else { } else {
String::from("Success") String::from(super::super::super::response::SUCCESSFUL)
}; };
(axum::http::StatusCode::OK, axum::Json(response)) (axum::http::StatusCode::OK, axum::Json(response))