src/protocolsupport/protocol/PacketDataSerializer.java

package protocolsupport.protocol;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufInputStream;
import io.netty.buffer.ByteBufOutputStream;
import io.netty.buffer.Unpooled;

import java.io.BufferedInputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutput;
import java.io.DataOutputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

import net.minecraft.server.v1_8_R1.GameProfileSerializer;
import net.minecraft.server.v1_8_R1.Item;
import net.minecraft.server.v1_8_R1.ItemStack;
import net.minecraft.server.v1_8_R1.NBTCompressedStreamTools;
import net.minecraft.server.v1_8_R1.NBTReadLimiter;
import net.minecraft.server.v1_8_R1.NBTTagCompound;
import net.minecraft.server.v1_8_R1.NBTTagString;

import org.bukkit.Material;
import org.bukkit.craftbukkit.v1_8_R1.inventory.CraftItemStack;
import org.spigotmc.LimitStream;
import org.spigotmc.SneakyThrow;

import protocolsupport.protocol.DataStorage.ProtocolVersion;

import com.mojang.authlib.GameProfile;

public class PacketDataSerializer extends net.minecraft.server.v1_8_R1.PacketDataSerializer {

	private ProtocolVersion version;

	public PacketDataSerializer(ByteBuf buf, ProtocolVersion version) {
		super(buf);
		this.version = version;
	}

	public ProtocolVersion getVersion() {
		return version;
	}


	@SuppressWarnings("deprecation")
	@Override
	public void a(ItemStack itemstack) {
		if ((itemstack == null) || (itemstack.getItem() == null)) {
			writeShort(-1);
		} else {
			int itemId = Item.getId(itemstack.getItem());
			if ((getVersion() == ProtocolVersion.MINECRAFT_1_6_4) || (getVersion() == ProtocolVersion.MINECRAFT_1_6_2)) {
				writeShort(protocolsupport.protocol.v_1_6.remappers.ItemIDRemapper.replaceItemId(itemId));
			} else if ((getVersion() == ProtocolVersion.MINECRAFT_1_7_10) || (getVersion() == ProtocolVersion.MINECRAFT_1_7_5)) {
				writeShort(protocolsupport.protocol.v_1_7.remappers.ItemIDRemapper.replaceItemId(itemId));
			} else if (getVersion() == ProtocolVersion.MINECRAFT_1_5_2) {
				writeShort(protocolsupport.protocol.v_1_5.remappers.ItemIDRemapper.replaceItemId(itemId));
			} else {
				writeShort(itemId);
			}
			writeByte(itemstack.count);
			writeShort(itemstack.getData());
			NBTTagCompound nbttagcompound = null;
			if (itemstack.getItem().usesDurability() || itemstack.getItem().p()) {
				itemstack = itemstack.cloneItemStack();
				CraftItemStack.setItemMeta(itemstack, CraftItemStack.getItemMeta(itemstack));
				nbttagcompound = itemstack.getTag();
			}
			if (
				(getVersion() == ProtocolVersion.MINECRAFT_1_7_5) ||
				(getVersion() == ProtocolVersion.MINECRAFT_1_6_4) ||
				(getVersion() == ProtocolVersion.MINECRAFT_1_6_2) ||
				(getVersion() == ProtocolVersion.MINECRAFT_1_5_2)
			) {
				if ((nbttagcompound != null) && (itemId == Material.SKULL_ITEM.getId())) {
					if (nbttagcompound.hasKeyOfType("SkullOwner", 10)) {
						GameProfile gameprofile = GameProfileSerializer.deserialize(nbttagcompound.getCompound("SkullOwner"));
						if (gameprofile.getName() != null) {
							nbttagcompound.set("SkullOwner", new NBTTagString(gameprofile.getName()));
						} else {
							nbttagcompound.remove("SkullOwner");
						}
					}
				}
			}
			this.a(nbttagcompound);
		}
	}

