Finish implementation of address book

2 files changed, 14 insertions, 2 deletions

diff --git a/gui/include/addresstablemodel.h b/gui/include/addresstablemodel.h
index f9ccab4eb6..1890260971 100644
--- a/gui/include/addresstablemodel.h
+++ b/gui/include/addresstablemodel.h
@@ -13,10 +13,10 @@ public:
     explicit AddressTableModel(QObject *parent = 0);
     ~AddressTableModel();
 
-    enum {
+    enum ColumnIndex {
         Label = 0,   /* User specified label */
         Address = 1  /* Bitcoin address */
-    } ColumnIndex;
+    };
 
     enum {
         TypeRole = Qt::UserRole
@@ -25,13 +25,22 @@ public:
     static const QString Send; /* Send addres */
     static const QString Receive; /* Receive address */
 
+    /* Overridden methods from QAbstractTableModel */
     int rowCount(const QModelIndex &parent) const;
     int columnCount(const QModelIndex &parent) const;
     QVariant data(const QModelIndex &index, int role) const;
     bool setData(const QModelIndex & index, const QVariant & value, int role);
     QVariant headerData(int section, Qt::Orientation orientation, int role) const;
     QModelIndex index(int row, int column, const QModelIndex & parent) const;
+    bool removeRows(int row, int count, const QModelIndex & parent = QModelIndex());
+
+    /* Add an address to the model.
+       Returns true on success, false otherwise.
+     */
+    bool addRow(const QString &type, const QString &label, const QString &address);
 
+    /* Update address list from core. Invalidates any indices.
+     */
     void updateList();
 private:
     AddressTablePriv *priv;
diff --git a/gui/include/editaddressdialog.h b/gui/include/editaddressdialog.h
index 3d8a5dcf8d..dd7766951b 100644
--- a/gui/include/editaddressdialog.h
+++ b/gui/include/editaddressdialog.h
@@ -29,10 +29,13 @@ public:
 
     void setModel(AddressTableModel *model);
     void loadRow(int row);
+    void saveCurrentRow();
 
 private:
     Ui::EditAddressDialog *ui;
     QDataWidgetMapper *mapper;
+    Mode mode;
+    AddressTableModel *model;
 };
 
 #endif // EDITADDRESSDIALOG_H