#include "common.h"
#include "ZCompressor.h"
#include "WorldSession.h"
#include "UpdateData.h"
#include "UpdateFields.h"
#include "Object.h"
#include "ObjMgr.h"
#include "UpdateMask.h"


void WorldSession::_HandleCompressedUpdateObjectOpcode(WorldPacket& recvPacket)
{
    //printf("-> COMPRESSED_UPDATE_OBJECT, pktlen=%u\n",recvPacket.size());
    //recvPacket.hexlike();
    uint32 realsize;
    recvPacket >> realsize;
    ZCompressor z;
    z.append(recvPacket.contents() + sizeof(uint32),recvPacket.size() - sizeof(uint32));
    z.Compressed(true);
    z.RealSize(realsize);
    z.Inflate();
    if(z.Compressed())
    {
        logerror("_HandleCompressedUpdateObjectOpcode(): Inflate() failed!");
        return;
    }
    WorldPacket wp;
    wp.SetOpcode(recvPacket.GetOpcode());
    wp.append(z.contents(),z.size());
 
    _HandleUpdateObjectOpcode(wp);
}

void WorldSession::_HandleUpdateObjectOpcode(WorldPacket& recvPacket)
{
	//recvPacket.hexlike();
	uint8 utype;
	uint8 unk8;
	uint32 usize, ublocks;
	uint64 uguid;
	recvPacket >> ublocks >> unk8;
	logdebug("UpdateObject: ublocks=%u unk=%u",ublocks,unk8);
	//while(true) // need to read full packet as soon as the structure is 100% known & implemented
	// for now reading first object is enough
	{
		recvPacket >> utype;
		logdebug("UpdateObject: utype=%u",utype);
		switch(utype)
		{
		case UPDATETYPE_VALUES:
			{
				uint8 blockcount, masksize;
				uint32 value, valuesCount = 1500;
				uguid = recvPacket.GetPackedGuid();
				recvPacket >> blockcount;
				masksize = blockcount * 4;
				logdebug("UPDATETYPE_VALUES: guid="I64FMT" blockcount=%u masksize=%d",uguid,blockcount, masksize);

				UpdateMask umask;
				umask.SetCount(masksize);
				uint32 *updateMask = new uint32[100];
				recvPacket.read((uint8*)updateMask, masksize);
				umask.SetMask(updateMask);

				for (uint32 i = 0; i < valuesCount; i++) // How do i get valuesCount?
				{
					if (umask.GetBit(i))
					{
						recvPacket >> value;
						//logdebug("Value (%d): %d", i, value);
					}
				}

			}
			break;

		case UPDATETYPE_MOVEMENT:
			{
                recvPacket >> uguid;
                Object *obj = objmgr.GetObj(uguid);
                if(obj)
				    this->_MovementUpdate(obj->GetTypeId(),uguid,recvPacket);
                else
                    logerror("Got UpdateObject_Movement for unknown object "I64FMT,uguid);
			}
			break;

		case UPDATETYPE_CREATE_OBJECT:
		case UPDATETYPE_CREATE_OBJECT2:
			{
				uguid = recvPacket.GetPackedGuid();
				uint8 objtypeid, flags;
				recvPacket >> objtypeid >> flags;
				logdebug("Create Object type %u with guid "I64FMT,objtypeid,uguid);

				this->_MovementUpdate(objtypeid, uguid, recvPacket);

				// (TODO) and then: Add object to objmgr
			}
			break;

		case UPDATETYPE_OUT_OF_RANGE_OBJECTS:
			recvPacket >> usize;
			for(uint16 i=0;i<usize;i++)
			{
				uguid = recvPacket.GetPackedGuid(); // not 100% sure if this is correct
				logdebug("GUID "I64FMT" out of range",uguid);
				// TODO: delete object from known objects list
			}
			break;

		default:
            logerror("Got unk updatetype 0x%X",utype);
			break;
		}
	}
}

void WorldSession::_MovementUpdate(uint8 objtypeid, uint64 uguid, WorldPacket& recvPacket)
{
	if(objtypeid==TYPEID_PLAYER)
	{
		uint32 flags, flags2, time;
		uint64 tguid,guid;
		float nul;
		float x, y, z, o;
		float tx, ty, tz, to;
		float speedWalk, speedRun, speedSwimBack, speedSwim, speedWalkBack, speedTurn;

		recvPacket >> guid >> flags >> flags2 >> time;

		if (flags2 & 0x02000000) // On a transport
		{
			recvPacket >> x >> y >> z >> o >> tguid >> tx >> ty >> tz >> to; 
		}
		else
		{
			recvPacket >> x >> y >> z >> o;
		}

		recvPacket >> nul;

		if(flags2 & 0x2000) // Self update
		{
			// What is this data used for?
			recvPacket >> nul;
			recvPacket >> nul;
			recvPacket >> nul;
			recvPacket >> nul;
		}

		recvPacket >> speedWalk >> speedRun >> speedSwimBack >> speedSwim >> speedWalkBack >> speedTurn;

		logdebug("TYPEID_PLAYER: OnTransport=%s x=%d y=%d z=%d o=%d", flags2 & 0x02000000 ? "true" : "false", x, y, z, o);
	}
	if(objtypeid==TYPEID_UNIT)
	{
		uint32 flags2, unk, posCount;
		float nul, unkf;
		float x, y, z, o;
		float speedWalk, speedRun, speedSwimBack, speedSwim, speedWalkBack, speedTurn;

		recvPacket >> flags2 >> unk >> x >> y >> z >> o >> nul;
		recvPacket >> speedWalk >> speedRun >> speedSwimBack >> speedSwim >> speedWalkBack >> speedTurn;

		if (flags2 & 0x400000)
		{
			recvPacket >> unk >> unk >> unk >> unk >> posCount;

			for (uint8 i = 0; i < posCount + 1; i++)
			{
				recvPacket >> unkf >> unkf >> unkf; // Some x, y, z value
			}
		}

		logdebug("TYPEID_UNIT: 0x400000 flag=%s x=%d y=%d z=%d o=%d", flags2 & 0x400000 ? "true" : "false", x, y, z, o);
	}
	if( (objtypeid==TYPEID_CORPSE) || (objtypeid==TYPEID_GAMEOBJECT) || (objtypeid==TYPEID_DYNAMICOBJECT))
	{
		float x, y, z, o;

		recvPacket >> x >> y >> z >> o;

		// TODO: Check for transport and corpse extra data
	}
}