PluralKit/PluralKit.Bot/Handlers/MessageEdited.cs

126 lines
4.9 KiB
C#
Raw Normal View History

2021-01-31 13:42:28 +00:00
using System;
using System.Threading.Tasks;
2020-06-14 20:19:12 +00:00
using App.Metrics;
2021-01-31 13:42:28 +00:00
using Myriad.Cache;
using Myriad.Extensions;
2020-12-22 12:15:26 +00:00
using Myriad.Gateway;
using Myriad.Rest;
2021-01-31 13:42:28 +00:00
using Myriad.Types;
using PluralKit.Core;
using Serilog;
namespace PluralKit.Bot
{
2020-12-22 12:15:26 +00:00
public class MessageEdited: IEventHandler<MessageUpdateEvent>
{
private readonly LastMessageCacheService _lastMessageCache;
private readonly ProxyService _proxy;
2020-06-13 17:36:43 +00:00
private readonly IDatabase _db;
2020-08-29 11:46:27 +00:00
private readonly ModelRepository _repo;
2020-06-14 20:19:12 +00:00
private readonly IMetrics _metrics;
2021-01-31 13:42:28 +00:00
private readonly Cluster _client;
private readonly IDiscordCache _cache;
private readonly Bot _bot;
private readonly DiscordApiClient _rest;
private readonly ILogger _logger;
2021-08-27 15:03:47 +00:00
public MessageEdited(LastMessageCacheService lastMessageCache, ProxyService proxy, IDatabase db, IMetrics metrics, ModelRepository repo, Cluster client, IDiscordCache cache, Bot bot, DiscordApiClient rest, ILogger logger)
{
_lastMessageCache = lastMessageCache;
_proxy = proxy;
_db = db;
2020-06-14 20:19:12 +00:00
_metrics = metrics;
2020-08-29 11:46:27 +00:00
_repo = repo;
2020-09-20 19:49:52 +00:00
_client = client;
2021-01-31 13:42:28 +00:00
_cache = cache;
_bot = bot;
_rest = rest;
_logger = logger.ForContext<MessageEdited>();
}
2020-12-22 12:15:26 +00:00
public async Task Handle(Shard shard, MessageUpdateEvent evt)
{
if (evt.Author.Value?.Id == await _cache.GetOwnUser()) return;
2021-08-27 15:03:47 +00:00
2021-01-31 13:42:28 +00:00
// Edit message events sometimes arrive with missing data; double-check it's all there
2021-08-27 15:03:47 +00:00
if (!evt.Content.HasValue || !evt.Author.HasValue || !evt.Member.HasValue)
2021-01-31 13:42:28 +00:00
return;
2021-08-27 15:03:47 +00:00
var channel = await _cache.GetChannel(evt.ChannelId);
2021-07-08 16:45:59 +00:00
if (!DiscordUtils.IsValidGuildChannel(channel))
2021-01-31 13:42:28 +00:00
return;
var guild = await _cache.GetGuild(channel.GuildId!.Value);
var lastMessage = _lastMessageCache.GetLastMessage(evt.ChannelId)?.Current;
2021-01-31 13:42:28 +00:00
// Only react to the last message in the channel
if (lastMessage?.Id != evt.Id)
2021-01-31 13:42:28 +00:00
return;
2021-08-27 15:03:47 +00:00
2021-01-31 13:42:28 +00:00
// Just run the normal message handling code, with a flag to disable autoproxying
MessageContext ctx;
using (_metrics.Measure.Timer.Time(BotMetrics.MessageContextQueryTime))
ctx = await _repo.GetMessageContext(evt.Author.Value!.Id, channel.GuildId!.Value, evt.ChannelId);
2021-01-31 13:42:28 +00:00
var equivalentEvt = await GetMessageCreateEvent(evt, lastMessage, channel);
var botPermissions = await _cache.PermissionsIn(channel.Id);
try
{
await _proxy.HandleIncomingMessage(shard, equivalentEvt, ctx, allowAutoproxy: false, guild: guild,
channel: channel, botPermissions: botPermissions);
}
// Catch any failed proxy checks so they get ignored in the global error handler
2021-08-27 15:03:47 +00:00
catch (ProxyService.ProxyChecksFailedException) { }
}
2021-03-22 07:17:10 +00:00
private async Task<MessageCreateEvent> GetMessageCreateEvent(MessageUpdateEvent evt, CachedMessage lastMessage, Channel channel)
{
var referencedMessage = await GetReferencedMessage(evt.ChannelId, lastMessage.ReferencedMessage);
var messageReference = lastMessage.ReferencedMessage != null
? new Message.Reference(channel.GuildId, evt.ChannelId, lastMessage.ReferencedMessage.Value)
: null;
2021-08-27 15:03:47 +00:00
var messageType = lastMessage.ReferencedMessage != null
? Message.MessageType.Reply
: Message.MessageType.Default;
2021-08-27 15:03:47 +00:00
2021-01-31 13:42:28 +00:00
// TODO: is this missing anything?
var equivalentEvt = new MessageCreateEvent
{
Id = evt.Id,
ChannelId = evt.ChannelId,
GuildId = channel.GuildId,
Author = evt.Author.Value,
Member = evt.Member.Value,
Content = evt.Content.Value,
Attachments = evt.Attachments.Value ?? Array.Empty<Message.Attachment>(),
MessageReference = messageReference,
2021-03-22 07:17:10 +00:00
ReferencedMessage = referencedMessage,
Type = messageType,
2021-01-31 13:42:28 +00:00
};
return equivalentEvt;
}
private async Task<Message?> GetReferencedMessage(ulong channelId, ulong? referencedMessageId)
{
if (referencedMessageId == null)
return null;
2021-08-27 15:03:47 +00:00
var botPermissions = await _cache.PermissionsIn(channelId);
if (!botPermissions.HasFlag(PermissionSet.ReadMessageHistory))
{
_logger.Warning("Tried to get referenced message in channel {ChannelId} to reply but bot does not have Read Message History",
channelId);
return null;
}
return await _rest.GetMessage(channelId, referencedMessageId.Value);
}
}
}