#include "md5\md5.h"
#include "aes\Rijndael.h"

#include "Base64CPP/Base64.h"
#include "jsoncpp\include\json\json.h"

//utf8 ת Unicode
std::wstring Utf82Unicode(const std::string& utf8string)
{
	int widesize = ::MultiByteToWideChar(CP_UTF8, 0, utf8string.c_str(), -1, NULL, 0);
	if (widesize == ERROR_NO_UNICODE_TRANSLATION)
	{
		throw std::exception("Invalid UTF-8 sequence.");
	}
	if (widesize == 0)
	{
		throw std::exception("Error in conversion.");
	}

	std::vector<wchar_t> resultstring(widesize);

	int convresult = ::MultiByteToWideChar(CP_UTF8, 0, utf8string.c_str(), -1, &resultstring[0], widesize);

	if (convresult != widesize)
	{
		throw std::exception("La falla!");
	}

	return std::wstring(&resultstring[0]);
}


//unicode תΪ ascii
std::string WideByte2Acsi(std::wstring& wstrcode)
{
	int asciisize = ::WideCharToMultiByte(CP_OEMCP, 0, wstrcode.c_str(), -1, NULL, 0, NULL, NULL);
	if (asciisize == ERROR_NO_UNICODE_TRANSLATION)
	{
		throw std::exception("Invalid UTF-8 sequence.");
	}
	if (asciisize == 0)
	{
		throw std::exception("Error in conversion.");
	}
	std::vector<char> resultstring(asciisize);
	int convresult = ::WideCharToMultiByte(CP_OEMCP, 0, wstrcode.c_str(), -1, &resultstring[0], asciisize, NULL, NULL);

	if (convresult != asciisize)
	{
		throw std::exception("La falla!");
	}

	return std::string(&resultstring[0]);
}


//utf-8 ת ascii
std::string UTF_82ASCII(std::string& strUtf8Code)
{
	std::string strRet("");
	//Ȱ utf8 תΪ unicode
	std::wstring wstr = Utf82Unicode(strUtf8Code);
	// unicode תΪ ascii
	strRet = WideByte2Acsi(wstr);
	return strRet;
}


///////////////////////////////////////////////////////////////////////


//ascii ת Unicode
std::wstring Acsi2WideByte(std::string& strascii)
{
	int widesize = MultiByteToWideChar(CP_ACP, 0, (char*)strascii.c_str(), -1, NULL, 0);
	if (widesize == ERROR_NO_UNICODE_TRANSLATION)
	{
		throw std::exception("Invalid UTF-8 sequence.");
	}
	if (widesize == 0)
	{
		throw std::exception("Error in conversion.");
	}
	std::vector<wchar_t> resultstring(widesize);
	int convresult = MultiByteToWideChar(CP_ACP, 0, (char*)strascii.c_str(), -1, &resultstring[0], widesize);


	if (convresult != widesize)
	{
		throw std::exception("La falla!");
	}

	return std::wstring(&resultstring[0]);
}


//Unicode ת Utf8
std::string Unicode2Utf8(const std::wstring& widestring)
{
	int utf8size = ::WideCharToMultiByte(CP_UTF8, 0, widestring.c_str(), -1, NULL, 0, NULL, NULL);
	if (utf8size == 0)
	{
		throw std::exception("Error in conversion.");
	}

	std::vector<char> resultstring(utf8size);

	int convresult = ::WideCharToMultiByte(CP_UTF8, 0, widestring.c_str(), -1, &resultstring[0], utf8size, NULL, NULL);

	if (convresult != utf8size)
	{
		throw std::exception("La falla!");
	}

	return std::string(&resultstring[0]);
}

//ascii ת Utf8
std::string ASCII2UTF_8(std::string& strAsciiCode)
{
	std::string strRet("");
	//Ȱ ascii תΪ unicode
	std::wstring wstr = Acsi2WideByte(strAsciiCode);
	// unicode תΪ utf8
	strRet = Unicode2Utf8(wstr);
	return strRet;
}

static const char* my_iv = "1234567812345678";
static const char* my_key = "1234567812345678";

static string encrypt_cbc(std::string in)
{
	// Encryptor 
	CRijndael rin;
	unsigned char * pOutPut = new unsigned char[in.length() + 16];
	rin.Init(CRijndael::CBC, CRijndael::EncryptDir, (const UINT8*)my_key,
		CRijndael::Key16Bytes, (const UINT8*)my_iv);
	int nSize = rin.PadEncrypt((const UINT8*)in.c_str(), in.length(), pOutPut);

	string encString = Base64::base64Encode(pOutPut, nSize);
	
	for (uint32 i = 0; i < encString.length(); i++){
		if (encString[i] == '+'){
			encString[i] = '-';
		}
	}
	return encString;
}
