114 lines
3.2 KiB
Rust
114 lines
3.2 KiB
Rust
use std::fs::exists;
|
|
use axum::{body::Body, extract::Path, response::IntoResponse, Extension, Json};
|
|
use crate::cookie::CookieJar;
|
|
use pathbufd::PathBufD;
|
|
use crate::{get_user_from_token, routes::api::v1::UpdateUploadAlt, State};
|
|
use super::auth::images::read_image;
|
|
use tetratto_core::model::{carp::CarpGraph, oauth, uploads::MediaType, ApiReturn, Error};
|
|
|
|
pub async fn get_request(
|
|
Path(id): Path<usize>,
|
|
Extension(data): Extension<State>,
|
|
) -> impl IntoResponse {
|
|
let data = &(data.read().await).0;
|
|
|
|
let upload = match data.get_upload_by_id(id).await {
|
|
Ok(u) => u,
|
|
Err(_) => {
|
|
return Err((
|
|
[("Content-Type", "image/svg+xml")],
|
|
Body::from(read_image(PathBufD::current().extend(&[
|
|
data.0.0.dirs.media.as_str(),
|
|
"images",
|
|
"default-banner.svg",
|
|
]))),
|
|
));
|
|
}
|
|
};
|
|
|
|
let path = upload.path(&data.0.0);
|
|
|
|
if !exists(&path).unwrap() {
|
|
return Err((
|
|
[("Content-Type", "image/svg+xml")],
|
|
Body::from(read_image(PathBufD::current().extend(&[
|
|
data.0.0.dirs.media.as_str(),
|
|
"images",
|
|
"default-banner.svg",
|
|
]))),
|
|
));
|
|
}
|
|
|
|
let bytes = read_image(path);
|
|
|
|
if upload.what == MediaType::Carpgraph {
|
|
// conver to svg and return
|
|
return Ok((
|
|
[("Content-Type", "image/svg+xml".to_string())],
|
|
Body::from(CarpGraph::from_bytes(bytes).to_svg()),
|
|
));
|
|
}
|
|
|
|
Ok(([("Content-Type", upload.what.mime())], Body::from(bytes)))
|
|
}
|
|
|
|
pub async fn get_json_request(
|
|
Path(id): Path<usize>,
|
|
Extension(data): Extension<State>,
|
|
) -> impl IntoResponse {
|
|
let data = &(data.read().await).0;
|
|
|
|
let upload = match data.get_upload_by_id(id).await {
|
|
Ok(u) => u,
|
|
Err(e) => return Json(e.into()),
|
|
};
|
|
|
|
Json(ApiReturn {
|
|
ok: true,
|
|
message: "Success".to_string(),
|
|
payload: Some(upload),
|
|
})
|
|
}
|
|
|
|
pub async fn delete_request(
|
|
jar: CookieJar,
|
|
Extension(data): Extension<State>,
|
|
Path(id): Path<usize>,
|
|
) -> impl IntoResponse {
|
|
let data = &(data.read().await).0;
|
|
let user = match get_user_from_token!(jar, data, oauth::AppScope::UserManageUploads) {
|
|
Some(ua) => ua,
|
|
None => return Json(Error::NotAllowed.into()),
|
|
};
|
|
|
|
match data.delete_upload_checked(id, &user).await {
|
|
Ok(_) => Json(ApiReturn {
|
|
ok: true,
|
|
message: "Upload deleted".to_string(),
|
|
payload: (),
|
|
}),
|
|
Err(e) => Json(e.into()),
|
|
}
|
|
}
|
|
|
|
pub async fn update_alt_request(
|
|
jar: CookieJar,
|
|
Extension(data): Extension<State>,
|
|
Path(id): Path<usize>,
|
|
Json(props): Json<UpdateUploadAlt>,
|
|
) -> impl IntoResponse {
|
|
let data = &(data.read().await).0;
|
|
let user = match get_user_from_token!(jar, data, oauth::AppScope::UserManageUploads) {
|
|
Some(ua) => ua,
|
|
None => return Json(Error::NotAllowed.into()),
|
|
};
|
|
|
|
match data.update_upload_alt(id, &user, &props.alt).await {
|
|
Ok(_) => Json(ApiReturn {
|
|
ok: true,
|
|
message: "Upload updated".to_string(),
|
|
payload: (),
|
|
}),
|
|
Err(e) => Json(e.into()),
|
|
}
|
|
}
|