From 8231c57bdf6b1c0b7c0894dd87ce55efd3977822 Mon Sep 17 00:00:00 2001 From: sam Date: Fri, 16 Aug 2024 00:51:19 +0200 Subject: [PATCH] feat: add message update handler --- .../Bot/Responders/MessageUpdateResponder.cs | 125 ++++++++++++++++++ .../Database/Queries/MessageRepository.cs | 31 +++++ .../Extensions/DiscordExtensions.cs | 3 + 3 files changed, 159 insertions(+) create mode 100644 Catalogger.Backend/Bot/Responders/MessageUpdateResponder.cs diff --git a/Catalogger.Backend/Bot/Responders/MessageUpdateResponder.cs b/Catalogger.Backend/Bot/Responders/MessageUpdateResponder.cs new file mode 100644 index 0000000..9e99e88 --- /dev/null +++ b/Catalogger.Backend/Bot/Responders/MessageUpdateResponder.cs @@ -0,0 +1,125 @@ +using Catalogger.Backend.Cache; +using Catalogger.Backend.Database; +using Catalogger.Backend.Database.Queries; +using Catalogger.Backend.Extensions; +using Catalogger.Backend.Services; +using Remora.Discord.API; +using Remora.Discord.API.Abstractions.Gateway.Events; +using Remora.Discord.API.Abstractions.Objects; +using Remora.Discord.API.Gateway.Events; +using Remora.Discord.API.Objects; +using Remora.Discord.Extensions.Embeds; +using Remora.Discord.Gateway.Responders; +using Remora.Results; + +namespace Catalogger.Backend.Bot.Responders; + +public class MessageUpdateResponder( + ILogger logger, + DatabaseContext db, + ChannelCacheService channelCache, + UserCacheService userCache, + MessageRepository messageRepository, + WebhookExecutorService webhookExecutor) : IResponder +{ + private readonly ILogger _logger = logger.ForContext(); + + public async Task RespondAsync(IMessageUpdate evt, CancellationToken ct = default) + { + // Discord only *very* recently changed message update events to have all fields, + // so we convert the event to a MessageCreate to avoid having to unwrap every single field + var msg = ConvertToMessageCreate(evt); + + if (!msg.GuildID.IsDefined()) + { + _logger.Debug("Received message create event for message {MessageId} despite it not being in a guild", + msg.ID); + return Result.Success; + } + + _logger.Debug("Guild is {GuildId}", msg.GuildID.Value); + var guildConfig = await db.GetGuildAsync(msg.GuildID.Value, ct); + + if (await messageRepository.IsMessageIgnoredAsync(msg.ID.Value, ct)) + { + _logger.Debug("Message {MessageId} should be ignored", msg.ID); + return Result.Success; + } + + var logChannel = webhookExecutor.GetLogChannel(guildConfig, LogChannelType.MessageUpdate, msg.ChannelID, + msg.Author.ID.Value); + if (logChannel == null) return Result.Success; + + try + { + var oldMessage = await messageRepository.GetMessageAsync(msg.ID.Value, ct); + if (oldMessage == null) + { + logger.Debug("Message {Id} was edited and should be logged but is not in the database", msg.ID); + return Result.Success; + } + + if (oldMessage.Content == msg.Content) return Result.Success; + + var user = msg.Author; + if (msg.Author.ID != oldMessage.UserId) + { + var systemAccount = await userCache.GetUserAsync(DiscordSnowflake.New(oldMessage.UserId)); + if (systemAccount != null) user = systemAccount; + } + + var embedBuilder = new EmbedBuilder() + .WithAuthor(user.Tag(), null, user.AvatarUrl()) + .WithTitle("Message edited") + .WithDescription(oldMessage.Content) + .WithColour(DiscordUtils.Purple) + .WithFooter($"ID: {msg.ID}") + .WithTimestamp(msg.ID.Timestamp); + + var fields = Enumerable.Range(0, msg.Content.Length / 1000) + .Select(i => msg.Content.Substring(i * 1000, 1000)) + .Select((s, i) => + new EmbedField($"New content{(i != 0 ? " (cont.)" : "")}", s, false)) + .ToList(); + embedBuilder.SetFields(fields); + + string channelMention; + if (!channelCache.TryGet(msg.ChannelID, out var channel)) + channelMention = $"<#{msg.ChannelID}>"; + else if (channel.Type is ChannelType.AnnouncementThread or ChannelType.PrivateThread + or ChannelType.PublicThread) + channelMention = + $"<#{channel.ParentID.Value}>\nID: {channel.ParentID.Value}\n\nThread: {channel.Name} (<#{channel.ID}>)"; + else channelMention = $"<#{channel.ID}>\nID: {channel.ID}"; + + embedBuilder.AddField("Channel", channelMention, true); + embedBuilder.AddField("Sender", $"<@{user.ID}>\n{user.Tag()}\nID: {user.ID}", true); + + if (oldMessage is { System: not null, Member: not null }) + { + embedBuilder.WithTitle($"Message by {msg.Author.Username} edited"); + embedBuilder.AddField("\u200b", "**PluralKit information**", false); + embedBuilder.AddField("System ID", oldMessage.System, true); + embedBuilder.AddField("Member ID", oldMessage.Member, true); + } + + embedBuilder.AddField("Link", $"https://discord.com/channels/{msg.GuildID}/{msg.ChannelID}/{msg.ID}"); + + await webhookExecutor.QueueLogAsync(logChannel.Value, embedBuilder.Build().GetOrThrow()); + return Result.Success; + } + finally + { + await messageRepository.UpdateMessageAsync(msg, ct); + } + } + + private static MessageCreate ConvertToMessageCreate(IMessageUpdate evt) => new(evt.GuildID, evt.Member, + evt.Mentions.GetOrThrow(), evt.ID.GetOrThrow(), evt.ChannelID.GetOrThrow(), evt.Author.GetOrThrow(), + evt.Content.GetOrThrow(), evt.Timestamp.GetOrThrow(), evt.EditedTimestamp.GetOrThrow(), IsTTS: false, + evt.MentionsEveryone.GetOrThrow(), evt.MentionedRoles.GetOrThrow(), evt.MentionedChannels, + evt.Attachments.GetOrThrow(), evt.Embeds.GetOrThrow(), evt.Reactions, evt.Nonce, evt.IsPinned.GetOrThrow(), + evt.WebhookID, evt.Type.GetOrThrow(), evt.Activity, evt.Application, evt.ApplicationID, evt.MessageReference, + evt.Flags, evt.ReferencedMessage, evt.Interaction, evt.Thread, evt.Components, evt.StickerItems, evt.Position, + evt.Resolved, evt.InteractionMetadata, evt.Poll); +} \ No newline at end of file diff --git a/Catalogger.Backend/Database/Queries/MessageRepository.cs b/Catalogger.Backend/Database/Queries/MessageRepository.cs index c090674..1d94a0c 100644 --- a/Catalogger.Backend/Database/Queries/MessageRepository.cs +++ b/Catalogger.Backend/Database/Queries/MessageRepository.cs @@ -36,6 +36,37 @@ public class MessageRepository(ILogger logger, DatabaseContext db, IEncryptionSe await db.SaveChangesAsync(ct); } + public async Task UpdateMessageAsync(IMessageCreate msg, CancellationToken ct = default) + { + _logger.Debug("Updating message {MessageId}", msg.ID); + + var tx = await db.Database.BeginTransactionAsync(ct); + var (isStored, _) = await HasProxyInfoAsync(msg.ID.Value); + if (!isStored) + { + await SaveMessageAsync(msg, ct); + } + else + { + var metadata = new Metadata(IsWebhook: msg.WebhookID.HasValue, + msg.Attachments.Select(a => new Attachment(a.Filename, a.Size, a.ContentType.Value))); + + var dbMsg = await db.Messages.FindAsync(msg.ID.Value); + if (dbMsg == null) throw new CataloggerError("Message was null despite HasProxyInfoAsync returning true"); + + dbMsg.EncryptedContent = await Task.Run( + () => encryptionService.Encrypt(string.IsNullOrWhiteSpace(msg.Content) ? "None" : msg.Content), ct); + dbMsg.EncryptedUsername = await Task.Run(() => encryptionService.Encrypt(msg.Author.Tag()), ct); + dbMsg.EncryptedMetadata = + await Task.Run(() => encryptionService.Encrypt(JsonSerializer.Serialize(metadata)), ct); + + db.Update(dbMsg); + await db.SaveChangesAsync(ct); + } + + await tx.CommitAsync(ct); + } + public async Task GetMessageAsync(ulong id, CancellationToken ct = default) { _logger.Debug("Retrieving message {MessageId}", id); diff --git a/Catalogger.Backend/Extensions/DiscordExtensions.cs b/Catalogger.Backend/Extensions/DiscordExtensions.cs index 8caab54..4e96c1d 100644 --- a/Catalogger.Backend/Extensions/DiscordExtensions.cs +++ b/Catalogger.Backend/Extensions/DiscordExtensions.cs @@ -50,6 +50,9 @@ public static class DiscordExtensions return result.Entity; } + public static T GetOrThrow(this Optional optional) => + optional.OrThrow(() => new CataloggerError("Optional was unset")); + public static async Task GetOrThrow(this Task> result) => (await result).GetOrThrow(); public static async Task UpdateMessageAsync(this IDiscordRestInteractionAPI interactionApi,