package net.minecraft.network;

import com.mojang.logging.LogUtils;
import net.minecraft.network.chat.Component;
import net.minecraft.network.protocol.PacketFlow;
import net.minecraft.network.protocol.common.ClientboundDisconnectPacket;
import org.slf4j.Logger;

/**
 * Variant of {@link Connection} that monitors the amount of received packets and disables receiving if the set limit is exceeded.
 */
public class RateKickingConnection extends Connection {
	private static final Logger LOGGER = LogUtils.getLogger();
	private static final Component EXCEED_REASON = Component.translatable("disconnect.exceeded_packet_rate");
	private final int rateLimitPacketsPerSecond;

	public RateKickingConnection(int rateLimitPacketsPerSecond) {
		super(PacketFlow.SERVERBOUND);
		this.rateLimitPacketsPerSecond = rateLimitPacketsPerSecond;
	}

	@Override
	protected void tickSecond() {
		super.tickSecond();
		float f = this.getAverageReceivedPackets();
		if (f > this.rateLimitPacketsPerSecond) {
			LOGGER.warn("Player exceeded rate-limit (sent {} packets per second)", f);
			this.send(new ClientboundDisconnectPacket(EXCEED_REASON), PacketSendListener.thenRun(() -> this.disconnect(EXCEED_REASON)));
			this.setReadOnly();
		}
	}
}