id3-image-rs/src/lib.rs

77 lines
3.0 KiB
Rust
Raw Normal View History

use std::path::Path;
2019-02-16 08:52:45 +00:00
use std::error::Error;
2019-02-24 15:46:47 +00:00
/// Embed the image from `image_filename` into `music_filename`, in-place. Any errors reading ID3
2019-02-24 15:41:24 +00:00
/// tags from the music file or parsing the image get propagated upwards.
///
2019-02-24 15:46:47 +00:00
/// The image is encoded as a JPEG with a 90% quality setting, and embedded as a "Front cover".
/// Tags get written as ID3v2.3.
2019-02-24 15:41:24 +00:00
///
2019-11-22 19:26:16 +00:00
pub fn embed_image(music_filename: &Path, image_filename: &Path) -> Result<(), Box<dyn Error>> {
2019-02-16 08:52:45 +00:00
let mut tag = id3::Tag::read_from_path(&music_filename).
map_err(|e| format!("Error reading music file {:?}: {}", music_filename, e))?;
2019-02-16 08:52:45 +00:00
let image = image::open(&image_filename).
map_err(|e| format!("Error reading image {:?}: {}", image_filename, e))?;
2019-02-16 08:52:45 +00:00
let mut encoded_image_bytes = Vec::new();
// Unwrap: Writing to a Vec should always succeed;
image.write_to(&mut encoded_image_bytes, image::ImageOutputFormat::JPEG(90)).unwrap();
tag.add_picture(id3::frame::Picture {
mime_type: "image/jpeg".to_string(),
2019-02-17 14:11:07 +00:00
picture_type: id3::frame::PictureType::CoverFront,
2019-02-16 08:52:45 +00:00
description: String::new(),
data: encoded_image_bytes,
});
tag.write_to_path(music_filename, id3::Version::Id3v23).
map_err(|e| format!("Error writing image to music file {:?}: {}", music_filename, e))?;
2019-02-16 08:52:45 +00:00
Ok(())
}
2019-02-24 15:46:47 +00:00
/// Extract the first found embedded image from `music_filename` and write it as a file with the
/// given `image_filename`. The image file will be silently overwritten if it exists.
2019-02-24 15:41:24 +00:00
///
/// Any errors from parsing id3 tags will be propagated. The function will also return an error if
/// there's no embedded images in the mp3 file.
///
2019-11-22 19:26:16 +00:00
pub fn extract_first_image(music_filename: &Path, image_filename: &Path) -> Result<(), Box<dyn Error>> {
2019-02-16 08:52:45 +00:00
let tag = id3::Tag::read_from_path(&music_filename).
map_err(|e| format!("Error reading music file {:?}: {}", music_filename, e))?;
2019-02-16 08:52:45 +00:00
let first_picture = tag.pictures().next();
if let Some(p) = first_picture {
match image::load_from_memory(&p.data) {
Ok(image) => {
image.save(&image_filename).
map_err(|e| format!("Couldn't write image file {:?}: {}", image_filename, e))?;
2019-02-16 08:52:45 +00:00
},
Err(e) => return Err(format!("Couldn't load image: {}", e).into()),
};
Ok(())
} else {
Err("No image found in music file".into())
}
2019-02-16 08:52:45 +00:00
}
2019-02-24 15:41:24 +00:00
/// Remove all embedded images from the given `music_filename`. In effect, this removes all tags of
/// type "APIC".
///
/// If the mp3 file's ID3 tags can't be parsed, the error will be propagated upwards.
///
2019-11-22 19:26:16 +00:00
pub fn remove_images(music_filename: &Path) -> Result<(), Box<dyn Error>> {
let mut tag = id3::Tag::read_from_path(&music_filename).
map_err(|e| format!("Error reading music file {:?}: {}", music_filename, e))?;
tag.remove("APIC");
tag.write_to_path(music_filename, id3::Version::Id3v23).
map_err(|e| format!("Error updating music file {:?}: {}", music_filename, e))?;
Ok(())
}