#pragma once

#include "common/SignalVectorModel.hpp"
#include "controllers/highlights/HighlightBadge.hpp"
#include "providers/twitch/TwitchBadges.hpp"

#include <QObject>

namespace chatterino {

class HighlightController;

class BadgeHighlightModel : public SignalVectorModel<HighlightBadge>
{
public:
    explicit BadgeHighlightModel(QObject *parent);

    enum Column {
        Badge = 0,
        ShowInMentions = 1,
        FlashTaskbar = 2,
        PlaySound = 3,
        SoundPath = 4,
        Color = 5
    };

protected:
    // vector into model row
    virtual HighlightBadge getItemFromRow(
        std::vector<QStandardItem *> &row,
        const HighlightBadge &original) override;

    virtual void getRowFromItem(const HighlightBadge &item,
                                std::vector<QStandardItem *> &row) override;
};

}  // namespace chatterino
