get, post, put, delete covered

1 files changed, 46 insertions, 10 deletions

diff --git a/src/routes/song.rs b/src/routes/song.rs
index 97513f6..850c759 100644
--- a/src/routes/song.rs
+++ b/src/routes/song.rs
@@ -1,6 +1,6 @@
-use crate::database::Song;
+use crate::database::{Delete, Song, SongPost, SongPut};
 use crate::AppState;
-use actix_web::{get, web, HttpResponse};
+use actix_web::{delete, get, post, put, web, HttpResponse};
 use serde::Deserialize;
 
 #[derive(Deserialize)]
@@ -16,7 +16,7 @@ pub async fn get_song(
     app_state: web::Data<AppState>,
     get_args: web::Query<SongQueryOptions>,
 ) -> HttpResponse {
-    let default: String = String::from("");
+    let default = String::from("");
 
     if get_args.id.is_some() {
         let id: &str = get_args.id.as_ref().unwrap_or(&default);
@@ -51,10 +51,46 @@ pub async fn get_song(
     }
 }
 
-// #[post("/song")]
-// pub async fn post_song(
-//     app_state: web::Data<AppState>,
-//     get_args: web::Query<SongQueryOptions>,
-// ) -> HttpResponse {
-//     HttpResponse::Ok().body("Post carried successfully")
-// }
+#[post("/song")]
+pub async fn post_song(
+    app_state: web::Data<AppState>,
+    post_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)),
+    }
+}
+
+#[put("/song")]
+pub async fn put_song(
+    app_state: web::Data<AppState>,
+    post_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)),
+    }
+}
+
+#[delete("/song")]
+pub async fn delete_song(
+    app_state: web::Data<AppState>,
+    post_data: web::Json<Delete>,
+) -> HttpResponse {
+    let id: i32 = post_data
+        .into_inner()
+        .id
+        .unwrap_or(String::default())
+        .parse::<i32>()
+        .unwrap_or(-1);
+
+    if id == -1 {
+        return HttpResponse::Ok().body("Invalid id value, code not executed\n");
+    }
+
+    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)),
+    }
+}