use std::collections::HashMap; use std::fmt; use std::fmt::Formatter; use serde_derive::{Deserialize, Serialize}; use teloxide::{prelude2::*}; use tracing::{debug, error, info, warn}; use warp::{Rejection, Reply}; use warp::reply::Json; use crate::{Bot, database, DbPool}; #[derive(Deserialize, Serialize)] pub struct SendMessage { token: String, message: String, } impl fmt::Display for SendMessage { fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result { write!(f, "SendMessage {{ token={}, message={} }}", self.token, self.message) } } #[derive(Deserialize, Serialize)] pub struct SendMessageResponse { success: bool, message: String, } impl fmt::Display for SendMessageResponse { fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result { write!(f, "SendMessageResponse {{ success={}, message={} }}", self.success, self.message) } } pub async fn get_handler(params: HashMap, db: DbPool, bot: Bot) -> std::result::Result { let token = params.get("token"); let message = params.get("message"); if token == None || message == None { return Ok(warp::reply::json(&SendMessageResponse { success: false, message: String::from(concat!("Missing parameters. ", "Usage: GET /?token=&message=")), })); } let message = SendMessage { token: token.unwrap().clone(), message: message.unwrap().clone(), }; handler(message, db, bot).await } pub async fn handler(req: SendMessage, db: DbPool, bot: Bot) -> std::result::Result { info!("Income API request: {}", req); let user = database::get_user_by_token(&db, req.token.as_str()).await; let response = match user { Ok(u) => match u { Some(user) => { info!("Send message to user {}.", user); let bot = bot.auto_send(); match bot.send_message(user.chat_id as i64, req.message).await { Ok(_) => SendMessageResponse { success: true, message: String::new(), }, Err(why) => { error!("Failed to send message to telegram: {:?}", why); SendMessageResponse { success: false, message: String::from("Failed to send message to telegram."), } } } } None => { warn!("Invalid token: {}.", req); SendMessageResponse { success: false, message: String::from("Invalid token."), } } }, Err(err) => { error!("Error when querying the database: {:?}.", err); SendMessageResponse { success: false, message: String::from("Invalid parameter."), } } }; debug!("Response: {}", response); Ok(warp::reply::json(&response)) }