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, name: Option, artist: Option, } #[get("/album")] pub async fn get_album( app_state: web::Data, get_args: web::Query, ) -> HttpResponse { /* 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 query_result: sqlx::Result> = app_state.database.select_album_by_id(id).await; return match query_result { Ok(song_list) => HttpResponse::Ok().json(song_list), Err(e) => HttpResponse::Ok().body(format!("{}", e)), }; } /* Handle N results */ let query_result: sqlx::Result, 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 query_result { 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, request_data: web::Json, ) -> HttpResponse { 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, request_data: web::Json, ) -> HttpResponse { 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, request_data: web::Json, ) -> HttpResponse { /* Check if ID is valid (return -1 if invalid) */ let id: i32 = request_data .into_inner() .id .unwrap_or(String::default()) .parse::() .unwrap_or(-1); if id == -1 { return HttpResponse::BadRequest().json(Response { message: "Invalid id value, code not executed\n".to_owned(), }); } get_response_from_query( app_state.database.delete_album(id).await, "POST".to_string(), ) }