mirror of
https://github.com/GeyserMC/Geyser.git
synced 2024-08-14 23:57:35 +00:00
279 lines
10 KiB
Java
279 lines
10 KiB
Java
|
/*
|
||
|
* Copyright (c) 2019-2020 GeyserMC. http://geysermc.org
|
||
|
*
|
||
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
|
* of this software and associated documentation files (the "Software"), to deal
|
||
|
* in the Software without restriction, including without limitation the rights
|
||
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
|
* copies of the Software, and to permit persons to whom the Software is
|
||
|
* furnished to do so, subject to the following conditions:
|
||
|
*
|
||
|
* The above copyright notice and this permission notice shall be included in
|
||
|
* all copies or substantial portions of the Software.
|
||
|
*
|
||
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||
|
* THE SOFTWARE.
|
||
|
*
|
||
|
* @author GeyserMC
|
||
|
* @link https://github.com/GeyserMC/Geyser
|
||
|
*/
|
||
|
|
||
|
package org.geysermc.connector.network.translators.chat;
|
||
|
|
||
|
import com.github.steveice10.mc.protocol.data.game.scoreboard.TeamColor;
|
||
|
import com.github.steveice10.mc.protocol.data.message.style.ChatColor;
|
||
|
import com.github.steveice10.mc.protocol.data.message.style.ChatFormat;
|
||
|
import net.kyori.adventure.text.Component;
|
||
|
import net.kyori.adventure.text.renderer.TranslatableComponentRenderer;
|
||
|
import net.kyori.adventure.text.serializer.gson.GsonComponentSerializer;
|
||
|
import net.kyori.adventure.text.serializer.legacy.LegacyComponentSerializer;
|
||
|
import net.kyori.adventure.translation.TranslationRegistry;
|
||
|
import org.geysermc.connector.network.session.GeyserSession;
|
||
|
import org.geysermc.connector.utils.LanguageUtils;
|
||
|
|
||
|
import java.util.*;
|
||
|
|
||
|
public class MessageTranslator {
|
||
|
|
||
|
// These are used for handling the translations of the messages
|
||
|
private static final TranslationRegistry REGISTRY = new MinecraftTranslationRegistry();
|
||
|
private static final TranslatableComponentRenderer<Locale> RENDERER = TranslatableComponentRenderer.usingTranslationSource(REGISTRY);
|
||
|
|
||
|
// Store team colors for player names
|
||
|
private static final Map<TeamColor, String> TEAM_COLORS = new HashMap<>();
|
||
|
|
||
|
static {
|
||
|
TEAM_COLORS.put(TeamColor.BLACK, getColor(ChatColor.BLACK));
|
||
|
TEAM_COLORS.put(TeamColor.DARK_BLUE, getColor(ChatColor.DARK_BLUE));
|
||
|
TEAM_COLORS.put(TeamColor.DARK_GREEN, getColor(ChatColor.DARK_GREEN));
|
||
|
TEAM_COLORS.put(TeamColor.DARK_AQUA, getColor(ChatColor.DARK_AQUA));
|
||
|
TEAM_COLORS.put(TeamColor.DARK_RED, getColor(ChatColor.DARK_RED));
|
||
|
TEAM_COLORS.put(TeamColor.DARK_PURPLE, getColor(ChatColor.DARK_PURPLE));
|
||
|
TEAM_COLORS.put(TeamColor.GOLD, getColor(ChatColor.GOLD));
|
||
|
TEAM_COLORS.put(TeamColor.GRAY, getColor(ChatColor.GRAY));
|
||
|
TEAM_COLORS.put(TeamColor.DARK_GRAY, getColor(ChatColor.DARK_GRAY));
|
||
|
TEAM_COLORS.put(TeamColor.BLUE, getColor(ChatColor.BLUE));
|
||
|
TEAM_COLORS.put(TeamColor.GREEN, getColor(ChatColor.GREEN));
|
||
|
TEAM_COLORS.put(TeamColor.AQUA, getColor(ChatColor.AQUA));
|
||
|
TEAM_COLORS.put(TeamColor.RED, getColor(ChatColor.RED));
|
||
|
TEAM_COLORS.put(TeamColor.LIGHT_PURPLE, getColor(ChatColor.LIGHT_PURPLE));
|
||
|
TEAM_COLORS.put(TeamColor.YELLOW, getColor(ChatColor.YELLOW));
|
||
|
TEAM_COLORS.put(TeamColor.WHITE, getColor(ChatColor.WHITE));
|
||
|
TEAM_COLORS.put(TeamColor.OBFUSCATED, getFormat(ChatFormat.OBFUSCATED));
|
||
|
TEAM_COLORS.put(TeamColor.BOLD, getFormat(ChatFormat.BOLD));
|
||
|
TEAM_COLORS.put(TeamColor.STRIKETHROUGH, getFormat(ChatFormat.STRIKETHROUGH));
|
||
|
TEAM_COLORS.put(TeamColor.ITALIC, getFormat(ChatFormat.ITALIC));
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Convert a Java message to the legacy format ready for bedrock
|
||
|
*
|
||
|
* @param message Java message
|
||
|
* @param locale Locale to use for translation strings
|
||
|
* @return Parsed and formatted message for bedrock
|
||
|
*/
|
||
|
public static String convertMessage(String message, String locale) {
|
||
|
Component component = GsonComponentSerializer.gson().deserialize(message);
|
||
|
|
||
|
// Get a Locale from the given locale string
|
||
|
Locale localeCode = Locale.forLanguageTag(locale.replace('_', '-'));
|
||
|
component = RENDERER.render(component, localeCode);
|
||
|
|
||
|
return LegacyComponentSerializer.legacySection().serialize(component);
|
||
|
}
|
||
|
|
||
|
public static String convertMessage(String message) {
|
||
|
return convertMessage(message, LanguageUtils.getDefaultLocale());
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Verifies the message is valid JSON in case it's plaintext. Works around GsonComponentSeraializer not using lenient mode.
|
||
|
* See https://wiki.vg/Chat for messages sent in lenient mode, and for a description on leniency.
|
||
|
*
|
||
|
* @param message Potentially lenient JSON message
|
||
|
* @param locale Locale to use for translation strings
|
||
|
* @return Bedrock formatted message
|
||
|
*/
|
||
|
public static String convertMessageLenient(String message, String locale) {
|
||
|
if (isMessage(message)) {
|
||
|
return convertMessage(message, locale);
|
||
|
} else {
|
||
|
String convertedMessage = convertMessage(convertToJavaMessage(message), locale);
|
||
|
|
||
|
// We have to do this since Adventure strips the starting reset character
|
||
|
if (message.startsWith(getColor(ChatColor.RESET))) {
|
||
|
convertedMessage = getColor(ChatColor.RESET) + convertedMessage;
|
||
|
}
|
||
|
|
||
|
return convertedMessage;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public static String convertMessageLenient(String message) {
|
||
|
return convertMessageLenient(message, LanguageUtils.getDefaultLocale());
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Convert a Bedrock message string back to a format Java can understand
|
||
|
*
|
||
|
* @param message Message to convert
|
||
|
* @return The formatted JSON string
|
||
|
*/
|
||
|
public static String convertToJavaMessage(String message) {
|
||
|
Component component = LegacyComponentSerializer.legacySection().deserialize(message);
|
||
|
return GsonComponentSerializer.gson().serialize(component);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Checks if the given text string is a JSON message
|
||
|
*
|
||
|
* @param text String to test
|
||
|
* @return True if its a valid message JSON string, false if not
|
||
|
*/
|
||
|
public static boolean isMessage(String text) {
|
||
|
if (text.trim().isEmpty()) {
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
try {
|
||
|
GsonComponentSerializer.gson().deserialize(text);
|
||
|
} catch (Exception ex) {
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Convert a {@link ChatColor} into a string for inserting into messages
|
||
|
*
|
||
|
* @param color {@link ChatColor} to convert
|
||
|
* @return The converted color string
|
||
|
*/
|
||
|
private static String getColor(String color) {
|
||
|
String base = "\u00a7";
|
||
|
switch (color) {
|
||
|
case ChatColor.BLACK:
|
||
|
base += "0";
|
||
|
break;
|
||
|
case ChatColor.DARK_BLUE:
|
||
|
base += "1";
|
||
|
break;
|
||
|
case ChatColor.DARK_GREEN:
|
||
|
base += "2";
|
||
|
break;
|
||
|
case ChatColor.DARK_AQUA:
|
||
|
base += "3";
|
||
|
break;
|
||
|
case ChatColor.DARK_RED:
|
||
|
base += "4";
|
||
|
break;
|
||
|
case ChatColor.DARK_PURPLE:
|
||
|
base += "5";
|
||
|
break;
|
||
|
case ChatColor.GOLD:
|
||
|
base += "6";
|
||
|
break;
|
||
|
case ChatColor.GRAY:
|
||
|
base += "7";
|
||
|
break;
|
||
|
case ChatColor.DARK_GRAY:
|
||
|
base += "8";
|
||
|
break;
|
||
|
case ChatColor.BLUE:
|
||
|
base += "9";
|
||
|
break;
|
||
|
case ChatColor.GREEN:
|
||
|
base += "a";
|
||
|
break;
|
||
|
case ChatColor.AQUA:
|
||
|
base += "b";
|
||
|
break;
|
||
|
case ChatColor.RED:
|
||
|
base += "c";
|
||
|
break;
|
||
|
case ChatColor.LIGHT_PURPLE:
|
||
|
base += "d";
|
||
|
break;
|
||
|
case ChatColor.YELLOW:
|
||
|
base += "e";
|
||
|
break;
|
||
|
case ChatColor.WHITE:
|
||
|
base += "f";
|
||
|
break;
|
||
|
case ChatColor.RESET:
|
||
|
base += "r";
|
||
|
break;
|
||
|
default:
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
return base;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Convert a {@link ChatFormat} into a string for inserting into messages
|
||
|
*
|
||
|
* @param format {@link ChatFormat} to convert
|
||
|
* @return The converted chat formatting string
|
||
|
*/
|
||
|
private static String getFormat(ChatFormat format) {
|
||
|
StringBuilder str = new StringBuilder();
|
||
|
String base = "\u00a7";
|
||
|
switch (format) {
|
||
|
case OBFUSCATED:
|
||
|
base += "k";
|
||
|
break;
|
||
|
case BOLD:
|
||
|
base += "l";
|
||
|
break;
|
||
|
case STRIKETHROUGH:
|
||
|
base += "m";
|
||
|
break;
|
||
|
case UNDERLINED:
|
||
|
base += "n";
|
||
|
break;
|
||
|
case ITALIC:
|
||
|
base += "o";
|
||
|
break;
|
||
|
default:
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
str.append(base);
|
||
|
|
||
|
return str.toString();
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Convert a team color to a chat color
|
||
|
*
|
||
|
* @param teamColor
|
||
|
* @return The chat color character
|
||
|
*/
|
||
|
public static String toChatColor(TeamColor teamColor) {
|
||
|
return TEAM_COLORS.getOrDefault(teamColor, "");
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Checks if the given message is over 256 characters (Java edition server chat limit) and sends a message to the user if it is
|
||
|
*
|
||
|
* @param message Message to check
|
||
|
* @param session {@link GeyserSession} for the user
|
||
|
* @return True if the message is too long, false if not
|
||
|
*/
|
||
|
public static boolean isTooLong(String message, GeyserSession session) {
|
||
|
if (message.length() > 256) {
|
||
|
session.sendMessage(LanguageUtils.getPlayerLocaleString("geyser.chat.too_long", session.getLocale(), message.length()));
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
return false;
|
||
|
}
|
||
|
}
|