PacketHandler.java

package mrolas.muchMoney.common.network;

import java.io.ByteArrayOutputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import mrolas.muchMoney.common.gui.container.ContainerCoinBank;
import mrolas.muchMoney.common.gui.container.ContainerCoinTrader;
import mrolas.muchMoney.common.gui.container.ContainerShopStop;
import mrolas.muchMoney.common.lib.Reference;
import mrolas.muchMoney.common.tileEntities.TileEntityCoinBank;
import mrolas.muchMoney.common.tileEntities.TileEntityCoinTrader;
import mrolas.muchMoney.common.tileEntities.TileEntityShopStop;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.Container;
import net.minecraft.network.INetworkManager;
import net.minecraft.network.packet.Packet250CustomPayload;

import com.google.common.io.ByteArrayDataInput;
import com.google.common.io.ByteStreams;

import cpw.mods.fml.common.network.IPacketHandler;
import cpw.mods.fml.common.network.PacketDispatcher;
import cpw.mods.fml.common.network.Player;

public class PacketHandler
    implements IPacketHandler
{
    @Override
    public void onPacketData(INetworkManager manager, Packet250CustomPayload packet, Player player)
    {
        ByteArrayDataInput reader = ByteStreams.newDataInput(packet.data);

        EntityPlayer entityPlayer = (EntityPlayer)player;

        byte packetId = reader.readByte();

        switch(packetId)
        {
            case 0:
                byte buttonId = reader.readByte();
                Container container = entityPlayer.openContainer;
                if (container != null)
                {
                    if (container instanceof ContainerCoinTrader)
                    {
                        TileEntityCoinTrader coinTrader = ((ContainerCoinTrader)container).getCoinTrader();
                        coinTrader.receiveButtonEvent(buttonId);
                    }
                    if (container instanceof ContainerCoinBank)
                    {
                        TileEntityCoinBank coinBank = ((ContainerCoinBank)container).getCoinBank();
                        coinBank.receiveButtonEvent(buttonId);
                    }
                    if (container instanceof ContainerShopStop)
                    {
                        TileEntityShopStop shopStop = ((ContainerShopStop)container).getShopStop();
                        shopStop.receiveButtonEvent(buttonId);
                    }
                }
                break;
        }
    }

    public static void sendButtonPacket (byte id)
    {
        ByteArrayOutputStream byteStream = new ByteArrayOutputStream();
        DataOutputStream dataStream = new DataOutputStream(byteStream);

        try
        {
            dataStream.writeByte((byte)0);
            dataStream.writeByte(id);

            PacketDispatcher.sendPacketToServer(PacketDispatcher.getPacket(Reference.CHANNEL, byteStream.toByteArray()));
        }
        catch (IOException e)
        {
            System.err.append("[Much Money] Failed to send button click packet");
        }
    }
}