Textractor_test/GUI/extenwindow.cpp

171 lines
5.4 KiB
C++
Raw Normal View History

#include "extenwindow.h"
#include "ui_extenwindow.h"
2019-02-16 13:25:19 +08:00
#include <concrt.h>
#include <QMenu>
#include <QFileDialog>
#include <QDragEnterEvent>
#include <QDropEvent>
2018-11-02 06:47:19 +08:00
#include <QMimeData>
#include <QUrl>
2019-02-28 00:33:17 +08:00
extern const char* EXTENSIONS;
extern const char* ADD_EXTENSION;
2019-06-05 11:14:46 +08:00
extern const char* INVALID_EXTENSION;
extern const char* CONFIRM_EXTENSION_OVERWRITE;
extern const char* EXTENSION_WRITE_ERROR;
2019-02-28 00:33:17 +08:00
extern const char* EXTEN_WINDOW_INSTRUCTIONS;
namespace
{
constexpr auto EXTEN_SAVE_FILE = u8"SavedExtensions.txt";
constexpr auto DEFAULT_EXTENSIONS = u8"Remove Repeated Characters>Remove Repeated Phrases>Regex Filter>Copy to Clipboard>Bing Translate>Extra Window>Extra Newlines";
2019-02-09 13:30:38 +08:00
struct Extension
{
2019-02-09 13:30:38 +08:00
std::wstring name;
wchar_t* (*callback)(wchar_t*, const InfoForExtension*);
};
Ui::ExtenWindow ui;
2019-02-16 13:25:19 +08:00
concurrency::reader_writer_lock extenMutex;
2019-02-09 13:30:38 +08:00
std::vector<Extension> extensions;
ExtenWindow* This = nullptr;
2019-06-05 11:14:46 +08:00
bool Load(QString extenName)
{
// Extension is dll and exports "OnNewSentence"
2019-06-05 11:14:46 +08:00
if (QTextFile(extenName + ".dll", QIODevice::ReadOnly).readAll().contains("OnNewSentence"))
{
if (HMODULE module = LoadLibraryW(S(extenName + ".dll").c_str()))
2019-06-05 11:14:46 +08:00
{
2019-09-17 05:14:59 +08:00
if (auto callback = (decltype(Extension::callback))GetProcAddress(module, "OnNewSentence"))
{
std::scoped_lock writeLock(extenMutex);
extensions.push_back({ S(extenName), callback });
return true;
}
FreeLibrary(module);
2019-06-05 11:14:46 +08:00
}
}
2019-06-05 11:14:46 +08:00
return false;
}
2019-02-09 13:30:38 +08:00
void Unload(int index)
{
2019-01-10 11:35:01 +08:00
std::scoped_lock writeLock(extenMutex);
FreeLibrary(GetModuleHandleW((extensions.at(index).name + L".dll").c_str()));
2019-02-09 13:30:38 +08:00
extensions.erase(extensions.begin() + index);
}
2018-11-02 02:46:37 +08:00
void Reorder(QStringList extenNames)
{
2019-01-10 11:35:01 +08:00
std::scoped_lock writeLock(extenMutex);
2019-02-09 13:30:38 +08:00
std::vector<Extension> extensions;
for (auto extenName : extenNames)
extensions.push_back(*std::find_if(::extensions.begin(), ::extensions.end(), [&](Extension extension) { return extension.name == S(extenName); }));
2019-02-09 13:30:38 +08:00
::extensions = extensions;
2018-11-02 02:46:37 +08:00
}
void Sync()
{
ui.extenList->clear();
QTextFile extenSaveFile(EXTEN_SAVE_FILE, QIODevice::WriteOnly | QIODevice::Truncate);
concurrency::reader_writer_lock::scoped_lock_read readLock(extenMutex);
for (auto extension : extensions)
{
ui.extenList->addItem(S(extension.name));
extenSaveFile.write((S(extension.name) + ">").toUtf8());
}
}
void Add(QFileInfo extenFile)
{
if (extenFile.suffix() == "dll")
{
if (extenFile.absolutePath() != QDir::currentPath())
{
if (QFile::exists(extenFile.fileName()) && QMessageBox::question(This, EXTENSIONS, CONFIRM_EXTENSION_OVERWRITE) == QMessageBox::Yes) QFile::remove(extenFile.fileName());
if (!QFile::copy(extenFile.absoluteFilePath(), extenFile.fileName())) QMessageBox::warning(This, EXTENSIONS, EXTENSION_WRITE_ERROR);
}
if (Load(extenFile.completeBaseName())) return Sync();
}
QMessageBox::information(This, EXTENSIONS, QString(INVALID_EXTENSION).arg(extenFile.fileName()));
}
void OpenMenu(QPoint point)
{
QAction addExtension(ADD_EXTENSION);
if (QMenu::exec({ &addExtension }, ui.extenList->mapToGlobal(point), nullptr, This))
if (QString extenFile = QFileDialog::getOpenFileName(This, ADD_EXTENSION, ".", EXTENSIONS + QString(" (*.dll)")); !extenFile.isEmpty()) Add(extenFile);
}
}
2019-09-05 00:29:48 +08:00
bool DispatchSentenceToExtensions(std::wstring& sentence, const InfoForExtension* sentenceInfo)
{
2019-02-19 09:57:10 +08:00
wchar_t* sentenceBuffer = (wchar_t*)HeapAlloc(GetProcessHeap(), HEAP_GENERATE_EXCEPTIONS, (sentence.size() + 1) * sizeof(wchar_t));
wcscpy_s(sentenceBuffer, sentence.size() + 1, sentence.c_str());
2019-02-16 13:25:19 +08:00
concurrency::reader_writer_lock::scoped_lock_read readLock(extenMutex);
2019-02-09 13:30:38 +08:00
for (const auto& extension : extensions)
2019-09-05 00:29:48 +08:00
if (*(sentenceBuffer = extension.callback(sentenceBuffer, sentenceInfo)) == L'\0') break;
sentence = sentenceBuffer;
HeapFree(GetProcessHeap(), 0, sentenceBuffer);
2019-02-19 09:57:10 +08:00
return !sentence.empty();
}
2019-06-21 13:29:48 +08:00
void CleanupExtensions()
{
std::scoped_lock writeLock(extenMutex);
for (auto extension : extensions) FreeLibrary(GetModuleHandleW((extension.name + L".dll").c_str()));
2019-06-21 13:29:48 +08:00
extensions.clear();
}
ExtenWindow::ExtenWindow(QWidget* parent) :
QMainWindow(parent, Qt::WindowCloseButtonHint)
{
This = this;
ui.setupUi(this);
ui.vboxLayout->addWidget(new QLabel(EXTEN_WINDOW_INSTRUCTIONS, this));
setWindowTitle(EXTENSIONS);
connect(ui.extenList, &QListWidget::customContextMenuRequested, OpenMenu);
ui.extenList->installEventFilter(this);
if (!QFile::exists(EXTEN_SAVE_FILE)) QTextFile(EXTEN_SAVE_FILE, QIODevice::WriteOnly).write(DEFAULT_EXTENSIONS);
for (auto extenName : QString(QTextFile(EXTEN_SAVE_FILE, QIODevice::ReadOnly).readAll()).split(">")) Load(extenName);
Sync();
}
ExtenWindow::~ExtenWindow() = default;
2018-12-29 01:14:56 +08:00
bool ExtenWindow::eventFilter(QObject* target, QEvent* event)
{
// See https://stackoverflow.com/questions/1224432/how-do-i-respond-to-an-internal-drag-and-drop-operation-using-a-qlistwidget/1528215
if (event->type() == QEvent::ChildRemoved)
{
2018-11-02 02:46:37 +08:00
QStringList extenNames;
for (int i = 0; i < ui.extenList->count(); ++i) extenNames.push_back(ui.extenList->item(i)->text());
2018-11-02 02:46:37 +08:00
Reorder(extenNames);
Sync();
}
2018-12-29 01:14:56 +08:00
return false;
}
void ExtenWindow::keyPressEvent(QKeyEvent* event)
{
if (event->key() == Qt::Key_Delete && ui.extenList->currentItem())
{
Unload(ui.extenList->currentIndex().row());
Sync();
}
}
2018-11-02 06:47:19 +08:00
void ExtenWindow::dragEnterEvent(QDragEnterEvent* event)
{
2018-11-02 06:47:19 +08:00
event->acceptProposedAction();
}
2018-12-29 01:14:56 +08:00
void ExtenWindow::dropEvent(QDropEvent* event)
{
for (auto file : event->mimeData()->urls()) Add(file.toLocalFile());
2018-11-02 06:47:19 +08:00
}