#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 = exe_name;
            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 = exe_name;
            CloseHandle(h);
            return true;
        }
    }

    CloseHandle(h);
    
    return false;
}


#endif