use std::env;
use dotenvy::dotenv;
use serde::{Deserialize, Serialize};
use tgbot::{
api::Client,
handler::{LongPoll, UpdateHandler},
types::{AnswerCallbackQuery, InlineKeyboardButton, Message, SendMessage, Update, UpdateType},
};
struct Handler {
client: Client,
}
#[derive(Deserialize, Serialize)]
struct CallbackData {
value: String,
}
impl CallbackData {
fn new<S: Into<String>>(value: S) -> Self {
Self { value: value.into() }
}
}
async fn handle_update(client: &Client, update: Update) -> Option<Message> {
match update.update_type {
UpdateType::Message(message) => {
let chat_id = message.chat.get_id();
if let Some(commands) = message.get_text().and_then(|text| text.get_bot_commands()) {
let command = &commands[0];
if command.command == "/start" {
let callback_data = CallbackData::new("Hello!");
let method = SendMessage::new(chat_id, "Press the button").with_reply_markup([[
InlineKeyboardButton::for_callback_data_struct("Greet", &callback_data).unwrap(),
]]);
return Some(client.execute(method).await.unwrap());
}
}
}
UpdateType::CallbackQuery(query) => {
let data = query.parse_data::<CallbackData>().unwrap().unwrap(); let method = AnswerCallbackQuery::new(query.id).with_text(data.value);
client.execute(method).await.unwrap();
}
_ => {}
}
None
}
impl UpdateHandler for Handler {
async fn handle(&self, update: Update) {
log::info!("Got an update: {update:?}");
if let Some(msg) = handle_update(&self.client, update).await {
log::info!("Message sent: {msg:?}");
}
}
}
#[tokio::main]
async fn main() {
dotenv().ok();
env_logger::init();
let token = env::var("TGBOT_TOKEN").expect("TGBOT_TOKEN is not set");
let client = Client::new(token).expect("Failed to create API");
LongPoll::new(client.clone(), Handler { client }).run().await;
}