From 8f014f0915c7bdc97573f953aa9a3ae25a5f953a Mon Sep 17 00:00:00 2001
From: Andrew Branson <andrew.branson@cern.ch>
Date: Thu, 18 Feb 2016 09:11:16 +0100
Subject: Rockwork 1.0

---
 rockwork/notificationsourcemodel.cpp | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'rockwork/notificationsourcemodel.cpp')

diff --git a/rockwork/notificationsourcemodel.cpp b/rockwork/notificationsourcemodel.cpp
index cbb75ca..83e87f0 100644
--- a/rockwork/notificationsourcemodel.cpp
+++ b/rockwork/notificationsourcemodel.cpp
@@ -1,5 +1,8 @@
 #include "notificationsourcemodel.h"
 
+#include <QStandardPaths>
+#include <QFileInfo>
+#include <QDir>
 #include <QSettings>
 #include <QDebug>
 
@@ -21,6 +24,8 @@ QVariant NotificationSourceModel::data(const QModelIndex &index, int role) const
         return item.m_displayName;
     case RoleEnabled:
         return item.m_enabled;
+    case RoleId:
+        return item.m_id;
     case RoleIcon:
         return item.m_icon;
     }
@@ -33,6 +38,7 @@ QHash<int, QByteArray> NotificationSourceModel::roleNames() const
     roles.insert(RoleName, "name");
     roles.insert(RoleEnabled, "enabled");
     roles.insert(RoleIcon, "icon");
+    roles.insert(RoleId, "id");
     return roles;
 }
 
@@ -59,10 +65,6 @@ void NotificationSourceModel::insert(const QString &sourceId, bool enabled)
     }
 }
 
-#include <QStandardPaths>
-#include <QFileInfo>
-#include <QDir>
-
 NotificationSourceItem NotificationSourceModel::fromDesktopFile(const QString &sourceId)
 {
     NotificationSourceItem ret;
-- 
cgit v1.2.3