/*
    Helper functions for Miranda-IM (www.miranda-im.org)
    Copyright 2006 P. Boon

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    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, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
#include "commonheaders.h"
#include "gen_helpers.h"

int ProtoServiceExists(const char *szModule, const char *szService) {

	char str[MAXMODULELABELLENGTH];
	mir_snprintf(str, sizeof(str), "%s%s", szModule, szService);

	return ServiceExists(str);
}

char *Hlp_GetProtocolNameA(const char *proto) {

	char protoname[256];

	if ( (!ProtoServiceExists(proto, PS_GETNAME)) || (CallProtoService(proto, PS_GETNAME, (WPARAM)sizeof(protoname), (LPARAM)protoname)) ) {
		return NULL;
	}

	return _strdup(protoname);
}

TCHAR *Hlp_GetProtocolName(const char *proto) {

	char protoname[256];

	if ( (!ProtoServiceExists(proto, PS_GETNAME)) || (CallProtoService(proto, PS_GETNAME, (WPARAM)sizeof(protoname), (LPARAM)protoname)) ) {
		return NULL;
	}
#ifdef UNICODE
	return a2u(protoname);
#else
	return _strdup(protoname);
#endif
}

char *Hlp_GetDlgItemTextA(HWND hwndDlg, int nIDDlgItem) {

	int len = SendDlgItemMessageA(hwndDlg, nIDDlgItem, WM_GETTEXTLENGTH, 0, 0);
	if (len < 0)
		return NULL;

	char *res = ( char* )malloc((len+1));
	ZeroMemory(res, (len+1));
	GetDlgItemTextA(hwndDlg, nIDDlgItem, res, len+1);

	return res;
}

TCHAR *Hlp_GetDlgItemText(HWND hwndDlg, int nIDDlgItem) {

	int len = SendDlgItemMessage(hwndDlg, nIDDlgItem, WM_GETTEXTLENGTH, 0, 0);
	if (len < 0)
		return NULL;

	TCHAR *res = ( TCHAR* )malloc((len+1)*sizeof(TCHAR));
	ZeroMemory(res, (len+1)*sizeof(TCHAR));
	GetDlgItemText(hwndDlg, nIDDlgItem, res, len+1);

	return res;
}

char *Hlp_GetWindowTextA(HWND hwndDlg) {

	int len = SendMessageA(hwndDlg, WM_GETTEXTLENGTH, 0, 0);
	if (len < 0)
		return NULL;

	char *res = ( char* )malloc((len+1));
	ZeroMemory(res, (len+1));
	GetWindowTextA(hwndDlg, res, len+1);

	return res;
}

TCHAR *Hlp_GetWindowText(HWND hwndDlg) {

	int len = SendMessage(hwndDlg, WM_GETTEXTLENGTH, 0, 0);
	if (len < 0)
		return NULL;

	TCHAR *res = ( TCHAR* )malloc((len+1)*sizeof(TCHAR));
	ZeroMemory(res, (len+1)*sizeof(TCHAR));
	GetWindowText(hwndDlg, res, len+1);

	return res;
}

/**
 * Modified from Miranda CList, clistsettings.c
 **/

char* u2a( wchar_t* src )
{
	int cbLen;
	char* result;

	if (src == NULL) {
		return NULL;
	}
	cbLen = WideCharToMultiByte( CP_ACP, 0, src, -1, NULL, 0, NULL, NULL );
	result = ( char* )malloc( cbLen+1 );
	if ( result == NULL )
		return NULL;

	WideCharToMultiByte( CP_ACP, 0, src, -1, result, cbLen, NULL, NULL );
	result[ cbLen ] = 0;
	return result;
}

wchar_t* a2u( char* src )
{
	int cbLen;
	wchar_t* result;
	
	if (src == NULL) {
		return NULL;
	}
	cbLen = MultiByteToWideChar( CP_ACP, 0, src, -1, NULL, 0 );
	result = ( wchar_t* )malloc( sizeof( wchar_t )*(cbLen+1));
	if ( result == NULL )
		return NULL;

	MultiByteToWideChar( CP_ACP, 0, src, -1, result, cbLen );
	result[ cbLen ] = 0;
	return result;
}

// Logging
static int WriteToDebugLogA(const char *szMsg) {

	int res;

	res = 0;
	if (ServiceExists(MS_NETLIB_LOG)) {
		res = CallService(MS_NETLIB_LOG, (WPARAM)NULL, (LPARAM)szMsg);
	} else {
		OutputDebugStringA(szMsg);
		OutputDebugStringA("\r\n");
	}
	
	return res;
}

