tsk-50: Added properties

Closes #49

Reviewed-on: #51
Co-authored-by: phoenix <kundeng00@pm.me>
Co-committed-by: phoenix <kundeng00@pm.me>

Cargo.lock

@@ -104,7 +104,7 @@ dependencies = [
 
 [[package]]
 name = "icarus_meta"
-version = "0.4.3"
+version = "0.4.4"
 dependencies = [
  "imghdr",
  "infer",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "icarus_meta"
-version = "0.4.3"
+version = "0.4.4"
 edition = "2024"
 rust-version = "1.90"
 

src/detection/mod.rs

@@ -6,3 +6,29 @@ pub struct FileType {
     pub mime: String,
     pub file_type: String,
 }
+
+/// Initializes a FileType given a filetype passed.
+pub fn init_filetype(filetype: &str) -> Result<FileType, std::io::Error> {
+    if filetype == song::constants::FLAC_TYPE {
+        Ok(FileType {
+            mime: String::from(song::constants::mime::FLAC),
+            file_type: String::from(song::constants::FLAC_TYPE),
+        })
+    } else if filetype == coverart::constants::PNG_TYPE {
+        Ok(FileType {
+            mime: String::from(coverart::constants::mime::PNG),
+            file_type: String::from(coverart::constants::PNG_TYPE),
+        })
+    } else if filetype == coverart::constants::JPEG_TYPE
+        || filetype == coverart::constants::JPG_TYPE
+    {
+        Ok(FileType {
+            mime: String::from(coverart::constants::mime::JPEG),
+            file_type: String::from(coverart::constants::JPEG_TYPE),
+        })
+    } else {
+        Err(std::io::Error::other(format!(
+            "Unsupported FileType: {filetype:?}"
+        )))
+    }
+}

src/detection/song.rs

@@ -20,7 +20,7 @@ pub fn file_type_from_filepath(
 }
 
 /// Gets the file type of a Song given it's data
-pub fn file_type_from_data(data: &Vec<u8>) -> Result<crate::detection::FileType, std::io::Error> {
+pub fn file_type_from_data(data: &[u8]) -> Result<crate::detection::FileType, std::io::Error> {
     match infer::get(data) {
         Some(kind) => {
             let mime = kind.mime_type();

src/properties/audio.rs

@@ -0,0 +1,16 @@
+use lofty::file::AudioFile;
+
+pub fn get_properties(songpath: &str) -> Result<lofty::flac::FlacProperties, std::io::Error> {
+    match std::fs::File::open(songpath) {
+        Ok(mut content) => {
+            match lofty::flac::FlacFile::read_from(&mut content, lofty::config::ParseOptions::new())
+            {
+                Ok(flac_file) => {
+                    Ok(*flac_file.properties())
+                }
+                Err(err) => Err(std::io::Error::other(err.to_string())),
+            }
+        }
+        Err(err) => Err(err),
+    }
+}

src/properties/mod.rs

@@ -1,21 +1,43 @@
-use lofty::file::AudioFile;
+pub mod audio;
 
-pub fn get_duration(song_path: &String) -> Result<std::time::Duration, std::io::Error> {
+
-    match std::fs::File::open(song_path) {
+#[derive(Clone, Debug, Default)]
-        Ok(mut content) => {
+pub struct SongProperties {
-            match lofty::flac::FlacFile::read_from(&mut content, lofty::config::ParseOptions::new())
+    pub duration: std::time::Duration,
-            {
+    pub sample_rate: u32,
-                Ok(flac_file) => {
+    pub bitrate: u32,
-                    let properties = flac_file.properties();
+    pub overall_bitrate: u32,
-                    Ok(properties.duration())
+    pub bit_depth: u8,
-                }
+    pub channels: u8,
-                Err(err) => Err(std::io::Error::other(err.to_string())),
+}
-            }
+
+
+pub fn get_song_properties(song_path: &str) -> Result<SongProperties, std::io::Error> {
+    match audio::get_properties(song_path) {
+        Ok(flac_properties) => {
+            Ok(SongProperties {
+                duration: flac_properties.duration(),
+                sample_rate: flac_properties.sample_rate(),
+                bitrate: flac_properties.audio_bitrate(),
+                overall_bitrate: flac_properties.overall_bitrate(),
+                bit_depth: flac_properties.bit_depth(),
+                channels: flac_properties.channels(),
+            })
         }
         Err(err) => Err(err),
     }
 }
 
+pub fn get_duration(song_path: &String) -> Result<std::time::Duration, std::io::Error> {
+    match get_song_properties(song_path) {
+        Ok(song_properties) => {
+            Ok(song_properties.duration)
+        }
+        Err(err) => Err(err),
+    }
+}
+
+
 #[cfg(test)]
 mod tests {
     use crate::test_util;