main: allow automatically run profiles when executables run

Issue: #160

1 files changed, 243 insertions, 0 deletions

diff --git a/facetracknoir/process_detector.cpp b/facetracknoir/process_detector.cpp
new file mode 100644
index 00000000..3ac6329b
--- /dev/null
+++ b/facetracknoir/process_detector.cpp
@@ -0,0 +1,243 @@
+#include "process_detector.h"
+#include "facetracknoir/ui.h"
+#include <QList>
+#include <QFileDialog>
+#include <QComboBox>
+#include <QString>
+#include <QHash>
+#include <QPushButton>
+
+void settings::set_game_list(const QString &game_list)
+{
+    QSettings settings(group::org);
+    settings.setValue("executable-list", game_list);
+}
+
+QString settings::get_game_list()
+{
+    QSettings settings(group::org);
+    return settings.value("executable-list").toString();
+}
+
+bool settings::is_enabled()
+{
+    QSettings settings(group::org);
+    return settings.value("executable-detector-enabled", false).toBool();
+}
+
+void settings::set_is_enabled(bool enabled)
+{
+    QSettings settings(group::org);
+    settings.setValue("executable-detector-enabled", enabled);
+}
+
+QHash<QString, QString> settings::split_process_names()
+{
+    QHash<QString, QString> ret;
+    QString str = get_game_list();
+    QStringList pairs = str.split('|');
+    for (auto pair : pairs)
+    {
+        QList<QString> tmp = pair.split(':');
+        if (tmp.count() != 2)
+            continue;
+        ret[tmp[0]] = tmp[1];
+    }
+    return ret;
+}
+
+void BrowseButton::browse()
+{
+    QFileDialog dialog(this);
+    dialog.setFileMode(QFileDialog::ExistingFile);
+    QString dir_path = QFileInfo(group::ini_pathname()).absolutePath();
+    QString filename = dialog.getOpenFileName(
+                this,
+                tr("Set executable name"),
+                dir_path,
+                tr("Executable (*.exe);;All Files (*)"));
+    MainWindow::set_working_directory();
+    filename = QFileInfo(filename).fileName();
+    twi->setText(filename);
+}
+
+process_detector::add_row(QString exe_name, QString profile)
+{
+    int i = ui.tableWidget->rowCount();
+    ui.tableWidget->insertRow(i);
+    
+    QComboBox* cb = new QComboBox();
+    cb->setSizePolicy(QSizePolicy::Minimum, QSizePolicy::Maximum);
+    cb->addItems(group::ini_list());
+    ui.tableWidget->setCellWidget(i, 1, cb);
+    
+    QTableWidgetItem* twi = new QTableWidgetItem(exe_name);
+    ui.tableWidget->setItem(i, 0, twi);
+    
+    {
+        BrowseButton* b = new BrowseButton(twi);
+        b->setText("...");
+        QObject::connect(b, SIGNAL(pressed()), b, SLOT(browse()));
+        ui.tableWidget->setCellWidget(i, 2, b);
+    }
+    
+    cb->setCurrentText(profile);
+    
+    return i;
+}
+
+void process_detector::add_items()
+{
+    auto names = s.split_process_names();
+    ui.tableWidget->clearContents();
+    auto keys = names.keys();
+    qSort(keys);
+    for (auto n : keys)
+        add_row(n, names[n]);
+}
+
+process_detector::process_detector(QWidget* parent) : QWidget(parent)
+{
+    ui.setupUi(this);
+    connect(ui.add, SIGNAL(pressed()), this, SLOT(add()));
+    connect(ui.remove, SIGNAL(pressed()), this, SLOT(remove()));
+    
+    add_items();
+    
+    QResizeEvent e(ui.tableWidget->size(), ui.tableWidget->size());
+    ui.tableWidget->resizeEvent(&e);
+    
+    settings s;
+    
+    ui.enabled->setChecked(s.is_enabled());
+}
+
+void process_detector::save()
+{
+    QString str;
+    
+    for (int i = 0; i < ui.tableWidget->rowCount(); i++)
+    {
+        auto exe = ui.tableWidget->item(i, 0)->text();
+        auto profile = reinterpret_cast<QComboBox*>(ui.tableWidget->cellWidget(i, 1))->currentText();
+        str += "|" + exe + ":" + profile;
+    }
+    
+    s.set_game_list(str);
+    s.set_is_enabled(ui.enabled->isChecked());
+}
+
+void process_detector::revert()
+{
+}
+
+void process_detector::add()
+{
+    add_row();
+}
+
+void process_detector::remove()
+{
+    int r = ui.tableWidget->currentRow();
+    if (r != -1)
+        ui.tableWidget->removeRow(r);
+}
+
+#ifdef _WIN32
+
+#include <windows.h>
+#include <TlHelp32.h>
+
+bool process_detector_worker::should_stop()
+{   
+    if (last_exe_name == "")
+        return false;
+ 
+    settings s;
+    
+    if (!s.is_enabled())
+    {
+        last_exe_name = "";
+        return false;
+    }
+    
+    HANDLE h = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
+    if (h == INVALID_HANDLE_VALUE)
+        return false;
+    
+    PROCESSENTRY32 e;
+
+    if (Process32First(h, &e) == TRUE)
+    {
+        QString exe_name(e.szExeFile);
+        if (exe_name == last_exe_name)
+        {
+            CloseHandle(h);
+            return false;
+        }
+    }
+    
+    while (Process32Next(h, &e) == TRUE)
+    {
+        QString exe_name(e.szExeFile);
+        if (exe_name == last_exe_name)
+        {
+            CloseHandle(h);
+            return false;
+        }
+    }
+
+    last_exe_name = "";
+    CloseHandle(h);
+    
+    return true;
+}
+
+bool process_detector_worker::config_to_start(QString& str)
+{
+    settings s;
+    if (!s.is_enabled())
+    {
+        last_exe_name = "";
+        return false;
+    }
+    
+    auto filenames = s.split_process_names();
+    
+    HANDLE h = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
+    if (h == INVALID_HANDLE_VALUE)
+        return false;
+    
+    PROCESSENTRY32 e;
+
+    if (Process32First(h, &e) == TRUE)
+    {
+        QString exe_name(e.szExeFile);
+        if (filenames.contains(exe_name))
+        {
+            str = filenames[exe_name];
+            last_exe_name = str;
+            CloseHandle(h);
+            return true;
+        }
+    }
+    
+    while (Process32Next(h, &e) == TRUE)
+    {
+        QString exe_name(e.szExeFile);
+        if (filenames.contains(exe_name))
+        {
+            str = filenames[exe_name];
+            last_exe_name = str;
+            CloseHandle(h);
+            return true;
+        }
+    }
+
+    CloseHandle(h);
+    
+    return false;
+}
+
+
+#endif