Add broadcast detection

HaHaWTH · Mar 3, 2024 · e699776 · e699776
1 parent 2a57d17
commit e699776
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 0 deletions.
diff --git a/src/main/java/io/wdsj/asw/AdvancedSensitiveWords.java b/src/main/java/io/wdsj/asw/AdvancedSensitiveWords.java
@@ -114,6 +114,9 @@ public void onEnable() {
         if (settingsManager.getProperty(PluginSettings.ENABLE_PLAYER_NAME_CHECK)) {
             getServer().getPluginManager().registerEvents(new PlayerLoginListener(), this);
         }
+        if (settingsManager.getProperty(PluginSettings.CHAT_BROADCAST_CHECK)) {
+            getServer().getPluginManager().registerEvents(new BroadCastListener(), this);
+        }
         long endTime = System.currentTimeMillis();
         getLogger().info("AdvancedSensitiveWords is enabled!(took " + (endTime - startTime) + "ms)");
         // bro, don't bytecode this, you can just disable it in the config TAT

diff --git a/src/main/java/io/wdsj/asw/listener/BroadCastListener.java b/src/main/java/io/wdsj/asw/listener/BroadCastListener.java
@@ -0,0 +1,39 @@
+package io.wdsj.asw.listener;
+
+import io.wdsj.asw.setting.PluginSettings;
+import io.wdsj.asw.util.Utils;
+import org.bukkit.event.EventHandler;
+import org.bukkit.event.EventPriority;
+import org.bukkit.event.Listener;
+import org.bukkit.event.server.BroadcastMessageEvent;
+
+import java.util.List;
+
+import static io.wdsj.asw.AdvancedSensitiveWords.*;
+import static io.wdsj.asw.util.TimingUtils.addProcessStatistic;
+import static io.wdsj.asw.util.Utils.messagesFilteredNum;
+
+public class BroadCastListener implements Listener {
+
+    @EventHandler(priority = EventPriority.LOWEST)
+    public void onBroadCast(BroadcastMessageEvent event) {
+        if (!isInitialized || !settingsManager.getProperty(PluginSettings.CHAT_BROADCAST_CHECK)) return;
+        String originalMessage = event.getMessage();
+        long startTime = System.currentTimeMillis();
+        List<String> censoredWordList = sensitiveWordBs.findAll(originalMessage);
+        if (!censoredWordList.isEmpty()) {
+            messagesFilteredNum.getAndIncrement();
+            String processedMessage = sensitiveWordBs.replace(originalMessage);
+            if (settingsManager.getProperty(PluginSettings.CHAT_METHOD).equalsIgnoreCase("cancel")) {
+                event.setCancelled(true);
+            } else {
+                event.setMessage(processedMessage);
+            }
+            if (settingsManager.getProperty(PluginSettings.LOG_VIOLATION)) {
+                Utils.logViolation("服务器广播消息(IP: 无)(Chat)(BroadCast)", originalMessage + censoredWordList);
+            }
+            long endTime = System.currentTimeMillis();
+            addProcessStatistic(endTime, startTime);
+        }
+    }
+}
diff --git a/src/main/java/io/wdsj/asw/setting/PluginSettings.java b/src/main/java/io/wdsj/asw/setting/PluginSettings.java
@@ -83,6 +83,8 @@ public class PluginSettings implements SettingsHolder {
     public static final Property<Boolean> CHAT_FAKE_MESSAGE_ON_CANCEL = newProperty("Chat.fakeMessageOnCancel", false);
     @Comment("是否发送消息提醒(和假消息冲突)")
     public static final Property<Boolean> CHAT_SEND_MESSAGE = newProperty("Chat.sendMessage", true);
+    @Comment("*是否启用服务器广播消息检测(仅提供取消和替换模式,配置跟随聊天检测)(不会触发API事件)")
+    public static final Property<Boolean> CHAT_BROADCAST_CHECK = newProperty("Chat.broadcastCheck", true);
     @Comment("是否开启聊天上下文检测(仅提供取消和假消息模式,配置跟随聊天检测)")
     public static final Property<Boolean> CHAT_CONTEXT_CHECK = newProperty("Chat.contextCheck", false);
     @Comment("最大检测上下文大小")