2022-01-28 14:21:07 +01:00
|
|
|
#[cfg(not(feature = "previews-image"))]
|
|
|
|
use anyhow::anyhow;
|
2022-01-24 17:04:27 +01:00
|
|
|
#[cfg(feature = "previews-image")]
|
|
|
|
use image::{io::Reader as ImageReader, GenericImageView};
|
|
|
|
use std::{cmp, path::Path};
|
|
|
|
|
|
|
|
use anyhow::Result;
|
|
|
|
|
|
|
|
use super::Previewable;
|
|
|
|
|
|
|
|
pub struct ImagePath<'a>(pub &'a Path);
|
|
|
|
|
|
|
|
impl<'a> Previewable for ImagePath<'a> {
|
2022-01-24 18:26:53 +01:00
|
|
|
fn get_thumbnail(&self) -> Result<Option<Vec<u8>>> {
|
2022-01-24 17:04:27 +01:00
|
|
|
#[cfg(feature = "previews-image")]
|
|
|
|
{
|
2022-03-16 11:30:06 +01:00
|
|
|
let file = std::fs::File::open(&self.0)?;
|
|
|
|
let mut bufreader = std::io::BufReader::new(&file);
|
|
|
|
let exifreader = exif::Reader::new();
|
|
|
|
let orientation = exifreader
|
|
|
|
.read_from_container(&mut bufreader)
|
|
|
|
.ok()
|
|
|
|
.and_then(|exif| {
|
|
|
|
exif.get_field(exif::Tag::Orientation, exif::In::PRIMARY)
|
|
|
|
.and_then(|f| match &f.value {
|
|
|
|
exif::Value::Short(shorts) => Some(shorts.clone()),
|
|
|
|
_ => None,
|
|
|
|
})
|
|
|
|
})
|
|
|
|
.and_then(|shorts| shorts.first().cloned());
|
|
|
|
|
2022-01-24 17:04:27 +01:00
|
|
|
let image = ImageReader::open(&self.0)?.decode()?;
|
2022-03-16 11:30:06 +01:00
|
|
|
let image = match orientation {
|
|
|
|
Some(3) => image.rotate180(),
|
|
|
|
Some(6) => image.rotate90(),
|
|
|
|
Some(8) => image.rotate270(),
|
|
|
|
_ => image
|
|
|
|
};
|
2022-01-24 17:04:27 +01:00
|
|
|
let (w, h) = image.dimensions();
|
2022-01-24 18:26:53 +01:00
|
|
|
if cmp::max(w, h) > 1024 {
|
|
|
|
let thumbnail = image.thumbnail(1024, 1024);
|
|
|
|
let thumbnail = thumbnail.into_rgba8();
|
|
|
|
let (w, h) = thumbnail.dimensions();
|
|
|
|
let encoder = webp::Encoder::from_rgba(&thumbnail, w, h);
|
|
|
|
let result = encoder.encode(90.0);
|
|
|
|
Ok(Some(result.to_vec()))
|
2022-01-24 17:04:27 +01:00
|
|
|
} else {
|
2022-01-24 18:26:53 +01:00
|
|
|
Ok(None)
|
|
|
|
}
|
2022-01-24 17:04:27 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(feature = "previews-image"))]
|
|
|
|
Err(anyhow!("Image preview support not enabled!"))
|
|
|
|
}
|
|
|
|
}
|