82 lines
2.4 KiB
Rust
82 lines
2.4 KiB
Rust
use axum::{
|
|
body::Bytes,
|
|
extract::{FromRequest, Request},
|
|
http::{StatusCode, header::CONTENT_TYPE},
|
|
};
|
|
use axum_extra::extract::Multipart;
|
|
use std::{fs::File, io::BufWriter};
|
|
|
|
/// An image extractor accepting:
|
|
/// * `multipart/form-data`
|
|
/// * `image/png`
|
|
/// * `image/jpeg`
|
|
/// * `image/avif`
|
|
/// * `image/webp`
|
|
pub struct Image(pub Bytes, pub String);
|
|
|
|
impl<S> FromRequest<S> for Image
|
|
where
|
|
Bytes: FromRequest<S>,
|
|
S: Send + Sync,
|
|
{
|
|
type Rejection = StatusCode;
|
|
|
|
async fn from_request(req: Request, state: &S) -> Result<Self, Self::Rejection> {
|
|
let Some(content_type) = req.headers().get(CONTENT_TYPE) else {
|
|
return Err(StatusCode::BAD_REQUEST);
|
|
};
|
|
|
|
let content_type = content_type.to_str().unwrap();
|
|
let content_type_string = content_type.to_string();
|
|
|
|
let body = if content_type.starts_with("multipart/form-data") {
|
|
let mut multipart = Multipart::from_request(req, state)
|
|
.await
|
|
.map_err(|_| StatusCode::BAD_REQUEST)?;
|
|
|
|
let Ok(Some(field)) = multipart.next_field().await else {
|
|
return Err(StatusCode::BAD_REQUEST);
|
|
};
|
|
|
|
field.bytes().await.map_err(|_| StatusCode::BAD_REQUEST)?
|
|
} else if (content_type == "image/avif")
|
|
| (content_type == "image/jpeg")
|
|
| (content_type == "image/png")
|
|
| (content_type == "image/webp")
|
|
| (content_type == "image/gif")
|
|
{
|
|
Bytes::from_request(req, state)
|
|
.await
|
|
.map_err(|_| StatusCode::BAD_REQUEST)?
|
|
} else {
|
|
return Err(StatusCode::BAD_REQUEST);
|
|
};
|
|
|
|
Ok(Self(body, content_type_string))
|
|
}
|
|
}
|
|
|
|
/// Create an image buffer given an input of `bytes`
|
|
pub fn save_buffer(path: &str, bytes: Vec<u8>, format: image::ImageFormat) -> std::io::Result<()> {
|
|
let pre_img_buffer = match image::load_from_memory(&bytes) {
|
|
Ok(i) => i,
|
|
Err(_) => {
|
|
return Err(std::io::Error::new(
|
|
std::io::ErrorKind::InvalidData,
|
|
"Image failed",
|
|
));
|
|
}
|
|
};
|
|
|
|
let file = File::create(path)?;
|
|
let mut writer = BufWriter::new(file);
|
|
|
|
if pre_img_buffer.write_to(&mut writer, format).is_err() {
|
|
return Err(std::io::Error::new(
|
|
std::io::ErrorKind::Other,
|
|
"Image conversion failed",
|
|
));
|
|
};
|
|
|
|
Ok(())
|
|
}
|