PacketSyncAnimal.java

package joshie.harvest.animals.packet;

import io.netty.buffer.ByteBuf;
import joshie.harvest.api.animals.AnimalStats;
import joshie.harvest.core.helpers.EntityHelper;
import joshie.harvest.core.helpers.MCClientHelper;
import joshie.harvest.core.network.Packet;
import joshie.harvest.core.network.Packet.Side;
import joshie.harvest.core.network.PenguinPacket;
import net.minecraft.entity.passive.EntityAnimal;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraftforge.fml.common.network.ByteBufUtils;

@Packet(Side.CLIENT)
public class PacketSyncAnimal extends PenguinPacket {
    private int id;
    private NBTTagCompound tag;

    public PacketSyncAnimal() {}
    public PacketSyncAnimal(int id, AnimalStats stats) {
        this.id = id;
        this.tag = (NBTTagCompound) stats.serializeNBT();
    }

    @Override
    public void toBytes(ByteBuf buf) {
        buf.writeInt(id);
        buf.writeBoolean(tag != null);
        if (tag != null) ByteBufUtils.writeTag(buf, tag);
    }

    @Override
    public void fromBytes(ByteBuf buf) {
        id = buf.readInt();
        if (buf.readBoolean()) {
            tag = ByteBufUtils.readTag(buf);
        }
    }

    @Override
    @SuppressWarnings("unchecked, ConstantConditions")
    public void handlePacket(EntityPlayer player) {
        EntityAnimal animal = getEntityAsAnimal();
        if (animal != null) {
            AnimalStats stats = EntityHelper.getStats(animal);
            if (stats != null) {
                stats.setEntity(animal).deserializeNBT(tag);
            }
        }
    }

    private EntityAnimal getEntityAsAnimal() {
        return (EntityAnimal) MCClientHelper.getWorld().getEntityByID(id);
    }
}