add afk feature

src/main/java/cc/fascinated/bat/features/afk/AfkFeature.java

@@ -0,0 +1,20 @@
+package cc.fascinated.bat.features.afk;
+
+import cc.fascinated.bat.features.Feature;
+import cc.fascinated.bat.features.afk.command.AfkCommand;
+import cc.fascinated.bat.service.CommandService;
+import lombok.NonNull;
+import org.springframework.context.ApplicationContext;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author Fascinated (fascinated7)
+ */
+@Component
+public class AfkFeature extends Feature {
+    public AfkFeature(@NonNull ApplicationContext context, @NonNull CommandService commandService) {
+        super("AFK", Category.GENERAL);
+
+        commandService.registerCommand(context.getBean(AfkCommand.class));
+    }
+}

src/main/java/cc/fascinated/bat/features/afk/AfkMentionListener.java

@@ -0,0 +1,35 @@
+package cc.fascinated.bat.features.afk;
+
+import cc.fascinated.bat.event.EventListener;
+import cc.fascinated.bat.features.afk.profile.AfkProfile;
+import cc.fascinated.bat.model.BatGuild;
+import cc.fascinated.bat.model.BatUser;
+import lombok.NonNull;
+import net.dv8tion.jda.api.entities.Message;
+import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+
+/**
+ * @author Fascinated (fascinated7)
+ */
+@Component
+public class AfkMentionListener implements EventListener {
+    @Override
+    public void onGuildMessageReceive(@NonNull BatGuild guild, @NonNull BatUser user, @NonNull MessageReceivedEvent event) {
+        Message message = event.getMessage();
+        List<User> mentionedUsers = message.getMentions().getUsers();
+        if (mentionedUsers.isEmpty()) {
+            return;
+        }
+        User mentionedUser = mentionedUsers.get(0);
+        AfkProfile profile = guild.getProfile(AfkProfile.class);
+        if (!profile.isAfk(mentionedUser.getId())) {
+            return;
+        }
+
+        event.getMessage().reply("%s is currently AFK: %s".formatted(mentionedUser.getAsMention(), profile.getAfkReason(mentionedUser.getId()))).queue();
+    }
+}

src/main/java/cc/fascinated/bat/features/afk/AfkReturnListener.java

@@ -0,0 +1,25 @@
+package cc.fascinated.bat.features.afk;
+
+import cc.fascinated.bat.event.EventListener;
+import cc.fascinated.bat.features.afk.profile.AfkProfile;
+import cc.fascinated.bat.model.BatGuild;
+import cc.fascinated.bat.model.BatUser;
+import lombok.NonNull;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author Fascinated (fascinated7)
+ */
+@Component
+public class AfkReturnListener implements EventListener {
+    @Override
+    public void onGuildMessageReceive(@NonNull BatGuild guild, @NonNull BatUser user, @NonNull MessageReceivedEvent event) {
+        AfkProfile profile = guild.getProfile(AfkProfile.class);
+        if (!profile.isAfk(user.getId())) {
+            return;
+        }
+        profile.removeAfkUser(guild, user.getId());
+        event.getMessage().reply("Welcome back, %s! You are no longer AFK.".formatted(user.getDiscordUser().getAsMention())).queue();
+    }
+}

src/main/java/cc/fascinated/bat/features/afk/command/AfkCommand.java

