import minecraft.networking.packets
from protocol import packets

def get_packets(old_get_packets):
    def wrapper(func, context):
        print('Monkey-patch worked.')
        mc_packets = func(context)

        # add any custom packets here
        mc_packets.add(packets.ChunkDataPacket)
        mc_packets.add(packets.AcknowledgePlayerDiggingPacket)
        mc_packets.add(packets.BlockBreakAnimationPacket)
        mc_packets.add(packets.SetSlotPacket)
        mc_packets.add(packets.PlayerDiggingPacket)
        mc_packets.add(packets.PickItemPacket)
        mc_packets.add(packets.HeldItemChangePacket)
        mc_packets.add(packets.OpenWindowPacket)
        mc_packets.add(packets.CloseWindowPacket)
        mc_packets.add(packets.ClickWindowPacket)
        mc_packets.add(packets.ClientWindowConfirmationPacket)
        mc_packets.add(packets.ServerWindowConfirmationPacket)
        mc_packets.add(packets.EntityMetadataPacket)
        mc_packets.add(packets.SpawnLivingEntityPacket)
        mc_packets.add(packets.EntityPositionRotationPacket)
        mc_packets.add(packets.DestroyEntitiesPacket)
        #mc_packets.add(packets.EntityActionPacket)
        mc_packets.add(packets.SpawnPlayerPacket)

        return mc_packets
    return lambda x: wrapper(old_get_packets, x)

minecraft.networking.packets.clientbound.play.get_packets = get_packets(minecraft.networking.packets.clientbound.play.get_packets)