diff --git a/src/callers/song.rs b/src/callers/song.rs index fdf78e0..ee52b08 100644 --- a/src/callers/song.rs +++ b/src/callers/song.rs @@ -77,6 +77,13 @@ pub mod request { } } } + + pub mod wipe_data_from_song_queue { + #[derive(Debug, serde::Deserialize, serde::Serialize)] + pub struct Request { + pub song_queue_id: uuid::Uuid + } + } } pub mod response { @@ -127,6 +134,14 @@ pub mod response { pub data: Vec, } } + + pub mod wipe_data_from_song_queue { + #[derive(Debug, Default, serde::Deserialize, serde::Serialize)] + pub struct Response { + pub message: String, + pub data: Vec + } + } } // TODO: Might make a distinction between year and date in a song's tag at some point @@ -459,6 +474,59 @@ mod song_queue { } } + pub async fn get_song_queue(pool: &sqlx::PgPool, id: &uuid::Uuid) -> Result { + let result = sqlx::query( + r#" + SELECT id, filename, status FROM "songQueue" WHERE id = $1 + "# + ) + .bind(&id) + .fetch_one(pool) + .await + .map_err(|e| { + eprintln!("Error querying data: {:?}", e); + }); + + match result { + Ok(row) => Ok(SongQueue { + id: row + .try_get("id") + .map_err(|_e| sqlx::Error::RowNotFound) + .unwrap(), + filename: row + .try_get("filename") + .map_err(|_e| sqlx::Error::RowNotFound) + .unwrap(), + status: row + .try_get("status") + .map_err(|_e| sqlx::Error::RowNotFound) + .unwrap(), + }), + Err(_err) => Err(sqlx::Error::RowNotFound), + } + } + + pub async fn wipe_data(pool: &sqlx::PgPool, id: &uuid::Uuid) -> Result { + let result = sqlx::query( + r#" + UPDATE "songQueue" SET data = NULL WHERE id = $1 RETURNING id; + "# + ) + .bind(&id) + .fetch_one(pool) + .await + .map_err(|e| { + eprintln!("Error updating record: {:?}", e); + }); + + match result { + Ok(row) => { + Ok(row.try_get("id").map_err(|_e| sqlx::Error::RowNotFound).unwrap()) + } + Err(_) => Err(sqlx::Error::RowNotFound) + } + } + pub async fn get_data(pool: &sqlx::PgPool, id: &uuid::Uuid) -> Result, sqlx::Error> { let result = sqlx::query( r#" @@ -736,4 +804,33 @@ pub mod endpoint { (axum::http::StatusCode::BAD_REQUEST, axum::Json(response)) } } + + pub async fn wipe_data_from_song_queue( + axum::Extension(pool): axum::Extension, + axum::Json(payload): axum::Json, + ) -> (axum::http::StatusCode, axum::Json,) { + let mut response = super::response::wipe_data_from_song_queue::Response::default(); + let id = payload.song_queue_id; + + match super::song_queue::get_song_queue(&pool, &id).await { + Ok(song_queue) => { + match super::song_queue::wipe_data(&pool, &song_queue.id).await { + Ok(wiped_id) => { + response.message = "Success"; + response.data.push(wiped_id); + + (axum::http::StatusCode::OK, axum::Json(response)) + } + Err(err) => { + response.message = err.to_string(); + (axum::http::StatusCode::NOT_FOUND, axum::Json(response)) + } + } + } + Err(err) => { + response.message = err.to_string(); + (axum::http::StatusCode::NOT_FOUND, axum::Json(response)) + } + } + } }