src/main/java/cc/fascinated/bat/command/impl/avatar/AvatarCommand.java

@@ -11,7 +11,7 @@ import org.springframework.stereotype.Component;
  * @author Nick (okNick)
  */
 @Component
-@CommandInfo(name = "avatar", description = "View the avatar of the guild or a user", guildOnly = false)
+@CommandInfo(name = "avatar", description = "View the avatar of the guild or a user")
 public class AvatarCommand extends BatCommand {
     @Autowired
     public AvatarCommand(@NonNull ApplicationContext context) {

src/main/java/cc/fascinated/bat/command/impl/avatar/GuildSubCommand.java

@@ -24,7 +24,7 @@ public class GuildSubCommand extends BatSubCommand {
 
         if (icon == null) {
             interaction.replyEmbeds(EmbedUtils.errorEmbed()
-                            .setDescription("**%s** does not have an avatar!".formatted(guild.getName()))
+                            .setDescription("%s does not have an avatar!".formatted(guild.getName()))
                             .build())
                     .queue();
             return;

src/main/java/cc/fascinated/bat/command/impl/banner/BannerCommand.java

@@ -11,7 +11,7 @@ import org.springframework.stereotype.Component;
  * @author Nick (okNick)
  */
 @Component
-@CommandInfo(name = "banner", description = "View the banner of the guild or a user", guildOnly = false)
+@CommandInfo(name = "banner", description = "View the banner of the guild or a user")
 public class BannerCommand extends BatCommand {
     @Autowired
     public BannerCommand(@NonNull ApplicationContext context) {

src/main/java/cc/fascinated/bat/command/impl/banner/GuildSubCommand.java

@@ -21,9 +21,10 @@ public class GuildSubCommand extends BatSubCommand {
     @Override
     public void execute(BatGuild guild, @NonNull BatUser user, @NonNull MessageChannel channel, Member member, @NonNull SlashCommandInteraction interaction) {
         ImageProxy banner = guild.getDiscordGuild().getBanner();
+
         if (banner == null) {
             interaction.replyEmbeds(EmbedUtils.errorEmbed()
-                            .setDescription("**%s** does not have a banner!".formatted(guild.getName()))
+                            .setDescription("%s does not have a banner!".formatted(guild.getName()))
                             .build())
                     .queue();
             return;

src/main/java/cc/fascinated/bat/command/impl/banner/UserSubCommand.java

@@ -38,9 +38,10 @@ public class UserSubCommand extends BatSubCommand {
 
         User target = userOption.getAsUser();
         ImageProxy banner = target.retrieveProfile().complete().getBanner();
+
         if (banner == null) {
             interaction.replyEmbeds(EmbedUtils.errorEmbed()
-                            .setDescription("**%s** does not have a banner!".formatted(target.getName()))
+                            .setDescription("%s does not have a banner!".formatted(target.getName()))
                             .build())
                     .queue();
             return;

src/main/java/cc/fascinated/bat/command/impl/fun/CatCommand.java

@@ -1,6 +1,7 @@
-package cc.fascinated.bat.command.impl.fun.image;
+package cc.fascinated.bat.command.impl.fun;
 
-import cc.fascinated.bat.command.BatSubCommand;
+import cc.fascinated.bat.command.BatCommand;
+import cc.fascinated.bat.command.Category;
 import cc.fascinated.bat.command.CommandInfo;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.common.WebRequest;
@@ -17,8 +18,8 @@ import org.springframework.stereotype.Component;
  * @author Fascinated (fascinated7)
  */
 @Component
-@CommandInfo(name = "cat", description = "Get a random cat image")
-public class CatSubCommand extends BatSubCommand {
+@CommandInfo(name = "cat", description = "Get a random cat image", category = Category.FUN, guildOnly = false)
+public class CatCommand extends BatCommand {
     @Override
     public void execute(BatGuild guild, @NonNull BatUser user, @NonNull MessageChannel channel, Member member, @NonNull SlashCommandInteraction interaction) {
         CatImageToken[] responseEntity = WebRequest.getAsEntity("https://api.thecatapi.com/v1/images/search", CatImageToken[].class);

src/main/java/cc/fascinated/bat/command/impl/fun/DogCommand.java

@@ -1,6 +1,7 @@
-package cc.fascinated.bat.command.impl.fun.image;
+package cc.fascinated.bat.command.impl.fun;
 
-import cc.fascinated.bat.command.BatSubCommand;
+import cc.fascinated.bat.command.BatCommand;
+import cc.fascinated.bat.command.Category;
 import cc.fascinated.bat.command.CommandInfo;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.common.WebRequest;
@@ -17,8 +18,8 @@ import org.springframework.stereotype.Component;
  * @author Fascinated (fascinated7)
  */
 @Component
-@CommandInfo(name = "dog", description = "Get a random dog image")
-public class DogSubCommand extends BatSubCommand {
+@CommandInfo(name = "dog", description = "Get a random dog image", category = Category.FUN, guildOnly = false)
+public class DogCommand extends BatCommand {
     @Override
     public void execute(BatGuild guild, @NonNull BatUser user, @NonNull MessageChannel channel, Member member, @NonNull SlashCommandInteraction interaction) {
         RandomImage responseEntity = WebRequest.getAsEntity("https://dog.ceo/api/breeds/image/random", RandomImage.class);

src/main/java/cc/fascinated/bat/command/impl/fun/FoxCommand.java

@@ -1,6 +1,7 @@
-package cc.fascinated.bat.command.impl.fun.image;
+package cc.fascinated.bat.command.impl.fun;
 
-import cc.fascinated.bat.command.BatSubCommand;
+import cc.fascinated.bat.command.BatCommand;
+import cc.fascinated.bat.command.Category;
 import cc.fascinated.bat.command.CommandInfo;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.common.WebRequest;
@@ -17,8 +18,8 @@ import org.springframework.stereotype.Component;
  * @author Nick (okNick)
  */
 @Component
-@CommandInfo(name = "fox", description = "Get a random fox image")
-public class FoxSubCommand extends BatSubCommand {
+@CommandInfo(name = "fox", description = "Get a random fox image", category = Category.FUN, guildOnly = false)
+public class FoxCommand extends BatCommand {
     @Override
     public void execute(BatGuild guild, @NonNull BatUser user, @NonNull MessageChannel channel, Member member, @NonNull SlashCommandInteraction interaction) {
         RandomFoxToken responseEntity = WebRequest.getAsEntity("https://randomfox.ca/floof/", RandomFoxToken.class);

src/main/java/cc/fascinated/bat/command/impl/fun/image/ImageCommand.java

@@ -1,23 +0,0 @@
-package cc.fascinated.bat.command.impl.fun.image;
-
-import cc.fascinated.bat.command.BatCommand;
-import cc.fascinated.bat.command.Category;
-import cc.fascinated.bat.command.CommandInfo;
-import lombok.NonNull;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.context.ApplicationContext;
-import org.springframework.stereotype.Component;
-
-/**
- * @author Fascinated (fascinated7)
- */
-@Component
-@CommandInfo(name = "image", description = "View a random image", guildOnly = false, category = Category.FUN)
-public class ImageCommand extends BatCommand {
-    @Autowired
-    public ImageCommand(@NonNull ApplicationContext context) {
-        super.addSubCommand(context.getBean(CatSubCommand.class));
-        super.addSubCommand(context.getBean(DogSubCommand.class));
-        super.addSubCommand(context.getBean(FoxSubCommand.class));
-    }
-}

src/main/java/cc/fascinated/bat/features/spotify/command/CurrentSubCommand.java

@@ -43,7 +43,7 @@ public class CurrentSubCommand extends BatSubCommand {
         }
 
         if (!spotifyService.hasTrackPlaying(user)) {
-            interaction.replyEmbeds(EmbedUtils.errorEmbed()
+            interaction.replyEmbeds(EmbedUtils.genericEmbed()
                             .setDescription("You are not currently playing a track.")
                             .build())
                     .queue();
@@ -55,23 +55,14 @@ public class CurrentSubCommand extends BatSubCommand {
         String trackUrl = "https://open.spotify.com/track/" + track.getId();
         String albumUrl = "https://open.spotify.com/album/" + album.getId();
 
-        StringBuilder artists = new StringBuilder();
-        for (int i = 0; i < track.getArtists().length; i++) {
-            artists.append("**[%s](%s)**".formatted(track.getArtists()[i].getName(), "https://open.spotify.com/artist/" + track.getArtists()[i].getId()));
-            if (i != track.getArtists().length - 1) {
-                artists.append(", ");
-            }
-        }
-
         String description =
                 "➜ Song: **[%s](%s)**\n".formatted(track.getName(), trackUrl) +
-                "➜ Album: **[%s](%s)**\n".formatted(album.getName(), albumUrl) +
-                "➜ Artist%s: %s\n".formatted(track.getArtists().length > 1 ? "s" : "", artists) +
-                "➜ Position: %s\n".formatted(getFormattedTime(currentlyPlaying));
+                        "➜ Album: **[%s](%s)**\n".formatted(album.getName(), albumUrl) +
+                        "➜ Position: %s\n".formatted(getFormattedTime(currentlyPlaying));
 
         Image albumCover = album.getImages()[0];
         interaction.replyEmbeds(EmbedUtils.genericEmbed()
-                .setAuthor("Listening to %s | %s".formatted(track.getName(), track.getArtists()[0].getName()), trackUrl)
+                .setAuthor("Listening to %s".formatted(track.getName()), trackUrl)
                 .setThumbnail(albumCover.getUrl())
                 .setDescription(description)
                 .build()).queue();

src/main/java/cc/fascinated/bat/features/spotify/command/LinkSubCommand.java

@@ -4,7 +4,6 @@ import cc.fascinated.bat.command.BatSubCommand;
 import cc.fascinated.bat.command.CommandInfo;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.event.EventListener;
-import cc.fascinated.bat.features.spotify.profile.SpotifyProfile;
 import cc.fascinated.bat.model.BatGuild;
 import cc.fascinated.bat.model.BatUser;
 import cc.fascinated.bat.service.SpotifyService;
@@ -38,15 +37,6 @@ public class LinkSubCommand extends BatSubCommand implements EventListener {
 
     @Override
     public void execute(BatGuild guild, @NonNull BatUser user, @NonNull MessageChannel channel, Member member, @NonNull SlashCommandInteraction interaction) {
-        SpotifyProfile profile = user.getProfile(SpotifyProfile.class);
-        if (profile.hasLinkedAccount()) {
-            interaction.replyEmbeds(EmbedUtils.errorEmbed()
-                    .setDescription("You have already linked your Spotify account!")
-                    .build())
-                    .queue();
-            return;
-        }
-
         interaction.replyEmbeds(EmbedUtils.genericEmbed()
                         .setDescription("You can link your Spotify account by clicking [here](%s)".formatted(spotifyService.getAuthorizationUrl()))
                 .build())

src/main/java/cc/fascinated/bat/features/spotify/command/PauseSubCommand.java

@@ -37,7 +37,7 @@ public class PauseSubCommand extends BatSubCommand {
         }
 
         if (!spotifyService.hasTrackPlaying(user)) {
-            interaction.replyEmbeds(EmbedUtils.errorEmbed()
+            interaction.replyEmbeds(EmbedUtils.genericEmbed()
                             .setDescription("You need to be playing a track to pause it.")
                             .build())
                     .queue();
@@ -45,7 +45,7 @@ public class PauseSubCommand extends BatSubCommand {
         }
 
         boolean didPause = spotifyService.pausePlayback(user);
-        interaction.replyEmbeds(EmbedUtils.successEmbed()
+        interaction.replyEmbeds(EmbedUtils.genericEmbed()
                         .setDescription(didPause ? "Paused the current track." : "The current track is already paused.")
                         .build())
                 .queue();

src/main/java/cc/fascinated/bat/features/spotify/command/ResumeSubCommand.java

@@ -37,15 +37,15 @@ public class ResumeSubCommand extends BatSubCommand {
         }
 
         if (!spotifyService.hasTrackPlaying(user)) {
-            interaction.replyEmbeds(EmbedUtils.errorEmbed()
-                            .setDescription("You need to be playing a track to resume it.")
+            interaction.replyEmbeds(EmbedUtils.genericEmbed()
+                            .setDescription("You need to be playing a track to pause it.")
                             .build())
                     .queue();
             return;
         }
 
         boolean didPause = spotifyService.resumePlayback(user);
-        interaction.replyEmbeds(EmbedUtils.successEmbed()
+        interaction.replyEmbeds(EmbedUtils.genericEmbed()
                         .setDescription(didPause ? "Resumed the current track." : "The current track is already playing.")
                         .build())
                 .queue();

src/main/java/cc/fascinated/bat/features/spotify/command/UnlinkSubCommand.java

@@ -32,7 +32,7 @@ public class UnlinkSubCommand extends BatSubCommand implements EventListener {
     public void execute(BatGuild guild, @NonNull BatUser user, @NonNull MessageChannel channel, Member member, @NonNull SlashCommandInteraction interaction) {
         SpotifyProfile profile = user.getProfile(SpotifyProfile.class);
         if (!profile.hasLinkedAccount()) {
-            interaction.replyEmbeds(EmbedUtils.errorEmbed()
+            interaction.replyEmbeds(EmbedUtils.genericEmbed()
                             .setDescription("You do not have a linked Spotify account.")
                             .build())
                     .queue();
@@ -41,7 +41,7 @@ public class UnlinkSubCommand extends BatSubCommand implements EventListener {
 
         profile.reset();
         userService.saveUser(user);
-        interaction.replyEmbeds(EmbedUtils.successEmbed()
+        interaction.replyEmbeds(EmbedUtils.genericEmbed()
                         .setDescription("Successfully unlinked your Spotify account.")
                         .build())
                 .queue();

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

@@ -75,9 +75,8 @@ public class CommandService extends ListenerAdapter {
 
         // Unregister all commands that Discord has but we don't
         jda.retrieveCommands().complete().forEach(command -> {
-            if (commands.containsKey(command.getName())
-                    && (commands.get(command.getName()).getCommandInfo().category().isHidden()
-                    || commands.get(command.getName()).getCommandInfo().botOwnerOnly())) {
+            CommandInfo commandInfo = commands.get(command.getName()).getCommandInfo();
+            if (commands.containsKey(command.getName()) && (commandInfo.category().isHidden() || commandInfo.botOwnerOnly())) {
                 jda.deleteCommandById(command.getId()).complete(); // Unregister the command on Discord
                 log.info("Unregistered hidden command \"{}\" from Discord", command.getName());
                 return;
@@ -112,7 +111,7 @@ public class CommandService extends ListenerAdapter {
             log.error("Unable to find the admin guild to register hidden commands");
         }
 
-        log.info("Registered {} slash commands in {}ms", discordCommands.size(), System.currentTimeMillis() - before);
+        log.info("Registered all slash commands in {}ms", System.currentTimeMillis() - before);
     }
 
     /**
@@ -150,9 +149,7 @@ public class CommandService extends ListenerAdapter {
 
         try {
             BatCommandExecutor executor = null;
-            CommandInfo commandInfo = command.getCommandInfo();
             List<Permission> requiredPermissions = new ArrayList<>();
-            boolean isSubCommand = false;
 
             // No args provided, use the main command executor
             if (event.getInteraction().getSubcommandName() == null) {
@@ -165,8 +162,6 @@ public class CommandService extends ListenerAdapter {
                         executor = subCommand.getValue();
                         requiredPermissions.addAll(Arrays.asList(subCommand.getValue().getCommandInfo().requiredPermissions()));
                         requiredPermissions.addAll(Arrays.asList(command.getCommandInfo().requiredPermissions())); // not sure if we'd want this, but it's here for now
-                        isSubCommand = true;
-                        commandInfo = subCommand.getValue().getCommandInfo();
                         break;
                     }
                 }
@@ -201,14 +196,6 @@ public class CommandService extends ListenerAdapter {
                     return;
                 }
             }
-
-            if (isSubCommand && commandInfo.guildOnly() && !ranInsideGuild) {
-                event.replyEmbeds(EmbedUtils.errorEmbed()
-                        .setDescription("This command can only be executed in a guild")
-                        .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/SpotifyService.java

@@ -6,8 +6,9 @@ import cc.fascinated.bat.model.BatUser;
 import lombok.Getter;
 import lombok.NonNull;
 import lombok.SneakyThrows;
-import lombok.extern.log4j.Log4j2;
 import net.jodah.expiringmap.ExpiringMap;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import se.michaelthelin.spotify.SpotifyApi;
@@ -24,8 +25,8 @@ import java.util.concurrent.TimeUnit;
  */
 @Service
 @Getter
-@Log4j2
 public class SpotifyService {
+    private static final Logger log = LoggerFactory.getLogger(SpotifyService.class);
     /**
      * The access token map.
      */
@@ -107,8 +108,12 @@ public class SpotifyService {
      */
     @SneakyThrows
     public boolean pausePlayback(BatUser user) {
-        getSpotifyApi(user).pauseUsersPlayback().build().execute();
-        return true;
+        try {
+            getSpotifyApi(user).pauseUsersPlayback().build().execute();
+            return true;
+        } catch (Exception e) {
+            return false;
+        }
     }
 
     /**
@@ -119,8 +124,12 @@ public class SpotifyService {
      */
     @SneakyThrows
     public boolean resumePlayback(BatUser user) {
-        getSpotifyApi(user).startResumeUsersPlayback().build().execute();
-        return true;
+        try {
+            getSpotifyApi(user).startResumeUsersPlayback().build().execute();
+            return true;
+        } catch (Exception e) {
+            return false;
+        }
     }
 
     /**
@@ -145,7 +154,7 @@ public class SpotifyService {
      * Links the user's Spotify account with their Discord account.
      *
      * @param user the user to link the account with
-     * @param key  the key to link the account with
+     * @param key the key to link the account with
      */
     public void linkAccount(BatUser user, String key) {
         AuthorizationCodeCredentials credentials = accessToken.get(key);
@@ -156,7 +165,6 @@ public class SpotifyService {
         SpotifyProfile profile = user.getProfile(SpotifyProfile.class);
         profile.setAccessToken(credentials.getAccessToken());
         profile.setRefreshToken(credentials.getRefreshToken());
-        profile.setExpiresAt(System.currentTimeMillis() + (credentials.getExpiresIn() * 1000));
         userService.saveUser(user);
     }
 
@@ -168,27 +176,24 @@ public class SpotifyService {
     @SneakyThrows
     public SpotifyApi getSpotifyApi(BatUser user) {
         SpotifyProfile profile = user.getProfile(SpotifyProfile.class);
-        ensureValidToken(profile, user);
-        return new SpotifyApi.Builder()
+        SpotifyApi api = new SpotifyApi.Builder()
+                .setClientId(clientId)
+                .setClientSecret(clientSecret)
                 .setAccessToken(profile.getAccessToken())
                 .setRefreshToken(profile.getRefreshToken())
                 .build();
-    }
 
-    /**
-     * Ensures the user has a valid Spotify access token.
-     *
-     * @param user the user to get the token for
-     */
-    @SneakyThrows
-    public void ensureValidToken(SpotifyProfile profile, BatUser user) {
-        if (profile.getExpiresAt() == null || profile.getExpiresAt() > System.currentTimeMillis()) {
-            return;
+        // Refresh the access token if it's expired
+        if (profile.getExpiresAt() == null || profile.getExpiresAt() < System.currentTimeMillis()) {
+            AuthorizationCodeCredentials credentials = api.authorizationCodeRefresh().build().execute();
+            profile.setAccessToken(credentials.getAccessToken());
+            profile.setRefreshToken(credentials.getRefreshToken());
+            profile.setExpiresAt(System.currentTimeMillis() + (credentials.getExpiresIn() * 1000));
+            api.setAccessToken(credentials.getAccessToken());
+            api.setRefreshToken(credentials.getRefreshToken());
+            userService.saveUser(user);
+            log.info("Refreshed spotify access token for user {}", user.getName());
         }
-        AuthorizationCodeCredentials credentials = spotifyApi.authorizationCodeRefresh().build().execute();
-        profile.setAccessToken(credentials.getAccessToken());
-        profile.setRefreshToken(credentials.getRefreshToken());
-        profile.setExpiresAt(System.currentTimeMillis() + (credentials.getExpiresIn() * 1000));
-        userService.saveUser(user);
+        return api;
     }
 }