From e8e475956b3d2a57981ad41536a1ddc1b7361591 Mon Sep 17 00:00:00 2001
From: Nicolas Werner <nicolas.werner@hotmail.de>
Date: Sat, 17 Dec 2022 01:11:19 +0100
Subject: [PATCH] Fix confetti in the sidebar

---
 src/Utils.cpp | 42 ++++++++++++------------------------------
 src/Utils.h   | 21 ++++++++++++++++++---
 2 files changed, 30 insertions(+), 33 deletions(-)

diff --git a/src/Utils.cpp b/src/Utils.cpp
index c19871bc..b0d84339 100644
--- a/src/Utils.cpp
+++ b/src/Utils.cpp
@@ -45,16 +45,17 @@ createDescriptionInfo(const Event &event, const QString &localUser, const QStrin
 
     const auto username = displayName;
     const auto ts       = QDateTime::fromMSecsSinceEpoch(msg.origin_server_ts);
-    auto body           = utils::event_body(event).trimmed();
+    auto body           = mtx::accessors::body(event);
     if (mtx::accessors::relations(event).reply_to())
-        body = QString::fromStdString(utils::stripReplyFromBody(body.toStdString()));
+        body = utils::stripReplyFromBody(body);
 
-    return DescInfo{QString::fromStdString(msg.event_id),
-                    sender,
-                    utils::messageDescription<T>(username, body, sender == localUser),
-                    utils::descriptiveTime(ts),
-                    msg.origin_server_ts,
-                    ts};
+    return DescInfo{
+      QString::fromStdString(msg.event_id),
+      sender,
+      utils::messageDescription<T>(username, QString::fromStdString(body), sender == localUser),
+      utils::descriptiveTime(ts),
+      msg.origin_server_ts,
+      ts};
 }
 
 std::string
@@ -216,6 +217,7 @@ utils::getMessageDescription(const TimelineEvent &event,
     using Notice     = mtx::events::RoomEvent<mtx::events::msg::Notice>;
     using Text       = mtx::events::RoomEvent<mtx::events::msg::Text>;
     using Video      = mtx::events::RoomEvent<mtx::events::msg::Video>;
+    using Confetti   = mtx::events::RoomEvent<mtx::events::msg::Confetti>;
     using CallInvite = mtx::events::RoomEvent<mtx::events::voip::CallInvite>;
     using CallAnswer = mtx::events::RoomEvent<mtx::events::voip::CallAnswer>;
     using CallHangUp = mtx::events::RoomEvent<mtx::events::voip::CallHangUp>;
@@ -236,6 +238,8 @@ utils::getMessageDescription(const TimelineEvent &event,
         return createDescriptionInfo<Text>(event, localUser, displayName);
     } else if (std::holds_alternative<Video>(event)) {
         return createDescriptionInfo<Video>(event, localUser, displayName);
+    } else if (std::holds_alternative<Confetti>(event)) {
+        return createDescriptionInfo<Confetti>(event, localUser, displayName);
     } else if (std::holds_alternative<CallInvite>(event)) {
         return createDescriptionInfo<CallInvite>(event, localUser, displayName);
     } else if (std::holds_alternative<CallAnswer>(event)) {
@@ -324,28 +328,6 @@ utils::levenshtein_distance(const std::string &s1, const std::string &s2)
     return *std::min_element(row1.begin(), row1.end());
 }
 
-QString
-utils::event_body(const mtx::events::collections::TimelineEvents &e)
-{
-    using namespace mtx::events;
-    if (auto ev = std::get_if<RoomEvent<msg::Audio>>(&e); ev != nullptr)
-        return QString::fromStdString(ev->content.body);
-    if (auto ev = std::get_if<RoomEvent<msg::Emote>>(&e); ev != nullptr)
-        return QString::fromStdString(ev->content.body);
-    if (auto ev = std::get_if<RoomEvent<msg::File>>(&e); ev != nullptr)
-        return QString::fromStdString(ev->content.body);
-    if (auto ev = std::get_if<RoomEvent<msg::Image>>(&e); ev != nullptr)
-        return QString::fromStdString(ev->content.body);
-    if (auto ev = std::get_if<RoomEvent<msg::Notice>>(&e); ev != nullptr)
-        return QString::fromStdString(ev->content.body);
-    if (auto ev = std::get_if<RoomEvent<msg::Text>>(&e); ev != nullptr)
-        return QString::fromStdString(ev->content.body);
-    if (auto ev = std::get_if<RoomEvent<msg::Video>>(&e); ev != nullptr)
-        return QString::fromStdString(ev->content.body);
-
-    return QString();
-}
-
 QPixmap
 utils::scaleImageToPixmap(const QImage &img, int size)
 {
diff --git a/src/Utils.h b/src/Utils.h
index e4c3ccb3..31a1a0de 100644
--- a/src/Utils.h
+++ b/src/Utils.h
@@ -89,9 +89,6 @@ firstChar(const QString &input);
 QString
 humanReadableFileSize(uint64_t bytes);
 
-QString
-event_body(const mtx::events::collections::TimelineEvents &event);
-
 //! Match widgets/events with a description message.
 template<class T>
 QString
@@ -107,6 +104,7 @@ messageDescription(const QString &username = QString(),
     using Sticker    = mtx::events::Sticker;
     using Text       = mtx::events::RoomEvent<mtx::events::msg::Text>;
     using Video      = mtx::events::RoomEvent<mtx::events::msg::Video>;
+    using Confetti   = mtx::events::RoomEvent<mtx::events::msg::Confetti>;
     using CallInvite = mtx::events::RoomEvent<mtx::events::voip::CallInvite>;
     using CallAnswer = mtx::events::RoomEvent<mtx::events::voip::CallAnswer>;
     using CallHangUp = mtx::events::RoomEvent<mtx::events::voip::CallHangUp>;
@@ -158,6 +156,23 @@ messageDescription(const QString &username = QString(),
         else
             return QCoreApplication::translate("message-description sent:", "%1: %2")
               .arg(username, body);
+    } else if (std::is_same<T, Confetti>::value) {
+        if (body.isEmpty()) {
+            if (isLocal)
+                return QCoreApplication::translate("message-description sent:",
+                                                   "You sent some confetti");
+            else
+                return QCoreApplication::translate("message-description sent:",
+                                                   "%1 sent some confetti")
+                  .arg(username);
+        } else {
+            if (isLocal)
+                return QCoreApplication::translate("message-description sent:", "You: %1")
+                  .arg(body);
+            else
+                return QCoreApplication::translate("message-description sent:", "%1: %2")
+                  .arg(username, body);
+        }
     } else if (std::is_same<T, Emote>::value) {
         return QStringLiteral("* %1 %2").arg(username, body);
     } else if (std::is_same<T, Encrypted>::value) {