mirror-chatterino2/src/providers/ffz/FfzEmotes.cpp

#include "providers/ffz/FfzEmotes.hpp"

#include <QJsonArray>

#include "common/NetworkRequest.hpp"
#include "common/Outcome.hpp"
#include "debug/Log.hpp"
#include "messages/Emote.hpp"
#include "messages/Image.hpp"

namespace chatterino {
namespace {
Url getEmoteLink(const QJsonObject &urls, const QString &emoteScale)
{
    auto emote = urls.value(emoteScale);
    if (emote.isUndefined()) {
        return {""};
    }

    assert(emote.isString());

    return {"https:" + emote.toString()};
}
void fillInEmoteData(const QJsonObject &urls, const EmoteName &name,
                     const QString &tooltip, Emote &emoteData)
{
    auto url1x = getEmoteLink(urls, "1");
    auto url2x = getEmoteLink(urls, "2");
    auto url3x = getEmoteLink(urls, "4");

    //, code, tooltip
    emoteData.name = name;
    emoteData.images =
        ImageSet{Image::fromUrl(url1x, 1), Image::fromUrl(url2x, 0.5),
                 Image::fromUrl(url3x, 0.25)};
    emoteData.tooltip = {tooltip};
}
EmotePtr cachedOrMake(Emote &&emote, const EmoteId &id)
{
    static std::unordered_map<EmoteId, std::weak_ptr<const Emote>> cache;
    static std::mutex mutex;

    return cachedOrMakeEmotePtr(std::move(emote), cache, mutex, id);
}
std::pair<Outcome, EmoteMap> parseGlobalEmotes(const QJsonObject &jsonRoot,
                                               const EmoteMap &currentEmotes)
{
    auto jsonSets = jsonRoot.value("sets").toObject();
    auto emotes = EmoteMap();

    for (auto jsonSet : jsonSets) {
        auto jsonEmotes = jsonSet.toObject().value("emoticons").toArray();

        for (auto jsonEmoteValue : jsonEmotes) {
            auto jsonEmote = jsonEmoteValue.toObject();

            auto name = EmoteName{jsonEmote.value("name").toString()};
            auto id = EmoteId{jsonEmote.value("id").toString()};
            auto urls = jsonEmote.value("urls").toObject();

            auto emote = Emote();
            fillInEmoteData(urls, name, name.string + "<br/>Global FFZ Emote",
                            emote);
            emote.homePage =
                Url{QString("https://www.frankerfacez.com/emoticon/%1-%2")
                        .arg(id.string)
                        .arg(name.string)};

            emotes[name] =
                cachedOrMakeEmotePtr(std::move(emote), currentEmotes);
        }
    }

    return {Success, std::move(emotes)};
}
std::pair<Outcome, EmoteMap> parseChannelEmotes(const QJsonObject &jsonRoot)
{
    auto jsonSets = jsonRoot.value("sets").toObject();
    auto emotes = EmoteMap();

    for (auto jsonSet : jsonSets) {
        auto jsonEmotes = jsonSet.toObject().value("emoticons").toArray();

        for (auto _jsonEmote : jsonEmotes) {
            auto jsonEmote = _jsonEmote.toObject();

            // margins
            auto id = EmoteId{QString::number(jsonEmote.value("id").toInt())};
            auto name = EmoteName{jsonEmote.value("name").toString()};
            auto urls = jsonEmote.value("urls").toObject();

            Emote emote;
            fillInEmoteData(urls, name, name.string + "<br/>Channel FFZ Emote",
                            emote);
            emote.homePage =
                Url{QString("https://www.frankerfacez.com/emoticon/%1-%2")
                        .arg(id.string)
                        .arg(name.string)};

            emotes[name] = cachedOrMake(std::move(emote), id);
        }
    }

    return {Success, std::move(emotes)};
}
}  // namespace

FfzEmotes::FfzEmotes()
    : global_(std::make_shared<EmoteMap>())
{
}

std::shared_ptr<const EmoteMap> FfzEmotes::emotes() const
{
    return this->global_.get();
}

boost::optional<EmotePtr> FfzEmotes::emote(const EmoteName &name) const
{
    auto emotes = this->global_.get();
    auto it = emotes->find(name);
    if (it != emotes->end()) return it->second;
    return boost::none;
}

void FfzEmotes::loadEmotes()
{
    QString url("https://api.frankerfacez.com/v1/set/global");

    NetworkRequest request(url);
    request.setCaller(QThread::currentThread());
    request.setTimeout(30000);

    request.onSuccess([this](auto result) -> Outcome {
        auto emotes = this->emotes();
        auto pair = parseGlobalEmotes(result.parseJson(), *emotes);
        if (pair.first)
            this->global_.set(
                std::make_shared<EmoteMap>(std::move(pair.second)));
        return pair.first;
    });

    request.execute();
}

void FfzEmotes::loadChannel(const QString &channelName,
                            std::function<void(EmoteMap &&)> callback)
{
    log("[FFZEmotes] Reload FFZ Channel Emotes for channel %s\n", channelName);

    NetworkRequest request("https://api.frankerfacez.com/v1/room/" +
                           channelName);
    request.setCaller(QThread::currentThread());
    request.setTimeout(3000);

    request.onSuccess([callback = std::move(callback)](auto result) -> Outcome {
        auto pair = parseChannelEmotes(result.parseJson());
        if (pair.first) callback(std::move(pair.second));
        return pair.first;
    });

    request.execute();
}

}  // namespace chatterino