Move showColumnsContextMenu function from kpmcore to partitionmanager.

src/gui/partitionmanagerwidget.cpp

@@ -24,6 +24,7 @@
 #include "gui/applyprogressdialog.h"
 #include "gui/insertdialog.h"
 #include "gui/editmountpointdialog.h"
+#include "util/guihelpers.h"
 
 #include <core/partition.h>
 #include <core/device.h>
@@ -51,8 +52,6 @@
 #include <util/report.h>
 #include <util/helpers.h>
 
-#include "util/guihelpers.h"
-
 #include <QFileDialog>
 #include <QLocale>
 #include <QPointer>

src/gui/smartdialogwidget.cpp

@@ -17,7 +17,7 @@
 
 #include "gui/smartdialogwidget.h"
 
-#include <util/helpers.h>
+#include "util/guihelpers.h"
 
 #include <KLocalizedString>
 

src/gui/treelog.cpp

@@ -18,10 +18,10 @@
 
 #include "gui/treelog.h"
 #include "gui/partitionmanagerwidget.h"
+#include "util/guihelpers.h"
 #include "config.h"
 
 #include <util/globallog.h>
-#include <util/helpers.h>
 
 #include <QDateTime>
 #include <QDebug>

src/util/guihelpers.cpp

@@ -21,14 +21,19 @@
 
 #include <backend/corebackendmanager.h>
 
+#include <QAction>
 #include <QApplication>
 #include <QFileInfo>
 #include <QIcon>
+#include <QHeaderView>
+#include <QMenu>
 #include <QPainter>
 #include <QPixmap>
 #include <QProcess>
+#include <QRect>
 #include <QStandardPaths>
 #include <QString>
+#include <QTreeWidget>
 
 #include <KLocalizedString>
 #include <KMessageBox>
@@ -167,6 +172,33 @@ Capacity::Unit preferredUnit()
     return static_cast<Capacity::Unit>(Config::preferredUnit());
 }
 
+void showColumnsContextMenu(const QPoint& p, QTreeWidget& tree)
+{
+    QMenu headerMenu(xi18nc("@title:menu", "Columns"));
+
+    QHeaderView* header = tree.header();
+
+    for (qint32 i = 0; i < tree.model()->columnCount(); i++) {
+        const int idx = header->logicalIndex(i);
+        const QString text = tree.model()->headerData(idx, Qt::Horizontal).toString();
+
+        QAction* action = headerMenu.addAction(text);
+        action->setCheckable(true);
+        action->setChecked(!header->isSectionHidden(idx));
+        action->setData(idx);
+        action->setEnabled(idx > 0);
+    }
+
+    QAction* action = headerMenu.exec(tree.header()->mapToGlobal(p));
+
+    if (action != nullptr) {
+        const bool hidden = !action->isChecked();
+        tree.setColumnHidden(action->data().toInt(), hidden);
+        if (!hidden)
+            tree.resizeColumnToContents(action->data().toInt());
+    }
+}
+
 namespace GuiHelpers
 {
 

src/util/guihelpers.h

@@ -23,13 +23,16 @@
 #include <fs/filesystem.h>
 #include <util/capacity.h>
 
-class QString;
 class QIcon;
+class QPoint;
+class QString;
+class QTreeWidget;
 
 bool checkPermissions();
 bool loadBackend();
 QIcon createFileSystemColor(FileSystem::Type type, quint32 size);
 Capacity::Unit preferredUnit();
+void showColumnsContextMenu(const QPoint& p, QTreeWidget& tree);
 namespace GuiHelpers
 {
 FileSystem::Type defaultFileSystem();