File indexing completed on 2024-05-19 05:06:58

 /*
     SPDX-FileCopyrightText: 2019 Thomas Baumgart <tbaumgart@kde.org>
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 
 #include "onlinebalanceproxymodel.h"
 #include "accountsmodel.h"
 #include "mymoneyfile.h"
 #include "securitiesmodel.h"
 
 // ----------------------------------------------------------------------------
 // QT Includes
 
 #include <QDate>
 
 // ----------------------------------------------------------------------------
 // KDE Includes
 
 #include <KLocalizedString>
 
 // ----------------------------------------------------------------------------
 // Project Includes
 
 #include "mymoneymoney.h"
 #include "journalmodel.h"
 #include "accountsmodel.h"
 #include "mymoneyenums.h"
 
 class OnlineBalanceProxyModelPrivate
 {
     Q_DECLARE_PUBLIC(OnlineBalanceProxyModel);
     OnlineBalanceProxyModel* q_ptr;
 
 public:
     OnlineBalanceProxyModelPrivate(OnlineBalanceProxyModel* qq)
         : q_ptr(qq)
     {
     }
 
     ~OnlineBalanceProxyModelPrivate() {}
 };
 
 OnlineBalanceProxyModel::OnlineBalanceProxyModel(QObject *parent)
     : QSortFilterProxyModel(parent)
     , d_ptr(new OnlineBalanceProxyModelPrivate(this))
 {
 }
 
 OnlineBalanceProxyModel::~OnlineBalanceProxyModel()
 {
     Q_D(OnlineBalanceProxyModel);
     delete d;
 }
 
 void OnlineBalanceProxyModel::setSourceModel(QAbstractItemModel* sourceModel)
 {
     QSortFilterProxyModel::setSourceModel(sourceModel);
 
     // make sure that data changes in the source model invalidate our filter
     connect(sourceModel, &QAbstractItemModel::dataChanged, this, [&]() {
         invalidateFilter();
     });
 }
 
 int OnlineBalanceProxyModel::columnCount(const QModelIndex& parent) const
 {
     Q_UNUSED(parent);
     return MyMoneyFile::instance()->journalModel()->columnCount();
 }
 
 QModelIndex OnlineBalanceProxyModel::index(int row, int column, const QModelIndex& parent) const
 {
     switch(column) {
     case JournalModel::Column::Balance:
         column = AccountsModel::Column::Balance;
         break;
     default:
         break;
     }
     return QSortFilterProxyModel::index(row, column, parent);
 }
 
 QVariant OnlineBalanceProxyModel::data(const QModelIndex& idx, int role) const
 {
     if (role == eMyMoney::Model::BaseModelRole) {
         return eMyMoney::Model::OnlineBalanceEntryRole;
     }
 
     if (idx.isValid()) {
         switch(role) {
         case eMyMoney::Model::DelegateRole:
             return static_cast<int>(eMyMoney::Delegates::Types::OnlineBalanceDelegate);
 
         case Qt::DisplayRole:
             switch (idx.column()) {
             case JournalModel::Column::EntryDate:
             case JournalModel::Column::Date:
                 return QSortFilterProxyModel::data(idx, eMyMoney::Model::AccountOnlineBalanceDateRole);
             case JournalModel::Column::Balance:
                 return QSortFilterProxyModel::data(idx, eMyMoney::Model::AccountBalanceRole);
             }
             break;
 
         case eMyMoney::Model::JournalSplitIdRole:
         case eMyMoney::Model::SplitAccountIdRole:
             return QSortFilterProxyModel::data(idx, eMyMoney::Model::IdRole);
 
         case eMyMoney::Model::TransactionPostDateRole:
         case eMyMoney::Model::TransactionEntryDateRole:
             return QSortFilterProxyModel::data(idx, eMyMoney::Model::AccountOnlineBalanceDateRole);
 
         case Qt::ForegroundRole:
             return QVariant();
 
         case Qt::FontRole:
             return QVariant();
 
         case Qt::TextAlignmentRole:
             switch(idx.column()) {
             case AccountsModel::Column::Balance:
                 return QVariant(Qt::AlignRight | Qt::AlignVCenter);
 
             default:
                 return QVariant(Qt::AlignLeft | Qt::AlignVCenter);
             }
             break;
         }
     }
 
     return QSortFilterProxyModel::data(idx, role);
 }
 
 Qt::ItemFlags OnlineBalanceProxyModel::flags(const QModelIndex& idx) const
 {
     Q_UNUSED(idx)
     return Qt::NoItemFlags;
 }
 
 bool OnlineBalanceProxyModel::filterAcceptsRow(int source_row, const QModelIndex& source_parent) const
 {
     // never show a favorite entry
     auto idx = sourceModel()->index(source_row, 0, source_parent);
     if (idx.data(eMyMoney::Model::AccountIsFavoriteIndexRole).toBool() == true)
         return false;
 
     // don't show an online balance if there isn't any for the account
     const auto dateValue = idx.data(eMyMoney::Model::AccountOnlineBalanceDateRole);
     if (!dateValue.isValid()) {
         return false;
     }
 
     return dateValue.toDate().isValid();
 }