From ed7f8ec0ec150d61dd5a3980839fa74297b2771b Mon Sep 17 00:00:00 2001 From: Blu3train Date: Mon, 10 Oct 2022 00:12:04 +0200 Subject: [PATCH 1/2] Regex Replacer Translated Text Extension --- deploy.ps1 | 1 + extensions/CMakeLists.txt | 2 + extensions/regexreplacertranslatedtext.cpp | 79 ++++++++++++++++++++++ 3 files changed, 82 insertions(+) create mode 100644 extensions/regexreplacertranslatedtext.cpp diff --git a/deploy.ps1 b/deploy.ps1 index ac13101..138fc74 100644 --- a/deploy.ps1 +++ b/deploy.ps1 @@ -53,6 +53,7 @@ foreach ($language in @{ "Lua", "Regex Filter", "Regex Replacer", + "Regex Replacer Translated Text", "Remove Repeated Characters", "Remove Repeated Phrases", "Remove Repeated Phrases 2", diff --git a/extensions/CMakeLists.txt b/extensions/CMakeLists.txt index 3e3d0a6..171690b 100644 --- a/extensions/CMakeLists.txt +++ b/extensions/CMakeLists.txt @@ -15,6 +15,7 @@ add_library(Google\ Translate MODULE googletranslate.cpp translatewrapper.cpp ne add_library(Lua MODULE lua.cpp extensionimpl.cpp) add_library(Regex\ Filter MODULE regexfilter.cpp extensionimpl.cpp) add_library(Regex\ Replacer MODULE regexreplacer.cpp extensionimpl.cpp) +add_library(Regex\ Replacer\ Translated\ Text MODULE regexreplacertranslatedtext.cpp extensionimpl.cpp) add_library(Remove\ Repeated\ Characters MODULE removerepeatchar.cpp extensionimpl.cpp) add_library(Remove\ Repeated\ Phrases MODULE removerepeatphrase.cpp extensionimpl.cpp) add_library(Remove\ Repeated\ Phrases\ 2 MODULE removerepeatphrase2.cpp extensionimpl.cpp) @@ -35,6 +36,7 @@ target_precompile_headers(Google\ Translate REUSE_FROM pch) target_precompile_headers(Lua REUSE_FROM pch) target_precompile_headers(Regex\ Filter REUSE_FROM pch) target_precompile_headers(Regex\ Replacer REUSE_FROM pch) +target_precompile_headers(Regex\ Replacer\ Translated\ Text REUSE_FROM pch) target_precompile_headers(Remove\ Repeated\ Characters REUSE_FROM pch) target_precompile_headers(Remove\ Repeated\ Phrases REUSE_FROM pch) target_precompile_headers(Remove\ Repeated\ Phrases\ 2 REUSE_FROM pch) diff --git a/extensions/regexreplacertranslatedtext.cpp b/extensions/regexreplacertranslatedtext.cpp new file mode 100644 index 0000000..b54f49e --- /dev/null +++ b/extensions/regexreplacertranslatedtext.cpp @@ -0,0 +1,79 @@ +#include "extension.h" +#include "module.h" +#include "blockmarkup.h" +#include + +extern const wchar_t* REGEX_REPLACER_INSTRUCTIONS; + +const char* REPLACE_SAVE_FILE = "SavedRegexReplacementsTranslatedText.txt"; + +std::atomic replaceFileLastWrite = {}; +concurrency::reader_writer_lock m; +std::vector> replacements; + +void UpdateReplacements() +{ + try + { + if (replaceFileLastWrite.exchange(std::filesystem::last_write_time(REPLACE_SAVE_FILE)) == std::filesystem::last_write_time(REPLACE_SAVE_FILE)) return; + std::scoped_lock lock(m); + replacements.clear(); + std::ifstream stream(REPLACE_SAVE_FILE, std::ios::binary); + BlockMarkupIterator savedFilters(stream, Array{ L"|REGEX|", L"|BECOMES|", L"|MODIFIER|" }); + while (auto read = savedFilters.Next()) + { + const auto& [regex, replacement, modifier] = read.value(); + try + { + replacements.emplace_back( + std::wregex(regex, modifier.find(L'i') == std::string::npos ? std::regex::ECMAScript : std::regex::icase), + replacement, + modifier.find(L'g') == std::string::npos ? std::regex_constants::format_first_only : std::regex_constants::format_default + ); + } + catch (std::regex_error) {} + } + } + catch (std::filesystem::filesystem_error) { replaceFileLastWrite.store({}); } +} + +BOOL WINAPI DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved) +{ + switch (ul_reason_for_call) + { + case DLL_PROCESS_ATTACH: + { + UpdateReplacements(); + if (replacements.empty()) + { + auto file = std::ofstream(REPLACE_SAVE_FILE, std::ios::binary) << "\xff\xfe"; + for (auto ch : std::wstring_view(REGEX_REPLACER_INSTRUCTIONS)) + file << (ch == L'\n' ? std::string_view("\r\0\n", 4) : std::string_view((char*)&ch, 2)); + SpawnThread([] { _spawnlp(_P_DETACH, "notepad", "notepad", REPLACE_SAVE_FILE, NULL); }); // show file to user + } + } + break; + case DLL_PROCESS_DETACH: + { + } + break; + } + return TRUE; +} + +bool ProcessSentence(std::wstring& sentence, SentenceInfo sentenceInfo) +{ + if (!sentenceInfo["current select"] || sentenceInfo["text number"] == 0) return false; + size_t posTranslation = sentence.find(L"\x200b \n"); + if (posTranslation == std::wstring::npos) return false; + posTranslation +=3; + + UpdateReplacements(); + + std::wstring sTranslated = sentence.substr(posTranslation, std::wstring::npos); + sentence = sentence.substr(0, posTranslation); + + concurrency::reader_writer_lock::scoped_lock_read readLock(m); + for (const auto& [regex, replacement, flags] : replacements) sentence += std::regex_replace(sTranslated, regex, replacement, flags); + return true; +} From 9b628b4d8730cf74899df594504068662baa59a5 Mon Sep 17 00:00:00 2001 From: Blu3train Date: Mon, 10 Oct 2022 00:44:21 +0200 Subject: [PATCH 2/2] code fix --- extensions/regexreplacertranslatedtext.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/extensions/regexreplacertranslatedtext.cpp b/extensions/regexreplacertranslatedtext.cpp index b54f49e..149beed 100644 --- a/extensions/regexreplacertranslatedtext.cpp +++ b/extensions/regexreplacertranslatedtext.cpp @@ -2,6 +2,7 @@ #include "module.h" #include "blockmarkup.h" #include +#include extern const wchar_t* REGEX_REPLACER_INSTRUCTIONS;