/**

  * vim: set ts=4 :

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

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

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

  *

  * This file is not open source and may not be copied without explicit

  * written permission of AlliedModders LLC.  This file may not be redistributed 

  * in whole or significant part.

  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php

  *

  * Version: $Id$

  */

 #include <stdlib.h>

 #include <ctype.h>

 #include <stdarg.h>

 #include "sm_stringutil.h"

 #include "Logger.h"

 #include "PluginSys.h"

 #include "Translator.h"

 #include "PlayerManager.h"

 #define LADJUST			0x00000004		/* left adjustment */

 #define ZEROPAD			0x00000080		/* zero (as opposed to blank) pad */

 #define UPPERDIGITS		0x00000200		/* make alpha digits uppercase */

 #define to_digit(c)		((c) - '0')

 #define is_digit(c)		((unsigned)to_digit(c) <= 9)

 #define CHECK_ARGS(x) \

 	if ((arg+x) > args) { \

 		pCtx->ThrowNativeErrorEx(SP_ERROR_PARAM, "String formatted incorrectly - parameter %d (total %d)", arg, args); \

 		return 0; \

 	}

 inline bool TryServerLanguage(const char *serverlang, unsigned int *langid)

 {

 	if (!g_Translator.GetLanguageByCode(serverlang, langid))

 	{

 		if (!g_Translator.GetLanguageByCode("en", langid))

 		{

 			return false;

 		}

 	}

 	return true;

 }

 inline bool TryTranslation(CPlugin *pl, const char *key, unsigned int langid, unsigned int langcount, Translation *pTrans)

 {

 	TransError err = Trans_BadLanguage;

 	CPhraseFile *phrfl;

 	for (size_t i=0; i<langcount && err!=Trans_Okay; i++)

 	{

 		phrfl = g_Translator.GetFileByIndex(pl->GetLangFileByIndex(i));

 		err = phrfl->GetTranslation(key, langid, pTrans);

 	}

 	return (err == Trans_Okay) ? true : false;

 }

 size_t Translate(char *buffer, size_t maxlen, IPluginContext *pCtx, const char *key, cell_t target, const cell_t *params, int *arg, bool *error)

 {

 	unsigned int langid;

 	const char *langname = NULL;

 	*error = false;

 	Translation pTrans;

 	CPlugin *pl = (CPlugin *)g_PluginSys.FindPluginByContext(pCtx->GetContext());

 	size_t langcount = pl->GetLangFileCount();

 	void *new_params[MAX_TRANSLATE_PARAMS];

 	unsigned int max_params = 0;

 try_serverlang:

 	if (target == LANG_SERVER)

 	{

 		langname = g_Translator.GetServerLanguageCode();

 		if (!TryServerLanguage(langname ? langname : "en", &langid))

 		{

 			pCtx->ThrowNativeError("Translation failure: English language not found");

 			goto error_out;

 		}

 	} else if ((target >= 1) && (target <= g_Players.GetMaxClients())) {

 		langname = g_Translator.GetServerLanguageCode(); /* :TODO: read player's lang */

 		if (!langname || !g_Translator.GetLanguageByCode(langname, &langid))

 		{

 			if (langname && !strcmp(langname, "en"))

 			{

 				pCtx->ThrowNativeError("Translation failure: English language not found");

 				goto error_out;

 			}

 			target = LANG_SERVER;

 			goto try_serverlang;

 		}

 	} else {

 		pCtx->ThrowNativeErrorEx(SP_ERROR_PARAM, "Translation failed: invalid client index %d", target);

 		goto error_out;

 	}

 	if (!TryTranslation(pl, key, langid, langcount, &pTrans))

 	{

 		if (target != LANG_SERVER)

 		{

 			target = LANG_SERVER;

 			goto try_serverlang;

 		} else {

 			if (!g_Translator.GetLanguageByCode("en", &langid)

 				|| !TryTranslation(pl, key, langid, langcount, &pTrans))

 			{

 				pCtx->ThrowNativeErrorEx(SP_ERROR_PARAM, "Language phrase \"%s\" not found", key);

 				goto error_out;

 			}

 		}

 	}

 	max_params = pTrans.fmt_count;

 	for (size_t i=0; i<max_params; i++)

 	{

 		pCtx->LocalToPhysAddr(params[*arg], reinterpret_cast<cell_t **>(&new_params[i]));

 		(*arg)++;

 		if ((*arg) + i > (size_t)params[0])

 		{

 			pCtx->ThrowNativeErrorEx(SP_ERROR_PARAMS_MAX, "Translation string formatted incorrectly - parameter %d (total %d)", (*arg) + i, params[0]);

 			goto error_out;

 		}

 	}

 	return g_Translator.Translate(buffer, maxlen, new_params, &pTrans);

 error_out:

 	*error = true;

 	return 0;

 }

 //:TODO: review this code before we choose a license

 void AddString(char **buf_p, size_t &maxlen, const char *string, int width, int prec)

 {

 	int size = 0;

 	char *buf;

 	static char nlstr[] = {'(','n','u','l','l',')','\0'};

 	buf = *buf_p;

 	if (string == NULL)

 	{

 		string = nlstr;

 		prec = -1;

 	}

 	if (prec >= 0)

 	{

 		for (size = 0; size < prec; size++) 

 		{

 			if (string[size] == '\0')

 			{

 				break;

 			}

 		}

 	} else {

 		while (string[size++]);

 		size--;

 	}

 	if (size > (int)maxlen)

 	{

 		size = maxlen;

 	}

 	maxlen -= size;

 	width -= size;

 	while (size--)

 	{

 		*buf++ = *string++;

 	}

 	while ((width-- > 0) && maxlen)

 	{

 		*buf++ = ' ';

 		maxlen--;

 	}

 	*buf_p = buf;

 }

 void AddFloat(char **buf_p, size_t &maxlen, double fval, int width, int prec)

 {

 	char text[32];

 	int digits;

 	double signedVal;

 	char *buf;

 	int val;

 	// get the sign

 	signedVal = fval;

 	if (fval < 0)

 	{

 		fval = -fval;

 	}

 	// write the float number

 	digits = 0;

 	val = (int)fval;

 	do

 	{

 		text[digits++] = '0' + val % 10;

 		val /= 10;

 	} while (val);

 	if (signedVal < 0)

 	{

 		text[digits++] = '-';

 	}

 	buf = *buf_p;

 	while ((digits < width) && maxlen)

 	{

 		*buf++ = ' ';

 		width--;

 		maxlen--;

 	}

 	while ((digits--) && maxlen)

 	{

 		*buf++ = text[digits];

 		maxlen--;

 	}

 	*buf_p = buf;

 	if (prec < 0)

 	{

 		prec = 6;

 	}

 	// write the fraction

 	digits = 0;

 	while (digits < prec)

 	{

 		fval -= (int)fval;

 		fval *= 10.0;

 		val = (int)fval;

 		text[digits++] = '0' + val % 10;

 	}

 	if ((digits > 0) && maxlen)

 	{

 		buf = *buf_p;

 		*buf++ = '.';

 		maxlen--;

 		for (prec = 0; maxlen && (prec < digits); prec++)

 		{

 			*buf++ = text[prec];

 			maxlen--;

 		}

 		*buf_p = buf;

 	}

 }

 void AddUInt(char **buf_p, size_t &maxlen, unsigned int val, int width, int flags)

 {

 	char text[32];

 	int digits;

 	char *buf;

 	digits = 0;

 	do

 	{

 		text[digits++] = '0' + val % 10;

 		val /= 10;

 	} while (val);

 	buf = *buf_p;

 	if (!(flags & LADJUST))

 	{

 		while (digits < width && maxlen)

 		{

 			*buf++ = (flags & ZEROPAD) ? '0' : ' ';

 			width--;

 			maxlen--;

 		}

 	}

 	while (digits-- && maxlen)

 	{

 		*buf++ = text[digits];

 		width--;

 		maxlen--;

 	}

 	if (flags & LADJUST)

 	{

 		while (width-- && maxlen)

 		{

 			*buf++ = (flags & ZEROPAD) ? '0' : ' ';

 			maxlen--;

 		}

 	}

 	*buf_p = buf;

 }

 void AddInt(char **buf_p, size_t &maxlen, int val, int width, int flags)

 {

 	char text[32];

 	int digits;

 	int signedVal;

 	char *buf;

 	unsigned int unsignedVal;

 	digits = 0;

 	signedVal = val;

 	if (val < 0)

 	{

 		/* we want the unsigned version */

 		unsignedVal = abs(val);

 	} else {

 		unsignedVal = val;

 	}

 	do

 	{

 		text[digits++] = '0' + unsignedVal % 10;

 		unsignedVal /= 10;

 	} while (unsignedVal);

 	if (signedVal < 0)

 	{

 		text[digits++] = '-';

 	}

 	buf = *buf_p;

 	if (!(flags & LADJUST))

 	{

 		while ((digits < width) && maxlen)

 		{

 			*buf++ = (flags & ZEROPAD) ? '0' : ' ';

 			width--;

 			maxlen--;

 		}

 	}

 	while (digits-- && maxlen)

 	{

 		*buf++ = text[digits];

 		width--;

 		maxlen--;

 	}

 	if (flags & LADJUST)

 	{

 		while (width-- && maxlen)

 		{

 			*buf++ = (flags & ZEROPAD) ? '0' : ' ';

 			maxlen--;

 		}

 	}

 	*buf_p = buf;

 }

 void AddHex(char **buf_p, size_t &maxlen, unsigned int val, int width, int flags)

 {

 	char text[32];

 	int digits;

 	char *buf;

 	char digit;

 	int hexadjust;

 	if (flags & UPPERDIGITS)

 	{

 		hexadjust = 'A' - '9' - 1;

 	} else {

 		hexadjust = 'a' - '9' - 1;

 	}

 	digits = 0;

 	do 

 	{

 		digit = ('0' + val % 16);

 		if (digit > '9')

 		{

 			digit += hexadjust;

 		}

 		text[digits++] = digit;

 		val /= 16;

 	} while(val);

 	buf = *buf_p;

 	if (!(flags & LADJUST))

 	{

 		while (digits < width && maxlen)

 		{

 			*buf++ = (flags & ZEROPAD) ? '0' : ' ';

 			width--;

 			maxlen--;

 		}

 	}

 	while (digits-- && maxlen)

 	{

 		*buf++ = text[digits];

 		width--;

 		maxlen--;

 	}

 	if (flags & LADJUST)

 	{

 		while (width-- && maxlen)

 		{

 			*buf++ = (flags & ZEROPAD) ? '0' : ' ';

 			maxlen--;

 		}

 	}

 	*buf_p = buf;

 }

 size_t gnprintf(char *buffer, size_t maxlen, const char *format, void **args)

 {

 	if (!buffer || !maxlen)

 	{

 		return 0;

 	}

 	int arg = 0;

 	char *buf_p;

 	char ch;

 	int flags;

 	int width;

 	int prec;

 	int n;

 	char sign;

 	const char *fmt;

 	size_t llen = maxlen - 1;

 	buf_p = buffer;

 	fmt = format;

 	while (true)

 	{

 		// run through the format string until we hit a '%' or '\0'

 		for (ch = *fmt; llen && ((ch = *fmt) != '\0') && (ch != '%'); fmt++)

 		{

 			*buf_p++ = ch;

 			llen--;

 		}

 		if ((ch == '\0') || (llen <= 0))

 		{

 			goto done;

 		}

 		// skip over the '%'

 		fmt++;

 		// reset formatting state

 		flags = 0;

 		width = 0;

 		prec = -1;

 		sign = '\0';

 rflag:

 		ch = *fmt++;

 reswitch:

 		switch(ch)

 		{

 		case '-':

 			{

 				flags |= LADJUST;

 				goto rflag;

 			}

 		case '.':

 			{

 				n = 0;

 				while(is_digit((ch = *fmt++)))

 				{

 					n = 10 * n + (ch - '0');

 				}

 				prec = (n < 0) ? -1 : n;

 				goto reswitch;

 			}

 		case '0':

 			{

 				flags |= ZEROPAD;

 				goto rflag;

 			}

 		case '1':

 		case '2':

 		case '3':

 		case '4':

 		case '5':

 		case '6':

 		case '7':

 		case '8':

 		case '9':

 			{

 				n = 0;

 				do

 				{

 					n = 10 * n + (ch - '0');

 					ch = *fmt++;

 				} while(is_digit(ch));

 				width = n;

 				goto reswitch;

 			}

 		case 'c':

 			{

 				if (!llen)

 				{

 					goto done;

 				}

 				char *c = (char *)args[arg];

 				*buf_p++ = *c;

 				llen--;

 				arg++;

 				break;

 			}

 		case 'd':

 		case 'i':

 			{

 				int *value = (int *)args[arg];

 				AddInt(&buf_p, llen, *value, width, flags);

 				arg++;

 				break;

 			}

 		case 'u':

 			{

 				unsigned int *value = (unsigned int *)args[arg];

 				AddUInt(&buf_p, llen, *value, width, flags);

 				arg++;

 				break;

 			}

 		case 'f':

 			{

 				float *value = (float *)args[arg];

 				AddFloat(&buf_p, llen, *value, width, prec);

 				arg++;

 				break;

 			}

 		case 's':

 			{

 				const char *str = (const char *)args[arg];

 				AddString(&buf_p, llen, str, width, prec);

 				arg++;

 				break;

 			}

 		case 'X':

 			{

 				unsigned int *value = (unsigned int *)args[arg];

 				flags |= UPPERDIGITS;

 				AddHex(&buf_p, llen, *value, width, flags);

 				arg++;

 				break;

 			}

 		case 'x':

 			{

 				unsigned int *value = (unsigned int *)args[arg];

 				AddHex(&buf_p, llen, *value, width, flags);

 				arg++;

 				break;

 			}

 		case '%':

 			{

 				if (!llen)

 				{

 					goto done;

 				}

 				*buf_p++ = ch;

 				llen--;

 				break;

 			}

 		case '\0':

 			{

 				if (!llen)

 				{

 					goto done;

 				}

 				*buf_p++ = '%';

 				llen--;

 				goto done;

 			}

 		default:

 			{

 				if (!llen)

 				{

 					goto done;

 				}

 				*buf_p++ = ch;

 				llen--;

 				break;

 			}

 		}

 	}

 done:

 	*buf_p = '\0';

 	return (maxlen - llen - 1);

 }

 size_t atcprintf(char *buffer, size_t maxlen, const char *format, IPluginContext *pCtx, const cell_t *params, int *param)

 {

 	if (!buffer || !maxlen)

 	{

 		return 0;

 	}

 	int arg;

 	int args = params[0];

 	char *buf_p;

 	char ch;

 	int flags;

 	int width;

 	int prec;

 	int n;

 	char sign;

 	const char *fmt;

 	size_t llen = maxlen - 1;

 	buf_p = buffer;

 	arg = *param;

 	fmt = format;

 	while (true)

 	{

 		// run through the format string until we hit a '%' or '\0'

 		for (ch = *fmt; llen && ((ch = *fmt) != '\0') && (ch != '%'); fmt++)

 		{

 			*buf_p++ = ch;

 			llen--;

 		}

 		if ((ch == '\0') || (llen <= 0))

 		{

 			goto done;

 		}

 		// skip over the '%'

 		fmt++;

 		// reset formatting state

 		flags = 0;

 		width = 0;

 		prec = -1;

 		sign = '\0';

 rflag:

 		ch = *fmt++;

 reswitch:

 		switch(ch)

 		{

 		case '-':

 			{

 				flags |= LADJUST;

 				goto rflag;

 			}

 		case '.':

 			{

 				n = 0;

 				while(is_digit((ch = *fmt++)))

 				{

 					n = 10 * n + (ch - '0');

 				}

 				prec = (n < 0) ? -1 : n;

 				goto reswitch;

 			}

 		case '0':

 			{

 				flags |= ZEROPAD;

 				goto rflag;

 			}

 		case '1':

 		case '2':

 		case '3':

 		case '4':

 		case '5':

 		case '6':

 		case '7':

 		case '8':

 		case '9':

 			{

 				n = 0;

 				do

 				{

 					n = 10 * n + (ch - '0');

 					ch = *fmt++;

 				} while(is_digit(ch));

 				width = n;

 				goto reswitch;

 			}

 		case 'c':

 			{

 				CHECK_ARGS(0);

 				if (!llen)

 				{

 					goto done;

 				}

 				char *c;

 				pCtx->LocalToString(params[arg], &c);

 				*buf_p++ = *c;

 				llen--;

 				arg++;

 				break;

 			}

 		case 'd':

 		case 'i':

 			{

 				CHECK_ARGS(0);

 				cell_t *value;

 				pCtx->LocalToPhysAddr(params[arg], &value);

 				AddInt(&buf_p, llen, static_cast<int>(*value), width, flags);

 				arg++;

 				break;

 			}

 		case 'u':

 			{

 				CHECK_ARGS(0);

 				cell_t *value;

 				pCtx->LocalToPhysAddr(params[arg], &value);

 				AddUInt(&buf_p, llen, static_cast<unsigned int>(*value), width, flags);

 				arg++;

 				break;

 			}

 		case 'f':

 			{

 				CHECK_ARGS(0);

 				cell_t *value;

 				pCtx->LocalToPhysAddr(params[arg], &value);

 				AddFloat(&buf_p, llen, sp_ctof(*value), width, prec);

 				arg++;

 				break;

 			}

 		case 's':

 			{

 				CHECK_ARGS(0);

 				char *str;

 				int err;

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

 				{

 					pCtx->ThrowNativeErrorEx(err, "Could not deference string");

 					return 0;

 				}

 				AddString(&buf_p, llen, str, width, prec);

 				arg++;

 				break;

 			}

 		case 'T':

 			{

 				CHECK_ARGS(0);

 				char *key;

 				bool error;

 				size_t res;

 				cell_t target = params[arg++];

 				pCtx->LocalToString(params[arg++], &key);

 				res = Translate(buf_p, llen, pCtx, key, target, params, &arg, &error);

 				if (error)

 				{

 					return 0;

 				}

 				buf_p += res;

 				llen -= res;

 				break;

 			}

 		case 't':

 			{

 				CHECK_ARGS(0);

 				char *key;

 				bool error;

 				size_t res;

 				cell_t target = g_SourceMod.GetGlobalTarget();

 				pCtx->LocalToString(params[arg++], &key);

 				res = Translate(buf_p, llen, pCtx, key, target, params, &arg, &error);

 				if (error)

 				{

 					return 0;

 				}

 				buf_p += res;

 				llen -= res;

 				break;

 			}

 		case 'X':

 			{

 				CHECK_ARGS(0);

 				cell_t *value;

 				pCtx->LocalToPhysAddr(params[arg], &value);

 				flags |= UPPERDIGITS;

 				AddHex(&buf_p, llen, static_cast<unsigned int>(*value), width, flags);

 				arg++;

 				break;

 			}

 		case 'x':

 			{

 				CHECK_ARGS(0);

 				cell_t *value;

 				pCtx->LocalToPhysAddr(params[arg], &value);

 				AddHex(&buf_p, llen, static_cast<unsigned int>(*value), width, flags);

 				arg++;

 				break;

 			}

 		case '%':

 			{

 				if (!llen)

 				{

 					goto done;

 				}

 				*buf_p++ = ch;

 				llen--;

 				break;

 			}

 		case '\0':

 			{

 				if (!llen)

 				{

 					goto done;

 				}

 				*buf_p++ = '%';

 				llen--;

 				goto done;

 			}

 		default:

 			{

 				if (!llen)

 				{

 					goto done;

 				}

 				*buf_p++ = ch;

 				llen--;

 				break;

 			}

 		}

 	}

 done:

 	*buf_p = '\0';

 	*param = arg;

 	return (maxlen - llen - 1);

 }

 const char *stristr(const char *str, const char *substr)

 {

 	if (!*substr)

 	{

 		return ((char *)str);

 	}

 	char *needle = (char *)substr;

 	char *prevloc = (char *)str;

 	char *haystack = (char *)str;

 	while (*haystack)

 	{

 		if (tolower(*haystack) == tolower(*needle))

 		{

 			haystack++;

 			if (!*++needle)

 			{

 				return prevloc;

 			}

 		} else {

 			haystack = ++prevloc;

 			needle = (char *)substr;

 		}

 	}

 	return NULL;

 }

 unsigned int strncopy(char *dest, const char *src, size_t count)

 {

 	if (!count)

 	{

 		return 0;

 	}

 	char *start = dest;

 	while ((*src) && (--count))

 	{