import { reactionRolesCmd, TReactionRolePair } from "../types"; import { commandTypeHelpers as ct } from "../../../commandTypes"; import { sendErrorMessage, sendSuccessMessage } from "src/pluginUtils"; import { TextChannel } from "eris"; import { RecoverablePluginError, ERRORS } from "src/RecoverablePluginError"; import { canUseEmoji } from "src/utils"; import { applyReactionRoleReactionsToMessage } from "../util/applyReactionRoleReactionsToMessage"; const CLEAR_ROLES_EMOJI = "❌"; export const InitReactionRolesCmd = reactionRolesCmd({ trigger: "reaction_roles", permission: "can_manage", signature: { messageId: ct.string(), reactionRolePairs: ct.string({ catchAll: true }), exclusive: ct.bool({ option: true, isSwitch: true, shortcut: "e" }), }, async run({ message: msg, args, pluginData }) { const savedMessage = await pluginData.state.savedMessages.find(args.messageId); if (!savedMessage) { sendErrorMessage(pluginData, msg.channel, "Unknown message"); return; } const channel = (await pluginData.guild.channels.get(savedMessage.channel_id)) as TextChannel; if (!channel || !(channel instanceof TextChannel)) { sendErrorMessage(pluginData, msg.channel, "Channel no longer exists"); return; } const targetMessage = await channel.getMessage(args.messageId); if (!targetMessage) { sendErrorMessage(pluginData, msg.channel, "Unknown message (2)"); return; } // Clear old reaction roles for the message from the DB await pluginData.state.reactionRoles.removeFromMessage(targetMessage.id); // Turn "emoji = role" pairs into an array of tuples of the form [emoji, roleId] // Emoji is either a unicode emoji or the snowflake of a custom emoji const emojiRolePairs: TReactionRolePair[] = args.reactionRolePairs .trim() .split("\n") .map(v => v.split(/(\s|[=,])+/).map(v => v.trim())) // tslint:disable-line .map( (pair): TReactionRolePair => { const customEmojiMatch = pair[0].match(/^$/); if (customEmojiMatch) { return [customEmojiMatch[2], pair[1], customEmojiMatch[1]]; } else { return pair as TReactionRolePair; } }, ); // Verify the specified emojis and roles are valid and usable for (const pair of emojiRolePairs) { if (pair[0] === CLEAR_ROLES_EMOJI) { sendErrorMessage( pluginData, msg.channel, `The emoji for clearing roles (${CLEAR_ROLES_EMOJI}) is reserved and cannot be used`, ); return; } try { if (!canUseEmoji(pluginData.client, pair[0])) { sendErrorMessage( pluginData, msg.channel, "I can only use regular emojis and custom emojis from servers I'm on", ); return; } } catch (e) { if (e instanceof RecoverablePluginError && e.code === ERRORS.INVALID_EMOJI) { sendErrorMessage(pluginData, msg.channel, `Invalid emoji: ${pair[0]}`); return; } throw e; } if (!pluginData.guild.roles.has(pair[1])) { sendErrorMessage(pluginData, msg.channel, `Unknown role ${pair[1]}`); return; } } // Save the new reaction roles to the database for (const pair of emojiRolePairs) { await pluginData.state.reactionRoles.add(channel.id, targetMessage.id, pair[0], pair[1], args.exclusive); } // Apply the reactions themselves const reactionRoles = await pluginData.state.reactionRoles.getForMessage(targetMessage.id); await applyReactionRoleReactionsToMessage(pluginData, targetMessage.channel.id, targetMessage.id, reactionRoles); sendSuccessMessage(pluginData, msg.channel, "Reaction roles added"); }, });