File indexing completed on 2024-05-05 05:01:21

 // SPDX-FileCopyrightText: 2021 Carson Black <uhhadd@gmail.com>
 // SPDX-License-Identifier: GPL-2.0-or-later
 
 #include "customemojimodel.h"
 
 #include <QImage>
 #include <QMimeDatabase>
 
 #include "emojimodel.h"
 
 #include <Quotient/csapi/account-data.h>
 #include <Quotient/csapi/content-repo.h>
 
 using namespace Quotient;
 
 void CustomEmojiModel::setConnection(NeoChatConnection *connection)
 {
     if (connection == m_connection) {
         return;
     }
     m_connection = connection;
     Q_EMIT connectionChanged();
     fetchEmojis();
 }
 
 NeoChatConnection *CustomEmojiModel::connection() const
 {
     return m_connection;
 }
 
 void CustomEmojiModel::fetchEmojis()
 {
     if (!m_connection) {
         return;
     }
 
     const auto &data = m_connection->accountData("im.ponies.user_emotes"_ls);
     if (data == nullptr) {
         return;
     }
     QJsonObject emojis = data->contentJson()["images"_ls].toObject();
 
     // TODO: Remove with stable migration
     const auto legacyEmojis = data->contentJson()["emoticons"_ls].toObject();
     for (const auto &emoji : legacyEmojis.keys()) {
         if (!emojis.contains(emoji)) {
             emojis[emoji] = legacyEmojis[emoji];
         }
     }
 
     beginResetModel();
     m_emojis.clear();
 
     for (const auto &emoji : emojis.keys()) {
         const auto &data = emojis[emoji];
 
         const auto e = emoji.startsWith(":"_ls) ? emoji : (QStringLiteral(":") + emoji + QStringLiteral(":"));
 
         m_emojis << CustomEmoji{e, data.toObject()["url"_ls].toString(), QRegularExpression(e)};
     }
 
     endResetModel();
 }
 
 void CustomEmojiModel::addEmoji(const QString &name, const QUrl &location)
 {
     using namespace Quotient;
 
     auto job = m_connection->uploadFile(location.toLocalFile());
 
     connect(job, &BaseJob::success, this, [name, location, job, this] {
         const auto &data = m_connection->accountData("im.ponies.user_emotes"_ls);
         auto json = data != nullptr ? data->contentJson() : QJsonObject();
         auto emojiData = json["images"_ls].toObject();
 
         QString url;
         url = job->contentUri().toString();
 
         QImage image(location.toLocalFile());
         QJsonObject imageInfo;
         imageInfo["w"_ls] = image.width();
         imageInfo["h"_ls] = image.height();
         imageInfo["mimetype"_ls] = QMimeDatabase().mimeTypeForFile(location.toLocalFile()).name();
         imageInfo["size"_ls] = image.sizeInBytes();
 
         emojiData[QStringLiteral("%1").arg(name)] = QJsonObject({
             {QStringLiteral("url"), url},
             {QStringLiteral("info"), imageInfo},
             {QStringLiteral("body"), location.fileName()},
             {"usage"_ls, "emoticon"_ls},
         });
 
         json["images"_ls] = emojiData;
         m_connection->setAccountData("im.ponies.user_emotes"_ls, json);
     });
 }
 
 void CustomEmojiModel::removeEmoji(const QString &name)
 {
     using namespace Quotient;
 
     const auto &data = m_connection->accountData("im.ponies.user_emotes"_ls);
     Q_ASSERT(data);
     auto json = data->contentJson();
     const QString _name = name.mid(1).chopped(1);
     auto emojiData = json["images"_ls].toObject();
 
     if (emojiData.contains(name)) {
         emojiData.remove(name);
         json["images"_ls] = emojiData;
     }
     if (emojiData.contains(_name)) {
         emojiData.remove(_name);
         json["images"_ls] = emojiData;
     }
     emojiData = json["emoticons"_ls].toObject();
     if (emojiData.contains(name)) {
         emojiData.remove(name);
         json["emoticons"_ls] = emojiData;
     }
     if (emojiData.contains(_name)) {
         emojiData.remove(_name);
         json["emoticons"_ls] = emojiData;
     }
     m_connection->setAccountData("im.ponies.user_emotes"_ls, json);
 }
 
 CustomEmojiModel::CustomEmojiModel(QObject *parent)
     : QAbstractListModel(parent)
 {
     connect(this, &CustomEmojiModel::connectionChanged, this, [this]() {
         if (!m_connection) {
             return;
         }
         CustomEmojiModel::fetchEmojis();
         connect(m_connection, &Connection::accountDataChanged, this, [this](const QString &id) {
             if (id != QStringLiteral("im.ponies.user_emotes")) {
                 return;
             }
             fetchEmojis();
         });
     });
     CustomEmojiModel::fetchEmojis();
 }
 
 QVariant CustomEmojiModel::data(const QModelIndex &idx, int role) const
 {
     const auto row = idx.row();
     if (row >= m_emojis.length()) {
         return QVariant();
     }
     const auto &data = m_emojis[row];
 
     switch (Roles(role)) {
     case Roles::ModelData:
         return QVariant::fromValue(Emoji(QStringLiteral("image://mxc/") + data.url.mid(6), data.name, true));
     case Roles::Name:
     case Roles::DisplayRole:
     case Roles::ReplacedTextRole:
         return data.name;
     case Roles::ImageURL:
         return QUrl(QStringLiteral("image://mxc/") + data.url.mid(6));
     case Roles::MxcUrl:
         return data.url.mid(6);
     default:
         return {};
     }
 
     return QVariant();
 }
 
 int CustomEmojiModel::rowCount(const QModelIndex &parent) const
 {
     Q_UNUSED(parent)
 
     return m_emojis.length();
 }
 
 QHash<int, QByteArray> CustomEmojiModel::roleNames() const
 {
     return {
         {Name, "name"},
         {ImageURL, "imageURL"},
         {ModelData, "modelData"},
         {MxcUrl, "mxcUrl"},
     };
 }
 
 QString CustomEmojiModel::preprocessText(QString text)
 {
     for (const auto &emoji : std::as_const(m_emojis)) {
         text.replace(
             emoji.regexp,
             QStringLiteral(R"(<img data-mx-emoticon="" src="%1" alt="%2" title="%2" height="32" vertical-align="middle" />)").arg(emoji.url, emoji.name));
     }
     return text;
 }
 
 QVariantList CustomEmojiModel::filterModel(const QString &filter)
 {
     QVariantList results;
     for (const auto &emoji : std::as_const(m_emojis)) {
         if (results.length() >= 10)
             break;
         if (!emoji.name.contains(filter, Qt::CaseInsensitive))
             continue;
 
         results << QVariant::fromValue(Emoji(QStringLiteral("image://mxc/") + emoji.url.mid(6), emoji.name, true));
     }
     return results;
 }
 
 #include "moc_customemojimodel.cpp"