Textractor_test/extensions/bingtranslate.cpp

137 lines
3.4 KiB
C++
Raw Normal View History

2019-02-11 10:46:39 +08:00
#include "extension.h"
2018-12-19 05:32:28 +08:00
#include "text.h"
2019-02-11 10:46:39 +08:00
#include "network.h"
#include <QInputDialog>
#include <QTimer>
QStringList languages
{
"English: en",
"Arabic: ar",
"Bosnian: bs-Latn",
"Bulgarian: bg",
"Catalan: ca",
"Chinese(Simplified): zh-CHS",
"Chinese(Traditional): zh-CHT",
"Croatian: hr",
"Czech: cs",
"Danish: da",
"Dutch: nl",
"Estonian: et",
"Finnish: fi",
"French: fr",
"German: de",
"Greek: el",
"Hebrew: he",
"Hindi: hi",
"Hungarian: hu",
"Indonesian: id",
"Italian: it",
"Japanese: ja",
"Klingon: tlh",
"Korean: ko",
"Latvian: lv",
"Lithuanian: lt",
"Malay: ms",
"Maltese: mt",
"Norwegian: no",
"Persian: fa",
"Polish: pl",
"Portuguese: pt",
"Romanian: ro",
"Russian: ru",
"Serbian: sr-Cyrl",
"Slovak: sk",
"Slovenian: sl",
"Spanish: es",
"Swedish: sv",
"Thai: th",
"Turkish: tr",
"Ukranian: uk",
"Urdu: ur",
"Vietnamese: vi",
"Welsh: cy"
};
2019-02-11 10:46:39 +08:00
std::wstring translateTo = L"en";
BOOL WINAPI DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
{
switch (ul_reason_for_call)
{
case DLL_PROCESS_ATTACH:
{
QTimer::singleShot(0, []
{
2018-11-04 17:31:49 +08:00
translateTo = QInputDialog::getItem(
nullptr,
2018-12-19 05:32:28 +08:00
SELECT_LANGUAGE,
BING_PROMPT,
2018-11-04 17:31:49 +08:00
languages,
0, false, nullptr,
Qt::WindowCloseButtonHint
).split(" ")[1].toStdWString();
});
}
break;
case DLL_PROCESS_DETACH:
{
}
break;
}
return TRUE;
}
2019-02-11 10:46:39 +08:00
// This function detects language and returns it if translateFrom is empty
std::wstring Translate(const std::wstring& text, std::wstring translateFrom, std::wstring translateTo)
{
2019-01-24 02:41:50 +08:00
static std::atomic<HINTERNET> internet = NULL;
2018-10-18 22:13:33 +08:00
if (!internet) internet = WinHttpOpen(L"Mozilla/5.0 Textractor", WINHTTP_ACCESS_TYPE_DEFAULT_PROXY, NULL, NULL, 0);
2019-02-11 10:46:39 +08:00
std::wstring escapedText;
for (unsigned char ch : WideStringToString(text))
{
2019-02-11 10:46:39 +08:00
wchar_t escapedChar[4] = {};
swprintf_s<4>(escapedChar, L"%%%02X", (int)ch);
2019-02-17 11:43:31 +08:00
escapedText += escapedChar;
}
2019-02-11 10:46:39 +08:00
std::wstring location = translateFrom.empty()
2019-02-17 11:43:31 +08:00
? L"/tdetect?text=" + escapedText
: L"/ttranslate?from=" + translateFrom + L"&to=" + translateTo + L"&text=" + escapedText;
std::wstring translation;
if (internet)
2019-02-11 10:46:39 +08:00
if (InternetHandle connection = WinHttpConnect(internet, L"www.bing.com", INTERNET_DEFAULT_HTTPS_PORT, 0))
if (InternetHandle request = WinHttpOpenRequest(connection, L"POST", location.c_str(), NULL, NULL, NULL, WINHTTP_FLAG_ESCAPE_DISABLE | WINHTTP_FLAG_SECURE))
if (WinHttpSendRequest(request, NULL, 0, NULL, 0, 0, NULL))
2019-02-11 10:46:39 +08:00
if (auto response = ReceiveHttpRequest(request))
if (translateFrom.empty()) translation = response.value();
2019-02-17 11:43:31 +08:00
// Response formatted as JSON: translation starts with :" and ends with "}
2019-02-11 10:46:39 +08:00
else if (std::wsmatch results; std::regex_search(response.value(), results, std::wregex(L":\"(.+)\"\\}"))) translation = results[1];
Escape(translation);
return translation;
}
bool ProcessSentence(std::wstring& sentence, SentenceInfo sentenceInfo)
{
2019-02-18 08:14:49 +08:00
if (sentenceInfo["text number"] == 0) return false;
2019-02-11 10:46:39 +08:00
static RateLimiter rateLimiter(30, 60 * 1000);
2019-01-22 03:22:47 +08:00
2019-02-11 10:46:39 +08:00
std::wstring translation;
if (!(rateLimiter.Request() || sentenceInfo["current select"])) translation = TOO_MANY_TRANS_REQUESTS;
else translation = Translate(sentence, Translate(sentence, L"", translateTo), translateTo);
2018-12-19 05:32:28 +08:00
if (translation.empty()) translation = TRANSLATION_ERROR;
2019-01-11 10:51:20 +08:00
sentence += L"\n" + translation;
return true;
2018-12-19 05:32:28 +08:00
}
2019-02-11 10:46:39 +08:00
TEST(
{
std::wstring test = L"こんにちは";
2019-02-13 12:54:33 +08:00
ProcessSentence(test, { SentenceInfo::DUMMY });
2019-02-11 10:46:39 +08:00
assert(test.find(L"Hello") != std::wstring::npos);
}
);