File indexing completed on 2024-05-26 05:08:38

 /*
     SPDX-FileCopyrightText: 2016-2019 Thomas Baumgart <tbaumgart@kde.org>
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "budgetsmodel.h"
 
 // ----------------------------------------------------------------------------
 // QT Includes
 
 #include <QDebug>
 #include <QString>
 #include <QDate>
 #include <QRegularExpression>
 #include <QRegularExpressionMatch>
 
 // ----------------------------------------------------------------------------
 // KDE Includes
 
 #include <KLocalizedString>
 
 // ----------------------------------------------------------------------------
 // Project Includes
 
 struct BudgetsModel::Private
 {
     Private()
         : fiscalYearStartMonth(1)
         , fiscalYearStartDay(1)
     {}
 
     int fiscalYearStartMonth;
     int fiscalYearStartDay;
 };
 
 BudgetsModel::BudgetsModel(QObject* parent, QUndoStack* undoStack)
     : MyMoneyModel<MyMoneyBudget>(parent, QStringLiteral("B"), BudgetsModel::ID_SIZE, undoStack)
     , d(new Private)
 {
     setObjectName(QLatin1String("BudgetsModel"));
 }
 
 BudgetsModel::~BudgetsModel()
 {
 }
 
 int BudgetsModel::columnCount(const QModelIndex& parent) const
 {
     Q_UNUSED(parent);
     return 2;
 }
 
 Qt::ItemFlags BudgetsModel::flags(const QModelIndex &index) const
 {
     if (!index.isValid())
         return Qt::ItemFlags();
     if (index.row() < 0 || index.row() >= rowCount(index.parent()))
         return Qt::ItemFlags();
 
     return Qt::ItemIsEnabled | Qt::ItemIsSelectable | Qt::ItemIsEditable;
 }
 
 QVariant BudgetsModel::headerData(int section, Qt::Orientation orientation, int role) const
 {
     if (orientation == Qt::Horizontal && role == Qt::DisplayRole) {
         switch(section) {
         case 0:
             return i18n("Budget");
 
         case 1:
             return i18nc("Budget year", "Year");
         }
     }
     return MyMoneyModelBase::headerData(section, orientation, role);
 }
 
 QVariant BudgetsModel::data(const QModelIndex& index, int role) const
 {
     if (!index.isValid())
         return QVariant();
     if (index.row() < 0 || index.row() >= rowCount(index.parent()))
         return QVariant();
 
     const MyMoneyBudget& budget = static_cast<TreeItem<MyMoneyBudget>*>(index.internalPointer())->constDataRef();
     switch (role) {
     case Qt::DisplayRole:
     case Qt::EditRole:
         switch(index.column()) {
         case Columns::Name:
             return budget.name();
 
         case Columns::Year:
             return budget.budgetStart().year();
 
         }
         return QVariant();
 
     case Qt::TextAlignmentRole:
         return QVariant(Qt::AlignLeft | Qt::AlignVCenter);
 
     case eMyMoney::Model::Roles::IdRole:
         return budget.id();
 
     case eMyMoney::Model::Roles::BudgetNameRole:
         return budget.name();
 
     default:
         break;
     }
     return QVariant();
 }
 
 bool BudgetsModel::setData(const QModelIndex& index, const QVariant& value, int role)
 {
     if (!index.isValid()) {
         return false;
     }
 
     if (index.row() < 0 || index.row() >= rowCount(index.parent()))
         return false;
 
     MyMoneyBudget& budget = static_cast<TreeItem<MyMoneyBudget>*>(index.internalPointer())->dataRef();
 
     switch(role) {
     case Qt::DisplayRole:
     case Qt::EditRole:
         switch(index.column()) {
         case Columns::Name:
             budget.setName(value.toString());
             break;
 
         case Columns::Year:
         {
             QDate date(value.toInt(), d->fiscalYearStartMonth, d->fiscalYearStartDay);
             if (date.isValid()) {
                 budget.setBudgetStart(QDate(value.toInt(), d->fiscalYearStartMonth, d->fiscalYearStartDay));
             } else {
                 return false;
             }
         }
         break;
 
         default:
             return false;
         }
         break;
 
     case eMyMoney::Model::Roles::BudgetNameRole:
         budget.setName(value.toString());
         break;
 
     default:
         return false;
     }
 
     setDirty();
     const auto topLeft = BudgetsModel::index(index.row(), 0);
     const auto bottomRight = BudgetsModel::index(index.row(), columnCount()-1);
     Q_EMIT dataChanged(topLeft, bottomRight);
 
     return true;
 }
 
 void BudgetsModel::setFiscalYearStart(int month, int day)
 {
     d->fiscalYearStartDay = day;
     d->fiscalYearStartMonth = month;
 }