2 files changed, 10 insertions, 2 deletions
diff --git a/plugin/src/Caelestia/filesystemmodel.cpp b/plugin/src/Caelestia/filesystemmodel.cpp
index 7cd2adc..8722cc0 100644
--- a/plugin/src/Caelestia/filesystemmodel.cpp
+++ b/plugin/src/Caelestia/filesystemmodel.cpp
@@ -18,7 +18,7 @@ QVariant FileSystemModel::data(const QModelIndex& index, int role) const {
         return QVariant();
     }
 
-    const FileSystemEntry* file = m_entries.at(index.row());
+    FileSystemEntry* file = m_entries.at(index.row());
     switch (role) {
     case FilePathRole:
         return file->path();
@@ -30,6 +30,10 @@ QVariant FileSystemModel::data(const QModelIndex& index, int role) const {
         return file->parentDir();
     case FileSizeRole:
         return file->size();
+    case FileIsDirRole:
+        return file->isDir();
+    case FileIsImageRole:
+        return file->isImage();
     default:
         return QVariant();
     }
@@ -42,6 +46,8 @@ QHash<int, QByteArray> FileSystemModel::roleNames() const {
     roles[FileNameRole] = "fileName";
     roles[ParentDirRole] = "parentDir";
     roles[FileSizeRole] = "fileSize";
+    roles[FileIsDirRole] = "fileIsDir";
+    roles[FileIsImageRole] = "fileIsImage";
     return roles;
 }
 
diff --git a/plugin/src/Caelestia/filesystemmodel.hpp b/plugin/src/Caelestia/filesystemmodel.hpp
index 10fce1d..42027e7 100644
--- a/plugin/src/Caelestia/filesystemmodel.hpp
+++ b/plugin/src/Caelestia/filesystemmodel.hpp
@@ -68,7 +68,9 @@ public:
         RelativeFilePathRole,
         FileNameRole,
         ParentDirRole,
-        FileSizeRole
+        FileSizeRole,
+        FileIsDirRole,
+        FileIsImageRole
     };
 
     enum Filter {