Working on profile picture

service/src/users/routes.rs

@@ -1,8 +1,10 @@
+use actix_multipart::Multipart;
 use actix_web::{get, post, delete, web, HttpResponse, ResponseError};
+use futures_util::StreamExt;
 
-use crate::auth::{JwtAuth, QueryUser, InsertUser};
+use crate::{auth::{JwtAuth, QueryUser, InsertUser}, error_handler::ServiceError, db::{upload_file, get_file, delete_file}};
 
-#[get("users/favorites")]
+#[get("/favorites")]
 async fn get_favorites(auth: JwtAuth) -> HttpResponse {
   match QueryUser::get_by_email(&auth.user.email) {
     Ok(user) => {
@@ -12,7 +14,7 @@ async fn get_favorites(auth: JwtAuth) -> HttpResponse {
   }
 }
 
-#[post("users/favorites/{icao}")]
+#[post("/favorites/{icao}")]
 async fn add_favorite(icao: web::Path<String>, auth: JwtAuth) -> HttpResponse {
   match QueryUser::get_by_email(&auth.user.email) {
     Ok(user) => {
@@ -33,7 +35,7 @@ async fn add_favorite(icao: web::Path<String>, auth: JwtAuth) -> HttpResponse {
   }
 }
 
-#[delete("users/favorites/{icao}")]
+#[delete("/favorites/{icao}")]
 async fn delete_favorite(icao: web::Path<String>, auth: JwtAuth) -> HttpResponse {
   let icao: String = icao.into_inner();
   match QueryUser::get_by_email(&auth.user.email) {
@@ -55,8 +57,95 @@ async fn delete_favorite(icao: web::Path<String>, auth: JwtAuth) -> HttpResponse
   }
 }
 
+#[post("/picture")]
+async fn set_picture(mut payload: Multipart, auth: JwtAuth) -> HttpResponse {
+  while let Some(item) = payload.next().await {
+    let mut bytes = web::BytesMut::new();
+    let mut field = match item {
+      Ok(field) => field,
+      Err(err) => return ResponseError::error_response(&err)
+    };
+    let content_type = field.content_disposition();
+    let filename = match content_type.get_filename() {
+      Some(name) => {
+        match name.split(".").last() {
+          Some(ext) => {
+            match ext {
+              "apng" | "avif" | "gif" | "jpg" | "jpeg" | "jfif" | "pjpeg" | "pjp" | "png" | "svg" | "webp" => name,
+              _ => return ResponseError::error_response(&ServiceError::new(400, "File extension is not supported".to_string()))
+            }
+          },
+          None => return ResponseError::error_response(&ServiceError::new(400, "Unknown file extension".to_string()))
+        }
+      },
+      None => return ResponseError::error_response(&ServiceError::new(400, "File name is not provided".to_string()))
+    };
+    let path = format!("users/{}/{}", auth.user.email, filename);
+
+    while let Some(chunk) = field.next().await {
+      let data = match chunk {
+        Ok(data) => data,
+        Err(err) => return ResponseError::error_response(&err)
+      };
+      bytes.extend_from_slice(&data);
+    }
+    match upload_file(&path, &bytes).await {
+      Ok(_) => {
+        match InsertUser::update_profile_picture(&auth.user.email, Some(&path)) {
+          Ok(_) => {},
+          Err(err) => return ResponseError::error_response(&err)
+        };
+      },
+      Err(err) => return ResponseError::error_response(&err)
+    };
+  }
+  HttpResponse::Ok().finish()
+}
+
+#[get("/picture")]
+async fn get_picture(auth: JwtAuth) -> HttpResponse {
+  let user = match QueryUser::get_by_email(&auth.user.email) {
+    Ok(user) => user,
+    Err(err) => return ResponseError::error_response(&err)
+  };
+  if let Some(path) = user.profile_picture {
+    match get_file(&path).await {
+      Ok(bytes) => HttpResponse::Ok().body(bytes),
+      Err(err) => ResponseError::error_response(&err)
+    }
+  } else {
+    HttpResponse::NotFound().finish()
+  }
+}
+
+#[delete("/picture")]
+async fn delete_picture(auth: JwtAuth) -> HttpResponse {
+  let user = match QueryUser::get_by_email(&auth.user.email) {
+    Ok(user) => user,
+    Err(err) => return ResponseError::error_response(&err)
+  };
+  if let Some(path) = user.profile_picture {
+    match delete_file(&path).await {
+      Ok(_) => {
+        match InsertUser::update_profile_picture(&auth.user.email, None) {
+          Ok(_) => HttpResponse::Ok().finish(),
+          Err(err) => ResponseError::error_response(&err)
+        }
+      },
+      Err(err) => ResponseError::error_response(&err)
+    }
+  } else {
+    HttpResponse::NotFound().finish()
+  }
+}
+
 pub fn init_routes(config: &mut web::ServiceConfig) {
-  config.service(get_favorites);
-  config.service(add_favorite);
-  config.service(delete_favorite);
+  config.service(web::scope("users")
+    .service(get_favorites)
+    .service(add_favorite)
+    .service(delete_favorite)
+    .service(set_picture)
+    .service(get_picture)
+    .service(delete_picture)
+  );
 }