87 lines
3.6 KiB
C#
87 lines
3.6 KiB
C#
using System;
|
|
using System.Threading.Tasks;
|
|
|
|
using App.Metrics;
|
|
|
|
using Myriad.Cache;
|
|
using Myriad.Extensions;
|
|
using Myriad.Gateway;
|
|
using Myriad.Rest;
|
|
using Myriad.Types;
|
|
|
|
using PluralKit.Core;
|
|
|
|
|
|
namespace PluralKit.Bot
|
|
{
|
|
public class MessageEdited: IEventHandler<MessageUpdateEvent>
|
|
{
|
|
private readonly LastMessageCacheService _lastMessageCache;
|
|
private readonly ProxyService _proxy;
|
|
private readonly IDatabase _db;
|
|
private readonly ModelRepository _repo;
|
|
private readonly IMetrics _metrics;
|
|
private readonly Cluster _client;
|
|
private readonly IDiscordCache _cache;
|
|
private readonly Bot _bot;
|
|
private readonly DiscordApiClient _rest;
|
|
|
|
|
|
public MessageEdited(LastMessageCacheService lastMessageCache, ProxyService proxy, IDatabase db, IMetrics metrics, ModelRepository repo, Cluster client, IDiscordCache cache, Bot bot, DiscordApiClient rest)
|
|
{
|
|
_lastMessageCache = lastMessageCache;
|
|
_proxy = proxy;
|
|
_db = db;
|
|
_metrics = metrics;
|
|
_repo = repo;
|
|
_client = client;
|
|
_cache = cache;
|
|
_bot = bot;
|
|
_rest = rest;
|
|
}
|
|
|
|
public async Task Handle(Shard shard, MessageUpdateEvent evt)
|
|
{
|
|
if (evt.Author.Value?.Id == _client.User?.Id) return;
|
|
|
|
// Edit message events sometimes arrive with missing data; double-check it's all there
|
|
if (!evt.Content.HasValue || !evt.Author.HasValue || !evt.Member.HasValue)
|
|
return;
|
|
|
|
var channel = _cache.GetChannel(evt.ChannelId);
|
|
if (channel.Type != Channel.ChannelType.GuildText)
|
|
return;
|
|
var guild = _cache.GetGuild(channel.GuildId!.Value);
|
|
var lastMessage = _lastMessageCache.GetLastMessage(evt.ChannelId);
|
|
|
|
// Only react to the last message in the channel
|
|
if (lastMessage?.mid != evt.Id)
|
|
return;
|
|
|
|
// Just run the normal message handling code, with a flag to disable autoproxying
|
|
MessageContext ctx;
|
|
await using (var conn = await _db.Obtain())
|
|
using (_metrics.Measure.Timer.Time(BotMetrics.MessageContextQueryTime))
|
|
ctx = await _repo.GetMessageContext(conn, evt.Author.Value!.Id, channel.GuildId!.Value, evt.ChannelId);
|
|
|
|
Message referencedMessage = (lastMessage.referenced_message != null) ? await _rest.GetMessage(evt.ChannelId, lastMessage.referenced_message.Value) : null;
|
|
|
|
// 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 = (lastMessage.referenced_message != null) ? new (channel.GuildId, evt.ChannelId, lastMessage.referenced_message.Value) : null,
|
|
ReferencedMessage = referencedMessage,
|
|
Type = (lastMessage.referenced_message != null) ? Message.MessageType.Reply : Message.MessageType.Default,
|
|
};
|
|
var botPermissions = _bot.PermissionsIn(channel.Id);
|
|
await _proxy.HandleIncomingMessage(shard, equivalentEvt, ctx, allowAutoproxy: false, guild: guild, channel: channel, botPermissions: botPermissions);
|
|
}
|
|
}
|
|
} |