ChestViewer/PacketHandler.java at master · dewfalse/ChestViewer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package chestviewer;

import cpw.mods.fml.common.network.ByteBufUtils;
import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.entity.player.EntityPlayerMP;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.network.PacketBuffer;
import net.minecraft.pathfinding.PathPoint;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.tileentity.TileEntityChest;

import java.io.*;

public class PacketHandler extends AbstractPacket {

    EnumCommand command;
    int x;
    int y;
    int z;
    ItemStack[] itemStacks = null;

    public PacketHandler() {

    }
    public PacketHandler(EnumCommand command, int x, int y, int z, ItemStack[] itemStacks) {
        this.command = command;
        this.x = x;
        this.y = y;
        this.z = z;
        this.itemStacks = itemStacks;
    }

    @Override
    public void encodeInto(ChannelHandlerContext ctx, ByteBuf buffer) {
        buffer.writeInt(command.ordinal());
        buffer.writeInt(x);
        buffer.writeInt(y);
        buffer.writeInt(z);
        if(itemStacks == null) {
            buffer.writeInt(0);
        }
        else {
            buffer.writeInt(itemStacks.length);
            for(ItemStack is : itemStacks) {
                ByteBufUtils.writeItemStack(buffer, is);
            }
        }
    }

    @Override
    public void decodeInto(ChannelHandlerContext ctx, ByteBuf buffer) {
        command = EnumCommand.values()[buffer.readInt()];
        x = buffer.readInt();
        y = buffer.readInt();
        z = buffer.readInt();
        int size = buffer.readInt();
        if(size == 0) {
            itemStacks = null;
        }
        else {
            itemStacks = new ItemStack[size];
            for(int i = 0; i < size; ++i) {
                itemStacks[i] = ByteBufUtils.readItemStack(buffer);
            }
        }
    }
    @Override
    public void handleClientSide(EntityPlayer player) {
        if(ChestViewer.instance == null) {
            return;
        }

        if(ChestViewer.instance.chestMap == null) {
            return;
        }

        ChestViewer.instance.chestMap.put(new PathPoint(x, y, z), itemStacks);
    }

    @Override
    public void handleServerSide(EntityPlayer player) {

        EntityPlayerMP entityPlayer = (EntityPlayerMP)player;
        TileEntity tileEntity = entityPlayer.worldObj.getTileEntity(x, y, z);
        if(tileEntity == null) {
            return;
        }

        if(tileEntity instanceof TileEntityChest) {
            TileEntityChest chest = (TileEntityChest) tileEntity;
            TileEntityChest[] chests = {chest.adjacentChestXNeg, chest.adjacentChestZNeg, chest, chest.adjacentChestXPos, chest.adjacentChestZPos};

            int invSize = 0;
            for (TileEntityChest c : chests) {
                if (c == null) {
                    continue;
                }
                invSize += c.getSizeInventory();
            }
            ItemStack[] itemStacks = new ItemStack[invSize];

            int index = 0;
            for (TileEntityChest c : chests) {
                if (c == null) {
                    continue;
                }

                for (int i = 0; i < c.getSizeInventory(); ++i) {
                    itemStacks[index++] = c.getStackInSlot(i);
                }
            }

            sendPacket(x, y, z, itemStacks);
        }
    }

	private void sendPacket(int x, int y, int z, ItemStack[] itemStacks) {
        ChestViewer.packetPipeline.sendPacketToAllPlayer(new PacketHandler(EnumCommand.RESPONSE, x, y, z, itemStacks));
	}


}