1 files changed, 28 insertions, 4 deletions

diff --git a/framework/src/entitymodel.h b/framework/src/entitymodel.h
index add66d78..e8c294a0 100644
--- a/framework/src/entitymodel.h
+++ b/framework/src/entitymodel.h
@@ -21,6 +21,8 @@
 #include "kube_export.h"
 #include <QSharedPointer>
 #include <QSortFilterProxyModel>
+#include <QSet>
+#include <QByteArray>
 
 namespace Sink {
     class Query;
@@ -43,13 +45,12 @@ public:
         SuccessStatus,
     };
     Q_ENUMS(Status)
-
     EntityModel(QObject *parent = Q_NULLPTR);
-    ~EntityModel();
+    virtual ~EntityModel();
 
-    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const Q_DECL_OVERRIDE;
+    virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
 
-    QHash<int, QByteArray> roleNames() const Q_DECL_OVERRIDE;
+    virtual QHash<int, QByteArray> roleNames() const override;
 
     void setAccountId(const QString &);
     QString accountId() const;
@@ -74,3 +75,26 @@ private:
     QString mAccountId;
     QString mType;
 };
+
+
+class KUBE_EXPORT CheckableEntityModel : public EntityModel {
+
+    Q_OBJECT
+
+    Q_PROPERTY (QSet<QByteArray> checkedEntities READ checkedEntities NOTIFY checkedEntitiesChanged)
+public:
+    CheckableEntityModel(QObject *parent = Q_NULLPTR);
+    virtual ~CheckableEntityModel();
+
+    QHash<int, QByteArray> roleNames() const override;
+    QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
+    bool setData(const QModelIndex &index, const QVariant &value, int role = Qt::EditRole) override;
+
+    QSet<QByteArray> checkedEntities() const;
+
+signals:
+    void checkedEntitiesChanged();
+
+private:
+    QSet<QByteArray> mCheckedEntities;
+};