rust-tg-bot-ext 1.0.0-rc.1

Application framework for Telegram bots -- handlers, filters, persistence, job queue
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
49

//! [`PollAnswerHandler`] -- handles updates that contain a poll answer.
//!
//! Ported from `python-telegram-bot`'s `PollAnswerHandler`. Matches any
//! update with `Update.poll_answer` set.

use std::future::Future;
use std::pin::Pin;
use std::sync::Arc;

use rust_tg_bot_raw::types::update::Update;

use super::base::{Handler, HandlerCallback, HandlerResult, MatchResult};

/// Handler for `Update.poll_answer`.
///
/// Bots receive poll answer updates only for polls they sent.
pub struct PollAnswerHandler {
    callback: HandlerCallback,
    block: bool,
}

impl PollAnswerHandler {
    /// Create a new `PollAnswerHandler`.
    pub fn new(callback: HandlerCallback, block: bool) -> Self {
        Self { callback, block }
    }
}

impl Handler for PollAnswerHandler {
    fn check_update(&self, update: &Update) -> Option<MatchResult> {
        if update.poll_answer().is_some() {
            Some(MatchResult::Empty)
        } else {
            None
        }
    }

    fn handle_update(
        &self,
        update: Arc<Update>,
        match_result: MatchResult,
    ) -> Pin<Box<dyn Future<Output = HandlerResult> + Send>> {
        (self.callback)(update, match_result)
    }

    fn block(&self) -> bool {
        self.block
    }
}