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, 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, pub user_id: Option, } } 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, } #[derive(Debug, Default, Deserialize, Serialize, ToSchema)] pub struct GetScheduledMessageResponse { pub message: String, pub data: Vec, } } pub mod endpoint { use crate::repo::message::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, axum::Json(payload): axum::Json, ) -> ( axum::http::StatusCode, axum::Json, ) { 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, axum::extract::Query(params): axum::extract::Query< super::request::GetScheduledMessageParams, >, ) -> ( axum::http::StatusCode, axum::Json, ) { 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) -> bool { scheduled_time .is_some_and(|t| t - time::OffsetDateTime::now_utc() >= time::Duration::minutes(10)) } }