1 files changed, 96 insertions, 0 deletions
diff --git a/src/api/song.rs b/src/api/song.rs
new file mode 100644
index 0000000..850c759
--- /dev/null
+++ b/src/api/song.rs
@@ -0,0 +1,96 @@
+use crate::database::{Delete, Song, SongPost, SongPut};
+use crate::AppState;
+use actix_web::{delete, get, post, put, web, HttpResponse};
+use serde::Deserialize;
+
+#[derive(Deserialize)]
+struct SongQueryOptions {
+    id: Option<String>,
+    name: Option<String>,
+    artist: Option<String>,
+    album: Option<String>,
+}
+
+#[get("/song")]
+pub async fn get_song(
+    app_state: web::Data<AppState>,
+    get_args: web::Query<SongQueryOptions>,
+) -> HttpResponse {
+    let default = String::from("");
+
+    if get_args.id.is_some() {
+        let id: &str = get_args.id.as_ref().unwrap_or(&default);
+        let search_attempt: sqlx::Result<Option<Song>> =
+            app_state.database.select_song_by_id(id).await;
+
+        return match search_attempt {
+            Ok(song_list) => HttpResponse::Ok().json(song_list),
+            Err(e) => HttpResponse::Ok().body(format!("{}", e)),
+        };
+    }
+
+    let search_attempt: 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
+        }
+        _ if get_args.album.is_some() => {
+            let album: &str = get_args.album.as_ref().unwrap_or(&default);
+            app_state.database.select_songs_by_album(album).await
+        }
+        _ if get_args.artist.is_some() => {
+            let artist: &str = get_args.artist.as_ref().unwrap_or(&default);
+            app_state.database.select_songs_by_artist(artist).await
+        }
+        _ => app_state.database.select_songs().await,
+    };
+
+    match search_attempt {
+        Ok(song_list) => HttpResponse::Ok().json(song_list),
+        Err(e) => HttpResponse::Ok().body(format!("{}", e)),
+    }
+}
+
+#[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)),
+    }
+}