From 011e4d522ed7b0df52cdeb014608e9881f8bc76b Mon Sep 17 00:00:00 2001
From: Mikaël Capelle <capelle.mikael@gmail.com>
Date: Fri, 1 Jan 2021 17:17:56 +0100
Subject: Fix setting the primary category of multiple mods at once.

---
 src/modlistviewactions.cpp | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

(limited to 'src/modlistviewactions.cpp')

diff --git a/src/modlistviewactions.cpp b/src/modlistviewactions.cpp
index 8d24a8d5..cb212e0b 100644
--- a/src/modlistviewactions.cpp
+++ b/src/modlistviewactions.cpp
@@ -956,7 +956,6 @@ void ModListViewActions::setCategories(const QModelIndexList& selected, const QM
 {
   ModInfo::Ptr refMod = ModInfo::getByIndex(ref.data(ModList::IndexRole).toInt());
   if (selected.size() > 1) {
-
     for (auto& idx : selected) {
       if (idx.row() != ref.row()) {
         setCategoriesIf(
@@ -982,6 +981,23 @@ void ModListViewActions::setCategories(const QModelIndexList& selected, const QM
   }
 }
 
+void ModListViewActions::setPrimaryCategory(const QModelIndexList& selected, int category, bool force)
+{
+  for (auto& idx : selected) {
+    ModInfo::Ptr info = ModInfo::getByIndex(idx.data(ModList::IndexRole).toInt());
+    if (force || info->categorySet(category)) {
+      info->setCategory(category, true);
+      info->setPrimaryCategory(category);
+    }
+  }
+
+  // reset the selection manually - still needed
+  auto viewIndices = m_view->indexModelToView(selected);
+  for (auto& idx : viewIndices) {
+    m_view->selectionModel()->select(idx, QItemSelectionModel::Select | QItemSelectionModel::Rows);
+  }
+}
+
 void ModListViewActions::openExplorer(const QModelIndexList& index) const
 {
   for (auto& idx : index) {
-- 
cgit v1.3.1