Add mass dehoist as a scheduled task

Fixes #258

Author: Erisa

Commands/DebugCmds.cs

@@ -139,8 +139,9 @@ public async Task Refresh(TextCommandContext ctx)
                 bool unlocks = await Tasks.LockdownTasks.CheckUnlocksAsync();
                 bool channelUpdateEvents = await Tasks.EventTasks.HandlePendingChannelUpdateEventsAsync();
                 bool channelDeleteEvents = await Tasks.EventTasks.HandlePendingChannelDeleteEventsAsync();
+                bool checkAndMassDehoist = await Tasks.MassDehoistTasks.CheckAndMassDehoistTask();
 
-                await msg.ModifyAsync($"Unban check result: `{bans}`\nUnmute check result: `{mutes}`\nPunishment message cleanup check result: `{punishmentMessages}`\nReminders check result: `{reminders}`\nRaidmode check result: `{raidmode}`\nUnlocks check result: `{unlocks}`\nPending Channel Update events check result: `{channelUpdateEvents}`\nPending Channel Delete events check result: `{channelDeleteEvents}`");
+                await msg.ModifyAsync($"Unban check result: `{bans}`\nUnmute check result: `{mutes}`\nPunishment message cleanup check result: `{punishmentMessages}`\nReminders check result: `{reminders}`\nRaidmode check result: `{raidmode}`\nUnlocks check result: `{unlocks}`\nPending Channel Update events check result: `{channelUpdateEvents}`\nPending Channel Delete events check result: `{channelDeleteEvents}`\nMass dehoist check result: `{checkAndMassDehoist}`");
             }
 
             [Command("sh")]

Commands/DehoistCmds.cs

@@ -139,18 +139,11 @@ public async Task MassDehoist(TextCommandContext ctx)
         {
             await ctx.RespondAsync($"{Program.cfgjson.Emoji.Loading} Working on it. This will take a while.");
             var msg = await ctx.GetResponseAsync();
-            var discordMembers = await ctx.Guild.GetAllMembersAsync().ToListAsync();
-            int failedCount = 0;
 
-            foreach (DiscordMember discordMember in discordMembers)
-            {
-                bool success = await DehoistHelpers.CheckAndDehoistMemberAsync(discordMember, ctx.User, true);
-                if (!success)
-                    failedCount++;
-            }
+            var (totalMembers, failedMembers) = await DehoistHelpers.MassDehoistAsync(ctx.Guild);
 
             _ = msg.DeleteAsync();
-            await ctx.Channel.SendMessageAsync(new DiscordMessageBuilder().WithContent($"{Program.cfgjson.Emoji.Success} Successfully dehoisted {discordMembers.Count() - failedCount} of {discordMembers.Count()} member(s)! (Check Audit Log for details)").WithReply(ctx.Message.Id, true, false));
+            await ctx.Channel.SendMessageAsync(new DiscordMessageBuilder().WithContent($"{Program.cfgjson.Emoji.Success} Successfully dehoisted {totalMembers - failedMembers} of {totalMembers} member(s)! (Check Audit Log for details)").WithReply(ctx.Message.Id, true, false));
         }
 
         [Command("massundehoisttextcmd")]

Helpers/DehoistHelpers.cs

@@ -165,5 +165,19 @@ await discordMember.ModifyAsync(a =>
                 return (success, isPermissionError, true);
             }
         }
+
+        public static async Task<(int totalMembers, int failedMembers)> MassDehoistAsync(DiscordGuild guild)
+        {
+            var discordMembers = await guild.GetAllMembersAsync().ToListAsync();
+            int failedCount = 0;
+
+            foreach (DiscordMember discordMember in discordMembers)
+            {
+                bool success = await CheckAndDehoistMemberAsync(discordMember, Program.discord.CurrentUser, true);
+                if (!success)
+                    failedCount++;
+            }
+            return (discordMembers.Count, failedCount);
+        }
     }
 }

Program.cs

@@ -263,6 +263,9 @@ static async Task Main(string[] _)
                         Tasks.EventTasks.HandlePendingChannelDeleteEventsAsync(),
                     ];
 
+                    // This one has its own time management, run it asynchronously and throw caution to the wind.
+                    Tasks.MassDehoistTasks.CheckAndMassDehoistTask();
+
                     // To prevent a future issue if checks take longer than 10 seconds,
                     // we only start the 10 second counter after all tasks have concluded.
                     await Task.WhenAll(taskList);

Tasks/MassDehoistTasks.cs

@@ -0,0 +1,45 @@
+namespace Cliptok.Tasks
+{
+    internal class MassDehoistTasks
+    {
+        public static async Task<bool> CheckAndMassDehoistTask()
+        {
+            var dbResult = await Program.db.StringGetAsync("lastMassDehoistRun");
+
+            if (dbResult.IsNull)
+            {
+                await MassDehoistAndUpdateTimeAsync();
+            }
+
+            var lastMassDehoistRun = DateTimeOffset.FromUnixTimeSeconds((int)dbResult);
+            var timeSinceLastRun = DateTime.Now - lastMassDehoistRun;
+
+            if (timeSinceLastRun > TimeSpan.FromHours(24))
+            {
+                await MassDehoistAndUpdateTimeAsync();
+                return true;
+            }
+            else
+            {
+                Program.discord.Logger.LogDebug("Mass dehoist task skipped. Last run was at {time} UTC", lastMassDehoistRun);
+                return false;
+            }
+        }
+
+        public static async Task<bool> MassDehoistAndUpdateTimeAsync()
+        {
+            await Program.db.StringSetAsync("lastMassDehoistRun", TimeHelpers.ToUnixTimestamp(DateTime.Now));
+            var (totalMembers, failedMembers) = await DehoistHelpers.MassDehoistAsync(Program.homeGuild);
+            if (totalMembers != failedMembers)
+            {
+                Program.discord.Logger.LogInformation("Successfully mass dehoisted {members} of {total} member(s)! Check Audit Log for details.", totalMembers - failedMembers, totalMembers);
+                return true;
+            }
+            else
+            {
+                Program.discord.Logger.LogDebug("Mass dehoist task completed with no members to dehoist.");
+                return false;
+            }
+        }
+    }
+}