1 files changed, 30 insertions, 22 deletions

diff --git a/src/api/song.rs b/src/api/song.rs
index 850c759..698f27a 100644
--- a/src/api/song.rs
+++ b/src/api/song.rs
@@ -1,8 +1,10 @@
+use crate::api::{get_response_from_query, Response};
 use crate::database::{Delete, Song, SongPost, SongPut};
 use crate::AppState;
 use actix_web::{delete, get, post, put, web, HttpResponse};
 use serde::Deserialize;
 
+/* Possible arguments ( /song?arg=value ) */
 #[derive(Deserialize)]
 struct SongQueryOptions {
     id: Option<String>,
@@ -16,20 +18,23 @@ pub async fn get_song(
     app_state: web::Data<AppState>,
     get_args: web::Query<SongQueryOptions>,
 ) -> 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<Song>> =
+        let query_result: sqlx::Result<Option<Song>> =
             app_state.database.select_song_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<Song>> = match true {
+    /* Handle N results */
+    let query_result: sqlx::Result<Vec<Song>> = match true {
         _ if get_args.name.is_some() => {
             let name: &str = get_args.name.as_ref().unwrap_or(&default);
             app_state.database.select_songs_by_name(name).await
@@ -45,7 +50,7 @@ pub async fn get_song(
         _ => app_state.database.select_songs().await,
     };
 
-    match search_attempt {
+    match query_result {
         Ok(song_list) => HttpResponse::Ok().json(song_list),
         Err(e) => HttpResponse::Ok().body(format!("{}", e)),
     }
@@ -54,31 +59,32 @@ pub async fn get_song(
 #[post("/song")]
 pub async fn post_song(
     app_state: web::Data<AppState>,
-    post_data: web::Json<SongPost>,
+    request_data: web::Json<SongPost>,
 ) -> HttpResponse {
-    match app_state.database.create_song(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_song(request_data.into_inner()).await,
+        "POST".to_string(),
+    )
 }
 
 #[put("/song")]
 pub async fn put_song(
     app_state: web::Data<AppState>,
-    post_data: web::Json<SongPut>,
+    request_data: web::Json<SongPut>,
 ) -> HttpResponse {
-    match app_state.database.edit_song(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_song(request_data.into_inner()).await,
+        "PUT".to_owned(),
+    )
 }
 
 #[delete("/song")]
 pub async fn delete_song(
     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())
@@ -86,11 +92,13 @@ pub async fn delete_song(
         .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".to_owned(),
+        });
     }
 
-    match app_state.database.delete_song(id).await {
-        Ok(_) => HttpResponse::Ok().body("Deletion succeeded\n"),
-        Err(e) => HttpResponse::Ok().body(format!("There was an issue in the request:\n{}", e)),
-    }
+    get_response_from_query(
+        app_state.database.delete_song(id).await,
+        "DELETE".to_owned(),
+    )
 }