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
|
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;
#[derive(Clone, Copy, Debug)]
pub enum CommandType {
Mention,
None,
Prefix,
}
#[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) {
let res = command::positions(&message.content, &self.configuration);
let positions = match res {
Some((positions, _kind)) => positions,
None => return,
};
// Ensure that the message length is at least longer than a prefix
// length. There's no point in checking further ahead if there's nothing
// _to_ check.
if positions.iter().all(|p| message.content.len() <= *p) {
return;
}
for position in positions {
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(position)
.1
.trim()
.split_whitespace()
.collect::<Vec<&str>>()
.get(i)
} {
Some(piece) => piece,
None => continue,
});
if let Some(command) = self.commands.get(&built) {
if let Some(check) = self.checks.get(&built) {
if !(check)(&context, &message) {
continue;
}
}
let command = command.clone();
thread::spawn(move || {
let args = message.content[position + built.len()..]
.split_whitespace()
.map(|arg| arg.to_owned())
.collect::<Vec<String>>();
(command)(context, message, args)
});
return;
}
}
}
}
pub fn on<F, S>(mut self, command_name: S, f: F) -> Self
where F: Fn(Context, Message, Vec<String>) + 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
}
}
|