using System.Threading.Tasks; using DSharpPlus; using DSharpPlus.Entities; using DSharpPlus.EventArgs; using DSharpPlus.Exceptions; using PluralKit.Core; namespace PluralKit.Bot { public class ReactionAdded: IEventHandler { private IDataStore _data; private EmbedService _embeds; public ReactionAdded(IDataStore data, EmbedService embeds) { _data = data; _embeds = embeds; } public async Task Handle(MessageReactionAddEventArgs evt) { await TryHandleProxyMessageReactions(evt); } private async ValueTask TryHandleProxyMessageReactions(MessageReactionAddEventArgs evt) { // Only proxies in guild text channels if (evt.Channel.Type != ChannelType.Text) return; // Sometimes we get events from users that aren't in the user cache // In that case we get a "broken" user object (where eg. calling IsBot throws an exception) // We just ignore all of those for now, should be quite rare... if (!evt.Client.TryGetCachedUser(evt.User.Id, out _)) return; // Ignore reactions from bots (we can't DM them anyway) if (evt.User.IsBot) return; FullMessage msg; switch (evt.Emoji.Name) { // Message deletion case "\u274C": // Red X if ((msg = await _data.GetMessage(evt.Message.Id)) != null) await HandleDeleteReaction(evt, msg); break; case "\u2753": // Red question mark case "\u2754": // White question mark if ((msg = await _data.GetMessage(evt.Message.Id)) != null) await HandleQueryReaction(evt, msg); break; case "\U0001F514": // Bell case "\U0001F6CE": // Bellhop bell case "\U0001F3D3": // Ping pong paddle (lol) case "\u23F0": // Alarm clock case "\u2757": // Exclamation mark if ((msg = await _data.GetMessage(evt.Message.Id)) != null) await HandlePingReaction(evt, msg); break; } } private async ValueTask HandleDeleteReaction(MessageReactionAddEventArgs evt, FullMessage msg) { if (!evt.Channel.BotHasAllPermissions(Permissions.ManageMessages)) return; // Can only delete your own message if (msg.Message.Sender != evt.User.Id) return; try { await evt.Message.DeleteAsync(); } catch (NotFoundException) { // Message was deleted by something/someone else before we got to it } await _data.DeleteMessage(evt.Message.Id); } private async ValueTask HandleQueryReaction(MessageReactionAddEventArgs evt, FullMessage msg) { // Try to DM the user info about the message var member = await evt.Guild.GetMemberAsync(evt.User.Id); try { await member.SendMessageAsync(embed: await _embeds.CreateMemberEmbed(msg.System, msg.Member, evt.Guild, LookupContext.ByNonOwner)); await member.SendMessageAsync(embed: await _embeds.CreateMessageInfoEmbed(evt.Client, msg)); } catch (UnauthorizedException) { } // No permissions to DM, can't check for this :( // And finally remove the original reaction (if we can) if (evt.Channel.BotHasAllPermissions(Permissions.ManageMessages)) await evt.Message.DeleteReactionAsync(evt.Emoji, evt.User); } private async ValueTask HandlePingReaction(MessageReactionAddEventArgs evt, FullMessage msg) { if (!evt.Channel.BotHasAllPermissions(Permissions.SendMessages)) return; // Check if the "pinger" has permission to send messages in this channel // (if not, PK shouldn't send messages on their behalf) var guildUser = await evt.Guild.GetMemberAsync(evt.User.Id); var requiredPerms = Permissions.AccessChannels | Permissions.SendMessages; if ((guildUser.PermissionsIn(evt.Channel) & requiredPerms) != requiredPerms) return; if (msg.System.PingsEnabled) { // If the system has pings enabled, go ahead var embed = new DiscordEmbedBuilder().WithDescription($"[Jump to pinged message]({evt.Message.JumpLink})"); await evt.Channel.SendMessageFixedAsync($"Psst, **{msg.Member.DisplayName()}** (<@{msg.Message.Sender}>), you have been pinged by <@{evt.User.Id}>.", embed: embed.Build(), new IMention[] {new UserMention(msg.Message.Sender) }); } else { // If not, tell them in DMs (if we can) try { await guildUser.SendMessageFixedAsync($"{Emojis.Error} {msg.Member.DisplayName()}'s system has disabled reaction pings. If you want to mention them anyway, you can copy/paste the following message:"); await guildUser.SendMessageFixedAsync($"`<@{msg.Message.Sender}>`"); } catch (UnauthorizedException) { } } // Finally, remove the original reaction (if we can) if (evt.Channel.BotHasAllPermissions(Permissions.ManageMessages)) await evt.Message.DeleteReactionAsync(evt.Emoji, evt.User); } } }