File indexing completed on 2024-05-12 05:07:56

 /*
     SPDX-FileCopyrightText: 2010-2014 Cristian Oneț <onet.cristian@gmail.com>
     SPDX-FileCopyrightText: 2017-2018 Łukasz Wojniłowicz <lukasz.wojnilowicz@gmail.com>
     SPDX-FileCopyrightText: 2019 Thomas Baumgart <tbaumgart@kde.org>
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "kmymoneyaccounttreeview.h"
 
 // ----------------------------------------------------------------------------
 // QT Includes
 
 #include <QObject>
 #include <QHeaderView>
 #include <QMouseEvent>
 #include <QPoint>
 
 // ----------------------------------------------------------------------------
 // KDE Includes
 
 #include <KConfigGroup>
 #include <KSharedConfig>
 
 // ----------------------------------------------------------------------------
 // Project Includes
 
 #include "mymoneyfile.h"
 #include "mymoneyinstitution.h"
 #include "mymoneyaccount.h"
 #include "accountsmodel.h"
 #include "institutionsmodel.h"
 #include "accountsproxymodel.h"
 #include "menuenums.h"
 #include "selectedobjects.h"
 
 class KMyMoneyAccountTreeViewPrivate
 {
     Q_DECLARE_PUBLIC(KMyMoneyAccountTreeView)
     Q_DISABLE_COPY_MOVE(KMyMoneyAccountTreeViewPrivate)
 
 public:
     KMyMoneyAccountTreeViewPrivate(KMyMoneyAccountTreeView *qq)
         : q_ptr(qq),
           proxyModel(new AccountsProxyModel(qq))
     {}
 
     void openIndex(const QModelIndex &index)
     {
         Q_Q(KMyMoneyAccountTreeView);
         if (index.isValid()) {
             QModelIndex baseIdx = MyMoneyFile::baseModel()->mapToBaseSource(index);
             // baseIdx could point into the accountsModel or the institutionsModel.
             // in case of the institutionsModel it is unclear if it is an account or
             // an institution. So we simply extract the id and check where we find
             // the object.
             const auto objId = baseIdx.data(eMyMoney::Model::IdRole).toString();
             auto idx = MyMoneyFile::instance()->accountsModel()->indexById(objId);
             if (idx.isValid() || (objId == MyMoneyAccount::stdAccName(eMyMoney::Account::Standard::Favorite))) {
                 // In case we should open an account or category we do it
                 // For the top level accounts we switch between expanded and collapsed view
                 if (index.parent().isValid()) {
                     Q_EMIT q->requestActionTrigger(eMenu::Action::OpenAccount);
                 } else {
                     q->setExpanded(index, !q->isExpanded(index));
                 }
             } else {
                 idx = MyMoneyFile::instance()->institutionsModel()->indexById(objId);
                 if (idx.isValid()) {
                     Q_EMIT q->requestActionTrigger(eMenu::Action::EditInstitution);
                 }
             }
         }
     }
 
     KMyMoneyAccountTreeView*  q_ptr;
     AccountsProxyModel*       proxyModel;
 };
 
 KMyMoneyAccountTreeView::KMyMoneyAccountTreeView(QWidget *parent)
     : KMyMoneyTreeView(parent)
     , d_ptr(new KMyMoneyAccountTreeViewPrivate(this))
 {
     setContextMenuPolicy(Qt::CustomContextMenu);            // allow context menu to be opened on tree items
     connect(this, &QWidget::customContextMenuRequested, this, &KMyMoneyAccountTreeView::customContextMenuRequested);
     setAllColumnsShowFocus(true);
     setAlternatingRowColors(true);
     setIconSize(QSize(22, 22));
     setSortingEnabled(true);
     QTreeView::header()->setStretchLastSection(false);
 
     connect(this, &KMyMoneyTreeView::startEdit, this, [&](const QModelIndex& idx) {
         Q_D(KMyMoneyAccountTreeView);
         d->openIndex(idx);
     });
 }
 
 KMyMoneyAccountTreeView::~KMyMoneyAccountTreeView()
 {
 }
 
 
 void KMyMoneyAccountTreeView::setModel(QAbstractItemModel* model)
 {
     Q_D(KMyMoneyAccountTreeView);
     d->proxyModel->setSourceModel(model);
     QTreeView::setModel(d->proxyModel);
 
     for (int column = 0; column < model->columnCount(); column++) {
         QVariant headerData = model->headerData(column, Qt::Horizontal, Qt::UserRole);
         QHeaderView::ResizeMode mode = QHeaderView::ResizeMode(headerData.toInt());
         QTreeView::header()->setSectionResizeMode(column, mode);
     }
 }
 
 void KMyMoneyAccountTreeView::setProxyModel(AccountsProxyModel* model)
 {
     Q_D(KMyMoneyAccountTreeView);
     // unlink a possible sourceModel
     QAbstractItemModel* sourceModel = d->proxyModel->sourceModel();
     if (sourceModel) {
         d->proxyModel->setSourceModel(nullptr);
     }
 
     // delete the old proxy
     d->proxyModel->deleteLater();
 
     // reparent the new proxy
     model->setParent(this);
 
     // and insert it into the chain
     d->proxyModel = model;
     d->proxyModel->setSourceModel(sourceModel);
     QTreeView::setModel(d->proxyModel);
 
 }
 
 AccountsProxyModel* KMyMoneyAccountTreeView::proxyModel() const
 {
     Q_D(const KMyMoneyAccountTreeView);
     return d->proxyModel;
 }
 
 void KMyMoneyAccountTreeView::customContextMenuRequested(const QPoint pos)
 {
     if (currentIndex().isValid() && (model()->flags(currentIndex()) & Qt::ItemIsSelectable)) {
         auto objId = currentIndex().data(eMyMoney::Model::IdRole).toString();
         const auto account = MyMoneyFile::instance()->accountsModel()->itemById(objId);
         if (!account.id().isEmpty()) {
             if (account.isIncomeExpense()) {
                 Q_EMIT requestCustomContextMenu(eMenu::Menu::Category, viewport()->mapToGlobal(pos));
             } else {
                 Q_EMIT requestCustomContextMenu(eMenu::Menu::Account, viewport()->mapToGlobal(pos));
             }
         }
         const auto institution = MyMoneyFile::instance()->institutionsModel()->itemById(objId);
         if (!institution.id().isEmpty()) {
             // the institutions model also reports accounts as institutions at this point.
             // We can differentiate between the two objects by looking at the parent of
             // the index. If it is valid, we have been called via an account,
             // if it is invalid the source is an institution.
             if (!currentIndex().parent().isValid()) {
                 Q_EMIT requestCustomContextMenu(eMenu::Menu::Institution, viewport()->mapToGlobal(pos));
             }
         }
     }
 }
 
 void KMyMoneyAccountTreeView::selectionChanged(const QItemSelection &selected, const QItemSelection &deselected)
 {
     SelectedObjects selections;
 
     QTreeView::selectionChanged(selected, deselected);
     if (!selected.isEmpty()) {
         QModelIndexList idxList = selected.indexes();
         if (!idxList.isEmpty()) {
             auto objId = selected.indexes().front().data(eMyMoney::Model::IdRole).toString();
             const auto account = MyMoneyFile::instance()->accountsModel()->itemById(objId);
             if (!account.id().isEmpty()) {
                 selections.addSelection(SelectedObjects::Account, account.id());
                 if (!account.institutionId().isEmpty()) {
                     selections.addSelection(SelectedObjects::Institution, account.institutionId());
                 }
             } else {
 
                 const auto institution = MyMoneyFile::instance()->institutionsModel()->itemById(objId);
                 if (!institution.id().isEmpty()) {
                     selections.addSelection(SelectedObjects::Institution, institution.id());
                 }
             }
         }
     }
     // since no object was selected reset the object selection
     Q_EMIT requestSelectionChange(selections);
 }