AltitudeBot/src/main/java/com/alttd/commandManager/commands/PollCommand/CommandPoll.java

package com.alttd.commandManager.commands.PollCommand;

import com.alttd.commandManager.CommandManager;
import com.alttd.commandManager.DiscordCommand;
import com.alttd.commandManager.SubCommand;
import com.alttd.permissions.PermissionManager;
import com.alttd.util.Logger;
import com.alttd.util.Util;
import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.JDA;
import net.dv8tion.jda.api.events.interaction.command.CommandAutoCompleteInteractionEvent;
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
import net.dv8tion.jda.api.interactions.commands.OptionType;
import net.dv8tion.jda.api.interactions.commands.build.Commands;
import net.dv8tion.jda.api.interactions.commands.build.SlashCommandData;
import net.dv8tion.jda.api.interactions.commands.build.SubcommandData;

import java.awt.*;
import java.util.HashMap;

public class CommandPoll extends DiscordCommand {

    private final HashMap<String, SubCommand> subCommandMap = new HashMap<>();

    public CommandPoll(JDA jda, CommandManager commandManager) {
        SlashCommandData slashCommandData = Commands.slash(getName(), "Create, edit, and manage polls")
                .addSubcommands(
                        new SubcommandData("add", "Add a new poll to a channel")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll should go into", true, true)
                                .addOption(OptionType.STRING, "title", "Title of the embed (max 256 characters)", true),
                        new SubcommandData("edit_title", "Edit the title of a poll")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll is in", true, true)
                                .addOption(OptionType.INTEGER, "message_id", "Id of the poll you're editing", true)
                                .addOption(OptionType.STRING, "title", "The new title for the poll (max 256 characters)", true),
                        new SubcommandData("edit_description", "Edit the description of a poll")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll is in", true, true)
                                .addOption(OptionType.INTEGER, "message_id", "Id of the poll you're editing", true)
                                .addOption(OptionType.STRING, "description", "The new description for the poll (max 2048 characters)", true),
                        new SubcommandData("add_button", "Add a button to a poll")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll is in", true, true)
                                .addOption(OptionType.INTEGER, "message_id", "Id of the poll you're adding a button to", true)
                                .addOption(OptionType.INTEGER, "button_row", "Row the button should go in (1-5)", true)
                                .addOption(OptionType.STRING, "button_name", "Name of the button you're adding"),
                        new SubcommandData("remove_button", "Remove a button from a poll")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll is in", true, true)
                                .addOption(OptionType.INTEGER, "message_id", "Id of the poll you're removing a button from", true)
                                .addOption(OptionType.STRING, "button_name", "Name of the button you're removing"),
                        new SubcommandData("open", "Open a poll")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll is in", true, true)
                                .addOption(OptionType.INTEGER, "message_id", "Id of the poll you're opening", true),
                        new SubcommandData("close", "Close a poll")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll is in", true, true)
                                .addOption(OptionType.INTEGER, "message_id", "Id of the poll you're closing", true),
                        new SubcommandData("results", "Get the results for a poll")
                                .addOption(OptionType.CHANNEL, "channel", "Channel this poll is in", true, true)
                                .addOption(OptionType.INTEGER, "message_id", "Id of the poll you want the results for", true));
        Util.registerSubcommands(subCommandMap, new SubCommandAdd(this),
                new SubCommandAddButton(this),
                new SubCommandClose(this),
                new SubCommandEditDescription(this),
                new SubCommandEditTitle(this),
                new SubCommandOpen(this),
                new SubCommandRemoveButton(this),
                new SubCommandResults(this));
        Util.registerCommand(commandManager, jda, slashCommandData, getName());
    }

    @Override
    public String getName() {
        return "poll";
    }

    @Override
    public void execute(SlashCommandInteractionEvent event) {
        if (event.getGuild() == null || event.getMember() == null) {
            event.replyEmbeds(Util.guildOnlyCommand(getName())).setEphemeral(true).queue();
            return;
        }
        if (PermissionManager.getInstance().hasPermission(event.getTextChannel(), event.getIdLong(), Util.getGroupIds(event.getMember()), getPermission())) {
            event.replyEmbeds(Util.noPermission(getName())).setEphemeral(true).queue();
            return;
        }

        String subcommandName = event.getInteraction().getSubcommandName();
        if (subcommandName == null) {
            Logger.severe("No subcommand found for %", getName());
            return;
        }

        SubCommand subCommand = subCommandMap.get(subcommandName);
        if (subCommand == null) {
            event.replyEmbeds(new EmbedBuilder()
                    .setTitle("Subcommand not found")
                    .setDescription("Unable to find subcommand `" + subcommandName + "`")
                    .setColor(Color.RED)
                    .build())
                    .setEphemeral(true)
                    .queue();
            return;
        }

        subCommand.execute(event);
    }

    @Override
    public void suggest(CommandAutoCompleteInteractionEvent event) {

    }

    @Override
    public String getHelpMessage() {
        return null;
    }
}