Get schedule message endpoint (#16)

Reviewed-on: phoenix/textsender_api#16

src/caller/message/scheduling.rs

@@ -15,6 +15,12 @@ pub mod request {
             self.scheduled.is_some() || !self.status.is_empty() || !self.user_id.is_nil()
         }
     }
+
+    #[derive(Debug, Default, serde::Deserialize, serde::Serialize, utoipa::ToSchema)]
+    pub struct GetScheduledMessageParams {
+        pub id: Option<uuid::Uuid>,
+        pub user_id: Option<uuid::Uuid>,
+    }
 }
 
 pub mod response {
@@ -26,6 +32,12 @@ pub mod response {
         pub message: String,
         pub data: Vec<textsender_models::message::scheduling::ScheduledMessage>,
     }
+
+    #[derive(Debug, Default, Deserialize, Serialize, ToSchema)]
+    pub struct GetScheduledMessageResponse {
+        pub message: String,
+        pub data: Vec<textsender_models::message::scheduling::ScheduledMessage>,
+    }
 }
 
 pub mod endpoint {
@@ -97,6 +109,67 @@ pub mod endpoint {
         }
     }
 
+    /// Endpoint to get Scheduled Message
+    #[utoipa::path(
+        get,
+        path = crate::caller::endpoints::GET_SCHEDULE_MESSAGE,
+        params(
+            ("id" = uuid::Uuid, Path, description = "Id of Scheduled Message"),
+            ("user_id" = uuid::Uuid, Path, description = "User Id associated with the Scheduled Message")
+            ),
+        responses(
+            (status = 200, description = "Scheduled Message found", body = super::response::GetScheduledMessageResponse),
+            (status = 400, description = "Error getting Scheduled Message", body = super::response::GetScheduledMessageResponse)
+        )
+    )]
+    pub async fn get_scheduled_messages(
+        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
+        axum::extract::Query(params): axum::extract::Query<
+            super::request::GetScheduledMessageParams,
+        >,
+    ) -> (
+        axum::http::StatusCode,
+        axum::Json<super::response::GetScheduledMessageResponse>,
+    ) {
+        let mut response = super::response::GetScheduledMessageResponse::default();
+
+        let scheduled_messages = match params.id {
+            Some(id) => match scheduling_repo::get(&pool, &id).await {
+                Ok(scheduled_message) => {
+                    vec![scheduled_message]
+                }
+                Err(err) => {
+                    eprintln!("Error: {err:?}");
+                    return (
+                        axum::http::StatusCode::INTERNAL_SERVER_ERROR,
+                        axum::Json(response),
+                    );
+                }
+            },
+            None => match params.user_id {
+                Some(user_id) => match scheduling_repo::get_with_user_id(&pool, &user_id).await {
+                    Ok(scheduled_messages) => scheduled_messages,
+                    Err(err) => {
+                        eprintln!("Error: {err:?}");
+                        return (
+                            axum::http::StatusCode::INTERNAL_SERVER_ERROR,
+                            axum::Json(response),
+                        );
+                    }
+                },
+                None => {
+                    response.message = String::from("Invalid parameter");
+                    return (axum::http::StatusCode::BAD_REQUEST, axum::Json(response));
+                }
+            },
+        };
+
+        response.data = scheduled_messages;
+        response.message = String::from(super::super::super::response::SUCCESSFUL);
+
+        (axum::http::StatusCode::OK, axum::Json(response))
+    }
+
     fn can_schedule(scheduled_time: &Option<time::OffsetDateTime>) -> bool {
         scheduled_time
             .is_some_and(|t| t - time::OffsetDateTime::now_utc() >= time::Duration::minutes(10))

src/caller/mod.rs

@@ -16,6 +16,8 @@ pub mod endpoints {
     pub const GET_MESSAGE: &str = "/api/v1/message";
     /// Constant for scheduling message endpoint
     pub const SCHEDULE_MESSAGE: &str = "/api/v1/schedule/message";
+    /// Constant for getting scheduled message endpoint
+    pub const GET_SCHEDULE_MESSAGE: &str = "/api/v1/schedule/message";
 }
 
 pub mod response {

src/config/mod.rs

@@ -130,6 +130,12 @@ pub mod init {
                     axum::middleware::from_fn(crate::auth::auth::<axum::body::Body>),
                 ),
             )
+            .route(
+                crate::caller::endpoints::GET_SCHEDULE_MESSAGE,
+                get(scheduling_endpoints::get_scheduled_messages).route_layer(
+                    axum::middleware::from_fn(crate::auth::auth::<axum::body::Body>),
+                ),
+            )
             .layer(cors::configure_cors().await)
     }
 

src/repo/scheduling.rs

@@ -25,3 +25,79 @@ pub async fn insert(
         Err(_) => Err(sqlx::Error::RowNotFound),
     }
 }
+
+pub async fn get(
+    pool: &sqlx::PgPool,
+    id: &uuid::Uuid,
+) -> Result<textsender_models::message::scheduling::ScheduledMessage, sqlx::Error> {
+    match sqlx::query(
+        r#"
+        SELECT id, scheduled, created, status, user_id FROM "scheduled_messages"
+        WHERE
+            id = $1;
+        "#,
+    )
+    .bind(id)
+    .fetch_one(pool)
+    .await
+    {
+        Ok(row) => match parse_row(&row).await {
+            Ok(scheduled_message) => Ok(scheduled_message),
+            Err(err) => Err(err),
+        },
+        Err(_) => Err(sqlx::Error::RowNotFound),
+    }
+}
+
+pub async fn get_with_user_id(
+    pool: &sqlx::PgPool,
+    user_id: &uuid::Uuid,
+) -> Result<Vec<textsender_models::message::scheduling::ScheduledMessage>, sqlx::Error> {
+    match sqlx::query(
+        r#"
+        SELECT id, scheduled, created, status, user_id FROM "scheduled_messages"
+        WHERE
+            user_id = $1;
+        "#,
+    )
+    .bind(user_id)
+    .fetch_all(pool)
+    .await
+    {
+        Ok(rows) => {
+            let mut messages: Vec<textsender_models::message::scheduling::ScheduledMessage> =
+                Vec::new();
+            for row in rows {
+                match parse_row(&row).await {
+                    Ok(scheduled_message) => {
+                        messages.push(scheduled_message);
+                    }
+                    Err(err) => {
+                        return Err(err);
+                    }
+                }
+            }
+
+            Ok(messages)
+        }
+        Err(_) => Err(sqlx::Error::RowNotFound),
+    }
+}
+
+async fn parse_row(
+    row: &sqlx::postgres::PgRow,
+) -> Result<textsender_models::message::scheduling::ScheduledMessage, sqlx::Error> {
+    let id: uuid::Uuid = row.try_get("id")?;
+    let scheduled: time::OffsetDateTime = row.try_get("scheduled")?;
+    let status: String = row.try_get("status")?;
+    let created: time::OffsetDateTime = row.try_get("created")?;
+    let user_id: uuid::Uuid = row.try_get("user_id")?;
+
+    Ok(textsender_models::message::scheduling::ScheduledMessage {
+        id,
+        scheduled: Some(scheduled),
+        created: Some(created),
+        status,
+        user_id,
+    })
+}