1 files changed, 94 insertions, 0 deletions
diff --git a/src/api/album.rs b/src/api/album.rs
new file mode 100644
index 0000000..d9d0d52
--- /dev/null
+++ b/src/api/album.rs
@@ -0,0 +1,94 @@
+use crate::database::{Album, AlbumPost, AlbumPut, Delete};
+use crate::AppState;
+use actix_web::{delete, get, post, put, web, HttpResponse};
+use serde::Deserialize;
+
+#[derive(Deserialize)]
+struct AlbumQueryOptions {
+    id: Option<String>,
+    name: Option<String>,
+    artist: Option<String>,
+}
+
+#[get("/album")]
+pub async fn get_album(
+    app_state: web::Data<AppState>,
+    get_args: web::Query<AlbumQueryOptions>,
+) -> 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<Album>> =
+            app_state.database.select_album_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<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
+        }
+        _ if get_args.artist.is_some() => {
+            let artist: &str = &get_args.artist.as_ref().unwrap_or(&default);
+            app_state.database.select_albums_by_artist(artist).await
+        }
+        _ => app_state.database.select_albums().await,
+    };
+
+    match search_attempt {
+        Ok(album_list) => HttpResponse::Ok().json(album_list),
+        Err(e) => HttpResponse::Ok().body(format!("{}", e)),
+    }
+}
+
+#[post("/album")]
+pub async fn post_album(
+    app_state: web::Data<AppState>,
+    post_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)),
+    }
+}
+
+#[put("/album")]
+pub async fn put_album(
+    app_state: web::Data<AppState>,
+    post_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)),
+    }
+}
+
+#[delete("/album")]
+pub async fn delete_album(
+    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_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)),
+    }
+}