Restructured the project structure back to the way it was, there's no reasonable way to split the modules

2025-09-11 01:38:27 -04:00 · 2024-04-26 22:26:24 +00:00
parent 6c9c8bf63e
commit e0819f760c
768 changed files with 192 additions and 1047 deletions
--- a/src/NadekoBot/Modules/Permissions/Blacklist/BlacklistCommands.cs
+++ b/src/NadekoBot/Modules/Permissions/Blacklist/BlacklistCommands.cs
@@ -0,0 +1,143 @@
+#nullable disable
+using NadekoBot.Modules.Permissions.Services;
+using Nadeko.Bot.Db.Models;
+
+namespace NadekoBot.Modules.Permissions;
+
+public partial class Permissions
+{
+    [Group]
+    public partial class BlacklistCommands : NadekoModule<BlacklistService>
+    {
+        private readonly DiscordSocketClient _client;
+
+        public BlacklistCommands(DiscordSocketClient client)
+            => _client = client;
+
+        private async Task ListBlacklistInternal(string title, BlacklistType type, int page = 0)
+        {
+            if (page < 0)
+                throw new ArgumentOutOfRangeException(nameof(page));
+
+            var list = _service.GetBlacklist();
+            var items = await list.Where(x => x.Type == type)
+                                  .Select(async i =>
+                                  {
+                                      try
+                                      {
+                                          return i.Type switch
+                                          {
+                                              BlacklistType.Channel => Format.Code(i.ItemId.ToString())
+                                                                       + " "
+                                                                       + (_client.GetChannel(i.ItemId)?.ToString()
+                                                                          ?? ""),
+                                              BlacklistType.User => Format.Code(i.ItemId.ToString())
+                                                                    + " "
+                                                                    + ((await _client.Rest.GetUserAsync(i.ItemId))
+                                                                       ?.ToString()
+                                                                       ?? ""),
+                                              BlacklistType.Server => Format.Code(i.ItemId.ToString())
+                                                                      + " "
+                                                                      + (_client.GetGuild(i.ItemId)?.ToString() ?? ""),
+                                              _ => Format.Code(i.ItemId.ToString())
+                                          };
+                                      }
+                                      catch
+                                      {
+                                          Log.Warning("Can't get {BlacklistType} [{BlacklistItemId}]",
+                                              i.Type,
+                                              i.ItemId);
+                                          return Format.Code(i.ItemId.ToString());
+                                      }
+                                  })
+                                  .WhenAll();
+
+            await ctx.SendPaginatedConfirmAsync(page,
+                curPage =>
+                {
+                    var pageItems = items.Skip(10 * curPage).Take(10).ToList();
+
+                    if (pageItems.Count == 0)
+                        return _eb.Create().WithOkColor().WithTitle(title).WithDescription(GetText(strs.empty_page));
+
+                    return _eb.Create().WithTitle(title).WithDescription(pageItems.Join('\n')).WithOkColor();
+                },
+                items.Length,
+                10);
+        }
+
+        [Cmd]
+        [OwnerOnly]
+        public Task UserBlacklist(int page = 1)
+        {
+            if (--page < 0)
+                return Task.CompletedTask;
+
+            return ListBlacklistInternal(GetText(strs.blacklisted_users), BlacklistType.User, page);
+        }
+
+        [Cmd]
+        [OwnerOnly]
+        public Task ChannelBlacklist(int page = 1)
+        {
+            if (--page < 0)
+                return Task.CompletedTask;
+
+            return ListBlacklistInternal(GetText(strs.blacklisted_channels), BlacklistType.Channel, page);
+        }
+
+        [Cmd]
+        [OwnerOnly]
+        public Task ServerBlacklist(int page = 1)
+        {
+            if (--page < 0)
+                return Task.CompletedTask;
+
+            return ListBlacklistInternal(GetText(strs.blacklisted_servers), BlacklistType.Server, page);
+        }
+
+        [Cmd]
+        [OwnerOnly]
+        public Task UserBlacklist(AddRemove action, ulong id)
+            => Blacklist(action, id, BlacklistType.User);
+
+        [Cmd]
+        [OwnerOnly]
+        public Task UserBlacklist(AddRemove action, IUser usr)
+            => Blacklist(action, usr.Id, BlacklistType.User);
+
+        [Cmd]
+        [OwnerOnly]
+        public Task ChannelBlacklist(AddRemove action, ulong id)
+            => Blacklist(action, id, BlacklistType.Channel);
+
+        [Cmd]
+        [OwnerOnly]
+        public Task ServerBlacklist(AddRemove action, ulong id)
+            => Blacklist(action, id, BlacklistType.Server);
+
+        [Cmd]
+        [OwnerOnly]
+        public Task ServerBlacklist(AddRemove action, IGuild guild)
+            => Blacklist(action, guild.Id, BlacklistType.Server);
+
+        private async Task Blacklist(AddRemove action, ulong id, BlacklistType type)
+        {
+            if (action == AddRemove.Add)
+                await _service.Blacklist(type, id);
+            else
+                await _service.UnBlacklist(type, id);
+
+            if (action == AddRemove.Add)
+            {
+                await ReplyConfirmLocalizedAsync(strs.blacklisted(Format.Code(type.ToString()),
+                    Format.Code(id.ToString())));
+            }
+            else
+            {
+                await ReplyConfirmLocalizedAsync(strs.unblacklisted(Format.Code(type.ToString()),
+                    Format.Code(id.ToString())));
+            }
+        }
+    }
+}