stoat-rs 0.2.5

Stoat API Wrapper
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use std::time::{Duration, SystemTime};

use stoat_database::events::server::ClientMessage;
use tokio::{select, time::sleep};
use ulid::Ulid;

use crate::context::Events;

pub fn created_at(id: &str) -> SystemTime {
    Ulid::from_string(id).expect("Malformed ID").datetime()
}

/// Automatically sends typing events while the inner future is running, this is useful for displaying that the bot is still processing the users input.
///
/// This will automatically end typing when the inner future is finished.
///
/// Also accessable via [`crate::ChannelExt::with_typing`].
///
/// ## Example:
/// ```rust
/// let output = with_typing(&ctx, channel_id, async move {
///     // Some long calculation
/// }).await;
///
/// ctx.reply()
///     .content(output)
///     .build()
///     .await?
/// ```
pub async fn with_typing<Fut: Future<Output = R>, R>(
    events: impl AsRef<Events>,
    channel_id: String,
    fut: Fut,
) -> R {
    let events = events.as_ref();

    let bg = {
        let events = events.clone();
        let channel_id = channel_id.clone();

        async move {
            loop {
                if let Err(e) = events.send_event(ClientMessage::BeginTyping {
                    channel: channel_id.clone(),
                }) {
                    log::error!("Error occurred in with_typing: {e:?}");
                };

                sleep(Duration::from_secs(10)).await;
            }
        }
    };

    select! {
        _ = bg => {
            unreachable!()
        },
        r = fut => {
            if let Err(e) = events.send_event(ClientMessage::EndTyping { channel: channel_id }) {
                log::error!("Error occurred in with_typing: {e:?}");
            };

            r
        }
    }
}