From e44838f4339b90817b5eba5df16230b02487f0cc Mon Sep 17 00:00:00 2001
From: Illia <fwrs@users.noreply.github.com>
Date: Sat, 10 Dec 2016 22:25:55 +0200
Subject: More config for CreateCommand, add various methods

Adds multiple configurations to the command builder, and adds methods
to various structs.

Context::get_current_user is a shortcut to retrieve the current user
from the cache.

Message::get_member retrieves the member object of the message, if sent
in a guild. Message::is_private checks if the message was sent in a
Group or PrivateChannel.

User::member retrieves the user's member object in a guild by Id;

Adds 6 configurations to the command builder:

- dm_only: whether the command can only be used in direct messages;
- guild_only: whether the command can only be used in guilds;
- help_available: whether the command should be displayed in the help
  list;
- max_args: specify the maximum number of arguments a command must be
  given;
- min_args: specify the minimum number of arguments a command must be
  given;
- required_permissions: the permissions a member must have to be able
  to use the command;
---
 src/ext/framework/create_command.rs | 49 +++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

(limited to 'src/ext/framework/create_command.rs')

diff --git a/src/ext/framework/create_command.rs b/src/ext/framework/create_command.rs
index 824c3dd..6d40592 100644
--- a/src/ext/framework/create_command.rs
+++ b/src/ext/framework/create_command.rs
@@ -5,6 +5,7 @@ use std::default::Default;
 use std::sync::Arc;
 use ::client::Context;
 use ::model::Message;
+use ::model::Permissions;
 
 pub struct CreateCommand(pub Command);
 
@@ -54,6 +55,48 @@ impl CreateCommand {
         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
+    }
+
+    /// 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
+    }
+
+    /// 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
+    }
+
+    /// 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
+    }
+
+    /// 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
+    }
+
+    /// Maximum amount of arguments that can be passed.
+    pub fn required_permissions(mut self, required_permissions: Permissions) -> Self {
+        self.0.required_permissions = required_permissions;
+
+        self
+    }
+
     /// A function that can be called when a command is received.
     ///
     /// See [`exec_str`] if you _only_ need to return a string on command use.
@@ -126,6 +169,12 @@ impl Default for Command {
             desc: None,
             usage: None,
             use_quotes: true,
+            min_args: None,
+            max_args: None,
+            required_permissions: Permissions::empty(),
+            dm_only: false,
+            guild_only: false,
+            help_available: true
         }
     }
 }
-- 
cgit v1.2.3