#include <QJsonDocument>
#include <QJsonObject>
#include <QFile>
#include <QDebug>
#include <QJsonArray>
#include <QJsonValue>
#include <QIntValidator>
#include <QFileInfo>
#include <QProcess>
#include <unistd.h>

#include "utils.h"
#include "inbound_settings.h"
#include "mainwindow.h"
#include "ui_inbound_settings.h"

inbound_settings_window::inbound_settings_window(MainWindow *parent) :
    QDialog(parent),
    ui(new Ui::inbound_settings_window)
{
    ui->setupUi(this);
    rootObj = loadRootObjFromConf();
    QJsonObject http = findValueFromJsonArray(rootObj.value("inbounds").toArray(), "tag", "http-in");
    QJsonObject socks = findValueFromJsonArray(rootObj.value("inbounds").toArray(), "tag", "socks-in");
    if(rootObj.value("v2suidEnabled").toBool()) {
        ui->rootCheckbox->setCheckState(Qt::Checked);
    }
    if(!http.isEmpty()) {
        ui->httpPortLE->setText(QString::number(http.value("port").toInt()));
        ui->httpCB->setCheckState(Qt::Checked);
    } else {
        ui->httpPortLE->setDisabled(true);
    }
    if(!socks.isEmpty()) {
        ui->socksPortLE->setText(QString::number(socks.value("port").toInt()));
        ui->socksCB->setCheckState(Qt::Checked);
    } else {
        ui->socksPortLE->setDisabled(true);
    }
    ui->httpPortLE->setValidator(new QIntValidator());
    ui->socksPortLE->setValidator(new QIntValidator());
    parentMW = parent;
}

inbound_settings_window::~inbound_settings_window()
{
    delete ui;
}

void inbound_settings_window::on_buttonBox_accepted()
{
    if(testCoreFiles()) {
        if(ui->httpPortLE->text().toInt() != ui->socksPortLE->text().toInt()) {
            QJsonArray inbounds;
            QJsonDocument modifiedDoc;
            inbounds = rootObj.value("inbounds").toArray();
            int socksId = getIndexInArrayByValue(inbounds, "tag", "socks-in");
            if(socksId != -1) {
                inbounds.removeAt(socksId);
            }
            int httpId = getIndexInArrayByValue(inbounds, "tag", "http-in");
            if(httpId != -1) {
                inbounds.removeAt(httpId);
            }
            rootObj.remove("inbounds");
            rootObj.remove("v2suidEnabled");
            if(ui->socksCB->isChecked()) {
                QJsonObject socks;
                QJsonObject settings;
                socks.insert("tag", "socks-in");
                socks.insert("port", ui->socksPortLE->text().toInt());
                socks.insert("listen", "127.0.0.1");
                socks.insert("protocol", "socks");
                settings.insert("auth", "noauth");
                settings.insert("udp", true);
                settings.insert("ip", "127.0.0.1");
                socks.insert("settings", QJsonValue(settings));
                inbounds.append(socks);
            }
            if(ui->httpCB->isChecked()) {
                QJsonObject http;
                QJsonObject settings;
                http.insert("tag", "http-in");
                http.insert("port", ui->httpPortLE->text().toInt());
                http.insert("listen", "127.0.0.1");
                http.insert("protocol", "http");
                settings.insert("auth", "noauth");
                settings.insert("udp", true);
                settings.insert("ip", "127.0.0.1");
                http.insert("settings", QJsonValue(settings));
                inbounds.append(http);
            }
            rootObj.insert("inbounds", QJsonValue(inbounds));
#ifndef _WIN32
            // Set UID and GID in *nix
            QFileInfo ray("v2ray");
            if(ui->rootCheckbox->isChecked() && ray.ownerId() != 0) {
                QProcess::execute("pkexec", QStringList() << "bash" << "-c" << "chown root:root " + QCoreApplication::applicationDirPath() + "/v2ray" + ";chmod +s " + QCoreApplication::applicationDirPath() + "/v2ray");
            } else if (!ui->rootCheckbox->isChecked() && ray.ownerId() == 0) {
                uid_t uid = getuid();
                gid_t gid = getgid();
                QProcess::execute("pkexec", QStringList() << "chown" << QString::number(uid) + ":" + QString::number(gid) << QCoreApplication::applicationDirPath() + "/v2ray");
            }
            ray.refresh();
            rootObj.insert("v2suidEnabled", ray.ownerId() == 0);
#else
            // No such uid gid thing on windows....
#endif
            modifiedDoc.setObject(rootObj);
            QByteArray byteArray = modifiedDoc.toJson(QJsonDocument::Indented);
            QFile confFile("conf/Hv2ray.config.json");
            if(!confFile.open(QIODevice::WriteOnly)) {
                qDebug() << "Cannot open Hv2ray.config.json for modifying";
            }
            confFile.write(byteArray);
            confFile.close();
            parentMW->on_restartButton_clicked();
        } else {
            alterMessage("Config error", "Port numbers cannot be the same!");
        }
    }
}


void inbound_settings_window::on_httpCB_stateChanged(int arg1)
{
    if(arg1 != Qt::Checked) {
        ui->httpPortLE->setDisabled(true);
    } else {
        ui->httpPortLE->setEnabled(true);
        ui->httpPortLE->setText("6666");
    }
}

void inbound_settings_window::on_socksCB_stateChanged(int arg1)
{
    if(arg1 != Qt::Checked) {
        ui->socksPortLE->setDisabled(true);
    } else {
        ui->socksPortLE->setEnabled(true);
        ui->socksPortLE->setText("1080");
    }
}