From f6c9380e9b0378eb1d80084ceaa902ea1327a752 Mon Sep 17 00:00:00 2001
From: 2 * r + 2 * t <61896496+soramanew@users.noreply.github.com>
Date: Sun, 31 Aug 2025 23:36:06 +1000
Subject: plugin/fsm: add new roles

---
 plugin/src/Caelestia/filesystemmodel.cpp | 8 +++++++-
 plugin/src/Caelestia/filesystemmodel.hpp | 4 +++-
 2 files changed, 10 insertions(+), 2 deletions(-)

(limited to 'plugin')

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 {
-- 
cgit v1.2.3-freya