Skip to content

Commit a01678c

Browse files
committed
Handle messages replaced by all msg filters, not only the attachment one
1 parent 9c2ee35 commit a01678c

File tree

2 files changed

+9
-8
lines changed

2 files changed

+9
-8
lines changed

src/main/java/net/discordjug/javabot/listener/filter/BlacklistedMessageAttachmentFilter.java

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
import lombok.RequiredArgsConstructor;
88
import net.discordjug.javabot.data.config.BotConfig;
99
import net.discordjug.javabot.data.config.GuildConfig;
10-
import net.discordjug.javabot.data.h2db.message_cache.MessageCache;
1110
import net.dv8tion.jda.api.EmbedBuilder;
1211
import net.dv8tion.jda.api.entities.Message;
1312
import net.dv8tion.jda.api.entities.MessageEmbed;
@@ -22,7 +21,6 @@
2221
public class BlacklistedMessageAttachmentFilter implements MessageFilter {
2322

2423
private final BotConfig botConfig;
25-
private final MessageCache cache;
2624

2725
@Override
2826
public MessageModificationStatus processMessage(MessageContent content) {
@@ -37,8 +35,6 @@ public MessageModificationStatus processMessage(MessageContent content) {
3735
MessageEmbed attachmentRemovedInfo = new EmbedBuilder()
3836
.setDescription("Disallowed attachments have been removed from this message.").build();
3937
embeds.add(attachmentRemovedInfo);
40-
cache.blockedMessages.add(event.getMessageIdLong());
41-
cache.sendBlacklistedAttachmentsMessageToLog(event.getGuild(), event.getChannel(), event.getMessage());
4238
return MessageModificationStatus.MODIFIED;
4339
} else {
4440
return MessageModificationStatus.NOT_MODIFIED;

src/main/java/net/discordjug/javabot/listener/filter/MessageFilterHandler.java

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,13 @@
11
package net.discordjug.javabot.listener.filter;
22

3+
import java.util.ArrayList;
4+
import java.util.List;
5+
6+
import org.jetbrains.annotations.NotNull;
7+
38
import lombok.RequiredArgsConstructor;
49
import net.discordjug.javabot.data.config.BotConfig;
10+
import net.discordjug.javabot.data.h2db.message_cache.MessageCache;
511
import net.discordjug.javabot.systems.moderation.AutoMod;
612
import net.discordjug.javabot.util.ExceptionLogger;
713
import net.discordjug.javabot.util.WebhookUtil;
@@ -10,10 +16,6 @@
1016
import net.dv8tion.jda.api.entities.channel.middleman.StandardGuildChannel;
1117
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
1218
import net.dv8tion.jda.api.hooks.ListenerAdapter;
13-
import org.jetbrains.annotations.NotNull;
14-
15-
import java.util.ArrayList;
16-
import java.util.List;
1719

1820
/**
1921
* This class is responsible for calling {@link MessageFilter}s on incoming messages and optionally replacing the message.
@@ -30,6 +32,7 @@ public class MessageFilterHandler extends ListenerAdapter {
3032
private final List<MessageFilter> filters;
3133
private final AutoMod autoMod;
3234
private final BotConfig botConfig;
35+
private final MessageCache cache;
3336

3437
@Override
3538
public void onMessageReceived(@NotNull MessageReceivedEvent event) {
@@ -56,6 +59,8 @@ public void onMessageReceived(@NotNull MessageReceivedEvent event) {
5659
}
5760

5861
if (handled) {
62+
cache.blockedMessages.add(event.getMessageIdLong());
63+
cache.sendBlacklistedAttachmentsMessageToLog(event.getGuild(), event.getChannel(), event.getMessage());
5964
IWebhookContainer webhookContainer = null;
6065
long threadId = 0;
6166
if (event.isFromType(ChannelType.TEXT)) {

0 commit comments

Comments
 (0)