automate 0.4.0

An asynchronous library to interact with Discord API and make bots
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

use crate::{Error, Snowflake, Identifiable};
use crate::gateway::*;
use percent_encoding::NON_ALPHANUMERIC;
use std::fmt::Write;

/// Any type which is identified by
/// a snowflake.
pub trait ExtractSnowflake {
    fn extract_snowflake(&self) -> Result<Snowflake, Error>;
}

/// Any type that can be put in a URL
/// but requires some kind of encoding.
/// Currently only useful for emojis.
pub trait WriteUrl {
    fn write_url(&self, buf: &mut String) -> Result<(), Error>;
}

macro_rules! automate_types {
    ($($struct:ty),*) => {
        $(
            impl ExtractSnowflake for $struct {
                fn extract_snowflake(&self) -> Result<Snowflake, Error> {
                    Ok(self.id)
                }
            }

            impl ExtractSnowflake for &$struct {
                fn extract_snowflake(&self) -> Result<Snowflake, Error> {
                    Ok(self.id)
                }
            }
        )*
    }
}

macro_rules! automate_enums {
    ($($struct:ty),*) => {
        $(
            impl ExtractSnowflake for $struct {
                fn extract_snowflake(&self) -> Result<Snowflake, Error> {
                    Ok(self.id())
                }
            }

            impl ExtractSnowflake for &$struct {
                fn extract_snowflake(&self) -> Result<Snowflake, Error> {
                    Ok(self.id())
                }
            }
        )*
    }
}

automate_types! {
    AuditLogEntry,
    Category, TextChannel, VoiceChannel, NewsChannel, StoreChannel,
    DirectChannel, GroupChannel,
    ChannelMention, MentionnedUser,
    Overwrite,
    Webhook,
    Guild, PartialGuild, UnavailableGuild,
    Role, PartialRole,
    Message, Attachment, MessageApplication,
    User, PartialUser,
    GuildEmoji
}

automate_enums! {
    Channel, GuildChannel, PrivateChannel
}

impl ExtractSnowflake for Snowflake {
    fn extract_snowflake(&self) -> Result<Snowflake, Error> {
        Ok(*self)
    }
}

impl ExtractSnowflake for Emoji {
    fn extract_snowflake(&self) -> Result<Snowflake, Error> {
        if let Some(id) = self.id {
            Ok(id)
        } else {
            Error::err("Emoji's id field is empty")
        }
    }
}

impl ExtractSnowflake for PartialEmoji {
    fn extract_snowflake(&self) -> Result<Snowflake, Error> {
        if let Some(id) = self.id {
            Ok(id)
        } else {
            Error::err("Emoji's id field is empty")
        }
    }
}

impl WriteUrl for Emoji {
    fn write_url(&self, buf: &mut String) -> Result<(), Error> {
        let penc = percent_encoding::percent_encode(self.name.as_bytes(), NON_ALPHANUMERIC);

        buf.reserve(self.name.len());
        buf.write_fmt(format_args!("{}", penc))?;

        if let Some(id) = self.id {
            buf.push(':');
            buf.push_str(&id.to_string());
        }

        Ok(())
    }
}

impl WriteUrl for PartialEmoji {
    fn write_url(&self, buf: &mut String) -> Result<(), Error> {
        let penc = percent_encoding::percent_encode(self.name.as_bytes(), NON_ALPHANUMERIC);

        buf.reserve(self.name.len());
        buf.write_fmt(format_args!("{}", penc))?;

        if let Some(id) = self.id {
            buf.push(':');
            buf.push_str(&id.to_string());
        }

        Ok(())
    }
}