teloxide/examples/simple_dialogue/src/main.rs

#[macro_use]
extern crate strum_macros;
#[macro_use]
extern crate smart_default;

use std::fmt::{self, Display, Formatter};
use teloxide::{
    prelude::*,
    types::{KeyboardButton, ReplyKeyboardMarkup},
};

// ============================================================================
// [Favourite music kinds]
// ============================================================================

#[derive(Copy, Clone, Display, EnumString)]
enum FavouriteMusic {
    Rock,
    Metal,
    Pop,
    Other,
}

impl FavouriteMusic {
    fn markup() -> ReplyKeyboardMarkup {
        ReplyKeyboardMarkup::default().append_row(vec![
            KeyboardButton::new("Rock"),
            KeyboardButton::new("Metal"),
            KeyboardButton::new("Pop"),
            KeyboardButton::new("Other"),
        ])
    }
}

// ============================================================================
// [A UserInfo's data]
// ============================================================================

// TODO: implement a type-safe UserInfo without lots of .unwrap
#[derive(Default)]
struct UserInfo {
    full_name: Option<String>,
    age: Option<u8>,
    favourite_music: Option<FavouriteMusic>,
}

impl Display for UserInfo {
    fn fmt(&self, f: &mut Formatter) -> Result<(), fmt::Error> {
        write!(
            f,
            "Your full name: {}, your age: {}, your favourite music: {}",
            self.full_name.as_ref().unwrap(),
            self.age.unwrap(),
            self.favourite_music.unwrap()
        )
    }
}

// ============================================================================
// [States of a dialogue]
// ============================================================================

#[derive(SmartDefault)]
enum State {
    #[default]
    Start,
    FullName,
    Age,
    FavouriteMusic,
}

// ============================================================================
// [Control a dialogue]
// ============================================================================

type Ctx = DialogueHandlerCtx<Message, State, UserInfo>;
type Res = Result<DialogueStage<State, UserInfo>, RequestError>;

async fn send_favourite_music_types(ctx: &Ctx) -> Result<(), RequestError> {
    ctx.answer("Good. Now choose your favourite music:")
        .reply_markup(FavouriteMusic::markup())
        .send()
        .await?;
    Ok(())
}

async fn start(mut ctx: Ctx) -> Res {
    ctx.answer("Let's start! First, what's your full name?")
        .send()
        .await?;
    state!(ctx, State::FullName);
    next(ctx.dialogue)
}

async fn full_name(mut ctx: Ctx) -> Res {
    ctx.answer("What a wonderful name! Your age?")
        .send()
        .await?;
    ctx.dialogue.data.full_name = Some(ctx.update.text().unwrap().to_owned());
    state!(ctx, State::Age);
    next(ctx.dialogue)
}

async fn age(mut ctx: Ctx) -> Res {
    match ctx.update.text().unwrap().parse() {
        Ok(ok) => {
            send_favourite_music_types(&ctx).await?;
            ctx.dialogue.data.age = Some(ok);
            state!(ctx, State::FavouriteMusic);
        }
        Err(_) => ctx
            .answer("Oh, please, enter a number!")
            .send()
            .await
            .map(|_| ())?,
    }

    next(ctx.dialogue)
}

async fn favourite_music(mut ctx: Ctx) -> Res {
    match ctx.update.text().unwrap().parse() {
        Ok(ok) => {
            ctx.dialogue.data.favourite_music = Some(ok);
            ctx.answer(format!("Fine. {}", ctx.dialogue.data))
                .send()
                .await?;
            exit()
        }
        Err(_) => {
            ctx.answer("Oh, please, enter from the keyboard!")
                .send()
                .await?;
            next(ctx.dialogue)
        }
    }
}

async fn handle_message(ctx: Ctx) -> Res {
    match ctx.dialogue.state {
        State::Start => start(ctx).await,
        State::FullName => full_name(ctx).await,
        State::Age => age(ctx).await,
        State::FavouriteMusic => favourite_music(ctx).await,
    }
}

// ============================================================================
// [Run!]
// ============================================================================

#[tokio::main]
async fn main() {
    std::env::set_var("RUST_LOG", "simple_dialogue=trace");
    std::env::set_var("RUST_LOG", "teloxide=error");
    pretty_env_logger::init();
    log::info!("Starting the simple_dialogue bot!");

    Dispatcher::new(Bot::new("YourAwesomeToken"))
        .message_handler(&DialogueDispatcher::new(|ctx| async move {
            handle_message(ctx)
                .await
                .expect("Something wrong with the bot!")
        }))
        .dispatch()
        .await;
}