Added GlobalWhitelist/Blacklist extension methods

This commit is contained in:
RogueException
2016-02-10 17:19:19 -04:00
parent 86addc2594
commit 9843298af8
2 changed files with 47 additions and 5 deletions

View File

@@ -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);
}
}
}

View File

@@ -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);
}
}
}