icarus/src/callers/coverart.rs

// TODO: Separate queue and coverart endpoints
#[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
pub struct CoverArtQueue {
    pub id: uuid::Uuid,
    pub song_queue_id: uuid::Uuid,
}

pub mod request {

    pub mod link {
        #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
        pub struct Request {
            pub coverart_id: uuid::Uuid,
            pub song_queue_id: uuid::Uuid,
        }
    }

    pub mod fetch_coverart_no_data {
        #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
        pub struct Params {
            pub id: Option<uuid::Uuid>,
            pub song_queue_id: Option<uuid::Uuid>,
        }
    }

    pub mod fetch_coverart_with_data {
        #[derive(Debug, serde::Deserialize, serde::Serialize)]
        pub struct Params {
            pub id: Option<uuid::Uuid>,
            pub song_queue_id: Option<uuid::Uuid>,
        }
    }

    pub mod create_coverart {
        #[derive(Debug, serde::Deserialize, serde::Serialize)]
        pub struct Request {
            pub song_id: uuid::Uuid,
            pub coverart_queue_id: uuid::Uuid,
        }
    }

    pub mod wipe_data_from_coverart_queue {
        #[derive(Debug, serde::Deserialize, serde::Serialize)]
        pub struct Request {
            pub coverart_queue_id: uuid::Uuid,
        }
    }
}

pub mod response {
    #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
    pub struct Response {
        pub message: String,
        pub data: Vec<uuid::Uuid>,
    }

    pub mod link {
        #[derive(Debug, serde::Deserialize, serde::Serialize)]
        pub struct Id {
            pub coverart_id: uuid::Uuid,
            pub song_queue_id: uuid::Uuid,
        }

        #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
        pub struct Response {
            pub message: String,
            pub data: Vec<Id>,
        }
    }

    pub mod fetch_coverart_no_data {
        #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
        pub struct Response {
            pub message: String,
            pub data: Vec<super::super::CoverArtQueue>,
        }
    }

    pub mod fetch_coverart_with_data {
        #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
        pub struct Response {
            pub message: String,
            pub data: Vec<Vec<u8>>,
        }
    }

    pub mod create_coverart {
        #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
        pub struct Response {
            pub message: String,
            pub data: Vec<icarus_models::coverart::CoverArt>,
        }
    }

    pub mod wipe_data_from_coverart_queue {
        #[derive(Debug, Default, serde::Deserialize, serde::Serialize)]
        pub struct Response {
            pub message: String,
            pub data: Vec<uuid::Uuid>,
        }
    }
}

pub mod db {
    use sqlx::Row;

    pub async fn insert(pool: &sqlx::PgPool, data: &Vec<u8>) -> Result<uuid::Uuid, sqlx::Error> {
        let result = sqlx::query(
            r#"
            INSERT INTO "coverartQueue" (data) VALUES($1) RETURNING id;
            "#,
        )
        .bind(data)
        .fetch_one(pool)
        .await
        .map_err(|e| {
            eprintln!("Error inserting: {e:?}");
        });

        match result {
            Ok(row) => {
                let id: uuid::Uuid = row
                    .try_get("id")
                    .map_err(|_e| sqlx::Error::RowNotFound)
                    .unwrap();
                Ok(id)
            }
            Err(_err) => Err(sqlx::Error::RowNotFound),
        }
    }

    pub async fn update(
        pool: &sqlx::PgPool,
        coverart_id: &uuid::Uuid,
        song_queue_id: &uuid::Uuid,
    ) -> Result<i32, sqlx::Error> {
        let result = sqlx::query(
            r#"
            UPDATE "coverartQueue" SET song_queue_id = $1 WHERE id = $2;
            "#,
        )
        .bind(song_queue_id)
        .bind(coverart_id)
        .execute(pool)
        .await;

        match result {
            Ok(_) => Ok(0),
            Err(_err) => Err(sqlx::Error::RowNotFound),
        }
    }

