1 files changed, 110 insertions, 0 deletions
diff --git a/src/emoji/EmojiModel.cpp b/src/emoji/EmojiModel.cpp
new file mode 100644
index 00000000..898a47c6
--- /dev/null
+++ b/src/emoji/EmojiModel.cpp
@@ -0,0 +1,110 @@
+#include "EmojiModel.h"
+
+#include <Cache.h>
+#include <MatrixClient.h>
+
+using namespace emoji;
+
+QHash<int, QByteArray>
+EmojiModel::roleNames() const
+{
+        static QHash<int, QByteArray> roles;
+
+        if (roles.isEmpty()) {
+                roles = QAbstractListModel::roleNames();
+                roles[static_cast<int>(EmojiModel::Roles::Unicode)] = QByteArrayLiteral("unicode");
+                roles[static_cast<int>(EmojiModel::Roles::ShortName)] =
+                  QByteArrayLiteral("shortName");
+                roles[static_cast<int>(EmojiModel::Roles::Category)] =
+                  QByteArrayLiteral("category");
+                roles[static_cast<int>(EmojiModel::Roles::Emoji)] = QByteArrayLiteral("emoji");
+        }
+
+        return roles;
+}
+
+int
+EmojiModel::rowCount(const QModelIndex &parent) const
+{
+        return parent == QModelIndex() ? Provider::emoji.count() : 0;
+}
+
+QVariant
+EmojiModel::data(const QModelIndex &index, int role) const
+{
+        if (hasIndex(index.row(), index.column(), index.parent())) {
+                switch (role) {
+                case Qt::DisplayRole:
+                case static_cast<int>(EmojiModel::Roles::Unicode):
+                        return Provider::emoji[index.row()].unicode();
+
+                case Qt::ToolTipRole:
+                case static_cast<int>(EmojiModel::Roles::ShortName):
+                        return Provider::emoji[index.row()].shortName();
+
+                case static_cast<int>(EmojiModel::Roles::Category):
+                        return QVariant::fromValue(Provider::emoji[index.row()].category());
+
+                case static_cast<int>(EmojiModel::Roles::Emoji):
+                        return QVariant::fromValue(Provider::emoji[index.row()]);
+                }
+        }
+
+        return {};
+}
+
+EmojiProxyModel::EmojiProxyModel(QObject *parent)
+  : QSortFilterProxyModel(parent)
+{}
+
+EmojiProxyModel::~EmojiProxyModel() {}
+
+Emoji::Category
+EmojiProxyModel::category() const
+{
+        return category_;
+}
+
+void
+EmojiProxyModel::setCategory(Emoji::Category cat)
+{
+        if (category_ == cat) {
+                return;
+        }
+
+        category_ = cat;
+        emit categoryChanged();
+
+        invalidateFilter();
+}
+
+QString
+EmojiProxyModel::filter() const
+{
+        return filterRegExp().pattern();
+}
+
+void
+EmojiProxyModel::setFilter(const QString &filter)
+{
+        if (filterRegExp().pattern() == filter) {
+                return;
+        }
+
+        setFilterWildcard(filter);
+        emit filterChanged();
+}
+
+bool
+EmojiProxyModel::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
+{
+        const QModelIndex index = sourceModel()->index(sourceRow, 0, sourceParent);
+        const Emoji emoji = index.data(static_cast<int>(EmojiModel::Roles::Emoji)).value<Emoji>();
+
+        // TODO: Add favorites / recently used
+        if (category_ != Emoji::Category::Search) {
+                return emoji.category() == category_;
+        }
+
+        return filterRegExp().isEmpty() ? true : filterRegExp().indexIn(emoji.shortName()) != -1;
+}