#include "StartPageLogic.h"
#include "ViewConfigLogic.h"

#include "core/errorstrings.h"
#include "configurators/ssh_configurator.h"
#include "configurators/vpn_configurator.h"
#include "../uilogic.h"
#include "utils.h"

#include <QFileDialog>
#include <QStandardPaths>

#ifdef Q_OS_ANDROID
#include <QtAndroid>
#include "platforms/android/android_controller.h"
#endif

namespace {
QString checkConfigFormat(const QString &config)
{
    const QString openVpnConfigPatternCli = "client";
    const QString openVpnConfigPatternProto1 = "proto tcp";
    const QString openVpnConfigPatternProto2 = "proto udp";
    const QString openVpnConfigPatternDriver1 = "dev tun";
    const QString openVpnConfigPatternDriver2 = "dev tap";

    if (config.contains(openVpnConfigPatternCli) &&
            (config.contains(openVpnConfigPatternProto1) || config.contains(openVpnConfigPatternProto2)) &&
            (config.contains(openVpnConfigPatternDriver1) || config.contains(openVpnConfigPatternDriver2))) {
        return "OpenVpn";
    }
    return "Amnezia";
}

}

StartPageLogic::StartPageLogic(UiLogic *logic, QObject *parent):
    PageLogicBase(logic, parent),
    m_pushButtonConnectEnabled{true},
    m_pushButtonConnectText{tr("Connect")},
    m_pushButtonConnectKeyChecked{false},
    m_labelWaitInfoVisible{true},
    m_pushButtonBackFromStartVisible{true},
    m_ipAddressPortRegex{Utils::ipAddressPortRegExp()}
{
#ifdef Q_OS_ANDROID
    // Set security screen for Android app
    QtAndroid::runOnAndroidThread([]() {
        QAndroidJniObject window = QtAndroid::androidActivity().callObjectMethod("getWindow", "()Landroid/view/Window;");
        if (window.isValid()){
            const int FLAG_SECURE = 8192;
            window.callMethod<void>("addFlags", "(I)V", FLAG_SECURE);
        }
    });
#endif
}

void StartPageLogic::onUpdatePage()
{
    set_lineEditStartExistingCodeText("");
    set_textEditSshKeyText("");
    set_lineEditIpText("");
    set_lineEditPasswordText("");
    set_textEditSshKeyText("");
    set_lineEditLoginText("");

    set_labelWaitInfoVisible(false);
    set_labelWaitInfoText("");

    set_pushButtonConnectKeyChecked(false);

    set_pushButtonBackFromStartVisible(uiLogic()->pagesStackDepth() > 0);
}

void StartPageLogic::onPushButtonConnect()
{
    if (pushButtonConnectKeyChecked()){
        if (lineEditIpText().isEmpty() ||
                lineEditLoginText().isEmpty() ||
                textEditSshKeyText().isEmpty() ) {
            set_labelWaitInfoText(tr("Please fill in all fields"));
            return;
        }
    }
    else {
        if (lineEditIpText().isEmpty() ||
                lineEditLoginText().isEmpty() ||
                lineEditPasswordText().isEmpty() ) {
            set_labelWaitInfoText(tr("Please fill in all fields"));
            return;
        }
    }

    ServerCredentials serverCredentials;
    serverCredentials.hostName = lineEditIpText();
    if (serverCredentials.hostName.contains(":")) {
        serverCredentials.port = serverCredentials.hostName.split(":").at(1).toInt();
        serverCredentials.hostName = serverCredentials.hostName.split(":").at(0);
    }
    serverCredentials.userName = lineEditLoginText();
    if (pushButtonConnectKeyChecked()){
        QString key = textEditSshKeyText();
        if (key.startsWith("ssh-rsa")) {
            emit uiLogic()->showPublicKeyWarning();
            return;
        }

        if (key.contains("OPENSSH") && key.contains("BEGIN") && key.contains("PRIVATE KEY")) {
            key = m_configurator->sshConfigurator->convertOpenSShKey(key);
        }

        serverCredentials.password = key;
    }
    else {
        serverCredentials.password = lineEditPasswordText();
    }

    set_pushButtonConnectEnabled(false);
    set_pushButtonConnectText(tr("Connecting..."));

    ErrorCode e = ErrorCode::NoError;
#ifdef Q_DEBUG
    //QString output = m_serverController->checkSshConnection(serverCredentials, &e);
#else
    QString output;
#endif

    bool ok = true;
    if (e) {
        set_labelWaitInfoVisible(true);
        set_labelWaitInfoText(errorString(e));
        ok = false;
    }
    else {
        if (output.contains("Please login as the user")) {
            output.replace("\n", "");
            set_labelWaitInfoVisible(true);
            set_labelWaitInfoText(output);
            ok = false;
        }
    }

    set_pushButtonConnectEnabled(true);
    set_pushButtonConnectText(tr("Connect"));

    uiLogic()->installCredentials = serverCredentials;
    if (ok) emit uiLogic()->goToPage(Page::NewServer);
}

