github-wownero/wowlet
1
0
Fork 0
mirror of https://github.com/wownero/wowlet.git synced 2026-10-21 17:18:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Clean up WalletKeysFilesModel

Browse Source
This commit is contained in:
tobtoht
2021-05-20 21:44:40 +02:00
parent b0834f8729
commit 3d646fb345
6 changed files with 81 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/utils/keysfiles.cpp
-216
View File
@@ -1,216 +0,0 @@
// SPDX-License-Identifier: BSD-3-Clause
// Copyright (c) 2014-2021, The Monero Project.
#include "keysfiles.h"
using namespace std::chrono;
WalletKeysFiles::WalletKeysFiles(const QFileInfo &info, int networkType, QString address) :
m_fileName(info.fileName()),
m_modified(info.lastModified().toSecsSinceEpoch()),
m_path(QDir::toNativeSeparators(info.absoluteFilePath())),
m_networkType(networkType),
m_address(std::move(address))
{
QFileInfo cacheFile = QFileInfo(info.absoluteFilePath().replace(QRegExp(".keys$"), ""));
qint64 cacheLastModified = cacheFile.lastModified().toSecsSinceEpoch();
if (cacheFile.exists()) {
m_modified = (cacheLastModified > m_modified) ? cacheLastModified : m_modified;
}
}
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;
}
int WalletKeysFiles::networkType() const {
return m_networkType;
}
WalletKeysFilesModel::WalletKeysFilesModel(QObject *parent)
: QAbstractTableModel(parent)
{
this->updateDirectories();
this->m_walletKeysFilesItemModel = qobject_cast<QAbstractItemModel *>(this);
}
void WalletKeysFilesModel::clear() {
beginResetModel();
m_walletKeyFiles.clear();
endResetModel();
}
void WalletKeysFilesModel::refresh() {
this->clear();
this->findWallets();
endResetModel();
}
void WalletKeysFilesModel::updateDirectories() { // TODO
this->walletDirectories.clear();
QDir defaultWalletDir = QDir(Utils::defaultWalletDir());
QString walletDir = defaultWalletDir.path();
defaultWalletDir.cdUp();
QString walletDirRoot = defaultWalletDir.path();
this->walletDirectories << walletDir;
this->walletDirectories << walletDirRoot;
this->walletDirectories << QDir::homePath();
this->walletDirectories.removeDuplicates();
}
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<milliseconds>(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 4;
}
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().replace(".keys", "");
case ModelColumns::Path: {
auto fp = walletKeyFile.path();
#if defined(Q_OS_MAC) || defined(Q_OS_LINUX)
if (fp.startsWith(QDir::homePath())) {
fp = QString("~/%1").arg(fp.remove(0, QDir::homePath().length() + 1));
}
#endif
return fp;
}
case ModelColumns::Modified:
return walletKeyFile.modified();
default:
break;
}
} else if(role == Qt::UserRole) {
switch(index.column()) {
case ModelColumns::NetworkType:
return static_cast<int>(walletKeyFile.networkType());
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();
}
WalletKeysFilesProxyModel::WalletKeysFilesProxyModel(QObject *parent, NetworkType::Type nettype)
: QSortFilterProxyModel(parent)
, m_nettype(nettype)
{
}
bool WalletKeysFilesProxyModel::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const {
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);
}
src/utils/keysfiles.h
-74
View File
@@ -1,74 +0,0 @@
// SPDX-License-Identifier: BSD-3-Clause
// Copyright (c) 2014-2021, The Monero Project.
#ifndef KEYSFILES_H
#define KEYSFILES_H
#include "libwalletqt/WalletManager.h"
#include "utils/networktype.h"
#include "utils/utils.h"
class WalletKeysFiles
{
public:
WalletKeysFiles(const QFileInfo &info, int networkType, QString address);
QString fileName() const;
qint64 modified() const;
QString path() const;
int networkType() const;
QString address() const;
private:
QString m_fileName;
qint64 m_modified;
QString m_path;
int m_networkType;
QString m_address;
};
class WalletKeysFilesModel : public QAbstractTableModel
{
Q_OBJECT
public:
enum ModelColumns {
NetworkType = 0,
FileName,
Path,
Modified
};
explicit WalletKeysFilesModel(QObject *parent = nullptr);
Q_INVOKABLE void refresh();
Q_INVOKABLE void clear();
void findWallets();
void addWalletKeysFile(const WalletKeysFiles &walletKeysFile);
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
int columnCount(const QModelIndex &parent = QModelIndex()) const override;
QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
QStringList walletDirectories;
private:
void updateDirectories();
QList<WalletKeysFiles> m_walletKeyFiles;
QAbstractItemModel *m_walletKeysFilesItemModel;
QSortFilterProxyModel m_walletKeysFilesModelProxy;
};
class WalletKeysFilesProxyModel : public QSortFilterProxyModel
{
Q_OBJECT
public:
explicit WalletKeysFilesProxyModel(QObject *parent, NetworkType::Type nettype);
bool filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const override;
private:
NetworkType::Type m_nettype;
};
#endif // KEYSFILES_H