feat: Added Buttons

Author: seailz

src/main/java/com/seailz/jdaframework/DiscordBot.java

@@ -1,5 +1,6 @@
 package com.seailz.jdaframework;
 
+import com.seailz.jdaframework.button.ButtonListener;
 import com.seailz.jdaframework.command.Command;
 import com.seailz.jdaframework.command.listener.CommandRunListener;
 import com.seailz.jdaframework.command.registry.CommandRegistry;
@@ -11,10 +12,13 @@
 import lombok.Setter;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.JDABuilder;
+import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
 import net.dv8tion.jda.api.hooks.EventListener;
 import net.dv8tion.jda.api.hooks.ListenerAdapter;
 
 import javax.security.auth.login.LoginException;
+import java.util.HashMap;
+import java.util.function.Consumer;
 
 /**
  * Main Discord Bot Class
@@ -26,6 +30,8 @@ public class DiscordBot {
 
     @Getter
     private static DiscordBot instance;
+    @Getter
+    private static HashMap<String, Consumer<ButtonInteractionEvent>> buttonRegistry;
     private String token;
     private JDA jda;
     private JDABuilder builder;
@@ -40,11 +46,14 @@ public DiscordBot(String token) {
         contextMenuRegistry = new ContextMenuRegistry();
         builder = JDABuilder.createDefault(token);
 
+        buttonRegistry = new HashMap<>();
+
         registerListeners(
                 new MessageContextMenuListener(),
                 new UserContextMenuListener(),
                 new CommandRunListener(),
-                new ModalListener()
+                new ModalListener(),
+                new ButtonListener()
         );
     }
 

src/main/java/com/seailz/jdaframework/button/ButtonListener.java

@@ -0,0 +1,15 @@
+package com.seailz.jdaframework.button;
+
+import com.seailz.jdaframework.DiscordBot;
+import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
+import net.dv8tion.jda.api.hooks.ListenerAdapter;
+import org.jetbrains.annotations.NotNull;
+
+public class ButtonListener extends ListenerAdapter {
+    @Override
+    public void onButtonInteraction(@NotNull ButtonInteractionEvent event) {
+        if (DiscordBot.getButtonRegistry().containsKey(event.getButton().getId())) {
+            DiscordBot.getButtonRegistry().get(event.getButton().getId()).accept(event);
+        }
+    }
+}

src/main/java/com/seailz/jdaframework/button/ButtonManager.java

@@ -0,0 +1,16 @@
+package com.seailz.jdaframework.button;
+
+import com.seailz.jdaframework.DiscordBot;
+import lombok.Getter;
+import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
+import net.dv8tion.jda.api.interactions.components.buttons.Button;
+
+import java.util.function.Consumer;
+
+public class ButtonManager {
+
+    public static void listen(Button button, Consumer<ButtonInteractionEvent> onClick) {
+        DiscordBot.getButtonRegistry().put(button.getId(), onClick);
+    }
+
+}

src/test/java/com/seailz/test/TestCommand.java

@@ -11,6 +11,6 @@
 public class TestCommand extends Command {
     @Override
     public void onCommand(SlashCommandInteractionEvent e) {
-        e.reply("hi").queue();
+        new TestModal().open(e.getMember(), e);
     }
 }

src/test/java/com/seailz/test/TestModal.java

@@ -0,0 +1,19 @@
+package com.seailz.test;
+
+import com.seailz.jdaframework.modals.Modal;
+import net.dv8tion.jda.api.interactions.components.text.TextInput;
+import net.dv8tion.jda.api.interactions.components.text.TextInputStyle;
+
+public class TestModal extends Modal {
+
+
+    public TestModal() {
+        super("test", "test");
+
+        TextInput lol = TextInput.create("lol", "lol", TextInputStyle.SHORT).build();
+        addComponent(member -> lol);
+        onSubmit((member, mappings, event) -> {
+            event.getTextChannel().sendMessage("asdf").queue();
+        });
+    }
+}