1 files changed, 123 insertions, 0 deletions
diff --git a/ui/qt/models/coloring_rules_delegate.cpp b/ui/qt/models/coloring_rules_delegate.cpp
new file mode 100644
index 0000000..b86ffa9
--- /dev/null
+++ b/ui/qt/models/coloring_rules_delegate.cpp
@@ -0,0 +1,123 @@
+/* coloring_rules_delegate.cpp
+ * Delegates for editing various coloring rule fields.
+ *
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#include <ui/qt/models/coloring_rules_delegate.h>
+#include <ui/qt/models/coloring_rules_model.h>
+#include <ui/qt/widgets/display_filter_edit.h>
+
+ColoringRulesDelegate::ColoringRulesDelegate(QObject *parent) : QStyledItemDelegate(parent)
+{
+}
+
+QWidget* ColoringRulesDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem&,
+                                  const QModelIndex &index) const
+{
+    switch (index.column())
+    {
+    case ColoringRulesModel::colName:
+    {
+        SyntaxLineEdit *editor = new SyntaxLineEdit(parent);
+        connect(editor, &SyntaxLineEdit::textChanged, this, &ColoringRulesDelegate::ruleNameChanged);
+        return editor;
+    }
+
+    case ColoringRulesModel::colFilter:
+        return new DisplayFilterEdit(parent);
+
+    default:
+        Q_ASSERT(FALSE);
+        return 0;
+    }
+
+    return 0;
+}
+
+void ColoringRulesDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const
+{
+    switch (index.column())
+    {
+    case ColoringRulesModel::colName:
+    {
+        SyntaxLineEdit *syntaxEdit = static_cast<SyntaxLineEdit*>(editor);
+        syntaxEdit->setText(index.model()->data(index, Qt::EditRole).toString());
+        break;
+    }
+    case ColoringRulesModel::colFilter:
+    {
+        DisplayFilterEdit *displayEdit = static_cast<DisplayFilterEdit*>(editor);
+        displayEdit->setText(index.model()->data(index, Qt::EditRole).toString());
+        break;
+    }
+    default:
+        QStyledItemDelegate::setEditorData(editor, index);
+        break;
+    }
+}
+
+void ColoringRulesDelegate::setModelData(QWidget *editor, QAbstractItemModel *model,
+                              const QModelIndex &index) const
+{
+    switch (index.column())
+    {
+    case ColoringRulesModel::colName:
+    {
+        SyntaxLineEdit *syntaxEdit = static_cast<SyntaxLineEdit*>(editor);
+        model->setData(index, syntaxEdit->text(), Qt::EditRole);
+        if (syntaxEdit->syntaxState() == SyntaxLineEdit::Invalid) {
+            QString error_text = tr("the \"@\" symbol will be ignored.");
+            emit invalidField(index, error_text);
+        }
+        else
+        {
+            emit validField(index);
+        }
+        break;
+    }
+    case ColoringRulesModel::colFilter:
+    {
+        DisplayFilterEdit *displayEdit = static_cast<DisplayFilterEdit*>(editor);
+        model->setData(index, displayEdit->text(), Qt::EditRole);
+        if ((displayEdit->syntaxState() == SyntaxLineEdit::Invalid) &&
+            (model->data(model->index(index.row(), ColoringRulesModel::colName), Qt::CheckStateRole) == Qt::Checked))
+        {
+            model->setData(model->index(index.row(), ColoringRulesModel::colName), Qt::Unchecked, Qt::CheckStateRole);
+            emit invalidField(index, displayEdit->syntaxErrorMessage());
+        }
+        else
+        {
+            emit validField(index);
+        }
+        break;
+    }
+    default:
+        QStyledItemDelegate::setModelData(editor, model, index);
+        break;
+    }
+}
+
+void ColoringRulesDelegate::updateEditorGeometry(QWidget *editor,
+        const QStyleOptionViewItem &option, const QModelIndex&) const
+{
+    editor->setGeometry(option.rect);
+}
+
+void ColoringRulesDelegate::ruleNameChanged(const QString name)
+{
+    SyntaxLineEdit *name_edit = qobject_cast<SyntaxLineEdit*>(QObject::sender());
+    if (!name_edit) return;
+
+    if (name.isEmpty()) {
+        name_edit->setSyntaxState(SyntaxLineEdit::Empty);
+    } else if (name.contains("@")) {
+        name_edit->setSyntaxState(SyntaxLineEdit::Invalid);
+    } else {
+        name_edit->setSyntaxState(SyntaxLineEdit::Valid);
+    }
+}