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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
|
pub use super::{Command, CommandType, CommandGroup};
use std::collections::HashMap;
use std::default::Default;
use std::sync::Arc;
use ::client::Context;
use ::model::{Message, Permissions};
pub struct CreateCommand(pub Command);
impl CreateCommand {
/// Adds a ratelimit bucket.
pub fn bucket(mut self, bucket: &str) -> Self {
self.0.bucket = Some(bucket.to_owned());
self
}
/// Adds an alias, allowing users to use the command under a different name.
pub fn known_as(mut self, name: &str) -> Self {
self.0.aliases.push(name.to_owned());
self
}
/// Adds multiple aliases.
pub fn batch_known_as(mut self, names: Vec<&str>) -> Self {
for n in names {
self.0.aliases.push(n.to_owned());
}
self
}
/// Adds a "check" to a command, which checks whether or not the command's
/// function should be called.
///
/// # Examples
///
/// Ensure that the user who created a message, calling a "ping" command,
/// is the owner.
///
/// ```rust,no_run
/// use serenity::client::{Client, Context};
/// use serenity::model::Message;
/// use std::env;
///
/// let mut client = Client::login(&env::var("DISCORD_TOKEN").unwrap());
///
/// client.with_framework(|f| f
/// .configure(|c| c.prefix("~"))
/// .command("ping", |c| c
/// .check(owner_check)
/// .desc("Replies to a ping with a pong")
/// .exec(ping)));
///
/// fn ping(_context: &mut Context, message: &Message, _args: Vec<String>) -> Result<(), String> {
/// let _ = message.channel_id.say("Pong!");
///
/// Ok(())
/// }
///
/// fn owner_check(_context: &mut Context, message: &Message) -> bool {
/// // replace with your user ID
/// message.author.id == 7
/// }
/// ```
pub fn check<F>(mut self, check: F) -> Self
where F: Fn(&mut Context, &Message) -> bool + Send + Sync + 'static {
self.0.checks.push(Box::new(check));
self
}
/// Description, used by other commands.
pub fn desc(mut self, desc: &str) -> Self {
self.0.desc = Some(desc.to_owned());
self
}
/// Whether command can be used only privately or not.
pub fn dm_only(mut self, dm_only: bool) -> Self {
self.0.dm_only = dm_only;
self
}
/// Example arguments, used by other commands.
pub fn example(mut self, example: &str) -> Self {
self.0.example = Some(example.to_owned());
self
}
/// A function that can be called when a command is received.
/// You can return `Err(string)` if there's an error.
///
/// See [`exec_str`] if you _only_ need to return a string on command use.
///
/// [`exec_str`]: #method.exec_str
pub fn exec<F>(mut self, func: F) -> Self
where F: Fn(&mut Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
self.0.exec = CommandType::Basic(Box::new(func));
self
}
/// Sets a function that's called when a command is called that can access
/// the internal HashMap of commands, used specifically for creating a help
/// command.
///
/// You can return `Err(string)` if there's an error.
pub fn exec_help<F>(mut self, f: F) -> Self
where F: Fn(&mut Context, &Message, HashMap<String, Arc<CommandGroup>>, &[String]) -> Result<(), String> + Send + Sync + 'static {
self.0.exec = CommandType::WithCommands(Box::new(f));
self
}
/// Sets a string to be sent in the channel of context on command. This can
/// be useful for an `about`, `invite`, `ping`, etc. command.
///
/// # Examples
///
/// Create a command named "ping" that returns "Pong!":
///
/// ```rust,ignore
/// client.with_framework(|f| f
/// .command("ping", |c| c.exec_str("Pong!")));
/// ```
pub fn exec_str(mut self, content: &str) -> Self {
self.0.exec = CommandType::StringResponse(content.to_owned());
self
}
/// Whether command can be used only in guilds or not.
pub fn guild_only(mut self, guild_only: bool) -> Self {
self.0.guild_only = guild_only;
self
}
/// Whether command should be displayed in help list or not, used by other commands.
pub fn help_available(mut self, help_available: bool) -> Self {
self.0.help_available = help_available;
self
}
/// Maximum amount of arguments that can be passed.
pub fn max_args(mut self, max_args: i32) -> Self {
self.0.max_args = Some(max_args);
self
}
/// Minumum amount of arguments that should be passed.
pub fn min_args(mut self, min_args: i32) -> Self {
self.0.min_args = Some(min_args);
self
}
/// Whether command can be used only privately or not.
pub fn owners_only(mut self, owners_only: bool) -> Self {
self.0.owners_only = owners_only;
self
}
/// The permissions that a user must have in the contextual channel in order
/// for the command to be processed.
pub fn required_permissions(mut self, permissions: Permissions) -> Self {
self.0.required_permissions = permissions;
self
}
/// Command usage schema, used by other commands.
pub fn usage(mut self, usage: &str) -> Self {
self.0.usage = Some(usage.to_owned());
self
}
/// Whether or not arguments should be parsed using the quotation parser.
///
/// Enabling this will parse `~command "this is arg 1" "this is arg 2"` as
/// two arguments: `this is arg 1` and `this is arg 2`.
///
/// Disabling this will parse `~command "this is arg 1" "this is arg 2"` as
/// eight arguments: `"this`, `is`, `arg`, `1"`, `"this`, `is`, `arg`, `2"`.
///
/// Refer to [`utils::parse_quotes`] for information on the parser.
///
/// [`utils::parse_quotes`]: ../../utils/fn.parse_quotes.html
pub fn use_quotes(mut self, use_quotes: bool) -> Self {
self.0.use_quotes = use_quotes;
self
}
}
impl Default for Command {
fn default() -> Command {
Command {
aliases: Vec::new(),
checks: Vec::default(),
exec: CommandType::Basic(Box::new(|_, _, _| Ok(()))),
desc: None,
usage: None,
example: None,
use_quotes: false,
min_args: None,
bucket: None,
max_args: None,
required_permissions: Permissions::empty(),
dm_only: false,
guild_only: false,
help_available: true,
owners_only: false,
}
}
}
|