void StartPageLogic::onPushButtonImport()
{
    importConnectionFromCode(lineEditStartExistingCodeText());
}

void StartPageLogic::onPushButtonImportOpenFile()
{
    QString fileName = QFileDialog::getOpenFileName(nullptr, tr("Open profile"),
        QStandardPaths::writableLocation(QStandardPaths::DocumentsLocation), "*.vpn *.ovpn");

    if (fileName.isEmpty()) return;

    QFile file(fileName);
    file.open(QIODevice::ReadOnly);
    QByteArray data = file.readAll();

    auto configFormat = checkConfigFormat(QString(data));
    if (configFormat == "OpenVpn") {
        importConnectionFromOpenVpnConfig(QString(data));
    } else {
        importConnectionFromCode(QString(data));
    }
}

bool StartPageLogic::importConnection(const QJsonObject &profile)
{
    ServerCredentials credentials;
    credentials.hostName = profile.value(config_key::hostName).toString();
    credentials.port = profile.value(config_key::port).toInt();
    credentials.userName = profile.value(config_key::userName).toString();
    credentials.password = profile.value(config_key::password).toString();

    if (credentials.isValid() || profile.contains(config_key::containers)) {
        // check config
        uiLogic()->pageLogic<ViewConfigLogic>()->set_configJson(profile);
        emit uiLogic()->goToPage(Page::ViewConfig);
    }
    else {
        qDebug() << "Failed to import profile";
        qDebug().noquote() << QJsonDocument(profile).toJson();
        return false;
    }

    if (!profile.contains(config_key::containers)) {
        uiLogic()->selectedServerIndex = m_settings->defaultServerIndex();
        uiLogic()->selectedDockerContainer = m_settings->defaultContainer(uiLogic()->selectedServerIndex);
        uiLogic()->onUpdateAllPages();

        emit uiLogic()->goToPage(Page::ServerContainers);
    }

    return true;
}

bool StartPageLogic::importConnectionFromCode(QString code)
{
    code.replace("vpn://", "");
    QByteArray ba = QByteArray::fromBase64(code.toUtf8(), QByteArray::Base64UrlEncoding | QByteArray::OmitTrailingEquals);

    QByteArray ba_uncompressed = qUncompress(ba);
    if (!ba_uncompressed.isEmpty()) {
        ba = ba_uncompressed;
    }

    QJsonObject o;
    o = QJsonDocument::fromJson(ba).object();
    if (!o.isEmpty()) {
        return importConnection(o);
    }

    o = QJsonDocument::fromBinaryData(ba).object();
    if (!o.isEmpty()) {
        return importConnection(o);
    }
    return false;
}

bool StartPageLogic::importConnectionFromQr(const QByteArray &data)
{
    QJsonObject dataObj = QJsonDocument::fromJson(data).object();
    if (!dataObj.isEmpty()) {
        return importConnection(dataObj);
    }

    QByteArray ba_uncompressed = qUncompress(data);
    if (!ba_uncompressed.isEmpty()) {
        return importConnection(QJsonDocument::fromJson(ba_uncompressed).object());
    }

    return false;
}

bool StartPageLogic::importConnectionFromOpenVpnConfig(const QString &config)
{
    QJsonObject openVpnConfig;
    openVpnConfig[config_key::config] = config;

    QJsonObject lastConfig;
    lastConfig[config_key::last_config] = QString(QJsonDocument(openVpnConfig).toJson());

    QJsonObject containers;
    containers.insert(config_key::container, QJsonValue("amnezia-openvpn"));
    containers.insert(config_key::openvpn, QJsonValue(lastConfig));

    QJsonArray arr;
    arr.push_back(containers);

    QJsonObject o;
    o[config_key::containers] = arr;
    o[config_key::defaultContainer] = "amnezia-openvpn";
    o[config_key::description] = "OpenVpn server";


    const static QRegularExpression dnsRegExp("dhcp-option DNS \\b\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\b");
    QRegularExpressionMatchIterator dnsMatch = dnsRegExp.globalMatch(config);
    if (dnsMatch.hasNext()) {
        o[config_key::dns1] = dnsMatch.next().captured(0).split(" ").at(2);
    }
    if (dnsMatch.hasNext()) {
        o[config_key::dns2] = dnsMatch.next().captured(0).split(" ").at(2);
    }

    const static QRegularExpression hostNameRegExp("remote \\b\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\.\\d{1,3}\\b");
    QRegularExpressionMatch hostNameMatch = hostNameRegExp.match(config);
    if (hostNameMatch.hasMatch()) {
        o[config_key::hostName] = hostNameMatch.captured(0).split(" ").at(1);
    }

    o[config_key::is_third_party_config] = true;

    return importConnection(o);
}