much stuff

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

@@ -0,0 +1,156 @@
+package cc.fascinated.bat.service;
+
+import cc.fascinated.bat.command.BatCommand;
+import cc.fascinated.bat.command.BatSubCommand;
+import cc.fascinated.bat.command.impl.global.beatsaber.scoresaber.LinkSubCommand;
+import cc.fascinated.bat.command.impl.global.beatsaber.scoresaber.ScoreSaberCommand;
+import cc.fascinated.bat.command.impl.global.beatsaber.scoresaber.UserSubCommand;
+import cc.fascinated.bat.common.EmbedUtils;
+import cc.fascinated.bat.model.BatGuild;
+import cc.fascinated.bat.model.user.BatUser;
+import lombok.NonNull;
+import lombok.extern.log4j.Log4j2;
+import net.dv8tion.jda.api.JDA;
+import net.dv8tion.jda.api.entities.Guild;
+import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
+import net.dv8tion.jda.api.hooks.ListenerAdapter;
+import net.dv8tion.jda.api.interactions.commands.OptionMapping;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.annotation.DependsOn;
+import org.springframework.stereotype.Service;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @author Fascinated (fascinated7)
+ */
+@Service @Log4j2
+@DependsOn("discordService")
+public class CommandService extends ListenerAdapter {
+    /**
+     * The registered commands
+     */
+    private final Map<String, BatCommand> commands = new HashMap<>();
+
+    /**
+     * The guild service to use
+     */
+    private final GuildService guildService;
+
+    /**
+     * The user service to use
+     */
+    private final UserService userService;
+
+    /**
+     * The application context to use
+     */
+    private final ApplicationContext context;
+
+    @Autowired
+    public CommandService(@NonNull GuildService guildService, @NonNull UserService userService, @NonNull ApplicationContext context) {
+        this.guildService = guildService;
+        this.userService = userService;
+        this.context = context;
+        DiscordService.JDA.addEventListener(this);
+
+        // Guild commands
+        // todo: add some, duh
+
+        // Global commands
+        registerCommand(context.getBean(ScoreSaberCommand.class)
+                .addSubCommand("link", context.getBean(LinkSubCommand.class))
+                .addSubCommand("user", context.getBean(UserSubCommand.class)
+        ));
+
+        registerSlashCommands(); // Register all slash commands
+    }
+
+    /**
+     * Registers a command
+     *
+     * @param command The command to register
+     */
+    public void registerCommand(@NonNull BatCommand command) {
+        commands.put(command.getName().toLowerCase(), command);
+    }
+
+    /**
+     * Registers all slash commands
+     */
+    public void registerSlashCommands() {
+        log.info("Registering all slash commands");
+        JDA jda = DiscordService.JDA;
+        long before = System.currentTimeMillis();
+
+        // Unregister all commands that Discord has but we don't
+        jda.retrieveCommands().complete().forEach(command -> {
+            if (commands.containsKey(command.getName())) {
+                return;
+            }
+
+            jda.deleteCommandById(command.getId()).complete(); // Unregister the command on Discord
+            log.info("Unregistered unknown command \"{}\" from Discord", command.getName());
+        });
+
+        // Register all commands
+        for (BatCommand command : commands.values()) {
+            if (command.getCommandData() == null) {
+                continue;
+            }
+
+            jda.upsertCommand(command.getCommandData()).complete(); // Register the command on Discord
+        }
+        log.info("Registered all slash commands in {}ms", System.currentTimeMillis() - before);
+    }
+
+    @Override
+    public void onSlashCommandInteraction(@NotNull SlashCommandInteractionEvent event) {
+        Guild discordGuild = event.getGuild();
+        if (discordGuild == null) {
+            return;
+        }
+        if (event.getUser().isBot()) {
+            return;
+        }
+        if (event.getMember() == null) {
+            return;
+        }
+
+        String commandName = event.getName();
+        BatCommand command = commands.get(commandName);
+        if (command == null) {
+            return;
+        }
+
+        BatGuild guild = guildService.getGuild(discordGuild.getId());
+        BatUser user = userService.getUser(event.getUser().getId());
+
+        // No args provided, use the main command executor
+        List<OptionMapping> options = event.getInteraction().getOptions();
+
+        try {
+            if (options.isEmpty()) {
+                command.execute(guild, user, event.getChannel().asTextChannel(), event.getMember(), event.getInteraction(), null);
+            }
+
+            // Check if the sub command exists
+            for (Map.Entry<String, BatSubCommand> subCommand : command.getSubCommands().entrySet()) {
+                for (OptionMapping option : options) {
+                    if (subCommand.getKey().equalsIgnoreCase(option.getName())) {
+                        subCommand.getValue().execute(guild, user, event.getChannel().asTextChannel(), event.getMember(), event.getInteraction(), option);
+                        break;
+                    }
+                }
+            }
+        } catch (Exception ex) {
+            log.error("An error occurred while executing command \"{}\"", commandName, ex);
+            event.replyEmbeds(EmbedUtils.buildErrorEmbed("An error occurred while executing the command\n\n" +
+                    ex.getLocalizedMessage()).build()).queue();
+        }
+    }
+}

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

