Add !tqsmute and /tqsmute

Erisa · Feb 15, 2024 · 3fe757d · 3fe757d
1 parent 85434a9
commit 3fe757d
Show file tree

Hide file tree

Showing 5 changed files with 198 additions and 30 deletions.
diff --git a/CommandChecks/HomeServerPerms.cs b/CommandChecks/HomeServerPerms.cs
@@ -1,4 +1,6 @@
-namespace Cliptok.CommandChecks
+using System.Formats.Tar;
+
+namespace Cliptok.CommandChecks
 {
     public class ServerPerms
     {
@@ -16,6 +18,7 @@ public enum ServerPermLevel
             Tier8,
             TierS,
             TierX,
+            TechnicalQueriesSlayer,
             TrialModerator,
             Moderator,
             Admin,
@@ -38,6 +41,8 @@ public static ServerPermLevel GetPermLevel(DiscordMember target)
                 return ServerPermLevel.Muted;
             else if (target.Roles.Contains(target.Guild.GetRole(Program.cfgjson.TrialModRole)))
                 return ServerPermLevel.TrialModerator;
+            else if (target.Roles.Contains(target.Guild.GetRole(Program.cfgjson.TqsRoleId)))
+                return ServerPermLevel.TechnicalQueriesSlayer;
             else if (target.Roles.Contains(target.Guild.GetRole(Program.cfgjson.TierRoles[9])))
                 return ServerPermLevel.TierX;
             else if (target.Roles.Contains(target.Guild.GetRole(Program.cfgjson.TierRoles[8])))

diff --git a/Commands/InteractionCommands/MuteInteractions.cs b/Commands/InteractionCommands/MuteInteractions.cs
@@ -89,5 +89,47 @@ public async Task UnmuteSlashCommand(
                     await ctx.RespondAsync($"{Program.cfgjson.Emoji.Error} That user doesn't appear to be muted, *and* an error occurred while attempting to unmute them anyway. Please contact the bot owner, the error has been logged.");
                 }
         }
+
+        [SlashCommand("tqsmute", "Temporarily mute a user in tech support channels.")]
+        [SlashRequireHomeserverPerm(ServerPermLevel.TechnicalQueriesSlayer)]
+        public async Task TqsMuteSlashCommand(
+            InteractionContext ctx,
+            [Option("user", "The user to mute.")] DiscordUser targetUser,
+            [Option("reason", "The reason for the mute.")] string reason)
+        {
+            await ctx.DeferAsync(ephemeral: true);
+
+            // Only allow usage in #tech-support, #tech-support-forum, and their threads
+            if (ctx.Channel.Id != Program.cfgjson.TechSupportChannel &&
+                ctx.Channel.Id != Program.cfgjson.SupportForumId &&
+                ctx.Channel.Parent.Id != Program.cfgjson.TechSupportChannel &&
+                ctx.Channel.Parent.Id != Program.cfgjson.SupportForumId)
+            {
+                await ctx.EditResponseAsync(new DiscordWebhookBuilder().WithContent($"{Program.cfgjson.Emoji.Error} This command can only be used in <#{Program.cfgjson.TechSupportChannel}>, <#{Program.cfgjson.SupportForumId}>, and threads in those channels!"));
+                return;
+            }
+
+            DiscordMember targetMember = default;
+            try
+            {
+                targetMember = await ctx.Guild.GetMemberAsync(targetUser.Id);
+            }
+            catch (DSharpPlus.Exceptions.NotFoundException)
+            {
+                // blah
+            }
+
+            if (targetMember != default && GetPermLevel(ctx.Member) == ServerPermLevel.TechnicalQueriesSlayer && (GetPermLevel(targetMember) >= ServerPermLevel.TechnicalQueriesSlayer || targetMember.IsBot))
+            {
+                await ctx.EditResponseAsync(new DiscordWebhookBuilder().WithContent($"{Program.cfgjson.Emoji.Error} {ctx.User.Mention}, you cannot mute other TQS or staff members."));
+                return;
+            }
+
+            // mute duration is static for TQS mutes
+            TimeSpan muteDuration = TimeSpan.FromHours(Program.cfgjson.TqsMuteDurationHours);
+
+            await MuteHelpers.MuteUserAsync(targetUser, reason, ctx.User.Id, ctx.Guild, ctx.Channel, muteDuration, true, true);
+            await ctx.EditResponseAsync(new DiscordWebhookBuilder().WithContent("Done. Please open a modmail thread for this user if you haven't already!"));
+        }
     }
 }
