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
|
#[macro_use]
extern crate serenity;
use serenity::client::Context;
use serenity::Client;
use serenity::model::Message;
use std::env;
fn main() {
// Configure the client with your Discord bot token in the environment.
let token = env::var("DISCORD_TOKEN")
.expect("Expected a token in the environment");
let mut client = Client::login_bot(&token);
client.on_ready(|_context, ready| {
println!("{} is connected!", ready.user.name);
});
// Commands are equivilant to:
// "~about"
// "~emoji cat"
// "~emoji dog"
// "~ping"
// "~some complex command"
client.with_framework(|f| f
.configure(|c| c
.allow_whitespace(true)
.on_mention(true)
.prefix("~"))
.before(|_context, message, command_name| {
println!("Got command '{}' by user '{}'",
command_name,
message.author.name);
})
.after(|_context, _message, command_name| {
println!("Processed command '{}'", command_name)
})
.on("ping", ping_command)
.set_check("ping", owner_check) // Ensure only the owner can run this
.on("emoji cat", cat_command)
.on("emoji dog", dog_command)
.on("multiply", multiply)
.on("some complex command", some_complex_command)
// Commands can be in closure-form as well
.on("about", |context, _message, _args| drop(context.say("A test bot"))));
let _ = client.start();
}
command!(cat_command(context, _msg, _arg) {
let _ = context.say(":cat:");
});
fn dog_command(context: &Context, _msg: &Message, _args: Vec<String>) {
let _ = context.say(":dog:");
}
// `Message::reply` is only compiled if the `methods` feature flag is enabled.
#[cfg(feature = "methods")]
fn ping_command(_context: &Context, message: &Message, _args: Vec<String>) {
let _ = message.reply("Pong!");
}
#[cfg(not(feature = "methods"))]
fn ping_command(context: &Context, message: &Message, _args: Vec<String>) {
if let Err(why) = context.say(&format!("{}: Pong!", message.author)) {
println!("Error sending message: {:?}", why);
}
}
fn owner_check(_context: &Context, message: &Message) -> bool {
// Replace 7 with your ID
message.author.id == 7
}
fn some_complex_command(context: &Context, _msg: &Message, args: Vec<String>) {
let _ = context.say(&format!("Arguments: {:?}", args));
}
command!(multiply(context, _message, args, first: f64, second: f64) {
let res = first * second;
let _ = context.say(&res.to_string());
println!("{:?}", args);
});
|