100 lines
4.6 KiB
Java
100 lines
4.6 KiB
Java
package wtf.hak.survivalfabric.mixin;
|
|
|
|
import net.minecraft.network.ClientConnection;
|
|
import net.minecraft.network.message.MessageType;
|
|
import net.minecraft.network.message.SignedMessage;
|
|
import net.minecraft.server.PlayerManager;
|
|
import net.minecraft.server.network.ConnectedClientData;
|
|
import net.minecraft.server.network.ServerPlayerEntity;
|
|
import net.minecraft.text.Text;
|
|
import net.minecraft.world.GameMode;
|
|
import org.spongepowered.asm.mixin.Mixin;
|
|
import org.spongepowered.asm.mixin.injection.At;
|
|
import org.spongepowered.asm.mixin.injection.Inject;
|
|
import org.spongepowered.asm.mixin.injection.ModifyArg;
|
|
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
|
|
import wtf.hak.survivalfabric.commands.SpectatorCommand;
|
|
import wtf.hak.survivalfabric.config.ConfigManager;
|
|
import wtf.hak.survivalfabric.utils.MathUtils;
|
|
|
|
import java.util.Set;
|
|
|
|
@Mixin(PlayerManager.class)
|
|
public abstract class PlayerManagerMixin {
|
|
|
|
/**
|
|
* Send join message to joined player
|
|
*/
|
|
@Inject(method = {"onPlayerConnect"}, at = {@At(value = "INVOKE", target = "Lnet/minecraft/server/PlayerManager;broadcast(Lnet/minecraft/text/Text;Z)V")})
|
|
public void onPlayerConnect(ClientConnection connection, ServerPlayerEntity player, ConnectedClientData clientData, CallbackInfo ci) {
|
|
if (ConfigManager.getConfig().joinMessageEnabled && !player.getServer().isSingleplayer()) {
|
|
Text text = Text.literal(String.format(ConfigManager.getConfig().joinMessage, player.getName().getString()));
|
|
player.sendMessage(text, false);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Modify join message broadcasted to server
|
|
*/
|
|
@ModifyArg(method = {"onPlayerConnect"}, at = @At(value = "INVOKE", target = "Lnet/minecraft/server/PlayerManager;broadcast(Lnet/minecraft/text/Text;Z)V"))
|
|
private Text onPlayerConnect(Text text) {
|
|
if (ConfigManager.getConfig().joinMessageEnabled) {
|
|
String name = text.getString().split(" ")[0];
|
|
return Text.literal(String.format(ConfigManager.getConfig().joinMessage, name));
|
|
} else
|
|
return text;
|
|
}
|
|
|
|
/**
|
|
* Get player out of specator mode if necessary
|
|
*/
|
|
@Inject(method = {"remove"}, at = {@At("HEAD")})
|
|
public void onPlayerLeave(ServerPlayerEntity player, CallbackInfo ci) {
|
|
if (SpectatorCommand.spectating.containsKey(player)) {
|
|
SpectatorCommand.LocationData loc = SpectatorCommand.spectating.remove(player);
|
|
player.teleport(loc.world, loc.x, loc.y, loc.z, Set.of(), loc.yaw, loc.pitch, false);
|
|
player.changeGameMode(GameMode.SURVIVAL);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Modify chat messages
|
|
*/
|
|
@Inject(method = {"broadcast(Lnet/minecraft/network/message/SignedMessage;Lnet/minecraft/server/network/ServerPlayerEntity;Lnet/minecraft/network/message/MessageType$Parameters;)V"}, at = {@At("HEAD")}, cancellable = true)
|
|
private void onBroadcast(SignedMessage message, ServerPlayerEntity sender, MessageType.Parameters parameters, CallbackInfo ci) {
|
|
if (sender != null) {
|
|
String rawMessage = message.getContent().getString().trim();
|
|
boolean isCalcEnabled = ConfigManager.getConfig().chatCalcEnabled;
|
|
boolean isMsgEnabled = ConfigManager.getConfig().chatMessageEnabled;
|
|
|
|
String processedMessage = rawMessage;
|
|
|
|
if (isCalcEnabled && MathUtils.hasSupportedOperator(rawMessage)) {
|
|
String expression = rawMessage.endsWith("=") ? rawMessage.substring(0, rawMessage.length() - 1).trim() : rawMessage;
|
|
try {
|
|
String result = String.valueOf(MathUtils.evaluateExpression(expression));
|
|
StringBuilder sb = new StringBuilder(rawMessage).append("§6");
|
|
|
|
if (rawMessage.contains(" ")) sb.append(" ");
|
|
if (!rawMessage.endsWith("=")) sb.append("=");
|
|
if (rawMessage.contains(" ")) sb.append(" ");
|
|
|
|
sb.append(result.endsWith(".0") ? result.substring(0, result.length() - 2) : result);
|
|
processedMessage = sb.toString();
|
|
} catch (Exception ignored) {}
|
|
}
|
|
|
|
if (isMsgEnabled) {
|
|
String formatted = String.format(ConfigManager.getConfig().chatMessage, sender.getName().getString(), processedMessage);
|
|
sender.getServer().getPlayerManager().broadcast(Text.literal(formatted), false);
|
|
} else if (isCalcEnabled) {
|
|
String formatted = "<" + sender.getName().getString() + "> " + processedMessage;
|
|
sender.getServer().getPlayerManager().broadcast(Text.literal(formatted), false);
|
|
}
|
|
|
|
ci.cancel();
|
|
}
|
|
|
|
}
|
|
}
|