File indexing completed on 2025-01-05 04:55:51

 /*
     ui/treewidget.cpp
 
     This file is part of libkleopatra
     SPDX-FileCopyrightText: 2022 g10 Code GmbH
     SPDX-FileContributor: Ingo Klöcker <dev@ingo-kloecker.de>
 
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include <config-libkleo.h>
 
 #include "treewidget.h"
 
 #include <KConfigGroup>
 #include <KLocalizedString>
 #include <KSharedConfig>
 
 #include <QContextMenuEvent>
 #include <QHeaderView>
 #include <QMenu>
 
 using namespace Kleo;
 
 class TreeWidget::Private
 {
     TreeWidget *q;
 
 public:
     QMenu *mHeaderPopup = nullptr;
     QList<QAction *> mColumnActions;
     QString mStateGroupName;
 
     Private(TreeWidget *qq)
         : q(qq)
     {
     }
 
     ~Private()
     {
         saveColumnLayout();
     }
     void saveColumnLayout();
 };
 
 TreeWidget::TreeWidget(QWidget *parent)
     : QTreeWidget::QTreeWidget(parent)
     , d{new Private(this)}
 {
     header()->installEventFilter(this);
 }
 
 TreeWidget::~TreeWidget() = default;
 
 void TreeWidget::Private::saveColumnLayout()
 {
     if (mStateGroupName.isEmpty()) {
         return;
     }
     auto config = KConfigGroup(KSharedConfig::openStateConfig(), mStateGroupName);
     auto header = q->header();
 
     QVariantList columnVisibility;
     QVariantList columnOrder;
     QVariantList columnWidths;
     const int headerCount = header->count();
     columnVisibility.reserve(headerCount);
     columnWidths.reserve(headerCount);
     columnOrder.reserve(headerCount);
     for (int i = 0; i < headerCount; ++i) {
         columnVisibility << QVariant(!q->isColumnHidden(i));
         columnWidths << QVariant(header->sectionSize(i));
         columnOrder << QVariant(header->visualIndex(i));
     }
 
     config.writeEntry("ColumnVisibility", columnVisibility);
     config.writeEntry("ColumnOrder", columnOrder);
     config.writeEntry("ColumnWidths", columnWidths);
 
     config.writeEntry("SortAscending", (int)header->sortIndicatorOrder());
     if (header->isSortIndicatorShown()) {
         config.writeEntry("SortColumn", header->sortIndicatorSection());
     } else {
         config.writeEntry("SortColumn", -1);
     }
 }
 
 bool TreeWidget::restoreColumnLayout(const QString &stateGroupName)
 {
     if (stateGroupName.isEmpty()) {
         return false;
     }
     d->mStateGroupName = stateGroupName;
     auto config = KConfigGroup(KSharedConfig::openStateConfig(), d->mStateGroupName);
     auto header = this->header();
 
     QVariantList columnVisibility = config.readEntry("ColumnVisibility", QVariantList());
     QVariantList columnOrder = config.readEntry("ColumnOrder", QVariantList());
     QVariantList columnWidths = config.readEntry("ColumnWidths", QVariantList());
 
     if (!columnVisibility.isEmpty() && !columnOrder.isEmpty() && !columnWidths.isEmpty()) {
         for (int i = 0; i < header->count(); ++i) {
             if (i >= columnOrder.size() || i >= columnWidths.size() || i >= columnVisibility.size()) {
                 // An additional column that was not around last time we saved.
                 // We default to hidden.
                 hideColumn(i);
                 continue;
             }
             bool visible = columnVisibility[i].toBool();
             int width = columnWidths[i].toInt();
             int order = columnOrder[i].toInt();
 
             header->resizeSection(i, width ? width : 100);
             header->moveSection(header->visualIndex(i), order);
 
             if (!visible) {
                 hideColumn(i);
             }
         }
     }
 
     int sortOrder = config.readEntry("SortAscending", (int)Qt::AscendingOrder);
     int sortColumn = config.readEntry("SortColumn", 0);
     if (sortColumn >= 0) {
         sortByColumn(sortColumn, (Qt::SortOrder)sortOrder);
     }
     return !columnVisibility.isEmpty() && !columnOrder.isEmpty() && !columnWidths.isEmpty();
 }
 
 bool TreeWidget::eventFilter(QObject *watched, QEvent *event)
 {
     Q_UNUSED(watched)
     if (event->type() == QEvent::ContextMenu) {
         auto e = static_cast<QContextMenuEvent *>(event);
 
         if (!d->mHeaderPopup) {
             d->mHeaderPopup = new QMenu(this);
             d->mHeaderPopup->setTitle(i18nc("@title:menu", "View Columns"));
             for (int i = 0; i < model()->columnCount(); ++i) {
                 QAction *tmp = d->mHeaderPopup->addAction(model()->headerData(i, Qt::Horizontal).toString());
                 tmp->setData(QVariant(i));
                 tmp->setCheckable(true);
                 d->mColumnActions << tmp;
             }
 
             connect(d->mHeaderPopup, &QMenu::triggered, this, [this](QAction *action) {
                 const int col = action->data().toInt();
                 if (action->isChecked()) {
                     showColumn(col);
                 } else {
                     hideColumn(col);
                 }
 
                 if (action->isChecked()) {
                     Q_EMIT columnEnabled(col);
                 } else {
                     Q_EMIT columnDisabled(col);
                 }
             });
         }
 
         for (QAction *action : std::as_const(d->mColumnActions)) {
             const int column = action->data().toInt();
             action->setChecked(!isColumnHidden(column));
         }
 
         d->mHeaderPopup->popup(mapToGlobal(e->pos()));
         return true;
     }
 
     return false;
 }
 
 QModelIndex TreeWidget::moveCursor(QAbstractItemView::CursorAction cursorAction, Qt::KeyboardModifiers modifiers)
 {
     // make column by column keyboard navigation with Left/Right possible by switching
     // the selection behavior to SelectItems before calling the parent class's moveCursor,
     // because it ignores MoveLeft/MoveRight if the selection behavior is SelectRows;
     // moreover, temporarily disable exanding of items to prevent expanding/collapsing
     // on MoveLeft/MoveRight
     if ((cursorAction != MoveLeft) && (cursorAction != MoveRight)) {
         return QTreeWidget::moveCursor(cursorAction, modifiers);
     }
 
     const auto savedSelectionBehavior = selectionBehavior();
     setSelectionBehavior(SelectItems);
     const auto savedItemsExpandable = itemsExpandable();
     setItemsExpandable(false);
 
     const auto result = QTreeWidget::moveCursor(cursorAction, modifiers);
 
     setItemsExpandable(savedItemsExpandable);
     setSelectionBehavior(savedSelectionBehavior);
 
     return result;
 }
 
 #include "moc_treewidget.cpp"