#include "db/ProxyEntity.hpp"
#include "fmt/includes.h"

#include <QFile>
#include <QDir>
#include <QFileInfo>
#include <QUrl>

#define WriteTempFile(fn, data)                                   \
    QDir dir;                                                     \
    if (!dir.exists("temp")) dir.mkdir("temp");                   \
    QFile f(QString("temp/") + fn);                               \
    bool ok = f.open(QIODevice::WriteOnly | QIODevice::Truncate); \
    if (ok) {                                                     \
        f.write(data);                                            \
    } else {                                                      \
        result.error = f.errorString();                           \
    }                                                             \
    f.close();                                                    \
    auto TempFile = QFileInfo(f).absoluteFilePath();

namespace NekoRay::fmt {
    // -1: Cannot use this config
    // 0: Internal
    // 1: Mapping External
    // 2: Direct External

    int NaiveBean::NeedExternal(bool isFirstProfile) {
        if (isFirstProfile) {
            if (dataStore->spmode_vpn) {
                return 1;
            }
            return 2;
        }
        return 1;
    }

    int HysteriaBean::NeedExternal(bool isFirstProfile) {
        auto hysteriaCore = [=] {
            if (isFirstProfile) {
                if (dataStore->spmode_vpn && protocol != hysteria_protocol_facktcp && hopPort.trimmed().isEmpty()) {
                    return 1;
                }
                return 2;
            } else {
                if (protocol == hysteria_protocol_facktcp || !hopPort.trimmed().isEmpty()) {
                    return -1;
                }
            }
            return 1;
        };

        if (IS_NEKO_BOX) {
            if (protocol == hysteria_protocol_udp && hopPort.trimmed().isEmpty()) {
                // sing-box support
                return 0;
            } else {
                // hysteria core support
                return hysteriaCore();
            }
        } else {
            return hysteriaCore();
        }
    }

    int CustomBean::NeedExternal(bool isFirstProfile) {
        if (core == "internal" || core == "internal-full") return 0;
        return 1;
    }

    ExternalBuildResult NaiveBean::BuildExternal(int mapping_port, int socks_port, int external_stat) {
        ExternalBuildResult result{dataStore->extraCore->Get("naive")};

        auto is_direct = external_stat == 2;
        auto domain_address = sni.isEmpty() ? serverAddress : sni;
        auto connect_address = is_direct ? serverAddress : "127.0.0.1";
        auto connect_port = is_direct ? serverPort : mapping_port;
        domain_address = WrapIPV6Host(domain_address);
        connect_address = WrapIPV6Host(connect_address);

        auto proxy_url = QUrl();
        proxy_url.setScheme(protocol);
        proxy_url.setUserName(username);
        proxy_url.setPassword(password);
        proxy_url.setPort(connect_port);
        proxy_url.setHost(domain_address);

        result.arguments += "--log";
        result.arguments += "--listen=socks://127.0.0.1:" + Int2String(socks_port);
        result.arguments += "--proxy=" + proxy_url.toString(QUrl::FullyEncoded);
        if (domain_address != connect_address)
            result.arguments += "--host-resolver-rules=MAP " + domain_address + " " + connect_address;
        if (insecure_concurrency > 0) result.arguments += "--insecure-concurrency=" + Int2String(insecure_concurrency);
        if (!extra_headers.trimmed().isEmpty()) result.arguments += "--extra-headers=" + extra_headers;
        if (!certificate.trimmed().isEmpty()) {
            WriteTempFile("naive_" + GetRandomString(10) + ".crt", certificate.toUtf8());
            result.env += "SSL_CERT_FILE=" + TempFile;
        }

        auto config_export = QStringList{result.program};
        config_export += result.arguments;
        result.config_export = QStringList2Command(config_export);

        return result;
    }

