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, name: Option, artist: Option, album: Option, } #[get("/song")] pub async fn get_song( 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_song_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> = 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 query_result { 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, request_data: web::Json, ) -> HttpResponse { 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, request_data: web::Json, ) -> HttpResponse { 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, 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".to_owned(), }); } get_response_from_query( app_state.database.delete_song(id).await, "DELETE".to_owned(), ) }