diff --git a/Commands/Mutes.cs b/Commands/Mutes.cs
@@ -12,6 +12,7 @@ public async Task UnmuteCmd(CommandContext ctx, [Description("The user you're tr
 
             // todo: store per-guild
             DiscordRole mutedRole = ctx.Guild.GetRole(Program.cfgjson.MutedRole);
+            DiscordRole tqsMutedRole = ctx.Guild.GetRole(Program.cfgjson.TqsMutedRole);
 
             DiscordMember member = default;
             try
@@ -23,7 +24,7 @@ public async Task UnmuteCmd(CommandContext ctx, [Description("The user you're tr
                 Program.discord.Logger.LogWarning(eventId: Program.CliptokEventID, exception: ex, message: "Failed to unmute {user} in {server} because they weren't in the server.", $"{DiscordHelpers.UniqueUsername(targetUser)}", ctx.Guild.Name);
             }
 
-            if ((await Program.db.HashExistsAsync("mutes", targetUser.Id)) || (member != default && member.Roles.Contains(mutedRole)))
+            if ((await Program.db.HashExistsAsync("mutes", targetUser.Id)) || (member != default && (member.Roles.Contains(mutedRole) || member.Roles.Contains(tqsMutedRole))))
             {
                 await MuteHelpers.UnmuteUserAsync(targetUser, reason);
                 await ctx.RespondAsync($"{Program.cfgjson.Emoji.Information} Successfully unmuted **{DiscordHelpers.UniqueUsername(targetUser)}**.");
@@ -93,5 +94,47 @@ public async Task MuteCmd(
 
             _ = MuteHelpers.MuteUserAsync(targetUser, reason, ctx.User.Id, ctx.Guild, ctx.Channel, muteDuration, true);
         }
+
+        [Command("tqsmute")]
+        [Description(
+            "Temporarily mutes a user, preventing them from sending messages in #tech-support and related channels until they're unmuted.")]
+        [HomeServer, RequireHomeserverPerm(ServerPermLevel.TechnicalQueriesSlayer)]
+        public async Task TqsMuteCmd(
+            CommandContext ctx, [Description("The user to mute")] DiscordUser targetUser,
+            [RemainingText, Description("The reason for the mute")] string reason = "No reason specified.")
+        {
+            // Only allow usage in #tech-support, #tech-support-forum, and their threads
+            if (ctx.Channel.Id != Program.cfgjson.TechSupportChannel &&
+                ctx.Channel.Id != Program.cfgjson.SupportForumId &&
+                ctx.Channel.Parent.Id != Program.cfgjson.TechSupportChannel &&
+                ctx.Channel.Parent.Id != Program.cfgjson.SupportForumId)
+            {
+                await ctx.RespondAsync($"{Program.cfgjson.Emoji.Error} This command can only be used in <#{Program.cfgjson.TechSupportChannel}>, <#{Program.cfgjson.SupportForumId}>, and threads in those channels!");
+                return;
+            }
+
+            DiscordMember targetMember = default;
+            try
+            {
+                targetMember = await ctx.Guild.GetMemberAsync(targetUser.Id);
+            }
+            catch (DSharpPlus.Exceptions.NotFoundException)
+            {
+                // blah
+            }
+
+            if (targetMember != default && GetPermLevel(ctx.Member) == ServerPermLevel.TechnicalQueriesSlayer && (GetPermLevel(targetMember) >= ServerPermLevel.TechnicalQueriesSlayer || targetMember.IsBot))
+            {
+                await ctx.Channel.SendMessageAsync($"{Program.cfgjson.Emoji.Error} {ctx.User.Mention}, you cannot mute other TQS or staff members.");
+                return;
+            }
+
+            await ctx.Message.DeleteAsync();
+
+            // mute duration is static for TQS mutes
+            TimeSpan muteDuration = TimeSpan.FromHours(Program.cfgjson.TqsMuteDurationHours);
+
+            MuteHelpers.MuteUserAsync(targetUser, reason, ctx.User.Id, ctx.Guild, ctx.Channel, muteDuration, true, true);
+        }
     }
 }