Merge pull request #518 from Smittyvb/room-stateEvents

2 files changed, 35 insertions, 0 deletions

diff --git a/lib/room.cpp b/lib/room.cpp
index 284eacd1..fac24e5e 100644
--- a/lib/room.cpp
+++ b/lib/room.cpp
@@ -226,6 +226,17 @@ public:
         return evt;
     }
 
+    QVector<const StateEventBase*> stateEventsOfType(const QString& evtType) const
+    {
+        QVector<const StateEventBase*> vals = QVector<const StateEventBase*>();
+        for (const auto* val : currentState) {
+            if (val->matrixType() == evtType) {
+                vals.append(val);
+            }
+        }
+        return vals;
+    }
+
     template <typename EventT>
     const EventT* getCurrentState(const QString& stateKey = {}) const
     {
@@ -1455,6 +1466,17 @@ const StateEventBase* Room::getCurrentState(const QString& evtType,
     return d->getCurrentState({ evtType, stateKey });
 }
 
+const QVector<const StateEventBase*>
+Room::stateEventsOfType(const QString& evtType) const
+{
+    return d->stateEventsOfType(evtType);
+}
+
+const QHash<StateEventKey, const StateEventBase*>& Room::currentState() const
+{
+    return d->currentState;
+}
+
 RoomEventPtr Room::decryptMessage(const EncryptedEvent& encryptedEvent)
 {
 #ifndef Quotient_E2EE_ENABLED
diff --git a/lib/room.h b/lib/room.h
index 65217290..85c51a87 100644
--- a/lib/room.h
+++ b/lib/room.h
@@ -757,6 +757,19 @@ public:
     Q_INVOKABLE const Quotient::StateEventBase*
     getCurrentState(const QString& evtType, const QString& stateKey = {}) const;
 
+    /// Get all state events in the room.
+    /*! This method returns all known state events that have occured in
+     * the room, as a mapping from the event type and state key to value.
+     */
+    const QHash<StateEventKey, const StateEventBase*>& currentState() const;
+
+    /// Get all state events in the room of a certain type.
+    /*! This method returns all known state events that have occured in
+     * the room of the given type.
+     */
+    Q_INVOKABLE const QVector<const StateEventBase*>
+    stateEventsOfType(const QString& evtType) const;
+
     /// Get a state event with the given event type and state key
     /*! This is a typesafe overload that accepts a C++ event type instead of
      * its Matrix name.