114 lines
6.0 KiB
Java
114 lines
6.0 KiB
Java
package com.alttd.commandManager.commands.PollCommand;
|
|
|
|
import com.alttd.commandManager.CommandManager;
|
|
import com.alttd.commandManager.DiscordCommand;
|
|
import com.alttd.commandManager.SubOption;
|
|
import com.alttd.util.Logger;
|
|
import com.alttd.util.Util;
|
|
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.CommandData;
|
|
import net.dv8tion.jda.api.interactions.commands.build.Commands;
|
|
import net.dv8tion.jda.api.interactions.commands.build.OptionData;
|
|
import net.dv8tion.jda.api.interactions.commands.build.SubcommandData;
|
|
|
|
import java.util.HashMap;
|
|
|
|
public class CommandPoll extends DiscordCommand {
|
|
|
|
private final HashMap<String, SubOption> subOptionsMap = new HashMap<>();
|
|
private final CommandData commandData;
|
|
|
|
public CommandPoll(JDA jda, CommandManager commandManager) {
|
|
commandData = 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)
|
|
.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)
|
|
.addOption(OptionType.STRING, "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)
|
|
.addOption(OptionType.STRING, "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)
|
|
.addOption(OptionType.STRING, "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)
|
|
.addOption(OptionType.STRING, "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)
|
|
.addOption(OptionType.STRING, "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)
|
|
.addOption(OptionType.STRING, "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)
|
|
.addOption(OptionType.STRING, "message_id", "Id of the poll you want the results for", true));
|
|
commandData.setDefaultEnabled(true);
|
|
Util.registerSubOptions(subOptionsMap,
|
|
new SubCommandAdd(null,this),
|
|
new SubCommandAddButton(null, this),
|
|
new SubCommandClose(null,this),
|
|
new SubCommandEditDescription(null, this),
|
|
new SubCommandEditTitle(null, this),
|
|
new SubCommandOpen(null, this),
|
|
new SubCommandRemoveButton(null, this),
|
|
new SubCommandResults(null,this));
|
|
Util.registerCommand(commandManager, jda, commandData, 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;
|
|
}
|
|
|
|
String subcommandName = event.getInteraction().getSubcommandGroup();
|
|
subcommandName = subcommandName == null ? event.getInteraction().getSubcommandName() : subcommandName;
|
|
if (subcommandName == null) {
|
|
Logger.severe("No subcommand found for %", getName());
|
|
return;
|
|
}
|
|
|
|
SubOption subOption = subOptionsMap.get(subcommandName);
|
|
if (subOption == null) {
|
|
event.replyEmbeds(Util.invalidSubcommand(subcommandName))
|
|
.setEphemeral(true)
|
|
.queue();
|
|
return;
|
|
}
|
|
|
|
subOption.execute(event);
|
|
}
|
|
|
|
@Override
|
|
public void suggest(CommandAutoCompleteInteractionEvent event) {
|
|
|
|
}
|
|
|
|
@Override
|
|
public String getHelpMessage() {
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public CommandData getCommandData() {
|
|
return commandData;
|
|
}
|
|
}
|