Store encryptedevent in decrypted roomevents

2 files changed, 17 insertions, 0 deletions

diff --git a/lib/events/roomevent.cpp b/lib/events/roomevent.cpp
index fb921af6..b99d1381 100644
--- a/lib/events/roomevent.cpp
+++ b/lib/events/roomevent.cpp
@@ -126,3 +126,16 @@ CallEventBase::CallEventBase(Event::Type type, const QJsonObject& json)
     if (callId().isEmpty())
         qCWarning(EVENTS) << id() << "is a call event with an empty call id";
 }
+
+void RoomEvent::setOriginalEvent(event_ptr_tt<RoomEvent> originalEvent)
+{
+    _originalEvent = std::move(originalEvent);
+}
+
+const QJsonObject RoomEvent::encryptedJson() const
+{
+    if(!_originalEvent) {
+        return {};
+    }
+    return _originalEvent->fullJson();
+}
diff --git a/lib/events/roomevent.h b/lib/events/roomevent.h
index 7f13f6f2..35527a62 100644
--- a/lib/events/roomevent.h
+++ b/lib/events/roomevent.h
@@ -60,11 +60,15 @@ public:
     //! callback for that in RoomEvent.
     void addId(const QString& newId);
 
+    void setOriginalEvent(event_ptr_tt<RoomEvent> originalEvent);
+    const QJsonObject encryptedJson() const;
+
 protected:
     void dumpTo(QDebug dbg) const override;
 
 private:
     event_ptr_tt<RedactionEvent> _redactedBecause;
+    event_ptr_tt<RoomEvent> _originalEvent;
 };
 using RoomEventPtr = event_ptr_tt<RoomEvent>;
 using RoomEvents = EventsArray<RoomEvent>;