chatterino2
Code Issues

Add ability to highlight messages based on user badges (#1704)

Browse Source 
Co-authored-by: Paweł <zneix@zneix.eu>
Co-authored-by: 23rd <23rd@vivaldi.net>
Co-authored-by: pajlada <rasmus.karlsson@pajlada.com>
This commit is contained in:
Daniel
2021-05-02 18:08:08 -04:00
committed by GitHub
parent 6ab5b13017
commit f6d9fb2aac
34 changed files with 956 additions and 115 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/providers/twitch/TwitchBadges.cpp
+179 -31
View File
@@ -7,60 +7,95 @@
#include "common/NetworkRequest.hpp"
#include "common/Outcome.hpp"
#include "common/QLogging.hpp"
#include "messages/Emote.hpp"
namespace chatterino {
TwitchBadges::TwitchBadges()
{
this->loadTwitchBadges();
}
void TwitchBadges::loadTwitchBadges()
{
assert(this->loaded_ == false);
static QString url(
"https://badges.twitch.tv/v1/badges/global/display?language=en");
NetworkRequest(url)
.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();
auto root = result.parseJson();
auto badgeSets = this->badgeSets_.access();
for (auto vIt = versions.begin(); vIt != versions.end(); ++vIt)
auto jsonSets = root.value("badge_sets").toObject();
for (auto sIt = jsonSets.begin(); sIt != jsonSets.end(); ++sIt)
{
auto versionObj = vIt.value().toObject();
auto key = sIt.key();
auto versions =
sIt.value().toObject().value("versions").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{versionObj.value("title").toString()},
Url{versionObj.value("click_url").toString()}};
// "title"
// "clickAction"
for (auto vIt = versions.begin(); vIt != versions.end();
++vIt)
{
auto versionObj = vIt.value().toObject();
(*badgeSets)[key][vIt.key()] =
std::make_shared<Emote>(emote);
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{versionObj.value("title").toString()},
Url{versionObj.value("click_url").toString()}};
// "title"
// "clickAction"
(*badgeSets)[key][vIt.key()] =
std::make_shared<Emote>(emote);
}
}
}
this->loaded();
return Success;
})
.onError([this](auto res) {
qCDebug(chatterinoTwitch)
<< "Error loading Twitch Badges:" << res.status();
// Despite erroring out, we still want to reach the same point
// Loaded should still be set to true to not build up an endless queue, and the quuee should still be flushed.
this->loaded();
})
.execute();
}
void TwitchBadges::loaded()
{
std::unique_lock loadedLock(this->loadedMutex_);
assert(this->loaded_ == false);
this->loaded_ = true;
// Flush callback queue
std::unique_lock queueLock(this->queueMutex_);
while (!this->callbackQueue_.empty())
{
auto callback = this->callbackQueue_.front();
this->callbackQueue_.pop();
this->getBadgeIcon(callback.first, callback.second);
}
}
boost::optional<EmotePtr> TwitchBadges::badge(const QString &set,
const QString &version) const
{
@@ -77,4 +112,117 @@ boost::optional<EmotePtr> TwitchBadges::badge(const QString &set,
return boost::none;
}
boost::optional<EmotePtr> TwitchBadges::badge(const QString &set) const
{
auto badgeSets = this->badgeSets_.access();
auto it = badgeSets->find(set);
if (it != badgeSets->end())
{
if (it->second.size() > 0)
{
return it->second.begin()->second;
}
}
return boost::none;
}
void TwitchBadges::getBadgeIcon(const QString &name, BadgeIconCallback callback)
{
{
std::shared_lock loadedLock(this->loadedMutex_);
if (!this->loaded_)
{
// Badges have not been loaded yet, store callback in a queue
std::unique_lock queueLock(this->queueMutex_);
this->callbackQueue_.push({name, std::move(callback)});
return;
}
}
{
std::shared_lock badgeLock(this->badgesMutex_);
if (this->badgesMap_.contains(name))
{
callback(name, this->badgesMap_[name]);
return;
}
}
// Split string in format "name1/version1,name2/version2" to "name1", "version1"
// If not in list+version form, name will remain the same
auto targetBadge = name.split(",").at(0).split("/");
const auto badge = targetBadge.size() == 2
? this->badge(targetBadge.at(0), targetBadge.at(1))
: this->badge(targetBadge.at(0));
if (badge)
{
this->loadEmoteImage(name, (*badge)->images.getImage3(),
std::move(callback));
}
}
void TwitchBadges::getBadgeIcon(const DisplayBadge &badge,
BadgeIconCallback callback)
{
this->getBadgeIcon(badge.badgeName(), std::move(callback));
}
void TwitchBadges::getBadgeIcons(const QList<DisplayBadge> &badges,
BadgeIconCallback callback)
{
for (const auto &item : badges)
{
this->getBadgeIcon(item, callback);
}
}
void TwitchBadges::loadEmoteImage(const QString &name, ImagePtr image,
BadgeIconCallback &&callback)
{
NetworkRequest(image->url().string)
.concurrent()
.cache()
.onSuccess([this, name, callback](auto result) -> Outcome {
auto data = result.getData();
// const cast since we are only reading from it
QBuffer buffer(const_cast<QByteArray *>(&data));
buffer.open(QIODevice::ReadOnly);
QImageReader reader(&buffer);
QImage image;
if (reader.imageCount() == 0 || !reader.read(&image))
{
return Failure;
}
auto icon = std::make_shared<QIcon>(QPixmap::fromImage(image));
{
std::unique_lock lock(this->badgesMutex_);
this->badgesMap_[name] = icon;
}
callback(name, icon);
return Success;
})
.execute();
}
TwitchBadges *TwitchBadges::instance_;
TwitchBadges *TwitchBadges::instance()
{
if (TwitchBadges::instance_ == nullptr)
{
TwitchBadges::instance_ = new TwitchBadges();
}
return TwitchBadges::instance_;
}
} // namespace chatterino