@@ -30,6 +30,7 @@ Categories::Categories(QObject* parent)
3030 m_roles[Categories::RoleName] = "name";
3131 m_roles[Categories::RoleIcon] = "icon";
3232 m_roles[Categories::RoleRenderer] = "renderer";
33+ m_roles[Categories::RoleContentType] = "content_type";
3334 m_roles[Categories::RoleHints] = "hints";
3435 m_roles[Categories::RoleResults] = "results";
3536 m_roles[Categories::RoleCount] = "count";
@@ -111,22 +112,31 @@ Categories::data(const QModelIndex& index, int role) const
111112 return QVariant();
112113 }
113114
114− if (role == RoleId) {
115+ switch (role) {
116+ case RoleId:
115117 return QVariant::fromValue(index.row());
116− } else if (role == RoleName) {
118+ case RoleName:
117119 return QVariant::fromValue(DeeListModel::data(index, 1)); //DISPLAY_NAME
118− } else if (role == RoleIcon) {
120+ case RoleIcon:
119121 return QVariant::fromValue(DeeListModel::data(index, 2)); //ICON_HINT
120− } else if (role == RoleRenderer) {
122+ case RoleRenderer:
121123 return QVariant::fromValue(DeeListModel::data(index, 3)); //RENDERER_NAME
122− } else if (role == RoleHints) {
124+ case RoleContentType:
125+ {
126+ auto hints = QVariant::fromValue(DeeListModel::data(index, 4)).toHash();
127+ return hints.contains("content-type") ?
128+ hints["content-type"] : QVariant(QString("default"));
129+ }
130+ case RoleHints:
123131 return QVariant::fromValue(DeeListModel::data(index, 4)); //HINTS
124− } else if (role == RoleResults) {
132+ case RoleResults:
125133 return QVariant::fromValue(getFilter(index.row()));
126− } else if (role == RoleCount) {
134+ case RoleCount:
135+ {
127136 CategoryFilter* filter = getFilter(index.row());
128137 return QVariant::fromValue(filter->rowCount());
129− } else {
138+ }
139+ default:
130140 return QVariant();
131141 }
132142}