68 lines
1.7 KiB
Rust
68 lines
1.7 KiB
Rust
use actix_web::http::header::ContentType;
|
|
use actix_web::Result;
|
|
use actix_web::{
|
|
web::{self, Data},
|
|
Scope,
|
|
};
|
|
use bytes::Bytes;
|
|
use shared_entity::data::{AppResponse, JsonAppResponse};
|
|
use shared_entity::error_code::ErrorCode;
|
|
|
|
use crate::biz::file_storage;
|
|
use crate::{component::auth::jwt::UserUuid, state::AppState};
|
|
|
|
pub fn file_storage_scope() -> Scope {
|
|
web::scope("/api/file_storage").service(
|
|
web::resource("/{path}")
|
|
.route(web::get().to(get_handler))
|
|
.route(web::put().to(put_handler))
|
|
.route(web::delete().to(delete_handler)),
|
|
)
|
|
}
|
|
|
|
async fn put_handler(
|
|
user_uuid: UserUuid,
|
|
state: Data<AppState>,
|
|
path: web::Path<String>,
|
|
file_data: Bytes,
|
|
content_type: web::Header<ContentType>,
|
|
) -> Result<JsonAppResponse<()>> {
|
|
let file_path = path.into_inner();
|
|
let mime = content_type.into_inner().0;
|
|
file_storage::put_object(
|
|
&state.pg_pool,
|
|
&state.s3_bucket,
|
|
&user_uuid,
|
|
&file_path,
|
|
&file_data,
|
|
mime,
|
|
)
|
|
.await?;
|
|
Ok(AppResponse::Ok().into())
|
|
}
|
|
|
|
async fn delete_handler(
|
|
user_uuid: UserUuid,
|
|
state: Data<AppState>,
|
|
path: web::Path<String>,
|
|
) -> Result<JsonAppResponse<()>> {
|
|
let file_path = path.into_inner();
|
|
file_storage::delete_object(&state.pg_pool, &state.s3_bucket, &user_uuid, &file_path).await?;
|
|
Ok(AppResponse::Ok().into())
|
|
}
|
|
|
|
async fn get_handler(
|
|
user_uuid: UserUuid,
|
|
state: Data<AppState>,
|
|
path: web::Path<String>,
|
|
) -> Result<Bytes> {
|
|
let file_path = path.into_inner();
|
|
match file_storage::get_object(&state.pg_pool, &state.s3_bucket, &user_uuid, &file_path).await {
|
|
Ok(data) => Ok(data),
|
|
Err(e) => match e.code {
|
|
ErrorCode::FileNotFound => Err(actix_web::error::ErrorNotFound(e)),
|
|
_ => Err(actix_web::error::ErrorInternalServerError(e)),
|
|
},
|
|
}
|
|
}
|