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

Co-authored-by: Paweł <zneix@zneix.eu> Co-authored-by: 23rd <23rd@vivaldi.net> Co-authored-by: pajlada <rasmus.karlsson@pajlada.com>
2021-05-02 18:08:08 -04:00
parent 6ab5b13017
commit f6d9fb2aac
34 changed files with 956 additions and 115 deletions
@@ -0,0 +1,71 @@
+#include "BadgePickerDialog.hpp"
+#include <QSizePolicy>
+#include "singletons/Resources.hpp"
+
+#include "providers/twitch/TwitchBadges.hpp"
+
+namespace chatterino {
+
+BadgePickerDialog::BadgePickerDialog(QList<DisplayBadge> badges,
+                                     QWidget *parent)
+    : QDialog(parent)
+{
+    this->dropdown_ = new QComboBox;
+    auto vbox = new QVBoxLayout(this);
+    auto buttonBox =
+        new QDialogButtonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel);
+
+    vbox->addWidget(this->dropdown_);
+    vbox->addWidget(buttonBox);
+
+    QObject::connect(buttonBox, &QDialogButtonBox::accepted, [this] {
+        this->accept();
+        this->close();
+    });
+    QObject::connect(buttonBox, &QDialogButtonBox::rejected, [this] {
+        this->reject();
+        this->close();
+    });
+
+    this->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Fixed);
+
+    this->setWindowFlags(
+        (this->windowFlags() & ~(Qt::WindowContextHelpButtonHint)) |
+        Qt::Dialog | Qt::MSWindowsFixedSizeDialogHint);
+
+    // Add items.
+    for (const auto &item : badges)
+    {
+        this->dropdown_->addItem(item.displayName(), item.badgeName());
+    }
+
+    const auto updateBadge = [=](int index) {
+        BadgeOpt badge;
+        if (index >= 0 && index < badges.size())
+        {
+            badge = badges[index];
+        }
+        this->currentBadge_ = badge;
+    };
+
+    QObject::connect(this->dropdown_,
+                     QOverload<int>::of(&QComboBox::currentIndexChanged),
+                     updateBadge);
+    updateBadge(0);
+
+    // Set icons.
+    TwitchBadges::instance()->getBadgeIcons(
+        badges,
+        [&dropdown = this->dropdown_](QString identifier, const QIconPtr icon) {
+            if (!dropdown)
+                return;
+
+            int index = dropdown->findData(identifier);
+            if (index != -1)
+            {
+                dropdown->setItemIcon(index, *icon);
+            }
+        });
+}
+
+}  // namespace chatterino