Add role self service commands

Geekbot.net/Commands/Role.cs

@@ -0,0 +1,122 @@
+using System;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using AngleSharp;
+using Discord;
+using Discord.Commands;
+using Geekbot.net.Lib;
+using StackExchange.Redis;
+
+namespace Geekbot.net.Commands
+{
+    [Group("role")]
+    public class Role : ModuleBase
+    {
+        private readonly IErrorHandler _errorHandler;
+        private readonly IDatabase _redis;
+        
+        public Role(IErrorHandler errorHandler, IDatabase redis)
+        {
+            _errorHandler = errorHandler;
+            _redis = redis;
+        }
+        
+        [Command(RunMode = RunMode.Async)]
+        [Remarks(CommandCategories.Helpers)]
+        [Summary("Get a list of all available roles.")]
+        public async Task getAllRoles()
+        {
+            try
+            {
+                var roles = _redis.HashGetAll($"{Context.Guild.Id}:RoleWhitelist");
+                if (roles.Length == 0)
+                {
+                    await ReplyAsync("There are no roles configured for this server");
+                    return;
+                }
+                var sb = new StringBuilder();
+                sb.AppendLine($"**Self Service Roles on {Context.Guild.Name}**");
+                sb.AppendLine("To get a role, use `!role name`");
+                foreach (var role in roles)
+                {
+                    sb.AppendLine($"- {role.Name}");
+                }
+                await ReplyAsync(sb.ToString());
+            }
+            catch (Exception e)
+            {
+                _errorHandler.HandleCommandException(e, Context);
+            }
+        }
+
+        [Command(RunMode = RunMode.Async)]
+        [Remarks(CommandCategories.Helpers)]
+        [Summary("Get a role by mentioning it.")]
+        public async Task giveRole([Summary("roleNickname")] string roleName)
+        {
+            try
+            {
+                if (_redis.HashExists($"{Context.Guild.Id}:RoleWhitelist", roleName))
+                {
+                    var guildUser = (IGuildUser) Context.User;
+                    var roleId = ulong.Parse(_redis.HashGet($"{Context.Guild.Id}:RoleWhitelist", roleName));
+                    var role = Context.Guild.Roles.First(r => r.Id == roleId);
+                    if (role == null)
+                    {
+                        await ReplyAsync("That role doesn't seem to exist");
+                        return;
+                    }
+                    if (guildUser.RoleIds.Contains(roleId))
+                    {
+                        guildUser.RemoveRoleAsync(role);
+                        await ReplyAsync($"Removed you from {role.Name}");
+                        return;
+                    }
+                    await guildUser.AddRoleAsync(role);
+                    await ReplyAsync($"Added you to {role.Name}");
+                    return;
+                }
+                await ReplyAsync("That role doesn't seem to exist");
+            }
+            catch (Exception e)
+            {
+                _errorHandler.HandleCommandException(e, Context);
+            }
+        }
+
+        [RequireUserPermission(GuildPermission.Administrator)]
+        [Command("add", RunMode = RunMode.Async)]
+        [Remarks(CommandCategories.Admin)]
+        [Summary("Add a role to the whitelist.")]
+        public async Task addRole([Summary("@role")] IRole role, [Summary("alias")] string roleName)
+        {
+            try
+            {
+                _redis.HashSet($"{Context.Guild.Id}:RoleWhitelist", new HashEntry[] { new HashEntry(roleName, role.Id.ToString()) });
+                await ReplyAsync($"Added {role.Name} to the whitelist");
+            }
+            catch (Exception e)
+            {
+                _errorHandler.HandleCommandException(e, Context);
+            }
+        }
+
+        [RequireUserPermission(GuildPermission.Administrator)]
+        [Command("remove", RunMode = RunMode.Async)]
+        [Remarks(CommandCategories.Admin)]
+        [Summary("Remove a role from the whitelist.")]
+        public async Task removeRole([Summary("roleNickname")] string roleName)
+        {
+            try
+            {
+                _redis.HashDelete($"{Context.Guild.Id}:RoleWhitelist", roleName);
+                await ReplyAsync($"Removed {roleName} from the whitelist");
+            }
+            catch (Exception e)
+            {
+                _errorHandler.HandleCommandException(e, Context);
+            }
+        }
+    }
+}

Geekbot.net/Lib/ErrorHandler.cs

@@ -6,26 +6,23 @@ namespace Geekbot.net.Lib
 {
     public class ErrorHandler : IErrorHandler
     {
-        private readonly ILogger logger;
+        private readonly ILogger _logger;
-//        private readonly IDMChannel botOwnerDmChannel;
 
-        public ErrorHandler(ILogger logger /*, IDMChannel botOwnerDmChannel*/)
+        public ErrorHandler(ILogger logger)
         {
-            this.logger = logger;
+            _logger = logger;
-//            this.botOwnerDmChannel = botOwnerDmChannel;
         }
 
         public void HandleCommandException(Exception e, ICommandContext Context, string errorMessage = "Something went wrong :confused:")
         {
             var errorMsg =
-                $"Error Occured while executing \"{Context.Message.Content}\", executed by \"{Context.User.Username}\", complete message was \"{Context.Message}\"";
+                $"Error Occured while executing \"{Context.Message.Content}\", executed by \"{Context.User.Username}\"";
-            logger.Error(e, errorMsg);
+            _logger.Error(e, errorMsg);
             if (!string.IsNullOrEmpty(errorMessage))
             {
                 Context.Channel.SendMessageAsync(errorMessage);
             }
-//            await botOwnerDmChannel.SendMessageAsync($"{errorMsg}```{e.StackTrace}```");
+
-//            await Context.Channel.SendMessageAsync("Something went wrong...");
         }
     }