Files
chatterino2/src/controllers/highlights/UserHighlightModel.cpp

130 lines
4.7 KiB
C++

#include "controllers/highlights/UserHighlightModel.hpp"
#include "Application.hpp"
#include "controllers/highlights/HighlightPhrase.hpp"
#include "providers/colors/ColorProvider.hpp"
#include "singletons/Settings.hpp"
#include "singletons/WindowManager.hpp"
#include "util/StandardItemHelper.hpp"
namespace chatterino {
// commandmodel
UserHighlightModel::UserHighlightModel(QObject *parent)
: SignalVectorModel<HighlightPhrase>(Column::COUNT, parent)
{
}
// turn vector item into model row
HighlightPhrase UserHighlightModel::getItemFromRow(
std::vector<QStandardItem *> &row, const HighlightPhrase &original)
{
// In order for old messages to update their highlight color, we need to
// update the highlight color here.
auto highlightColor = original.getColor();
*highlightColor =
row[Column::Color]->data(Qt::DecorationRole).value<QColor>();
return HighlightPhrase{
row[Column::Pattern]->data(Qt::DisplayRole).toString().trimmed(),
row[Column::ShowInMentions]->data(Qt::CheckStateRole).toBool(),
row[Column::FlashTaskbar]->data(Qt::CheckStateRole).toBool(),
row[Column::PlaySound]->data(Qt::CheckStateRole).toBool(),
row[Column::UseRegex]->data(Qt::CheckStateRole).toBool(),
row[Column::CaseSensitive]->data(Qt::CheckStateRole).toBool(),
row[Column::SoundPath]->data(Qt::UserRole).toString(),
highlightColor};
}
void UserHighlightModel::afterInit()
{
// User highlight settings for your own messages
std::vector<QStandardItem *> messagesRow = this->createRow();
setBoolItem(messagesRow[Column::Pattern],
getSettings()->enableSelfMessageHighlight.getValue(), true,
false);
messagesRow[Column::Pattern]->setData("Your messages (automatic)",
Qt::DisplayRole);
setBoolItem(messagesRow[Column::ShowInMentions],
getSettings()->showSelfMessageHighlightInMentions.getValue(),
true, false);
messagesRow[Column::FlashTaskbar]->setFlags({});
messagesRow[Column::PlaySound]->setFlags({});
messagesRow[Column::UseRegex]->setFlags({});
messagesRow[Column::CaseSensitive]->setFlags({});
messagesRow[Column::SoundPath]->setFlags({});
auto selfColor =
ColorProvider::instance().color(ColorType::SelfMessageHighlight);
setColorItem(messagesRow[Column::Color], *selfColor, false);
this->insertCustomRow(
messagesRow, HighlightModel::UserHighlightRowIndexes::SelfMessageRow);
}
void UserHighlightModel::customRowSetData(
const std::vector<QStandardItem *> &row, int column, const QVariant &value,
int role, int rowIndex)
{
switch (column)
{
case Column::Pattern: {
if (role == Qt::CheckStateRole)
{
if (rowIndex ==
HighlightModel::UserHighlightRowIndexes::SelfMessageRow)
{
getSettings()->enableSelfMessageHighlight.setValue(
value.toBool());
}
}
}
break;
case Column::ShowInMentions: {
if (role == Qt::CheckStateRole)
{
if (rowIndex ==
HighlightModel::UserHighlightRowIndexes::SelfMessageRow)
{
getSettings()->showSelfMessageHighlightInMentions.setValue(
value.toBool());
}
}
}
break;
case Column::Color: {
// Custom color
if (role == Qt::DecorationRole)
{
auto colorName = value.value<QColor>().name(QColor::HexArgb);
if (rowIndex ==
HighlightModel::UserHighlightRowIndexes::SelfMessageRow)
{
// Update the setting with the new value
getSettings()->selfMessageHighlightColor.setValue(
colorName);
}
}
}
break;
}
getApp()->getWindows()->forceLayoutChannelViews();
}
// row into vector item
void UserHighlightModel::getRowFromItem(const HighlightPhrase &item,
std::vector<QStandardItem *> &row)
{
setStringItem(row[Column::Pattern], item.getPattern());
setBoolItem(row[Column::ShowInMentions], item.showInMentions());
setBoolItem(row[Column::FlashTaskbar], item.hasAlert());
setBoolItem(row[Column::PlaySound], item.hasSound());
setBoolItem(row[Column::UseRegex], item.isRegex());
setBoolItem(row[Column::CaseSensitive], item.isCaseSensitive());
setFilePathItem(row[Column::SoundPath], item.getSoundUrl());
setColorItem(row[Column::Color], *item.getColor());
}
} // namespace chatterino