@@ -6,6 +6,7 @@ import net.dv8tion.jda.api.JDA;
 import net.dv8tion.jda.api.JDABuilder;
 import net.dv8tion.jda.api.entities.Activity;
 import net.dv8tion.jda.api.requests.GatewayIntent;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
@@ -19,10 +20,13 @@ public class DiscordService {
     /**
      * The JDA instance
      */
-    private final JDA jda;
+    public static JDA JDA;
 
-    public DiscordService(@Value("${discord.token}") String token) throws Exception {
-        jda = JDABuilder.createLight(token, EnumSet.of(
+    @Autowired
+    public DiscordService(
+            @Value("${discord.token}") String token
+    ) throws Exception {
+        JDA = JDABuilder.createLight(token, EnumSet.of(
                         GatewayIntent.GUILD_MESSAGES,
                         GatewayIntent.MESSAGE_CONTENT
                 )).build()
@@ -32,11 +36,12 @@ public class DiscordService {
         TimerUtils.scheduleRepeating(this::updateActivity, 0, 1000 * 60 * 5);
     }
 
+
     /**
      * Updates the activity of the bot
      */
     public void updateActivity() {
-        int guildCount = jda.getGuilds().size();
-        jda.getPresence().setActivity(Activity.playing("with %s guilds".formatted(guildCount)));
+        int guildCount = JDA.getGuilds().size();
+        JDA.getPresence().setActivity(Activity.playing("with %s guilds".formatted(guildCount)));
     }
 }

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

@@ -1,12 +1,13 @@
 package cc.fascinated.bat.service;
 
-import cc.fascinated.bat.model.Guild;
+import cc.fascinated.bat.model.BatGuild;
 import cc.fascinated.bat.repository.GuildRepository;
 import lombok.NonNull;
 import lombok.extern.log4j.Log4j2;
 import net.dv8tion.jda.api.events.guild.GuildJoinEvent;
 import net.dv8tion.jda.api.hooks.ListenerAdapter;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.DependsOn;
 import org.springframework.stereotype.Service;
 
 import java.util.Optional;
@@ -15,23 +16,17 @@ import java.util.Optional;
  * @author Fascinated (fascinated7)
  */
 @Service @Log4j2
+@DependsOn("discordService")
 public class GuildService extends ListenerAdapter {
     /**
      * The guild repository to use
      */
     private final GuildRepository guildRepository;
 
-    /**
-     * The discord service to use
-     */
-    private final DiscordService discordService;
-
     @Autowired
-    public GuildService(@NonNull GuildRepository guildRepository, @NonNull DiscordService discordService) {
+    public GuildService(@NonNull GuildRepository guildRepository) {
         this.guildRepository = guildRepository;
-        this.discordService = discordService;
-
-        discordService.getJda().addEventListener(this);
+        DiscordService.JDA.addEventListener(this);
     }
 
     /**
@@ -40,19 +35,28 @@ public class GuildService extends ListenerAdapter {
      * @param id The ID of the guild
      * @return The guild
      */
-    public Guild getGuild(@NonNull String id) {
+    public BatGuild getGuild(@NonNull String id) {
         long start = System.currentTimeMillis();
-        Optional<Guild> optionalGuild = guildRepository.findById(id);
+        Optional<BatGuild> optionalGuild = guildRepository.findById(id);
         if (optionalGuild.isPresent()) {
             return optionalGuild.get();
         }
-        Guild guild = guildRepository.save(new Guild(id));
+        BatGuild guild = guildRepository.save(new BatGuild(id));
         log.info("Created guild \"{}\" in {}ms", id, System.currentTimeMillis() - start);
         return guild;
     }
 
+    /**
+     * Saves a guild
+     *
+     * @param guild The guild to save
+     */
+    public void saveGuild(@NonNull BatGuild guild) {
+        guildRepository.save(guild);
+    }
+
     @Override
-    public void onGuildJoin(GuildJoinEvent event) {
+    public final void onGuildJoin(GuildJoinEvent event) {
         log.info("Joined guild \"{}\"", event.getGuild().getId());
         getGuild(event.getGuild().getId()); // Ensure the guild is in the database
     }

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

@@ -0,0 +1,85 @@
+package cc.fascinated.bat.service;
+
+import cc.fascinated.bat.common.DateUtils;
+import cc.fascinated.bat.common.WebRequest;
+import cc.fascinated.bat.exception.BadRequestException;
+import cc.fascinated.bat.exception.ResourceNotFoundException;
+import cc.fascinated.bat.model.beatsaber.scoresaber.ScoreSaberAccountToken;
+import cc.fascinated.bat.model.beatsaber.scoresaber.ScoreSaberPageMetadataToken;
+import cc.fascinated.bat.model.beatsaber.scoresaber.ScoreSaberPlayerScoreToken;
+import cc.fascinated.bat.model.beatsaber.scoresaber.ScoreSaberScoresPageToken;
+import cc.fascinated.bat.model.user.profiles.ScoreSaberProfile;
+import lombok.extern.log4j.Log4j2;
+import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+@Service @Log4j2(topic = "ScoreSaber Service")
+public class ScoreSaberService {
+    private static final String SCORESABER_API = "https://scoresaber.com/api/";
+    private static final String GET_PLAYER_ENDPOINT = SCORESABER_API + "player/%s/full";
+    private static final String GET_PLAYER_SCORES_ENDPOINT = SCORESABER_API + "player/%s/scores?limit=100&sort=%s&page=%s&withMetadata=true";
+
+    /**
+     * Gets the account from the ScoreSaber API.
+     *
+     * @param id The id of the account.
+     * @return The account.
+     * @throws ResourceNotFoundException If the account is not found.
+     * @throws cc.fascinated.bat.exception.RateLimitException If the ScoreSaber rate limit is reached.
+     */
+    public ScoreSaberAccountToken getAccount(String id) {
+        ScoreSaberAccountToken account = WebRequest.getAsEntity(String.format(GET_PLAYER_ENDPOINT, id), ScoreSaberAccountToken.class);
+        if (account == null) { // Check if the account doesn't exist.
+            log.info("Account with id '{}' not found.", id);
+            throw new ResourceNotFoundException("Account with id '%s' not found.".formatted(id));
+        }
+        if (account.isBanned()) {
+            throw new BadRequestException("Account with id '%s' is banned.".formatted(id));
+        }
+        if (account.isInactive()) {
+            throw new BadRequestException("Account with id '%s' is inactive.".formatted(id));
+        }
+        return account;
+    }
+
+    /**
+     * Gets the scores for the account.
+     *
+     * @param profile The profile.
+     * @param page The page to get the scores from.
+     * @return The scores.
+     */
+    public ScoreSaberScoresPageToken getPageScores(ScoreSaberProfile profile, int page) {
+        log.info("Fetching scores for account '{}' from page {}.", profile.getId(), page);
+        ScoreSaberScoresPageToken pageToken = WebRequest.getAsEntity(String.format(GET_PLAYER_SCORES_ENDPOINT, profile.getId(), "recent", page), ScoreSaberScoresPageToken.class);
+        if (pageToken == null) { // Check if the page doesn't exist.
+            return null;
+        }
+        // Sort the scores by newest time set.
+        pageToken.setPlayerScores(Arrays.stream(pageToken.getPlayerScores())
+                .sorted((a, b) -> DateUtils.getDateFromString(b.getScore().getTimeSet()).compareTo(DateUtils.getDateFromString(a.getScore().getTimeSet())))
+                .toArray(ScoreSaberPlayerScoreToken[]::new));
+        return pageToken;
+    }
+
+    /**
+     * Gets the scores for the account.
+     *
+     * @param profile The profile.
+     * @return The scores.
+     */
+    public List<ScoreSaberScoresPageToken> getScores(ScoreSaberProfile profile) {
+        List<ScoreSaberScoresPageToken> scores = new ArrayList<>(List.of(getPageScores(profile, 1)));
+        ScoreSaberPageMetadataToken metadata = scores.get(0).getMetadata();
+        int totalPages = (int) Math.ceil((double) metadata.getTotal() / metadata.getItemsPerPage());
+        log.info("Fetching {} pages of scores for account '{}'.", totalPages, profile.getId());
+        for (int i = 2; i <= totalPages; i++) {
+            scores.add(getPageScores(profile, i));
+        }
+
+        return scores;
+    }
+}

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

@@ -0,0 +1,54 @@
+package cc.fascinated.bat.service;
+
+import cc.fascinated.bat.model.user.BatUser;
+import cc.fascinated.bat.repository.UserRepository;
+import lombok.NonNull;
+import lombok.extern.log4j.Log4j2;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.DependsOn;
+import org.springframework.stereotype.Service;
+
+import java.util.Optional;
+
+/**
+ * @author Fascinated (fascinated7)
+ */
+@Service @Log4j2
+@DependsOn("discordService")
+public class UserService {
+    /**
+     * The user repository to use
+     */
+    private final UserRepository userRepository;
+
+    @Autowired
+    public UserService(@NonNull UserRepository userRepository) {
+        this.userRepository = userRepository;
+    }
+
+    /**
+     * Gets a user by their ID
+     *
+     * @param id The ID of the user
+     * @return The user
+     */
+    public BatUser getUser(@NonNull String id) {
+        long start = System.currentTimeMillis();
+        Optional<BatUser> optionalUser = userRepository.findById(id);
+        if (optionalUser.isPresent()) {
+            return optionalUser.get();
+        }
+        BatUser user = userRepository.save(new BatUser(id));
+        log.info("Created user \"{}\" in {}ms", id, System.currentTimeMillis() - start);
+        return user;
+    }
+
+    /**
+     * Saves a user
+     *
+     * @param user The user to save
+     */
+    public void saveUser(@NonNull BatUser user) {
+        userRepository.save(user);
+    }
+}