contracttablemodel.h

Go to the documentation of this file.

#ifndef CONTRACTTABLEMODEL_H
#define CONTRACTTABLEMODEL_H

#include <QAbstractTableModel>
#include <QStringList>

class ContractTablePriv;
class WalletModel;

class CWallet;

class ContractTableModel : public QAbstractTableModel
{
    Q_OBJECT

public:
    explicit ContractTableModel(CWallet *wallet, WalletModel *parent = 0);
    ~ContractTableModel();

    enum ColumnIndex {
        Label = 0,   
        Address = 1, 
        ABI = 2      
    };

    enum EditStatus {
        OK = 0,                     
        NO_CHANGES,             
        DUPLICATE_ADDRESS,      
    };

    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 the added address on success, and an empty string otherwise.
     */
    QString addRow(const QString &label, const QString &address, const QString &abi);

    /* Label for address in contract book, if not found return empty string.
     */
    QString labelForAddress(const QString &address) const;

    /* ABI for address in contract book, if not found return empty string.
     */
    QString abiForAddress(const QString &address) const;

    /* Look up row index of an address in the model.
       Return -1 if not found.
     */
    int lookupAddress(const QString &address) const;

    EditStatus getEditStatus() const { return editStatus; }

    void resetEditStatus();

private:
    WalletModel *walletModel;
    CWallet *wallet;
    ContractTablePriv *priv;
    QStringList columns;
    EditStatus editStatus;

    void emitDataChanged(int index);

    void updateEditStatus(EditStatus status);

public Q_SLOTS:
    /* Update address list from core.
     */
    void updateEntry(const QString &address, const QString &label, const QString &abi, int status);

    friend class ContractTablePriv;
};

#endif // CONTRACTTABLEMODEL_H