/**
 * CyrTranslit: the Cyrillic transliteration plug-in for Miranda IM.
 * Copyright 2005 Ivan Krechetov. 
 * 
 * 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 "plugin.h"
#include "MirandaContact.h"
#include "TransliterationProtocol.h"
#include "TransliterationMap.h"

namespace CyrTranslit
{

char *TransliterationProtocol::MODULE_NAME = "ProtoCyrTranslitByIKR";

//------------------------------------------------------------------------------

void TransliterationProtocol::initialize()
{
	PROTOCOLDESCRIPTOR pd = { sizeof(pd) };
	pd.szName = MODULE_NAME;
	pd.type = PROTOTYPE_TRANSLATION;
	CallService(MS_PROTO_REGISTERMODULE, 0, reinterpret_cast<LPARAM>(&pd));

	CreateProtoServiceFunction(MODULE_NAME, PSS_MESSAGE,    sendMessageA);
	CreateProtoServiceFunction(MODULE_NAME, PSS_MESSAGE"W", sendMessageW);
}

//------------------------------------------------------------------------------
void TransliterationProtocol::TranslateMessageUTF(WPARAM wParam, LPARAM lParam)
{
	CCSDATA *ccs = reinterpret_cast<CCSDATA*>(lParam);

	wchar_t* txtWdecoded = mir_utf8decodeW(reinterpret_cast<const char*>(ccs->lParam));
	std::wstring txtW = txtWdecoded;
	mir_free(txtWdecoded);

	txtW = TransliterationMap::getInstance().cyrillicToLatin(txtW);

	char* txtUTFencoded = mir_utf8encodeW(txtW.c_str());
	std::string txtUTF = txtUTFencoded;
	mir_free(txtUTFencoded);

	ccs->lParam = reinterpret_cast<LPARAM>(mir_alloc(txtUTF.length()));
	strcpy(reinterpret_cast<char*>(ccs->lParam), txtUTF.c_str());
}

void TransliterationProtocol::TranslateMessageW(WPARAM wParam, LPARAM lParam)
{
	CCSDATA *ccs = reinterpret_cast<CCSDATA*>(lParam);

	std::string txtA = reinterpret_cast<const char*>(ccs->lParam);
	int txtAlen = (int)(sizeof(txtA[0]) * (txtA.length() + 1));
	txtA = TransliterationMap::getInstance().cyrillicToLatin(txtA);

	std::wstring txtW = reinterpret_cast<const wchar_t*>(ccs->lParam + txtAlen);
	txtW = TransliterationMap::getInstance().cyrillicToLatin(txtW);

	txtAlen = (int)(sizeof(txtA[0]) * (txtA.length() + 1));
	const DWORD newSize = static_cast<DWORD>(txtAlen + (sizeof(txtW[0]) * (txtW.length() + 1)));

	ccs->lParam = reinterpret_cast<LPARAM>(mir_alloc(newSize));

	strcpy(reinterpret_cast<char*>(ccs->lParam), txtA.c_str());
	wcscpy(reinterpret_cast<wchar_t*>(ccs->lParam + txtAlen), txtW.c_str());
}

void TransliterationProtocol::TranslateMessageA(WPARAM wParam, LPARAM lParam)
{
	CCSDATA *ccs = reinterpret_cast<CCSDATA*>(lParam);

	std::string txt = reinterpret_cast<const char*>(ccs->lParam);
	txt = TransliterationMap::getInstance().cyrillicToLatin(txt);

	const DWORD newSize = static_cast<DWORD>(txt.length() + 1);

	ccs->lParam = reinterpret_cast<LPARAM>(mir_alloc(newSize));

	strcpy(reinterpret_cast<char*>(ccs->lParam), txt.c_str());
}

INT_PTR TransliterationProtocol::sendMessageW(WPARAM wParam, LPARAM lParam)
{
	CCSDATA *ccs = reinterpret_cast<CCSDATA*>(lParam);
	if ( !MirandaContact::bIsActive(ccs->hContact))
		return CallService(MS_PROTO_CHAINSEND, wParam, lParam);

	LPARAM oldlParam = ccs->lParam;

	TranslateMessageW(wParam, lParam);

	int ret = CallService(MS_PROTO_CHAINSEND /* "W" */, wParam, lParam);

	mir_free(reinterpret_cast<void*>(ccs->lParam));
	ccs->lParam = oldlParam;

	return ret;
}
//------------------------------------------------------------------------------

INT_PTR TransliterationProtocol::sendMessageA(WPARAM wParam, LPARAM lParam)
{
	CCSDATA *ccs = reinterpret_cast<CCSDATA*>(lParam);
	if ( !MirandaContact::bIsActive(ccs->hContact))
		return CallService(MS_PROTO_CHAINSEND, wParam, lParam);

	LPARAM oldlParam = ccs->lParam;
	bool msgProcessed = true;

	if (ccs->wParam & PREF_UTF)
		TranslateMessageUTF(wParam, lParam);
	else if (ccs->wParam & PREF_UNICODE)
		TranslateMessageW(wParam, lParam);
	else
		TranslateMessageA(wParam, lParam);

	int ret = CallService(MS_PROTO_CHAINSEND, wParam, lParam);

	if (msgProcessed) {
		mir_free(reinterpret_cast<void*>(ccs->lParam));
		ccs->lParam = oldlParam;
	}

	return ret;
}

}