From f6740316dfcdb287b019b4258df2213c31965d42 Mon Sep 17 00:00:00 2001
From: Kitsune Ral <Kitsune-Ral@users.sf.net>
Date: Wed, 26 Dec 2018 19:36:00 +0900
Subject: PendingEventItem: add FileUploaded status and setFileUploaded helper
 function

---
 lib/eventitem.cpp | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'lib/eventitem.cpp')

diff --git a/lib/eventitem.cpp b/lib/eventitem.cpp
index 79ef769c..8ec3fe48 100644
--- a/lib/eventitem.cpp
+++ b/lib/eventitem.cpp
@@ -17,3 +17,29 @@
  */
 
 #include "eventitem.h"
+
+#include "events/roommessageevent.h"
+#include "events/roomavatarevent.h"
+
+using namespace QMatrixClient;
+
+void PendingEventItem::setFileUploaded(const QUrl& remoteUrl)
+{
+    // TODO: eventually we might introduce hasFileContent to RoomEvent,
+    // and unify the code below.
+    if (auto* rme = getAs<RoomMessageEvent>())
+    {
+        Q_ASSERT(rme->hasFileContent());
+        rme->editContent([remoteUrl] (EventContent::TypedBase& ec) {
+            ec.fileInfo()->url = remoteUrl;
+        });
+    }
+    if (auto* rae = getAs<RoomAvatarEvent>())
+    {
+        Q_ASSERT(rae->content().fileInfo());
+        rae->editContent([remoteUrl] (EventContent::FileInfo& fi) {
+            fi.url = remoteUrl;
+        });
+    }
+    setStatus(EventStatus::FileUploaded);
+}
-- 
cgit v1.2.3