Add search predicates for badges and sub tiers (#4013)

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: Rasmus Karlsson <rasmus.karlsson@pajlada.com>
2022-10-01 08:30:29 -04:00
parent b5241670ae
commit bfcc9ff7a4
7 changed files with 191 additions and 0 deletions
@@ -0,0 +1,48 @@
+#include "messages/search/BadgePredicate.hpp"
+
+#include "util/Qt.hpp"
+
+namespace chatterino {
+
+BadgePredicate::BadgePredicate(const QStringList &badges)
+{
+    // Check if any comma-seperated values were passed and transform those
+    for (const auto &entry : badges)
+    {
+        for (const auto &badge : entry.split(',', Qt::SkipEmptyParts))
+        {
+            // convert short form name of certain badges to formal name
+            if (badge.compare("mod", Qt::CaseInsensitive) == 0)
+            {
+                this->badges_ << "moderator";
+            }
+            else if (badge.compare("sub", Qt::CaseInsensitive) == 0)
+            {
+                this->badges_ << "subscriber";
+            }
+            else if (badge.compare("prime", Qt::CaseInsensitive) == 0)
+            {
+                this->badges_ << "premium";
+            }
+            else
+            {
+                this->badges_ << badge;
+            }
+        }
+    }
+}
+
+bool BadgePredicate::appliesTo(const Message &message)
+{
+    for (const Badge &badge : message.badges)
+    {
+        if (badges_.contains(badge.key_, Qt::CaseInsensitive))
+        {
+            return true;
+        }
+    }
+
+    return false;
+}
+
+}  // namespace chatterino