.btr and .sclr added, cleanup

2025-09-12 02:08:27 -04:00 · 2024-11-20 17:14:48 +00:00
parent 3178074828
commit 796086538a
119 changed files with 9154 additions and 646 deletions
--- a/src/NadekoBot/Modules/Administration/Role/ButtonRolesService.cs
+++ b/src/NadekoBot/Modules/Administration/Role/ButtonRolesService.cs
@@ -0,0 +1,154 @@
+using LinqToDB;
+using LinqToDB.EntityFrameworkCore;
+using LinqToDB.SqlQuery;
+using NadekoBot.Common.ModuleBehaviors;
+using NadekoBot.Db.Models;
+using NCalc;
+
+namespace NadekoBot.Modules.Administration.Services;
+
+public sealed class ButtonRolesService : INService, IReadyExecutor
+{
+    private const string BTN_PREFIX = "n:btnrole:";
+
+    private readonly DbService _db;
+    private readonly DiscordSocketClient _client;
+    private readonly IBotCreds _creds;
+
+    public ButtonRolesService(IBotCreds creds, DiscordSocketClient client, DbService db)
+    {
+        _creds = creds;
+        _client = client;
+        _db = db;
+    }
+
+
+    public Task OnReadyAsync()
+    {
+        _client.InteractionCreated += OnInteraction;
+
+        return Task.CompletedTask;
+    }
+
+    private async Task OnInteraction(SocketInteraction inter)
+    {
+        if (inter is not SocketMessageComponent smc)
+            return;
+
+        if (!smc.Data.CustomId.StartsWith(BTN_PREFIX))
+            return;
+
+        await inter.DeferAsync();
+
+        _ = Task.Run(async () =>
+        {
+            await using var uow = _db.GetDbContext();
+            var buttonRole = await uow.GetTable<ButtonRole>()
+                                      .Where(x => x.ButtonId == smc.Data.CustomId && x.MessageId == smc.Message.Id)
+                                      .FirstOrDefaultAsyncLinqToDB();
+
+            if (buttonRole is null)
+                return;
+
+            var guild = _client.GetGuild(buttonRole.GuildId);
+            if (guild is null)
+                return;
+
+            var role = guild.GetRole(buttonRole.RoleId);
+            if (role is null)
+                return;
+
+            if (smc.User is not IGuildUser user)
+                return;
+
+            if (user.GetRoles().Any(x => x.Id == role.Id))
+            {
+                await user.RemoveRoleAsync(role.Id);
+                return;
+            }
+
+            await user.AddRoleAsync(role.Id);
+        });
+    }
+
+    public async Task<bool> AddButtonRole(
+        ulong guildId,
+        ulong channelId,
+        ulong roleId,
+        ulong messageId,
+        IEmote emote
+    )
+    {
+        await using var uow = _db.GetDbContext();
+
+        // up to 25 per message
+        if (await uow.GetTable<ButtonRole>()
+                     .Where(x => x.MessageId == messageId)
+                     .CountAsyncLinqToDB()
+            >= 25)
+            return false;
+
+
+        var emoteStr = emote.ToString()!;
+        var guid = Guid.NewGuid();
+        await uow.GetTable<ButtonRole>()
+                 .InsertOrUpdateAsync(() => new ButtonRole()
+                     {
+                         GuildId = guildId,
+                         ChannelId = channelId,
+                         RoleId = roleId,
+                         MessageId = messageId,
+                         Position =
+                             uow
+                                 .GetTable<ButtonRole>()
+                                 .Any(x => x.MessageId == messageId)
+                                 ? uow.GetTable<ButtonRole>()
+                                      .Where(x => x.MessageId == messageId)
+                                      .Max(x => x.Position)
+                                 : 1,
+                         Emote = emoteStr,
+                         Label = string.Empty,
+                         ButtonId = $"{BTN_PREFIX}:{guildId}:{guid}"
+                     },
+                     _ => new()
+                     {
+                         Emote = emoteStr,
+                         Label = string.Empty,
+                         ButtonId = $"{BTN_PREFIX}:{guildId}:{guid}"
+                     },
+                     () => new()
+                     {
+                         RoleId = roleId,
+                         MessageId = messageId,
+                     });
+
+        return true;
+    }
+
+    public async Task<IReadOnlyList<ButtonRole>> RemoveButtonRoles(ulong guildId, ulong messageId)
+    {
+        await using var uow = _db.GetDbContext();
+        return await uow.GetTable<ButtonRole>()
+                        .Where(x => x.GuildId == guildId && x.MessageId == messageId)
+                        .DeleteWithOutputAsync();
+    }
+
+    public async Task<ButtonRole?> RemoveButtonRole(ulong guildId, ulong messageId, ulong roleId)
+    {
+        await using var uow = _db.GetDbContext();
+        var deleted = await uow.GetTable<ButtonRole>()
+                               .Where(x => x.GuildId == guildId && x.MessageId == messageId && x.RoleId == roleId)
+                               .DeleteWithOutputAsync();
+
+        return deleted.FirstOrDefault();
+    }
+
+    public async Task<IReadOnlyList<ButtonRole>> GetButtonRoles(ulong guildId, ulong? messageId)
+    {
+        await using var uow = _db.GetDbContext();
+        return await uow.GetTable<ButtonRole>()
+                        .Where(x => x.GuildId == guildId && (messageId == null || x.MessageId == messageId))
+                        .OrderBy(x => x.Id)
+                        .ToListAsyncLinqToDB();
+    }
+}