pict-rs/src/validate/magick.rs

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>,
    bytes: Bytes,
) -> Result<impl AsyncRead + Unpin, MagickError> {
    convert(
        input.magick_format(),
        output.magick_format(),
        false,
        quality,
        bytes,
    )
    .await
}

pub(super) async fn convert_animation(
    input: AnimationFormat,
    output: AnimationFormat,
    quality: Option<u8>,
    bytes: Bytes,
) -> Result<impl AsyncRead + Unpin, MagickError> {
    convert(
        input.magick_format(),
        output.magick_format(),
        true,
        quality,
        bytes,
    )
    .await
}

pub(super) async fn convert_video(
    input: AnimationFormat,
    output: OutputVideoFormat,
    bytes: Bytes,
) -> Result<impl AsyncRead + Unpin, MagickError> {
    convert(
        input.magick_format(),
        output.magick_format(),
        true,
        None,
        bytes,
    )
    .await
}

async fn convert(
    input: &'static str,
    output: &'static str,
    coalesce: bool,
    quality: Option<u8>,
    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", "-strip", "-auto-orient", &input_arg];

    if let Some(quality) = &quality {
        args.extend(["-quality", quality]);
    }

    if coalesce {
        args.push("-coalesce");
    }

    args.push(&output_arg);

    let reader = Process::run("magick", &args)?.read();

    let clean_reader = crate::tmp_file::cleanup_tmpfile(reader, input_file);

    Ok(Box::pin(clean_reader))
}