// SPDX-License-Identifier: LGPL-3.0-or-later

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
*
* Copyright (C) 2013 - 2024, nymea GmbH
* Copyright (C) 2024 - 2025, chargebyte austria GmbH
*
* This file is part of libnymea-app.
*
* libnymea-app is free software: you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* libnymea-app is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with libnymea-app. If not, see <https://www.gnu.org/licenses/>.
*
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#include "nymeadiscovery.h"
#include "upnpdiscovery.h"
#include "zeroconfdiscovery.h"
#include "bluetoothservicediscovery.h"
#include "../nymeahost.h"

#include <QUuid>
#include <QBluetoothUuid>
#include <QUrlQuery>
#include <QSettings>
//#include <QNetworkConfigurationManager>
//#include <QNetworkSession>

#include "logging.h"
NYMEA_LOGGING_CATEGORY(dcDiscovery, "Discovery")

NymeaDiscovery::NymeaDiscovery(QObject *parent) : QObject(parent)
{
    m_nymeaHosts = new NymeaHosts(this);

    loadFromDisk();
}

NymeaDiscovery::~NymeaDiscovery()
{
}

bool NymeaDiscovery::discovering() const
{
    return m_discovering;
}

void NymeaDiscovery::setDiscovering(bool discovering)
{
    if (m_discovering == discovering)
        return;

    m_discovering = discovering;
    if (discovering) {
        if (m_zeroconfDiscoveryEnabled) {
            if (!m_zeroConf) {
                m_zeroConf = new ZeroconfDiscovery(m_nymeaHosts, this);
            }
        }

        // Start UPnP discovery
        if (m_upnpDiscoveryEnabled) {
            if (!m_upnp) {
                m_upnp = new UpnpDiscovery(m_nymeaHosts, this);
            }
            m_upnp->discover();
        }

        // Start Bluetooth discovery if HW is available
        if (m_bluetoothDiscoveryEnabled) {
            if (!m_bluetooth) {
                m_bluetooth = new BluetoothServiceDiscovery(m_nymeaHosts, this);
            }
            m_bluetooth->discover();
        }

    } else {

        if (m_upnp) {
            m_upnp->stopDiscovery();
        }

        if (m_bluetooth) {
            m_bluetooth->stopDiscovery();
        }
    }

    emit discoveringChanged();
}

NymeaHosts *NymeaDiscovery::nymeaHosts() const
{
    return m_nymeaHosts;
}

void NymeaDiscovery::cacheHost(NymeaHost *host)
{
    QSettings settings;
    settings.beginGroup("HostCache");
    settings.remove(host->uuid().toString());
    settings.beginGroup(host->uuid().toString());
    settings.setValue("name", host->name());
    QList<Connection*> connections;
    Connection *remoteConnection = host->connections()->bestMatch(Connection::BearerTypeCloud);
    if (remoteConnection) {
        connections.append(remoteConnection);
    }
    Connection *loopbackConnection = host->connections()->bestMatch(Connection::BearerTypeLoopback);
    if (loopbackConnection) {
        connections.append(loopbackConnection);
    }
    Connection *lanConnection = host->connections()->bestMatch(Connection::BearerTypeLan);
    if (lanConnection) {
        connections.append(lanConnection);
    }
    Connection *wanConnection = host->connections()->bestMatch(Connection::BearerTypeWan);
    if (wanConnection) {
        connections.append(wanConnection);
    }
    Connection *btConnection = host->connections()->bestMatch(Connection::BearerTypeBluetooth);
    if (btConnection) {
        connections.append(btConnection);
    }
    int i = 0;
    foreach (Connection *connection, connections) {
        settings.beginGroup(QString::number(i++));
        settings.setValue("url", connection->url());
        settings.setValue("bearerType", connection->bearerType());
        settings.setValue("secure", connection->secure());
        settings.setValue("displayName", connection->displayName());
        settings.setValue("manual", connection->manual());
        settings.endGroup();
    }
    settings.endGroup();
}

bool NymeaDiscovery::zeroconfDiscoveryEnable() const
{
    return m_zeroconfDiscoveryEnabled;
}

bool NymeaDiscovery::bluetoothDiscoveryEnabled() const
{
    return m_bluetoothDiscoveryEnabled;
}

bool NymeaDiscovery::upnpDiscoveryEnabled() const
{
    return m_upnpDiscoveryEnabled;
}

void NymeaDiscovery::setZeroconfDiscoveryEnabled(bool zeroconfDiscoveryEnabled)
{
    if (m_zeroconfDiscoveryEnabled  != zeroconfDiscoveryEnabled) {
        m_zeroconfDiscoveryEnabled = zeroconfDiscoveryEnabled;
        emit zeroconfDiscoveryEnabledChanged(m_zeroconfDiscoveryEnabled);
        if (!m_zeroconfDiscoveryEnabled && m_zeroConf && m_zeroConf->discovering()) {
            m_zeroConf->deleteLater();
            m_zeroConf = nullptr;
        }
    }
}

void NymeaDiscovery::setBluetoothDiscoveryEnabled(bool bluetoothDiscoveryEnabled)
{
    if (m_bluetoothDiscoveryEnabled != bluetoothDiscoveryEnabled) {
        m_bluetoothDiscoveryEnabled = bluetoothDiscoveryEnabled;
        emit bluetoothDiscoveryEnabledChanged(m_bluetoothDiscoveryEnabled);
        if (!m_bluetoothDiscoveryEnabled && m_bluetooth && m_bluetooth->discovering()) {
            m_bluetooth->stopDiscovery();
        }
    }
}

void NymeaDiscovery::setUpnpDiscoveryEnabled(bool upnpDiscoveryEnabled)
{
    if (m_upnpDiscoveryEnabled != upnpDiscoveryEnabled) {
        m_upnpDiscoveryEnabled = upnpDiscoveryEnabled;
        emit upnpDiscoveryEnabledChanged(m_upnpDiscoveryEnabled);
        if (!m_upnpDiscoveryEnabled && m_upnp && m_upnp->discovering()) {
            m_upnp->stopDiscovery();
        }
    }
}

void NymeaDiscovery::loadFromDisk()
{
    QSettings settings;
    settings.beginGroup("HostCache");
    foreach (const QString &serverUuid, settings.childGroups()) {
        settings.beginGroup(serverUuid);
        NymeaHost* host = m_nymeaHosts->find(QUuid(serverUuid));
        if (!host) {
            host = new NymeaHost(m_nymeaHosts);
            host->setName(settings.value("name").toString());
            host->setUuid(QUuid(serverUuid));
            m_nymeaHosts->addHost(host);
        }
        qCDebug(dcDiscovery()) << "Loaded Host from cache" << host->name() << host->uuid();
        foreach (const QString &group, settings.childGroups()) {
            settings.beginGroup(group);
            QString url = settings.value("url").toString();
            Connection* connection = host->connections()->find(url);
            if (!connection) {
                Connection::BearerType bearerType = static_cast<Connection::BearerType>(settings.value("bearerType").toInt());
                bool secure = settings.value("secure").toBool();
                QString displayName = settings.value("displayName").toString();
                connection = new Connection(url, bearerType, secure, displayName, host);
                connection->setManual(settings.value("manual").toBool());
                host->connections()->addConnection(connection);
                qCDebug(dcDiscovery()) << "|- Connection:" << group << connection->url() << connection->bearerType() << "secure:" << connection->secure();
            }
            settings.endGroup();
        }
        settings.endGroup();
    }

}

void NymeaDiscovery::updateActiveBearers()
{
}