package fr.impuuu.havenrp.packets;

import cpw.mods.fml.common.network.FMLEmbeddedChannel;
import cpw.mods.fml.common.network.FMLOutboundHandler;
import cpw.mods.fml.common.network.FMLOutboundHandler.OutboundTarget;
import cpw.mods.fml.common.network.NetworkRegistry;
import cpw.mods.fml.common.network.internal.FMLProxyPacket;
import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;
import io.netty.channel.Channel;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandler.Sharable;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.codec.MessageToMessageCodec;
import io.netty.util.Attribute;
import java.util.Collections;
import java.util.Comparator;
import java.util.EnumMap;
import java.util.LinkedList;
import java.util.List;
import net.minecraft.client.Minecraft;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.entity.player.EntityPlayerMP;
import net.minecraft.network.INetHandler;
import net.minecraft.network.NetHandlerPlayServer;

@ChannelHandler.Sharable
public class PacketHelper
  extends MessageToMessageCodec<FMLProxyPacket, AbstractPacketHelper>
{
  private EnumMap<Side, FMLEmbeddedChannel> channels;
  private LinkedList<Class<? extends AbstractPacketHelper>> packets = new LinkedList();
  private boolean isPostInitialised = false;

  public boolean registerPacket(Class<? extends AbstractPacketHelper> clazz)
  {
    if (this.packets.size() > 256) {
      return false;
    }
    if (this.packets.contains(clazz)) {
      return false;
    }
    if (this.isPostInitialised) {
      return false;
    }
    this.packets.add(clazz);
    return true;
  }

  protected void encode(ChannelHandlerContext ctx, AbstractPacketHelper msg, List<Object> out)
    throws Exception
  {
    ByteBuf buffer = Unpooled.buffer();
    Class<? extends AbstractPacketHelper> clazz = msg.getClass();
    if (!this.packets.contains(msg.getClass())) {
      throw new NullPointerException("No Packet Registered for: " + msg.getClass().getCanonicalName());
    }
    byte discriminator = (byte)this.packets.indexOf(clazz);
    buffer.writeByte(discriminator);
    msg.encodeInto(ctx, buffer);
    FMLProxyPacket proxyPacket = new FMLProxyPacket(buffer.copy(), (String)ctx.channel().attr(NetworkRegistry.FML_CHANNEL).get());
    out.add(proxyPacket);
  }

  protected void decode(ChannelHandlerContext ctx, FMLProxyPacket msg, List<Object> out)
    throws Exception
  {
    ByteBuf payload = msg.payload();
    byte discriminator = payload.readByte();
    Class<? extends AbstractPacketHelper> clazz = (Class)this.packets.get(discriminator);
    if (clazz == null) {
      throw new NullPointerException("No packet registered for discriminator: " + discriminator);
    }
    AbstractPacketHelper pkt = (AbstractPacketHelper)clazz.newInstance();
    pkt.decodeInto(ctx, payload.slice());
    switch (cpw.mods.fml.common.FMLCommonHandler.instance().getEffectiveSide())
    {
    case CLIENT:
      EntityPlayer player = getClientPlayer();
      pkt.handleClientSide(player);
      break;
    case SERVER:
      INetHandler netHandler = (INetHandler)ctx.channel().attr(NetworkRegistry.NET_HANDLER).get();
      EntityPlayer player1 = ((NetHandlerPlayServer)netHandler).playerEntity;
      pkt.handleServerSide(player1);
    }
  }

  public void initalise()
  {
    this.channels = NetworkRegistry.INSTANCE.newChannel("HavenRP", new ChannelHandler[] { this });
    registerPackets();
  }

  public void registerPackets()
  {
    registerPacket(PacketMessage.class);
  }

  public void postInitialise()
  {
    if (this.isPostInitialised) {
      return;
    }
    this.isPostInitialised = true;
    Collections.sort(this.packets, new Comparator()
    {
      public int compare(Class<? extends AbstractPacketHelper> clazz1, Class<? extends AbstractPacketHelper> clazz2)
      {
        int com = String.CASE_INSENSITIVE_ORDER.compare(clazz1.getCanonicalName(), clazz2.getCanonicalName());
        if (com == 0) {
          com = clazz1.getCanonicalName().compareTo(clazz2.getCanonicalName());
        }
        return com;
      }

      public int compare(Object o1, Object o2)
      {
        return 0;
      }
    });
  }

  @SideOnly(Side.CLIENT)
  private EntityPlayer getClientPlayer()
  {
    return Minecraft.getMinecraft().thePlayer;
  }

  public void sendToAll(AbstractPacketHelper message)
  {
    ((FMLEmbeddedChannel)this.channels.get(Side.SERVER)).attr(FMLOutboundHandler.FML_MESSAGETARGET).set(FMLOutboundHandler.OutboundTarget.ALL);
    ((FMLEmbeddedChannel)this.channels.get(Side.SERVER)).writeAndFlush(message);
  }

  public void sendTo(AbstractPacketHelper message, EntityPlayerMP player)
  {
    ((FMLEmbeddedChannel)this.channels.get(Side.SERVER)).attr(FMLOutboundHandler.FML_MESSAGETARGET).set(FMLOutboundHandler.OutboundTarget.PLAYER);
    ((FMLEmbeddedChannel)this.channels.get(Side.SERVER)).attr(FMLOutboundHandler.FML_MESSAGETARGETARGS).set(player);
    ((FMLEmbeddedChannel)this.channels.get(Side.SERVER)).writeAndFlush(message);
  }

  public void sendToServer(AbstractPacketHelper message)
  {
    ((FMLEmbeddedChannel)this.channels.get(Side.CLIENT)).attr(FMLOutboundHandler.FML_MESSAGETARGET).set(FMLOutboundHandler.OutboundTarget.TOSERVER);
    ((FMLEmbeddedChannel)this.channels.get(Side.CLIENT)).writeAndFlush(message);
  }
}