Refactor highlight blacklist to use the model system

Fix #560

src/controllers/highlights/HighlightBlacklistModel.cpp

@@ -0,0 +1,33 @@
+#include "controllers/highlights/HighlightBlacklistModel.hpp"
+
+#include "Application.hpp"
+#include "singletons/Settings.hpp"
+#include "util/StandardItemHelper.hpp"
+
+namespace chatterino {
+
+// commandmodel
+HighlightBlacklistModel::HighlightBlacklistModel(QObject *parent)
+    : SignalVectorModel<HighlightBlacklistUser>(2, parent)
+{
+}
+
+// turn a vector item into a model row
+HighlightBlacklistUser HighlightBlacklistModel::getItemFromRow(
+    std::vector<QStandardItem *> &row, const HighlightBlacklistUser &original)
+{
+    // key, regex
+
+    return HighlightBlacklistUser{row[0]->data(Qt::DisplayRole).toString(),
+                                  row[1]->data(Qt::CheckStateRole).toBool()};
+}
+
+// turns a row in the model into a vector item
+void HighlightBlacklistModel::getRowFromItem(const HighlightBlacklistUser &item,
+                                             std::vector<QStandardItem *> &row)
+{
+    setStringItem(row[0], item.getPattern());
+    setBoolItem(row[1], item.isRegex());
+}
+
+}  // namespace chatterino

src/controllers/highlights/HighlightBlacklistModel.hpp

@@ -0,0 +1,28 @@
+#pragma once
+
+#include <QObject>
+
+#include "common/SignalVectorModel.hpp"
+#include "controllers/highlights/HighlightBlacklistUser.hpp"
+
+namespace chatterino {
+
+class HighlightController;
+
+class HighlightBlacklistModel : public SignalVectorModel<HighlightBlacklistUser>
+{
+    explicit HighlightBlacklistModel(QObject *parent);
+
+protected:
+    // turn a vector item into a model row
+    virtual HighlightBlacklistUser getItemFromRow(std::vector<QStandardItem *> &row,
+                                                  const HighlightBlacklistUser &original) override;
+
+    // turns a row in the model into a vector item
+    virtual void getRowFromItem(const HighlightBlacklistUser &item,
+                                std::vector<QStandardItem *> &row) override;
+
+    friend class HighlightController;
+};
+
+}  // namespace chatterino

src/controllers/highlights/HighlightBlacklistUser.hpp

@@ -0,0 +1,101 @@
+#pragma once
+
+#include "common/SerializeCustom.hpp"
+#include "util/RapidjsonHelpers.hpp"
+
+#include <QRegularExpression>
+#include <QString>
+#include <pajlada/settings/serialize.hpp>
+
+#include <memory>
+
+namespace chatterino {
+
+class HighlightBlacklistUser
+{
+    QString pattern_;
+    bool isRegex_;
+    QRegularExpression regex_;
+
+public:
+    bool operator==(const HighlightBlacklistUser &other) const
+    {
+        return std::tie(this->pattern_, this->isRegex_) == std::tie(other.pattern_, other.isRegex_);
+    }
+
+    HighlightBlacklistUser(const QString &pattern, bool isRegex = false)
+        : pattern_(pattern)
+        , isRegex_(isRegex)
+        , regex_(isRegex ? pattern : "", QRegularExpression::CaseInsensitiveOption |
+                                             QRegularExpression::UseUnicodePropertiesOption)
+    {
+    }
+
+    const QString &getPattern() const
+    {
+        return this->pattern_;
+    }
+
+    bool isRegex() const
+    {
+        return this->isRegex_;
+    }
+
+    bool isValidRegex() const
+    {
+        return this->isRegex() && this->regex_.isValid();
+    }
+
+    bool isMatch(const QString &subject) const
+    {
+        if (this->isRegex()) {
+            if (this->isValidRegex()) {
+                return this->regex_.match(subject).hasMatch();
+            }
+
+            return false;
+        }
+
+        return subject.toLower() == this->pattern_.toLower();
+    }
+};
+
+}  // namespace chatterino
+
+namespace pajlada {
+namespace Settings {
+
+template <>
+struct Serialize<chatterino::HighlightBlacklistUser> {
+    static rapidjson::Value get(const chatterino::HighlightBlacklistUser &value,
+                                rapidjson::Document::AllocatorType &a)
+    {
+        rapidjson::Value ret(rapidjson::kObjectType);
+
+        AddMember(ret, "pattern", value.getPattern(), a);
+        AddMember(ret, "regex", value.isRegex(), a);
+
+        return ret;
+    }
+};
+
+template <>
+struct Deserialize<chatterino::HighlightBlacklistUser> {
+    static chatterino::HighlightBlacklistUser get(const rapidjson::Value &value)
+    {
+        QString pattern;
+        bool isRegex = false;
+
+        if (!value.IsObject()) {
+            return chatterino::HighlightBlacklistUser(pattern, isRegex);
+        }
+
+        chatterino::rj::getSafe(value, "pattern", pattern);
+        chatterino::rj::getSafe(value, "regex", isRegex);
+
+        return chatterino::HighlightBlacklistUser(pattern, isRegex);
+    }
+};
+
+}  // namespace Settings
+}  // namespace pajlada

