round 2 of crazy new feature with a global manager setup for listening to a message sent

Author: ayyhimself

src/main/java/net/discordjug/javabot/data/config/GuildConfig.java

@@ -18,8 +18,10 @@
 import java.lang.reflect.Field;
 import java.nio.file.Files;
 import java.nio.file.Path;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Optional;
+import java.util.Set;
 
 /**
  * A collection of guild-specific configuration items, each of which represents
@@ -56,8 +58,7 @@ public GuildConfig(Guild guild, Path file) {
 		this.starboardConfig = new StarboardConfig();
 		this.messageCacheConfig = new MessageCacheConfig();
 		this.serverLockConfig = new ServerLockConfig();
-		this.blacklistedMessageExtensions = List.of("application/x-msdos-program", "application/java-archive",
-				"application/zip");
+		this.blacklistedMessageExtensions = List.of("jar", "exe", "zip");
 		this.setGuild(guild);
 	}
 

src/main/java/net/discordjug/javabot/listener/BlacklistedMessageAttachmentListener.java

@@ -49,7 +49,7 @@ public void onMessageReceived(@NotNull MessageReceivedEvent event) {
 									List.of(ActionRow.of(
 											Button.link(m.getJumpUrl(), "Jump to Message"),
 											Button.secondary(InteractionUtils.createDeleteInteractionId(event.getAuthor().getIdLong(), m.getAuthor().getIdLong()), "\uD83D\uDDD1️"))),
-									null));
+									null, null));
 					}, e -> ExceptionLogger.capture(e, getClass().getSimpleName())));
 			}
 		}

src/main/java/net/discordjug/javabot/listener/HugListener.java

@@ -0,0 +1,35 @@
+package net.discordjug.javabot.listener.handler;
+
+import lombok.RequiredArgsConstructor;
+import net.discordjug.javabot.data.config.BotConfig;
+import net.discordjug.javabot.data.config.GuildConfig;
+import net.dv8tion.jda.api.EmbedBuilder;
+import net.dv8tion.jda.api.entities.Message;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+
+@Component
+@RequiredArgsConstructor
+public class BlacklistedMessageAttachmentFilter implements MessageFormatFilter {
+
+    private final BotConfig botConfig;
+
+    @Override
+    public boolean processMessage(MessageReceivedEvent event, StringBuilder builder, List<Message.Attachment> attachments, List<MessageEmbed> embeds) {
+        GuildConfig guildConfig = botConfig.get(event.getGuild());
+        List<String> blacklistedMessageExtensions = guildConfig.getBlacklistedMessageExtensions();
+        boolean removed = attachments.removeIf(attachment -> blacklistedMessageExtensions.contains(attachment.getFileExtension()));
+        if (removed) {
+            MessageEmbed attachmentRemovedInfo = new EmbedBuilder()
+                    .setDescription("Disallowed attachments have been removed from this message.")
+                    .build();
+            embeds.add(attachmentRemovedInfo);
+            return true;
+        } else {
+            return false;
+        }
+    }
+}

src/main/java/net/discordjug/javabot/listener/MessageLinkListener.java

@@ -0,0 +1,73 @@
+package net.discordjug.javabot.listener.handler;
+
+import net.dv8tion.jda.api.entities.Message;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.Objects;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+@Component
+public class HugFilter implements MessageFormatFilter {
+
+    private static final Pattern FUCKER = Pattern.compile(
+            "(fuck)(ing|er|ed|k+)?",
+            Pattern.CASE_INSENSITIVE
+    );
+
+    @Override
+    public boolean processMessage(MessageReceivedEvent event, StringBuilder builder, List<Message.Attachment> attachments, List<MessageEmbed> embeds) {
+        replaceFucks(builder);
+        return true;
+    }
+
+    public static void replaceFucks(StringBuilder sb) {
+        Matcher matcher = FUCKER.matcher(sb);
+        int offset = 0;
+
+        while (matcher.find()) {
+            String theFuck = matcher.group(1);
+            String suffix = Objects.requireNonNullElse(matcher.group(2), "");
+
+            String processedSuffix = switch (suffix.toLowerCase()) {
+                case "er", "ed", "ing" -> copyCase(suffix, 0, 'g') + suffix;
+                case "" -> "";
+                default -> copyCase(suffix, "g".repeat(suffix.length()));
+            };
+
+            String replacement = processHug(theFuck) + processedSuffix;
+
+            int start = matcher.start() + offset;
+            int end = matcher.end() + offset;
+
+            sb.replace(start, end, replacement);
+
+            offset += replacement.length() - (end - start);
+        }
+    }
+
+    private static String processHug(String originalText) {
+        // FucK -> HuG, FuCk -> Hug, etc.
+        return String.valueOf(copyCase(originalText, 0, 'h')) +
+                copyCase(originalText, 1, 'u') +
+                copyCase(originalText, 3, 'g');
+    }
+
+    private static String copyCase(String source, String toChange) {
+        if (source.length() != toChange.length()) {
+            throw new IllegalArgumentException("lengths differ");
+        }
+        StringBuilder sb = new StringBuilder();
+        for (int i = 0; i < source.length(); i++) {
+            sb.append(copyCase(source, i, toChange.charAt(i)));
+        }
+        return sb.toString();
+    }
+
+    private static char copyCase(String original, int index, char newChar) {
+        return Character.isUpperCase(original.charAt(index)) ? Character.toUpperCase(newChar) : Character.toLowerCase(newChar);
+    }
+}

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

@@ -0,0 +1,11 @@
+package net.discordjug.javabot.listener.handler;
+
+import net.dv8tion.jda.api.entities.Message;
+import net.dv8tion.jda.api.entities.MessageEmbed;
+import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
+
+import java.util.List;
+
+public interface MessageFormatFilter {
+    boolean processMessage(MessageReceivedEvent event, StringBuilder builder, List<Message.Attachment> attachments, List<MessageEmbed> embeds);
+}