role cmd can now give mod tester role

xiaoxiao921 · Sep 25, 2020 · 24e0330 · 24e0330
1 parent 009d3c8
commit 24e0330
Showing 1 changed file with 41 additions and 13 deletions.
diff --git a/CHEF/Components/Commands/GiveRole.cs b/CHEF/Components/Commands/GiveRole.cs
@@ -14,29 +14,57 @@ public async Task GiveRole(
             [Summary("The role you want to give / remove access to.")]
             string roleName)
         {
-            if (roleName == "nsfw")
-            {
-                var user = Context.User;
-                var nsfwRole = Context.Guild.Roles.FirstOrDefault(x => x.Name == roleName);
-                var nsfwRoleId = nsfwRole?.Id;
+            var user = Context.User;
+            roleName = roleName.ToLowerInvariant();
 
-                if (user is IGuildUser gUser && nsfwRoleId != null)
+            if (user is IGuildUser gUser)
+            {
+                if (roleName == "nsfw")
                 {
-                    if (gUser.RoleIds.Any(id => id == nsfwRoleId))
+                    var nsfwRole = Context.Guild.Roles.FirstOrDefault(x => x.Name == roleName);
+                    var nsfwRoleId = nsfwRole?.Id;
+
+                    if (nsfwRoleId != null)
                     {
-                        await gUser.RemoveRoleAsync(nsfwRole);
+                        if (gUser.RoleIds.Any(id => id == nsfwRoleId))
+                        {
+                            await gUser.RemoveRoleAsync(nsfwRole);
+                        }
+                        else
+                        {
+                            await gUser.AddRoleAsync(nsfwRole);
+                        }
+
+                        await Context.Message.AddReactionAsync(Emote.Parse("<:KappaPride:570231271645511692>"));
                     }
-                    else
+                }
+                else if (roleName == "guinea pig" || roleName == "mod tester")
+                {
+                    var modTesterRole = Context.Guild.Roles.FirstOrDefault(x => x.Name == "guinea pig / mod tester");
+                    var modTesterRoleId = modTesterRole?.Id;
+
+                    if (modTesterRoleId != null)
                     {
-                        await gUser.AddRoleAsync(nsfwRole);
-                    }
+                        if (gUser.RoleIds.Any(id => id == modTesterRoleId))
+                        {
+                            await gUser.RemoveRoleAsync(modTesterRole);
+                        }
+                        else
+                        {
+                            await gUser.AddRoleAsync(modTesterRole);
+                        }
 
-                    await Context.Message.AddReactionAsync(Emote.Parse("<:KappaPride:570231271645511692>"));
+                        await Context.Message.AddReactionAsync(new Emoji("✅"));
+                    }
+                }
+                else
+                {
+                    await Context.Channel.SendMessageAsync("Can only give/remove nsfw or the mod tester role for now.");
                 }
             }
             else
             {
-                await Context.Channel.SendMessageAsync("Can only give/remove nsfw role for now.");
+                await Context.Channel.SendMessageAsync("Just tried to give a role to someone who is not in the server.");
             }
         }
     }