Fix /suggestions reset command and other button errors

Author: TechnoVisionDev

src/main/java/technobot/commands/suggestions/SuggestionsCommand.java

@@ -5,19 +5,27 @@
 import net.dv8tion.jda.api.entities.Emoji;
 import net.dv8tion.jda.api.entities.Guild;
 import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
+import net.dv8tion.jda.api.exceptions.ContextException;
+import net.dv8tion.jda.api.exceptions.ErrorHandler;
 import net.dv8tion.jda.api.interactions.commands.OptionMapping;
 import net.dv8tion.jda.api.interactions.commands.OptionType;
 import net.dv8tion.jda.api.interactions.commands.build.OptionData;
 import net.dv8tion.jda.api.interactions.commands.build.SubcommandData;
+import net.dv8tion.jda.api.interactions.components.ActionRow;
 import net.dv8tion.jda.api.interactions.components.buttons.Button;
+import net.dv8tion.jda.api.interactions.components.buttons.ButtonStyle;
+import net.dv8tion.jda.api.requests.ErrorResponse;
 import technobot.TechnoBot;
 import technobot.commands.Category;
 import technobot.commands.Command;
 import technobot.data.GuildData;
 import technobot.handlers.SuggestionHandler;
+import technobot.listeners.ButtonListener;
 import technobot.util.embeds.EmbedUtils;
 
 import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.TimeUnit;
 
 /**
  * Admin command to setup and modify the suggestion board.
@@ -110,13 +118,24 @@ public void execute(SlashCommandInteractionEvent event) {
                 return;
             }
             case "reset" -> {
-                String userID = event.getUser().getId();
+                long userID = event.getUser().getIdLong();
                 text = "Would you like to reset the suggestions system?\nThis will delete **ALL** data!";
-                event.getHook().sendMessageEmbeds(EmbedUtils.createDefault(text))
-                        .addActionRow(
-                                Button.success("suggestions:yes:"+userID, Emoji.fromMarkdown("\u2714")),
-                                Button.danger("suggestions:no:"+userID, Emoji.fromUnicode("\u2716")))
-                        .queue();
+                List<Button> components = new ArrayList<>();
+                components.add(Button.success("suggestions:yes:"+userID, Emoji.fromMarkdown("\u2714")));
+                components.add(Button.danger("suggestions:no:"+userID, Emoji.fromUnicode("\u2716")));
+                ButtonListener.buttons.put(userID, components);
+                event.getHook().sendMessageEmbeds(EmbedUtils.createDefault(text)).addActionRow(components).queue(interactionHook -> {
+                    // Timer task to disable buttons and clear cache after 3 minutes
+                    Runnable task = () -> {
+                        List<Button> actionRow = ButtonListener.buttons.get(userID);
+                        for (int i = 0; i < actionRow.size(); i++) {
+                            actionRow.set(i, actionRow.get(i).asDisabled());
+                        }
+                        interactionHook.editMessageComponents(ActionRow.of(actionRow)).queue(null, new ErrorHandler().ignore(ErrorResponse.UNKNOWN_MESSAGE));
+                        ButtonListener.buttons.remove(userID);
+                    };
+                    ButtonListener.executor.schedule(task, 3, TimeUnit.MINUTES);
+                });
                 return;
             }
         }

src/main/java/technobot/commands/utility/HelpCommand.java

@@ -3,13 +3,15 @@
 import net.dv8tion.jda.api.EmbedBuilder;
 import net.dv8tion.jda.api.entities.MessageEmbed;
 import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
+import net.dv8tion.jda.api.exceptions.ErrorHandler;
 import net.dv8tion.jda.api.interactions.commands.OptionMapping;
 import net.dv8tion.jda.api.interactions.commands.OptionType;
 import net.dv8tion.jda.api.interactions.commands.build.OptionData;
 import net.dv8tion.jda.api.interactions.commands.build.SubcommandData;
 import net.dv8tion.jda.api.interactions.components.ActionRow;
 import net.dv8tion.jda.api.interactions.components.buttons.Button;
 import net.dv8tion.jda.api.interactions.components.buttons.ButtonStyle;
+import net.dv8tion.jda.api.requests.ErrorResponse;
 import net.dv8tion.jda.api.requests.restaction.interactions.ReplyCallbackAction;
 import technobot.TechnoBot;
 import technobot.commands.Category;
@@ -85,7 +87,7 @@ public void execute(SlashCommandInteractionEvent event) {
                         List<Button> actionRow = ButtonListener.buttons.get(userID);
                         actionRow.set(0, actionRow.get(0).asDisabled());
                         actionRow.set(2, actionRow.get(2).asDisabled());
-                        interactionHook.editOriginalComponents(ActionRow.of(actionRow)).queue();
+                        interactionHook.editOriginalComponents(ActionRow.of(actionRow)).queue(null, new ErrorHandler().ignore(ErrorResponse.UNKNOWN_MESSAGE));
                         ButtonListener.buttons.remove(userID);
                         ButtonListener.menus.remove(userID);
                     };

src/main/java/technobot/listeners/ButtonListener.java

@@ -24,29 +24,29 @@ public class ButtonListener extends ListenerAdapter {
     @Override
     public void onButtonInteraction(ButtonInteractionEvent event) {
         // Check that these are 'help' buttons
-        String[] args = event.getComponentId().split(":");
+        String[] pressedArgs = event.getComponentId().split(":");
 
         // Check if user owns this menu
-        long userID = Long.parseLong(args[2]);
+        long userID = Long.parseLong(pressedArgs[2]);
         if (userID != event.getUser().getIdLong()) return;
         List<Button> components = buttons.get(userID);
         if (components == null) return;
+        String[] storedArgs = components.get(0).getId().split(":");
 
-        if (args[0].equals("suggestions")) {
-            if (args[1].equals("yes")) {
+        if (pressedArgs[0].equals("suggestions") && storedArgs[0].equals("suggestions")) {
+            if (pressedArgs[1].equals("yes")) {
                 event.deferEdit().queue();
                 GuildData.get(event.getGuild()).suggestionHandler.reset();
                 MessageEmbed embed = EmbedUtils.createSuccess("Suggestion system was successfully reset!");
                 event.getHook().editOriginalComponents(new ArrayList<>()).setEmbeds(embed).queue();
-            } else if (args[1].equals("no")) {
+            } else if (pressedArgs[1].equals("no")) {
                 event.deferEdit().queue();
                 MessageEmbed embed = EmbedUtils.createError("Suggestion system was **NOT** reset!");
                 event.getHook().editOriginalComponents(new ArrayList<>()).setEmbeds(embed).queue();
             }
         }
-
-        if (args[0].equals("help")) {
-            if (args[1].equals("next")) {
+        else if (pressedArgs[0].equals("help") && storedArgs[0].equals("help")) {
+            if (pressedArgs[1].equals("next")) {
                 // Move to next embed
                 int page = Integer.parseInt(components.get(1).getId().split(":")[2]) + 1;
                 List<MessageEmbed> embeds = menus.get(userID);
@@ -60,7 +60,7 @@ public void onButtonInteraction(ButtonInteractionEvent event) {
                     buttons.put(userID, components);
                     event.editComponents(ActionRow.of(components)).setEmbeds(embeds.get(page)).queue();
                 }
-            } else if (args[1].equals("prev")) {
+            } else if (pressedArgs[1].equals("prev")) {
                 // Move to previous embed
                 int page = Integer.parseInt(components.get(1).getId().split(":")[2]) - 1;
                 List<MessageEmbed> embeds = menus.get(userID);