package net.minecraft.network.protocol.game;

import net.minecraft.network.FriendlyByteBuf;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.network.protocol.Packet;
import net.minecraft.network.protocol.PacketType;

public class ServerboundPlayerInputPacket implements Packet<ServerGamePacketListener> {
	public static final StreamCodec<FriendlyByteBuf, ServerboundPlayerInputPacket> STREAM_CODEC = Packet.codec(
		ServerboundPlayerInputPacket::write, ServerboundPlayerInputPacket::new
	);
	private static final int FLAG_JUMPING = 1;
	private static final int FLAG_SHIFT_KEY_DOWN = 2;
	/**
	 * Positive for left strafe, negative for right
	 */
	private final float xxa;
	private final float zza;
	private final boolean isJumping;
	private final boolean isShiftKeyDown;

	public ServerboundPlayerInputPacket(float xxa, float zza, boolean isJumping, boolean isShiftKeyDown) {
		this.xxa = xxa;
		this.zza = zza;
		this.isJumping = isJumping;
		this.isShiftKeyDown = isShiftKeyDown;
	}

	private ServerboundPlayerInputPacket(FriendlyByteBuf buffer) {
		this.xxa = buffer.readFloat();
		this.zza = buffer.readFloat();
		byte b = buffer.readByte();
		this.isJumping = (b & 1) > 0;
		this.isShiftKeyDown = (b & 2) > 0;
	}

	/**
	 * Writes the raw packet data to the data stream.
	 */
	private void write(FriendlyByteBuf buffer) {
		buffer.writeFloat(this.xxa);
		buffer.writeFloat(this.zza);
		byte b = 0;
		if (this.isJumping) {
			b = (byte)(b | 1);
		}

		if (this.isShiftKeyDown) {
			b = (byte)(b | 2);
		}

		buffer.writeByte(b);
	}

	@Override
	public PacketType<ServerboundPlayerInputPacket> type() {
		return GamePacketTypes.SERVERBOUND_PLAYER_INPUT;
	}

	/**
	 * Passes this Packet on to the NetHandler for processing.
	 */
	public void handle(ServerGamePacketListener handler) {
		handler.handlePlayerInput(this);
	}

	public float getXxa() {
		return this.xxa;
	}

	public float getZza() {
		return this.zza;
	}

	public boolean isJumping() {
		return this.isJumping;
	}

	public boolean isShiftKeyDown() {
		return this.isShiftKeyDown;
	}
}