#include "ServerSettingsLogic.h"
#include "vpnconnection.h"

#include "../uilogic.h"
#include "ServerListLogic.h"
#include "ShareConnectionLogic.h"
#include "VpnLogic.h"

#include "core/errorstrings.h"
#include <core/servercontroller.h>

ServerSettingsLogic::ServerSettingsLogic(UiLogic *logic, QObject *parent):
    PageLogicBase(logic, parent),
    m_pageServerSettingsEnabled{true},
    m_labelServerSettingsWaitInfoVisible{true},
    m_pushButtonServerSettingsClearVisible{true},
    m_pushButtonServerSettingsClearClientCacheVisible{true},
    m_pushButtonServerSettingsShareFullVisible{true},
    m_pushButtonServerSettingsClearText{tr("Clear server from Amnezia software")},
    m_pushButtonServerSettingsClearClientCacheText{tr("Clear client cached profile")}
{

}

void ServerSettingsLogic::updateServerSettingsPage()
{
    set_labelServerSettingsWaitInfoVisible(false);
    set_labelServerSettingsWaitInfoText("");
    set_pushButtonServerSettingsClearVisible(m_settings.haveAuthData(uiLogic()->selectedServerIndex));
    set_pushButtonServerSettingsClearClientCacheVisible(m_settings.haveAuthData(uiLogic()->selectedServerIndex));
    set_pushButtonServerSettingsShareFullVisible(m_settings.haveAuthData(uiLogic()->selectedServerIndex));
    QJsonObject server = m_settings.server(uiLogic()->selectedServerIndex);
    QString port = server.value(config_key::port).toString();
    set_labelServerSettingsServerText(QString("%1@%2%3%4")
                                     .arg(server.value(config_key::userName).toString())
                                     .arg(server.value(config_key::hostName).toString())
                                     .arg(port.isEmpty() ? "" : ":")
                                     .arg(port));
    set_lineEditServerSettingsDescriptionText(server.value(config_key::description).toString());
    QString selectedContainerName = m_settings.defaultContainerName(uiLogic()->selectedServerIndex);
    set_labelServerSettingsCurrentVpnProtocolText(tr("Protocol: ") + selectedContainerName);
}

void ServerSettingsLogic::onPushButtonServerSettingsClearServer()
{
    set_pageServerSettingsEnabled(false);
    set_pushButtonServerSettingsClearText(tr("Uninstalling Amnezia software..."));

    if (m_settings.defaultServerIndex() == uiLogic()->selectedServerIndex) {
        uiLogic()->vpnLogic()->onDisconnect();
    }

    ErrorCode e = ServerController::removeAllContainers(m_settings.serverCredentials(uiLogic()->selectedServerIndex));
    ServerController::disconnectFromHost(m_settings.serverCredentials(uiLogic()->selectedServerIndex));
    if (e) {
        uiLogic()->setDialogConnectErrorText(
                    tr("Error occurred while configuring server.") + "\n" +
                    errorString(e) + "\n" +
                    tr("See logs for details."));
        emit uiLogic()->showConnectErrorDialog();
    }
    else {
        set_labelServerSettingsWaitInfoVisible(true);
        set_labelServerSettingsWaitInfoText(tr("Amnezia server successfully uninstalled"));
    }

    m_settings.setContainers(uiLogic()->selectedServerIndex, {});
    m_settings.setDefaultContainer(uiLogic()->selectedServerIndex, DockerContainer::None);

    set_pageServerSettingsEnabled(true);
    set_pushButtonServerSettingsClearText(tr("Clear server from Amnezia software"));
}

void ServerSettingsLogic::onPushButtonServerSettingsForgetServer()
{
    if (m_settings.defaultServerIndex() == uiLogic()->selectedServerIndex && uiLogic()->m_vpnConnection->isConnected()) {
        uiLogic()->vpnLogic()->onDisconnect();
    }
    m_settings.removeServer(uiLogic()->selectedServerIndex);

    if (m_settings.defaultServerIndex() == uiLogic()->selectedServerIndex) {
        m_settings.setDefaultServer(0);
    }
    else if (m_settings.defaultServerIndex() > uiLogic()->selectedServerIndex) {
        m_settings.setDefaultServer(m_settings.defaultServerIndex() - 1);
    }

    if (m_settings.serversCount() == 0) {
        m_settings.setDefaultServer(-1);
    }


    uiLogic()->selectedServerIndex = -1;

    uiLogic()->serverListLogic()->updateServersListPage();

    if (m_settings.serversCount() == 0) {
        uiLogic()->setStartPage(Page::Start);
    }
    else {
        uiLogic()->closePage();
    }
}

void ServerSettingsLogic::onPushButtonServerSettingsClearClientCacheClicked()
{
    set_pushButtonServerSettingsClearClientCacheText(tr("Cache cleared"));

    const auto &containers = m_settings.containers(uiLogic()->selectedServerIndex);
    for (DockerContainer container: containers.keys()) {
        m_settings.clearLastConnectionConfig(uiLogic()->selectedServerIndex, container);
    }

    QTimer::singleShot(3000, this, [this]() {
        set_pushButtonServerSettingsClearClientCacheText(tr("Clear client cached profile"));
    });
}

void ServerSettingsLogic::onLineEditServerSettingsDescriptionEditingFinished()
{
    const QString &newText = lineEditServerSettingsDescriptionText();
    QJsonObject server = m_settings.server(uiLogic()->selectedServerIndex);
    server.insert(config_key::description, newText);
    m_settings.editServer(uiLogic()->selectedServerIndex, server);
    uiLogic()->serverListLogic()->updateServersListPage();
}

void ServerSettingsLogic::onPushButtonServerSettingsShareFullClicked()
{
    uiLogic()->shareConnectionLogic()->updateSharingPage(uiLogic()->selectedServerIndex, m_settings.serverCredentials(uiLogic()->selectedServerIndex), DockerContainer::None);
    uiLogic()->goToPage(Page::ShareConnection);
}