init
This commit is contained in:
77
libertas_api/src/handlers/media_handlers.rs
Normal file
77
libertas_api/src/handlers/media_handlers.rs
Normal file
@@ -0,0 +1,77 @@
|
||||
use axum::{
|
||||
Router,
|
||||
extract::{DefaultBodyLimit, Multipart, State},
|
||||
http::StatusCode,
|
||||
response::Json,
|
||||
routing::post,
|
||||
};
|
||||
use futures::TryStreamExt;
|
||||
use libertas_core::{error::CoreError, models::Media, schema::UploadMediaData};
|
||||
use serde::Serialize;
|
||||
use std::io;
|
||||
|
||||
use crate::{error::ApiError, middleware::auth::UserId, state::AppState};
|
||||
|
||||
#[derive(Serialize)]
|
||||
pub struct MediaResponse {
|
||||
id: uuid::Uuid,
|
||||
storage_path: String,
|
||||
original_filename: String,
|
||||
mime_type: String,
|
||||
hash: String,
|
||||
}
|
||||
|
||||
impl From<Media> for MediaResponse {
|
||||
fn from(media: Media) -> Self {
|
||||
Self {
|
||||
id: media.id,
|
||||
storage_path: media.storage_path,
|
||||
original_filename: media.original_filename,
|
||||
mime_type: media.mime_type,
|
||||
hash: media.hash,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn media_routes() -> Router<AppState> {
|
||||
Router::new()
|
||||
.route("/", post(upload_media))
|
||||
.layer(DefaultBodyLimit::max(250 * 1024 * 1024))
|
||||
}
|
||||
|
||||
async fn upload_media(
|
||||
State(state): State<AppState>,
|
||||
UserId(user_id): UserId,
|
||||
mut multipart: Multipart,
|
||||
) -> Result<(StatusCode, Json<MediaResponse>), ApiError> {
|
||||
let field = multipart
|
||||
.next_field()
|
||||
.await
|
||||
.map_err(|e| CoreError::Validation(format!("Multipart error: {}", e)))?
|
||||
.ok_or(ApiError::from(CoreError::Validation(
|
||||
"No file provided in 'file' field".to_string(),
|
||||
)))?;
|
||||
|
||||
let filename = field.file_name().unwrap_or("unknown_file").to_string();
|
||||
let mime_type = field
|
||||
.content_type()
|
||||
.unwrap_or("application/octet-stream")
|
||||
.to_string();
|
||||
|
||||
let stream = field.map_err(|e| io::Error::new(io::ErrorKind::Other, e));
|
||||
|
||||
let boxed_stream: Box<
|
||||
dyn futures::Stream<Item = Result<bytes::Bytes, std::io::Error>> + Send + Unpin,
|
||||
> = Box::new(stream);
|
||||
|
||||
let upload_data = UploadMediaData {
|
||||
owner_id: user_id,
|
||||
filename,
|
||||
mime_type,
|
||||
stream: boxed_stream,
|
||||
};
|
||||
|
||||
let media = state.media_service.upload_media(upload_data).await?;
|
||||
|
||||
Ok((StatusCode::CREATED, Json(media.into())))
|
||||
}
|
||||
Reference in New Issue
Block a user