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

Update examples/dialogue_bot

Browse source
This commit is contained in:
Temirkhan Myrzamadi 2020-05-24 01:06:04 +06:00
parent b0940df9bd
commit ddd94543d1
1 changed files with 20 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
examples/dialogue_bot/src/main.rs
View file

@ -79,7 +79,7 @@ impl ReceiveFavouriteMusicState {
#[derive(Display)] #[derive(Display)]
#[display( #[display(
"Your full name: {full_name}, your age: {age}, your favourite music: \ "Your full name: {full_name}, your age: {age}, your favourite music: \
{favourite_music}" {favourite_music}"
)] )]
struct ExitState { struct ExitState {
@ -105,11 +105,15 @@ type Cx<State> = DialogueDispatcherHandlerCx<Message, State, Infallible>;
type Res = ResponseResult<DialogueStage<Dialogue>>; type Res = ResponseResult<DialogueStage<Dialogue>>;
async fn start(cx: Cx<()>) -> Res { async fn start(cx: Cx<()>) -> Res {
let DialogueDispatcherHandlerCx { cx, .. } = cx;
cx.answer("Let's start! First, what's your full name?").send().await?; cx.answer("Let's start! First, what's your full name?").send().await?;
next(Dialogue::ReceiveFullName) next(Dialogue::ReceiveFullName)
} }
async fn full_name(cx: Cx<()>) -> Res { async fn full_name(cx: Cx<()>) -> Res {
let DialogueDispatcherHandlerCx { cx, .. } = cx;
match cx.update.text() { match cx.update.text() {
None => { None => {
cx.answer("Please, send me a text message!").send().await?; cx.answer("Please, send me a text message!").send().await?;
@ -125,57 +129,59 @@ async fn full_name(cx: Cx<()>) -> Res {
} }
async fn age(cx: Cx<ReceiveAgeState>) -> Res { async fn age(cx: Cx<ReceiveAgeState>) -> Res {
let DialogueDispatcherHandlerCx { cx, dialogue } = cx;
let dialogue = dialogue.unwrap();
match cx.update.text().unwrap().parse() { match cx.update.text().unwrap().parse() {
Ok(age) => { Ok(age) => {
cx.answer("Good. Now choose your favourite music:") cx.answer("Good. Now choose your favourite music:")
.reply_markup(FavouriteMusic::markup()) .reply_markup(FavouriteMusic::markup())
.send() .send()
.await?; .await?;
next(Dialogue::ReceiveFavouriteMusic(cx.dialogue.unwrap().up(age))) next(Dialogue::ReceiveFavouriteMusic(dialogue.up(age)))
} }
Err(_) => { Err(_) => {
cx.answer("Oh, please, enter a number!").send().await?; cx.answer("Oh, please, enter a number!").send().await?;
next(Dialogue::ReceiveAge(cx.dialogue.unwrap())) next(Dialogue::ReceiveAge(dialogue))
} }
} }
} }
async fn favourite_music(cx: Cx<ReceiveFavouriteMusicState>) -> Res { async fn favourite_music(cx: Cx<ReceiveFavouriteMusicState>) -> Res {
let DialogueDispatcherHandlerCx { cx, dialogue } = cx;
let dialogue = dialogue.unwrap();
match cx.update.text().unwrap().parse() { match cx.update.text().unwrap().parse() {
Ok(favourite_music) => { Ok(favourite_music) => {
cx.answer(format!( cx.answer(format!("Fine. {}", dialogue.up(favourite_music)))
"Fine. {}",
cx.dialogue.clone().unwrap().up(favourite_music)
))
.send() .send()
.await?; .await?;
exit() exit()
} }
Err(_) => { Err(_) => {
cx.answer("Oh, please, enter from the keyboard!").send().await?; cx.answer("Oh, please, enter from the keyboard!").send().await?;
next(Dialogue::ReceiveFavouriteMusic(cx.dialogue.unwrap())) next(Dialogue::ReceiveFavouriteMusic(dialogue))
} }
} }
} }
async fn handle_message(cx: Cx<Dialogue>) -> Res { async fn handle_message(cx: Cx<Dialogue>) -> Res {
let DialogueDispatcherHandlerCx { bot, update, dialogue } = cx; let DialogueDispatcherHandlerCx { cx, dialogue } = cx;
// You need handle the error instead of panicking in real-world code, maybe // You need handle the error instead of panicking in real-world code, maybe
// send diagnostics to a development chat. // send diagnostics to a development chat.
match dialogue.expect("Failed to get dialogue info from storage") { match dialogue.expect("Failed to get dialogue info from storage") {
Dialogue::Start => { Dialogue::Start => {
start(DialogueDispatcherHandlerCx::new(bot, update, ())).await start(DialogueDispatcherHandlerCx::new(cx, ())).await
} }
Dialogue::ReceiveFullName => { Dialogue::ReceiveFullName => {
full_name(DialogueDispatcherHandlerCx::new(bot, update, ())).await full_name(DialogueDispatcherHandlerCx::new(cx, ())).await
} }
Dialogue::ReceiveAge(s) => { Dialogue::ReceiveAge(s) => {
age(DialogueDispatcherHandlerCx::new(bot, update, s)).await age(DialogueDispatcherHandlerCx::new(cx, s)).await
} }
Dialogue::ReceiveFavouriteMusic(s) => { Dialogue::ReceiveFavouriteMusic(s) => {
favourite_music(DialogueDispatcherHandlerCx::new(bot, update, s)) favourite_music(DialogueDispatcherHandlerCx::new(cx, s)).await
.await
} }
} }
} }