2 files changed, 113 insertions, 0 deletions
diff --git a/src/model/AddressBookModel.cpp b/src/model/AddressBookModel.cpp
new file mode 100644
index 00000000..5e78200b
--- /dev/null
+++ b/src/model/AddressBookModel.cpp
@@ -0,0 +1,75 @@
+#include "AddressBookModel.h"
+#include "AddressBook.h"
+#include <QDebug>
+#include <QHash>
+#include <wallet/wallet2_api.h>
+
+AddressBookModel::AddressBookModel(QObject *parent, AddressBook *addressBook)
+    : QAbstractListModel(parent) , m_addressBook(addressBook)
+{
+    qDebug(__FUNCTION__);
+    connect(m_addressBook,SIGNAL(refreshStarted()),this,SLOT(startReset()));
+    connect(m_addressBook,SIGNAL(refreshFinished()),this,SLOT(endReset()));
+
+}
+
+void AddressBookModel::startReset(){
+    qDebug(__FUNCTION__);
+    beginResetModel();
+}
+void AddressBookModel::endReset(){
+    qDebug(__FUNCTION__);
+    endResetModel();
+}
+
+int AddressBookModel::rowCount(const QModelIndex &parent) const
+{
+    return m_addressBook->count();
+}
+
+QVariant AddressBookModel::data(const QModelIndex &index, int role) const
+{
+    if (!index.isValid())
+        return QVariant();
+
+    if (index.row() < 0 || (unsigned)index.row() >= m_addressBook->count()) {
+        return QVariant();
+    }
+
+    Bitmonero::AddressBookRow * ar = m_addressBook->getRow(index.row());
+
+    QVariant result = "";
+    switch (role) {
+    case AddressBookAddressRole:
+        result = QString::fromStdString(ar->getAddress());
+        break;
+    case AddressBookDescriptionRole:
+        result = QString::fromStdString(ar->getDescription());
+        break;
+    case AddressBookPaymentIdRole:
+        result = QString::fromStdString(ar->getPaymentId());
+        break;
+    case AddressBookRowIdRole:
+        result = ar->getRowId();
+        break;
+    }
+
+    return result;
+}
+
+bool AddressBookModel::deleteRow(int row)
+{
+    m_addressBook->deleteRow(row);
+}
+
+QHash<int, QByteArray> AddressBookModel::roleNames() const
+{
+    QHash<int, QByteArray> roleNames = QAbstractListModel::roleNames();
+    roleNames.insert(AddressBookAddressRole, "address");
+    roleNames.insert(AddressBookPaymentIdRole, "paymentId");
+    roleNames.insert(AddressBookDescriptionRole, "description");
+    roleNames.insert(AddressBookRowIdRole, "rowId");
+
+
+    return roleNames;
+}
diff --git a/src/model/AddressBookModel.h b/src/model/AddressBookModel.h
new file mode 100644
index 00000000..d6fd01e8
--- /dev/null
+++ b/src/model/AddressBookModel.h
@@ -0,0 +1,38 @@
+#ifndef ADDRESSBOOKMODEL_H
+#define ADDRESSBOOKMODEL_H
+
+#include <QAbstractListModel>
+//#include "wallet/wallet2_api.h" // we need to have an access to the Bitmonero::Wallet::AddressBook
+
+class AddressBook;
+
+class AddressBookModel : public QAbstractListModel
+{
+    Q_OBJECT
+
+public:
+    enum AddressBookRowRole {
+        AddressBookRole = Qt::UserRole + 1, // for the AddressBookRow object;
+        AddressBookAddressRole,
+        AddressBookDescriptionRole,
+        AddressBookPaymentIdRole,
+        AddressBookRowIdRole,
+    };
+    Q_ENUM(AddressBookRowRole)
+
+    AddressBookModel(QObject *parent, AddressBook * addressBook);
+
+    int rowCount(const QModelIndex &parent = QModelIndex()) const override;
+    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
+    Q_INVOKABLE bool deleteRow(int row);
+    virtual QHash<int, QByteArray> roleNames() const  override;
+
+public slots:
+    void startReset();
+    void endReset();
+
+private:
+    AddressBook * m_addressBook;
+};
+
+#endif // ADDRESSBOOKMODEL_H