    pub async fn get_coverart_queue_with_id(
        pool: &sqlx::PgPool,
        id: &uuid::Uuid,
    ) -> Result<super::CoverArtQueue, sqlx::Error> {
        let result = sqlx::query(
            r#"
            SELECT id, song_queue_id FROM "coverartQueue" WHERE id = $1;
            "#,
        )
        .bind(id)
        .fetch_one(pool)
        .await
        .map_err(|e| {
            eprintln!("Error querying data: {e:?}");
        });

        match result {
            Ok(row) => Ok(super::CoverArtQueue {
                id: row
                    .try_get("id")
                    .map_err(|_e| sqlx::Error::RowNotFound)
                    .unwrap(),
                song_queue_id: row
                    .try_get("song_queue_id")
                    .map_err(|_e| sqlx::Error::RowNotFound)
                    .unwrap(),
            }),
            Err(_) => Err(sqlx::Error::RowNotFound),
        }
    }

    pub async fn get_coverart_queue_with_song_queue_id(
        pool: &sqlx::PgPool,
        song_queue_id: &uuid::Uuid,
    ) -> Result<super::CoverArtQueue, sqlx::Error> {
        let result = sqlx::query(
            r#"
            SELECT id, song_queue_id FROM "coverartQueue" WHERE song_queue_id = $1;
            "#,
        )
        .bind(song_queue_id)
        .fetch_one(pool)
        .await
        .map_err(|e| {
            eprintln!("Error querying data: {e:?}");
        });

        match result {
            Ok(row) => Ok(super::CoverArtQueue {
                id: row
                    .try_get("id")
                    .map_err(|_e| sqlx::Error::RowNotFound)
                    .unwrap(),
                song_queue_id: row
                    .try_get("song_queue_id")
                    .map_err(|_e| sqlx::Error::RowNotFound)
                    .unwrap(),
            }),
            Err(_) => Err(sqlx::Error::RowNotFound),
        }
    }

    pub async fn get_coverart_queue_data_with_id(
        pool: &sqlx::PgPool,
        id: &uuid::Uuid,
    ) -> Result<Vec<u8>, sqlx::Error> {
        let result = sqlx::query(
            r#"
            SELECT data FROM "coverartQueue" WHERE id = $1;
            "#,
        )
        .bind(id)
        .fetch_one(pool)
        .await
        .map_err(|e| {
            eprintln!("Error querying data: {e:?}");
        });

        match result {
            Ok(row) => Ok(row
                .try_get("data")
                .map_err(|_e| sqlx::Error::RowNotFound)
                .unwrap()),
            Err(_) => Err(sqlx::Error::RowNotFound),
        }
    }

    pub async fn get_coverart_queue_data_with_song_queue_id(
        pool: &sqlx::PgPool,
        song_queue_id: &uuid::Uuid,
    ) -> Result<Vec<u8>, sqlx::Error> {
        let result = sqlx::query(
            r#"
            SELECT data FROM "coverartQueue" WHERE song_queue_id = $1;
            "#,
        )
        .bind(song_queue_id)
        .fetch_one(pool)
        .await
        .map_err(|e| {
            eprintln!("Error querying data: {e}");
        });

        match result {
            Ok(row) => Ok(row
                .try_get("data")
                .map_err(|_e| sqlx::Error::RowNotFound)
                .unwrap()),
            Err(_) => Err(sqlx::Error::RowNotFound),
        }
    }

    pub async fn wipe_data(
        pool: &sqlx::PgPool,
        coverart_queue_id: &uuid::Uuid,
    ) -> Result<uuid::Uuid, sqlx::Error> {
        let result = sqlx::query(
            r#"
            UPDATE "coverartQueue" SET data = NULL WHERE id = $1 RETURNING id;
            "#,
        )
        .bind(coverart_queue_id)
        .fetch_one(pool)
        .await
        .map_err(|e| {
            eprintln!("Error updating query: {e}");
        });

        match result {
            Ok(row) => Ok(row
                .try_get("id")
                .map_err(|_e| sqlx::Error::RowNotFound)
                .unwrap()),
            Err(_) => Err(sqlx::Error::RowNotFound),
        }
    }
}

