charcoal-client 0.1.1

Client library for Hearth: A LavaLink alternative with 30X Higher performance
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

//! Provides ClientBuilder extension for super easy use with serenity
use futures::executor;
use std::sync::Arc;

use crate::{init_charcoal, Charcoal, CharcoalConfig};
use serenity::prelude::TypeMapKey;
// pub use serenity::client::ClientBuilder;
pub use serenity::client::ClientBuilder;
use serenity::*;
use tokio::sync::Mutex;

pub struct CharcoalKey;

impl TypeMapKey for CharcoalKey {
    type Value = Arc<Mutex<Charcoal>>;
}

pub trait SerenityInit {
    #[must_use]
    /// Initializes charcoal and registers it in the Serenity type-map
    fn register_charcoal(self, broker: String, config: CharcoalConfig) -> Self;
}

impl SerenityInit for ClientBuilder {
    fn register_charcoal(self, broker: String, config: CharcoalConfig) -> Self {
        let c = init_charcoal(broker, config);
        self.type_map_insert::<CharcoalKey>(executor::block_on(c))
    }
}

#[macro_export]
macro_rules! get_handler_from_serenity_mutable {
    ($ctx: expr,$msg: expr,$reference: ident) => {
        let r = $ctx.data.read().await;
        // Get the GuildID
        let guild = $msg.guild(&$ctx.cache).unwrap();
        let guild_id = guild.id;
        // Get the charcoal manager from the serenity typemap
        let manager = r.get::<CharcoalKey>();
        let mut mx = manager.unwrap().lock().await;
        // Get the PlayerObject
        let mut players = mx.players.write().await;
        $reference = players.get_mut(&guild_id.to_string());
    };
}

#[macro_export]
macro_rules! get_handler_from_serenity {
    ($ctx: expr,$msg: expr,$reference: ident) => {
        let r = $ctx.data.read().await;
        // Get the GuildID
        let guild = $msg.guild(&$ctx.cache).unwrap();
        let guild_id = guild.id;
        // Get the charcoal manager from the serenity typemap
        let manager = r.get::<CharcoalKey>();
        let mut mx = manager.unwrap().lock().await;
        // Get the PlayerObject
        let players = mx.players.read().await;
        $reference = players.get(&guild_id.to_string());
    };
}