🤖 An elegant Telegram bots framework for Rust https://docs.rs/teloxide
Find a file
2020-07-16 19:00:03 +06:00
.github/workflows Update ci.yml 2020-07-04 21:13:09 +06:00
examples Show strongly typed behaviour in examples/simple_commands_bot 2020-07-07 01:20:09 +06:00
media Create new screenshots from MacOS 2020-02-19 04:27:24 +03:00
src Add BotBuilder 2020-07-16 18:52:39 +06:00
tests Return Arc<RedisServer> 2020-07-04 21:02:07 +06:00
.gitignore Simplify .gitignore 2020-03-05 11:23:20 +06:00
Cargo.toml Fix build 2020-07-16 18:59:33 +06:00
CHANGELOG.md changelog 2020-07-06 16:05:53 +03:00
CODE_STYLE.md Update CODE_STYLE.md 2020-02-19 21:50:49 +06:00
CONTRIBUTING.md Simpler form of using nightly rustfmt 2020-04-19 18:43:12 +03:00
ICON.png Improve the ICON.png quality 2019-10-15 16:16:56 +06:00
LICENSE Update LICENSE 2020-02-22 02:56:02 +06:00
logo.svg Make logo svg 2019-12-30 12:14:05 +03:00
README.md Rework the 'Features' section (README.md) 2020-07-11 20:52:29 +06:00
rustfmt.toml Fixes 2020-02-19 04:54:41 +06:00

teloxide

A full-featured framework that empowers you to easily build Telegram bots using the async/.await syntax in Rust. It handles all the difficult stuff so you can focus only on your business logic.

Table of contents

Features

  • Functioal reactive design. teloxide has functional reactive design, allowing you to declaratively manipulate streams of updates from Telegram using filters, maps, folds, zips, and a lot of other adaptors.
  • Persistence. Dialogues management is independent of how/where dialogues are stored: you can just replace one line and make them persistent. Out-of-the-box storages include Redis.
  • Strongly typed bot commands. You can describe bot commands as enumerations, and then they'll be automatically constructed from strings. Just like you describe JSON structures in serde-json and command-line arguments in structopt.

Setting up your environment

  1. Download Rust.
  2. Create a new bot using @Botfather to get a token in the format 123456789:blablabla.
  3. Initialise the TELOXIDE_TOKEN environmental variable to your token:
# Unix-like
$ export TELOXIDE_TOKEN=<Your token here>

# Windows
$ set TELOXIDE_TOKEN=<Your token here>
  1. Be sure that you are up to date:
# If you're using stable
$ rustup update stable
$ rustup override set stable

# If you're using nightly
$ rustup update nightly
$ rustup override set nightly
  1. Execute cargo new my_bot, enter the directory and put these lines into your Cargo.toml:
[dependencies]
teloxide = "0.2.0"
log = "0.4.8"
tokio = "0.2.11"
pretty_env_logger = "0.4.0"

API overview

The ping-pong bot

This bot has a single message handler, which answers "pong" to each incoming message:

(Full)

use teloxide::prelude::*;

#[tokio::main]
async fn main() {
    teloxide::enable_logging!();
    log::info!("Starting ping_pong_bot!");

    let bot = Bot::from_env();

    Dispatcher::new(bot)
        .messages_handler(|rx: DispatcherHandlerRx<Message>| {
            rx.for_each(|message| async move {
                message.answer("pong").send().await.log_on_error().await;
            })
        })
        .dispatch()
        .await;
}

Commands

Commands are defined similar to how we define CLI using structopt and JSON structures in serde-json. The following bot accepts either /username YourUsername, /usernameandage YourUsername YourAge and shows the usage guide on /help:

(Full)

// Imports are omitted...

#[derive(BotCommand)]
#[command(rename = "lowercase", description = "These commands are supported:")]
enum Command {
    #[command(description = "display this text.")]
    Help,
    #[command(description = "handle a username.")]
    Username(String),
    #[command(
        description = "handle a username and an age.",
        parse_with = "split"
    )]
    UsernameAndAge { username: String, age: u8 },
}

async fn answer(
    cx: UpdateWithCx<Message>,
    command: Command,
) -> ResponseResult<()> {
    match command {
        Command::Help => cx.answer(Command::descriptions()).send().await?,
        Command::Username(username) => {
            cx.answer_str(format!("Your username is @{}.", username)).await?
        }
        Command::UsernameAndAge { username, age } => {
            cx.answer_str(format!(
                "Your username is @{} and age is {}.",
                username, age
            ))
            .await?
        }
    };

    Ok(())
}

async fn handle_commands(rx: DispatcherHandlerRx<Message>) {
    rx.commands::<Command, &str>(panic!("Insert here your bot's name"))
        .for_each_concurrent(None, |(cx, command)| async move {
            answer(cx, command).await.log_on_error().await;
        })
        .await;
}

#[tokio::main]
async fn main() {
    // Setup is omitted...
}

Dialogues

A dialogue is described by an enumeration, where each variant is one of possible dialogue's states. There are also transition functions, which turn a dialogue from one state to another, thereby forming an FSM.

States and transition functions are placed into separated modules. For example:

(dialogue_bot/src/states.rs)

// Imports are omitted...

#[derive(Default)]
pub struct StartState;

pub struct ReceiveFullNameState {
    rest: StartState,
}