    ExternalBuildResult HysteriaBean::BuildExternal(int mapping_port, int socks_port, int external_stat) {
        ExternalBuildResult result{dataStore->extraCore->Get("hysteria")};

        QJsonObject config;

        // determine server format
        auto is_direct = external_stat == 2;
        auto sniGen = sni;
        if (sni.isEmpty() && !IsIpAddress(serverAddress)) sniGen = serverAddress;

        auto server = serverAddress;
        if (!hopPort.trimmed().isEmpty()) {
            server = WrapIPV6Host(server) + ":" + hopPort;
        } else {
            server = WrapIPV6Host(server) + ":" + Int2String(serverPort);
        }
        config["server"] = is_direct ? server : "127.0.0.1:" + Int2String(mapping_port);

        // listen
        config["socks5"] = QJsonObject{
            {"listen", "127.0.0.1:" + Int2String(socks_port)},
        };

        // misc

        config["retry"] = 5;
        config["fast_open"] = true;
        config["lazy_start"] = true;
        config["obfs"] = obfsPassword;
        config["up_mbps"] = uploadMbps;
        config["down_mbps"] = downloadMbps;

        if (authPayloadType == hysteria_auth_base64) config["auth"] = authPayload;
        if (authPayloadType == hysteria_auth_string) config["auth_str"] = authPayload;

        if (protocol == hysteria_protocol_facktcp) config["protocol"] = "faketcp";
        if (protocol == hysteria_protocol_wechat_video) config["protocol"] = "wechat-video";

        if (!sniGen.isEmpty()) config["server_name"] = sniGen;
        if (!alpn.isEmpty()) config["alpn"] = alpn;

        if (!caText.trimmed().isEmpty()) {
            WriteTempFile("hysteria_" + GetRandomString(10) + ".crt", caText.toUtf8());
            config["ca"] = TempFile;
        }

        if (allowInsecure) config["insecure"] = true;
        if (streamReceiveWindow > 0) config["recv_window_conn"] = streamReceiveWindow;
        if (connectionReceiveWindow > 0) config["recv_window"] = connectionReceiveWindow;
        if (disableMtuDiscovery) config["disable_mtu_discovery"] = true;
        config["hop_interval"] = hopInterval;

        //

        result.config_export = QJsonObject2QString(config, false);
        WriteTempFile("hysteria_" + GetRandomString(10) + ".json", result.config_export.toUtf8());
        result.arguments = QStringList{"--no-check", "-c", TempFile};

        return result;
    }

    ExternalBuildResult CustomBean::BuildExternal(int mapping_port, int socks_port, int external_stat) {
        ExternalBuildResult result{dataStore->extraCore->Get(core)};

        result.arguments = command; // TODO split?

        for (int i = 0; i < result.arguments.length(); i++) {
            auto arg = result.arguments[i];
            arg = arg.replace("%mapping_port%", Int2String(mapping_port));
            arg = arg.replace("%socks_port%", Int2String(socks_port));
            arg = arg.replace("%server_addr%", serverAddress);
            arg = arg.replace("%server_port%", Int2String(serverPort));
            result.arguments[i] = arg;
        }

        if (!config_simple.trimmed().isEmpty()) {
            auto config = config_simple;
            config = config.replace("%mapping_port%", Int2String(mapping_port));
            config = config.replace("%socks_port%", Int2String(socks_port));
            config = config.replace("%server_addr%", serverAddress);
            config = config.replace("%server_port%", Int2String(serverPort));

            // suffix
            QString suffix;
            if (!config_suffix.isEmpty()) {
                suffix = "." + config_suffix;
            } else if (!QString2QJsonObject(config).isEmpty()) {
                // trojan-go: unsupported config format: xxx.tmp. use .yaml or .json instead.
                suffix = ".json";
            }

            // write config
            WriteTempFile("custom_" + GetRandomString(10) + suffix, config.toUtf8());
            for (int i = 0; i < result.arguments.count(); i++) {
                result.arguments[i] = result.arguments[i].replace("%config%", TempFile);
            }

            result.config_export = config;
        }

        return result;
    }
} // namespace NekoRay::fmt