mirror of
https://git.asonix.dog/asonix/pict-rs
synced 2024-11-10 06:25:00 +00:00
94 lines
2.4 KiB
Rust
94 lines
2.4 KiB
Rust
|
use crate::{
|
||
|
concurrent_processor::CancelSafeProcessor,
|
||
|
config::ImageFormat,
|
||
|
details::Details,
|
||
|
error::Error,
|
||
|
ffmpeg::{InputFormat, ThumbnailFormat},
|
||
|
repo::{Alias, FullRepo},
|
||
|
store::Store,
|
||
|
};
|
||
|
use actix_web::web::Bytes;
|
||
|
use std::path::PathBuf;
|
||
|
use tokio::io::AsyncReadExt;
|
||
|
|
||
|
pub(crate) async fn generate<R: FullRepo, S: Store + 'static>(
|
||
|
repo: &R,
|
||
|
store: &S,
|
||
|
format: ImageFormat,
|
||
|
alias: Alias,
|
||
|
thumbnail_path: PathBuf,
|
||
|
thumbnail_args: Vec<String>,
|
||
|
hash: R::Bytes,
|
||
|
) -> Result<(Details, Bytes), Error> {
|
||
|
let process_fut = process(
|
||
|
repo,
|
||
|
store,
|
||
|
format,
|
||
|
alias,
|
||
|
thumbnail_path.clone(),
|
||
|
thumbnail_args,
|
||
|
hash.clone(),
|
||
|
);
|
||
|
|
||
|
let (details, bytes) =
|
||
|
CancelSafeProcessor::new(hash.as_ref(), thumbnail_path, process_fut).await?;
|
||
|
|
||
|
Ok((details, bytes))
|
||
|
}
|
||
|
|
||
|
async fn process<R: FullRepo, S: Store + 'static>(
|
||
|
repo: &R,
|
||
|
store: &S,
|
||
|
format: ImageFormat,
|
||
|
alias: Alias,
|
||
|
thumbnail_path: PathBuf,
|
||
|
thumbnail_args: Vec<String>,
|
||
|
hash: R::Bytes,
|
||
|
) -> Result<(Details, Bytes), Error> {
|
||
|
let permit = crate::PROCESS_SEMAPHORE.acquire().await?;
|
||
|
|
||
|
let identifier = if let Some(identifier) = repo
|
||
|
.still_identifier_from_alias::<S::Identifier>(&alias)
|
||
|
.await?
|
||
|
{
|
||
|
identifier
|
||
|
} else {
|
||
|
let identifier = repo.identifier(hash.clone()).await?;
|
||
|
let mut reader = crate::ffmpeg::thumbnail(
|
||
|
store.clone(),
|
||
|
identifier,
|
||
|
InputFormat::Mp4,
|
||
|
ThumbnailFormat::Jpeg,
|
||
|
)
|
||
|
.await?;
|
||
|
let motion_identifier = store.save_async_read(&mut reader).await?;
|
||
|
|
||
|
repo.relate_motion_identifier(hash.clone(), &motion_identifier)
|
||
|
.await?;
|
||
|
|
||
|
motion_identifier
|
||
|
};
|
||
|
|
||
|
let mut processed_reader =
|
||
|
crate::magick::process_image_store_read(store.clone(), identifier, thumbnail_args, format)?;
|
||
|
|
||
|
let mut vec = Vec::new();
|
||
|
processed_reader.read_to_end(&mut vec).await?;
|
||
|
let bytes = Bytes::from(vec);
|
||
|
|
||
|
drop(permit);
|
||
|
|
||
|
let details = Details::from_bytes(bytes.clone(), format.as_hint()).await?;
|
||
|
|
||
|
let identifier = store.save_bytes(bytes.clone()).await?;
|
||
|
repo.relate_details(&identifier, &details).await?;
|
||
|
repo.relate_variant_identifier(
|
||
|
hash,
|
||
|
thumbnail_path.to_string_lossy().to_string(),
|
||
|
&identifier,
|
||
|
)
|
||
|
.await?;
|
||
|
|
||
|
Ok((details, bytes)) as Result<(Details, Bytes), Error>
|
||
|
}
|