src/controllers/highlights/HighlightController.cpp

@@ -1,6 +1,7 @@
 #include "HighlightController.hpp"
 
 #include "Application.hpp"
+#include "controllers/highlights/HighlightBlacklistModel.hpp"
 #include "controllers/highlights/HighlightModel.hpp"
 #include "widgets/dialogs/NotificationPopup.hpp"
 
@@ -32,6 +33,26 @@ HighlightModel *HighlightController::createModel(QObject *parent)
     return model;
 }
 
+HighlightBlacklistModel *HighlightController::createBlacklistModel(QObject *parent)
+{
+    auto *model = new HighlightBlacklistModel(parent);
+    model->init(&this->blacklistedUsers);
+
+    return model;
+}
+
+bool HighlightController::blacklistContains(const QString &username)
+{
+    std::vector<HighlightBlacklistUser> blacklistItems = this->blacklistedUsers.getVector();
+    for (const auto &blacklistedUser : blacklistItems) {
+        if (blacklistedUser.isMatch(username)) {
+            return true;
+        }
+    }
+
+    return false;
+}
+
 void HighlightController::addHighlight(const MessagePtr &msg)
 {
     //    static NotificationPopup popup;

src/controllers/highlights/HighlightController.hpp

@@ -1,6 +1,7 @@
 #pragma once
 
 #include "common/SignalVector.hpp"
+#include "controllers/highlights/HighlightBlacklistUser.hpp"
 #include "controllers/highlights/HighlightPhrase.hpp"
 #include "messages/Message.hpp"
 #include "singletons/Settings.hpp"
@@ -8,6 +9,7 @@
 namespace chatterino {
 
 class HighlightModel;
+class HighlightBlacklistModel;
 
 class HighlightController
 {
@@ -17,8 +19,12 @@ public:
     void initialize();
 
     UnsortedSignalVector<HighlightPhrase> phrases;
+    UnsortedSignalVector<HighlightBlacklistUser> blacklistedUsers;
 
     HighlightModel *createModel(QObject *parent);
+    HighlightBlacklistModel *createBlacklistModel(QObject *parent);
+
+    bool blacklistContains(const QString &username);
 
     void addHighlight(const MessagePtr &msg);
 
@@ -27,6 +33,7 @@ private:
 
     ChatterinoSetting<std::vector<HighlightPhrase>> highlightsSetting = {
         "/highlighting/highlights"};
+    ChatterinoSetting<std::vector<HighlightPhrase>> blacklistSetting = {"/highlighting/blacklist"};
 };
 
 }  // namespace chatterino