Add lock to encrypted messages

2 files changed, 7 insertions, 0 deletions

diff --git a/src/timeline2/TimelineModel.cpp b/src/timeline2/TimelineModel.cpp

index 9537649b..36b768ba 100644
--- a/src/timeline2/TimelineModel.cpp
+++ b/src/timeline2/TimelineModel.cpp
@@ -293,6 +293,7 @@ TimelineModel::roleNames() const
           {ProportionalHeight, "proportionalHeight"},
           {Id, "id"},
           {State, "state"},
+          {IsEncrypted, "isEncrypted"},
         };
 }
 int
@@ -391,6 +392,11 @@ TimelineModel::data(const QModelIndex &index, int role) const
                         return qml_mtx_events::Read;
                 else
                         return qml_mtx_events::Received;
+        case IsEncrypted: {
+                auto tempEvent = events[id];
+                return boost::get<mtx::events::EncryptedEvent<mtx::events::msg::Encrypted>>(
+                         &tempEvent) != nullptr;
+        }
         default:
                 return QVariant();
         }
diff --git a/src/timeline2/TimelineModel.h b/src/timeline2/TimelineModel.h

index 7723ef66..3d55f206 100644
--- a/src/timeline2/TimelineModel.h
+++ b/src/timeline2/TimelineModel.h
@@ -132,6 +132,7 @@ public:
                 ProportionalHeight,
                 Id,
                 State,
+                IsEncrypted,
         };
 
         QHash<int, QByteArray> roleNames() const override;