vylbot-app/commands/poll.js

// Required components
const { command } = require('vylbot-core');
const { MessageEmbed } = require('discord.js');
const emojiRegex = require('emoji-regex/RGI_Emoji');

// Command variables
const embedColor = "0x3050ba";

// Command class
class poll extends command {
    constructor() {
        // Set the command's run method, description, category, and example usage
        super("poll");
        super.description = "Generates a poll with reaction numbers";
        super.category = "General";
        super.usage = "<title>;<option 1>;<option 2>...";
    }

    // Run method
    poll(context) {
        // Get the command's arguments, and split them by a semicolon rather than a space
        // This allows the variables to be able to use spaces in them
        let args = context.arguments;
        const argsJoined = args.join(' ');
        args = argsJoined.split(';');

        // If the argument has 3 or more arguments and less than 11 arguments
        // This allows the title and 2-9 options
        if (args.length >= 3 && args.length < 11) {
            // Set the title to the first argument
            const title = args[0];
            let optionString = "";

            // Array used to get the numbers as their words
            // arrayOfNumbers[n] = "n written in full words"
            const arrayOfNumbers = [
                ':zero:',
                ':one:',
                ':two:',
                ':three:',
                ':four:',
                ':five:',
                ':six:',
                ':seven:',
                ':eight:',
                ':nine:'
            ];

            // Array containing the numbers as their emoji
            const reactionEmojis = ["0️⃣", "1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6️⃣", "7️⃣", "8️⃣", "9️⃣"];

            // Loop through all the arguments after the title
            // Add them to the optionString, with their index turned into a number emoji
            // Example: :one: Option 1
            for (let i = 1; i < args.length; i++) {
                // If the option contains an emoji, replace the emoji with it
                const regex = emojiRegex();
                const match = regex.exec(args[i]);

                if (match) {
                    const emoji = match[0];
                    reactionEmojis[i] = emoji;
                    arrayOfNumbers[i] = '';
                }

                optionString += `${arrayOfNumbers[i]} ${args[i]}\n`;
            }

            // Create the embed with the title at the top of the description with the options below
            const embed = new MessageEmbed()
                .setColor(embedColor)
                .setDescription(`**${title}**\n\n${optionString}`);

            // Send the embed and then react with the numbers for users to react with,
            // the bot will determine how many to react with for the amount of options inputted
            context.message.channel.send(embed).then(message => {
                if (args.length == 2) {
                    message.react(reactionEmojis[1]);
                } else if (args.length == 3) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]));
                } else if (args.length == 4) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]))
                        .then(() => message.react(reactionEmojis[3]));
                } else if (args.length == 5) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]))
                        .then(() => message.react(reactionEmojis[3]))
                        .then(() => message.react(reactionEmojis[4]));
                } else if (args.length == 6) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]))
                        .then(() => message.react(reactionEmojis[3]))
                        .then(() => message.react(reactionEmojis[4]))
                        .then(() => message.react(reactionEmojis[5]));
                } else if (args.length == 7) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]))
                        .then(() => message.react(reactionEmojis[3]))
                        .then(() => message.react(reactionEmojis[4]))
                        .then(() => message.react(reactionEmojis[5]))
                        .then(() => message.react(reactionEmojis[6]));
                } else if (args.length == 8) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]))
                        .then(() => message.react(reactionEmojis[3]))
                        .then(() => message.react(reactionEmojis[4]))
                        .then(() => message.react(reactionEmojis[5]))
                        .then(() => message.react(reactionEmojis[6]))
                        .then(() => message.react(reactionEmojis[7]));
                } else if (args.length == 9) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]))
                        .then(() => message.react(reactionEmojis[3]))
                        .then(() => message.react(reactionEmojis[4]))
                        .then(() => message.react(reactionEmojis[5]))
                        .then(() => message.react(reactionEmojis[6]))
                        .then(() => message.react(reactionEmojis[7]))
                        .then(() => message.react(reactionEmojis[8]));
                } else if (args.length == 10) {
                    message.react(reactionEmojis[1])
                        .then(() => message.react(reactionEmojis[2]))
                        .then(() => message.react(reactionEmojis[3]))
                        .then(() => message.react(reactionEmojis[4]))
                        .then(() => message.react(reactionEmojis[5]))
                        .then(() => message.react(reactionEmojis[6]))
                        .then(() => message.react(reactionEmojis[7]))
                        .then(() => message.react(reactionEmojis[8]))
                        .then(() => message.react(reactionEmojis[9]));
                }
            }).catch(console.error);

            // Delete the message
            context.message.delete();
        } else if (args.length >= 11) { // If the user inputted more than 9 options
            const errorEmbed = new MessageEmbed()
                .setDescription("The poll command can only accept up to 9 options");

            context.message.channel.send(errorEmbed);
        } else { // If the user didn't give enough data
            const errorEmbed = new MessageEmbed()
                .setDescription("Please use the correct usage: <title>;<option 1>;<option 2>... (separate options with semicolons)");

            context.message.channel.send(errorEmbed);
        }
    }
}

module.exports = poll;