/**

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

 * 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 "CLogger.h"

#include "PluginSys.h"

#include "CTranslator.h"

#include "CPlayerManager.h"

#define LADJUST			0x00000004		/* left adjustment */

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

#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_PARAMS_MAX, "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->GetFileByIndex(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;

	char *langname = NULL;

	*error = false;

try_serverlang:

	if (target == LANG_SERVER)

	{

		langname = "en"; //:TODO: read serverlang

		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 = "en"; //: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;

	}

	Translation pTrans;

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

	size_t langcount = pl->GetLangFileCount();

	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;

			}

		}

	}

	void *new_params[MAX_TRANSLATE_PARAMS];

	unsigned int 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;

}

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 '%':

			{

				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;