rename group
This commit is contained in:
ChomeNS
380
src/main/java/me/chayapak1/chomens_bot/plugins/ChatPlugin.java
Normal file
380
src/main/java/me/chayapak1/chomens_bot/plugins/ChatPlugin.java
Normal file
@@ -0,0 +1,380 @@
|
||||
package me.chayapak1.chomens_bot.plugins;
|
||||
|
||||
import com.github.steveice10.mc.protocol.packet.ingame.clientbound.ClientboundDisguisedChatPacket;
|
||||
import com.github.steveice10.mc.protocol.packet.ingame.clientbound.ClientboundPlayerChatPacket;
|
||||
import com.github.steveice10.mc.protocol.packet.ingame.clientbound.ClientboundSystemChatPacket;
|
||||
import com.github.steveice10.mc.protocol.packet.ingame.serverbound.ServerboundChatCommandPacket;
|
||||
import com.github.steveice10.mc.protocol.packet.ingame.serverbound.ServerboundChatPacket;
|
||||
import com.github.steveice10.packetlib.Session;
|
||||
import com.github.steveice10.packetlib.packet.Packet;
|
||||
import me.chayapak1.chomens_bot.Bot;
|
||||
import me.chayapak1.chomens_bot.chatParsers.CreayunChatParser;
|
||||
import me.chayapak1.chomens_bot.chatParsers.KaboomChatParser;
|
||||
import me.chayapak1.chomens_bot.chatParsers.MinecraftChatParser;
|
||||
import me.chayapak1.chomens_bot.chatParsers.U203aChatParser;
|
||||
import me.chayapak1.chomens_bot.data.chat.ChatParser;
|
||||
import me.chayapak1.chomens_bot.data.PlayerEntry;
|
||||
import me.chayapak1.chomens_bot.data.chat.PlayerMessage;
|
||||
import me.chayapak1.chomens_bot.util.ComponentUtilities;
|
||||
import me.chayapak1.chomens_bot.util.IllegalCharactersUtilities;
|
||||
import me.chayapak1.chomens_bot.util.UUIDUtilities;
|
||||
import net.kyori.adventure.text.Component;
|
||||
import net.kyori.adventure.text.TextComponent;
|
||||
import net.kyori.adventure.text.TranslatableComponent;
|
||||
import net.kyori.adventure.text.serializer.gson.GsonComponentSerializer;
|
||||
|
||||
import java.time.Instant;
|
||||
import java.util.*;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
public class ChatPlugin extends Bot.Listener {
|
||||
public final Pattern CHAT_SPLIT_PATTERN;
|
||||
public static final Pattern COLOR_CODE_PATTERN = Pattern.compile("(&[a-f0-9rlonmk])", Pattern.MULTILINE);
|
||||
public static final Pattern COLOR_CODE_END_PATTERN = Pattern.compile("^.*&[a-f0-9rlonmk]$", Pattern.MULTILINE);
|
||||
|
||||
private final Bot bot;
|
||||
|
||||
private final List<ChatParser> chatParsers;
|
||||
|
||||
private final List<String> queue = new ArrayList<>();
|
||||
|
||||
public final int queueDelay;
|
||||
|
||||
private final List<Listener> listeners = new ArrayList<>();
|
||||
|
||||
public ChatPlugin (Bot bot) {
|
||||
this.bot = bot;
|
||||
this.CHAT_SPLIT_PATTERN = Pattern.compile("\\G\\s*([^\\r\\n]{1," + (bot.options.creayun ? 126 : 254) + "}(?=\\s|$)|[^\\r\\n]{254})"); // thanks HBot for the regex <3
|
||||
|
||||
queueDelay = bot.options.chatQueueDelay;
|
||||
|
||||
bot.addListener(this);
|
||||
|
||||
chatParsers = new ArrayList<>();
|
||||
chatParsers.add(new MinecraftChatParser(bot));
|
||||
chatParsers.add(new KaboomChatParser(bot));
|
||||
chatParsers.add(new U203aChatParser(bot));
|
||||
chatParsers.add(new CreayunChatParser(bot));
|
||||
|
||||
bot.executor.scheduleAtFixedRate(this::sendChatTick, 0, queueDelay, TimeUnit.MILLISECONDS);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void packetReceived (Session session, Packet packet) {
|
||||
if (packet instanceof ClientboundSystemChatPacket) packetReceived((ClientboundSystemChatPacket) packet);
|
||||
else if (packet instanceof ClientboundPlayerChatPacket) packetReceived((ClientboundPlayerChatPacket) packet);
|
||||
else if (packet instanceof ClientboundDisguisedChatPacket) packetReceived((ClientboundDisguisedChatPacket) packet);
|
||||
}
|
||||
|
||||
public void packetReceived (ClientboundSystemChatPacket packet) {
|
||||
final Component component = packet.getContent();
|
||||
|
||||
if (
|
||||
component instanceof TextComponent t_component &&
|
||||
t_component.content().length() > 15_000
|
||||
) return;
|
||||
|
||||
if (component instanceof TranslatableComponent t_component) {
|
||||
final String key = t_component.key();
|
||||
|
||||
if (
|
||||
key.equals("advMode.setCommand.success") ||
|
||||
key.equals("advMode.notAllowed")
|
||||
) return;
|
||||
}
|
||||
|
||||
PlayerMessage playerMessage = null;
|
||||
|
||||
for (ChatParser parser : chatParsers) {
|
||||
playerMessage = parser.parse(component);
|
||||
if (playerMessage != null) break;
|
||||
}
|
||||
|
||||
final String string = ComponentUtilities.stringify(component);
|
||||
final String ansi = ComponentUtilities.stringifyAnsi(component);
|
||||
|
||||
for (Listener listener : listeners) {
|
||||
final boolean bool1 = listener.systemMessageReceived(component, string, ansi);
|
||||
|
||||
if (!bool1) break;
|
||||
|
||||
if (playerMessage != null) {
|
||||
final boolean bool2 = listener.playerMessageReceived(playerMessage);
|
||||
|
||||
if (!bool2) break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private String getTranslationByChatType (int chatType) {
|
||||
String translation = null;
|
||||
|
||||
// maybe use the registry in the login packet? too lazy.,.,,.
|
||||
switch (chatType) {
|
||||
case 0 -> translation = "chat.type.text"; // normal vanilla chat message
|
||||
case 1 -> translation = "chat.type.emote"; // /me
|
||||
case 2 -> translation = "commands.message.display.incoming"; // player that received /w message
|
||||
case 3 -> translation = "commands.message.display.outgoing"; // player that sent /w message
|
||||
case 4 -> translation = "%s"; // idk what this is but when the kaboom chat it uses this chat type
|
||||
case 5 -> translation = "chat.type.announcement"; // /say
|
||||
case 6 -> translation = "chat.type.team.text"; // player that received /teammsg message
|
||||
case 7 -> translation = "chat.type.team.sent"; // player that sent /teammsg message
|
||||
}
|
||||
|
||||
return translation;
|
||||
}
|
||||
|
||||
public void packetReceived (ClientboundPlayerChatPacket packet) {
|
||||
final UUID senderUUID = packet.getSender();
|
||||
|
||||
final PlayerEntry entry = bot.players.getEntry(senderUUID);
|
||||
|
||||
if (entry == null) return;
|
||||
|
||||
final PlayerMessage playerMessage = new PlayerMessage(
|
||||
entry,
|
||||
packet.getName(),
|
||||
Component.text(packet.getContent())
|
||||
);
|
||||
|
||||
final Component unsignedContent = packet.getUnsignedContent();
|
||||
|
||||
final String translation = getTranslationByChatType(packet.getChatType());
|
||||
|
||||
for (Listener listener : listeners) {
|
||||
final boolean bool = listener.playerMessageReceived(playerMessage);
|
||||
|
||||
if (!bool) break;
|
||||
|
||||
if (translation != null && unsignedContent == null) {
|
||||
TranslatableComponent component = Component.translatable(translation);
|
||||
|
||||
if (translation.equals("chat.type.team.text") || translation.equals("chat.type.team.sent")) { // ohio
|
||||
component = component.args(
|
||||
Component.translatable("chat.square_brackets").args(bot.team.teamsByPlayer.get(bot.profile.getName()).displayName),
|
||||
playerMessage.displayName,
|
||||
playerMessage.contents
|
||||
);
|
||||
} else {
|
||||
component = component.args(playerMessage.displayName, playerMessage.contents);
|
||||
}
|
||||
|
||||
final String string = ComponentUtilities.stringify(component);
|
||||
final String ansi = ComponentUtilities.stringifyAnsi(component);
|
||||
|
||||
final boolean _bool = listener.systemMessageReceived(component, string, ansi);
|
||||
|
||||
if (!_bool) break;
|
||||
} else {
|
||||
final String string = ComponentUtilities.stringify(unsignedContent);
|
||||
final String ansi = ComponentUtilities.stringifyAnsi(unsignedContent);
|
||||
|
||||
final boolean _bool = listener.systemMessageReceived(unsignedContent, string, ansi);
|
||||
|
||||
if (!_bool) break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void packetReceived (ClientboundDisguisedChatPacket packet) {
|
||||
try {
|
||||
final String translation = getTranslationByChatType(packet.getChatType());
|
||||
|
||||
final Component component = packet.getMessage();
|
||||
|
||||
PlayerMessage parsedFromMessage = null;
|
||||
|
||||
for (ChatParser parser : chatParsers) {
|
||||
parsedFromMessage = parser.parse(component);
|
||||
if (parsedFromMessage != null) break;
|
||||
}
|
||||
|
||||
if (translation != null && parsedFromMessage == null) {
|
||||
final Component name = packet.getName();
|
||||
final Component content = packet.getMessage();
|
||||
|
||||
TranslatableComponent translatableComponent = Component.translatable(translation);
|
||||
|
||||
if (translation.equals("chat.type.team.text") || translation.equals("chat.type.team.sent")) { // ohio
|
||||
translatableComponent = translatableComponent.args(
|
||||
Component.translatable("chat.square_brackets").args(bot.team.teamsByPlayer.get(bot.profile.getName()).displayName),
|
||||
name,
|
||||
content
|
||||
);
|
||||
} else {
|
||||
translatableComponent = translatableComponent.args(name, content);
|
||||
}
|
||||
|
||||
final String string = ComponentUtilities.stringify(translatableComponent);
|
||||
final String ansi = ComponentUtilities.stringifyAnsi(translatableComponent);
|
||||
|
||||
for (Listener listener : listeners) {
|
||||
final boolean bool = listener.systemMessageReceived(translatableComponent, string, ansi);
|
||||
|
||||
if (!bool) break;
|
||||
}
|
||||
|
||||
for (ChatParser parser : chatParsers) {
|
||||
final PlayerMessage parsed = parser.parse(translatableComponent);
|
||||
|
||||
if (parsed == null) continue;
|
||||
|
||||
final PlayerMessage playerMessage = new PlayerMessage(parsed.sender, packet.getName(), parsed.contents);
|
||||
|
||||
for (Listener listener : listeners) {
|
||||
final boolean bool = listener.playerMessageReceived(playerMessage);
|
||||
|
||||
if (!bool) break;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if (parsedFromMessage == null) return;
|
||||
|
||||
final PlayerMessage playerMessage = new PlayerMessage(parsedFromMessage.sender, packet.getName(), parsedFromMessage.contents);
|
||||
|
||||
final String string = ComponentUtilities.stringify(component);
|
||||
final String ansi = ComponentUtilities.stringifyAnsi(component);
|
||||
|
||||
for (Listener listener : listeners) {
|
||||
final boolean bool1 = listener.playerMessageReceived(playerMessage);
|
||||
|
||||
if (!bool1) break;
|
||||
|
||||
final boolean bool2 = listener.systemMessageReceived(component, string, ansi);
|
||||
|
||||
if (!bool2) break;
|
||||
}
|
||||
}
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
private void sendChatTick () {
|
||||
if (queue.size() > 100) queue.clear(); // detects spam, like spamming *echo for example
|
||||
|
||||
if (queue.isEmpty()) return;
|
||||
|
||||
final String message = queue.get(0);
|
||||
|
||||
if (message.startsWith("/")) {
|
||||
String removedMessage = message.substring(1);
|
||||
|
||||
final String[] splittedSpace = removedMessage.split("\\s+"); // [minecraft:test, arg1, arg2, ...]
|
||||
final String[] splittedColon = splittedSpace[0].split(":"); // [minecraft, test]
|
||||
if (bot.options.removeNamespaces && splittedColon.length >= 2) {
|
||||
removedMessage = String.join(":", Arrays.copyOfRange(splittedColon, 1, splittedColon.length));
|
||||
|
||||
if (splittedSpace.length > 1) {
|
||||
removedMessage += " ";
|
||||
removedMessage += String.join(" ", Arrays.copyOfRange(splittedSpace, 1, splittedSpace.length));
|
||||
}
|
||||
}
|
||||
|
||||
sendCommandInstantly(removedMessage);
|
||||
} else {
|
||||
sendChatInstantly(message);
|
||||
}
|
||||
|
||||
queue.remove(0);
|
||||
}
|
||||
|
||||
public void sendCommandInstantly (String command) {
|
||||
bot.session.send(new ServerboundChatCommandPacket(
|
||||
command,
|
||||
Instant.now().toEpochMilli(),
|
||||
0L,
|
||||
Collections.emptyList(),
|
||||
0,
|
||||
new BitSet()
|
||||
));
|
||||
}
|
||||
|
||||
public void sendChatInstantly (String message) {
|
||||
bot.session.send(new ServerboundChatPacket(
|
||||
message,
|
||||
Instant.now().toEpochMilli(),
|
||||
0L,
|
||||
null,
|
||||
0,
|
||||
new BitSet()
|
||||
));
|
||||
}
|
||||
|
||||
public void clearQueue () { queue.clear(); }
|
||||
|
||||
public void send (String message) {
|
||||
final Matcher splitMatcher = CHAT_SPLIT_PATTERN.matcher(message);
|
||||
|
||||
String lastColor = "";
|
||||
|
||||
boolean isFirst = true;
|
||||
|
||||
// kinda broken but whatever
|
||||
while (splitMatcher.find()) {
|
||||
final String eachMessage = splitMatcher.group(1);
|
||||
|
||||
final Matcher eachMessageMatcher = CHAT_SPLIT_PATTERN.matcher(eachMessage);
|
||||
|
||||
while (eachMessageMatcher.find()) {
|
||||
String strippedMessage = IllegalCharactersUtilities.stripIllegalCharacters(eachMessageMatcher.group(1));
|
||||
|
||||
if (strippedMessage.trim().isEmpty()) continue;
|
||||
|
||||
final Matcher colorCodeEndMatcher = COLOR_CODE_END_PATTERN.matcher(strippedMessage);
|
||||
|
||||
if (colorCodeEndMatcher.find()) strippedMessage = strippedMessage.substring(0, strippedMessage.length() - 2);
|
||||
|
||||
if (!isFirst) {
|
||||
final Matcher colorCodeEndMatcher2 = COLOR_CODE_END_PATTERN.matcher(message);
|
||||
|
||||
Matcher colorCodeMatcher;
|
||||
|
||||
if (!colorCodeEndMatcher2.find()) colorCodeMatcher = COLOR_CODE_PATTERN.matcher(message);
|
||||
else colorCodeMatcher = COLOR_CODE_PATTERN.matcher(message.substring(0, message.length() - 2));
|
||||
|
||||
while (colorCodeMatcher.find()) lastColor = colorCodeMatcher.group();
|
||||
}
|
||||
|
||||
queue.add(
|
||||
lastColor + strippedMessage // the regex has 254 (comes from 256 - 2 (color code length)) so we can do this here
|
||||
);
|
||||
|
||||
isFirst = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void tellraw (Component component, String targets) {
|
||||
if (bot.options.useChat) {
|
||||
if (!targets.equals("@a")) return; // worst fix of all time!1!
|
||||
|
||||
final String stringified = ComponentUtilities.stringifyMotd(component).replace("§", "&");
|
||||
send(stringified);
|
||||
} else {
|
||||
bot.core.run("minecraft:tellraw " + targets + " " + GsonComponentSerializer.gson().serialize(component));
|
||||
}
|
||||
}
|
||||
|
||||
public void tellraw (Component component, UUID uuid) { tellraw(component, UUIDUtilities.selector(uuid)); }
|
||||
|
||||
public void tellraw (Component component) { tellraw(component, "@a"); }
|
||||
|
||||
public void actionBar (Component component, String targets) {
|
||||
if (bot.options.useChat) return;
|
||||
bot.core.run("minecraft:title " + targets + " actionbar " + GsonComponentSerializer.gson().serialize(component));
|
||||
}
|
||||
|
||||
public void actionBar (Component component, UUID uuid) { actionBar(component, UUIDUtilities.selector(uuid)); }
|
||||
|
||||
public void actionBar (Component component) { actionBar(component, "@a"); }
|
||||
|
||||
public void addListener (Listener listener) { listeners.add(listener); }
|
||||
|
||||
public static class Listener {
|
||||
public boolean playerMessageReceived (PlayerMessage message) { return true; }
|
||||
public boolean systemMessageReceived (Component component, String string, String ansi) { return true; }
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user