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
|
pub use super::{Args, Command, CommandGroup, CommandOptions, CommandError};
use client::Context;
use model::{Message, Permissions};
use std::sync::Arc;
pub enum FnOrCommand {
Fn(fn(&mut Context, &Message, Args) -> Result<(), CommandError>),
Command(Arc<Command>),
}
pub struct CreateCommand(pub CommandOptions, pub FnOrCommand);
impl CreateCommand {
/// Adds multiple aliases.
pub fn batch_known_as<T: ToString, It: IntoIterator<Item=T>>(mut self, names: It) -> Self {
self.0
.aliases
.extend(names.into_iter().map(|n| n.to_string()));
self
}
/// Adds a ratelimit bucket.
pub fn bucket(mut self, bucket: &str) -> Self {
self.0.bucket = Some(bucket.to_string());
self
}
/// Adds a "check" to a command, which checks whether or not the command's
/// function should be called.
///
/// These checks are bypassed for commands sent by the application owner.
///
/// # Examples
///
/// Ensure that the user who created a message, calling a "ping" command,
/// is the owner.
///
/// ```rust,no_run
/// # use serenity::prelude::EventHandler;
/// # struct Handler;
/// # impl EventHandler for Handler {}
/// use serenity::client::{Client, Context};
/// use serenity::framework::standard::{
/// Args,
/// CommandOptions,
/// CommandError,
/// StandardFramework,
/// };
/// use serenity::model::Message;
/// use std::env;
///
/// let token = env::var("DISCORD_TOKEN").unwrap();
/// let mut client = Client::new(&token, Handler).unwrap();
///
/// client.with_framework(StandardFramework::new()
/// .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: Args) -> Result<(),
/// CommandError> {
/// message.channel_id.say("Pong!")?;
///
/// Ok(())
/// }
///
/// fn owner_check(_context: &mut Context, message: &Message, _: &mut Args, _:
/// &CommandOptions) -> 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, &mut Args, &CommandOptions) -> 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_string());
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_string());
self
}
/// A function that can be called when a command is received.
/// You can return `Err(string)` if there's an error.
pub fn exec(mut self, func: fn(&mut Context, &Message, Args) -> Result<(), CommandError>) -> Self {
self.1 = FnOrCommand::Fn(func);
self
}
/// Like [`exec`] but accepts a `Command` directly.
///
/// [`exec`]: #method.exec
pub fn cmd<C: Command + 'static>(mut self, c: C) -> Self {
self.1 = FnOrCommand::Command(Arc::new(c));
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
}
/// 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_string());
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
}
/// Exact number of arguments that should be passed.
pub fn num_args(mut self, num_args: i32) -> Self {
self.0.min_args = Some(num_args);
self.0.max_args = Some(num_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_string());
self
}
/// Sets roles that are allowed to use the command.
pub fn allowed_roles<T: ToString, It: IntoIterator<Item=T>>(mut self, allowed_roles: It) -> Self {
self.0.allowed_roles = allowed_roles.into_iter().map(|x| x.to_string()).collect();
self
}
pub(crate) fn finish(self) -> Arc<Command> {
let CreateCommand(options, fc) = self;
match fc {
FnOrCommand::Fn(func) => {
struct A(Arc<CommandOptions>, fn(&mut Context, &Message, Args) -> Result<(), CommandError>);
impl Command for A {
fn execute(&self, c: &mut Context, m: &Message, a: Args) -> Result<(), CommandError> {
(self.1)(c, m, a)
}
fn options(&self) -> Arc<CommandOptions> { Arc::clone(&self.0) }
}
Arc::new(A(Arc::new(options), func))
},
FnOrCommand::Command(cmd) => cmd,
}
}
}
|