Rewrite /afk backend to be guild specific

Author: TechnoVisionDev

src/main/java/technobot/commands/utility/AfkCommand.java

@@ -4,7 +4,6 @@
 import net.dv8tion.jda.api.interactions.commands.OptionMapping;
 import net.dv8tion.jda.api.interactions.commands.OptionType;
 import net.dv8tion.jda.api.interactions.commands.build.OptionData;
-import net.dv8tion.jda.internal.utils.tuple.Pair;
 import technobot.TechnoBot;
 import technobot.commands.Category;
 import technobot.commands.Command;
@@ -30,7 +29,7 @@ public AfkCommand(TechnoBot bot) {
     public void execute(SlashCommandInteractionEvent event) {
         OptionMapping option = event.getOption("message");
         String message = option != null ? option.getAsString() : "";
-        AfkListener.AFK_MESSAGES.put(event.getUser(), Pair.of(message, new Date().toInstant()));
-        event.reply(":wave: | See you later!").queue();
+        AfkListener.AFK_MESSAGES.put(event.getMember(), new AfkListener.AfkStatus(message, new Date().toInstant()));
+        event.reply(":wave: | **"+event.getMember().getEffectiveName()+"** has gone AFK. See you later!").queue();
     }
 }

src/main/java/technobot/listeners/AfkListener.java

@@ -1,10 +1,9 @@
 package technobot.listeners;
 
 import net.dv8tion.jda.api.EmbedBuilder;
-import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.entities.Member;
 import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
 import net.dv8tion.jda.api.hooks.ListenerAdapter;
-import net.dv8tion.jda.internal.utils.tuple.Pair;
 import org.jetbrains.annotations.NotNull;
 import technobot.util.embeds.EmbedColor;
 
@@ -20,7 +19,7 @@ public class AfkListener extends ListenerAdapter {
 
 
     /** Map of User objects to a pair containing the AFK message to be sent and the time created */
-    public static HashMap<User, Pair<String, Instant>> AFK_MESSAGES = new HashMap<>();
+    public static HashMap<Member, AfkStatus> AFK_MESSAGES = new HashMap<>();
 
     /**
      * Sends AFK messages if necessary.
@@ -30,26 +29,33 @@ public class AfkListener extends ListenerAdapter {
     @Override
     public void onMessageReceived(@NotNull MessageReceivedEvent event) {
         // Check if AFK user has returned
-        User messageAuthor = event.getMember().getUser();
-        if (AFK_MESSAGES.containsKey(messageAuthor)) {
-            AFK_MESSAGES.remove(messageAuthor);
+        if (AFK_MESSAGES.containsKey(event.getMember())) {
+            AFK_MESSAGES.remove(event.getMember());
             event.getMessage().addReaction("\uD83D\uDC4B").queue();
             return;
         }
 
         // Check mentions against AFK users
-        for (User user : event.getMessage().getMentions().getUsers()) {
-            Pair<String, Instant> message = AFK_MESSAGES.get(user);
-            if (message != null) {
+        for (Member member : event.getMessage().getMentions().getMembers()) {
+            AfkStatus status = AFK_MESSAGES.get(member);
+            if (status != null) {
                 // Mentioned user was AFK -- send message response
                 EmbedBuilder embed = new EmbedBuilder()
                         .setColor(EmbedColor.DEFAULT.color)
-                        .setAuthor(user.getName()+" is currently AFK", null, user.getEffectiveAvatarUrl())
-                        .setDescription(message.getLeft())
-                        .setTimestamp(message.getRight());
+                        .setAuthor(member.getEffectiveName()+" is currently AFK", null, member.getEffectiveAvatarUrl())
+                        .setDescription(status.message())
+                        .setTimestamp(status.timestamp());
                 event.getChannel().sendMessageEmbeds(embed.build()).queue();
                 return;
             }
         }
     }
+
+    /**
+     * represents an AFK status message and timestamp.
+     *
+     * @param message the afk message to be sent.
+     * @param timestamp the time the user when afk.
+     */
+    public record AfkStatus(String message, Instant timestamp) {}
 }