add manual check command

SylveonDeko · Jul 31, 2023 · 6534d0a · 6534d0a
1 parent 149104d
commit 6534d0a
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/Mewdeko/Modules/Permissions/BlacklistCommands.cs b/src/Mewdeko/Modules/Permissions/BlacklistCommands.cs
@@ -29,6 +29,13 @@ public class BlacklistCommands : MewdekoSubmodule<BlacklistService>
         [Cmd, Aliases]
         public Task ServerBlacklist(AddRemove action, IGuild guild, [Remainder] string? reason) => Blacklist(action, guild.Id, BlacklistType.Server, reason);
 
+        [Cmd, Aliases]
+        public async Task ManualBlacklistCheck()
+        {
+            await ctx.Channel.SendConfirmAsync("Sending manual check...");
+            await Service.SendManualCheck();
+        }
+
         private async Task Blacklist(AddRemove action, ulong id, BlacklistType type, string? reason)
         {
             switch (action)

diff --git a/src/Mewdeko/Modules/Permissions/Services/BlacklistService.cs b/src/Mewdeko/Modules/Permissions/Services/BlacklistService.cs
@@ -13,15 +13,16 @@ public sealed class BlacklistService : IEarlyBehavior, INService
 
     private readonly TypedKey<BlacklistEntry[]> blPubKey = new("blacklist.reload");
     public IList<BlacklistEntry> BlacklistEntries;
+    private readonly TypedKey<bool> blPrivKey = new("blacklist.reload.priv");
 
     public BlacklistService(DbService db, IPubSub pubSub, EventHandler handler, DiscordSocketClient client)
     {
         this.db = db;
         this.pubSub = pubSub;
         this.client = client;
-
         Reload(false);
         this.pubSub.Sub(blPubKey, OnReload);
+        this.pubSub.Sub(blPrivKey, ManualCheck);
         handler.JoinedGuild += CheckBlacklist;
         client.Ready += CheckAllGuilds;
         _ = CheckAllGuilds();
@@ -39,6 +40,12 @@ public BlacklistService(DbService db, IPubSub pubSub, EventHandler handler, Disc
         });
     }
 
+    private ValueTask ManualCheck(bool _)
+    {
+        CheckAllGuilds();
+        return default;
+    }
+
     private Task CheckAllGuilds()
     {
         _ = Task.Run(async () =>
@@ -59,6 +66,12 @@ private Task CheckAllGuilds()
         return Task.CompletedTask;
     }
 
+    public Task SendManualCheck()
+    {
+        this.pubSub.Pub(blPrivKey, true);
+        return Task.CompletedTask;
+    }
+
     private async Task CheckBlacklist(IGuild arg)
     {
         var channels = await arg.GetTextChannelsAsync();