adityapk00
6 years ago
committed by
GitHub
GitHub
14 changed files with 602 additions and 380 deletions
@ -0,0 +1,319 @@ |
|||||
|
#include "connection.h" |
||||
|
#include "mainwindow.h" |
||||
|
#include "settings.h" |
||||
|
#include "ui_connection.h" |
||||
|
#include "rpc.h" |
||||
|
|
||||
|
#include "precompiled.h" |
||||
|
|
||||
|
using json = nlohmann::json; |
||||
|
/*
|
||||
|
class LoadingDialog : public QDialog { |
||||
|
//Q_OBJECT
|
||||
|
public: |
||||
|
LoadingDialog(QWidget* parent); |
||||
|
~LoadingDialog(); |
||||
|
public slots: |
||||
|
void reject(); |
||||
|
}; |
||||
|
|
||||
|
LoadingDialog::LoadingDialog(QWidget* parent) : QDialog(parent) {} |
||||
|
LoadingDialog::~LoadingDialog() {} |
||||
|
void LoadingDialog::reject() { |
||||
|
//event->ignore();
|
||||
|
} |
||||
|
*/ |
||||
|
ConnectionLoader::ConnectionLoader(MainWindow* main, RPC* rpc) { |
||||
|
this->main = main; |
||||
|
this->rpc = rpc; |
||||
|
|
||||
|
d = new QDialog(main); |
||||
|
connD = new Ui_ConnectionDialog(); |
||||
|
connD->setupUi(d); |
||||
|
|
||||
|
// Center on screen
|
||||
|
QRect screenGeometry = QApplication::desktop()->screenGeometry(d); |
||||
|
int x = (screenGeometry.width() - d->width()) / 2; |
||||
|
int y = (screenGeometry.height() - d->height()) / 2; |
||||
|
d->move(x, y); |
||||
|
connD->buttonBox->setEnabled(false); |
||||
|
d->show(); |
||||
|
} |
||||
|
|
||||
|
ConnectionLoader::~ConnectionLoader() { |
||||
|
delete d; |
||||
|
delete connD; |
||||
|
} |
||||
|
|
||||
|
void ConnectionLoader::loadConnection() { |
||||
|
// Priority 1: Try to connect to detect zcash.conf and connect to it.
|
||||
|
bool isZcashConfPresent = false; |
||||
|
auto config = autoDetectZcashConf(); |
||||
|
|
||||
|
// If not autodetected, go and read the UI Settings
|
||||
|
if (config.get() != nullptr) { |
||||
|
isZcashConfPresent = true; |
||||
|
} else { |
||||
|
config = loadFromSettings(); |
||||
|
|
||||
|
if (config.get() == nullptr) { |
||||
|
// Nothing configured, show an error
|
||||
|
auto explanation = QString() |
||||
|
% "A zcash.conf was not found on this machine.\n\n" |
||||
|
% "If you are connecting to a remote/non-standard node " |
||||
|
% "please set the host/port and user/password in the File->Settings menu."; |
||||
|
|
||||
|
showError(explanation); |
||||
|
rpc->setConnection(nullptr); |
||||
|
|
||||
|
return; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
auto connection = makeConnection(config); |
||||
|
refreshZcashdState(connection); |
||||
|
} |
||||
|
|
||||
|
Connection* ConnectionLoader::makeConnection(std::shared_ptr<ConnectionConfig> config) { |
||||
|
QNetworkAccessManager* client = new QNetworkAccessManager(main); |
||||
|
|
||||
|
QUrl myurl; |
||||
|
myurl.setScheme("http"); |
||||
|
myurl.setHost(config.get()->host); |
||||
|
myurl.setPort(config.get()->port.toInt()); |
||||
|
|
||||
|
QNetworkRequest* request = new QNetworkRequest(); |
||||
|
request->setUrl(myurl); |
||||
|
request->setHeader(QNetworkRequest::ContentTypeHeader, "text/plain"); |
||||
|
|
||||
|
QString userpass = config.get()->rpcuser % ":" % config.get()->rpcpassword; |
||||
|
QString headerData = "Basic " + userpass.toLocal8Bit().toBase64(); |
||||
|
request->setRawHeader("Authorization", headerData.toLocal8Bit()); |
||||
|
|
||||
|
return new Connection(main, client, request, config); |
||||
|
} |
||||
|
|
||||
|
void ConnectionLoader::refreshZcashdState(Connection* connection) { |
||||
|
json payload = { |
||||
|
{"jsonrpc", "1.0"}, |
||||
|
{"id", "someid"}, |
||||
|
{"method", "getinfo"} |
||||
|
}; |
||||
|
connection->doRPC(payload, |
||||
|
[=] (auto) { |
||||
|
// Success
|
||||
|
d->hide(); |
||||
|
rpc->setConnection(connection); |
||||
|
}, |
||||
|
[=] (auto reply, auto res) { |
||||
|
auto err = reply->error(); |
||||
|
// Failed, see what it is.
|
||||
|
qDebug() << err << ":" << QString::fromStdString(res.dump()); |
||||
|
|
||||
|
if (err == QNetworkReply::NetworkError::ConnectionRefusedError) { |
||||
|
auto isZcashConfFound = connection->config.get()->usingZcashConf; |
||||
|
auto explanation = QString() |
||||
|
% (isZcashConfFound ? "A zcash.conf file was found, but a" : "A") |
||||
|
% " connection to zcashd could not be established.\n\n" |
||||
|
% "If you are connecting to a remote/non-standard node " |
||||
|
% "please set the host/port and user/password in the File->Settings menu"; |
||||
|
|
||||
|
showError(explanation); |
||||
|
} else if (err == QNetworkReply::NetworkError::AuthenticationRequiredError) { |
||||
|
auto explanation = QString() |
||||
|
% "Authentication failed. The username / password you specified was " |
||||
|
% "not accepted by zcashd. Try changing it in the File->Settings menu"; |
||||
|
|
||||
|
showError(explanation); |
||||
|
} else if (err == QNetworkReply::NetworkError::InternalServerError && !res.is_discarded()) { |
||||
|
// The server is loading, so just poll until it succeeds
|
||||
|
QString status = QString::fromStdString(res["error"]["message"]); |
||||
|
|
||||
|
QIcon icon = QApplication::style()->standardIcon(QStyle::SP_MessageBoxInformation); |
||||
|
connD->icon->setPixmap(icon.pixmap(128, 128)); |
||||
|
connD->status->setText(status); |
||||
|
|
||||
|
// Refresh after one second
|
||||
|
QTimer::singleShot(1000, [=]() { this->refreshZcashdState(connection); }); |
||||
|
} |
||||
|
} |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
void ConnectionLoader::showError(QString explanation) { |
||||
|
QIcon icon = QApplication::style()->standardIcon(QStyle::SP_MessageBoxCritical); |
||||
|
connD->icon->setPixmap(icon.pixmap(128, 128)); |
||||
|
connD->status->setText(explanation); |
||||
|
connD->title->setText(""); |
||||
|
|
||||
|
connD->buttonBox->setEnabled(true); |
||||
|
} |
||||
|
|
||||
|
/*
|
||||
|
int ConnectionLoader::getProgressFromStatus(QString status) { |
||||
|
if (status.startsWith("Loading block")) return 20; |
||||
|
if (status.startsWith("Verifying")) return 40; |
||||
|
if (status.startsWith("Loading Wallet")) return 60; |
||||
|
if (status.startsWith("Activating")) return 80; |
||||
|
if (status.startsWith("Rescanning")) return 90; |
||||
|
return 0; |
||||
|
} |
||||
|
*/ |
||||
|
|
||||
|
/**
|
||||
|
* Try to automatically detect a zcash.conf file in the correct location and load parameters |
||||
|
*/ |
||||
|
std::shared_ptr<ConnectionConfig> ConnectionLoader::autoDetectZcashConf() { |
||||
|
|
||||
|
#ifdef Q_OS_LINUX |
||||
|
auto confLocation = QStandardPaths::locate(QStandardPaths::HomeLocation, ".zcash/zcash.conf"); |
||||
|
#elif defined(Q_OS_DARWIN) |
||||
|
auto confLocation = QStandardPaths::locate(QStandardPaths::HomeLocation, "/Library/Application Support/Zcash/zcash.conf"); |
||||
|
#else |
||||
|
auto confLocation = QStandardPaths::locate(QStandardPaths::AppDataLocation, "../../Zcash/zcash.conf"); |
||||
|
#endif |
||||
|
|
||||
|
confLocation = QDir::cleanPath(confLocation); |
||||
|
|
||||
|
if (confLocation.isNull()) { |
||||
|
// No zcash file, just return with nothing
|
||||
|
return nullptr; |
||||
|
} |
||||
|
|
||||
|
QFile file(confLocation); |
||||
|
if (!file.open(QIODevice::ReadOnly)) { |
||||
|
qDebug() << file.errorString(); |
||||
|
return nullptr; |
||||
|
} |
||||
|
|
||||
|
QTextStream in(&file); |
||||
|
|
||||
|
auto zcashconf = new ConnectionConfig(); |
||||
|
zcashconf->host = "127.0.0.1"; |
||||
|
zcashconf->connType = ConnectionType::DetectedConfExternalZcashD; |
||||
|
zcashconf->usingZcashConf = true; |
||||
|
|
||||
|
Settings::getInstance()->setUsingZcashConf(confLocation); |
||||
|
|
||||
|
while (!in.atEnd()) { |
||||
|
QString line = in.readLine(); |
||||
|
auto s = line.indexOf("="); |
||||
|
QString name = line.left(s).trimmed().toLower(); |
||||
|
QString value = line.right(line.length() - s - 1).trimmed(); |
||||
|
|
||||
|
if (name == "rpcuser") { |
||||
|
zcashconf->rpcuser = value; |
||||
|
} |
||||
|
if (name == "rpcpassword") { |
||||
|
zcashconf->rpcpassword = value; |
||||
|
} |
||||
|
if (name == "rpcport") { |
||||
|
zcashconf->port = value; |
||||
|
} |
||||
|
if (name == "testnet" && |
||||
|
value == "1" && |
||||
|
zcashconf->port.isEmpty()) { |
||||
|
zcashconf->port = "18232"; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// If rpcport is not in the file, and it was not set by the testnet=1 flag, then go to default
|
||||
|
if (zcashconf->port.isEmpty()) zcashconf->port = "8232"; |
||||
|
|
||||
|
file.close(); |
||||
|
|
||||
|
return std::shared_ptr<ConnectionConfig>(zcashconf); |
||||
|
} |
||||
|
|
||||
|
/**
|
||||
|
* Load connection settings from the UI, which indicates an unknown, external zcashd |
||||
|
*/ |
||||
|
std::shared_ptr<ConnectionConfig> ConnectionLoader::loadFromSettings() { |
||||
|
// Load from the QT Settings.
|
||||
|
QSettings s; |
||||
|
|
||||
|
auto host = s.value("connection/host").toString(); |
||||
|
auto port = s.value("connection/port").toString(); |
||||
|
auto username = s.value("connection/rpcuser").toString(); |
||||
|
auto password = s.value("connection/rpcpassword").toString(); |
||||
|
|
||||
|
if (username.isEmpty() || password.isEmpty()) |
||||
|
return nullptr; |
||||
|
|
||||
|
auto uiConfig = new ConnectionConfig{ host, port, username, password, false, ConnectionType::UISettingsZCashD }; |
||||
|
|
||||
|
return std::shared_ptr<ConnectionConfig>(uiConfig); |
||||
|
} |
||||
|
|
||||
|
|
||||
|
|
||||
|
|
||||
|
|
||||
|
|
||||
|
|
||||
|
|
||||
|
Connection::Connection(MainWindow* m, QNetworkAccessManager* c, QNetworkRequest* r, std::shared_ptr<ConnectionConfig> conf) { |
||||
|
this->restclient = c; |
||||
|
this->request = r; |
||||
|
this->config = conf; |
||||
|
this->main = m; |
||||
|
} |
||||
|
|
||||
|
Connection::~Connection() { |
||||
|
delete restclient; |
||||
|
delete request; |
||||
|
} |
||||
|
|
||||
|
|
||||
|
|
||||
|
void Connection::doRPC(const json& payload, const std::function<void(json)>& cb, |
||||
|
const std::function<void(QNetworkReply*, const json&)>& ne) { |
||||
|
QNetworkReply *reply = restclient->post(*request, QByteArray::fromStdString(payload.dump())); |
||||
|
|
||||
|
QObject::connect(reply, &QNetworkReply::finished, [=] { |
||||
|
reply->deleteLater(); |
||||
|
|
||||
|
if (reply->error() != QNetworkReply::NoError) { |
||||
|
auto parsed = json::parse(reply->readAll(), nullptr, false); |
||||
|
ne(reply, parsed); |
||||
|
|
||||
|
return; |
||||
|
} |
||||
|
|
||||
|
auto parsed = json::parse(reply->readAll(), nullptr, false); |
||||
|
if (parsed.is_discarded()) { |
||||
|
ne(reply, "Unknown error"); |
||||
|
} |
||||
|
|
||||
|
cb(parsed["result"]); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
void Connection::doRPCWithDefaultErrorHandling(const json& payload, const std::function<void(json)>& cb) { |
||||
|
doRPC(payload, cb, [=] (auto reply, auto parsed) { |
||||
|
if (!parsed.is_discarded() && !parsed["error"]["message"].is_null()) { |
||||
|
showTxError(QString::fromStdString(parsed["error"]["message"])); |
||||
|
} else { |
||||
|
showTxError(reply->errorString()); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
void Connection::doRPCIgnoreError(const json& payload, const std::function<void(json)>& cb) { |
||||
|
doRPC(payload, cb, [=] (auto, auto) { |
||||
|
// Ignored error handling
|
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
void Connection::showTxError(const QString& error) { |
||||
|
if (error.isNull()) return; |
||||
|
|
||||
|
QMessageBox msg(main); |
||||
|
msg.setIcon(QMessageBox::Icon::Critical); |
||||
|
msg.setWindowTitle("Transaction Error"); |
||||
|
|
||||
|
msg.setText("There was an error sending the transaction. The error was: \n\n" |
||||
|
+ error); |
||||
|
|
||||
|
msg.exec(); |
||||
|
} |
||||
@ -0,0 +1,125 @@ |
|||||
|
#ifndef CONNECTION_H |
||||
|
#define CONNECTION_H |
||||
|
|
||||
|
#include "mainwindow.h" |
||||
|
#include "ui_connection.h" |
||||
|
#include "precompiled.h" |
||||
|
|
||||
|
using json = nlohmann::json; |
||||
|
|
||||
|
class RPC; |
||||
|
|
||||
|
enum ConnectionType { |
||||
|
DetectedConfExternalZcashD = 1, |
||||
|
UISettingsZCashD, |
||||
|
InternalZcashD |
||||
|
}; |
||||
|
|
||||
|
struct ConnectionConfig { |
||||
|
QString host; |
||||
|
QString port; |
||||
|
QString rpcuser; |
||||
|
QString rpcpassword; |
||||
|
bool usingZcashConf; |
||||
|
|
||||
|
ConnectionType connType; |
||||
|
}; |
||||
|
|
||||
|
class Connection; |
||||
|
|
||||
|
class ConnectionLoader { |
||||
|
|
||||
|
public: |
||||
|
ConnectionLoader(MainWindow* main, RPC* rpc); |
||||
|
~ConnectionLoader(); |
||||
|
|
||||
|
void loadConnection(); |
||||
|
|
||||
|
private: |
||||
|
std::shared_ptr<ConnectionConfig> autoDetectZcashConf(); |
||||
|
std::shared_ptr<ConnectionConfig> loadFromSettings(); |
||||
|
|
||||
|
Connection* makeConnection(std::shared_ptr<ConnectionConfig> config); |
||||
|
|
||||
|
void refreshZcashdState(Connection* connection); |
||||
|
int getProgressFromStatus(QString status); |
||||
|
|
||||
|
void showError(QString explanation); |
||||
|
|
||||
|
QDialog* d; |
||||
|
Ui_ConnectionDialog* connD; |
||||
|
|
||||
|
MainWindow* main; |
||||
|
RPC* rpc; |
||||
|
}; |
||||
|
|
||||
|
/**
|
||||
|
* Represents a connection to a zcashd. It may even start a new zcashd if needed. |
||||
|
* This is also a UI class, so it may show a dialog waiting for the connection. |
||||
|
*/ |
||||
|
class Connection { |
||||
|
public: |
||||
|
Connection(MainWindow* m, QNetworkAccessManager* c, QNetworkRequest* r, std::shared_ptr<ConnectionConfig> conf); |
||||
|
~Connection(); |
||||
|
|
||||
|
QNetworkAccessManager* restclient; |
||||
|
QNetworkRequest* request; |
||||
|
std::shared_ptr<ConnectionConfig> config; |
||||
|
MainWindow* main; |
||||
|
|
||||
|
void doRPC(const json& payload, const std::function<void(json)>& cb, |
||||
|
const std::function<void(QNetworkReply*, const json&)>& ne); |
||||
|
void doRPCWithDefaultErrorHandling(const json& payload, const std::function<void(json)>& cb); |
||||
|
void doRPCIgnoreError(const json& payload, const std::function<void(json)>& cb) ; |
||||
|
|
||||
|
void showTxError(const QString& error); |
||||
|
|
||||
|
// Batch method. Note: Because of the template, it has to be in the header file.
|
||||
|
template<class T> |
||||
|
void doBatchRPC(const QList<T>& payloads, |
||||
|
std::function<json(T)> payloadGenerator, |
||||
|
std::function<void(QMap<T, json>*)> cb) { |
||||
|
auto responses = new QMap<T, json>(); // zAddr -> list of responses for each call.
|
||||
|
int totalSize = payloads.size(); |
||||
|
|
||||
|
for (auto item: payloads) { |
||||
|
json payload = payloadGenerator(item); |
||||
|
|
||||
|
QNetworkReply *reply = restclient->post(*request, QByteArray::fromStdString(payload.dump())); |
||||
|
|
||||
|
QObject::connect(reply, &QNetworkReply::finished, [=] { |
||||
|
reply->deleteLater(); |
||||
|
|
||||
|
auto all = reply->readAll(); |
||||
|
auto parsed = json::parse(all.toStdString(), nullptr, false); |
||||
|
|
||||
|
if (reply->error() != QNetworkReply::NoError) { |
||||
|
qDebug() << QString::fromStdString(parsed.dump()); |
||||
|
qDebug() << reply->errorString(); |
||||
|
|
||||
|
(*responses)[item] = json::object(); // Empty object
|
||||
|
} else { |
||||
|
if (parsed.is_discarded()) { |
||||
|
(*responses)[item] = json::object(); // Empty object
|
||||
|
} else { |
||||
|
(*responses)[item] = parsed["result"]; |
||||
|
} |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
auto waitTimer = new QTimer(main); |
||||
|
QObject::connect(waitTimer, &QTimer::timeout, [=]() { |
||||
|
if (responses->size() == totalSize) { |
||||
|
waitTimer->stop(); |
||||
|
|
||||
|
cb(responses); |
||||
|
|
||||
|
waitTimer->deleteLater(); |
||||
|
} |
||||
|
}); |
||||
|
waitTimer->start(100); |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
#endif |
||||
Loading…
Reference in new issue