mirror of
https://git.wownero.com/wowlet/wowlet.git
synced 2024-08-15 01:03:14 +00:00
177 lines
5 KiB
C++
177 lines
5 KiB
C++
|
// SPDX-License-Identifier: BSD-3-Clause
|
||
|
// Copyright (c) 2014-2020, The Monero Project.
|
||
|
|
||
|
#include "AddressBookModel.h"
|
||
|
#include "AddressBook.h"
|
||
|
#include "ModelUtils.h"
|
||
|
#include "utils/utils.h"
|
||
|
#include <wallet/api/wallet2_api.h>
|
||
|
|
||
|
#include <QDebug>
|
||
|
#include <QFileDialog>
|
||
|
#include <QDir>
|
||
|
#include <QFontDatabase>
|
||
|
|
||
|
AddressBookModel::AddressBookModel(QObject *parent, AddressBook *addressBook)
|
||
|
: QAbstractTableModel(parent),
|
||
|
m_addressBook(addressBook),
|
||
|
m_showFullAddresses(false)
|
||
|
{
|
||
|
connect(m_addressBook, &AddressBook::refreshStarted, this, &AddressBookModel::startReset);
|
||
|
connect(m_addressBook, &AddressBook::refreshFinished, this, &AddressBookModel::endReset);
|
||
|
m_contactIcon = QIcon(":/assets/images/person.svg");
|
||
|
}
|
||
|
|
||
|
void AddressBookModel::startReset(){
|
||
|
beginResetModel();
|
||
|
}
|
||
|
|
||
|
void AddressBookModel::endReset(){
|
||
|
endResetModel();
|
||
|
}
|
||
|
|
||
|
int AddressBookModel::rowCount(const QModelIndex &) const
|
||
|
{
|
||
|
return m_addressBook->count();
|
||
|
}
|
||
|
|
||
|
int AddressBookModel::columnCount(const QModelIndex &parent) const
|
||
|
{
|
||
|
if (parent.isValid()) {
|
||
|
return 0;
|
||
|
}
|
||
|
return COUNT;
|
||
|
}
|
||
|
|
||
|
bool AddressBookModel::isShowFullAddresses() const {
|
||
|
return m_showFullAddresses;
|
||
|
}
|
||
|
|
||
|
void AddressBookModel::setShowFullAddresses(bool show)
|
||
|
{
|
||
|
m_showFullAddresses = show;
|
||
|
emit dataChanged(index(0, 0), index(rowCount() - 1, columnCount() - 1));
|
||
|
}
|
||
|
|
||
|
bool AddressBookModel::setData(const QModelIndex &index, const QVariant &value, int role)
|
||
|
{
|
||
|
if (index.isValid() && role == Qt::EditRole) {
|
||
|
const int row = index.row();
|
||
|
|
||
|
switch (index.column()) {
|
||
|
case Description:
|
||
|
m_addressBook->setDescription(row, value.toString());
|
||
|
break;
|
||
|
default:
|
||
|
return false;
|
||
|
}
|
||
|
emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
QVariant AddressBookModel::data(const QModelIndex &index, int role) const
|
||
|
{
|
||
|
QVariant result;
|
||
|
|
||
|
bool found = m_addressBook->getRow(index.row(), [this, &result, &role, &index](const AddressBookInfo &row) {
|
||
|
if (role == Qt::DisplayRole || role == Qt::EditRole || role == Qt::UserRole) {
|
||
|
switch (index.column()) {
|
||
|
case Address:
|
||
|
{
|
||
|
QString address = row.address();
|
||
|
if (!m_showFullAddresses && role != Qt::UserRole) {
|
||
|
address = ModelUtils::displayAddress(address);
|
||
|
}
|
||
|
result = address;
|
||
|
break;
|
||
|
}
|
||
|
case Description:
|
||
|
result = row.description();
|
||
|
break;
|
||
|
default:
|
||
|
qCritical() << "Invalid column" << index.column();
|
||
|
}
|
||
|
}
|
||
|
else if (role == Qt::FontRole) {
|
||
|
switch (index.column()) {
|
||
|
case Address:
|
||
|
result = ModelUtils::getMonospaceFont();
|
||
|
}
|
||
|
}
|
||
|
else if (role == Qt::DecorationRole) {
|
||
|
switch (index.column()) {
|
||
|
case Description: {
|
||
|
return QVariant(m_contactIcon); // @TODO: does not actually work
|
||
|
}
|
||
|
default: {
|
||
|
return QVariant();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
return QVariant();
|
||
|
});
|
||
|
if (!found) {
|
||
|
qCritical("%s: internal error: invalid index %d", __FUNCTION__, index.row());
|
||
|
}
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
Qt::ItemFlags AddressBookModel::flags(const QModelIndex &index) const
|
||
|
{
|
||
|
if (!index.isValid())
|
||
|
return Qt::ItemIsEnabled;
|
||
|
|
||
|
if (index.column() == Description)
|
||
|
return QAbstractTableModel::flags(index) | Qt::ItemIsEditable;
|
||
|
|
||
|
return QAbstractTableModel::flags(index);
|
||
|
}
|
||
|
|
||
|
QVariant AddressBookModel::headerData(int section, Qt::Orientation orientation, int role) const
|
||
|
{
|
||
|
if (role != Qt::DisplayRole) {
|
||
|
return QVariant();
|
||
|
}
|
||
|
if (orientation == Qt::Horizontal)
|
||
|
{
|
||
|
switch(section) {
|
||
|
case Address:
|
||
|
return QString("Address");
|
||
|
case Description:
|
||
|
return QString("Name");
|
||
|
default:
|
||
|
return QVariant();
|
||
|
}
|
||
|
}
|
||
|
return QVariant();
|
||
|
}
|
||
|
|
||
|
bool AddressBookModel::deleteRow(int row)
|
||
|
{
|
||
|
return m_addressBook->deleteRow(row);
|
||
|
}
|
||
|
|
||
|
int AddressBookModel::lookupPaymentID(const QString &payment_id) const
|
||
|
{
|
||
|
return m_addressBook->lookupPaymentID(payment_id);
|
||
|
}
|
||
|
|
||
|
bool AddressBookModel::writeCSV(const QString &path) {
|
||
|
QString csv = "";
|
||
|
for(int i = 0; i < this->rowCount(); i++) {
|
||
|
QModelIndex index = this->index(i, 0);
|
||
|
const auto description = this->data(index.siblingAtColumn(AddressBookModel::Description), Qt::UserRole).toString().replace("\"", "");
|
||
|
const auto address = this->data(index.siblingAtColumn(AddressBookModel::Address), Qt::UserRole).toString();
|
||
|
if(address.isEmpty()) continue;
|
||
|
csv += QString("%1,\"%2\"\n").arg(address).arg(description);
|
||
|
}
|
||
|
if(csv.isEmpty())
|
||
|
return false;
|
||
|
csv = QString("address,description\n%1").arg(csv);
|
||
|
return Utils::fileWrite(path, csv);
|
||
|
}
|