tsk-45: Identify type of Song (#48)

Closes #45

Reviewed-on: phoenix/icarus_meta#48
Co-authored-by: phoenix <kundeng00@pm.me>
Co-committed-by: phoenix <kundeng00@pm.me>
This commit is contained in:
phoenix
2025-10-22 02:26:18 +00:00
committed by phoenix
parent 71374eb731
commit c0607597f1
6 changed files with 256 additions and 10 deletions
+35 -8
View File
@@ -1,8 +1,16 @@
/// Gets the file type of a CoverArt given it's path
pub fn file_type_from_filepath(filepath: &str) -> Result<String, std::io::Error> {
pub fn file_type_from_filepath(
filepath: &str,
) -> Result<crate::detection::FileType, std::io::Error> {
match imghdr::from_file(filepath) {
Ok(Some(imghdr::Type::Jpeg)) => Ok(String::from("jpeg")),
Ok(Some(imghdr::Type::Png)) => Ok(String::from("png")),
Ok(Some(imghdr::Type::Jpeg)) => Ok(crate::detection::FileType {
mime: String::from(constants::mime::JPEG),
file_type: String::from(constants::JPEG_TYPE),
}),
Ok(Some(imghdr::Type::Png)) => Ok(crate::detection::FileType {
mime: String::from(constants::mime::PNG),
file_type: String::from(constants::PNG_TYPE),
}),
Ok(None) => Err(std::io::Error::other("Image file not supported")),
Err(err) => Err(err),
_ => Err(std::io::Error::other("Image file not supported")),
@@ -10,15 +18,32 @@ pub fn file_type_from_filepath(filepath: &str) -> Result<String, std::io::Error>
}
/// Gets the file type of a CoverArt given it's data
pub fn file_type_from_data(data: &Vec<u8>) -> Result<String, std::io::Error> {
pub fn file_type_from_data(data: &Vec<u8>) -> Result<crate::detection::FileType, std::io::Error> {
match imghdr::from_bytes(data) {
Some(imghdr::Type::Jpeg) => Ok(String::from("jpeg")),
Some(imghdr::Type::Png) => Ok(String::from("png")),
Some(imghdr::Type::Jpeg) => Ok(crate::detection::FileType {
mime: String::from(constants::mime::JPEG),
file_type: String::from(constants::JPEG_TYPE),
}),
Some(imghdr::Type::Png) => Ok(crate::detection::FileType {
mime: String::from(constants::mime::PNG),
file_type: String::from(constants::PNG_TYPE),
}),
None => Err(std::io::Error::other("Image file not supported")),
_ => Err(std::io::Error::other("Image file not supported")),
}
}
pub mod constants {
pub const PNG_TYPE: &str = "png";
pub const JPEG_TYPE: &str = "jpeg";
pub const JPG_TYPE: &str = "jpg";
pub mod mime {
pub const JPEG: &str = "image/jpeg";
pub const PNG: &str = "image/png";
}
}
#[cfg(test)]
mod tests {
#[test]
@@ -30,7 +55,8 @@ mod tests {
match super::file_type_from_filepath(&filepath) {
Ok(filetype) => {
assert_eq!(
filetype, "png",
filetype.file_type,
super::constants::PNG_TYPE,
"The file type of the CoverArt should have been png"
);
}
@@ -50,7 +76,8 @@ mod tests {
match super::file_type_from_data(&data) {
Ok(filetype) => {
assert_eq!(
filetype, "png",
filetype.file_type,
super::constants::PNG_TYPE,
"The file type of the CoverArt should have been png"
);
}