wowlet/src/utils/nodes.h
dsc 133700160a Feather, a free Monero desktop wallet
Co-Authored-By: tobtoht <thotbot@protonmail.com>
2020-10-08 04:31:22 +02:00

138 lines
3.7 KiB
C++

// SPDX-License-Identifier: BSD-3-Clause
// Copyright (c) 2020, The Monero Project.
#ifndef FEATHER_NODES_H
#define FEATHER_NODES_H
#include <QtCore>
#include <QTimer>
#include <QRegExp>
#include <QApplication>
#include <QtNetwork>
#include <QNetworkAccessManager>
#include <QNetworkReply>
#include "model/NodeModel.h"
#include "utils/utils.h"
#include "utils/config.h"
enum NodeSource {
websocket = 0,
custom
};
struct FeatherNode {
FeatherNode(QString _address, unsigned int height, bool online) : height(height), online(online){
// wonky ipv4/host parsing, should be fine(tm)(c).
if(_address.isEmpty()) return;
_address = _address.replace("https://", "");
_address = _address.replace("http://", "");
if(_address.contains("@")){ // authentication, user/pass
const auto spl = _address.split("@");
const auto &creds = spl.at(0);
if(creds.contains(":")) {
const auto _spl = creds.split(":");
this->username = _spl.at(0).trimmed().replace(" ", "");
this->password = _spl.at(1).trimmed().replace(" ", "");
}
_address = spl.at(1);
}
if(!_address.contains(":"))
_address += ":18089";
this->address = _address;
if(this->address.contains(".onion"))
tor = true;
this->full = this->generateFull();
};
QString address;
QString full;
unsigned int height;
bool online = false;
QString username;
QString password;
bool cached = false;
bool custom = false;
bool tor = false;
bool isConnecting = false;
bool isActive = false;
QString generateFull() {
QString auth;
if(!this->username.isEmpty() && !this->password.isEmpty())
auth = QString("%1:%2@").arg(this->username).arg(this->password);
return QString("%1%2").arg(auth).arg(this->address);
}
QString as_url() {
return QString("http://%1/get_info").arg(this->full);
}
bool operator == (const FeatherNode &other) const {
return this->full == other.full;
}
};
class Nodes : public QObject {
Q_OBJECT
public:
explicit Nodes(AppContext *ctx, QNetworkAccessManager *networkAccessManager, QObject *parent = nullptr);
void loadConfig();
void writeConfig();
void stopTimer();
NodeSource source();
FeatherNode connection();
QList<FeatherNode> customNodes();
NodeModel *modelWebsocket;
NodeModel *modelCustom;
QStringList activityLog;
public slots:
void onWalletClosing();
void connectToNode();
void connectToNode(FeatherNode node);
void onWSNodesReceived(const QList<QSharedPointer<FeatherNode>>& nodes);
void onNodeSourceChanged(NodeSource nodeSource);
void setCustomNodes(QList<FeatherNode> nodes);
private slots:
void onConnectionTimer();
signals:
void WSNodeExhausted();
void nodeExhausted();
void updateStatus(const QString &msg);
private:
AppContext *m_ctx = nullptr;
NodeSource m_source = NodeSource::websocket;
QNetworkAccessManager *m_networkAccessManager = nullptr;
QJsonObject m_configJson;
QList<FeatherNode> m_customNodes;
QList<FeatherNode> m_websocketNodes;
FeatherNode m_connection; // current active connection, if any
QTimer *m_connectionTimer = new QTimer(this);
time_t m_connectionAttemptTime = 0;
QStringList m_connectionAttempts;
bool m_wsNodesReceived = false;
bool m_wsExhaustedWarningEmitted = true;
bool m_customExhaustedWarningEmitted = true;
FeatherNode pickEligibleNode();
void updateModels();
void resetLocalState();
void exhausted();
void WSNodeExhaustedWarning();
void nodeExhaustedWarning();
};
#endif //FEATHER_NODES_H