Merge pull request #19 from ice-games/feat/guildcommnads

Guild Commands

Author: seailz

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

@@ -13,6 +13,7 @@
 import lombok.Setter;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.JDABuilder;
+import net.dv8tion.jda.api.entities.Guild;
 import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
 import net.dv8tion.jda.api.events.interaction.component.SelectMenuInteractionEvent;
 import net.dv8tion.jda.api.hooks.EventListener;
@@ -121,4 +122,17 @@ public DiscordBot registerCommands(Command... commands) {
         }
         return this;
     }
+
+    /**
+     * Registers commands. These commands will <b>ONLY</b> show up in the guild you register them in, and not in DMs or any other server.
+     * @param guild The guild to register them for
+     * @param commands The commands to register
+     * @return The bot instance
+     */
+    public DiscordBot registerGuildCommand(Guild guild, Command... commands) {
+        for (Command command : commands) {
+            registry.registerGuildCommand(command, guild);
+        }
+        return this;
+    }
 }

src/main/java/com/seailz/jdaframework/command/registry/CommandRegistry.java

@@ -4,6 +4,7 @@
 import lombok.Getter;
 import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.JDABuilder;
+import net.dv8tion.jda.api.entities.Guild;
 import net.dv8tion.jda.api.requests.restaction.CommandCreateAction;
 
 import java.util.HashMap;
@@ -34,4 +35,18 @@ public void registerCommand(JDA jda, Command command) {
         commands.put(command.getName(), command);
     }
 
+    public void registerGuildCommand(Command command, Guild guild) {
+        if (commands.containsValue(command)) return;
+        if (command.getOptions().isEmpty()) {
+            guild.upsertCommand(command.getName(), command.getDescription()).queue();
+        } else {
+            CommandCreateAction action = guild.upsertCommand(command.getName(), command.getDescription());
+            command.getOptions().forEach(option -> {
+                action.addOption(option.getType(), option.getName(), option.getDescription(), option.isRequired());
+            });
+            action.queue();
+        }
+        commands.put(command.getName(), command);
+    }
+
 }