Skip to content

0xfe/mobot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MOBOT

Crates.io Documentation License: MIT

MOBOT is a Telegram chat framework written in Rust. It comes with a fully native implementation of the Telegram Bot API.

Features

  • Raw Telegram API with support for Messages, Channels, Stickers, Callbacks, and more.
  • Web framework style Routing API with support for message-based routing and nested handler stacks.
  • Easy application state management. MOBOT makes sure your handler gets the right state for each chat.
  • Integrated test infrastructure (FakeBot), to simplify writing unit tests for your bots.
  • Support for progress bars, inline keyboards, "Typing..." indicators, etc. See demo video below.

Demo and Examples

This is a demo of a devops bot built with the MOBOT framework.

MOBOT.demo.mp4

Other bots built with MOBOT

  • WordleBot - A Wordle Telegram Bot
  • melatonin-bot - Telegram bot that reminds users about streams of selected NijiEN livers in ~20 minutes
  • <Add Your's Here>

Hello World!

Add mobot to your Cargo.toml.

cargo add mobot

Example Bot that replies with "Hello world!" to every message. Working example in src/bin/hello.rs.

use mobot::*;

#[tokio::main]
async fn main() {
    let client = Client::new(std::env::var("TELEGRAM_TOKEN").unwrap().into());
    let mut router = Router::<()>::new(client);

    router.add_route(Route::Default, |_, _| async move {
        Ok(Action::ReplyText("Hello world!".into()))
    });
    router.start().await;
}

Documentation

See full API documentation at https://docs.rs/mobot.

See examples in src/bin.

Testing

MOBOT is packaged with fake::FakeAPI, a library to simplify unit testing your bots. FakeAPI can be plugged into mobot::Client using the with_post_handler hook. See example below from router_test.rs.

async fn it_works() {
    mobot::init_logger();

    // Create a FakeAPI and attach it to the client. Any Telegram requests are now forwarded
    // to `fakeserver` instead.
    let fakeserver = fake::FakeAPI::new();
    let client = Client::new("token".to_string().into()).with_post_handler(fakeserver.clone());

    // Keep the Telegram poll timeout short for testing. The default Telegram poll timeout is 60s.
    let mut router = Router::new(client).with_poll_timeout_s(1);

    // Since we're passing ownership of the Router to a background task, grab the
    // shutdown channels so we can shut it down from this task.
    let (shutdown_notifier, shutdown_tx) = router.shutdown();

    // Our bot is a ping bot. Add the handler to the router (see bin/ping.rs).
    router.add_route(Route::Default, handle_chat_event);

    // Start the router in a background task.
    tokio::spawn(async move {
        info!("Starting router...");
        router.start().await;
    });

    // We're in the foreground. Create a new chat session with the bot, providing your
    // username. This shows up in the `from` field of messages.
    let chat = fakeserver.create_chat("qubyte").await;

    // Send the message "ping1", expect the response "pong(1): ping1"
    chat.send_text("ping1").await.unwrap();
    assert_eq!(
        chat.recv_update().await.unwrap().to_string(),
        "pong(1): ping1"
    );

    // Send the message "ping2", expect the response "pong(2): ping2"
    chat.send_text("ping2").await.unwrap();
    assert_eq!(
        chat.recv_update().await.unwrap().to_string(),
        "pong(2): ping2"
    );

    // Optional: validate there's no more messages from the bot, by waiting two seconds
    // for more messages.
    assert!(
        tokio::time::timeout(Duration::from_millis(2000), chat.recv_update())
            .await
            .is_err()
    );

    // All done shutdown the router, and wait for it to complete.
    info!("Shutting down...");
    shutdown_tx.send(()).await.unwrap();
    shutdown_notifier.notified().await;
}

Extending MOBOT

Adding new Telegram API calls

Adding support for additional APIs is straightforward. It involves creating structs for the request and response in lib/api, and adding a method to API.

For example, to add support for the sendSticker Telegram API:

Create the SendStickerRequest struct

Make sure to derive the BotRequest trait to the request.

#[derive(Debug, Serialize, Clone, BotRequest)]
pub struct SendStickerRequest {
    /// Unique identifier for the target chat or username of the target
    pub chat_id: i64,

    /// Sticker to send. Pass a file_id as String to send a file that
    pub sticker: String,

    /// Sends the message silently. Users will receive a notification with
    #[serde(skip_serializing_if = "Option::is_none")]
    pub disable_notification: Option<bool>,

    /// If the message is a reply, ID of the original message
    #[serde(skip_serializing_if = "Option::is_none")]
    pub reply_to_message_id: Option<i64>,
}

Add the send_sticker method call to API

impl API {
    pub async fn send_sticker(&self, req: &SendStickerRequest) -> anyhow::Result<Message> {
        self.client.post("sendSticker", req).await
    }
}

Test and send me a Pull Request

  • Add a test in tests/. If necessary update lib/fake.rs for client testing.
  • Add example code to src/bin/.
  • Commit and send me a PR!

Cutting releases

Install Cargo Release - crate install cargo-release.

Run:

cargo update
cargo test
cargo doc
git commit -a -m '...'

# Cut release
cargo release --execute patch

External Dependencies

This crate requires OpenSSL and pkg-config:

  • On Linux: sudo apt-get install pkg-config libssl-dev
  • On Mac: nothing to do!

TODO

  • Default logging handler
  • Default auth handler
  • Dialoge/script framework
  • Garbage Collect old sessions

License

MIT License Copyright 2023 Mohit Muthanna Cheppudira

Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

About

mobot: A Telegram Bot Library in Rust

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages