seppuku/src/main/java/me/rigamortis/seppuku/impl/module/misc/ChatFilterModule.java

package me.rigamortis.seppuku.impl.module.misc;

import com.mojang.realmsclient.gui.ChatFormatting;
import me.rigamortis.seppuku.api.event.EventStageable;
import me.rigamortis.seppuku.api.event.network.EventReceivePacket;
import me.rigamortis.seppuku.api.module.Module;
import me.rigamortis.seppuku.api.util.StringUtil;
import me.rigamortis.seppuku.api.value.Value;
import net.minecraft.client.Minecraft;
import net.minecraft.network.play.server.SPacketChat;
import net.minecraft.util.text.TextComponentString;
import team.stiff.pomelo.impl.annotated.handler.annotation.Listener;

import java.util.ArrayList;
import java.util.List;

/**
 * Author Seth
 * 5/25/2019 @ 4:34 AM.
 */
public final class ChatFilterModule extends Module {

    public final Value<Boolean> unicode = new Value<>("Unicode", new String[]{"uc"}, "Reverts \"Fancy Chat\" characters back into normal ones. ", true);
    public final Value<Boolean> broadcasts = new Value<>("Broadcasts", new String[]{"broadcast", "broad", "bc"}, "Prevents displaying chat messages that begin with [SERVER].", false);
    public final Value<Boolean> russian = new Value<>("Russian", new String[]{"russiantext", "rus", "r"}, "Prevents displaying russian-character containing messages.", false);
    public final Value<Boolean> asian = new Value<>("Asian", new String[]{"asiantext", "asia", "chinese", "japanese", "korean"}, "Prevents displaying \"CJK Unified Ideograph\"-character containing messages (chinese, korean, jap...).", false);
    public final Value<Boolean> spam = new Value<>("Spam", new String[]{"sp", "s"}, "Attempts to prevent spam by checking recent chat messages for duplicates.", true);
    public final Value<Boolean> death = new Value<>("Death", new String[]{"dead", "d"}, "Attempts to prevent death messages.", false);
    public final Value<Boolean> blue = new Value<>("BlueText", new String[]{"Blue", "b"}, "Cancels blue-text containing messages.", false);
    public final Value<Boolean> green = new Value<>("GreenText", new String[]{"Green", "g"}, "Cancels green-text containing messages.", false);

    private final List<String> cache = new ArrayList<>();

    public ChatFilterModule() {
        super("ChatFilter", new String[]{"CFilter"}, "Filters out annoying chat messages", "NONE", -1, ModuleType.MISC);
    }

    @Listener
    public void onDisable() {
        super.onDisable();
        this.cache.clear();
    }

    @Listener
    public void receivePacket(EventReceivePacket event) {
        if (event.getStage() == EventStageable.EventStage.PRE) {
            if (event.getPacket() instanceof SPacketChat) {
                final SPacketChat packet = (SPacketChat) event.getPacket();
                boolean is9b9tOr2b2t = false;

                if (!Minecraft.getMinecraft().isSingleplayer()) {
                    if (Minecraft.getMinecraft().getCurrentServerData() != null) {
                        final String currentServerIP = Minecraft.getMinecraft().getCurrentServerData().serverIP;
                        is9b9tOr2b2t = currentServerIP.equalsIgnoreCase("2b2t.org") || currentServerIP.equalsIgnoreCase("2b2t.com") || currentServerIP.equalsIgnoreCase("9b9t.com") || currentServerIP.equalsIgnoreCase("9b9t.org");
                    }
                }

                if (is9b9tOr2b2t) {
                    if (this.death.getValue()) {
                        if (packet.getChatComponent().getFormattedText().contains("\2474") || packet.getChatComponent().getFormattedText().contains("\247c")) {
                            event.setCanceled(true);
                        }
                    }

                    if (this.broadcasts.getValue()) {
                        if (packet.getChatComponent().getFormattedText().startsWith("\2475[SERVER]")) {
                            event.setCanceled(true);
                        }

                        if (packet.getChatComponent().getFormattedText().contains("\2472")) {
                            event.setCanceled(true);
                        }
                    }

                    if (this.spam.getValue()) {
                        final String chat = packet.getChatComponent().getUnformattedText();

                        if (this.cache.size() > 0) {
                            for (String s : this.cache) {
                                final double diff = StringUtil.levenshteinDistance(s, chat);

                                if (diff >= 0.75f) {
                                    event.setCanceled(true);
                                }
                            }
                        }

                        this.cache.add(chat);

                        if (this.cache.size() >= 10) {
                            this.cache.remove(0);
                        }
                    }

                    if (this.blue.getValue()) {
                        if (packet.getChatComponent().getFormattedText().contains(ChatFormatting.BLUE + "")) {
                            event.setCanceled(true);
                        }
                    }

                    if (this.green.getValue()) {
                        if (packet.getChatComponent().getFormattedText().contains(ChatFormatting.GREEN + "")) {
                            event.setCanceled(true);
                        }
                    }
                }

                if (this.russian.getValue()) {
                    for (int i = 0; i < packet.getChatComponent().getFormattedText().length(); i++) {
                        if (Character.UnicodeBlock.of(packet.getChatComponent().getFormattedText().charAt(i)).equals(Character.UnicodeBlock.CYRILLIC)) {
                            event.setCanceled(true);
                        }
                    }
                }

                if (this.asian.getValue()) {
                    for (int i = 0; i < packet.getChatComponent().getFormattedText().length(); i++) {
                        final Character.UnicodeBlock block = Character.UnicodeBlock.of(packet.getChatComponent().getFormattedText().charAt(i));

                        if (block.equals(Character.UnicodeBlock.CJK_UNIFIED_IDEOGRAPHS) ||
                                block.equals(Character.UnicodeBlock.CJK_UNIFIED_IDEOGRAPHS_EXTENSION_A) ||
                                block.equals(Character.UnicodeBlock.CJK_UNIFIED_IDEOGRAPHS_EXTENSION_B) ||
                                block.equals(Character.UnicodeBlock.CJK_UNIFIED_IDEOGRAPHS_EXTENSION_C) ||
                                block.equals(Character.UnicodeBlock.CJK_UNIFIED_IDEOGRAPHS_EXTENSION_D)) {
                            event.setCanceled(true);
                        }
                    }
                }

                if (this.unicode.getValue()) {
                    if (packet.getChatComponent() instanceof TextComponentString) {
                        final TextComponentString component = (TextComponentString) packet.getChatComponent();

                        final StringBuilder sb = new StringBuilder();

                        boolean containsUnicode = false;

                        for (String s : component.getFormattedText().split(" ")) {
                            StringBuilder line = new StringBuilder();
                            for (char c : s.toCharArray()) {
                                if (c >= 0xFEE0) {
                                    c -= 0xFEE0;
                                    containsUnicode = true;
                                }
                                line.append(c);
                            }
                            sb.append(line).append(" ");
                        }

                        if (containsUnicode) {
                            packet.chatComponent = new TextComponentString(sb.toString());
                        }
                    }
                }
            }
        }
    }

}