: QObject(parent)

, webSocket(new QWebSocket(QString(), QWebSocketProtocol::VersionLatest, this))

{

connect(qApp, &QCoreApplication::aboutToQuit, this, &WebsocketClient::stop);

connect(webSocket, &QWebSocket::stateChanged, this, &WebsocketClient::onStateChanged);

connect(webSocket, &QWebSocket::connected, this, &WebsocketClient::onConnected);

connect(webSocket, &QWebSocket::disconnected, this, &WebsocketClient::onDisconnected);

connect(webSocket, QOverload<QAbstractSocket::SocketError>::of(&QWebSocket::error), this, &WebsocketClient::onError);

connect(webSocket, &QWebSocket::binaryMessageReceived, this, &WebsocketClient::onbinaryMessageReceived);

// Keep websocket connection alive

connect(&m_pingTimer, &QTimer::timeout, [this]{

if (webSocket->state() == QAbstractSocket::ConnectedState) {

webSocket->ping();

}

});

m_pingTimer.setInterval(30 * 1000);

connect(&m_connectionTimeout, &QTimer::timeout, this, &WebsocketClient::onConnectionTimeout);

m_websocketUrlIndex = QRandomGenerator::global()->bounded(m_websocketUrls[this->networkType()].length());

this->nextWebsocketUrl();

if (webSocket->state() == QAbstractSocket::ConnectedState) {

webSocket->sendBinaryMessage(data);

}

if (m_stopped) {

return;

}

if (conf()->get(Config::offlineMode).toBool()) {

return;

}

if (conf()->get(Config::disableWebsocket).toBool()) {

return;

}

// connect & reconnect on errors/close

auto state = webSocket->state();

if (state != QAbstractSocket::ConnectedState && state != QAbstractSocket::ConnectingState) {

qDebug() << "WebSocket connect:" << m_url.url();

webSocket->open(m_url);

}

qDebug() << Q_FUNC_INFO;

m_stopped = true;

webSocket->close();

m_connectionTimeout.stop();

m_pingTimer.stop();

qDebug() << "WebSocket disconnected";

this->nextWebsocketUrl();

QTimer::singleShot(1000, [this]{this->start();});

if (state == QAbstractSocket::ConnectingState) {

m_connectionTimeout.start(m_timeout*1000);

}

else if (state == QAbstractSocket::ConnectedState) {

m_connectionTimeout.stop();

m_pingTimer.start();

}

qCritical() << "WebSocket error: " << error;

auto state = webSocket->state();

if (state == QAbstractSocket::ConnectedState || state == QAbstractSocket::ConnectingState) {

webSocket->abort();

}

Config::Proxy networkType = this->networkType();

m_websocketUrlIndex = (m_websocketUrlIndex+1)%m_websocketUrls[networkType].length();

m_url = m_websocketUrls[networkType][m_websocketUrlIndex];

if (conf()->get(Config::proxy).toInt() == Config::Proxy::Tor && conf()->get(Config::torOnlyAllowOnion).toBool()) {

// Websocket performance with onion services is abysmal, connect to clearnet server unless instructed otherwise

return Config::Proxy::Tor;

}

else if (conf()->get(Config::proxy).toInt() == Config::Proxy::i2p) {

return Config::Proxy::i2p;

}

else {

return Config::Proxy::None;

}

qWarning() << "Websocket connection timeout";

m_timeout = std::min(m_timeout + 5, 60);

m_connectionTimeout.setInterval(m_timeout*1000);

this->onDisconnected();

if (!Utils::validateJSON(message)) {

qCritical() << "Could not interpret WebSocket message as JSON";

return;

}

QJsonDocument doc = QJsonDocument::fromJson(message);

QJsonObject object = doc.object();

if(!object.contains("cmd") || !object.contains("data")) {

qCritical() << "Invalid WebSocket message received";

return;

}

emit WSMessage(object);