318 lines
10 KiB
C++
318 lines
10 KiB
C++
/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
|
|
*
|
|
* Copyright 2013 - 2020, nymea GmbH
|
|
* Contact: contact@nymea.io
|
|
*
|
|
* This file is part of nymea.
|
|
* This project including source code and documentation is protected by
|
|
* copyright law, and remains the property of nymea GmbH. All rights, including
|
|
* reproduction, publication, editing and translation, are reserved. The use of
|
|
* this project is subject to the terms of a license agreement to be concluded
|
|
* with nymea GmbH in accordance with the terms of use of nymea GmbH, available
|
|
* under https://nymea.io/license
|
|
*
|
|
* GNU General Public License Usage
|
|
* Alternatively, this project may be redistributed and/or modified under the
|
|
* terms of the GNU General Public License as published by the Free Software
|
|
* Foundation, GNU version 3. This project 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 General
|
|
* Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* this project. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
* For any further details and any questions please contact us under
|
|
* contact@nymea.io or see our FAQ/Licensing Information on
|
|
* https://nymea.io/license/faq
|
|
*
|
|
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
|
|
|
|
#include "nymeadiscovery.h"
|
|
#include "upnpdiscovery.h"
|
|
#include "zeroconfdiscovery.h"
|
|
#include "bluetoothservicediscovery.h"
|
|
#include "connection/awsclient.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();
|
|
|
|
m_cloudPollTimer.setInterval(5000);
|
|
connect(&m_cloudPollTimer, &QTimer::timeout, this, [this](){
|
|
if (m_awsClient && m_awsClient->isLoggedIn()) {
|
|
m_awsClient->fetchDevices();
|
|
}
|
|
});
|
|
|
|
}
|
|
|
|
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();
|
|
}
|
|
|
|
// start polling cloud
|
|
m_cloudPollTimer.start();
|
|
// If we're logged in, poll right away
|
|
if (m_awsClient && m_awsClient->isLoggedIn()) {
|
|
m_awsClient->fetchDevices();
|
|
}
|
|
} else {
|
|
|
|
if (m_upnp) {
|
|
m_upnp->stopDiscovery();
|
|
}
|
|
|
|
if (m_bluetooth) {
|
|
m_bluetooth->stopDiscovery();
|
|
}
|
|
|
|
m_cloudPollTimer.stop();
|
|
}
|
|
|
|
emit discoveringChanged();
|
|
}
|
|
|
|
NymeaHosts *NymeaDiscovery::nymeaHosts() const
|
|
{
|
|
return m_nymeaHosts;
|
|
}
|
|
|
|
AWSClient *NymeaDiscovery::awsClient() const
|
|
{
|
|
return m_awsClient;
|
|
}
|
|
|
|
void NymeaDiscovery::setAwsClient(AWSClient *awsClient)
|
|
{
|
|
if (m_awsClient != awsClient) {
|
|
m_awsClient = awsClient;
|
|
emit awsClientChanged();
|
|
}
|
|
|
|
if (m_awsClient) {
|
|
connect(m_awsClient, &AWSClient::devicesFetched, this, &NymeaDiscovery::syncCloudDevices);
|
|
}
|
|
}
|
|
|
|
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.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::syncCloudDevices()
|
|
{
|
|
for (int i = 0; i < m_awsClient->awsDevices()->rowCount(); i++) {
|
|
AWSDevice *d = m_awsClient->awsDevices()->get(i);
|
|
NymeaHost *host = m_nymeaHosts->find(d->id());
|
|
bool alreayAdded = host != nullptr;
|
|
if (!alreayAdded) {
|
|
host = new NymeaHost();
|
|
host->setUuid(d->id());
|
|
}
|
|
host->setName(d->name());
|
|
QUrl url;
|
|
url.setScheme("cloud");
|
|
url.setHost(d->id());
|
|
Connection *conn = host->connections()->find(url);
|
|
if (!conn) {
|
|
conn = new Connection(url, Connection::BearerTypeCloud, true, d->id());
|
|
qCDebug(dcDiscovery) << "Adding new connection to host:" << host->name() << conn->url().toString();
|
|
host->connections()->addConnection(conn);
|
|
}
|
|
conn->setOnline(d->online());
|
|
if (!alreayAdded) {
|
|
qCDebug(dcDiscovery) << "Adding new host:" << d->name() << d->id();
|
|
m_nymeaHosts->addHost(host);
|
|
}
|
|
}
|
|
|
|
QList<NymeaHost*> hostsToRemove;
|
|
for (int i = 0; i < m_nymeaHosts->rowCount(); i++) {
|
|
NymeaHost *host = m_nymeaHosts->get(i);
|
|
for (int j = 0; j < host->connections()->rowCount(); j++) {
|
|
if (host->connections()->get(j)->bearerType() == Connection::BearerTypeCloud) {
|
|
if (m_awsClient->awsDevices()->getDevice(host->uuid().toString()) == nullptr) {
|
|
host->connections()->removeConnection(j);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if (host->connections()->rowCount() == 0) {
|
|
hostsToRemove.append(host);
|
|
}
|
|
}
|
|
while (!hostsToRemove.isEmpty()) {
|
|
m_nymeaHosts->removeHost(hostsToRemove.takeFirst());
|
|
}
|
|
}
|
|
|
|
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);
|
|
host->connections()->addConnection(connection);
|
|
qCDebug(dcDiscovery()) << "|- Connection:" << group << connection->url() << connection->bearerType() << "secure:" << connection->secure();
|
|
}
|
|
settings.endGroup();
|
|
}
|
|
settings.endGroup();
|
|
}
|
|
|
|
}
|
|
|
|
void NymeaDiscovery::updateActiveBearers()
|
|
{
|
|
}
|
|
|