Feat/delete messages upto 24 hrs

application/src/main/java/org/togetherjava/tjbot/features/Features.java

@@ -31,6 +31,9 @@
 import org.togetherjava.tjbot.features.moderation.audit.AuditCommand;
 import org.togetherjava.tjbot.features.moderation.audit.ModAuditLogRoutine;
 import org.togetherjava.tjbot.features.moderation.audit.ModAuditLogWriter;
+import org.togetherjava.tjbot.features.moderation.history.MessageHistoryRoutine;
+import org.togetherjava.tjbot.features.moderation.history.MessageListener;
+import org.togetherjava.tjbot.features.moderation.history.PurgeHistoryCommand;
 import org.togetherjava.tjbot.features.moderation.modmail.ModMailCommand;
 import org.togetherjava.tjbot.features.moderation.scam.ScamBlocker;
 import org.togetherjava.tjbot.features.moderation.scam.ScamHistoryPurgeRoutine;
@@ -106,6 +109,7 @@ public static Collection<Feature> createFeatures(JDA jda, Database database, Con
             .add(new AutoPruneHelperRoutine(config, helpSystemHelper, modAuditLogWriter, database));
         features.add(new HelpThreadAutoArchiver(helpSystemHelper));
         features.add(new LeftoverBookmarksCleanupRoutine(bookmarksSystem));
+        features.add(new MessageHistoryRoutine(database));
 
         // Message receivers
         features.add(new TopHelpersMessageListener(database, config));
@@ -119,6 +123,7 @@ public static Collection<Feature> createFeatures(JDA jda, Database database, Con
         features.add(new CodeMessageManualDetection(codeMessageHandler));
         features.add(new SlashCommandEducator());
         features.add(new PinnedNotificationRemover(config));
+        features.add(new MessageListener(database));
 
         // Event receivers
         features.add(new RejoinModerationRoleListener(actionsStore, config));
@@ -159,6 +164,7 @@ public static Collection<Feature> createFeatures(JDA jda, Database database, Con
         features.add(new BookmarksCommand(bookmarksSystem));
         features.add(new ChatGptCommand(chatGptService));
         features.add(new JShellCommand(jshellEval));
+        features.add(new PurgeHistoryCommand(database));
 
         FeatureBlacklist<Class<?>> blacklist = blacklistConfig.normal();
         return features.stream().filter(f -> blacklist.isEnabled(f.getClass())).toList();

application/src/main/java/org/togetherjava/tjbot/features/moderation/history/MessageHistoryRoutine.java

@@ -0,0 +1,47 @@
+package org.togetherjava.tjbot.features.moderation.history;
+
+import static org.togetherjava.tjbot.db.generated.Tables.MESSAGE_HISTORY;
+
+import java.time.Instant;
+import java.time.temporal.ChronoUnit;
+import java.util.concurrent.TimeUnit;
+import net.dv8tion.jda.api.JDA;
+import org.jooq.impl.UpdatableRecordImpl;
+import org.togetherjava.tjbot.db.Database;
+import org.togetherjava.tjbot.features.Routine;
+
+/**
+ * Routine that records some metadata for every message received.
+ */
+public class MessageHistoryRoutine implements Routine {
+
+    private static final int SCHEDULE_INTERVAL_SECONDS = 30;
+    private final Database database;
+
+    /**
+     * Creates a new instance.
+     *
+     * @param database the database that records some metadata on each message received.
+     */
+    public MessageHistoryRoutine(Database database) {
+        this.database = database;
+    }
+
+
+    @Override
+    public Schedule createSchedule() {
+        return new Schedule(ScheduleMode.FIXED_RATE, 0, SCHEDULE_INTERVAL_SECONDS,
+                TimeUnit.SECONDS);
+    }
+
+    @Override
+    public void runRoutine(JDA jda) {
+        Instant now = Instant.now();
+        Instant twoHourBack = now.minus(2, ChronoUnit.HOURS);
+
+        database.write(context -> context.selectFrom(MESSAGE_HISTORY)
+            .where(MESSAGE_HISTORY.SENT_AT.lessThan(twoHourBack))
+            .stream()
+            .forEach(UpdatableRecordImpl::delete));
+    }
+}

application/src/main/java/org/togetherjava/tjbot/features/moderation/history/MessageListener.java

@@ -0,0 +1,46 @@
+package org.togetherjava.tjbot.features.moderation.history;
+
+import static org.togetherjava.tjbot.db.generated.Tables.MESSAGE_HISTORY;
+
+import java.time.Instant;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import org.togetherjava.tjbot.db.Database;
+import org.togetherjava.tjbot.features.MessageReceiverAdapter;
+
+/**
+ * Listens for new message throughout the guild, then stores some metadata for each message in a
+ * database.
+ */
+public class MessageListener extends MessageReceiverAdapter {
+    private final Database database;
+
+    /**
+     * Creates a new instance.
+     *
+     * @param database this database to record some metadata for each message received throughout
+     *        guild.
+     */
+    public MessageListener(Database database) {
+        this.database = database;
+    }
+
+    @Override
+    public void onMessageReceived(MessageReceivedEvent event) {
+        updateHistory(event);
+    }
+
+    private void updateHistory(MessageReceivedEvent event) {
+        long guildId = event.getGuild().getIdLong();
+        long channelId = event.getChannel().getIdLong();
+        long messageId = event.getMessageIdLong();
+        long authorId = event.getAuthor().getIdLong();
+
+        database.write(context -> context.newRecord(MESSAGE_HISTORY)
+            .setSentAt(Instant.now())
+            .setGuildId(guildId)
+            .setChannelId(channelId)
+            .setMessageId(messageId)
+            .setAuthorId(authorId)
+            .insert());
+    }
+}

application/src/main/java/org/togetherjava/tjbot/features/moderation/history/PurgeHistoryCommand.java

@@ -0,0 +1,123 @@
+package org.togetherjava.tjbot.features.moderation.history;
+
+import static org.togetherjava.tjbot.db.generated.Tables.MESSAGE_HISTORY;
+
+import java.time.Instant;
+import java.time.temporal.ChronoUnit;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Objects;
+import net.dv8tion.jda.api.JDA;
+import net.dv8tion.jda.api.entities.Member;
+import net.dv8tion.jda.api.entities.Role;
+import net.dv8tion.jda.api.entities.User;
+import net.dv8tion.jda.api.entities.channel.unions.MessageChannelUnion;
+import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
+import net.dv8tion.jda.api.interactions.InteractionHook;
+import net.dv8tion.jda.api.interactions.commands.OptionMapping;
+import net.dv8tion.jda.api.interactions.commands.OptionType;
+import org.togetherjava.tjbot.db.Database;
+import org.togetherjava.tjbot.features.CommandVisibility;
+import org.togetherjava.tjbot.features.SlashCommandAdapter;
+
+public class PurgeHistoryCommand extends SlashCommandAdapter {
+
+    private static final String USER_OPTION = "user";
+    private static final String REASON_OPTION = "reason";
+    private static final String COMMAND_NAME = "purge";
+    private final Database database;
+
+    /**
+     * Creates an instance of the command.
+     *
+     * @param database the database to retrieve message records of a target user
+     */
+
+    public PurgeHistoryCommand(Database database) {
+        super(COMMAND_NAME, "delete user message history upto an hour", CommandVisibility.GUILD);
+        getData()
+            .addOption(OptionType.USER, USER_OPTION,
+                    "user who's messages you want to purge within an hour", true)
+            .addOption(OptionType.STRING, REASON_OPTION, "Reason for purging user messages", true);
+        this.database = database;
+    }
+
+    @Override
+    public void onSlashCommand(SlashCommandInteractionEvent event) {
+        event.deferReply().queue();
+
+        OptionMapping targetOption =
+                Objects.requireNonNull(event.getOption(USER_OPTION), "target is null");
+        Member author = Objects.requireNonNull(event.getMember(), "author is null");
+        String reason = Objects.requireNonNull(event.getOption(REASON_OPTION).getAsString(),
+                "reason is null");
+
+        handleHistory(event, author, reason, targetOption, event.getHook());
+    }
+
+    private void handleHistory(SlashCommandInteractionEvent event, Member author, String reason,
+            OptionMapping targetOption, InteractionHook hook) {
+        Instant now = Instant.now();
+        Instant oneHourBack = now.minus(1, ChronoUnit.HOURS);
+
+        User target = targetOption.getAsUser();
+        String sourceChannelId = event.getChannel().getId();
+
+        if (!validateHierarchy(author, targetOption)) {
+            hook.sendMessage("Target user has a higher role than you").setEphemeral(true).queue();
+            return;
+        }
+
+        List<String> messageIdsForDeletion = new ArrayList<>();
+
+        database.write(context -> context.selectFrom(MESSAGE_HISTORY)
+            .where(MESSAGE_HISTORY.AUTHOR_ID.equal(target.getIdLong())
+                .and(MESSAGE_HISTORY.CHANNEL_ID.equal(Long.valueOf(sourceChannelId)))
+                .and(MESSAGE_HISTORY.SENT_AT.greaterOrEqual(oneHourBack)))
+            .stream()
+            .forEach(messageHistoryRecord -> {
+                messageIdsForDeletion.add(String.valueOf(messageHistoryRecord.getMessageId()));
+                messageHistoryRecord.delete();
+            }));
+
+        handleDelete(messageIdsForDeletion, event.getJDA(), event.getChannel(), event.getHook());
+    }
+
+    private void handleDelete(List<String> messageIdsForDeletion, JDA jda,
+            MessageChannelUnion channel, InteractionHook hook) {
+
+        if (messageIdsForDeletion.isEmpty()) {
+            handleEmptyMessageHistory(hook);
+            return;
+        }
+
+        if (hasSingleElement(messageIdsForDeletion)) {
+            String messageId = messageIdsForDeletion.get(0);
+
+            Objects.requireNonNull(jda.getTextChannelById(channel.getId()), "channel was not found")
+                .deleteMessageById(messageId)
+                .queue();
+        }
+
+        channel.purgeMessagesById(messageIdsForDeletion);
+        hook.sendMessage("Messages purged from user").setEphemeral(true).queue();
+    }
+
+    private boolean hasSingleElement(List<String> messageIdsForDeletion) {
+        return messageIdsForDeletion.size() == 1;
+    }
+
+    private void handleEmptyMessageHistory(InteractionHook hook) {
+        hook.sendMessage("User has no message history in this channel within last one hour")
+            .setEphemeral(true)
+            .queue();
+    }
+
+    private boolean validateHierarchy(Member author, OptionMapping target) {
+        int highestRole = 0;
+        Role targetUserRole = target.getAsMember().getRoles().get(highestRole);
+        Role authorRole = author.getRoles().get(highestRole);
+
+        return targetUserRole.getPosition() >= authorRole.getPosition();
+    }
+}

application/src/main/java/org/togetherjava/tjbot/features/moderation/history/package-info.java

@@ -0,0 +1,9 @@
+/**
+ * This package offers moderation command to delete messages from a user within last hour.
+ */
+@MethodsReturnNonnullByDefault
+@ParametersAreNonnullByDefault
+package org.togetherjava.tjbot.features.moderation.history;
+
+import javax.annotation.ParametersAreNonnullByDefault;
+import org.togetherjava.tjbot.annotations.MethodsReturnNonnullByDefault;

application/src/main/resources/db/V14__Add_Message_History.sql

@@ -0,0 +1,9 @@
+CREATE TABLE message_history
+(
+    id           INTEGER   NOT NULL PRIMARY KEY AUTOINCREMENT,
+    sent_at      TIMESTAMP NOT NULL,
+    guild_id     BIGINT    NOT NULL,
+    channel_id   BIGINT    NOT NULL,
+    message_id   BIGINT    NOT NULL,
+    author_id    BIGINT    NOT NULL
+)