aboutsummaryrefslogtreecommitdiff
path: root/src/ext/framework
diff options
context:
space:
mode:
authorTaavi <[email protected]>2017-01-02 03:38:36 +0300
committerzeyla <[email protected]>2017-01-01 16:38:36 -0800
commit18cc152769a35cd1705f045b9815cc90446034ef (patch)
treeb33d18158b62ad5d89cd35f4f547aadb509369e2 /src/ext/framework
parentFix command macro signatures (diff)
downloadserenity-18cc152769a35cd1705f045b9815cc90446034ef.tar.xz
serenity-18cc152769a35cd1705f045b9815cc90446034ef.zip
Implement context message queueing
Also the dreaded `ctx <<= "something"` which is actually a mistake.
Diffstat (limited to 'src/ext/framework')
-rw-r--r--src/ext/framework/command.rs16
-rw-r--r--src/ext/framework/configuration.rs2
-rw-r--r--src/ext/framework/create_command.rs10
-rw-r--r--src/ext/framework/create_group.rs4
-rw-r--r--src/ext/framework/help_commands.rs6
-rw-r--r--src/ext/framework/mod.rs38
6 files changed, 40 insertions, 36 deletions
diff --git a/src/ext/framework/command.rs b/src/ext/framework/command.rs
index 92e6e93..70ee4ff 100644
--- a/src/ext/framework/command.rs
+++ b/src/ext/framework/command.rs
@@ -5,14 +5,14 @@ use ::model::Message;
use ::model::Permissions;
use std::collections::HashMap;
-pub type Check = Fn(&Context, &Message) -> bool + Send + Sync + 'static;
-pub type Exec = Fn(&Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static;
-pub type Help = Fn(&Context, &Message, HashMap<String, Arc<CommandGroup>>, Vec<String>) -> Result<(), String> + Send + Sync + 'static;
-pub type BeforeHook = Fn(&Context, &Message, &String) -> bool + Send + Sync + 'static;
-pub type AfterHook = Fn(&Context, &Message, &String, Result<(), String>) + Send + Sync + 'static;
+pub type Check = Fn(&mut Context, &Message) -> bool + Send + Sync + 'static;
+pub type Exec = Fn(&mut Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static;
+pub type Help = Fn(&mut Context, &Message, HashMap<String, Arc<CommandGroup>>, Vec<String>) -> Result<(), String> + Send + Sync + 'static;
+pub type BeforeHook = Fn(&mut Context, &Message, &String) -> bool + Send + Sync + 'static;
+pub type AfterHook = Fn(&mut Context, &Message, &String, Result<(), String>) + Send + Sync + 'static;
#[doc(hidden)]
pub type InternalCommand = Arc<Command>;
-pub type PrefixCheck = Fn(&Context) -> Option<String> + Send + Sync + 'static;
+pub type PrefixCheck = Fn(&mut Context) -> Option<String> + Send + Sync + 'static;
#[doc(hidden)]
pub enum CommandOrAlias {
@@ -71,7 +71,7 @@ pub struct Command {
impl Command {
pub fn new<F>(f: F) -> Self
- where F: Fn(&Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
+ where F: Fn(&mut Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
Command {
aliases: Vec::new(),
checks: Vec::default(),
@@ -92,7 +92,7 @@ impl Command {
}
}
-pub fn positions(ctx: &Context, content: &str, conf: &Configuration) -> Option<Vec<usize>> {
+pub fn positions(ctx: &mut Context, content: &str, conf: &Configuration) -> Option<Vec<usize>> {
if !conf.prefixes.is_empty() || conf.dynamic_prefix.is_some() {
// Find out if they were mentioned. If not, determine if the prefix
// was used. If not, return None.
diff --git a/src/ext/framework/configuration.rs b/src/ext/framework/configuration.rs
index d4ebb3e..611775a 100644
--- a/src/ext/framework/configuration.rs
+++ b/src/ext/framework/configuration.rs
@@ -177,7 +177,7 @@ impl Configuration {
/// Sets the prefix to respond to. This can either be a single- or
/// multi-char string.
pub fn dynamic_prefix<F>(mut self, dynamic_prefix: F) -> Self
- where F: Fn(&Context) -> Option<String> + Send + Sync + 'static {
+ where F: Fn(&mut Context) -> Option<String> + Send + Sync + 'static {
self.dynamic_prefix = Some(Box::new(dynamic_prefix));
self
diff --git a/src/ext/framework/create_command.rs b/src/ext/framework/create_command.rs
index c758f4a..4d1e1a5 100644
--- a/src/ext/framework/create_command.rs
+++ b/src/ext/framework/create_command.rs
@@ -54,19 +54,19 @@ impl CreateCommand {
/// .desc("Replies to a ping with a pong")
/// .exec(ping)));
///
- /// fn ping(context: &Context, _message: &Message, _args: Vec<String>) -> Result<(), String> {
+ /// fn ping(context: &mut Context, _message: &Message, _args: Vec<String>) -> Result<(), String> {
/// let _ = context.say("Pong!");
///
/// Ok(())
/// }
///
- /// fn owner_check(_context: &Context, message: &Message) -> bool {
+ /// 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(&Context, &Message) -> bool + Send + Sync + 'static {
+ where F: Fn(&mut Context, &Message) -> bool + Send + Sync + 'static {
self.0.checks.push(Box::new(check));
self
@@ -100,7 +100,7 @@ impl CreateCommand {
///
/// [`exec_str`]: #method.exec_str
pub fn exec<F>(mut self, func: F) -> Self
- where F: Fn(&Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
+ where F: Fn(&mut Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
self.0.exec = CommandType::Basic(Box::new(func));
self
@@ -112,7 +112,7 @@ impl CreateCommand {
///
/// You can return Err(string) if there's an error.
pub fn exec_help<F>(mut self, f: F) -> Self
- where F: Fn(&Context, &Message, HashMap<String, Arc<CommandGroup>>, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
+ where F: Fn(&mut Context, &Message, HashMap<String, Arc<CommandGroup>>, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
self.0.exec = CommandType::WithCommands(Box::new(f));
self
diff --git a/src/ext/framework/create_group.rs b/src/ext/framework/create_group.rs
index 465224d..03fc33e 100644
--- a/src/ext/framework/create_group.rs
+++ b/src/ext/framework/create_group.rs
@@ -21,7 +21,7 @@ pub struct CreateGroup(pub CommandGroup);
/// framework.group("Information", |g| g
/// .prefix("info")
/// .command("name", |c| c
-/// .exec_str("meew0")))
+/// .exec_str("Hakase")))
/// ```
impl CreateGroup {
/// Adds a command to group.
@@ -45,7 +45,7 @@ impl CreateGroup {
/// Adds a command to group with simplified API.
/// You can return Err(string) if there's an error.
pub fn on<F>(mut self, command_name: &str, f: F) -> Self
- where F: Fn(&Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
+ where F: Fn(&mut Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static {
let cmd = Arc::new(Command::new(f));
self.0.commands.insert(command_name.to_owned(), CommandOrAlias::Command(cmd));
diff --git a/src/ext/framework/help_commands.rs b/src/ext/framework/help_commands.rs
index 230d78a..119e362 100644
--- a/src/ext/framework/help_commands.rs
+++ b/src/ext/framework/help_commands.rs
@@ -7,7 +7,7 @@ use ::client::Context;
use ::model::Message;
use ::utils::Colour;
-fn error_embed(ctx: &Context, message: &Message, input: &str) {
+fn error_embed(ctx: &mut Context, message: &Message, input: &str) {
let _ = ctx.send_message(message.channel_id, |m| m
.embed(|e| e
.colour(Colour::dark_red())
@@ -26,7 +26,7 @@ fn remove_aliases(cmds: &HashMap<String, CommandOrAlias>) -> HashMap<&String, &I
result
}
-pub fn with_embeds(ctx: &Context,
+pub fn with_embeds(ctx: &mut Context,
message: &Message,
groups: HashMap<String, Arc<CommandGroup>>,
args: Vec<String>) -> Result<(), String> {
@@ -154,7 +154,7 @@ pub fn with_embeds(ctx: &Context,
Ok(())
}
-pub fn plain(ctx: &Context,
+pub fn plain(ctx: &mut Context,
_: &Message,
groups: HashMap<String, Arc<CommandGroup>>,
args: Vec<String>) -> Result<(), String> {
diff --git a/src/ext/framework/mod.rs b/src/ext/framework/mod.rs
index f1d5020..77c874f 100644
--- a/src/ext/framework/mod.rs
+++ b/src/ext/framework/mod.rs
@@ -113,28 +113,32 @@ use ::client::CACHE;
#[macro_export]
macro_rules! command {
($fname:ident($c:ident) $b:block) => {
- pub fn $fname($c: &$crate::client::Context, _: &$crate::model::Message, _: Vec<String>) -> ::std::result::Result<(), String> {
+ #[allow(unused_mut)]
+ pub fn $fname(mut $c: &mut $crate::client::Context, _: &$crate::model::Message, _: Vec<String>) -> ::std::result::Result<(), String> {
$b
Ok(())
}
};
($fname:ident($c:ident, $m:ident) $b:block) => {
- pub fn $fname($c: &$crate::client::Context, $m: &$crate::model::Message, _: Vec<String>) -> ::std::result::Result<(), String> {
+ #[allow(unused_mut)]
+ pub fn $fname(mut $c: &mut $crate::client::Context, $m: &$crate::model::Message, _: Vec<String>) -> ::std::result::Result<(), String> {
$b
Ok(())
}
};
($fname:ident($c:ident, $m:ident, $a:ident) $b:block) => {
- pub fn $fname($c: &$crate::client::Context, $m: &$crate::model::Message, $a: Vec<String>) -> ::std::result::Result<(), String> {
+ #[allow(unused_mut)]
+ pub fn $fname(mut $c: &mut $crate::client::Context, $m: &$crate::model::Message, $a: Vec<String>) -> ::std::result::Result<(), String> {
$b
Ok(())
}
};
($fname:ident($c:ident, $m:ident, $a:ident, $($name:ident: $t:ty),*) $b:block) => {
- pub fn $fname($c: &$crate::client::Context, $m: &$crate::model::Message, $a: Vec<String>) -> ::std::result::Result<(), String> {
+ #[allow(unused_mut)]
+ pub fn $fname(mut $c: &mut $crate::client::Context, $m: &$crate::model::Message, $a: Vec<String>) -> ::std::result::Result<(), String> {
let mut i = $a.iter();
let mut arg_counter = 0;
@@ -260,7 +264,7 @@ impl Framework {
#[allow(cyclomatic_complexity)]
#[doc(hidden)]
- pub fn dispatch(&mut self, context: Context, message: Message) {
+ pub fn dispatch(&mut self, mut context: Context, message: Message) {
match self.configuration.account_type {
AccountType::Selfbot => {
if message.author.id != self.user_info.0 {
@@ -281,7 +285,7 @@ impl Framework {
},
AccountType::Any => {}
}
- let res = command::positions(&context, &message.content, &self.configuration);
+ let res = command::positions(&mut context, &message.content, &self.configuration);
let positions = match res {
Some(mut positions) => {
@@ -437,7 +441,7 @@ impl Framework {
}
for check in &command.checks {
- if !(check)(&context, &message) {
+ if !(check)(&mut context, &message) {
if let Some(ref message) = self.configuration.invalid_check_message {
let _ = context.say(message);
}
@@ -516,27 +520,27 @@ impl Framework {
thread::spawn(move || {
if let Some(before) = before {
- if !(before)(&context, &message, &built) && !is_owner {
+ if !(before)(&mut context, &message, &built) && !is_owner {
return;
}
}
let result = match command.exec {
CommandType::StringResponse(ref x) => {
- let _ = &context.say(x);
+ let _ = &mut context.say(x);
Ok(())
},
CommandType::Basic(ref x) => {
- (x)(&context, &message, args)
+ (x)(&mut context, &message, args)
},
CommandType::WithCommands(ref x) => {
- (x)(&context, &message, groups, args)
+ (x)(&mut context, &message, groups, args)
}
};
if let Some(after) = after {
- (after)(&context, &message, &built, result);
+ (after)(&mut context, &message, &built, result);
}
});
@@ -563,7 +567,7 @@ impl Framework {
/// [`command`]: #method.command
/// [module-level documentation]: index.html
pub fn on<F, S>(mut self, command_name: S, f: F) -> Self
- where F: Fn(&Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static,
+ where F: Fn(&mut Context, &Message, Vec<String>) -> Result<(), String> + Send + Sync + 'static,
S: Into<String> {
{
let ungrouped = self.groups.entry("Ungrouped".to_owned())
@@ -636,7 +640,7 @@ impl Framework {
/// Specify the function to be called prior to every command's execution.
/// If that function returns true, the command will be executed.
pub fn before<F>(mut self, f: F) -> Self
- where F: Fn(&Context, &Message, &String) -> bool + Send + Sync + 'static {
+ where F: Fn(&mut Context, &Message, &String) -> bool + Send + Sync + 'static {
self.before = Some(Arc::new(f));
self
@@ -645,7 +649,7 @@ impl Framework {
/// Specify the function to be called after every command's execution.
/// Fourth argument exists if command returned an error which you can handle.
pub fn after<F>(mut self, f: F) -> Self
- where F: Fn(&Context, &Message, &String, Result<(), String>) + Send + Sync + 'static {
+ where F: Fn(&mut Context, &Message, &String, Result<(), String>) + Send + Sync + 'static {
self.after = Some(Arc::new(f));
self
@@ -675,14 +679,14 @@ impl Framework {
/// let _ = context.say("Pong!");
/// });
///
- /// fn owner_check(_context: &Context, message: &Message) -> bool {
+ /// fn owner_check(_context: &mut Context, message: &Message) -> bool {
/// // replace with your user ID
/// message.author.id == 7
/// }
/// ```
#[deprecated(since="0.1.2", note="Use the `CreateCommand` builder's `check` instead.")]
pub fn set_check<F, S>(mut self, command: S, check: F) -> Self
- where F: Fn(&Context, &Message) -> bool + Send + Sync + 'static,
+ where F: Fn(&mut Context, &Message) -> bool + Send + Sync + 'static,
S: Into<String> {
{
let ungrouped = self.groups.entry("Ungrouped".to_owned())