Add a reset option to /greetings command

Author: TechnoVisionDev

src/main/java/technobot/commands/greetings/GreetingsCommand.java

@@ -2,19 +2,30 @@
 
 import net.dv8tion.jda.api.Permission;
 import net.dv8tion.jda.api.entities.ChannelType;
+import net.dv8tion.jda.api.entities.Emoji;
 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.requests.ErrorResponse;
 import technobot.TechnoBot;
 import technobot.commands.Category;
 import technobot.commands.Command;
 import technobot.data.GuildData;
 import technobot.data.cache.Greetings;
 import technobot.handlers.GreetingHandler;
+import technobot.listeners.ButtonListener;
 import technobot.util.embeds.EmbedUtils;
 
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+import java.util.concurrent.TimeUnit;
+
 /**
  * Command that displays and modifies greetings config.
  *
@@ -28,10 +39,11 @@ public GreetingsCommand(TechnoBot bot) {
         this.description = "Modify this server's greetings config.";
         this.category = Category.GREETINGS;
         this.permission = Permission.MANAGE_SERVER;
-        this.subCommands.add(new SubcommandData("channel", "Sets a channel to send welcome messages to.")
+        this.subCommands.add(new SubcommandData("channel", "Set a channel to send welcome messages to.")
                 .addOptions(new OptionData(OptionType.CHANNEL, "channel", "The channel to send welcome messages to")
                         .setChannelTypes(ChannelType.TEXT, ChannelType.NEWS)));
-        this.subCommands.add(new SubcommandData("config", "Displays the greetings config for this server."));
+        this.subCommands.add(new SubcommandData("config", "Display the greetings config for this server."));
+        this.subCommands.add(new SubcommandData("reset", "Reset all greetings data and settings."));
     }
 
     @Override
@@ -59,6 +71,28 @@ public void execute(SlashCommandInteractionEvent event) {
                 event.getHook().sendMessage(text).queue();
                 return;
             }
+            case "reset" -> {
+                long userID = event.getUser().getIdLong();
+                String uuid = userID + ":" + UUID.randomUUID();
+                text = "Would you like to reset the greeting system?\nThis will delete **ALL** data!";
+                List<Button> components = new ArrayList<>();
+                components.add(Button.success("greetings:yes:"+uuid, Emoji.fromMarkdown("\u2714")));
+                components.add(Button.danger("greetings:no:"+uuid, Emoji.fromUnicode("\u2716")));
+                ButtonListener.buttons.put(uuid, 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(uuid);
+                        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(uuid);
+                    };
+                    ButtonListener.executor.schedule(task, 3, TimeUnit.MINUTES);
+                });
+                return;
+            }
         }
         event.getHook().sendMessageEmbeds(EmbedUtils.createDefault(text)).queue();
     }

src/main/java/technobot/handlers/GreetingHandler.java

@@ -5,6 +5,8 @@
 import net.dv8tion.jda.api.entities.Guild;
 import org.bson.conversions.Bson;
 import technobot.TechnoBot;
+import technobot.data.GuildData;
+import technobot.data.cache.Config;
 import technobot.data.cache.Greetings;
 
 /**
@@ -148,4 +150,12 @@ public Long getChannel() {
     public Greetings getConfig() {
         return greetings;
     }
+
+    /**
+     * Resets all config data for greeting system.
+     */
+    public void reset() {
+        greetings = new Greetings(guild.getIdLong());
+        bot.database.greetings.replaceOne(filter, greetings);
+    }
 }

src/main/java/technobot/handlers/LevelingHandler.java

@@ -75,7 +75,7 @@ public void resetProfile(long userID) {
      */
     public void resetAll() {
         leaderboard.clear();
-        Bson query = Filters.and(Filters.eq("guild", guild.getIdLong()));
+        Bson query = Filters.eq("guild", guild.getIdLong());
         bot.database.leveling.deleteMany(query);
         Config config = new Config(guild.getIdLong());
         bot.database.config.replaceOne(query, config);

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

@@ -36,19 +36,7 @@ public void onButtonInteraction(ButtonInteractionEvent event) {
         if (components == null) return;
         String[] storedArgs = components.get(0).getId().split(":");
 
-        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 (pressedArgs[1].equals("no")) {
-                event.deferEdit().queue();
-                MessageEmbed embed = EmbedUtils.createError("Suggestion system was **NOT** reset!");
-                event.getHook().editOriginalComponents(new ArrayList<>()).setEmbeds(embed).queue();
-            }
-        }
-        else if (pressedArgs[0].equals("help") && storedArgs[0].equals("help")) {
+        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;
@@ -79,5 +67,29 @@ else if (pressedArgs[0].equals("help") && storedArgs[0].equals("help")) {
                 }
             }
         }
+        else 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 (pressedArgs[1].equals("no")) {
+                event.deferEdit().queue();
+                MessageEmbed embed = EmbedUtils.createError("Suggestion system was **NOT** reset!");
+                event.getHook().editOriginalComponents(new ArrayList<>()).setEmbeds(embed).queue();
+            }
+        }
+        else if (pressedArgs[0].equals("greetings") && storedArgs[0].equals("greetings")) {
+            if (pressedArgs[1].equals("yes")) {
+                event.deferEdit().queue();
+                GuildData.get(event.getGuild()).greetingHandler.reset();
+                MessageEmbed embed = EmbedUtils.createSuccess("Greeting system was successfully reset!");
+                event.getHook().editOriginalComponents(new ArrayList<>()).setEmbeds(embed).queue();
+            } else if (pressedArgs[1].equals("no")) {
+                event.deferEdit().queue();
+                MessageEmbed embed = EmbedUtils.createError("Greeting system was **NOT** reset!");
+                event.getHook().editOriginalComponents(new ArrayList<>()).setEmbeds(embed).queue();
+            }
+        }
     }
 }