mobot 0.3.10

A Telegram framework for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use anyhow::bail;
use async_trait::async_trait;

use crate::{
    handler::{BotHandlerFn, BotState},
    Action, Event, State,
};

/// This is a basic implementation of a handler that checks if the user is authorized.
pub struct AuthHandler {
    pub authorized_users: Vec<String>,
}

impl AuthHandler {
    pub fn new(authorized_users: Vec<String>) -> Self {
        Self { authorized_users }
    }
}

#[async_trait]
impl<S: BotState> BotHandlerFn<S> for AuthHandler {
    async fn run(&self, event: Event, _: State<S>) -> Result<Action, anyhow::Error> {
        if !self.authorized_users.contains(
            event
                .update
                .from_user()?
                .username
                .as_ref()
                .ok_or(anyhow::anyhow!("No username"))?,
        ) {
            bail!(
                "Unauthorized user: {}",
                event
                    .update
                    .from_user()?
                    .username
                    .as_ref()
                    .unwrap_or(&"__unknown__".to_string())
            );
        }

        Ok(Action::Next)
    }
}

pub fn auth_handler<S: BotState>(authorized_users: Vec<String>) -> Box<dyn BotHandlerFn<S>> {
    Box::new(AuthHandler::new(authorized_users))
}