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;
import net.minecraft.world.entity.player.Abilities;

public class ServerboundPlayerAbilitiesPacket implements Packet<ServerGamePacketListener> {
	public static final StreamCodec<FriendlyByteBuf, ServerboundPlayerAbilitiesPacket> STREAM_CODEC = Packet.codec(
		ServerboundPlayerAbilitiesPacket::write, ServerboundPlayerAbilitiesPacket::new
	);
	private static final int FLAG_FLYING = 2;
	private final boolean isFlying;

	public ServerboundPlayerAbilitiesPacket(Abilities abilities) {
		this.isFlying = abilities.flying;
	}

	private ServerboundPlayerAbilitiesPacket(FriendlyByteBuf buffer) {
		byte b = buffer.readByte();
		this.isFlying = (b & 2) != 0;
	}

	/**
	 * Writes the raw packet data to the data stream.
	 */
	private void write(FriendlyByteBuf buffer) {
		byte b = 0;
		if (this.isFlying) {
			b = (byte)(b | 2);
		}

		buffer.writeByte(b);
	}

	@Override
	public PacketType<ServerboundPlayerAbilitiesPacket> type() {
		return GamePacketTypes.SERVERBOUND_PLAYER_ABILITIES;
	}

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

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