From 9843298af8fe268dcca67764858edf289cf94935 Mon Sep 17 00:00:00 2001
From: RogueException <rogueexception@gmail.com>
Date: Wed, 10 Feb 2016 17:19:19 -0400
Subject: [PATCH] Added GlobalWhitelist/Blacklist extension methods

---
 .../Userlist/BlacklistExtensions.cs           | 25 +++++++++++++++--
 .../Userlist/WhitelistExtensions.cs           | 27 ++++++++++++++++---
 2 files changed, 47 insertions(+), 5 deletions(-)

diff --git a/src/Discord.Net.Commands/Permissions/Userlist/BlacklistExtensions.cs b/src/Discord.Net.Commands/Permissions/Userlist/BlacklistExtensions.cs
index 41527a50d..21de4076d 100644
--- a/src/Discord.Net.Commands/Permissions/Userlist/BlacklistExtensions.cs
+++ b/src/Discord.Net.Commands/Permissions/Userlist/BlacklistExtensions.cs
@@ -1,4 +1,6 @@
-namespace Discord.Commands.Permissions.Userlist
+using System.Collections.Generic;
+
+namespace Discord.Commands.Permissions.Userlist
 {
     public static class BlacklistExtensions
     {
@@ -23,5 +25,24 @@
 			service.Root.AddCheck(new BlacklistChecker(service.Client));
 			return service;
 		}
-	}
+
+        public static IEnumerable<ulong> GetBlacklistedUserIds(this DiscordClient client)
+            => client.Services.Get<BlacklistService>().UserIds;
+        public static void BlacklistUser(this DiscordClient client, User user)
+        {
+            client.Services.Get<BlacklistService>().Add(user.Id);
+        }
+        public static void BlacklistUser(this DiscordClient client, ulong userId)
+        {
+            client.Services.Get<BlacklistService>().Add(userId);
+        }
+        public static void UnBlacklistUser(this DiscordClient client, User user)
+        {
+            client.Services.Get<BlacklistService>().Remove(user.Id);
+        }
+        public static void UnBlacklistUser(this DiscordClient client, ulong userId)
+        {
+            client.Services.Get<BlacklistService>().Remove(userId);
+        }
+    }
 }
diff --git a/src/Discord.Net.Commands/Permissions/Userlist/WhitelistExtensions.cs b/src/Discord.Net.Commands/Permissions/Userlist/WhitelistExtensions.cs
index 2e29ee47a..eaa136075 100644
--- a/src/Discord.Net.Commands/Permissions/Userlist/WhitelistExtensions.cs
+++ b/src/Discord.Net.Commands/Permissions/Userlist/WhitelistExtensions.cs
@@ -1,4 +1,6 @@
-namespace Discord.Commands.Permissions.Userlist
+using System.Collections.Generic;
+
+namespace Discord.Commands.Permissions.Userlist
 {
     public static class WhitelistExtensions
     {
@@ -22,6 +24,25 @@
 		{
 			service.Root.AddCheck(new BlacklistChecker(service.Client));
 			return service;
-		}
-	}
+        }
+
+        public static IEnumerable<ulong> GetWhitelistedUserIds(this DiscordClient client)
+            => client.Services.Get<WhitelistService>().UserIds;
+        public static void WhitelistUser(this DiscordClient client, User user)
+        {
+            client.Services.Get<WhitelistService>().Add(user.Id);
+        }
+        public static void WhitelistUser(this DiscordClient client, ulong userId)
+        {
+            client.Services.Get<WhitelistService>().Add(userId);
+        }
+        public static void UnWhitelistUser(this DiscordClient client, User user)
+        {
+            client.Services.Get<WhitelistService>().Remove(user.Id);
+        }
+        public static void RemoveFromWhitelist(this DiscordClient client, ulong userId)
+        {
+            client.Services.Get<WhitelistService>().Remove(userId);
+        }
+    }
 }