From 420f9bdaa5a5022ff1d769f1d44a689a6fea12a4 Mon Sep 17 00:00:00 2001
From: acdenisSK <acdenissk69@gmail.com>
Date: Sat, 8 Jul 2017 00:43:09 +0200
Subject: Implement attaching reasons to bans

---
 src/model/guild/guild_id.rs | 28 ++++++++++++++++++++++-----
 src/model/guild/member.rs   | 46 +++++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 67 insertions(+), 7 deletions(-)

(limited to 'src/model')

diff --git a/src/model/guild/guild_id.rs b/src/model/guild/guild_id.rs
index 08538b1..68212c7 100644
--- a/src/model/guild/guild_id.rs
+++ b/src/model/guild/guild_id.rs
@@ -1,5 +1,6 @@
 use std::fmt::{Display, Formatter, Result as FmtResult};
 use ::model::*;
+use ::model::guild::BanOptions;
 
 #[cfg(feature="cache")]
 use ::CACHE;
@@ -45,13 +46,30 @@ impl GuildId {
     /// [`Guild::ban`]: struct.Guild.html#method.ban
     /// [`User`]: struct.User.html
     /// [Ban Members]: permissions/constant.BAN_MEMBERS.html
-    pub fn ban<U: Into<UserId>>(&self, user: U, delete_message_days: u8)
+    pub fn ban<U: Into<UserId>, BO: Into<BanOptions>>(&self, user: U, ban_options: BO)
         -> Result<()> {
-        if delete_message_days > 7 {
-            return Err(Error::Model(ModelError::DeleteMessageDaysAmount(delete_message_days)));
+        
+        use self::BanOptions::*;
+
+        match ban_options.into() {
+            DeleteMessageDays(dmd) => {
+                if dmd > 7 {
+                    return Err(Error::Model(ModelError::DeleteMessageDaysAmount(dmd)));
+                }
+
+                http::ban_user(self.0, user.into().0, dmd, "")
+            },
+            DMDReason(dmd, reason) => {
+                if dmd > 7 {
+                    return Err(Error::Model(ModelError::DeleteMessageDaysAmount(dmd)));
+                }
+
+                http::ban_user(self.0, user.into().0, dmd, &*reason)
+            },
+            Reason(reason) => {
+                http::ban_user(self.0, user.into().0, 0, &*reason)
+            },
         }
-
-        http::ban_user(self.0, user.into().0, delete_message_days)
     }
 
     /// Gets a list of the guild's bans.
diff --git a/src/model/guild/member.rs b/src/model/guild/member.rs
index 444d4af..ebf99d0 100644
--- a/src/model/guild/member.rs
+++ b/src/model/guild/member.rs
@@ -16,6 +16,36 @@ use ::builder::EditMember;
 #[cfg(all(feature="cache", feature="model", feature="utils"))]
 use ::utils::Colour;
 
+pub enum BanOptions {
+    DeleteMessageDays(u8),
+    DMDReason(u8, String),
+    Reason(String),
+}
+
+impl From<u8> for BanOptions {
+    fn from(dmd: u8) -> Self {
+        BanOptions::DeleteMessageDays(dmd)
+    }
+}
+
+impl<'a> From<&'a str> for BanOptions {
+    fn from(reason: &'a str) -> Self {
+        BanOptions::Reason(reason.to_string())
+    }
+}
+
+impl From<String> for BanOptions {
+    fn from(reason: String) -> Self {
+        BanOptions::Reason(reason)
+    }
+}
+
+impl<'a> From<(u8, &'a str)> for BanOptions {
+    fn from(dmdreason: (u8, &'a str)) -> Self {
+        BanOptions::DMDReason(dmdreason.0, dmdreason.1.to_string())
+    }
+}
+
 /// Information about a member of a guild.
 #[derive(Clone, Debug, Deserialize)]
 pub struct Member {
@@ -102,8 +132,20 @@ impl Member {
     ///
     /// [Ban Members]: permissions/constant.BAN_MEMBERS.html
     #[cfg(feature="cache")]
-    pub fn ban(&self, delete_message_days: u8) -> Result<()> {
-        http::ban_user(self.guild_id.0, self.user.read().unwrap().id.0, delete_message_days)
+    pub fn ban<BO: Into<BanOptions>>(&self, ban_options: BO) -> Result<()> {
+        use self::BanOptions::*;
+
+        match ban_options.into() {
+            DeleteMessageDays(dmd) => {
+                http::ban_user(self.guild_id.0, self.user.read().unwrap().id.0, dmd, "")
+            },
+            DMDReason(dmd, reason) => {
+                http::ban_user(self.guild_id.0, self.user.read().unwrap().id.0, dmd, &*reason)
+            },
+            Reason(reason) => {
+                http::ban_user(self.guild_id.0, self.user.read().unwrap().id.0, 0, &*reason)
+            },
+        }
     }
 
     /// Determines the member's colour.
-- 
cgit v1.2.3