pub mod cov_db {
    use sqlx::Row;

    pub async fn create(
        pool: &sqlx::PgPool,
        coverart: &icarus_models::coverart::CoverArt,
        song_id: &uuid::Uuid,
    ) -> Result<uuid::Uuid, sqlx::Error> {
        let result = sqlx::query(
            r#"
            INSERT INTO "coverart" (title, path, song_id) VALUES($1, $2, $3) RETURNING id;
            "#,
        )
        .bind(&coverart.title)
        .bind(&coverart.path)
        .bind(song_id)
        .fetch_one(pool)
        .await
        .map_err(|e| {
            eprintln!("Error inserting: {e}");
        });

        match result {
            Ok(row) => {
                let id: uuid::Uuid = row
                    .try_get("id")
                    .map_err(|_e| sqlx::Error::RowNotFound)
                    .unwrap();
                Ok(id)
            }
            Err(_err) => Err(sqlx::Error::RowNotFound),
        }
    }
}

pub mod endpoint {
    use std::io::Write;

    use axum::response::IntoResponse;

    pub async fn queue(
        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
        mut multipart: axum::extract::Multipart,
    ) -> (
        axum::http::StatusCode,
        axum::Json<super::response::Response>,
    ) {
        let mut response = super::response::Response::default();

        match multipart.next_field().await {
            Ok(Some(field)) => {
                let name = field.name().unwrap().to_string();
                let file_name = field.file_name().unwrap().to_string();
                let content_type = field.content_type().unwrap().to_string();
                let data = field.bytes().await.unwrap();
                let raw_data = data.to_vec();

                println!(
                    "Received file '{}' (name = '{}', content-type = '{}', size = {})",
                    file_name,
                    name,
                    content_type,
                    data.len()
                );

                match super::db::insert(&pool, &raw_data).await {
                    Ok(id) => {
                        response.message = String::from("Successful");
                        response.data.push(id);
                        (axum::http::StatusCode::OK, axum::Json(response))
                    }
                    Err(err) => {
                        response.message = err.to_string();
                        (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                    }
                }
            }
            Ok(None) => (axum::http::StatusCode::BAD_REQUEST, axum::Json(response)),
            Err(err) => {
                response.message = err.to_string();
                (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
            }
        }
    }

    pub async fn link(
        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
        axum::Json(payload): axum::Json<super::request::link::Request>,
    ) -> (
        axum::http::StatusCode,
        axum::Json<super::response::link::Response>,
    ) {
        let mut response = super::response::link::Response::default();
        let id = payload.coverart_id;
        let song_id = payload.song_queue_id;

        match super::db::update(&pool, &id, &song_id).await {
            Ok(_o) => {
                response.data.push(super::response::link::Id {
                    song_queue_id: song_id,
                    coverart_id: id,
                });

                (axum::http::StatusCode::OK, axum::Json(response))
            }
            Err(err) => {
                response.message = err.to_string();
                (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
            }
        }
    }

    pub async fn fetch_coverart_no_data(
        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
        axum::extract::Query(params): axum::extract::Query<
            super::request::fetch_coverart_no_data::Params,
        >,
    ) -> (
        axum::http::StatusCode,
        axum::Json<super::response::fetch_coverart_no_data::Response>,
    ) {
        let mut response = super::response::fetch_coverart_no_data::Response::default();

        match params.id {
            Some(id) => match super::db::get_coverart_queue_with_id(&pool, &id).await {
                Ok(cover_art_queue) => {
                    response.message = String::from("Successful");
                    response.data.push(cover_art_queue);
                    (axum::http::StatusCode::OK, axum::Json(response))
                }
                Err(err) => {
                    response.message = err.to_string();
                    (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                }
            },
            _ => match params.song_queue_id {
                Some(song_queue_id) => {
                    match super::db::get_coverart_queue_with_song_queue_id(&pool, &song_queue_id)
                        .await
                    {
                        Ok(cover_art_queue) => {
                            response.message = String::from("Successful");
                            response.data.push(cover_art_queue);
                            (axum::http::StatusCode::OK, axum::Json(response))
                        }
                        Err(err) => {
                            response.message = err.to_string();
                            (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                        }
                    }
                }
                None => {
                    response.message = String::from("No valid id provided");
                    (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                }
            },
        }
    }

    pub async fn fetch_coverart_with_data(
        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
        axum::extract::Path(id): axum::extract::Path<uuid::Uuid>,
    ) -> (axum::http::StatusCode, axum::response::Response) {
        match super::db::get_coverart_queue_data_with_id(&pool, &id).await {
            Ok(data) => {
                let bytes = axum::body::Bytes::from(data);
                let mut response = bytes.into_response();
                let headers = response.headers_mut();
                // TODO: Address this hard coding for the coverart content type
                headers.insert(axum::http::header::CONTENT_TYPE, "image".parse().unwrap());
                // TODO: Make the conent disposition more dynamic
                headers.insert(
                    axum::http::header::CONTENT_DISPOSITION,
                    format!("attachment; filename=\"{id}.jpg\"")
                        .parse()
                        .unwrap(),
                );

                (axum::http::StatusCode::OK, response)
            }
            Err(_err) => (
                axum::http::StatusCode::BAD_REQUEST,
                axum::response::Response::default(),
            ),
        }
    }

    pub async fn create_coverart(
        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
        axum::Json(payload): axum::Json<super::request::create_coverart::Request>,
    ) -> (
        axum::http::StatusCode,
        axum::Json<super::response::create_coverart::Response>,
    ) {
        let mut response = super::response::create_coverart::Response::default();
        let id = payload.coverart_queue_id;

        match super::db::get_coverart_queue_data_with_id(&pool, &id).await {
            Ok(data) => {
                let song_id = payload.song_id;
                match crate::callers::song::song_db::get_song(&pool, &song_id).await {
                    Ok(song) => {
                        let directory = icarus_envy::environment::get_root_directory().await;
                        let dir = std::path::Path::new(&directory);

                        // TODO: Make this random and the file extension should not be hard coded
                        let filename = format!("{}-coverart.jpeg", &song.filename[..8]);
                        let save_path = dir.join(&filename);
                        let path = String::from(save_path.to_str().unwrap());
                        let mut coverart =
                            icarus_models::coverart::init::init_coverart_only_path(path);
                        coverart.title = song.album.clone();
                        coverart.data = data;

                        let mut file = std::fs::File::create(&save_path).unwrap();
                        file.write_all(&coverart.data).unwrap();

                        match super::cov_db::create(&pool, &coverart, &song.id).await {
                            Ok(id) => {
                                coverart.id = id;
                                response.message = String::from("Successful");
                                response.data.push(coverart);

                                (axum::http::StatusCode::OK, axum::Json(response))
                            }
                            Err(err) => {
                                response.message = err.to_string();
                                (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                            }
                        }
                    }
                    Err(err) => {
                        response.message = err.to_string();
                        (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                    }
                }
            }
            Err(err) => {
                response.message = err.to_string();
                (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
            }
        }
    }

    pub async fn wipe_data_from_coverart_queue(
        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
        axum::Json(payload): axum::Json<super::request::wipe_data_from_coverart_queue::Request>,
    ) -> (
        axum::http::StatusCode,
        axum::Json<super::response::wipe_data_from_coverart_queue::Response>,
    ) {
        let mut response = super::response::wipe_data_from_coverart_queue::Response::default();
        let coverart_queue_id = payload.coverart_queue_id;

        match super::db::get_coverart_queue_with_id(&pool, &coverart_queue_id).await {
            Ok(coverart_queue) => match super::db::wipe_data(&pool, &coverart_queue.id).await {
                Ok(id) => {
                    response.message = String::from("Success");
                    response.data.push(id);
                    (axum::http::StatusCode::OK, axum::Json(response))
                }
                Err(err) => {
                    response.message = err.to_string();
                    (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                }
            },
            Err(err) => {
                response.message = err.to_string();
                (axum::http::StatusCode::NOT_FOUND, axum::Json(response))
            }
        }
    }
}