package net.minecraft.network.protocol.game;

import net.minecraft.core.Holder;
import net.minecraft.network.RegistryFriendlyByteBuf;
import net.minecraft.network.codec.StreamCodec;
import net.minecraft.network.protocol.Packet;
import net.minecraft.network.protocol.PacketType;
import net.minecraft.world.effect.MobEffect;
import net.minecraft.world.effect.MobEffectInstance;

public class ClientboundUpdateMobEffectPacket implements Packet<ClientGamePacketListener> {
	public static final StreamCodec<RegistryFriendlyByteBuf, ClientboundUpdateMobEffectPacket> STREAM_CODEC = Packet.codec(
		ClientboundUpdateMobEffectPacket::write, ClientboundUpdateMobEffectPacket::new
	);
	private static final int FLAG_AMBIENT = 1;
	private static final int FLAG_VISIBLE = 2;
	private static final int FLAG_SHOW_ICON = 4;
	private static final int FLAG_BLEND = 8;
	private final int entityId;
	private final Holder<MobEffect> effect;
	private final int effectAmplifier;
	private final int effectDurationTicks;
	private final byte flags;

	public ClientboundUpdateMobEffectPacket(int entityId, MobEffectInstance effect, boolean blend) {
		this.entityId = entityId;
		this.effect = effect.getEffect();
		this.effectAmplifier = effect.getAmplifier();
		this.effectDurationTicks = effect.getDuration();
		byte b = 0;
		if (effect.isAmbient()) {
			b = (byte)(b | 1);
		}

		if (effect.isVisible()) {
			b = (byte)(b | 2);
		}

		if (effect.showIcon()) {
			b = (byte)(b | 4);
		}

		if (blend) {
			b = (byte)(b | 8);
		}

		this.flags = b;
	}

	private ClientboundUpdateMobEffectPacket(RegistryFriendlyByteBuf buffer) {
		this.entityId = buffer.readVarInt();
		this.effect = MobEffect.STREAM_CODEC.decode(buffer);
		this.effectAmplifier = buffer.readVarInt();
		this.effectDurationTicks = buffer.readVarInt();
		this.flags = buffer.readByte();
	}

	private void write(RegistryFriendlyByteBuf buffer) {
		buffer.writeVarInt(this.entityId);
		MobEffect.STREAM_CODEC.encode(buffer, this.effect);
		buffer.writeVarInt(this.effectAmplifier);
		buffer.writeVarInt(this.effectDurationTicks);
		buffer.writeByte(this.flags);
	}

	@Override
	public PacketType<ClientboundUpdateMobEffectPacket> type() {
		return GamePacketTypes.CLIENTBOUND_UPDATE_MOB_EFFECT;
	}

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

	public int getEntityId() {
		return this.entityId;
	}

	public Holder<MobEffect> getEffect() {
		return this.effect;
	}

	public int getEffectAmplifier() {
		return this.effectAmplifier;
	}

	public int getEffectDurationTicks() {
		return this.effectDurationTicks;
	}

	public boolean isEffectVisible() {
		return (this.flags & 2) != 0;
	}

	public boolean isEffectAmbient() {
		return (this.flags & 1) != 0;
	}

	public boolean effectShowsIcon() {
		return (this.flags & 4) != 0;
	}

	public boolean shouldBlend() {
		return (this.flags & 8) != 0;
	}
}