JSPresetUnit.cpp

#include "JSPresetUnit.h"

#include "D2Ptrs.h"
#include "CriticalSections.h"
#include "D2Helpers.h"

EMPTY_CTOR(presetunit)

void presetunit_finalize(JSFreeOp *fop, JSObject *obj)
{
	myPresetUnit *pUnit = (myPresetUnit*)JS_GetPrivate(obj);

	if(pUnit)
	{
		JS_SetPrivate(obj, NULL);
		delete pUnit;
	}
}

JSAPI_PROP(presetunit_getProperty)
{
	myPresetUnit* pUnit = (myPresetUnit*)JS_GetPrivate(cx, obj);

	if(!pUnit)
		return JS_TRUE;

	jsval ID;
	JS_IdToValue(cx,id,&ID);
	switch(JSVAL_TO_INT(ID))
	{
		case PUNIT_TYPE:
			vp.setInt32(pUnit->dwType);
			break;
		case PUNIT_ROOMX:
			vp.setInt32(pUnit->dwRoomX);
			break;
		case PUNIT_ROOMY:
			vp.setInt32(pUnit->dwRoomY);
			break;
		case PUNIT_X:
			vp.setInt32(pUnit->dwPosX);
			break;
		case PUNIT_Y:
			vp.setInt32(pUnit->dwPosY);
			break;
		case PUNIT_ID:
			vp.setInt32(pUnit->dwId);
			break;
		case PUINT_LEVEL:
			vp.setInt32(pUnit->dwLevel);
		default:
			break;
	}
	return JS_TRUE;
}

JSAPI_FUNC(my_getPresetUnits)
{
	if(!WaitForGameReady())
		THROW_WARNING(cx, vp,  "Game not ready");

	if(argc < 1)
	{
		JS_SET_RVAL(cx, vp, JSVAL_FALSE);
		return JS_TRUE;
	}

	uint32 levelId;
	JS_BeginRequest(cx);
	JS_ValueToECMAUint32(cx, JS_ARGV(cx, vp)[0], &levelId);
	JS_EndRequest(cx);
	Level* pLevel = GetLevel(levelId);

	if(!pLevel)
		THROW_ERROR(cx, "getPresetUnits failed, couldn't access the level!");

	uint nClassId = NULL;
	uint nType = NULL;

	if(argc >= 2)
		nType = JSVAL_TO_INT(JS_ARGV(cx, vp)[1]);
	if(argc >= 3)
		nClassId = JSVAL_TO_INT(JS_ARGV(cx, vp)[2]);

	AutoCriticalRoom* cRoom = new AutoCriticalRoom;

	bool bAddedRoom = FALSE;
	DWORD dwArrayCount = NULL;

	JSObject* pReturnArray = JS_NewArrayObject(cx, 0, NULL);
	JS_BeginRequest(cx);
	JS_AddRoot(cx, &pReturnArray);
	for(Room2 *pRoom = pLevel->pRoom2First; pRoom; pRoom = pRoom->pRoom2Next)
	{
		bAddedRoom = FALSE;

		if(!pRoom->pPreset)
		{
			D2COMMON_AddRoomData(D2CLIENT_GetPlayerUnit()->pAct, pLevel->dwLevelNo, pRoom->dwPosX, pRoom->dwPosY, D2CLIENT_GetPlayerUnit()->pPath->pRoom1);
			bAddedRoom = TRUE;
		}
		
		for(PresetUnit* pUnit = pRoom->pPreset; pUnit; pUnit = pUnit->pPresetNext)
		{
			// Does it fit?
			if((nType == NULL || pUnit->dwType == nType) && (nClassId == NULL || pUnit->dwTxtFileNo == nClassId))
			{
				myPresetUnit* mypUnit = new myPresetUnit;

				mypUnit->dwPosX = pUnit->dwPosX;
				mypUnit->dwPosY = pUnit->dwPosY;
				mypUnit->dwRoomX = pRoom->dwPosX;
				mypUnit->dwRoomY = pRoom->dwPosY;
				mypUnit->dwType = pUnit->dwType;
				mypUnit->dwId = pUnit->dwTxtFileNo;
				mypUnit->dwLevel = levelId;

				JSObject* unit = BuildObject(cx, &presetunit_class, NULL, presetunit_props, mypUnit);
				if(!unit)
				{
					delete mypUnit;
					JS_EndRequest(cx);
					delete cRoom;
					THROW_ERROR(cx, "Failed to build object?");
				}

				jsval a = OBJECT_TO_JSVAL(unit);
				JS_SetElement(cx, pReturnArray, dwArrayCount, &a);

				dwArrayCount++;
			}
		}

		if(bAddedRoom)
		{
			D2COMMON_RemoveRoomData(D2CLIENT_GetPlayerUnit()->pAct, pLevel->dwLevelNo, pRoom->dwPosX, pRoom->dwPosY, D2CLIENT_GetPlayerUnit()->pPath->pRoom1);
			bAddedRoom = FALSE;			
		}
	}
	JS_SET_RVAL(cx, vp, OBJECT_TO_JSVAL(pReturnArray));
	
	JS_RemoveRoot(cx, &pReturnArray);
	JS_EndRequest(cx);
	delete cRoom;
	return JS_TRUE;
}

