Route for model uploading files

1 files changed, 37 insertions, 2 deletions

diff --git a/src/routes/model.rs b/src/routes/model.rs
index 4ac2688..66f8ec6 100644
--- a/src/routes/model.rs
+++ b/src/routes/model.rs
@@ -1,15 +1,23 @@
+use crate::config::MAX_UPLOAD_FILE_SIZE;
 use crate::errors::AppError;
+use crate::files::upload;
 use crate::models::{
     auth::Claims,
     model::{Model, ModelCreate, ModelUser},
 };
 use crate::pagination::Pagination;
-use axum::{extract::Query, routing::get, Json, Router};
+use axum::{
+    extract::{ContentLengthLimit, Multipart, Path, Query},
+    routing::{get, post},
+    Json, Router,
+};
 use serde::Serialize;
 
 /// Create routes for `/v1/models/` namespace
 pub fn create_route() -> Router {
-    Router::new().route("/", get(list_models).post(create_model))
+    Router::new()
+        .route("/", get(list_models).post(create_model))
+        .route("/:id/upload", post(upload_model_file))
 }
 
 #[derive(Serialize)]
@@ -47,3 +55,30 @@ async fn create_model(
 
     Ok(Json(model_new))
 }
+
+/// Upload a file for a model
+async fn upload_model_file(
+    claims: Claims,
+    Path(model_id): Path<i32>,
+    ContentLengthLimit(multipart): ContentLengthLimit<Multipart, { MAX_UPLOAD_FILE_SIZE }>,
+) -> Result<String, AppError> {
+    let model = match Model::find_by_id(model_id).await {
+        Ok(model) => model,
+        Err(_) => {
+            return Err(AppError::NotFound("Model not found".to_string()));
+        }
+    };
+
+    if model.author_id() != claims.user_id {
+        return Err(AppError::Unauthorized);
+    }
+
+    match upload(multipart, vec!["stl"]).await {
+        Ok(saved_file) => {
+            return Ok(format!("Uploaded {}", saved_file));
+        }
+        Err(e) => {
+            return Err(e);
+        }
+    }
+}