pub struct ReceiveAgeState {
    rest: ReceiveFullNameState,
    full_name: String,
}

pub struct ReceiveFavouriteMusicState {
    rest: ReceiveAgeState,
    age: u8,
}

#[derive(Display)]
#[display(
    "Your full name: {rest.rest.full_name}, your age: {rest.age}, your \
     favourite music: {favourite_music}"
)]
pub struct ExitState {
    rest: ReceiveFavouriteMusicState,
    favourite_music: FavouriteMusic,
}

up!(
    StartState -> ReceiveFullNameState,
    ReceiveFullNameState + [full_name: String] -> ReceiveAgeState,
    ReceiveAgeState + [age: u8] -> ReceiveFavouriteMusicState,
    ReceiveFavouriteMusicState + [favourite_music: FavouriteMusic] -> ExitState,
);

#[derive(SmartDefault, From)]
pub enum Dialogue {
    #[default]
    Start(StartState),
    ReceiveFullName(ReceiveFullNameState),
    ReceiveAge(ReceiveAgeState),
    ReceiveFavouriteMusic(ReceiveFavouriteMusicState),
}

The handy up! macro automatically generates functions that complete one state to another by appending a field.

(dialogue_bot/src/transitions.rs)

// Imports are omitted...

pub type Cx = UpdateWithCx<Message>;
pub type Out = TransitionOut<Dialogue>;

async fn start(cx: Cx, state: StartState) -> Out {
    cx.answer_str("Let's start! First, what's your full name?").await?;
    next(state.up())
}

async fn receive_full_name(cx: Cx, state: ReceiveFullNameState) -> Out {
    match cx.update.text_owned() {
        Some(full_name) => {
            cx.answer_str("What a wonderful name! Your age?").await?;
            next(state.up(full_name))
        }
        _ => {
            cx.answer_str("Please, enter a text message!").await?;
            next(state)
        }
    }
}

async fn receive_age(cx: Cx, state: ReceiveAgeState) -> Out {
    match cx.update.text().map(str::parse) {
        Some(Ok(age)) => {
            cx.answer("Good. Now choose your favourite music:")
                .reply_markup(FavouriteMusic::markup())
                .send()
                .await?;
            next(state.up(age))
        }
        _ => {
            cx.answer_str("Please, enter a number!").await?;
            next(state)
        }
    }
}

async fn receive_favourite_music(
    cx: Cx,
    state: ReceiveFavouriteMusicState,
) -> Out {
    match cx.update.text().map(str::parse) {
        Some(Ok(favourite_music)) => {
            cx.answer_str(format!("Fine. {}", state.up(favourite_music)))
                .await?;
            exit()
        }
        _ => {
            cx.answer_str("Please, enter from the keyboard!").await?;
            next(state)
        }
    }
}

pub async fn dispatch(cx: Cx, dialogue: Dialogue) -> Out {
    match dialogue {
        Dialogue::Start(state) => start(cx, state).await,
        Dialogue::ReceiveFullName(state) => receive_full_name(cx, state).await,
        Dialogue::ReceiveAge(state) => receive_age(cx, state).await,
        Dialogue::ReceiveFavouriteMusic(state) => {
            receive_favourite_music(cx, state).await
        }
    }
}

(dialogue_bot/src/favourite_music.rs)

// Imports are omitted...

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

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

(dialogue_bot/src/main.rs)

// Imports are omitted...

#[tokio::main]
async fn main() {
    teloxide::enable_logging!();
    log::info!("Starting dialogue_bot!");

    let bot = Bot::from_env();

    Dispatcher::new(bot)
        .messages_handler(DialogueDispatcher::new(
            |input: TransitionIn<Dialogue, Infallible>| async move {
                // Unwrap without panic because of std::convert::Infallible.
                dispatch(input.cx, input.dialogue.unwrap())
                    .await
                    .expect("Something wrong with the bot!")
            },
        ))
        .dispatch()
        .await;
}

More examples!

Recommendations

  • Use this pattern:
#[tokio::main]
async fn main() {
    run().await;
}

async fn run() {
    // Your logic here...
}

Instead of this:

#[tokio::main]
async fn main() {
    // Your logic here...
}

The second one produces very strange compiler messages because of the #[tokio::main] macro. However, the examples in this README use the second variant for brevity.

FAQ

Q: Where I can ask questions?

A: Issues is a good place for well-formed questions, for example, about the library design, enhancements, bug reports. But if you can't compile your bot due to compilation errors and need quick help, feel free to ask in our official group.

Q: Why Rust?

A: Most programming languages have their own implementations of Telegram bots frameworks, so why not Rust? We think Rust provides enough good ecosystem and the language itself to be suitable for writing bots.

Q: Can I use webhooks?

A: teloxide doesn't provide special API for working with webhooks due to their nature with lots of subtle settings. Instead, you setup your webhook by yourself, as shown in webhook_ping_pong_bot.

Associated links:

Q: Can I use different loggers?

A: Of course, you can. The enable_logging! and enable_logging_with_filter! macros are just convenient utilities, not necessary to use them. You can setup a different logger, for example, fern, as usual, e.g. teloxide has no specific requirements as it depends only on log.

Community bots

Feel free to push your own bot into our collection!

Contributing

See CONRIBUTING.md.