1 files changed, 34 insertions, 26 deletions

diff --git a/src/api/album.rs b/src/api/album.rs
index d9d0d52..3f91cd0 100644
--- a/src/api/album.rs
+++ b/src/api/album.rs
@@ -1,8 +1,10 @@
+use crate::api::{get_response_from_query, Response};
 use crate::database::{Album, AlbumPost, AlbumPut, Delete};
 use crate::AppState;
 use actix_web::{delete, get, post, put, web, HttpResponse};
 use serde::Deserialize;
 
+/* Possible arguments ( /album?arg=value ) */
 #[derive(Deserialize)]
 struct AlbumQueryOptions {
     id: Option<String>,
@@ -15,19 +17,23 @@ pub async fn get_album(
     app_state: web::Data<AppState>,
     get_args: web::Query<AlbumQueryOptions>,
 ) -> HttpResponse {
-    let default = String::from("");
+    /* Avoid lifespan issues */
+    let default = String::default();
 
+    /* Handle individual result for search-by-id */
     if get_args.id.is_some() {
         let id: &str = get_args.id.as_ref().unwrap_or(&default);
-        let search_attempt: sqlx::Result<Option<Album>> =
+        let query_result: sqlx::Result<Option<Album>> =
             app_state.database.select_album_by_id(id).await;
 
-        return match search_attempt {
+        return match query_result {
             Ok(song_list) => HttpResponse::Ok().json(song_list),
             Err(e) => HttpResponse::Ok().body(format!("{}", e)),
         };
     }
-    let search_attempt: sqlx::Result<Vec<Album>, sqlx::Error> = match true {
+
+    /* Handle N results */
+    let query_result: sqlx::Result<Vec<Album>, sqlx::Error> = match true {
         _ if get_args.name.is_some() => {
             let name: &str = &get_args.name.as_ref().unwrap_or(&default);
             app_state.database.select_albums_by_name(name).await
@@ -39,7 +45,7 @@ pub async fn get_album(
         _ => app_state.database.select_albums().await,
     };
 
-    match search_attempt {
+    match query_result {
         Ok(album_list) => HttpResponse::Ok().json(album_list),
         Err(e) => HttpResponse::Ok().body(format!("{}", e)),
     }
@@ -48,35 +54,35 @@ pub async fn get_album(
 #[post("/album")]
 pub async fn post_album(
     app_state: web::Data<AppState>,
-    post_data: web::Json<AlbumPost>,
+    request_data: web::Json<AlbumPost>,
 ) -> HttpResponse {
-    match app_state
-        .database
-        .create_album(post_data.into_inner())
-        .await
-    {
-        Ok(_) => HttpResponse::Ok().body("Post succeeded\n"),
-        Err(e) => HttpResponse::Ok().body(format!("{}", e)),
-    }
+    get_response_from_query(
+        app_state
+            .database
+            .create_album(request_data.into_inner())
+            .await,
+        "POST".to_string(),
+    )
 }
 
 #[put("/album")]
 pub async fn put_album(
     app_state: web::Data<AppState>,
-    post_data: web::Json<AlbumPut>,
+    request_data: web::Json<AlbumPut>,
 ) -> HttpResponse {
-    match app_state.database.edit_album(post_data.into_inner()).await {
-        Ok(_) => HttpResponse::Ok().body("Put succeeded\n"),
-        Err(e) => HttpResponse::Ok().body(format!("{}", e)),
-    }
+    get_response_from_query(
+        app_state.database.edit_album(request_data.into_inner()).await,
+        "PUT".to_string(),
+    )
 }
 
 #[delete("/album")]
 pub async fn delete_album(
     app_state: web::Data<AppState>,
-    post_data: web::Json<Delete>,
+    request_data: web::Json<Delete>,
 ) -> HttpResponse {
-    let id: i32 = post_data
+    /* Check if ID is valid (return -1 if invalid) */
+    let id: i32 = request_data
         .into_inner()
         .id
         .unwrap_or(String::default())
@@ -84,11 +90,13 @@ pub async fn delete_album(
         .unwrap_or(-1);
 
     if id == -1 {
-        return HttpResponse::Ok().body("Invalid id value, code not executed\n");
+        return HttpResponse::BadRequest().json(Response {
+            message: "Invalid id value, code not executed\n".to_owned(),
+        });
     }
 
-    match app_state.database.delete_album(id).await {
-        Ok(_) => HttpResponse::Ok().body("Deletion succeeded\n".to_owned()),
-        Err(e) => HttpResponse::Ok().body(format!("There was an issue in the request:\n{}", e)),
-    }
+    get_response_from_query(
+        app_state.database.delete_album(id).await,
+        "POST".to_string(),
+    )
 }