JSAPI_FUNC(my_getPresetUnit)
{
	if(!WaitForGameReady())
		THROW_WARNING(cx, vp,  "Game not ready");

	if(argc < 1)
	{
		JS_SET_RVAL(cx, vp, JSVAL_FALSE);
		return JS_TRUE;
	}

	uint32 levelId;
	JS_BeginRequest(cx);
	JS_ValueToECMAUint32(cx, JS_ARGV(cx, vp)[0], &levelId);
	JS_EndRequest(cx);
	Level* pLevel = GetLevel(levelId);

	if(!pLevel)
		THROW_ERROR(cx, "getPresetUnits failed, couldn't access the level!");

	DWORD nClassId = NULL;
	DWORD nType = NULL;

	if(argc >= 2)
		nType = JSVAL_TO_INT(JS_ARGV(cx, vp)[1]);
	if(argc >= 3)
		nClassId = JSVAL_TO_INT(JS_ARGV(cx, vp)[2]);

	AutoCriticalRoom* cRoom = new AutoCriticalRoom;

	bool bAddedRoom = FALSE;

	for(Room2 *pRoom = pLevel->pRoom2First; pRoom; pRoom = pRoom->pRoom2Next) {

		bAddedRoom = FALSE;

		if(!pRoom->pRoom1)
		{
			D2COMMON_AddRoomData(D2CLIENT_GetPlayerUnit()->pAct, pLevel->dwLevelNo, pRoom->dwPosX, pRoom->dwPosY, D2CLIENT_GetPlayerUnit()->pPath->pRoom1);
			bAddedRoom = TRUE;
		}

		for(PresetUnit* pUnit = pRoom->pPreset; pUnit; pUnit = pUnit->pPresetNext)
		{
			// Does it fit?
			if((nType == NULL || pUnit->dwType == nType) && (nClassId == NULL || pUnit->dwTxtFileNo == nClassId))
			{
				// Yes it fits! Return it
				myPresetUnit* mypUnit = new myPresetUnit;

				mypUnit->dwPosX = pUnit->dwPosX;
				mypUnit->dwPosY = pUnit->dwPosY;
				mypUnit->dwRoomX = pRoom->dwPosX;
				mypUnit->dwRoomY = pRoom->dwPosY;
				mypUnit->dwType = pUnit->dwType;
				mypUnit->dwId = pUnit->dwTxtFileNo;
				mypUnit->dwLevel = levelId;

				JSObject* obj = BuildObject(cx, &presetunit_class, NULL, presetunit_props, mypUnit);
				if(!obj)
				{
					delete cRoom;
					delete mypUnit;
					THROW_ERROR(cx, "Failed to create presetunit object");
				}
				JS_SET_RVAL(cx, vp, OBJECT_TO_JSVAL(obj));
				delete cRoom;
				return JS_TRUE;
			}
		}

		if(bAddedRoom)
		{
			D2COMMON_RemoveRoomData(D2CLIENT_GetPlayerUnit()->pAct, pLevel->dwLevelNo, pRoom->dwPosX, pRoom->dwPosY, D2CLIENT_GetPlayerUnit()->pPath->pRoom1);
			bAddedRoom = FALSE;			
		}
	}
	JS_SET_RVAL(cx, vp, JSVAL_FALSE);
	delete cRoom;
	return JS_TRUE;
}