Feather, a free Monero desktop wallet

Co-Authored-By: tobtoht <thotbot@protonmail.com>

src/model/AddressBookModel.cpp

@@ -0,0 +1,176 @@
+// 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);
+}