int AddDebugLogMessageA(const char* fmt, ...) {
	
	int res;
	char szText[MAX_DEBUG], szFinal[MAX_DEBUG];
	va_list va;
	
	va_start(va,fmt);
	_vsnprintf(szText, sizeof(szText), fmt, va);
	va_end(va);
#ifdef MODULENAME
	mir_snprintf(szFinal, sizeof(szFinal), "%s: %s", MODULENAME, szText);
#else
	strncpy(szFinal, szText, sizeof(szFinal));
#endif
	res = WriteToDebugLogA(szFinal);

	return res;
}

int AddDebugLogMessage(const TCHAR* fmt, ...) {

	int res;
	TCHAR tszText[MAX_DEBUG], tszFinal[MAX_DEBUG];
	char *szFinal;
	va_list va;

	va_start(va,fmt);
	_vsntprintf(tszText, sizeof(tszText), fmt, va);
	va_end(va);
#ifdef MODULENAME
	mir_sntprintf(tszFinal, SIZEOF(tszFinal), _T("%s: %s"), MODULENAME, tszText);
#else
	_tcsncpy(tszFinal, tszText, sizeof(tszFinal));
#endif

#ifdef UNICODE
	szFinal = u2a(tszFinal);
#else
	szFinal = _strdup(tszFinal);
#endif
	res = WriteToDebugLogA(szFinal);
	free(szFinal);

	return res;
}

int AddErrorLogMessageA(const char* fmt, ...) {

	int res;
	char szText[MAX_DEBUG], szFinal[MAX_DEBUG];
	va_list va;
	
	va_start(va,fmt);
	_vsnprintf(szText, sizeof(szText), fmt, va);
	va_end(va);
#ifdef MODULENAME
	mir_snprintf(szFinal, sizeof(szFinal), "%s: %s", MODULENAME, szText);
#else
	strncpy(szFinal, szText, sizeof(szFinal));
#endif
	res = WriteToDebugLogA(szFinal);
	MessageBoxA(NULL, szFinal, "Error", MB_OK|MB_ICONERROR);

	return res;
}

int AddErrorLogMessage(const TCHAR* fmt, ...) {
	
	int res;
	TCHAR tszText[MAX_DEBUG], tszFinal[MAX_DEBUG];
	char *szFinal;
	va_list va;

	va_start(va,fmt);
	_vsntprintf(tszText, sizeof(tszText), fmt, va);
	va_end(va);
#ifdef MODULENAME
	mir_sntprintf(tszFinal, SIZEOF(tszFinal), _T("%s: %s"), MODULENAME, tszText);
#else
	_tcsncpy(tszFinal, tszText, sizeof(tszFinal));
#endif

#ifdef UNICODE
	szFinal = u2a(tszFinal);
#else
	szFinal = _strdup(tszFinal);
#endif
	res = WriteToDebugLogA(szFinal);
	MessageBoxA(NULL, szFinal, "Error", MB_OK|MB_ICONERROR);
	free(szFinal);

	return res;
}


//
int ttoi(TCHAR *string)
{
	return ( string == NULL ) ? 0 : _ttoi( string );
}

TCHAR *itot(int num) {

	char tRes[32];

	// check this
	_itoa(num, tRes, 10);

#ifdef UNICODE
	return a2u(tRes);
#else
	return _strdup(tRes);
#endif
}

/////////////////////////////////////////////////////////////////////////////////////////
// Utf8Decode - converts UTF8-encoded string to the UCS2/MBCS format

void Utf8Decode( char* str, wchar_t** ucs2 )
{
	int len;
	wchar_t* tempBuf;

	if ( str == NULL )
		return;

	len = strlen( str );
	if ( len < 2 ) {
		if ( ucs2 != NULL ) {
			*ucs2 = ( wchar_t* )malloc(( len+1 )*sizeof( wchar_t ));
			MultiByteToWideChar( CP_ACP, 0, str, len, *ucs2, len );
			( *ucs2 )[ len ] = 0;
		}
		return;
	}

	tempBuf = ( wchar_t* )alloca(( len+1 )*sizeof( wchar_t ));
	{
		wchar_t* d = tempBuf;
		BYTE* s = ( BYTE* )str;

		while( *s )
		{
			if (( *s & 0x80 ) == 0 ) {
				*d++ = *s++;
				continue;
			}

			if (( s[0] & 0xE0 ) == 0xE0 && ( s[1] & 0xC0 ) == 0x80 && ( s[2] & 0xC0 ) == 0x80 ) {
				*d++ = (( WORD )( s[0] & 0x0F) << 12 ) + ( WORD )(( s[1] & 0x3F ) << 6 ) + ( WORD )( s[2] & 0x3F );
				s += 3;
				continue;
			}

			if (( s[0] & 0xE0 ) == 0xC0 && ( s[1] & 0xC0 ) == 0x80 ) {
				*d++ = ( WORD )(( s[0] & 0x1F ) << 6 ) + ( WORD )( s[1] & 0x3F );
				s += 2;
				continue;
			}

			*d++ = *s++;
		}

		*d = 0;
	}

	if ( ucs2 != NULL ) {
		int fullLen = ( len+1 )*sizeof( wchar_t );
		*ucs2 = ( wchar_t* )malloc( fullLen );
		memcpy( *ucs2, tempBuf, fullLen );
	}

   WideCharToMultiByte( CP_ACP, 0, tempBuf, -1, str, len, NULL, NULL );
}

