minecraft-src/net/minecraft/network/chat/PlayerChatMessage.java

package net.minecraft.network.chat;

import com.google.common.primitives.Ints;
import com.mojang.serialization.MapCodec;
import com.mojang.serialization.codecs.RecordCodecBuilder;
import java.security.SignatureException;
import java.time.Duration;
import java.time.Instant;
import java.util.Objects;
import java.util.Optional;
import java.util.UUID;
import java.util.stream.Collectors;
import net.minecraft.Util;
import net.minecraft.util.SignatureValidator;
import net.minecraft.util.SignatureUpdater.Output;
import org.jetbrains.annotations.Nullable;

public record PlayerChatMessage(
	SignedMessageLink link, @Nullable MessageSignature signature, SignedMessageBody signedBody, @Nullable Component unsignedContent, FilterMask filterMask
) {
	public static final MapCodec<PlayerChatMessage> MAP_CODEC = RecordCodecBuilder.mapCodec(
		instance -> instance.group(
				SignedMessageLink.CODEC.fieldOf("link").forGetter(PlayerChatMessage::link),
				MessageSignature.CODEC.optionalFieldOf("signature").forGetter(playerChatMessage -> Optional.ofNullable(playerChatMessage.signature)),
				SignedMessageBody.MAP_CODEC.forGetter(PlayerChatMessage::signedBody),
				ComponentSerialization.CODEC.optionalFieldOf("unsigned_content").forGetter(playerChatMessage -> Optional.ofNullable(playerChatMessage.unsignedContent)),
				FilterMask.CODEC.optionalFieldOf("filter_mask", FilterMask.PASS_THROUGH).forGetter(PlayerChatMessage::filterMask)
			)
			.apply(
				instance,
				(signedMessageLink, optional, signedMessageBody, optional2, filterMask) -> new PlayerChatMessage(
					signedMessageLink, (MessageSignature)optional.orElse(null), signedMessageBody, (Component)optional2.orElse(null), filterMask
				)
			)
	);
	private static final UUID SYSTEM_SENDER = Util.NIL_UUID;
	public static final Duration MESSAGE_EXPIRES_AFTER_SERVER = Duration.ofMinutes(5L);
	public static final Duration MESSAGE_EXPIRES_AFTER_CLIENT = MESSAGE_EXPIRES_AFTER_SERVER.plus(Duration.ofMinutes(2L));

	public static PlayerChatMessage system(String content) {
		return unsigned(SYSTEM_SENDER, content);
	}

	public static PlayerChatMessage unsigned(UUID sender, String content) {
		SignedMessageBody signedMessageBody = SignedMessageBody.unsigned(content);
		SignedMessageLink signedMessageLink = SignedMessageLink.unsigned(sender);
		return new PlayerChatMessage(signedMessageLink, null, signedMessageBody, null, FilterMask.PASS_THROUGH);
	}

	public PlayerChatMessage withUnsignedContent(Component message) {
		Component component = !message.equals(Component.literal(this.signedContent())) ? message : null;
		return new PlayerChatMessage(this.link, this.signature, this.signedBody, component, this.filterMask);
	}

	public PlayerChatMessage removeUnsignedContent() {
		return this.unsignedContent != null ? new PlayerChatMessage(this.link, this.signature, this.signedBody, null, this.filterMask) : this;
	}

	public PlayerChatMessage filter(FilterMask mask) {
		return this.filterMask.equals(mask) ? this : new PlayerChatMessage(this.link, this.signature, this.signedBody, this.unsignedContent, mask);
	}

	public PlayerChatMessage filter(boolean shouldFilter) {
		return this.filter(shouldFilter ? this.filterMask : FilterMask.PASS_THROUGH);
	}

	public PlayerChatMessage removeSignature() {
		SignedMessageBody signedMessageBody = SignedMessageBody.unsigned(this.signedContent());
		SignedMessageLink signedMessageLink = SignedMessageLink.unsigned(this.sender());
		return new PlayerChatMessage(signedMessageLink, null, signedMessageBody, this.unsignedContent, this.filterMask);
	}

	public static void updateSignature(Output output, SignedMessageLink link, SignedMessageBody body) throws SignatureException {
		output.update(Ints.toByteArray(1));
		link.updateSignature(output);
		body.updateSignature(output);
	}

	public boolean verify(SignatureValidator validator) {
		return this.signature != null && this.signature.verify(validator, output -> updateSignature(output, this.link, this.signedBody));
	}

	public String signedContent() {
		return this.signedBody.content();
	}

	public Component decoratedContent() {
		return (Component)Objects.requireNonNullElseGet(this.unsignedContent, () -> Component.literal(this.signedContent()));
	}

	public Instant timeStamp() {
		return this.signedBody.timeStamp();
	}

	public long salt() {
		return this.signedBody.salt();
	}

	public boolean hasExpiredServer(Instant timestamp) {
		return timestamp.isAfter(this.timeStamp().plus(MESSAGE_EXPIRES_AFTER_SERVER));
	}

	public boolean hasExpiredClient(Instant timestamp) {
		return timestamp.isAfter(this.timeStamp().plus(MESSAGE_EXPIRES_AFTER_CLIENT));
	}

	public UUID sender() {
		return this.link.sender();
	}

	public boolean isSystem() {
		return this.sender().equals(SYSTEM_SENDER);
	}

	public boolean hasSignature() {
		return this.signature != null;
	}

	public boolean hasSignatureFrom(UUID uuid) {
		return this.hasSignature() && this.link.sender().equals(uuid);
	}

	public boolean isFullyFiltered() {
		return this.filterMask.isFullyFiltered();
	}

	public static String describeSigned(PlayerChatMessage chatMessage) {
		return "'"
			+ chatMessage.signedBody.content()
			+ "' @ "
			+ chatMessage.signedBody.timeStamp()
			+ "\n - From: "
			+ chatMessage.link.sender()
			+ "/"
			+ chatMessage.link.sessionId()
			+ ", message #"
			+ chatMessage.link.index()
			+ "\n - Salt: "
			+ chatMessage.signedBody.salt()
			+ "\n - Signature: "
			+ MessageSignature.describe(chatMessage.signature)
			+ "\n - Last Seen: [\n"
			+ (String)chatMessage.signedBody
				.lastSeen()
				.entries()
				.stream()
				.map(messageSignature -> "     " + MessageSignature.describe(messageSignature) + "\n")
				.collect(Collectors.joining())
			+ " ]\n";
	}
}