/**

 * vim: set ts=4 :

 * ================================================================

 * SourceMod

 * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.

 * ================================================================

 *

 * This program is free software; you can redistribute it and/or 

 * modify it under the terms of the GNU General Public License, 

 * version 3.0, as published by the Free Software Foundation.

 * 

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program.  If not, see <http://www.gnu.org/licenses/>.

 *

 * As a special exception, AlliedModders LLC gives you permission to 

 * link the code of this program (as well as its derivative works) to 

 * "Half-Life 2," the "Source Engine," the "SourcePawn JIT," and any 

 * Game MODs that run on software by the Valve Corporation.  You must 

 * obey the GNU General Public License in all respects for all other 

 * code used. Additionally, AlliedModders LLC grants this exception 

 * to all derivative works. AlliedModders LLC defines further 

 * exceptions, found in LICENSE.txt (as of this writing, version 

 * JULY-31-2007), or <http://www.sourcemod.net/license.php>.

 *

 * Version: $Id$

 */

#include "sourcemod.h"

#include "sourcemm_api.h"

#include "sm_stringutil.h"

#include "HandleSys.h"

#include <KeyValues.h>

HandleType_t g_KeyValueType;

struct KeyValueStack 

{

	KeyValues *pBase;

	CStack<KeyValues *> pCurRoot;

};

class KeyValueNatives : 

	public SMGlobalClass,

	public IHandleTypeDispatch

{

public:

	void OnSourceModAllInitialized()

	{

		g_KeyValueType = g_HandleSys.CreateType("KeyValues", this, 0, NULL, NULL, g_pCoreIdent, NULL);

	}

	void OnSourceModShutdown()

	{

		g_HandleSys.RemoveType(g_KeyValueType, g_pCoreIdent);

		g_KeyValueType = 0;

	}

	void OnHandleDestroy(HandleType_t type, void *object)

	{

		KeyValueStack *pStk = reinterpret_cast<KeyValueStack *>(object);

		pStk->pBase->deleteThis();

		delete pStk;

	}

};

KeyValues *SourceModBase::ReadKeyValuesHandle(Handle_t hndl, HandleError *err, bool root)

{

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		if (err)

		{

			*err = herr;

		}

		return NULL;

	}

	if (err)

	{

		*err = HandleError_None;

	}

	return (root) ? pStk->pBase : pStk->pCurRoot.front();

}

static cell_t smn_KvSetString(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	char *key, *value;

	pCtx->LocalToStringNULL(params[2], &key);

	pCtx->LocalToString(params[3], &value);

	pStk->pCurRoot.front()->SetString(key, value);

	return 1;

}

static cell_t smn_KvSetNum(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	char *key;

	pCtx->LocalToStringNULL(params[2], &key);

	pStk->pCurRoot.front()->SetInt(key, params[3]);

	return 1;

}

static cell_t smn_KvSetUInt64(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	char *key;

	cell_t *addr;

	uint64 value;

	pCtx->LocalToStringNULL(params[2], &key);

	pCtx->LocalToPhysAddr(params[3], &addr);

	value = static_cast<uint64>(*addr);

	pStk->pCurRoot.front()->SetUint64(key, value);

	return 1;

}

static cell_t smn_KvSetFloat(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	char *key;

	pCtx->LocalToStringNULL(params[2], &key);

	pStk->pCurRoot.front()->SetFloat(key, sp_ctof(params[3]));

	return 1;

}

static cell_t smn_KvSetColor(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	char *key;

	pCtx->LocalToStringNULL(params[2], &key);

	Color color(params[3], params[4], params[5], params[6]);

	pStk->pCurRoot.front()->SetColor(key, color);

	return 1;

}

static cell_t smn_KvGetString(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	const char *value;

	char *key, *defvalue;

	pCtx->LocalToStringNULL(params[2], &key);

	pCtx->LocalToString(params[5], &defvalue);

	value = pStk->pCurRoot.front()->GetString(key, defvalue);

	pCtx->StringToLocalUTF8(params[3], params[4], value, NULL);

	return 1;

}

static cell_t smn_KvGetNum(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	int value;

	char *key;

	pCtx->LocalToStringNULL(params[2], &key);

	value = pStk->pCurRoot.front()->GetInt(key, params[3]);

	return value;

}

static cell_t smn_KvGetFloat(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	float value;

	char *key;

	pCtx->LocalToStringNULL(params[2], &key);

	value = pStk->pCurRoot.front()->GetFloat(key, sp_ctof(params[3]));

	return sp_ftoc(value);

}

