From eafbab6ae128dd5b14f145ba9214a6673e982951 Mon Sep 17 00:00:00 2001
From: Nicolas Werner <nicolas.werner@hotmail.de>
Date: Wed, 21 Jul 2021 13:37:57 +0200
Subject: Add menu to enable or disable stickers globally

---
 src/ImagePackListModel.cpp | 76 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)
 create mode 100644 src/ImagePackListModel.cpp

(limited to 'src/ImagePackListModel.cpp')

diff --git a/src/ImagePackListModel.cpp b/src/ImagePackListModel.cpp
new file mode 100644
index 00000000..89f1f68e
--- /dev/null
+++ b/src/ImagePackListModel.cpp
@@ -0,0 +1,76 @@
+// SPDX-FileCopyrightText: 2021 Nheko Contributors
+//
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "ImagePackListModel.h"
+
+#include <QQmlEngine>
+
+#include "Cache_p.h"
+#include "SingleImagePackModel.h"
+
+ImagePackListModel::ImagePackListModel(const std::string &roomId, QObject *parent)
+  : QAbstractListModel(parent)
+  , room_id(roomId)
+{
+        auto packs_ = cache::client()->getImagePacks(room_id, std::nullopt);
+
+        for (const auto &pack : packs_) {
+                packs.push_back(
+                  QSharedPointer<SingleImagePackModel>(new SingleImagePackModel(pack)));
+        }
+}
+
+int
+ImagePackListModel::rowCount(const QModelIndex &) const
+{
+        return (int)packs.size();
+}
+
+QHash<int, QByteArray>
+ImagePackListModel::roleNames() const
+{
+        return {
+          {Roles::DisplayName, "displayName"},
+          {Roles::AvatarUrl, "avatarUrl"},
+          {Roles::FromAccountData, "fromAccountData"},
+          {Roles::FromCurrentRoom, "fromCurrentRoom"},
+          {Roles::StateKey, "statekey"},
+          {Roles::RoomId, "roomid"},
+        };
+}
+
+QVariant
+ImagePackListModel::data(const QModelIndex &index, int role) const
+{
+        if (hasIndex(index.row(), index.column(), index.parent())) {
+                const auto &pack = packs.at(index.row());
+                switch (role) {
+                case Roles::DisplayName:
+                        return pack->packname();
+                case Roles::AvatarUrl:
+                        return pack->avatarUrl();
+                case Roles::FromAccountData:
+                        return pack->roomid().isEmpty();
+                case Roles::FromCurrentRoom:
+                        return pack->roomid().toStdString() == this->room_id;
+                case Roles::StateKey:
+                        return pack->statekey();
+                case Roles::RoomId:
+                        return pack->roomid();
+                default:
+                        return {};
+                }
+        }
+        return {};
+}
+
+SingleImagePackModel *
+ImagePackListModel::packAt(int row)
+{
+        if (row < 0 || static_cast<size_t>(row) >= packs.size())
+                return {};
+        auto e = packs.at(row).get();
+        QQmlEngine::setObjectOwnership(e, QQmlEngine::CppOwnership);
+        return e;
+}
-- 
cgit 1.5.1