	@Override
	public void a(final NBTTagCompound nbttagcompound) {
		if (getVersion() != ProtocolVersion.MINECRAFT_1_8) {
			if (nbttagcompound == null) {
				writeShort(-1);
			} else {
				final byte[] abyte = write(nbttagcompound);
				writeShort(abyte.length);
				this.writeBytes(abyte);
			}
		} else if (nbttagcompound == null) {
			writeByte(0);
		} else {
			final ByteBufOutputStream out = new ByteBufOutputStream(Unpooled.buffer());
			NBTCompressedStreamTools.a(nbttagcompound, (DataOutput) new DataOutputStream(out));
			this.writeBytes(out.buffer());
			out.buffer().release();
		}
	}

	@Override
	public NBTTagCompound h() {
		if (getVersion() != ProtocolVersion.MINECRAFT_1_8) {
			final short short1 = readShort();
			if (short1 < 0) {
				return null;
			}
			final byte[] abyte = new byte[short1];
			this.readBytes(abyte);
			return read(abyte, new NBTReadLimiter(2097152L));
		} else {
			final int index = this.readerIndex();
			if (readByte() == 0) {
				return null;
			}
			this.readerIndex(index);
			return NBTCompressedStreamTools.a(new DataInputStream(new ByteBufInputStream(this)), new NBTReadLimiter(2097152L));
		}
	}

	@Override
	public String c(int limit) {
		if (
			(getVersion() == ProtocolVersion.MINECRAFT_1_6_4) ||
			(getVersion() == ProtocolVersion.MINECRAFT_1_6_2) ||
			(getVersion() == ProtocolVersion.MINECRAFT_1_5_2)
		) {
			int length = readUnsignedShort();
			return new String(readBytes(length * 2).array(), StandardCharsets.UTF_16BE);
		} else {
			return super.c(limit);
		}
	}

	@Override
	public net.minecraft.server.v1_8_R1.PacketDataSerializer a(String string) {
		if (
			(getVersion() == ProtocolVersion.MINECRAFT_1_6_4) ||
			(getVersion() == ProtocolVersion.MINECRAFT_1_6_2) ||
			(getVersion() == ProtocolVersion.MINECRAFT_1_5_2)
		) {
			writeShort(string.length());
			writeBytes(string.getBytes(StandardCharsets.UTF_16BE));
		} else {
			super.a(string);
		}
		return this;
	}

	public int readVarInt() {
		return e();
	}

	public void writeVarInt(int varInt) {
		b(varInt);
	}

	public String readString(int limit) {
		return c(limit);
	}

	public void writeString(String string) {
		a(string);
	}

	private static NBTTagCompound read(final byte[] data, final NBTReadLimiter nbtreadlimiter) {
		try {
			final DataInputStream datainputstream = new DataInputStream(new BufferedInputStream(new LimitStream(new GZIPInputStream(new ByteArrayInputStream(data)), nbtreadlimiter)));
			NBTTagCompound nbttagcompound;
			try {
				nbttagcompound = NBTCompressedStreamTools.a(datainputstream, nbtreadlimiter);
			} finally {
				datainputstream.close();
			}
			return nbttagcompound;
		} catch (IOException ex) {
			SneakyThrow.sneaky(ex);
			return null;
		}
	}

	private static byte[] write(final NBTTagCompound nbttagcompound) {
		try {
			final ByteArrayOutputStream bytearrayoutputstream = new ByteArrayOutputStream();
			final DataOutputStream dataoutputstream = new DataOutputStream(new GZIPOutputStream(bytearrayoutputstream));
			try {
				NBTCompressedStreamTools.a(nbttagcompound, (DataOutput) dataoutputstream);
			} finally {
				dataoutputstream.close();
			}
			return bytearrayoutputstream.toByteArray();
		} catch (IOException ex) {
			SneakyThrow.sneaky(ex);
			return null;
		}
	}

}