mirror of
https://github.com/Artikash/Textractor.git
synced 2024-12-24 01:14:12 +08:00
33 lines
1.1 KiB
C++
33 lines
1.1 KiB
C++
|
#include "extensions.h"
|
||
|
|
#include <map>
|
||
|
|
|
||
|
|
std::map<int, ExtensionFunction> extensions;
|
||
|
|
|
||
|
|
std::map<int, std::wstring> LoadExtensions()
|
||
|
|
{
|
||
|
|
std::map<int, std::wstring> extensionNames;
|
||
|
|
wchar_t path[MAX_PATH];
|
||
|
|
wchar_t* end = path + GetModuleFileNameW(nullptr, path, MAX_PATH);
|
||
|
|
while (*(--end) != L'\\');
|
||
|
|
*(end + 1) = L'*';
|
||
|
|
*(end + 2) = L'\0';
|
||
|
|
WIN32_FIND_DATAW fileData;
|
||
|
|
HANDLE file = FindFirstFileW(path, &fileData);
|
||
|
|
do
|
||
|
|
if (!(fileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
||
|
|
if (wcsstr(fileData.cFileName, L"_nexthooker_extension.dll"))
|
||
|
|
if (GetProcAddress(LoadLibraryW(fileData.cFileName), "OnNewSentence"))
|
||
|
|
{
|
||
|
|
extensions[std::wcstol(fileData.cFileName, nullptr, 10)] = (ExtensionFunction)GetProcAddress(LoadLibraryW(fileData.cFileName), "OnNewSentence");
|
||
|
|
extensionNames[std::wcstol(fileData.cFileName, nullptr, 10)] = fileData.cFileName;
|
||
|
|
}
|
||
|
|
while (FindNextFileW(file, &fileData) != 0);
|
||
|
|
return extensionNames;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::wstring DispatchSentenceToExtensions(std::wstring sentence, std::unordered_map<std::string, int> miscInfo)
|
||
|
|
{
|
||
|
|
for (auto extension : extensions) sentence = extension.second(sentence, miscInfo);
|
||
|
|
return sentence;
|
||
|
|
}
|