/**

 * vim: set ts=4 :

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

 * SourceMod

 * Copyright (C) 2013 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 "UserMessages.h"

#ifdef USE_PROTOBUF_USERMESSAGES

#include "HandleSys.h"

#include "UserMessagePBHelpers.h"

#include "smn_usermsgs.h"

// Assumes pbuf message handle is param 1, gets message as msg

#define GET_MSG_FROM_HANDLE_OR_ERR()                  \

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

	HandleError herr;                                 \

	HandleSecurity sec;                               \

	SMProtobufMessage *msg;                           \

	                                                  \

	sec.pOwner = NULL;                                \

	sec.pIdentity = g_pCoreIdent;                     \

	                                                  \

	if ((herr=g_HandleSys.ReadHandle(hndl, g_ProtobufType, &sec, (void **)&msg)) \

		!= HandleError_None)                          \

	{                                                 \

		return pCtx->ThrowNativeError("Invalid protobuf message handle %x (error %d)", hndl, herr); \

	}

// Assumes message field name is param 2, gets as strField

#define GET_FIELD_NAME_OR_ERR()                                           \

	int err;                                                              \

	char *strField;                                                       \

	if ((err=pCtx->LocalToString(params[2], &strField)) != SP_ERROR_NONE) \

	{                                                                     \

		pCtx->ThrowNativeErrorEx(err, NULL);                              \

		return 0;                                                         \

	}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	int ret;

	if (!msg->GetInt32OrUnsigned(strField, &ret))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return ret;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	float ret;

	if (!msg->GetFloatOrDouble(strField, &ret))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return sp_ftoc(ret);

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	bool ret;

	if (!msg->GetBool(strField, &ret))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return ret ? 1 : 0;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	char *buf;

	pCtx->LocalToPhysAddr(params[3], (cell_t **)&buf);

	if (!msg->GetString(strField, buf, params[4]))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	Color clr;

	if (!msg->GetColor(strField, &clr))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = clr.r();

	out[1] = clr.g();

	out[2] = clr.b();

	out[3] = clr.a();

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	QAngle ang;

	if (!msg->GetQAngle(strField, &ang))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = sp_ftoc(ang.x);

	out[1] = sp_ftoc(ang.y);

	out[2] = sp_ftoc(ang.z);

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	Vector vec;

	if (!msg->GetVector(strField, &vec))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = sp_ftoc(vec.x);

	out[1] = sp_ftoc(vec.y);

	out[2] = sp_ftoc(vec.z);

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	Vector2D vec;

	if (!msg->GetVector2D(strField, &vec))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = sp_ftoc(vec.x);

	out[1] = sp_ftoc(vec.y);

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	int cnt = msg->GetRepeatedFieldCount(strField);

	if (cnt == -1)

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return cnt;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	int ret;

	if (!msg->GetRepeatedInt32OrUnsigned(strField, params[3], &ret))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return ret;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	float ret;

	if (!msg->GetRepeatedFloatOrDouble(strField, params[3], &ret))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return sp_ftoc(ret);

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	bool ret;

	if (!msg->GetRepeatedBool(strField, params[3], &ret))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return ret ? 1 : 0;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	char *buf;

	pCtx->LocalToPhysAddr(params[4], (cell_t **)&buf);

	if (!msg->GetRepeatedString(strField, params[3], buf, params[5]))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	Color clr;

	if (!msg->GetRepeatedColor(strField, params[3], &clr))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = clr.r();

	out[1] = clr.g();

	out[2] = clr.b();

	out[3] = clr.a();

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	QAngle ang;

	if (!msg->GetRepeatedQAngle(strField, params[3], &ang))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = sp_ftoc(ang.x);

	out[1] = sp_ftoc(ang.y);

	out[2] = sp_ftoc(ang.z);

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	Vector vec;

	if (!msg->GetRepeatedVector(strField, params[3], &vec))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = sp_ftoc(vec.x);

	out[1] = sp_ftoc(vec.y);

	out[2] = sp_ftoc(vec.z);

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *out;

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

	Vector2D vec;

	if (!msg->GetRepeatedVector2D(strField, params[3], &vec))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\"[%d] for message \"%s\"", strField, params[3], msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	out[0] = sp_ftoc(vec.x);

	out[1] = sp_ftoc(vec.y);

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	if (!msg->SetInt32OrUnsigned(strField, params[3]))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	if (!msg->SetFloatOrDouble(strField, sp_ctof(params[3])))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	bool value = (params[3] == 0 ? false : true);

	if (!msg->SetBool(strField, value))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	char *strValue;

	if ((err=pCtx->LocalToString(params[3], &strValue)) != SP_ERROR_NONE)

	{

		pCtx->ThrowNativeErrorEx(err, NULL);

		return 0;

	}

	if (!msg->SetString(strField, strValue))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *clrParams;

	if ((err=pCtx->LocalToPhysAddr(params[3], &clrParams)) != SP_ERROR_NONE)

	{

		pCtx->ThrowNativeErrorEx(err, NULL);

		return 0;

	}

	Color clr(

		clrParams[0],

		clrParams[1],

		clrParams[2],

		clrParams[3]);

	if (!msg->SetColor(strField, clr))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *angParams;

	if ((err=pCtx->LocalToPhysAddr(params[3], &angParams)) != SP_ERROR_NONE)

	{

		pCtx->ThrowNativeErrorEx(err, NULL);

		return 0;

	}

	QAngle ang(

		sp_ctof(angParams[0]),

		sp_ctof(angParams[1]),

		sp_ctof(angParams[2]));

	if (!msg->SetQAngle(strField, ang))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *vecParams;

	if ((err=pCtx->LocalToPhysAddr(params[3], &vecParams)) != SP_ERROR_NONE)

	{

		pCtx->ThrowNativeErrorEx(err, NULL);

		return 0;

	}

	Vector vec(

		sp_ctof(vecParams[0]),

		sp_ctof(vecParams[1]),

		sp_ctof(vecParams[2]));

	if (!msg->SetVector(strField, vec))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	cell_t *vecParams;

	if ((err=pCtx->LocalToPhysAddr(params[3], &vecParams)) != SP_ERROR_NONE)

	{

		pCtx->ThrowNativeErrorEx(err, NULL);

		return 0;

	}

	Vector2D vec(

		sp_ctof(vecParams[0]),

		sp_ctof(vecParams[1]));

	if (!msg->SetVector2D(strField, vec))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	if (!msg->AddInt32OrUnsigned(strField, params[3]))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	if (!msg->AddFloatOrDouble(strField, sp_ctof(params[3])))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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

{

	GET_MSG_FROM_HANDLE_OR_ERR();

	GET_FIELD_NAME_OR_ERR();

	bool value = (params[3] == 0 ? false : true);

	if (!msg->AddBool(strField, value))

	{

		return pCtx->ThrowNativeError("Invalid field \"%s\" for message \"%s\"", strField, msg->GetProtobufMessage()->GetTypeName().c_str());

	}

	return 1;

}

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