add: move upload server to buckets

This commit is contained in:
trisua 2025-08-21 00:30:29 -04:00
parent 8116307ba0
commit 75fe720f21
83 changed files with 351 additions and 458 deletions

View file

@ -42,11 +42,16 @@ pub async fn create_request(
}
let upload = match data
.create_upload(MediaUpload::new(MediaType::Webp, user.id))
.2
.create_upload(MediaUpload::new(
MediaType::Webp,
user.id,
"ads".to_string(),
))
.await
{
Ok(x) => x,
Err(e) => return Json(e.into()),
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
};
match data
@ -55,9 +60,11 @@ pub async fn create_request(
{
Ok(_) => {
// write image
if let Err(e) =
save_webp_buffer(&upload.path(&data.0.0).to_string(), file.to_vec(), None)
{
if let Err(e) = save_webp_buffer(
&upload.path(&data.2.0.0.directory).to_string(),
file.to_vec(),
None,
) {
return Json(Error::MiscError(e.to_string()).into());
}

View file

@ -63,10 +63,11 @@ pub async fn get_request(
};
let upload = data
.2
.get_upload_by_id(emoji.0.unwrap().upload_id)
.await
.unwrap();
let path = upload.path(&data.0.0);
let path = upload.path(&data.2.0.0.directory);
if !exists(&path).unwrap() {
return Err((
@ -80,7 +81,7 @@ pub async fn get_request(
}
Ok((
[("Content-Type", upload.what.mime())],
[("Content-Type", upload.metadata.what.mime())],
Body::from(read_image(path)),
))
}
@ -119,6 +120,7 @@ pub async fn create_request(
// create upload
let upload = match data
.2
.create_upload(MediaUpload::new(
if img.1 == "image/gif" {
MediaType::Gif
@ -126,11 +128,12 @@ pub async fn create_request(
MediaType::Webp
},
user.id,
"emojis".to_string(),
))
.await
{
Ok(u) => u,
Err(e) => return Json(e.into()),
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
};
// create emoji
@ -147,13 +150,15 @@ pub async fn create_request(
{
Ok(_) => {
if is_animated {
if let Err(e) = upload.write(&data.0.0, &img.0) {
if let Err(e) = upload.write(&data.2.0.0.directory, &img.0) {
return Json(Error::MiscError(e.to_string()).into());
}
} else {
if let Err(e) =
save_webp_buffer(&upload.path(&data.0.0).to_string(), img.0.to_vec(), None)
{
if let Err(e) = save_webp_buffer(
&upload.path(&data.2.0.0.directory).to_string(),
img.0.to_vec(),
None,
) {
return Json(Error::MiscError(e.to_string()).into());
}
}
@ -165,8 +170,8 @@ pub async fn create_request(
})
}
Err(e) => {
if let Err(e) = upload.remove(&data.0.0) {
return Json(e.into());
if let Err(e) = upload.remove(&data.2.0.0.directory) {
return Json(Error::MiscError(e.to_string()).into());
}
Json(e.into())

View file

@ -146,11 +146,16 @@ pub async fn create_request(
for _ in 0..images.len() {
props.uploads.push(
match data
.create_upload(MediaUpload::new(MediaType::Webp, props.owner))
.2
.create_upload(MediaUpload::new(
MediaType::Webp,
props.owner,
"post_media".to_string(),
))
.await
{
Ok(u) => u.id,
Err(e) => return Json(e.into()),
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
},
);
}
@ -164,22 +169,24 @@ pub async fn create_request(
let image = match images.get(i) {
Some(img) => img,
None => {
if let Err(e) = data.delete_upload(*upload_id).await {
return Json(e.into());
if let Err(e) = data.2.delete_upload(*upload_id).await {
return Json(Error::MiscError(e.to_string()).into());
}
continue;
}
};
let upload = match data.get_upload_by_id(*upload_id).await {
let upload = match data.2.get_upload_by_id(*upload_id).await {
Ok(u) => u,
Err(e) => return Json(e.into()),
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
};
if let Err(e) =
save_webp_buffer(&upload.path(&data.0.0).to_string(), image.to_vec(), None)
{
if let Err(e) = save_webp_buffer(
&upload.path(&data.2.0.0.directory).to_string(),
image.to_vec(),
None,
) {
return Json(Error::MiscError(e.to_string()).into());
}
}

View file

@ -706,9 +706,7 @@ pub fn routes() -> Router {
delete(notes::delete_by_dir_request),
)
// uploads
.route("/uploads/{id}", get(uploads::get_request))
.route("/uploads/{id}", delete(uploads::delete_request))
.route("/uploads/{id}/data", get(uploads::get_json_request))
.route("/uploads/{id}/alt", post(uploads::update_alt_request))
// services
.route("/services", get(services::list_request))

View file

@ -368,27 +368,34 @@ pub async fn update_uploads_request(
}
let upload = match data
.create_upload(MediaUpload::new(MediaType::Webp, user.id))
.2
.create_upload(MediaUpload::new(
MediaType::Webp,
user.id,
"product_imgs".to_string(),
))
.await
{
Ok(x) => x,
Err(e) => return Json(e.into()),
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
};
product.uploads.thumbnails.push(upload.id);
// write image
if let Err(e) =
save_webp_buffer(&upload.path(&data.0.0).to_string(), file.to_vec(), None)
{
if let Err(e) = save_webp_buffer(
&upload.path(&data.2.0.0.directory).to_string(),
file.to_vec(),
None,
) {
return Json(Error::MiscError(e.to_string()).into());
}
}
ProductUploadTarget::Reward => {
// remove old
if product.uploads.reward != 0 {
if let Err(e) = data.delete_upload(product.uploads.reward).await {
return Json(e.into());
if let Err(e) = data.2.delete_upload(product.uploads.reward).await {
return Json(Error::MiscError(e.to_string()).into());
}
}
@ -403,19 +410,26 @@ pub async fn update_uploads_request(
}
let upload = match data
.create_upload(MediaUpload::new(MediaType::Webp, user.id))
.2
.create_upload(MediaUpload::new(
MediaType::Webp,
user.id,
"product_imgs".to_string(),
))
.await
{
Ok(x) => x,
Err(e) => return Json(e.into()),
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
};
product.uploads.reward = upload.id;
// write image
if let Err(e) =
save_webp_buffer(&upload.path(&data.0.0).to_string(), file.to_vec(), None)
{
if let Err(e) = save_webp_buffer(
&upload.path(&data.2.0.0.directory).to_string(),
file.to_vec(),
None,
) {
return Json(Error::MiscError(e.to_string()).into());
}
}
@ -462,8 +476,8 @@ pub async fn remove_thumbnail_request(
None => return Json(Error::GeneralNotFound("thumbnail".to_string()).into()),
};
if let Err(e) = data.delete_upload(*thumbnail).await {
return Json(e.into());
if let Err(e) = data.2.delete_upload(*thumbnail).await {
return Json(Error::MiscError(e.to_string()).into());
}
product.uploads.thumbnails.remove(req.idx);

View file

@ -1,74 +1,7 @@
use std::fs::exists;
use axum::{body::Body, extract::Path, response::IntoResponse, Extension, Json};
use axum::{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),
})
}
use tetratto_core::model::{oauth, ApiReturn, Error};
pub async fn delete_request(
jar: CookieJar,
@ -81,13 +14,22 @@ pub async fn delete_request(
None => return Json(Error::NotAllowed.into()),
};
match data.delete_upload_checked(id, &user).await {
let upload = match data.2.get_upload_by_id(id).await {
Ok(x) => x,
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
};
if user.id != upload.owner {
return Json(Error::NotAllowed.into());
}
match data.2.delete_upload(id).await {
Ok(_) => Json(ApiReturn {
ok: true,
message: "Upload deleted".to_string(),
payload: (),
}),
Err(e) => Json(e.into()),
Err(e) => Json(Error::MiscError(e.to_string()).into()),
}
}
@ -103,12 +45,22 @@ pub async fn update_alt_request(
None => return Json(Error::NotAllowed.into()),
};
match data.update_upload_alt(id, &user, &props.alt).await {
let mut upload = match data.2.get_upload_by_id(id).await {
Ok(x) => x,
Err(e) => return Json(Error::MiscError(e.to_string()).into()),
};
if user.id != upload.owner {
return Json(Error::NotAllowed.into());
}
upload.metadata.alt = props.alt;
match data.2.update_upload_metadata(id, upload.metadata).await {
Ok(_) => Json(ApiReturn {
ok: true,
message: "Upload updated".to_string(),
payload: (),
}),
Err(e) => Json(e.into()),
Err(e) => Json(Error::MiscError(e.to_string()).into()),
}
}

View file

@ -122,10 +122,17 @@ pub async fn settings_request(
Err(e) => return Err(Html(render_error(e, &jar, &data, &None).await)),
};
let uploads = match data.0.get_uploads_by_owner(profile.id, 12, req.page).await {
let uploads = match data
.0
.2
.get_uploads_by_owner(profile.id, 12, req.page)
.await
{
Ok(ua) => ua,
Err(e) => {
return Err(Html(render_error(e, &jar, &data, &None).await));
return Err(Html(
render_error(Error::MiscError(e.to_string()), &jar, &data, &None).await,
));
}
};