static cell_t smn_KvGetColor(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	Color color;

	char *key;

	cell_t *r, *g, *b, *a;

	pCtx->LocalToStringNULL(params[2], &key);

	pCtx->LocalToPhysAddr(params[3], &r);

	pCtx->LocalToPhysAddr(params[4], &g);

	pCtx->LocalToPhysAddr(params[5], &b);

	pCtx->LocalToPhysAddr(params[6], &a);

	color = pStk->pCurRoot.front()->GetColor(key);

	*r = color.r();

	*g = color.g();

	*b = color.b();

	*a = color.a();

	return 1;

}

static cell_t smn_KvGetUInt64(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	char *key;

	cell_t *addr, *defvalue;

	uint64 value;

	pCtx->LocalToStringNULL(params[2], &key);

	pCtx->LocalToPhysAddr(params[3], &addr);

	pCtx->LocalToPhysAddr(params[4], &defvalue);

	value = pStk->pCurRoot.front()->GetUint64(key, static_cast<uint64>(*defvalue));

	*reinterpret_cast<uint64 *>(addr) = value;

	return 1;

}

static cell_t smn_CreateKeyValues(IPluginContext *pCtx, const cell_t *params)

{

	KeyValueStack *pStk;

	char *name, *firstkey, *firstvalue;

	bool is_empty;

	pCtx->LocalToString(params[1], &name);

	pCtx->LocalToString(params[2], &firstkey);

	pCtx->LocalToString(params[3], &firstvalue);

	is_empty = (firstkey[0] == '\0');

	pStk = new KeyValueStack;

	pStk->pBase = new KeyValues(name, is_empty ? NULL : firstkey, (is_empty||(firstvalue[0]=='\0')) ? NULL : firstvalue);

	pStk->pCurRoot.push(pStk->pBase);

	return g_HandleSys.CreateHandle(g_KeyValueType, pStk, pCtx->GetIdentity(), g_pCoreIdent, NULL);

}

static cell_t smn_KvJumpToKey(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	char *name;	

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	pCtx->LocalToString(params[2], &name);

	KeyValues *pSubKey = pStk->pCurRoot.front();

	pSubKey = pSubKey->FindKey(name, (params[3]) ? true : false);

	if (!pSubKey)

	{

		return 0;

	}

	pStk->pCurRoot.push(pSubKey);

	return 1;

}

static cell_t smn_KvGotoFirstSubKey(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	KeyValues *pSubKey = pStk->pCurRoot.front();

	KeyValues *pFirstSubKey;

	if (params[2])

	{

		pFirstSubKey = pSubKey->GetFirstTrueSubKey();

	} else {

		pFirstSubKey = pSubKey->GetFirstSubKey();

	}

	if (!pFirstSubKey)

	{

		return 0;

	}

	pStk->pCurRoot.push(pFirstSubKey);

	return 1;

}

static cell_t smn_KvGotoNextKey(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	KeyValues *pSubKey = pStk->pCurRoot.front();

	if (params[2])

	{

		pSubKey = pSubKey->GetNextTrueSubKey();

	} else {

		pSubKey = pSubKey->GetNextKey();

	}

	if (!pSubKey)

	{

		return 0;

	}

	pStk->pCurRoot.pop();

	pStk->pCurRoot.push(pSubKey);

	return 1;	

}

static cell_t smn_KvJumpNextSubKey(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	KeyValues *pSubKey = pStk->pCurRoot.front();

	KeyValues *pNextKey = pSubKey->GetNextKey();

	if (!pNextKey)

	{

		return 0;

	}

	pStk->pCurRoot.push(pNextKey);

	return 1;	

}

static cell_t smn_KvGoBack(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}

	if (pStk->pCurRoot.size() == 1)

	{

		return 0;

	}

	pStk->pCurRoot.pop();

	return 1;

}

static cell_t smn_KvRewind(IPluginContext *pCtx, const cell_t *params)

{

	Handle_t hndl = static_cast<Handle_t>(params[1]);

	HandleError herr;

	HandleSecurity sec;

	KeyValueStack *pStk;

	sec.pOwner = NULL;

	sec.pIdentity = g_pCoreIdent;

	if ((herr=g_HandleSys.ReadHandle(hndl, g_KeyValueType, &sec, (void **)&pStk))

		!= HandleError_None)

	{

		return pCtx->ThrowNativeError("Invalid key value handle %x (error %d)", hndl, herr);

	}