Add CTS to /roles (#233)

Author: FloatingMilkshake

Commands/InteractionCommands/RoleInteractions.cs

@@ -16,19 +16,12 @@ internal class RoleSlashCommands
             [SlashCommand("grant", "Opt into a role.")]
             public async Task GrantRole(
                 InteractionContext ctx,
-                [Choice("Windows 11 Insiders (Canary)", "insiderCanary")]
-                [Choice("Windows 11 Insiders (Dev)", "insiderDev")]
-                [Choice("Windows 11 Insiders (Beta)", "insiderBeta")]
-                [Choice("Windows 11 Insiders (Release Preview)", "insiderRP")]
-                [Choice("Windows 10 Insiders (Release Preview)", "insider10RP")]
-                [Choice("Windows 10 Insiders (Beta)", "insider10Beta")]
-                [Choice("Patch Tuesday", "patchTuesday")]
-                [Choice("Giveaways", "giveaways")]
+                [Autocomplete(typeof(RolesAutocompleteProvider))]
                 [Option("role", "The role to opt into.")] string role)
             {
                 DiscordMember member = ctx.Member;
 
-                var roleId = role switch
+                ulong roleId = role switch
                 {
                     "insiderCanary" => Program.cfgjson.UserRoles.InsiderCanary,
                     "insiderDev" => Program.cfgjson.UserRoles.InsiderDev,
@@ -38,8 +31,21 @@ public async Task GrantRole(
                     "insider10Beta" => Program.cfgjson.UserRoles.Insider10Beta,
                     "patchTuesday" => Program.cfgjson.UserRoles.PatchTuesday,
                     "giveaways" => Program.cfgjson.UserRoles.Giveaways,
-                    _ => throw new NotSupportedException()
+                    "cts" => Program.cfgjson.CommunityTechSupportRoleID,
+                    _ => 0
                 };
+                
+                if (roleId == 0)
+                {
+                    await ctx.RespondAsync($"{Program.cfgjson.Emoji.Error} Invalid role! Please choose from the list.", ephemeral: true);
+                    return;
+                }
+                
+                if (roleId == Program.cfgjson.CommunityTechSupportRoleID && await GetPermLevelAsync(ctx.Member) < ServerPermLevel.TechnicalQueriesSlayer)
+                {
+                    await ctx.RespondAsync($"{Program.cfgjson.Emoji.NoPermissions} You must be a TQS member to get the CTS role!", ephemeral: true);
+                    return;
+                }
 
                 var roleData = await ctx.Guild.GetRoleAsync(roleId);
 
@@ -50,19 +56,12 @@ public async Task GrantRole(
             [SlashCommand("remove", "Opt out of a role.")]
             public async Task RemoveRole(
                 InteractionContext ctx,
-                [Choice("Windows 11 Insiders (Canary)", "insiderCanary")]
-                [Choice("Windows 11 Insiders (Dev)", "insiderDev")]
-                [Choice("Windows 11 Insiders (Beta)", "insiderBeta")]
-                [Choice("Windows 11 Insiders (Release Preview)", "insiderRP")]
-                [Choice("Windows 10 Insiders (Release Preview)", "insider10RP")]
-                [Choice("Windows 10 Insiders (Beta)", "insider10Beta")]
-                [Choice("Patch Tuesday", "patchTuesday")]
-                [Choice("Giveaways", "giveaways")]
+                [Autocomplete(typeof(RolesAutocompleteProvider))]
                 [Option("role", "The role to opt out of.")] string role)
             {
                 DiscordMember member = ctx.Member;
 
-                var roleId = role switch
+                ulong roleId = role switch
                 {
                     "insiderCanary" => Program.cfgjson.UserRoles.InsiderCanary,
                     "insiderDev" => Program.cfgjson.UserRoles.InsiderDev,
@@ -72,14 +71,55 @@ public async Task RemoveRole(
                     "insider10Beta" => Program.cfgjson.UserRoles.Insider10Beta,
                     "patchTuesday" => Program.cfgjson.UserRoles.PatchTuesday,
                     "giveaways" => Program.cfgjson.UserRoles.Giveaways,
-                    _ => throw new NotSupportedException()
+                    "cts" => Program.cfgjson.CommunityTechSupportRoleID,
+                    _ => 0
                 };
+                
+                if (roleId == 0)
+                {
+                    await ctx.RespondAsync($"{Program.cfgjson.Emoji.Error} Invalid role! Please choose from the list.", ephemeral: true);
+                    return;
+                }
 
                 var roleData = await ctx.Guild.GetRoleAsync(roleId);
 
                 await member.RevokeRoleAsync(roleData, $"/roles remove used by {DiscordHelpers.UniqueUsername(ctx.User)}");
                 await ctx.RespondAsync($"{Program.cfgjson.Emoji.Success} The role {roleData.Mention} has been successfully removed!", ephemeral: true, mentions: false);
             }
         }
+        
+        internal class RolesAutocompleteProvider : IAutocompleteProvider
+        {
+            public async Task<IEnumerable<DiscordAutoCompleteChoice>> Provider(AutocompleteContext ctx)
+            {
+                Dictionary<string, string> options = new()
+                    {
+                        { "Windows 11 Insiders (Canary)", "insiderCanary" },
+                        { "Windows 11 Insiders (Dev)", "insiderDev" },
+                        { "Windows 11 Insiders (Beta)", "insiderBeta" },
+                        { "Windows 11 Insiders (Release Preview)", "insiderRP" },
+                        { "Windows 10 Insiders (Release Preview)", "insider10RP" },
+                        { "Windows 10 Insiders (Beta)", "insider10Beta" },
+                        { "Patch Tuesday", "patchTuesday" },
+                        { "Giveaways", "giveaways" },
+                        { "Community Tech Support (CTS)", "cts" }
+                    };
+                
+                var memberHasTqs = await GetPermLevelAsync(ctx.Member) >= ServerPermLevel.TechnicalQueriesSlayer;
+                
+                List<DiscordAutoCompleteChoice> list = new();
+                
+                foreach (var option in options)
+                {
+                    if (ctx.FocusedOption.Value.ToString() == "" || option.Key.Contains(ctx.FocusedOption.Value.ToString(), StringComparison.OrdinalIgnoreCase))
+                    {
+                        if (option.Value == "cts" && !memberHasTqs) continue;
+                        list.Add(new DiscordAutoCompleteChoice(option.Key, option.Value));
+                    }
+                }
+                
+                return list;
+            }
+        }
     }
 }