summaryrefslogtreecommitdiff
path: root/plugins/CyrTranslit/src/TransliterationProtocol.cpp
blob: c4d6649f2cfa2a6e6f3eec6cf1cef80a935d2def (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/**
 * 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 "stdafx.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, sendMessage);
}

//------------------------------------------------------------------------------
void TransliterationProtocol::TranslateMessageUTF(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, 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, 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::sendMessage(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;
}

}