From c80bc7d7e1c260191162500e14d55b51f842b520 Mon Sep 17 00:00:00 2001
From: phoenix <kundeng00@pm.me>
Date: Sat, 19 Apr 2025 01:07:49 +0000
Subject: [PATCH] remove_picture (#23)

Reviewed-on: https://git.kundeng.us/phoenix/icarus_meta/pulls/23
Co-authored-by: phoenix <kundeng00@pm.me>
Co-committed-by: phoenix <kundeng00@pm.me>
---
 Cargo.toml  |  2 +-
 src/meta.rs | 65 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 66 insertions(+), 1 deletion(-)

diff --git a/Cargo.toml b/Cargo.toml
index 7f589f8..133ca6a 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "icarus_meta"
-version = "0.1.10"
+version = "0.1.11"
 edition = "2024"
 rust-version = "1.86"
 
diff --git a/src/meta.rs b/src/meta.rs
index 42c3661..63f687b 100644
--- a/src/meta.rs
+++ b/src/meta.rs
@@ -118,6 +118,37 @@ pub mod coverart {
             Err(err) => Err(err),
         }
     }
+
+    pub fn remove_coverart(song_filepath: &String) -> Result<Vec<u8>, std::io::Error> {
+        match std::fs::File::open(song_filepath) {
+            Ok(mut file) => {
+                match lofty::flac::FlacFile::read_from(
+                    &mut file,
+                    lofty::config::ParseOptions::new(),
+                ) {
+                    Ok(mut flac_file) => {
+                        let pictures = flac_file.pictures();
+                        let res = pictures.to_vec();
+                        if !res.is_empty() {
+                            let picture = &res[0];
+                            flac_file.remove_picture(0);
+                            Ok(picture.clone().0.into_data())
+                        } else {
+                            Err(std::io::Error::new(
+                                std::io::ErrorKind::NotFound,
+                                "No pictures found",
+                            ))
+                        }
+                    }
+                    Err(err) => Err(std::io::Error::new(
+                        std::io::ErrorKind::InvalidData,
+                        err.to_string(),
+                    )),
+                }
+            }
+            Err(err) => Err(err),
+        }
+    }
 }
 
 pub fn get_meta(t: types::Type, filepath: &String) -> Result<String, std::io::Error> {
@@ -999,5 +1030,39 @@ mod tests {
                 }
             };
         }
+
+        #[test]
+        fn test_remove_picture() {
+            let filename = util::get_filename(1);
+            let dir = String::from(util::TESTFILEDIRECTORY);
+
+            let temp_file = tempfile::tempdir().expect("Could not create test directory");
+            let test_dir = String::from(temp_file.path().to_str().unwrap());
+            let test_filename = String::from("track09.flac");
+            let new_filepath = get_full_path(&test_dir, &test_filename).unwrap();
+
+            match file_exists(&dir, &filename) {
+                Ok(_) => {
+                    let filepath = get_full_path(&dir, &filename).unwrap();
+
+                    match util::copy_file(&filepath, &new_filepath) {
+                        Ok(_o) => match coverart::remove_coverart(&new_filepath) {
+                            Ok(bytes) => {
+                                assert_eq!(false, bytes.is_empty(), "This should not be empty");
+                            }
+                            Err(err) => {
+                                assert!(false, "Error: {:?}", err);
+                            }
+                        },
+                        Err(err) => {
+                            assert!(false, "Error: {:?}", err);
+                        }
+                    }
+                }
+                Err(err) => {
+                    assert!(false, "Error: File does not exist {:?}", err.to_string());
+                }
+            };
+        }
     }
 }