@@ -0,0 +1,42 @@
+package cc.fascinated.bat.features.afk.command;
+
+import cc.fascinated.bat.command.BatCommand;
+import cc.fascinated.bat.common.MemberUtils;
+import cc.fascinated.bat.features.afk.profile.AfkProfile;
+import cc.fascinated.bat.model.BatGuild;
+import cc.fascinated.bat.model.BatUser;
+import lombok.NonNull;
+import net.dv8tion.jda.api.entities.Member;
+import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel;
+import net.dv8tion.jda.api.interactions.commands.OptionMapping;
+import net.dv8tion.jda.api.interactions.commands.OptionType;
+import net.dv8tion.jda.api.interactions.commands.SlashCommandInteraction;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author Fascinated (fascinated7)
+ */
+@Component
+public class AfkCommand extends BatCommand {
+    public AfkCommand() {
+        super("afk", "Sets your AFK status");
+        super.addOption(OptionType.STRING, "reason", "The reason for being AFK", false);
+    }
+
+    @Override
+    public void execute(BatGuild guild, @NonNull BatUser user, @NonNull MessageChannel channel, Member member, @NonNull SlashCommandInteraction interaction) {
+        AfkProfile profile = guild.getProfile(AfkProfile.class);
+        String reason = null;
+        OptionMapping reasonOption = interaction.getOption("reason");
+        if (reasonOption != null) {
+            reason = reasonOption.getAsString();
+        }
+
+        profile.addAfkUser(guild, member.getId(), reason);
+        interaction.reply("You are now AFK: %s%s".formatted(
+                profile.getAfkReason(member.getId()),
+                MemberUtils.hasPermissionToEdit(guild, user) ? "" :
+                        "\n\n*I do not have enough permissions to edit your user, and therefore cannot update your nickname*"
+        )).queue();
+    }
+}

src/main/java/cc/fascinated/bat/features/afk/profile/AfkProfile.java

@@ -0,0 +1,99 @@
+package cc.fascinated.bat.features.afk.profile;
+
+import cc.fascinated.bat.common.Profile;
+import cc.fascinated.bat.model.BatGuild;
+import net.dv8tion.jda.api.entities.Guild;
+import net.dv8tion.jda.api.entities.Member;
+import org.springframework.stereotype.Component;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * @author Fascinated (fascinated7)
+ */
+@Component
+public class AfkProfile extends Profile {
+    private static final String DEFAULT_REASON = "Away";
+
+    /**
+     * The AFK users in this guild
+     */
+    private Map<String, String> afkUsers;
+
+    /**
+     * Adds a user to the AFK list
+     *
+     * @param guild the guild enable afk mode for
+     * @param userId the user ID to add
+     * @param reason the reason for being AFK
+     */
+    public void addAfkUser(BatGuild guild, String userId, String reason) {
+        if (afkUsers == null) {
+            afkUsers = new HashMap<>();
+        }
+        afkUsers.put(userId, reason == null ? DEFAULT_REASON : reason);
+
+        Guild discordGuild = guild.getDiscordGuild();
+        Member member = discordGuild.getMemberById(userId);
+        if (member == null) {
+            return;
+        }
+        try {
+            member.modifyNickname("[AFK] " + member.getEffectiveName()).queue();
+        } catch (Exception ignored) {}
+    }
+
+    /**
+     * Removes a user from the AFK list
+     *
+     * @param guild the guild to remove the user from
+     * @param userId the user ID to remove
+     */
+    public void removeAfkUser(BatGuild guild, String userId) {
+        if (afkUsers == null) {
+            afkUsers = new HashMap<>();
+        }
+        afkUsers.remove(userId);
+
+        Guild discordGuild = guild.getDiscordGuild();
+        Member member = discordGuild.getMemberById(userId);
+        if (member == null) {
+            return;
+        }
+        try {
+            member.modifyNickname(member.getEffectiveName().replace("[AFK] ", "")).queue();
+        } catch (Exception ignored) {}
+    }
+
+    /**
+     * Gets the reason for being AFK
+     *
+     * @param userId the user ID to get the reason for
+     * @return the reason for being AFK
+     */
+    public String getAfkReason(String userId) {
+        if (afkUsers == null) {
+            afkUsers = new HashMap<>();
+        }
+        return afkUsers.get(userId);
+    }
+
+    /**
+     * Checks if a user is AFK
+     *
+     * @param userId the user ID to check
+     * @return if the user is AFK
+     */
+    public boolean isAfk(String userId) {
+        if (afkUsers == null) {
+            afkUsers = new HashMap<>();
+        }
+        return afkUsers.containsKey(userId);
+    }
+
+    @Override
+    public void reset() {
+        afkUsers = new HashMap<>();
+    }
+}