Feather, a free Monero desktop wallet

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

src/model/SubaddressModel.cpp

@@ -0,0 +1,168 @@
+// SPDX-License-Identifier: BSD-3-Clause
+// Copyright (c) 2014-2020, The Monero Project.
+
+#include "SubaddressModel.h"
+#include "Subaddress.h"
+#include "ModelUtils.h"
+#include <wallet/api/wallet2_api.h>
+
+#include <QHash>
+#include <QPoint>
+#include <QColor>
+#include <QBrush>
+#include <QFont>
+
+SubaddressModel::SubaddressModel(QObject *parent, Subaddress *subaddress)
+    : QAbstractTableModel(parent),
+    m_subaddress(subaddress),
+    m_showFullAddresses(false)
+{
+    connect(m_subaddress, &Subaddress::refreshStarted, this, &SubaddressModel::startReset);
+    connect(m_subaddress, &Subaddress::refreshFinished, this, &SubaddressModel::endReset);
+}
+
+void SubaddressModel::startReset(){
+    beginResetModel();
+}
+
+void SubaddressModel::endReset(){
+    endResetModel();
+}
+
+int SubaddressModel::rowCount(const QModelIndex &parent) const
+{
+    if (parent.isValid()) {
+        return 0;
+    } else {
+        return m_subaddress->count();
+    }
+}
+
+int SubaddressModel::columnCount(const QModelIndex &parent) const
+{
+    if (parent.isValid()) {
+        return 0;
+    }
+    return ModelColumn::COUNT;
+}
+
+QVariant SubaddressModel::data(const QModelIndex &index, int role) const
+{
+    if (!index.isValid() || index.row() < 0 || static_cast<quint64>(index.row()) >= m_subaddress->count())
+        return QVariant();
+
+    QVariant result;
+
+    bool found = m_subaddress->getRow(index.row(), [this, &index, &role, &result](const Monero::SubaddressRow &subaddress) {
+        if (role == Qt::DisplayRole || role == Qt::EditRole || role == Qt::UserRole){
+            result = parseSubaddressRow(subaddress, index, role);
+        }
+        else if (role == Qt::BackgroundRole) {
+            if (subaddress.isUsed()) {
+                result = QBrush(QColor(255,100,100));
+            }
+        }
+        else if (role == Qt::FontRole) {
+            switch(index.column()) {
+                case Address:
+                {
+                   result = ModelUtils::getMonospaceFont();
+                }
+            }
+        }
+    });
+
+    if (!found)
+    {
+        qCritical("%s: internal error: invalid index %d", __FUNCTION__, index.row());
+    }
+
+    return result;
+}
+
+QVariant SubaddressModel::parseSubaddressRow(const Monero::SubaddressRow &subaddress, const QModelIndex &index, int role) const
+{
+    switch (index.column()) {
+        case Address:
+        {
+            QString address = QString::fromStdString(subaddress.getAddress());
+            if (!m_showFullAddresses && role != Qt::UserRole) {
+                address = ModelUtils::displayAddress(address);
+            }
+            return address;
+        }
+        case Label:
+            return index.row() == 0 ? tr("Primary address") : QString::fromStdString(subaddress.getLabel());
+        case isUsed:
+            return subaddress.isUsed();
+        default:
+            qCritical() << "Invalid column" << index.column();
+            return QVariant();
+    }
+}
+
+
+QVariant SubaddressModel::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 Label:
+                return QString("Label");
+            case isUsed:
+                return QString("Used");
+            default:
+                return QVariant();
+        }
+    }
+    return QVariant();
+}
+
+bool SubaddressModel::setData(const QModelIndex &index, const QVariant &value, int role)
+{
+    if (index.isValid() && role == Qt::EditRole) {
+        const int row = index.row();
+
+        switch (index.column()) {
+            case Label:
+                m_subaddress->setLabel(0, row, value.toString()); // Todo: get actual account index
+                break;
+            default:
+                return false;
+        }
+        emit dataChanged(index, index, {Qt::DisplayRole, Qt::EditRole});
+
+        return true;
+    }
+    return false;
+}
+
+void SubaddressModel::setShowFullAddresses(bool show)
+{
+    m_showFullAddresses = show;
+    emit dataChanged(index(0, 0), index(rowCount() - 1, columnCount() - 1));
+}
+
+Qt::ItemFlags SubaddressModel::flags(const QModelIndex &index) const
+{
+    if (!index.isValid())
+        return Qt::ItemIsEnabled;
+
+    if (index.column() == Label && index.row() != 0)
+        return QAbstractTableModel::flags(index) | Qt::ItemIsEditable;
+
+    return QAbstractTableModel::flags(index);
+}
+
+bool SubaddressModel::isShowFullAddresses() const {
+    return m_showFullAddresses;
+}
+
+int SubaddressModel::unusedLookahead() const {
+    return m_subaddress->unusedLookahead();
+}