expand RequireSpecificGuildAttribute to work with more than one server and add another server to work with Pirate and PirateBan

Author: GreemDev

src/Bot/Helpers/TextCommandHelper.cs

@@ -47,7 +47,7 @@ public static bool IsAccessibleToGuild(this Command command, ulong id)
         return command.Checks
                    .TryGetFirst(x => x is RequireSpecificGuildAttribute, out var attr)
                && attr is RequireSpecificGuildAttribute rsga
-               && rsga.GuildId == id;
+               && rsga.GuildIds.Contains(id);
     }
 
     public static async ValueTask<EmbedBuilder> CreateCommandEmbedAsync(Command command, VolteContext ctx)

src/Bot/Systems/Commands/Text/Attributes/Checks/RequireSpecificGuildAttribute.cs

@@ -1,13 +1,13 @@
 namespace Volte.Systems.Commands.Text;
 
-public class RequireSpecificGuildAttribute(ulong guildId) : CheckAttribute
+public class RequireSpecificGuildAttribute(ulong[] guildIds) : CheckAttribute
 {
-    public ulong GuildId => guildId;
+    public ulong[] GuildIds => guildIds;
 
     public override ValueTask<CheckResult> CheckAsync(CommandContext context)
     {
         var ctx = context.Cast<VolteContext>();
-        if (ctx.Guild.Id == guildId) 
+        if (GuildIds.Contains(ctx.Guild.Id))
             return CheckResult.Successful;
 
         return CheckResult.Failed("Insufficient permission.");

src/Bot/Systems/Commands/Text/Modules/Moderation/PirateBanCommand.cs

@@ -4,7 +4,7 @@ public partial class ModerationModule
 {
     [Command("PirateBan", "PBan")]
     [Description("Bans the user with a very long-winded message as to why piracy is not supported. Content is retrieved from the 'emulationisnotpiracy' tag.")]
-    [RequireSpecificGuild(1294443224030511104)]
+    [RequireSpecificGuild([1294443224030511104, 1325735818714943498])]
     public async Task<ActionResult> PirateBanAsync(
         [CheckHierarchy, EnsureNotSelf, Description("The target user.")] RestUser user)
     {

src/Bot/Systems/Commands/Text/Modules/Moderation/PirateCommand.cs

@@ -2,25 +2,39 @@
 
 public partial class ModerationModule
 {
-    public const ulong VerifiedSwitchOwnerRoleId = 1334992661198930001;
-    public const ulong NoSupportRoleId = 1298451667863470120;
+    public readonly Dictionary<Snowflake, Snowflake> GuildsToVerifiedSwitchOwnerRoleIds = new()
+    {
+        { 1294443224030511104, 1334992661198930001 }, // R
+        { 1325735818714943498, 1354326584550625460 } // C
+    };
+    
+    public readonly Dictionary<Snowflake, Snowflake> GuildsToPirateRoleIds = new()
+    {
+        { 1294443224030511104, 1298451667863470120 }, // R
+        { 1325735818714943498, 1405060402362060911 } // C
+    };
 
     [Command("Pirate", "NoSupport")]
-    [Description("Warns the target user for Rule 4, gives them the No Support role, and, if they have it, takes away Verified Switch Owner.")]
-    [RequireSpecificGuild(1294443224030511104)]
+    [Description("Warns the target user for piracy, gives them the server's pirate role, and, if they have it, takes away Verified Switch Owner.")]
+    [RequireSpecificGuild([1294443224030511104, 1325735818714943498])]
     public async Task<ActionResult> PirateAsync([CheckHierarchy, EnsureNotSelf, Description("The target user.")] SocketGuildUser member)
     {
-        await member.WarnAsync(Context, "Rule #4");
-        if (member.HasRole(VerifiedSwitchOwnerRoleId))
-            await member.RemoveRoleAsync(VerifiedSwitchOwnerRoleId);
+        await member.WarnAsync(Context, Context.Guild.Id is 1294443224030511104 ? "Rule #4" : "Piracy");
+
+        var verifiedRole = GuildsToVerifiedSwitchOwnerRoleIds[Context.Guild.Id];
+        
+        if (member.HasRole(verifiedRole))
+            await member.RemoveRoleAsync(verifiedRole);
 
-        if (!member.HasRole(NoSupportRoleId))
-            await member.AddRoleAsync(NoSupportRoleId);
+        var pirateRole = GuildsToPirateRoleIds[Context.Guild.Id];
+        
+        if (!member.HasRole(pirateRole))
+            await member.AddRoleAsync(pirateRole);
 
-        return Ok($"Successfully warned **{member}** for piracy, and gave them <@&1298451667863470120>.",
+        return Ok($"Successfully warned **{member}** for piracy, and gave them <@&{pirateRole.Raw}>.",
             _ => ModerationService.OnModActionCompleteAsync(ModActionEventArgs.InContext(Context)
                 .WithActionType(ModActionType.Warn)
                 .WithTarget(member)
-                .WithReason("#rules #4")));
+                .WithReason(Context.Guild.Id is 1294443224030511104 ? "#rules #4" : "Piracy")));
     }
 }