2020-10-07 10:36:04 +00:00
|
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
// Copyright (c) 2014-2020, The Monero Project.
|
|
|
|
|
|
|
|
#include <QFile>
|
|
|
|
#include <QFileInfo>
|
|
|
|
#include <QDir>
|
|
|
|
#include <QDebug>
|
|
|
|
#include <QUrl>
|
|
|
|
#include <QtConcurrent/QtConcurrent>
|
|
|
|
#include <QMutex>
|
|
|
|
#include <QString>
|
|
|
|
|
|
|
|
#include "appcontext.h"
|
|
|
|
#include "utils/config.h"
|
|
|
|
#include "utils/networktype.h"
|
|
|
|
#include "utils/utils.h"
|
|
|
|
#include "keysfiles.h"
|
|
|
|
|
|
|
|
using namespace std::chrono;
|
|
|
|
|
|
|
|
|
2020-12-18 22:39:27 +00:00
|
|
|
WalletKeysFiles::WalletKeysFiles(const QFileInfo &info, int networkType, QString address) :
|
2020-10-07 10:36:04 +00:00
|
|
|
m_fileName(info.fileName()),
|
|
|
|
m_modified(info.lastModified().toSecsSinceEpoch()),
|
|
|
|
m_path(QDir::toNativeSeparators(info.absoluteFilePath())),
|
|
|
|
m_networkType(networkType),
|
|
|
|
m_address(std::move(address)) {}
|
|
|
|
|
|
|
|
QString WalletKeysFiles::fileName() const {
|
|
|
|
return m_fileName;
|
|
|
|
}
|
|
|
|
|
|
|
|
qint64 WalletKeysFiles::modified() const {
|
|
|
|
return m_modified;
|
|
|
|
}
|
|
|
|
|
|
|
|
QString WalletKeysFiles::address() const {
|
|
|
|
return m_address;
|
|
|
|
}
|
|
|
|
|
|
|
|
QString WalletKeysFiles::path() const {
|
|
|
|
return m_path;
|
|
|
|
}
|
|
|
|
|
2020-12-18 22:39:27 +00:00
|
|
|
int WalletKeysFiles::networkType() const {
|
2020-10-07 10:36:04 +00:00
|
|
|
return m_networkType;
|
|
|
|
}
|
|
|
|
|
|
|
|
WalletKeysFilesModel::WalletKeysFilesModel(AppContext *ctx, QObject *parent) :
|
|
|
|
QAbstractTableModel(parent),
|
|
|
|
m_ctx(ctx) {
|
|
|
|
this->walletDirectories << m_ctx->defaultWalletDirRoot;
|
|
|
|
auto walletPath = config()->get(Config::walletPath).toString();
|
|
|
|
if(!walletPath.isEmpty() && Utils::fileExists(walletPath)) {
|
|
|
|
QDir d = QFileInfo(walletPath).absoluteDir();
|
|
|
|
this->walletDirectories << d.absolutePath();
|
|
|
|
}
|
|
|
|
|
|
|
|
this->walletDirectories << m_ctx->homeDir;
|
|
|
|
this->walletDirectories.removeDuplicates();
|
|
|
|
|
|
|
|
this->m_walletKeysFilesItemModel = qobject_cast<QAbstractItemModel *>(this);
|
|
|
|
}
|
|
|
|
|
|
|
|
void WalletKeysFilesModel::clear() {
|
|
|
|
beginResetModel();
|
|
|
|
m_walletKeyFiles.clear();
|
|
|
|
endResetModel();
|
|
|
|
}
|
|
|
|
|
|
|
|
void WalletKeysFilesModel::refresh() {
|
|
|
|
this->clear();
|
|
|
|
this->findWallets();
|
|
|
|
}
|
|
|
|
|
|
|
|
void WalletKeysFilesModel::findWallets() {
|
|
|
|
qDebug() << "wallet .keys search initiated";
|
|
|
|
auto now = high_resolution_clock::now();
|
|
|
|
|
|
|
|
QRegExp rx("*.keys");
|
|
|
|
rx.setPatternSyntax(QRegExp::Wildcard);
|
|
|
|
QStringList walletPaths;
|
|
|
|
|
|
|
|
for(auto i = 0; i != this->walletDirectories.length(); i++) {
|
|
|
|
// Scan default wallet dir (~/Monero/)
|
|
|
|
walletPaths << Utils::fileFind(rx, this->walletDirectories[i], 0, i == 0 ? 2 : 0, 200);
|
|
|
|
}
|
|
|
|
|
|
|
|
walletPaths.removeDuplicates();
|
|
|
|
for(const auto &walletPath: walletPaths) {
|
|
|
|
QFile walletPathFile(walletPath);
|
|
|
|
if(walletPathFile.size() <= 0)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
QFileInfo fileInfo(walletPath);
|
|
|
|
const QString absPath = fileInfo.absoluteFilePath();
|
|
|
|
const QString path = fileInfo.path();
|
|
|
|
const QString baseName = fileInfo.baseName();
|
|
|
|
const QString basePath = QString("%1/%2").arg(path).arg(baseName);
|
|
|
|
QString addr = QString("");
|
|
|
|
quint8 networkType = NetworkType::MAINNET;
|
|
|
|
|
|
|
|
if (Utils::fileExists(basePath + ".address.txt")) {
|
|
|
|
QFile file(basePath + ".address.txt");
|
|
|
|
file.open(QFile::ReadOnly | QFile::Text);
|
|
|
|
const QString _address = QTextCodec::codecForMib(106)->toUnicode(file.readAll());
|
|
|
|
|
|
|
|
if (!_address.isEmpty()) {
|
|
|
|
addr = _address;
|
|
|
|
if (addr.startsWith("5") || addr.startsWith("7"))
|
|
|
|
networkType = NetworkType::STAGENET;
|
|
|
|
else if (addr.startsWith("9") || addr.startsWith("B"))
|
|
|
|
networkType = NetworkType::TESTNET;
|
|
|
|
}
|
|
|
|
file.close();
|
|
|
|
}
|
|
|
|
|
|
|
|
this->addWalletKeysFile(WalletKeysFiles(fileInfo, networkType, std::move(addr)));
|
|
|
|
}
|
|
|
|
|
|
|
|
auto duration = duration_cast<microseconds>(high_resolution_clock::now() - now).count();
|
|
|
|
qDebug() << QString("wallet .keys search completed in %1 ms").arg(duration);
|
|
|
|
}
|
|
|
|
|
|
|
|
void WalletKeysFilesModel::addWalletKeysFile(const WalletKeysFiles &walletKeysFile) {
|
|
|
|
beginInsertRows(QModelIndex(), rowCount(), rowCount());
|
|
|
|
m_walletKeyFiles << walletKeysFile;
|
|
|
|
endInsertRows();
|
|
|
|
}
|
|
|
|
|
|
|
|
int WalletKeysFilesModel::rowCount(const QModelIndex & parent) const {
|
|
|
|
return parent.isValid() ? 0 : m_walletKeyFiles.count();
|
|
|
|
}
|
|
|
|
|
|
|
|
int WalletKeysFilesModel::columnCount(const QModelIndex &) const {
|
|
|
|
return 3;
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariant WalletKeysFilesModel::data(const QModelIndex &index, int role) const {
|
|
|
|
if (index.row() < 0 || index.row() >= m_walletKeyFiles.count())
|
|
|
|
return QVariant();
|
|
|
|
|
|
|
|
const auto &walletKeyFile = m_walletKeyFiles[index.row()];
|
|
|
|
|
|
|
|
if (role == Qt::DisplayRole) {
|
|
|
|
switch(index.column()) {
|
|
|
|
case ModelColumns::NetworkType: {
|
|
|
|
auto c = static_cast<NetworkType::Type>(walletKeyFile.networkType());
|
|
|
|
if (c == NetworkType::Type::STAGENET)
|
|
|
|
return QString("stage");
|
|
|
|
else if (c == NetworkType::Type::TESTNET)
|
|
|
|
return QString("test");
|
|
|
|
return QString("main");
|
|
|
|
}
|
|
|
|
case ModelColumns::FileName:
|
|
|
|
return walletKeyFile.fileName();
|
|
|
|
case ModelColumns::Path: {
|
|
|
|
auto fp = walletKeyFile.path();
|
|
|
|
#if defined(Q_OS_MAC) || defined(Q_OS_LINUX)
|
|
|
|
if (fp.startsWith(m_ctx->homeDir))
|
|
|
|
fp = QString("~/%1").arg(fp.remove(0, m_ctx->homeDir.length() + 1));
|
|
|
|
#endif
|
|
|
|
return fp;
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
} else if(role == Qt::UserRole) {
|
|
|
|
switch(index.column()) {
|
|
|
|
case ModelColumns::NetworkType:
|
2020-12-18 22:39:27 +00:00
|
|
|
return static_cast<int>(walletKeyFile.networkType());
|
2020-10-07 10:36:04 +00:00
|
|
|
case ModelColumns::FileName:
|
|
|
|
return walletKeyFile.fileName();
|
|
|
|
case ModelColumns::Modified:
|
|
|
|
return (int)walletKeyFile.modified();
|
|
|
|
case ModelColumns::Path:
|
|
|
|
return walletKeyFile.path();
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
|
|
|
|
QVariant WalletKeysFilesModel::headerData(int section, Qt::Orientation orientation, int role) const {
|
|
|
|
if (role != Qt::DisplayRole)
|
|
|
|
return QVariant();
|
|
|
|
if (orientation == Qt::Horizontal) {
|
|
|
|
switch(section) {
|
|
|
|
case NetworkType:
|
|
|
|
return QString("Type");
|
|
|
|
case FileName:
|
|
|
|
return QString("Name");
|
|
|
|
case Path:
|
|
|
|
return QString("Path");
|
|
|
|
default:
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return QVariant();
|
|
|
|
}
|
|
|
|
|
2020-12-18 22:39:27 +00:00
|
|
|
WalletKeysFilesProxyModel::WalletKeysFilesProxyModel(QObject *parent, NetworkType::Type nettype)
|
|
|
|
: QSortFilterProxyModel(parent)
|
|
|
|
, m_nettype(nettype)
|
|
|
|
{
|
|
|
|
}
|
2020-10-07 10:36:04 +00:00
|
|
|
|
|
|
|
bool WalletKeysFilesProxyModel::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const {
|
2020-12-18 22:39:27 +00:00
|
|
|
QModelIndex nettypeIndex = sourceModel()->index(sourceRow, WalletKeysFilesModel::NetworkType, sourceParent);
|
|
|
|
NetworkType::Type nettype = static_cast<NetworkType::Type>(sourceModel()->data(nettypeIndex, Qt::UserRole).toInt());
|
|
|
|
|
|
|
|
return (m_nettype == nettype);
|
2020-10-07 10:36:04 +00:00
|
|
|
}
|