Implemented automatic post-closing upon user leaving

Author: jasonlessenich

src/main/java/net/javadiscord/javabot/listener/UserLeaveListener.java

@@ -1,20 +1,33 @@
 package net.javadiscord.javabot.listener;
 
+import lombok.extern.slf4j.Slf4j;
 import net.dv8tion.jda.api.entities.Guild;
 import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.entities.UserSnowflake;
+import net.dv8tion.jda.api.entities.channel.concrete.ForumChannel;
+import net.dv8tion.jda.api.entities.channel.concrete.ThreadChannel;
 import net.dv8tion.jda.api.events.guild.member.GuildMemberRemoveEvent;
 import net.dv8tion.jda.api.hooks.ListenerAdapter;
 import net.javadiscord.javabot.data.config.BotConfig;
 
 import lombok.RequiredArgsConstructor;
+import net.javadiscord.javabot.data.config.guild.HelpConfig;
+import net.javadiscord.javabot.data.h2db.DbActions;
+import net.javadiscord.javabot.systems.help.HelpManager;
+import net.javadiscord.javabot.systems.help.dao.HelpAccountRepository;
+import net.javadiscord.javabot.systems.help.dao.HelpTransactionRepository;
 import org.jetbrains.annotations.NotNull;
 
 /**
  * Listens for the {@link GuildMemberRemoveEvent}.
  */
+@Slf4j
 @RequiredArgsConstructor
 public class UserLeaveListener extends ListenerAdapter {
+	private final DbActions dbActions;
 	private final BotConfig botConfig;
+	private final HelpAccountRepository helpAccountRepository;
+	private final HelpTransactionRepository helpTransactionRepository;
 
 	@Override
 	public void onGuildMemberRemove(@NotNull GuildMemberRemoveEvent event) {
@@ -31,6 +44,18 @@ public void onGuildMemberRemove(@NotNull GuildMemberRemoveEvent event) {
 	 * @param guild The guild they're leaving.
 	 */
 	private void unreserveAllChannels(User user, Guild guild) {
-		// TODO: Implement Forum
+		HelpConfig config = botConfig.get(guild).getHelpConfig();
+		ForumChannel forum = config.getHelpForumChannel();
+		if (forum != null) {
+			for (ThreadChannel post : forum.getThreadChannels()) {
+				if (post.isArchived() || post.isLocked()) continue;
+				if (post.getOwnerIdLong() == user.getIdLong()) {
+					HelpManager manager = new HelpManager(post, dbActions, botConfig, helpAccountRepository, helpTransactionRepository);
+					manager.close(UserSnowflake.fromId(guild.getSelfMember().getIdLong()), "User left the server");
+				}
+			}
+		} else {
+			log.warn("Could not find forum channel for guild {}", guild.getName());
+		}
 	}
 }

src/main/java/net/javadiscord/javabot/systems/help/HelpManager.java

@@ -2,10 +2,13 @@
 
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
+import net.dv8tion.jda.api.EmbedBuilder;
 import net.dv8tion.jda.api.entities.Guild;
 import net.dv8tion.jda.api.entities.Member;
 import net.dv8tion.jda.api.entities.Message;
+import net.dv8tion.jda.api.entities.MessageEmbed;
 import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.entities.UserSnowflake;
 import net.dv8tion.jda.api.entities.channel.concrete.ThreadChannel;
 import net.dv8tion.jda.api.entities.emoji.Emoji;
 import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
@@ -107,6 +110,17 @@ public void close(IReplyCallback callback, boolean withHelpers, @Nullable String
 				.queue(s -> postThread.getManager().setLocked(true).setArchived(true).queue());
 	}
 
+	/**
+	 * Closes the {@link HelpManager#postThread}.
+	 *
+	 * @param closedBy    The {@link UserSnowflake} that closed this post.
+	 * @param reason      The reason for closing this post.
+	 */
+	public void close(UserSnowflake closedBy, @Nullable String reason) {
+		postThread.sendMessageEmbeds(buildPostClosedEmbed(closedBy, reason))
+				.queue(s -> postThread.getManager().setLocked(true).setArchived(true).queue());
+	}
+
 	/**
 	 * Thanks a single user.
 	 *
@@ -154,6 +168,14 @@ public boolean isForumEligibleToBeUnreserved(@NotNull Interaction interaction) {
 				hasMemberHelperRole(interaction.getGuild(), interaction.getMember()) || hasMemberStaffRole(interaction.getGuild(), interaction.getMember());
 	}
 
+	private @NotNull MessageEmbed buildPostClosedEmbed(@NotNull UserSnowflake closedBy, @Nullable String reason) {
+		return new EmbedBuilder()
+				.setTitle("Post Closed")
+				.setDescription("This post has been closed by %s%s".formatted(closedBy.getAsMention(), reason != null ? " for the following reason:\n> " + reason : "."))
+				.setColor(Responses.Type.INFO.getColor())
+				.build();
+	}
+
 	private boolean hasMemberStaffRole(@NotNull Guild guild, @Nullable Member member) {
 		return member != null && member.getRoles().contains(botConfig.get(guild).getModerationConfig().getStaffRole());
 	}