using Boyfriend.Data; using Boyfriend.Services; using DiffPlex.DiffBuilder; using JetBrains.Annotations; using Remora.Discord.API.Abstractions.Gateway.Events; using Remora.Discord.API.Abstractions.Objects; using Remora.Discord.API.Abstractions.Rest; using Remora.Discord.Caching; using Remora.Discord.Caching.Services; using Remora.Discord.Extensions.Embeds; using Remora.Discord.Gateway.Responders; using Remora.Results; namespace Boyfriend.Responders; /// /// Handles logging the difference between an edited message's old and new content /// to a guild's if one is set. /// [UsedImplicitly] public class MessageEditedResponder : IResponder { private readonly CacheService _cacheService; private readonly IDiscordRestChannelAPI _channelApi; private readonly GuildDataService _dataService; private readonly IDiscordRestUserAPI _userApi; public MessageEditedResponder( CacheService cacheService, IDiscordRestChannelAPI channelApi, GuildDataService dataService, IDiscordRestUserAPI userApi) { _cacheService = cacheService; _channelApi = channelApi; _dataService = dataService; _userApi = userApi; } public async Task RespondAsync(IMessageUpdate gatewayEvent, CancellationToken ct = default) { if (!gatewayEvent.GuildID.IsDefined(out var guildId)) return Result.FromSuccess(); var cfg = await _dataService.GetSettings(guildId, ct); if (GuildSettings.PrivateFeedbackChannel.Get(cfg).Empty()) return Result.FromSuccess(); if (!gatewayEvent.Content.IsDefined(out var newContent)) return Result.FromSuccess(); if (!gatewayEvent.EditedTimestamp.IsDefined(out var timestamp)) return Result.FromSuccess(); // The message wasn't actually edited if (!gatewayEvent.ChannelID.IsDefined(out var channelId)) return Result.FromError(new ArgumentNullError(nameof(gatewayEvent.ChannelID))); if (!gatewayEvent.ID.IsDefined(out var messageId)) return Result.FromError(new ArgumentNullError(nameof(gatewayEvent.ID))); var cacheKey = new KeyHelpers.MessageCacheKey(channelId, messageId); var messageResult = await _cacheService.TryGetValueAsync( cacheKey, ct); if (!messageResult.IsDefined(out var message)) return Result.FromError(messageResult); if (message.Content == newContent) return Result.FromSuccess(); // Custom event responders are called earlier than responders responsible for message caching // This means that subsequent edit logs may contain the wrong content // We can work around this by evicting the message from the cache await _cacheService.EvictAsync(cacheKey, ct); // However, since we evicted the message, subsequent edits won't have a cached instance to work with // Getting the message will put it back in the cache, resolving all issues // We don't need to await this since the result is not needed // NOTE: Because this is not awaited, there may be a race condition depending on how fast clients are able to edit their messages // NOTE: Awaiting this might not even solve this if the same responder is called asynchronously _ = _channelApi.GetChannelMessageAsync(channelId, messageId, ct); var currentUserResult = await _userApi.GetCurrentUserAsync(ct); if (!currentUserResult.IsDefined(out var currentUser)) return Result.FromError(currentUserResult); var diff = InlineDiffBuilder.Diff(message.Content, newContent); Messages.Culture = GuildSettings.Language.Get(cfg); var embed = new EmbedBuilder() .WithSmallTitle(string.Format(Messages.CachedMessageEdited, message.Author.GetTag()), message.Author) .WithDescription($"https://discord.com/channels/{guildId}/{channelId}/{messageId}\n{diff.AsMarkdown()}") .WithUserFooter(currentUser) .WithTimestamp(timestamp.Value) .WithColour(ColorsList.Yellow) .Build(); if (!embed.IsDefined(out var built)) return Result.FromError(embed); return (Result)await _channelApi.CreateMessageAsync( GuildSettings.PrivateFeedbackChannel.Get(cfg), embeds: new[] { built }, allowedMentions: Boyfriend.NoMentions, ct: ct); } }