From 6c4cb405b18638e586ac34521259d5ee34de0df8 Mon Sep 17 00:00:00 2001
From: "Sebastian J. Wolf" <sebastian@ygriega.de>
Date: Thu, 13 Aug 2020 00:20:28 +0200
Subject: [PATCH] Process connection status updates

---
 src/tdlibreceiver.cpp | 11 +++++++++++
 src/tdlibreceiver.h   |  2 ++
 src/tdlibwrapper.cpp  | 29 ++++++++++++++++++++++++++++-
 src/tdlibwrapper.h    | 12 ++++++++++++
 4 files changed, 53 insertions(+), 1 deletion(-)

diff --git a/src/tdlibreceiver.cpp b/src/tdlibreceiver.cpp
index d21c354..76d2a0f 100644
--- a/src/tdlibreceiver.cpp
+++ b/src/tdlibreceiver.cpp
@@ -43,6 +43,10 @@ void TDLibReceiver::processReceivedDocument(const QJsonDocument &receivedJsonDoc
     if (objectTypeName == "updateAuthorizationState") {
         this->processUpdateAuthorizationState(receivedInformation);
     }
+
+    if (objectTypeName == "updateConnectionState") {
+        this->processUpdateConnectionState(receivedInformation);
+    }
 }
 
 void TDLibReceiver::processUpdateOption(const QVariantMap &receivedInformation)
@@ -65,3 +69,10 @@ void TDLibReceiver::processUpdateAuthorizationState(const QVariantMap &receivedI
     qDebug() << "[TDLibReceiver] Authorization state changed: " << authorizationState;
     emit authorizationStateChanged(authorizationState);
 }
+
+void TDLibReceiver::processUpdateConnectionState(const QVariantMap &receivedInformation)
+{
+    QString connectionState = receivedInformation.value("state").toMap().value("@type").toString();
+    qDebug() << "[TDLibReceiver] Connection state changed: " << connectionState;
+    emit connectionStateChanged(connectionState);
+}
diff --git a/src/tdlibreceiver.h b/src/tdlibreceiver.h
index e1fed92..90e5a09 100644
--- a/src/tdlibreceiver.h
+++ b/src/tdlibreceiver.h
@@ -21,6 +21,7 @@ signals:
     void versionDetected(const QString &version);
     void authorizationStateChanged(const QString &authorizationState);
     void optionUpdated(const QString &optionName, const QVariant &optionValue);
+    void connectionStateChanged(const QString &connectionState);
 
 private:
     void *tdLibClient;
@@ -30,6 +31,7 @@ private:
     void processReceivedDocument(const QJsonDocument &receivedJsonDocument);
     void processUpdateOption(const QVariantMap &receivedInformation);
     void processUpdateAuthorizationState(const QVariantMap &receivedInformation);
+    void processUpdateConnectionState(const QVariantMap &receivedInformation);
 };
 
 #endif // TDLIBRECEIVER_H
diff --git a/src/tdlibwrapper.cpp b/src/tdlibwrapper.cpp
index f1e824c..96dee15 100644
--- a/src/tdlibwrapper.cpp
+++ b/src/tdlibwrapper.cpp
@@ -39,6 +39,7 @@ TDLibWrapper::TDLibWrapper(QObject *parent) : QObject(parent)
     connect(this->tdLibReceiver, SIGNAL(versionDetected(QString)), this, SLOT(handleVersionDetected(QString)));
     connect(this->tdLibReceiver, SIGNAL(authorizationStateChanged(QString)), this, SLOT(handleAuthorizationStateChanged(QString)));
     connect(this->tdLibReceiver, SIGNAL(optionUpdated(QString, QVariant)), this, SLOT(handleOptionUpdated(QString, QVariant)));
+    connect(this->tdLibReceiver, SIGNAL(connectionStateChanged(QString)), this, SLOT(handleConnectionStateChanged(QString)));
 
     this->tdLibReceiver->start();
 }
@@ -70,6 +71,11 @@ TDLibWrapper::AuthorizationState TDLibWrapper::getAuthorizationState()
     return this->authorizationState;
 }
 
+TDLibWrapper::ConnectionState TDLibWrapper::getConnectionState()
+{
+    return this->connectionState;
+}
+
 void TDLibWrapper::handleVersionDetected(const QString &version)
 {
     this->version = version;
@@ -78,7 +84,7 @@ void TDLibWrapper::handleVersionDetected(const QString &version)
 
 void TDLibWrapper::handleAuthorizationStateChanged(const QString &authorizationState)
 {
-        if (authorizationState == "authorizationStateClosed") {
+    if (authorizationState == "authorizationStateClosed") {
         this->authorizationState = AuthorizationState::Closed;
     }
 
@@ -134,6 +140,27 @@ void TDLibWrapper::handleOptionUpdated(const QString &optionName, const QVariant
     emit optionUpdated(optionName, optionValue);
 }
 
+void TDLibWrapper::handleConnectionStateChanged(const QString &connectionState)
+{
+    if (connectionState == "connectionStateConnecting") {
+        this->connectionState = ConnectionState::Connecting;
+    }
+    if (connectionState == "connectionStateConnectingToProxy") {
+        this->connectionState = ConnectionState::ConnectingToProxy;
+    }
+    if (connectionState == "connectionStateReady") {
+        this->connectionState = ConnectionState::IsReady;
+    }
+    if (connectionState == "connectionStateUpdating") {
+        this->connectionState = ConnectionState::Updating;
+    }
+    if (connectionState == "connectionStateWaitingForNetwork") {
+        this->connectionState = ConnectionState::WaitingForNetwork;
+    }
+
+    emit connectionStateChanged(this->connectionState);
+}
+
 void TDLibWrapper::setInitialParameters()
 {
     qDebug() << "[TDLibWrapper] Sending initial parameters to TD Lib";
diff --git a/src/tdlibwrapper.h b/src/tdlibwrapper.h
index e8ad3bf..951099d 100644
--- a/src/tdlibwrapper.h
+++ b/src/tdlibwrapper.h
@@ -30,25 +30,37 @@ public:
         WaitTdlibParameters
     };
 
+    enum ConnectionState {
+        Connecting,
+        ConnectingToProxy,
+        IsReady,
+        Updating,
+        WaitingForNetwork
+    };
+
     Q_INVOKABLE void sendRequest(const QVariantMap &requestObject);
     Q_INVOKABLE QString getVersion();
     Q_INVOKABLE TDLibWrapper::AuthorizationState getAuthorizationState();
+    Q_INVOKABLE TDLibWrapper::ConnectionState getConnectionState();
 
 signals:
     void versionDetected(const QString &version);
     void authorizationStateChanged(const TDLibWrapper::AuthorizationState &authorizationState);
     void optionUpdated(const QString &optionName, const QVariant &optionValue);
+    void connectionStateChanged(const TDLibWrapper::ConnectionState &connectionState);
 
 public slots:
     void handleVersionDetected(const QString &version);
     void handleAuthorizationStateChanged(const QString &authorizationState);
     void handleOptionUpdated(const QString &optionName, const QVariant &optionValue);
+    void handleConnectionStateChanged(const QString &connectionState);
 
 private:
     void *tdLibClient;
     TDLibReceiver *tdLibReceiver;
     QString version;
     TDLibWrapper::AuthorizationState authorizationState;
+    TDLibWrapper::ConnectionState connectionState;
     QVariantMap options;
 
     void setInitialParameters();