OpenSource/teloxide
OpenSource/teloxide
1
0
Fork 0
mirror of https://github.com/teloxide/teloxide.git synced 2024-12-22 06:25:10 +01:00
Code Issues Projects Releases Packages Wiki Activity

Refactor the composite state example a bit

Browse source
This commit is contained in:
Hirrolot 2024-09-23 01:58:49 -04:00
parent de34970e70
commit 782ec00f32
No known key found for this signature in database
GPG key ID: F0C33B48BD883C81
1 changed files with 12 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
crates/teloxide/examples/composite_state.rs
View file

@ -96,26 +96,20 @@ fn schema() -> UpdateHandler {
teloxide::filter_command::<Command, _>()
.branch(dptree::case![Command::Start].endpoint(ask_full_name)),
)
.branch(dptree::case![GlobalState::Idle].endpoint(handle_configured_user_message))
.branch(dptree::case![GlobalState::Idle].endpoint(on_idle))
.branch(
// Its essential not to use
// `dptree::case![GlobalState::UserSetup(UserSetup::ReceiveFullName)]` directly,
// this won't work. Each nested enum requires it's own `branch`
// scope. Actually, each `dptree::case![..]` introduces the inner
// enum value to the `DependencyMap`, so there is an option
// to branch on the inner values freely.
dptree::case![GlobalState::UserSetup(_state)]
// This macro passes the `state` variable further. In order to pattern-match on it,
// we need to use `dptree::case!` two times again.
dptree::case![GlobalState::UserSetup(state)]
.branch(
dptree::case![UserSetup::ReceiveFullName]
.map(|text: String| FullName(text))
.endpoint(ask_age),
dptree::case![UserSetup::ReceiveFullName].map(FullName).endpoint(ask_age),
)
.branch(
dptree::case![UserSetup::ReceiveAge { full_name }]
.endpoint(finish_user_setup),
),
)
.branch(dptree::endpoint(handle_unconfigured_user_message)),
.branch(dptree::endpoint(fallback)),
)
}
@ -145,22 +139,22 @@ async fn finish_user_setup(
let _age = match age.parse::<u8>() {
Ok(age) => age,
Err(_err) => {
bot.send_message(message.chat.id, "Please, enter your age").await?;
bot.send_message(message.chat.id, "Please, enter your real age.").await?;
return Ok(());
}
};
bot.send_message(message.chat.id, "You've successfully passed setup").await?;
bot.send_message(message.chat.id, "Your accout is set up!").await?;
dialogue.update(GlobalState::Idle).await?;
Ok(())
}
async fn handle_configured_user_message(bot: Bot, message: Message) -> HandlerResult {
bot.send_message(message.chat.id, "Hi, configured user!").await?;
async fn on_idle(bot: Bot, message: Message) -> HandlerResult {
bot.send_message(message.chat.id, "You are all set.").await?;
Ok(())
}
async fn handle_unconfigured_user_message(bot: Bot, message: Message) -> HandlerResult {
bot.send_message(message.chat.id, "Use /start to setup your account").await?;
async fn fallback(bot: Bot, message: Message) -> HandlerResult {
bot.send_message(message.chat.id, "Use /start to setup your account.").await?;
Ok(())
}