92 lines
2.3 KiB
Rust
92 lines
2.3 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
use teloxide::types::InlineKeyboardButton;
|
|
|
|
use crate::{
|
|
db::{callback_info::CallbackInfo, CallDB},
|
|
BotResult,
|
|
};
|
|
|
|
#[macro_export]
|
|
macro_rules! single_button_markup {
|
|
($button:expr) => {
|
|
InlineKeyboardMarkup {
|
|
inline_keyboard: vec![vec![$button]],
|
|
}
|
|
};
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! stacked_buttons_markup {
|
|
($( $button:expr ),+) => {
|
|
InlineKeyboardMarkup {
|
|
inline_keyboard: vec![
|
|
$(
|
|
vec![$button],
|
|
)*
|
|
],
|
|
}
|
|
};
|
|
}
|
|
|
|
#[macro_export]
|
|
macro_rules! buttons_markup {
|
|
($( $buttons:expr ),+) => {
|
|
InlineKeyboardMarkup {
|
|
inline_keyboard: vec![
|
|
$(
|
|
$buttons.into_iter().collect::<Vec<_>>(),
|
|
)*
|
|
],
|
|
}
|
|
};
|
|
}
|
|
|
|
pub async fn create_callback_button<C, D>(
|
|
literal: &str,
|
|
callback: C,
|
|
db: &mut D,
|
|
) -> BotResult<InlineKeyboardButton>
|
|
where
|
|
C: Serialize + for<'a> Deserialize<'a> + Send + Sync,
|
|
D: CallDB + Send,
|
|
{
|
|
let text = db
|
|
.get_literal_value(literal)
|
|
.await?
|
|
.unwrap_or("Please, set content of this message".into());
|
|
let ci = CallbackInfo::new_with_literal(callback, literal.to_string())
|
|
.store(db)
|
|
.await?;
|
|
|
|
Ok(InlineKeyboardButton::new(
|
|
text,
|
|
teloxide::types::InlineKeyboardButtonKind::CallbackData(ci.get_id()),
|
|
))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use teloxide::types::InlineKeyboardButton;
|
|
use teloxide::types::InlineKeyboardMarkup;
|
|
|
|
#[test]
|
|
fn test_buttons_markup() {
|
|
let button1 = InlineKeyboardButton::new(
|
|
"Button 1",
|
|
teloxide::types::InlineKeyboardButtonKind::CallbackData("callback1".into()),
|
|
);
|
|
let button2 = InlineKeyboardButton::new(
|
|
"Button 2",
|
|
teloxide::types::InlineKeyboardButtonKind::CallbackData("callback2".into()),
|
|
);
|
|
|
|
let markup = buttons_markup!([button1.clone(), button2.clone()], [button1.clone()]);
|
|
|
|
assert_eq!(markup.inline_keyboard.len(), 2);
|
|
assert_eq!(markup.inline_keyboard[0][0].text, "Button 1");
|
|
assert_eq!(markup.inline_keyboard[0][1].text, "Button 2");
|
|
assert_eq!(markup.inline_keyboard[1][0].text, "Button 1");
|
|
}
|
|
}
|