1 files changed, 119 insertions, 0 deletions
diff --git a/libs/installer_bsplugins/src/BSPluginList/PluginListStyledItemDelegate.cpp b/libs/installer_bsplugins/src/BSPluginList/PluginListStyledItemDelegate.cpp
new file mode 100644
index 0000000..141ef17
--- /dev/null
+++ b/libs/installer_bsplugins/src/BSPluginList/PluginListStyledItemDelegate.cpp
@@ -0,0 +1,119 @@
+#include "PluginListStyledItemDelegate.h"
+#include "PluginListModel.h"
+#include "PluginListView.h"
+
+#include <QApplication>
+
+namespace BSPluginList
+{
+
+PluginListStyledItemDelegate::PluginListStyledItemDelegate(PluginListView* view)
+    : QStyledItemDelegate(view), m_View{view}
+{}
+
+void PluginListStyledItemDelegate::paint(QPainter* painter,
+                                         const QStyleOptionViewItem& option,
+                                         const QModelIndex& index) const
+{
+  QStyleOptionViewItem opt(option);
+
+  if (index.column() == 0) {
+    if (!index.model()->hasChildren(index) || !index.data().isValid()) {
+      opt.rect.adjust(-m_View->indentation(), 0, 0, 0);
+    }
+  }
+
+  const auto color    = m_View->markerColor(index);
+  opt.backgroundBrush = color;
+
+  const auto widget = opt.widget;
+  const auto style  = widget ? widget->style() : QApplication::style();
+
+  if (!index.siblingAtColumn(0).data().isValid()) {
+    QStyleOptionFrame optFrame;
+    optFrame.initFrom(widget);
+    optFrame.rect         = opt.rect;
+    optFrame.frameShape   = QFrame::HLine;
+    optFrame.lineWidth    = 0;
+    optFrame.midLineWidth = 1;
+    optFrame.state |= QStyle::State_Sunken;
+
+    QApplication::style()->drawControl(QStyle::CE_ShapedFrame, &optFrame, painter,
+                                       widget);
+    return;
+  }
+
+  const auto plugin =
+      index.data(PluginListModel::InfoRole).value<const TESData::FileInfo*>();
+
+  bool disabledText = false;
+  switch (index.column()) {
+  case PluginListModel::COL_NAME:
+    disabledText = plugin && plugin->isAlwaysEnabled();
+    break;
+  case PluginListModel::COL_PRIORITY:
+  case PluginListModel::COL_MODINDEX:
+    disabledText = plugin && plugin->forceLoaded();
+    break;
+  }
+
+  if (disabledText) {
+    opt.palette.setBrush(QPalette::Text,
+                         opt.palette.brush(QPalette::Disabled, QPalette::Text));
+  }
+
+  // HACK: we can't normally have a disabled checkbox on a selectable row, so create a
+  // margin for it and then draw one manually
+  bool drawCheck = false;
+  bool enabled   = false;
+  if (index.column() == 0 && !index.data(Qt::CheckStateRole).isValid()) {
+    if (plugin) {
+      drawCheck = true;
+      if (plugin->forceLoaded() || plugin->forceEnabled()) {
+        enabled = true;
+      }
+    }
+
+    // set decoration size to use as text margin
+    opt.decorationSize.setWidth(
+        style->pixelMetric(QStyle::PM_IndicatorWidth, &opt, widget));
+  }
+
+  QStyledItemDelegate::paint(painter, opt, index);
+
+  // draw check on top
+  if (drawCheck) {
+    opt.features |= QStyleOptionViewItem::HasCheckIndicator;
+
+    QRect checkRect =
+        style->subElementRect(QStyle::SE_ItemViewItemCheckIndicator, &opt, widget);
+    opt.rect = checkRect;
+
+    opt.state &= ~QStyle::State_Enabled;
+    if (enabled) {
+      opt.state |= QStyle::State_On;
+    }
+
+    style->drawPrimitive(QStyle::PE_IndicatorItemViewItemCheck, &opt, painter, widget);
+  }
+}
+
+void PluginListStyledItemDelegate::initStyleOption(QStyleOptionViewItem* option,
+                                                   const QModelIndex& index) const
+{
+  const auto backgroundColor = option->backgroundBrush.color();
+  QStyledItemDelegate::initStyleOption(option, index);
+
+  // HACK: create a text margin where the checkbox should be
+  if (index.column() == 0 && !index.data(Qt::CheckStateRole).isValid()) {
+    if (index.data(PluginListModel::InfoRole).isValid()) {
+      option->features |= QStyleOptionViewItem::HasDecoration;
+    }
+  }
+
+  if (backgroundColor.isValid()) {
+    option->backgroundBrush = backgroundColor;
+  }
+}
+
+}  // namespace BSPluginList