wowlet/src/model/AddressBookModel.cpp

194 lines
5.6 KiB
C++
Raw Normal View History

// SPDX-License-Identifier: BSD-3-Clause
2020-12-26 19:56:06 +00:00
// Copyright (c) 2014-2021, The Monero Project.
#include "AddressBookModel.h"
#include "AddressBook.h"
#include "ModelUtils.h"
#include "utils/utils.h"
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);
}
2020-10-21 06:25:02 +00:00
QMap<QString, QString> AddressBookModel::readCSV(const QString &path) {
if(!Utils::fileExists(path)) {
return QMap<QString, QString>();
}
QString csv = Utils::barrayToString(Utils::fileOpen(path));
QTextStream stream(&csv);
QMap<QString, QString> map;
2020-10-21 06:25:02 +00:00
while(!stream.atEnd()) {
QStringList line = stream.readLine().split(",");
if(line.length() != 2) {
continue;
}
QString address = line.at(0);
QString description = line.at(1);
description = description.replace("\"", "");
if(!description.isEmpty() && !address.isEmpty()) {
map[description] = address;
2020-10-21 06:25:02 +00:00
}
}
return map;
}