use crate::{config::Format, error::UploadError, ffmpeg::InputFormat, magick::ValidInputType}; use actix_web::web::Bytes; use tokio::io::AsyncRead; pub(crate) fn image_webp() -> mime::Mime { "image/webp".parse().unwrap() } pub(crate) fn video_mp4() -> mime::Mime { "video/mp4".parse().unwrap() } pub(crate) async fn validate_image_bytes( bytes: Bytes, prescribed_format: Option, ) -> Result<(mime::Mime, Box), UploadError> { let input_type = crate::magick::input_type_bytes(bytes.clone()).await?; match (prescribed_format, input_type) { (_, ValidInputType::Gif) => Ok(( video_mp4(), Box::new(crate::ffmpeg::to_mp4_bytes(bytes, InputFormat::Gif)?), )), (_, ValidInputType::Mp4) => Ok(( video_mp4(), Box::new(crate::ffmpeg::to_mp4_bytes(bytes, InputFormat::Mp4)?), )), (Some(Format::Jpeg) | None, ValidInputType::Jpeg) => Ok(( mime::IMAGE_JPEG, Box::new(crate::exiftool::clear_metadata_bytes_read(bytes)?), )), (Some(Format::Png) | None, ValidInputType::Png) => Ok(( mime::IMAGE_PNG, Box::new(crate::exiftool::clear_metadata_bytes_read(bytes)?), )), (Some(Format::Webp) | None, ValidInputType::Webp) => Ok(( image_webp(), Box::new(crate::magick::clear_metadata_bytes_read(bytes)?), )), (Some(format), _) => Ok(( format.to_mime(), Box::new(crate::magick::convert_bytes_read(bytes, format)?), )), } }