From ea6b19b3077dad0280e32aa762137728432e01f4 Mon Sep 17 00:00:00 2001
From: Nicolas Werner <nicolas.werner@hotmail.de>
Date: Mon, 27 Dec 2021 06:23:36 +0100
Subject: Add an ugly custom emote completer using ~

---
 src/Cache.cpp                        | 34 ++++++++++++++++++----------------
 src/CombinedImagePackModel.cpp       |  4 +++-
 src/timeline/TimelineViewManager.cpp |  5 +++++
 3 files changed, 26 insertions(+), 17 deletions(-)

(limited to 'src')

diff --git a/src/Cache.cpp b/src/Cache.cpp
index dd26157c..6db768d3 100644
--- a/src/Cache.cpp
+++ b/src/Cache.cpp
@@ -3580,24 +3580,26 @@ Cache::getImagePacks(const std::string &room_id, std::optional<bool> stickers)
     auto addPack = [&infos, stickers](const mtx::events::msc2545::ImagePack &pack,
                                       const std::string &source_room,
                                       const std::string &state_key) {
-        if (!pack.pack || !stickers.has_value() ||
-            (stickers.value() ? pack.pack->is_sticker() : pack.pack->is_emoji())) {
-            ImagePackInfo info;
-            info.source_room = source_room;
-            info.state_key   = state_key;
-            info.pack.pack   = pack.pack;
-
-            for (const auto &img : pack.images) {
-                if (stickers.has_value() && img.second.overrides_usage() &&
-                    (stickers ? !img.second.is_sticker() : !img.second.is_emoji()))
-                    continue;
-
-                info.pack.images.insert(img);
-            }
+        bool pack_matches = !stickers.has_value() ||
+                            (stickers.value() ? pack.pack->is_sticker() : pack.pack->is_emoji());
+
+        ImagePackInfo info;
+        info.source_room = source_room;
+        info.state_key   = state_key;
+        info.pack.pack   = pack.pack;
+
+        for (const auto &img : pack.images) {
+            if (stickers.has_value() &&
+                (img.second.overrides_usage()
+                   ? (!stickers.value() ? img.second.is_sticker() : img.second.is_emoji())
+                   : !pack_matches))
+                continue;
 
-            if (!info.pack.images.empty())
-                infos.push_back(std::move(info));
+            info.pack.images.insert(img);
         }
+
+        if (!info.pack.images.empty())
+            infos.push_back(std::move(info));
     };
 
     // packs from account data
diff --git a/src/CombinedImagePackModel.cpp b/src/CombinedImagePackModel.cpp
index 9a52f810..2a8959c0 100644
--- a/src/CombinedImagePackModel.cpp
+++ b/src/CombinedImagePackModel.cpp
@@ -56,7 +56,9 @@ CombinedImagePackModel::data(const QModelIndex &index, int role) const
     if (hasIndex(index.row(), index.column(), index.parent())) {
         switch (role) {
         case CompletionModel::CompletionRole:
-            return QString::fromStdString(images[index.row()].image.url);
+            return QString("<img data-mx-emoticon height=32 src=\"%1\" alt=\"%2\" title=\"%2\">")
+              .arg(QString::fromStdString(images[index.row()].image.url).toHtmlEscaped(),
+                   images[index.row()].shortcode);
         case Roles::Url:
             return QString::fromStdString(images[index.row()].image.url);
         case CompletionModel::SearchRole:
diff --git a/src/timeline/TimelineViewManager.cpp b/src/timeline/TimelineViewManager.cpp
index 9303ee40..eb30fe8c 100644
--- a/src/timeline/TimelineViewManager.cpp
+++ b/src/timeline/TimelineViewManager.cpp
@@ -550,6 +550,11 @@ TimelineViewManager::completerFor(QString completerName, QString roomId)
         auto proxy        = new CompletionProxyModel(stickerModel, 1, static_cast<size_t>(-1) / 4);
         stickerModel->setParent(proxy);
         return proxy;
+    } else if (completerName == "customEmoji") {
+        auto stickerModel = new CombinedImagePackModel(roomId.toStdString(), false);
+        auto proxy        = new CompletionProxyModel(stickerModel);
+        stickerModel->setParent(proxy);
+        return proxy;
     }
     return nullptr;
 }
-- 
cgit 1.5.1