/////////////////////////////////////////////////////////////////////////////////////////
// Utf8Encode - converts MBCS string to the UTF8-encoded format

char* Utf8Encode( const char* src )
{
	int len;
	char* result;
	wchar_t* tempBuf;

	if ( src == NULL )
		return NULL;

	len = strlen( src );
	result = ( char* )malloc( len*3 + 1 );
	if ( result == NULL )
		return NULL;

	tempBuf = ( wchar_t* )alloca(( len+1 )*sizeof( wchar_t ));
	MultiByteToWideChar( CP_ACP, 0, src, -1, tempBuf, len );
	tempBuf[ len ] = 0;
	{
		wchar_t* s = tempBuf;
		BYTE*		d = ( BYTE* )result;

		while( *s ) {
			int U = *s++;

			if ( U < 0x80 ) {
				*d++ = ( BYTE )U;
			}
			else if ( U < 0x800 ) {
				*d++ = 0xC0 + (( U >> 6 ) & 0x3F );
				*d++ = 0x80 + ( U & 0x003F );
			}
			else {
				*d++ = 0xE0 + ( U >> 12 );
				*d++ = 0x80 + (( U >> 6 ) & 0x3F );
				*d++ = 0x80 + ( U & 0x3F );
		}	}

		*d = 0;
	}

	return result;
}

/////////////////////////////////////////////////////////////////////////////////////////
// Utf8Encode - converts UCS2 string to the UTF8-encoded format

char* Utf8EncodeUcs2( const wchar_t* src )
{
	int len = wcslen( src );
	char* result = ( char* )malloc( len*3 + 1 );
	if ( result == NULL )
		return NULL;

	{	const wchar_t* s = src;
		BYTE*	d = ( BYTE* )result;

		while( *s ) {
			int U = *s++;

			if ( U < 0x80 ) {
				*d++ = ( BYTE )U;
			}
			else if ( U < 0x800 ) {
				*d++ = 0xC0 + (( U >> 6 ) & 0x3F );
				*d++ = 0x80 + ( U & 0x003F );
			}
			else {
				*d++ = 0xE0 + ( U >> 12 );
				*d++ = 0x80 + (( U >> 6 ) & 0x3F );
				*d++ = 0x80 + ( U & 0x3F );
		}	}

		*d = 0;
	}

	return result;
}

// Helper functions that need MODULENAME
#define SETTING_NOENCODINGCHECK		"NoEncodingCheck"

int Hlp_UnicodeCheck(char *szPluginName, BOOL bForce, const char *szModule) {

#ifndef _DEBUG
	char *ptr;
	char szVersionText[256];

	if (!CallService(MS_SYSTEM_GETVERSIONTEXT, (WPARAM)sizeof(szVersionText), (LPARAM)szVersionText)) {
		ptr = strstr(szVersionText, "Unicode");
#ifdef UNICODE
		if ( (ptr == NULL) && (!DBGetContactSettingByte(NULL, szModule, SETTING_NOENCODINGCHECK, 0)) ) {
			if (bForce) {
				MessageBoxA(NULL, "You are running the ANSI version Miranda. Please use the ANSI build of this plugin.", szPluginName, MB_OK);
				
				return -1;
			}
			else {
				MessageBoxA(NULL, "You are running the ANSI version Miranda. It's recommened to use the ANSI build of this plugin.", szPluginName, MB_OK);
				DBWriteContactSettingByte(NULL, szModule, SETTING_NOENCODINGCHECK, 1);

				return 0;
			}
		}
#else
		if ( (ptr != NULL) && (!DBGetContactSettingByte(NULL, szModule, SETTING_NOENCODINGCHECK, 0)) ) {
			if (bForce) {
				MessageBoxA(NULL, "You are running the UNICODE version Miranda. Please use the UNICODE build of this plugin.", szPluginName, MB_OK);
			
				return -1;
			}
			else {
				MessageBoxA(NULL, "You are running the UNICODE version Miranda. It's recommened to use the UNICODE build of this plugin.", szPluginName, MB_OK);
				DBWriteContactSettingByte(NULL, szModule, SETTING_NOENCODINGCHECK, 1);

				return 0;
			}
		}
#endif
	}
#endif

	return 0;
}