neochat/src/models/emojimodel.cpp

// SPDX-FileCopyrightText: 2017 Konstantinos Sideris <siderisk@auth.gr>
// SPDX-License-Identifier: GPL-3.0-or-later

#include <QVariant>

#include "emojimodel.h"
#include "emojitones.h"
#include <QDebug>

#include <algorithm>

#include "customemojimodel.h"
#include <KLocalizedString>

using namespace Qt::StringLiterals;

EmojiModel::EmojiModel(QObject *parent)
    : QAbstractListModel(parent)
    , m_config(KSharedConfig::openStateConfig())
    , m_configGroup(KConfigGroup(m_config, u"Editor"_s))
{
    if (_emojis.isEmpty()) {
#include "emojis.h"
    }
}

int EmojiModel::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent);
    int total = 0;
    for (const auto &category : std::as_const(_emojis)) {
        total += category.count();
    }
    return total;
}

QVariant EmojiModel::data(const QModelIndex &index, int role) const
{
    auto row = index.row();
    for (const auto &category : std::as_const(_emojis)) {
        if (row >= category.count()) {
            row -= category.count();
            continue;
        }
        auto emoji = category[row].value<Emoji>();
        switch (role) {
        case ShortNameRole:
            return u":%1:"_s.arg(emoji.shortName);
        case UnicodeRole:
        case ReplacedTextRole:
            return emoji.unicode;
        case InvalidRole:
            return u"invalid"_s;
        case DisplayRole:
            return u"%2   :%1:"_s.arg(emoji.shortName, emoji.unicode);
        case DescriptionRole:
            return emoji.description;
        }
    }
    return {};
}

QHash<int, QByteArray> EmojiModel::roleNames() const
{
    return {{ShortNameRole, "shortName"}, {UnicodeRole, "unicode"}};
}

QStringList EmojiModel::lastUsedEmojis() const
{
    return m_configGroup.readEntry(u"lastUsedEmojis"_s, QStringList());
}

QVariantList EmojiModel::filterModel(const QString &filter, bool limit)
{
    auto emojis = CustomEmojiModel::instance().filterModel(filter);
    emojis += filterModelNoCustom(filter, limit);
    return emojis;
}

QVariantList EmojiModel::filterModelNoCustom(const QString &filter, bool limit)
{
    QVariantList result;

    const auto &values = _emojis.values();
    for (const auto &e : values) {
        for (const auto &variant : e) {
            const auto &emoji = qvariant_cast<Emoji>(variant);
            if (emoji.shortName.contains(filter, Qt::CaseInsensitive)) {
                result.append(variant);
                if (result.length() > 10 && limit) {
                    return result;
                }
            }
        }
    }
    return result;
}

void EmojiModel::emojiUsed(const QVariant &modelData)
{
    auto list = lastUsedEmojis();
    const auto emoji = modelData.value<Emoji>();

    auto it = list.begin();
    while (it != list.end()) {
        if (*it == emoji.shortName) {
            it = list.erase(it);
        } else {
            it++;
        }
    }

    list.push_front(emoji.shortName);

    m_configGroup.writeEntry(u"lastUsedEmojis"_s, list);

    Q_EMIT historyChanged();
}

QVariantList EmojiModel::emojis(Category category) const
{
    if (category == History) {
        return emojiHistory();
    }
    if (category == HistoryNoCustom) {
        QVariantList list;
        const auto &history = emojiHistory();
        for (const auto &e : history) {
            auto emoji = qvariant_cast<Emoji>(e);
            if (!emoji.isCustom) {
                list.append(e);
            }
        }
        return list;
    }
    if (category == Custom) {
        return CustomEmojiModel::instance().filterModel({});
    }
    return _emojis[category];
}

QVariantList EmojiModel::tones(const QString &baseEmoji) const
{
    if (baseEmoji.endsWith(u"tone"_s)) {
        return EmojiTones::_tones.values(baseEmoji.split(u":"_s)[0]);
    }
    return EmojiTones::_tones.values(baseEmoji);
}

QHash<EmojiModel::Category, QVariantList> EmojiModel::_emojis;

QVariantList EmojiModel::categories() const
{
    return QVariantList{
        {QVariantMap{
            {u"category"_s, EmojiModel::HistoryNoCustom},
            {u"name"_s, i18nc("Previously used emojis", "History")},
            {u"emoji"_s, u"⌛️"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Smileys},
            {u"name"_s, i18nc("'Smileys' is a category of emoji", "Smileys")},
            {u"emoji"_s, u"😏"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::People},
            {u"name"_s, i18nc("'People' is a category of emoji", "People")},
            {u"emoji"_s, u"🙋‍♂️"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Nature},
            {u"name"_s, i18nc("'Nature' is a category of emoji", "Nature")},
            {u"emoji"_s, u"🌲"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Food},
            {u"name"_s, i18nc("'Food' is a category of emoji", "Food")},
            {u"emoji"_s, u"🍛"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Activities},
            {u"name"_s, i18nc("'Activities' is a category of emoji", "Activities")},
            {u"emoji"_s, u"🚁"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Travel},
            {u"name"_s, i18nc("'Travel' is  a category of emoji", "Travel")},
            {u"emoji"_s, u"🚅"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Objects},
            {u"name"_s, i18nc("'Objects' is a category of emoji", "Objects")},
            {u"emoji"_s, u"💡"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Symbols},
            {u"name"_s, i18nc("'Symbols' is a category of emoji", "Symbols")},
            {u"emoji"_s, u"🔣"_s},
        }},
        {QVariantMap{
            {u"category"_s, EmojiModel::Flags},
            {u"name"_s, i18nc("'Flags' is a category of emoji", "Flags")},
            {u"emoji"_s, u"🏁"_s},
        }},
    };
}

QVariantList EmojiModel::categoriesWithCustom() const
{
    auto cats = categories();
    cats.removeAt(0);
    cats.insert(0,
                QVariantMap{
                    {u"category"_s, EmojiModel::History},
                    {u"name"_s, i18nc("Previously used emojis", "History")},
                    {u"emoji"_s, u"⌛️"_s},
                });
    cats.insert(1,
                QVariantMap{
                    {u"category"_s, EmojiModel::Custom},
                    {u"name"_s, i18nc("'Custom' is a category of emoji", "Custom")},
                    {u"emoji"_s, u"🖼️"_s},
                });
    ;
    return cats;
}

QVariantList EmojiModel::emojiHistory() const
{
    QVariantList list;
    const auto &lastUsed = lastUsedEmojis();
    for (const auto &historicEmoji : lastUsed) {
        for (const auto &emojiCategory : std::as_const(_emojis)) {
            for (const auto &emoji : emojiCategory) {
                if (qvariant_cast<Emoji>(emoji).shortName == historicEmoji) {
                    list.append(emoji);
                }
            }
        }
    }
    return list;
}

#include "moc_emojimodel.cpp"