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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

use super::handle;
use crate::{connectors::Connector, errors, event_loop::Webhook};
use hyper::{
    service::{make_service_fn, service_fn},
    Server,
};
use std::{convert::Infallible, net::SocketAddr, sync::Arc};
use tokio::time::timeout;
use tracing::instrument;

/// Configures the HTTP webhook server.
#[must_use = "webhook server needs to be `start`ed to run the event loop"]
pub struct Http<'a, C> {
    webhook: Webhook<'a, C>,
}

impl<'a, C> Http<'a, C> {
    pub(crate) const fn new(webhook: Webhook<'a, C>) -> Self {
        Self { webhook }
    }
}

impl<'a, C: Connector + Clone> Http<'a, C> {
    /// Starts the server.
    #[instrument(name = "http_webhook", skip(self))]
    pub async fn start(self) -> Result<Infallible, errors::HttpWebhook> {
        let Webhook {
            event_loop,
            ip,
            port,
            updates_url,
            url,
            certificate,
            max_connections,
            allowed_updates,
            request_timeout,
        } = self.webhook;

        let set_webhook = event_loop
            .bot
            .set_webhook(url, certificate, max_connections, allowed_updates)
            .call();

        timeout(request_timeout, set_webhook).await??;

        let bot = Arc::new(event_loop.bot.clone());
        let event_loop = Arc::new(event_loop);
        let addr = SocketAddr::new(ip, port);
        let updates_url = Arc::new(updates_url);

        Server::bind(&addr)
            .serve(make_service_fn(move |_| {
                let bot = Arc::clone(&bot);
                let event_loop = Arc::clone(&event_loop);
                let updates_url = Arc::clone(&updates_url);

                async move {
                    let service = service_fn(move |request| {
                        handle(
                            Arc::clone(&bot),
                            Arc::clone(&event_loop),
                            request,
                            Arc::clone(&updates_url),
                        )
                    });

                    Ok::<_, hyper::Error>(service)
                }
            }))
            .await?;

        unreachable!("[tbot] The webhook server unexpectedly returned.");
    }
}