75 lines
3.1 KiB
Java
75 lines
3.1 KiB
Java
package com.alttd.commandManager.commands.PollCommand;
|
|
|
|
import com.alttd.commandManager.DiscordCommand;
|
|
import com.alttd.commandManager.SubCommand;
|
|
import com.alttd.commandManager.SubCommandGroup;
|
|
import com.alttd.database.queries.Poll.Poll;
|
|
import com.alttd.util.OptionMappingParsing;
|
|
import com.alttd.util.Util;
|
|
import net.dv8tion.jda.api.EmbedBuilder;
|
|
import net.dv8tion.jda.api.entities.Guild;
|
|
import net.dv8tion.jda.api.entities.Message;
|
|
import net.dv8tion.jda.api.entities.channel.middleman.GuildMessageChannel;
|
|
import net.dv8tion.jda.api.events.interaction.command.CommandAutoCompleteInteractionEvent;
|
|
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
|
|
import net.dv8tion.jda.api.interactions.AutoCompleteQuery;
|
|
import net.dv8tion.jda.api.interactions.InteractionHook;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.stream.Collectors;
|
|
|
|
public class SubCommandEditTitle extends SubCommand {
|
|
protected SubCommandEditTitle(SubCommandGroup parentGroup, DiscordCommand parent) {
|
|
super(parentGroup, parent);
|
|
}
|
|
|
|
@Override
|
|
public String getName() {
|
|
return "edit_title";
|
|
}
|
|
|
|
@Override
|
|
public void execute(SlashCommandInteractionEvent event) {
|
|
PollChannel pollChannel = PollUtil.getPollHandleErrors(event, getName());
|
|
if (pollChannel == null)
|
|
return;
|
|
|
|
String title = OptionMappingParsing.getString("title", event, getName());
|
|
if (title == null || title.length() > 256) {
|
|
if (title == null)
|
|
event.replyEmbeds(Util.genericErrorEmbed("Error", "No title found")).setEphemeral(true).queue();
|
|
else
|
|
event.replyEmbeds(Util.genericErrorEmbed("Error", "Title too long")).setEphemeral(true).queue();
|
|
return;
|
|
}
|
|
|
|
event.replyEmbeds(Util.genericWaitingEmbed("Waiting...", "Editing poll...")).setEphemeral(true).queue(hook -> {
|
|
pollChannel.textChannel().retrieveMessageById(pollChannel.poll().getPollId()).queue(message -> updatePoll(message, title, hook),
|
|
error -> hook.editOriginalEmbeds(Util.genericErrorEmbed("Error", "Unable to find message with id [" + pollChannel.poll().getPollId() + "].")).queue());
|
|
});
|
|
}
|
|
|
|
private void updatePoll(Message message, String title, InteractionHook hook) {
|
|
EmbedBuilder firstEmbedBuilder = Util.getFirstEmbedBuilder(message);
|
|
if (firstEmbedBuilder == null) {
|
|
hook.editOriginalEmbeds(Util.genericErrorEmbed("Error", "Unable to get embed from poll message."))
|
|
.queue();
|
|
return;
|
|
}
|
|
message.editMessageEmbeds(firstEmbedBuilder.setTitle(title).build()).queue(
|
|
resultMessage -> hook.editOriginalEmbeds(Util.genericSuccessEmbed("Success", "Updated the poll title.")).queue(),
|
|
error -> hook.editOriginalEmbeds(Util.genericErrorEmbed("Error", "Unable to edit poll message.")).queue());
|
|
}
|
|
|
|
@Override
|
|
public void suggest(CommandAutoCompleteInteractionEvent event) {
|
|
PollUtil.handleSuggestMessageId(event);
|
|
}
|
|
|
|
@Override
|
|
public String getHelpMessage() {
|
|
return null;
|
|
}
|
|
}
|