fourtf
95 lines
2.8 KiB
C++
95 lines
2.8 KiB
C++
#include "FfzBadges.hpp"
|
|
|
|
#include <QJsonArray>
|
|
#include <QJsonObject>
|
|
#include <QJsonValue>
|
|
#include <QThread>
|
|
#include <QUrl>
|
|
#include <map>
|
|
#include <shared_mutex>
|
|
#include "common/NetworkRequest.hpp"
|
|
#include "common/Outcome.hpp"
|
|
#include "messages/Emote.hpp"
|
|
|
|
namespace chatterino {
|
|
|
|
void FfzBadges::initialize(Settings &settings, Paths &paths)
|
|
{
|
|
this->loadFfzBadges();
|
|
}
|
|
|
|
boost::optional<EmotePtr> FfzBadges::getBadge(const UserId &id)
|
|
{
|
|
std::shared_lock lock(this->mutex_);
|
|
|
|
auto it = this->badgeMap.find(id.string);
|
|
if (it != this->badgeMap.end())
|
|
{
|
|
return this->badges[it->second];
|
|
}
|
|
return boost::none;
|
|
}
|
|
boost::optional<QColor> FfzBadges::getBadgeColor(const UserId &id)
|
|
{
|
|
std::shared_lock lock(this->mutex_);
|
|
|
|
auto badgeIt = this->badgeMap.find(id.string);
|
|
if (badgeIt != this->badgeMap.end())
|
|
{
|
|
auto colorIt = this->colorMap.find(badgeIt->second);
|
|
if (colorIt != this->colorMap.end())
|
|
{
|
|
return colorIt->second;
|
|
}
|
|
return boost::none;
|
|
}
|
|
return boost::none;
|
|
}
|
|
|
|
void FfzBadges::loadFfzBadges()
|
|
{
|
|
static QUrl url("https://api.frankerfacez.com/v1/badges/ids");
|
|
|
|
NetworkRequest(url)
|
|
.onSuccess([this](auto result) -> Outcome {
|
|
std::unique_lock lock(this->mutex_);
|
|
|
|
auto jsonRoot = result.parseJson();
|
|
int index = 0;
|
|
for (const auto &jsonBadge_ : jsonRoot.value("badges").toArray())
|
|
{
|
|
auto jsonBadge = jsonBadge_.toObject();
|
|
auto jsonUrls = jsonBadge.value("urls").toObject();
|
|
|
|
auto emote = Emote{
|
|
EmoteName{},
|
|
ImageSet{
|
|
Url{QString("https:") + jsonUrls.value("1").toString()},
|
|
Url{QString("https:") + jsonUrls.value("2").toString()},
|
|
Url{QString("https:") +
|
|
jsonUrls.value("4").toString()}},
|
|
Tooltip{jsonBadge.value("title").toString()}, Url{}};
|
|
|
|
this->badges.push_back(
|
|
std::make_shared<const Emote>(std::move(emote)));
|
|
this->colorMap[index] =
|
|
QColor(jsonBadge.value("color").toString());
|
|
|
|
auto badgeId = QString::number(jsonBadge.value("id").toInt());
|
|
for (const auto &user : jsonRoot.value("users")
|
|
.toObject()
|
|
.value(badgeId)
|
|
.toArray())
|
|
{
|
|
this->badgeMap[QString::number(user.toInt())] = index;
|
|
}
|
|
++index;
|
|
}
|
|
|
|
return Success;
|
|
})
|
|
.execute();
|
|
}
|
|
|
|
} // namespace chatterino
|