179 lines
5.2 KiB
C++
179 lines
5.2 KiB
C++
#include "qtcommon.h"
|
|
#include "translatewrapper.h"
|
|
#include "devtools.h"
|
|
|
|
extern const wchar_t* ERROR_START_CHROME;
|
|
extern const wchar_t* TRANSLATION_ERROR;
|
|
|
|
const char* TRANSLATION_PROVIDER = "DevTools DeepL Translate";
|
|
const char* GET_API_KEY_FROM = nullptr;
|
|
std::wstring currTranslateTo;
|
|
|
|
extern const QStringList languagesTo
|
|
{
|
|
"Bulgarian",
|
|
"Chinese (Simplified)",
|
|
"Czech",
|
|
"Danish",
|
|
"Dutch",
|
|
"English (American)",
|
|
"English (British)",
|
|
"Estonian",
|
|
"Finnish",
|
|
"French",
|
|
"German",
|
|
"Greek",
|
|
"Hungarian",
|
|
"Indonesian",
|
|
"Italian",
|
|
"Japanese",
|
|
"Korean",
|
|
"Latvian",
|
|
"Lithuanian",
|
|
"Norwegian",
|
|
"Polish",
|
|
"Portuguese",
|
|
"Portuguese (Brazilian)",
|
|
"Romanian",
|
|
"Russian",
|
|
"Slovak",
|
|
"Slovenian",
|
|
"Spanish",
|
|
"Swedish",
|
|
"Turkish",
|
|
"Ukrainian"
|
|
},
|
|
languagesFrom =
|
|
{
|
|
"Bulgarian",
|
|
"Chinese",
|
|
"Czech",
|
|
"Danish",
|
|
"Dutch",
|
|
"English",
|
|
"Estonian",
|
|
"Finnish",
|
|
"French",
|
|
"German",
|
|
"Greek",
|
|
"Hungarian",
|
|
"Indonesian",
|
|
"Italian",
|
|
"Japanese",
|
|
"Korean",
|
|
"Latvian",
|
|
"Lithuanian",
|
|
"Norwegian",
|
|
"Polish",
|
|
"Portuguese",
|
|
"Romanian",
|
|
"Russian",
|
|
"Slovak",
|
|
"Slovenian",
|
|
"Spanish",
|
|
"Swedish",
|
|
"Turkish",
|
|
"Ukrainian"
|
|
};
|
|
extern const std::unordered_map<std::wstring, std::wstring> codes
|
|
{
|
|
{ { L"Bulgarian" }, { L"bg-BG" } },
|
|
{ { L"Chinese" }, { L"zh" } },
|
|
{ { L"Chinese (Simplified)" }, { L"zh-ZH" } },
|
|
{ { L"Czech" }, { L"cs-CS" } },
|
|
{ { L"Danish" }, { L"da-DA" } },
|
|
{ { L"Dutch" }, { L"nl-NL" } },
|
|
{ { L"English" }, { L"en" } },
|
|
{ { L"English (American)" }, { L"en-US" } },
|
|
{ { L"English (British)" }, { L"en-GB" } },
|
|
{ { L"Estonian" }, { L"et-ET" } },
|
|
{ { L"Finnish" }, { L"fi-FI" } },
|
|
{ { L"French" }, { L"fr-FR" } },
|
|
{ { L"German" }, { L"de-DE" } },
|
|
{ { L"Greek" }, { L"el-EL" } },
|
|
{ { L"Hungarian" }, { L"hu-HU" } },
|
|
{ { L"Indonesian" }, { L"id-ID" } },
|
|
{ { L"Italian" }, { L"it-IT" } },
|
|
{ { L"Japanese" }, { L"ja-JA" } },
|
|
{ { L"Korean" }, { L"ko-KO" } },
|
|
{ { L"Latvian" }, { L"lv-LV" } },
|
|
{ { L"Lithuanian" }, { L"lt-LT" } },
|
|
{ { L"Norwegian" }, { L"nb-NB" } },
|
|
{ { L"Polish" }, { L"pl-PL" } },
|
|
{ { L"Portuguese" }, { L"pt-PT" } },
|
|
{ { L"Portuguese (Brazilian)" }, { L"pt-BR" } },
|
|
{ { L"Romanian" }, { L"ro-RO" } },
|
|
{ { L"Russian" }, { L"ru-RU" } },
|
|
{ { L"Slovak" }, { L"sk-SK" } },
|
|
{ { L"Slovenian" }, { L"sl-SL" } },
|
|
{ { L"Spanish" }, { L"es-ES" } },
|
|
{ { L"Swedish" }, { L"sv-SV" } },
|
|
{ { L"Turkish" }, { L"tr-TR" } },
|
|
{ { L"Ukrainian" }, { L"uk-UK" } },
|
|
{ { L"?" }, { L"auto" } }
|
|
};
|
|
|
|
bool translateSelectedOnly = true, useRateLimiter = true, rateLimitSelected = false, useCache = true, useFilter = true;
|
|
int tokenCount = 30, rateLimitTimespan = 60000, maxSentenceSize = 2500;
|
|
|
|
BOOL WINAPI DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
|
|
{
|
|
switch (ul_reason_for_call)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
{
|
|
DevTools::Initialize();
|
|
}
|
|
break;
|
|
case DLL_PROCESS_DETACH:
|
|
{
|
|
DevTools::Close();
|
|
}
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
std::wstring htmlDecode (std::wstring text) {
|
|
const std::wstring enc[] = { L"&", L"<", L">" };
|
|
const std::wstring dec[] = { L"&", L"<", L">" };
|
|
|
|
size_t pos;
|
|
for(int j = 0; j < 3; j++) {
|
|
do {
|
|
pos = text.find(enc[j]);
|
|
if (pos != std::wstring::npos)
|
|
text.replace (pos,enc[j].length(),dec[j]);
|
|
} while (pos != std::wstring::npos);
|
|
}
|
|
return text;
|
|
}
|
|
|
|
std::pair<bool, std::wstring> Translate(const std::wstring& text, TranslationParam tlp)
|
|
{
|
|
if (!DevTools::Connected()) return { false, FormatString(L"%s: %s", TRANSLATION_ERROR, ERROR_START_CHROME) };
|
|
// DevTools can't handle concurrent translations yet
|
|
static std::mutex translationMutex;
|
|
std::scoped_lock lock(translationMutex);
|
|
std::wstring escaped; // DeepL breaks with slash in input
|
|
for (auto ch : text) ch == '/' ? escaped += L"\\/" : escaped += ch;
|
|
if (currTranslateTo == tlp.translateTo)
|
|
DevTools::SendRequest("Page.navigate", FormatString(LR"({"url":"https://www.deepl.com/en/translator#%s/%s/%s"})", (tlp.translateFrom == L"?") ? codes.at(tlp.translateFrom) : codes.at(tlp.translateFrom).substr(0, 2), codes.at(tlp.translateTo).substr(0, 2), Escape(escaped)));
|
|
else
|
|
{
|
|
currTranslateTo = tlp.translateTo;
|
|
for (int retry = 0; ++retry < 20; Sleep(100))
|
|
if (Copy(DevTools::SendRequest("Runtime.evaluate", LR"({"expression":"document.readyState"})")[L"result"][L"value"].String()) == L"complete") break;
|
|
DevTools::SendRequest("Page.navigate", FormatString(LR"({"url":"https://www.deepl.com/en/translator#%s/%s/%s"})", (tlp.translateFrom == L"?") ? codes.at(tlp.translateFrom) : codes.at(tlp.translateFrom).substr(0, 2), codes.at(tlp.translateTo), Escape(escaped)));
|
|
}
|
|
|
|
for (int retry = 0; ++retry < 100; Sleep(100))
|
|
if (auto translation = Copy(DevTools::SendRequest("Runtime.evaluate",
|
|
LR"({"expression":"document.querySelector('#target-dummydiv').innerHTML.trim() ","returnByValue":true})"
|
|
)[L"result"][L"value"].String())) if (!translation->empty()) return { true, htmlDecode(translation.value()) };
|
|
if (auto errorMessage = Copy(DevTools::SendRequest("Runtime.evaluate",
|
|
LR"({"expression":"document.querySelector('div.lmt__system_notification').innerHTML","returnByValue":true})"
|
|
)[L"result"][L"value"].String())) return { false, FormatString(L"%s: %s", TRANSLATION_ERROR, errorMessage.value()) };
|
|
return { false, TRANSLATION_ERROR };
|
|
}
|