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
|
mod command;
mod configuration;
pub use self::command::Command;
pub use self::configuration::Configuration;
use self::command::InternalCommand;
use std::collections::HashMap;
use std::sync::Arc;
use std::thread;
use ::client::Context;
use ::model::Message;
#[allow(type_complexity)]
#[derive(Default)]
pub struct Framework {
configuration: Configuration,
commands: HashMap<String, InternalCommand>,
checks: HashMap<String, Arc<Fn(&Context, &Message) -> bool + Send + Sync + 'static>>,
pub initialized: bool,
}
impl Framework {
pub fn configure<F>(mut self, f: F) -> Self
where F: FnOnce(Configuration) -> Configuration {
self.configuration = f(self.configuration);
self.initialized = true;
self
}
#[doc(hidden)]
pub fn dispatch(&mut self, context: Context, message: Message) {
// Determine the point at which the prefix ends, and the command starts.
let pos = if let Some(ref prefix) = self.configuration.prefix {
let mut mention_ends = None;
if let Some(ref mentions) = self.configuration.on_mention {
for mention in mentions {
if !message.content.starts_with(&mention[..]) {
continue;
}
mention_ends = Some(mention.len() + 1);
break;
}
}
if let Some(mention_ends) = mention_ends {
mention_ends
} else if !message.content.starts_with(prefix) {
return;
} else {
prefix.len()
}
} else {
0
};
// Ensure that the message length is at least longer than the prefix
// length. There's no point in checking further ahead if there's nothing
// to check.
if message.content.len() <= pos {
return;
}
let mut built = String::new();
for i in 0..self.configuration.depth {
if i > 0 {
built.push(' ');
}
built.push_str(match {
message.content
.split_at(pos)
.1
.split_whitespace()
.collect::<Vec<&str>>()
.get(i)
} {
Some(piece) => piece,
None => return,
});
if let Some(command) = self.commands.get(&built) {
if let Some(check) = self.checks.get(&built) {
if !(check)(&context, &message) {
return;
}
}
let command = command.clone();
thread::spawn(move || {
(command)(context, message)
});
return;
}
}
}
pub fn on<F, S>(mut self, command_name: S, f: F) -> Self
where F: Fn(Context, Message) + Send + Sync + 'static,
S: Into<String> {
self.commands.insert(command_name.into(), Arc::new(f));
self.initialized = true;
self
}
pub fn set_check<F, S>(mut self, command: S, check: F) -> Self
where F: Fn(&Context, &Message) -> bool + Send + Sync + 'static,
S: Into<String> {
self.checks.insert(command.into(), Arc::new(check));
self.initialized = true;
self
}
}
|