schedtxt_api/src/caller/message/scheduling.rs

pub mod request {
    use serde::{Deserialize, Serialize};
    use utoipa::ToSchema;

    #[derive(Debug, Deserialize, Serialize, ToSchema)]
    pub struct ScheduleMessageRequest {
        #[serde(with = "time::serde::rfc3339::option")]
        pub scheduled: Option<time::OffsetDateTime>,
        pub status: String,
        pub user_id: uuid::Uuid,
    }

    impl ScheduleMessageRequest {
        pub fn is_valid(&self) -> bool {
            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 {
    use serde::{Deserialize, Serialize};
    use utoipa::ToSchema;

    #[derive(Debug, Default, Deserialize, Serialize, ToSchema)]
    pub struct ScheduleMessageResponse {
        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 {
    use crate::repo::scheduling as scheduling_repo;

    /// Endpoint to create a Scheudled Message
    #[utoipa::path(
    post,
    path = crate::caller::endpoints::SCHEDULE_MESSAGE,
    request_body(
        content = super::request::ScheduleMessageRequest,
        description = "Data needed to create a Scheduled Message",
        content_type = "application/json"
        ),
    responses(
        (status = 201, description = "Scheduled Message created", body = super::response::ScheduleMessageResponse),
        (status = 400, description = "Error", body = super::response::ScheduleMessageResponse),
        (status = 500, description = "Error creating Scheduled Message", body = super::response::ScheduleMessageResponse)
    )
)]
    pub async fn schedule_message(
        axum::Extension(pool): axum::Extension<sqlx::PgPool>,
        axum::Json(payload): axum::Json<super::request::ScheduleMessageRequest>,
    ) -> (
        axum::http::StatusCode,
        axum::Json<super::response::ScheduleMessageResponse>,
    ) {
        let mut response = super::response::ScheduleMessageResponse::default();

        if payload.is_valid() {
            if payload.status != textsender_models::message::scheduling::PENDING {
                response.message = String::from("scheduled message must be pending");
                (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
            } else {
                if can_schedule(&payload.scheduled) {
                    let mut scheduled_message =
                        textsender_models::message::scheduling::ScheduledMessage {
                            scheduled: payload.scheduled,
                            status: payload.status,
                            ..Default::default()
                        };
                    match scheduling_repo::insert(&pool, &scheduled_message, &payload.user_id).await
                    {
                        Ok((id, created)) => {
                            scheduled_message.id = id;
                            scheduled_message.created = Some(created);
                            scheduled_message.user_id = payload.user_id;
                            response.message = String::from("Message scheduled");
                            response.data.push(scheduled_message);
                            (axum::http::StatusCode::CREATED, axum::Json(response))
                        }
                        Err(err) => {
                            eprintln!("Error: {err:?}");
                            response.message = String::from("Error inserting");
                            (
                                axum::http::StatusCode::INTERNAL_SERVER_ERROR,
                                axum::Json(response),
                            )
                        }
                    }
                } else {
                    response.message = String::from("Cannot schedule message");
                    (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
                }
            }
        } else {
            response.message = String::from("Request body is not valid");
            (axum::http::StatusCode::BAD_REQUEST, axum::Json(response))
        }
    }

    /// 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))
    }
}