2 files changed, 102 insertions, 0 deletions
diff --git a/src/model/SubaddressAccountModel.cpp b/src/model/SubaddressAccountModel.cpp
new file mode 100644
index 00000000..fa2ce0b2
--- /dev/null
+++ b/src/model/SubaddressAccountModel.cpp
@@ -0,0 +1,66 @@
+#include "SubaddressAccountModel.h"
+#include "SubaddressAccount.h"
+#include <QDebug>
+#include <QHash>
+#include <wallet/api/wallet2_api.h>
+
+SubaddressAccountModel::SubaddressAccountModel(QObject *parent, SubaddressAccount *subaddressAccount)
+    : QAbstractListModel(parent), m_subaddressAccount(subaddressAccount)
+{
+    qDebug(__FUNCTION__);
+    connect(m_subaddressAccount,SIGNAL(refreshStarted()),this,SLOT(startReset()));
+    connect(m_subaddressAccount,SIGNAL(refreshFinished()),this,SLOT(endReset()));
+}
+
+void SubaddressAccountModel::startReset(){
+    qDebug("SubaddressAccountModel::startReset");
+    beginResetModel();
+}
+void SubaddressAccountModel::endReset(){
+    qDebug("SubaddressAccountModel::endReset");
+    endResetModel();
+}
+
+int SubaddressAccountModel::rowCount(const QModelIndex &parent) const
+{
+    return m_subaddressAccount->count();
+}
+
+QVariant SubaddressAccountModel::data(const QModelIndex &index, int role) const
+{
+    if (!index.isValid() || index.row() < 0 || (unsigned)index.row() >= m_subaddressAccount->count())
+        return {};
+
+    Monero::SubaddressAccountRow * sr = m_subaddressAccount->getRow(index.row());
+
+    QVariant result = "";
+    switch (role) {
+    case SubaddressAccountAddressRole:
+        result = QString::fromStdString(sr->getAddress());
+        break;
+    case SubaddressAccountLabelRole:
+        result = QString::fromStdString(sr->getLabel());
+        break;
+    case SubaddressAccountBalanceRole:
+        result = QString::fromStdString(sr->getBalance());
+        break;
+    case SubaddressAccountUnlockedBalanceRole:
+        result = QString::fromStdString(sr->getUnlockedBalance());
+        break;
+    }
+
+    return result;
+}
+
+QHash<int, QByteArray> SubaddressAccountModel::roleNames() const
+{
+    static QHash<int, QByteArray> roleNames;
+    if (roleNames.empty())
+    {
+        roleNames.insert(SubaddressAccountAddressRole, "address");
+        roleNames.insert(SubaddressAccountLabelRole, "label");
+        roleNames.insert(SubaddressAccountBalanceRole, "balance");
+        roleNames.insert(SubaddressAccountUnlockedBalanceRole, "unlockedBalance");
+    }
+    return roleNames;
+}
diff --git a/src/model/SubaddressAccountModel.h b/src/model/SubaddressAccountModel.h
new file mode 100644
index 00000000..e78a3f84
--- /dev/null
+++ b/src/model/SubaddressAccountModel.h
@@ -0,0 +1,36 @@
+#ifndef SUBADDRESSACCOUNTMODEL_H
+#define SUBADDRESSACCOUNTMODEL_H
+
+#include <QAbstractListModel>
+
+class SubaddressAccount;
+
+class SubaddressAccountModel : public QAbstractListModel
+{
+    Q_OBJECT
+
+public:
+    enum SubaddressAccountRowRole {
+        SubaddressAccountRole = Qt::UserRole + 1, // for the SubaddressAccountRow object;
+        SubaddressAccountAddressRole,
+        SubaddressAccountLabelRole,
+        SubaddressAccountBalanceRole,
+        SubaddressAccountUnlockedBalanceRole,
+    };
+    Q_ENUM(SubaddressAccountRowRole)
+
+    SubaddressAccountModel(QObject *parent, SubaddressAccount *subaddressAccount);
+
+    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
+    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
+    QHash<int, QByteArray> roleNames() const  override;
+
+public slots:
+    void startReset();
+    void endReset();
+
+private:
+    SubaddressAccount *m_subaddressAccount;
+};
+
+#endif // SUBADDRESSACCOUNTMODEL_H