Add conversion of images when configured
Signed-off-by: trivernis <trivernis@protonmail.com>pull/4/head
parent
3acfe9c6e2
commit
63ea60b10a
GPG Key ID:
DFFFCC2C7A02DB45
@ -0,0 +1,176 @@
|
||||
use crate::utils::caching::CacheStorage;
|
||||
use crate::utils::downloads::download_path;
|
||||
use image::imageops::FilterType;
|
||||
use image::io::Reader as ImageReader;
|
||||
use image::{GenericImageView, ImageFormat, ImageResult};
|
||||
use mime::Mime;
|
||||
use parking_lot::Mutex;
|
||||
use rayon::prelude::*;
|
||||
use std::io;
|
||||
use std::io::Cursor;
|
||||
use std::path::PathBuf;
|
||||
use std::sync::Arc;
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct ImageConverter {
|
||||
images: Vec<Arc<Mutex<PendingImage>>>,
|
||||
target_format: Option<ImageFormat>,
|
||||
target_size: Option<(u32, u32)>,
|
||||
}
|
||||
|
||||
impl ImageConverter {
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
images: Vec::new(),
|
||||
target_format: None,
|
||||
target_size: None,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn set_target_size(&mut self, target_size: (u32, u32)) {
|
||||
self.target_size = Some(target_size)
|
||||
}
|
||||
|
||||
pub fn set_target_format(&mut self, target_format: ImageFormat) {
|
||||
self.target_format = Some(target_format);
|
||||
}
|
||||
|
||||
/// Adds an image to convert
|
||||
pub fn add_image(&mut self, path: PathBuf) -> Arc<Mutex<PendingImage>> {
|
||||
let image = Arc::new(Mutex::new(PendingImage::new(path)));
|
||||
self.images.push(image.clone());
|
||||
|
||||
image
|
||||
}
|
||||
|
||||
/// Converts all images
|
||||
pub fn convert_all(&mut self) {
|
||||
self.images.par_iter().for_each(|image| {
|
||||
let mut image = image.lock();
|
||||
if let Err(e) = image.convert(self.target_format.clone(), self.target_size.clone()) {
|
||||
log::error!("Failed to embed image {:?}: {}", image.path, e)
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct PendingImage {
|
||||
pub path: PathBuf,
|
||||
pub data: Option<Vec<u8>>,
|
||||
cache: CacheStorage,
|
||||
pub mime: Mime,
|
||||
}
|
||||
|
||||
impl PendingImage {
|
||||
pub fn new(path: PathBuf) -> Self {
|
||||
let mime = get_mime(&path);
|
||||
|
||||
Self {
|
||||
path,
|
||||
data: None,
|
||||
cache: CacheStorage::new(),
|
||||
mime,
|
||||
}
|
||||
}
|
||||
|
||||
/// Converts the image to the specified target format (specified by target_extension)
|
||||
pub fn convert(
|
||||
&mut self,
|
||||
target_format: Option<ImageFormat>,
|
||||
target_size: Option<(u32, u32)>,
|
||||
) -> ImageResult<()> {
|
||||
let format = target_format
|
||||
.or_else(|| {
|
||||
self.path
|
||||
.extension()
|
||||
.and_then(|extension| ImageFormat::from_extension(extension))
|
||||
})
|
||||
.unwrap_or(ImageFormat::Png);
|
||||
let output_path = self.get_output_path(format, target_size);
|
||||
self.mime = get_mime(&output_path);
|
||||
|
||||
if self.cache.has_file(&output_path) {
|
||||
self.data = Some(self.cache.read(&output_path)?)
|
||||
} else {
|
||||
self.convert_image(format, target_size)?;
|
||||
|
||||
if let Some(data) = &self.data {
|
||||
self.cache.write(&output_path, data)?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Converts the image
|
||||
fn convert_image(
|
||||
&mut self,
|
||||
format: ImageFormat,
|
||||
target_size: Option<(u32, u32)>,
|
||||
) -> ImageResult<()> {
|
||||
let mut image = ImageReader::open(self.get_path()?)?.decode()?;
|
||||
|
||||
if let Some((width, height)) = target_size {
|
||||
let dimensions = image.dimensions();
|
||||
|
||||
if dimensions.0 > width || dimensions.1 > height {
|
||||
image = image.resize(width, height, FilterType::Lanczos3);
|
||||
}
|
||||
}
|
||||
let data = Vec::new();
|
||||
let mut writer = Cursor::new(data);
|
||||
|
||||
image.write_to(&mut writer, format)?;
|
||||
self.data = Some(writer.into_inner());
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Returns the path of the file
|
||||
fn get_path(&self) -> io::Result<PathBuf> {
|
||||
if !self.path.exists() {
|
||||
if self.cache.has_file(&self.path) {
|
||||
return Ok(self.cache.get_file_path(&self.path));
|
||||
}
|
||||
if let Some(data) = download_path(self.path.to_string_lossy().to_string()) {
|
||||
self.cache.write(&self.path, data)?;
|
||||
return Ok(self.cache.get_file_path(&self.path));
|
||||
}
|
||||
}
|
||||
Ok(self.path.clone())
|
||||
}
|
||||
|
||||
/// Returns the output file name after converting the image
|
||||
fn get_output_path(
|
||||
&self,
|
||||
target_format: ImageFormat,
|
||||
target_size: Option<(u32, u32)>,
|
||||
) -> PathBuf {
|
||||
let mut path = self.path.clone();
|
||||
let mut file_name = path.file_stem().unwrap().to_string_lossy().to_string();
|
||||
let extension = target_format.extensions_str()[0];
|
||||
let type_name = format!("{:?}", target_format);
|
||||
|
||||
if let Some(target_size) = target_size {
|
||||
file_name += &*format!("-{}-{}", target_size.0, target_size.1);
|
||||
}
|
||||
file_name += format!("-{}-converted", type_name).as_str();
|
||||
path.set_file_name(file_name);
|
||||
path.set_extension(extension);
|
||||
|
||||
path
|
||||
}
|
||||
}
|
||||
|
||||
fn get_mime(path: &PathBuf) -> Mime {
|
||||
let mime = mime_guess::from_ext(
|
||||
path.clone()
|
||||
.extension()
|
||||
.and_then(|e| e.to_str())
|
||||
.unwrap_or("png"),
|
||||
)
|
||||
.first()
|
||||
.unwrap_or(mime::IMAGE_PNG);
|
||||
mime
|
||||
}
|
||||
@ -1,4 +1,5 @@
|
||||
pub mod caching;
|
||||
pub mod downloads;
|
||||
pub mod image_converting;
|
||||
pub mod macros;
|
||||
pub mod parsing;
|
||||
|
||||
Loading…
Reference in New Issue