fix message

Reviewed-on: Fascinated/Bat#1

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")
+@CommandInfo(name = "avatar", description = "View the avatar of the guild or a user", guildOnly = false)
 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")
+@CommandInfo(name = "banner", description = "View the banner of the guild or a user", guildOnly = false)
 public class BannerCommand extends BatCommand {
     @Autowired
     public BannerCommand(@NonNull ApplicationContext context) {

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

@@ -21,10 +21,9 @@ 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,10 +38,9 @@ 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/image/CatSubCommand.java

@@ -1,7 +1,6 @@
-package cc.fascinated.bat.command.impl.fun;
+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.BatSubCommand;
 import cc.fascinated.bat.command.CommandInfo;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.common.WebRequest;
@@ -18,8 +17,8 @@ import org.springframework.stereotype.Component;
  * @author Fascinated (fascinated7)
  */
 @Component
-@CommandInfo(name = "cat", description = "Get a random cat image", category = Category.FUN, guildOnly = false)
-public class CatCommand extends BatCommand {
+@CommandInfo(name = "cat", description = "Get a random cat image")
+public class CatSubCommand extends BatSubCommand {
     @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/image/DogSubCommand.java

@@ -1,7 +1,6 @@
-package cc.fascinated.bat.command.impl.fun;
+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.BatSubCommand;
 import cc.fascinated.bat.command.CommandInfo;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.common.WebRequest;
@@ -18,8 +17,8 @@ import org.springframework.stereotype.Component;
  * @author Fascinated (fascinated7)
  */
 @Component
-@CommandInfo(name = "dog", description = "Get a random dog image", category = Category.FUN, guildOnly = false)
-public class DogCommand extends BatCommand {
+@CommandInfo(name = "dog", description = "Get a random dog image")
+public class DogSubCommand extends BatSubCommand {
     @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/image/FoxSubCommand.java

@@ -1,7 +1,6 @@
-package cc.fascinated.bat.command.impl.fun;
+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.BatSubCommand;
 import cc.fascinated.bat.command.CommandInfo;
 import cc.fascinated.bat.common.EmbedUtils;
 import cc.fascinated.bat.common.WebRequest;
@@ -18,8 +17,8 @@ import org.springframework.stereotype.Component;
  * @author Nick (okNick)
  */
 @Component
-@CommandInfo(name = "fox", description = "Get a random fox image", category = Category.FUN, guildOnly = false)
-public class FoxCommand extends BatCommand {
+@CommandInfo(name = "fox", description = "Get a random fox image")
+public class FoxSubCommand extends BatSubCommand {
     @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

@@ -0,0 +1,23 @@
+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.genericEmbed()
+            interaction.replyEmbeds(EmbedUtils.errorEmbed()
                             .setDescription("You are not currently playing a track.")
                             .build())
                     .queue();
@@ -55,14 +55,23 @@ 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) +
-                        "➜ Position: %s\n".formatted(getFormattedTime(currentlyPlaying));
+                "➜ 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));
 
         Image albumCover = album.getImages()[0];
         interaction.replyEmbeds(EmbedUtils.genericEmbed()
-                .setAuthor("Listening to %s".formatted(track.getName()), trackUrl)
+                .setAuthor("Listening to %s | %s".formatted(track.getName(), track.getArtists()[0].getName()), trackUrl)
                 .setThumbnail(albumCover.getUrl())
                 .setDescription(description)
                 .build()).queue();

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

@@ -4,6 +4,7 @@ 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;
@@ -37,6 +38,15 @@ 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.genericEmbed()
+            interaction.replyEmbeds(EmbedUtils.errorEmbed()
                             .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.genericEmbed()
+        interaction.replyEmbeds(EmbedUtils.successEmbed()
                         .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.genericEmbed()
-                            .setDescription("You need to be playing a track to pause it.")
+            interaction.replyEmbeds(EmbedUtils.errorEmbed()
+                            .setDescription("You need to be playing a track to resume it.")
                             .build())
                     .queue();
             return;
         }
 
         boolean didPause = spotifyService.resumePlayback(user);
-        interaction.replyEmbeds(EmbedUtils.genericEmbed()
+        interaction.replyEmbeds(EmbedUtils.successEmbed()
                         .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.genericEmbed()
+            interaction.replyEmbeds(EmbedUtils.errorEmbed()
                             .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.genericEmbed()
+        interaction.replyEmbeds(EmbedUtils.successEmbed()
                         .setDescription("Successfully unlinked your Spotify account.")
                         .build())
                 .queue();

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

@@ -75,8 +75,9 @@ public class CommandService extends ListenerAdapter {
 
         // Unregister all commands that Discord has but we don't
         jda.retrieveCommands().complete().forEach(command -> {
-            CommandInfo commandInfo = commands.get(command.getName()).getCommandInfo();
-            if (commands.containsKey(command.getName()) && (commandInfo.category().isHidden() || commandInfo.botOwnerOnly())) {
+            if (commands.containsKey(command.getName())
+                    && (commands.get(command.getName()).getCommandInfo().category().isHidden()
+                    || commands.get(command.getName()).getCommandInfo().botOwnerOnly())) {
                 jda.deleteCommandById(command.getId()).complete(); // Unregister the command on Discord
                 log.info("Unregistered hidden command \"{}\" from Discord", command.getName());
                 return;
@@ -111,7 +112,7 @@ public class CommandService extends ListenerAdapter {
             log.error("Unable to find the admin guild to register hidden commands");
         }
 
-        log.info("Registered all slash commands in {}ms", System.currentTimeMillis() - before);
+        log.info("Registered {} slash commands in {}ms", discordCommands.size(), System.currentTimeMillis() - before);
     }
 
     /**
@@ -149,7 +150,9 @@ 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) {
@@ -162,6 +165,8 @@ 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;
                     }
                 }
@@ -196,6 +201,14 @@ 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,9 +6,8 @@ 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;
@@ -25,8 +24,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.
      */
@@ -108,12 +107,8 @@ public class SpotifyService {
      */
     @SneakyThrows
     public boolean pausePlayback(BatUser user) {
-        try {
-            getSpotifyApi(user).pauseUsersPlayback().build().execute();
-            return true;
-        } catch (Exception e) {
-            return false;
-        }
+        getSpotifyApi(user).pauseUsersPlayback().build().execute();
+        return true;
     }
 
     /**
@@ -124,12 +119,8 @@ public class SpotifyService {
      */
     @SneakyThrows
     public boolean resumePlayback(BatUser user) {
-        try {
-            getSpotifyApi(user).startResumeUsersPlayback().build().execute();
-            return true;
-        } catch (Exception e) {
-            return false;
-        }
+        getSpotifyApi(user).startResumeUsersPlayback().build().execute();
+        return true;
     }
 
     /**
@@ -154,7 +145,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);
@@ -165,6 +156,7 @@ 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);
     }
 
@@ -176,24 +168,27 @@ public class SpotifyService {
     @SneakyThrows
     public SpotifyApi getSpotifyApi(BatUser user) {
         SpotifyProfile profile = user.getProfile(SpotifyProfile.class);
-        SpotifyApi api = new SpotifyApi.Builder()
-                .setClientId(clientId)
-                .setClientSecret(clientSecret)
+        ensureValidToken(profile, user);
+        return new SpotifyApi.Builder()
                 .setAccessToken(profile.getAccessToken())
                 .setRefreshToken(profile.getRefreshToken())
                 .build();
+    }
 
-        // 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());
+    /**
+     * 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;
         }
-        return api;
+        AuthorizationCodeCredentials credentials = spotifyApi.authorizationCodeRefresh().build().execute();
+        profile.setAccessToken(credentials.getAccessToken());
+        profile.setRefreshToken(credentials.getRefreshToken());
+        profile.setExpiresAt(System.currentTimeMillis() + (credentials.getExpiresIn() * 1000));
+        userService.saveUser(user);
     }
 }