mirror-chatterino2/src/providers/twitch/TwitchBadges.cpp

#include "TwitchBadges.hpp"

#include <QJsonArray>
#include <QJsonObject>
#include <QJsonValue>
#include <QThread>

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

namespace chatterino {

void TwitchBadges::loadTwitchBadges()
{
    static QString url(
        "https://badges.twitch.tv/v1/badges/global/display?language=en");

    NetworkRequest req(url);
    req.setCaller(QThread::currentThread());
    req.onSuccess([this](auto result) -> Outcome {
        auto root = result.parseJson();
        auto badgeSets = this->badgeSets_.access();

        auto jsonSets = root.value("badge_sets").toObject();
        for (auto sIt = jsonSets.begin(); sIt != jsonSets.end(); ++sIt) {
            auto key = sIt.key();
            auto versions = sIt.value().toObject().value("versions").toObject();

            for (auto vIt = versions.begin(); vIt != versions.end(); ++vIt) {
                auto versionObj = vIt.value().toObject();

                auto emote = Emote{
                    {""},
                    ImageSet{
                        Image::fromUrl(
                            {versionObj.value("image_url_1x").toString()}, 1),
                        Image::fromUrl(
                            {versionObj.value("image_url_2x").toString()}, .5),
                        Image::fromUrl(
                            {versionObj.value("image_url_4x").toString()}, .25),
                    },
                    Tooltip{root.value("description").toString()},
                    Url{root.value("clickURL").toString()}};
                // "title"
                // "clickAction"

                log("{} {}", key, vIt.key());

                (*badgeSets)[key][vIt.key()] = std::make_shared<Emote>(emote);
            }
        }

        return Success;
    });

    req.execute();
}

boost::optional<EmotePtr> TwitchBadges::badge(const QString &set,
                                              const QString &version) const
{
    auto badgeSets = this->badgeSets_.access();
    auto it = badgeSets->find(set);
    if (it != badgeSets->end()) {
        auto it2 = it->second.find(version);
        if (it2 != it->second.end()) {
            return it2->second;
        }
    }
    return boost::none;
}

}  // namespace chatterino