add feature toggling

src/main/java/cc/fascinated/bat/service/CommandService.java

@@ -4,6 +4,7 @@ import cc.fascinated.bat.Consts;
 import cc.fascinated.bat.command.*;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.config.Config;
+import cc.fascinated.bat.features.FeatureProfile;
 import cc.fascinated.bat.model.BatGuild;
 import cc.fascinated.bat.model.BatUser;
 import lombok.Getter;
@@ -81,9 +82,8 @@ public class CommandService extends ListenerAdapter {
                 return;
             }
             jda.retrieveCommands().complete().forEach(command -> jda.deleteCommandById(command.getId()).complete());
-            adminGuild.updateCommands().addCommands(commands.values().stream()
-                    .map(BatCommand::getCommandData).toList()).complete();
-            log.info("Registered {} slash commands in {}ms (DEV MODE)", commands.size(), System.currentTimeMillis() - before);
+            List<Command> registeredCommands = adminGuild.updateCommands().addCommands(commands.values().stream().map(BatCommand::getCommandData).toList()).complete();
+            log.info("Registered {} slash commands in {}ms (DEV MODE)", registeredCommands.size(), System.currentTimeMillis() - before);
             return;
         }
 
@@ -123,7 +123,6 @@ public class CommandService extends ListenerAdapter {
         } else {
             log.error("Unable to find the admin guild to register hidden commands");
         }
-
         log.info("Registered {} slash commands in {}ms", discordCommands.size(), System.currentTimeMillis() - before);
     }
 
@@ -221,6 +220,16 @@ public class CommandService extends ListenerAdapter {
                 return;
             }
 
+            if (guild != null) {
+                FeatureProfile featureProfile = guild.getFeatureProfile();
+                if (featureProfile.isFeatureDisabled(command.getFeature())) {
+                    event.replyEmbeds(EmbedUtils.errorEmbed()
+                            .setDescription("This command has been disabled by the guild owner")
+                            .build()).setEphemeral(true).queue();
+                    return;
+                }
+            }
+
             log.info("Executing command \"{}\" for user \"{}\"", commandName, user.getDiscordUser().getName());
             executor.execute(guild, user, ranInsideGuild ? event.getChannel().asTextChannel() : event.getChannel().asPrivateChannel(),
                     event.getMember(), event.getInteraction());

src/main/java/cc/fascinated/bat/service/FeatureService.java

@@ -2,6 +2,8 @@ package cc.fascinated.bat.service;
 
 import cc.fascinated.bat.command.BatCommand;
 import cc.fascinated.bat.features.Feature;
+import cc.fascinated.bat.features.command.FeatureCommand;
+import jakarta.annotation.PostConstruct;
 import lombok.Getter;
 import lombok.NonNull;
 import lombok.extern.log4j.Log4j2;
@@ -11,7 +13,9 @@ import org.springframework.context.annotation.DependsOn;
 import org.springframework.stereotype.Service;
 
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author Fascinated (fascinated7)
@@ -21,17 +25,28 @@ import java.util.List;
 @Log4j2
 @DependsOn("commandService")
 public class FeatureService {
+    public static FeatureService INSTANCE;
+    private final ApplicationContext context;
+    private final CommandService commandService;
+
     /**
      * The registered features
      */
-    private final List<Feature> features = new ArrayList<>();
+    private final Map<String, Feature> features = new HashMap<>();
 
     @Autowired
     public FeatureService(@NonNull ApplicationContext context, @NonNull CommandService commandService) {
+        this.context = context;
+        this.commandService = commandService;
+        INSTANCE = this;
+    }
+
+    @PostConstruct
+    public void init() {
         context.getBeansOfType(Feature.class)
                 .values()
                 .forEach((feature) -> {
-                    features.add(context.getBean(feature.getClass()));
+                    features.put(feature.getName().toLowerCase(), feature);
                 });
 
         context.getBeansOfType(BatCommand.class)
@@ -40,6 +55,27 @@ public class FeatureService {
                     commandService.registerCommand(context.getBean(command.getClass()));
                 });
 
+        commandService.registerCommand(context.getBean(FeatureCommand.class));
         commandService.registerSlashCommands(); // Register all slash commands
     }
+
+    /**
+     * Gets a feature by name
+     *
+     * @param name The name of the feature
+     * @return The feature
+     */
+    public Feature getFeature(@NonNull String name) {
+        return features.get(name.toLowerCase());
+    }
+
+    /**
+     * Checks if a feature is registered
+     *
+     * @param name The name of the feature
+     * @return Whether the feature is registered
+     */
+    public boolean isFeature(@NonNull String name) {
+        return features.containsKey(name.toLowerCase());
+    }
 }