summary refs log tree commit diff
path: root/src/emoji/EmojiModel.cpp
blob: 85c2dd34c3395792b0a83676854fcfac068770b1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#include "EmojiModel.h"

#include <Cache.h>
#include <MatrixClient.h>

#include "CompletionModelRoles.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 CompletionModel::CompletionRole:
                case static_cast<int>(EmojiModel::Roles::Unicode):
                        return Provider::emoji[index.row()].unicode;

                case Qt::ToolTipRole:
                case CompletionModel::SearchRole:
                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() {}

EmojiCategory
EmojiProxyModel::category() const
{
        return category_;
}

void
EmojiProxyModel::setCategory(EmojiCategory 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_ != EmojiCategory::Search) {
                return emoji.category == category_;
        }

        return filterRegExp().isEmpty() ? true : filterRegExp().indexIn(emoji.shortName) != -1;
}