111 lines
2.5 KiB
Rust
111 lines
2.5 KiB
Rust
use actix_web::web::Bytes;
|
|
use tokio::io::AsyncRead;
|
|
|
|
use crate::{
|
|
formats::{AnimationFormat, ImageFormat, OutputVideoFormat},
|
|
magick::MagickError,
|
|
process::Process,
|
|
};
|
|
|
|
pub(super) async fn convert_image(
|
|
input: ImageFormat,
|
|
output: ImageFormat,
|
|
quality: Option<u8>,
|
|
timeout: u64,
|
|
bytes: Bytes,
|
|
) -> Result<impl AsyncRead + Unpin, MagickError> {
|
|
convert(
|
|
input.magick_format(),
|
|
output.magick_format(),
|
|
false,
|
|
quality,
|
|
timeout,
|
|
bytes,
|
|
)
|
|
.await
|
|
}
|
|
|
|
pub(super) async fn convert_animation(
|
|
input: AnimationFormat,
|
|
output: AnimationFormat,
|
|
quality: Option<u8>,
|
|
timeout: u64,
|
|
bytes: Bytes,
|
|
) -> Result<impl AsyncRead + Unpin, MagickError> {
|
|
convert(
|
|
input.magick_format(),
|
|
output.magick_format(),
|
|
true,
|
|
quality,
|
|
timeout,
|
|
bytes,
|
|
)
|
|
.await
|
|
}
|
|
|
|
pub(super) async fn convert_video(
|
|
input: AnimationFormat,
|
|
output: OutputVideoFormat,
|
|
timeout: u64,
|
|
bytes: Bytes,
|
|
) -> Result<impl AsyncRead + Unpin, MagickError> {
|
|
convert(
|
|
input.magick_format(),
|
|
output.magick_format(),
|
|
true,
|
|
None,
|
|
timeout,
|
|
bytes,
|
|
)
|
|
.await
|
|
}
|
|
|
|
async fn convert(
|
|
input: &'static str,
|
|
output: &'static str,
|
|
coalesce: bool,
|
|
quality: Option<u8>,
|
|
timeout: u64,
|
|
bytes: Bytes,
|
|
) -> Result<impl AsyncRead + Unpin, MagickError> {
|
|
let input_file = crate::tmp_file::tmp_file(None);
|
|
let input_file_str = input_file.to_str().ok_or(MagickError::Path)?;
|
|
|
|
crate::store::file_store::safe_create_parent(&input_file)
|
|
.await
|
|
.map_err(MagickError::CreateDir)?;
|
|
|
|
let mut tmp_one = crate::file::File::create(&input_file)
|
|
.await
|
|
.map_err(MagickError::CreateFile)?;
|
|
tmp_one
|
|
.write_from_bytes(bytes)
|
|
.await
|
|
.map_err(MagickError::Write)?;
|
|
tmp_one.close().await.map_err(MagickError::CloseFile)?;
|
|
|
|
let input_arg = format!("{input}:{input_file_str}");
|
|
let output_arg = format!("{output}:-");
|
|
let quality = quality.map(|q| q.to_string());
|
|
|
|
let mut args = vec!["convert"];
|
|
|
|
if coalesce {
|
|
args.push("-coalesce");
|
|
}
|
|
|
|
args.extend(["-strip", "-auto-orient", &input_arg]);
|
|
|
|
if let Some(quality) = &quality {
|
|
args.extend(["-quality", quality]);
|
|
}
|
|
|
|
args.push(&output_arg);
|
|
|
|
let reader = Process::run("magick", &args, timeout)?.read();
|
|
|
|
let clean_reader = crate::tmp_file::cleanup_tmpfile(reader, input_file);
|
|
|
|
Ok(Box::pin(clean_reader))
|
|
}
|