File indexing completed on 2025-01-05 04:49:48

 /*
  * SPDX-FileCopyrightText: 2016 Daniel Vrátil <dvratil@kde.org>
  *
  * SPDX-License-Identifier: GPL-2.0-or-later
  *
  */
 
 #include "pimcalendarsmodel.h"
 #include "settingschangenotifier.h"
 
 #include <Akonadi/CollectionFetchScope>
 #include <Akonadi/EntityDisplayAttribute>
 #include <Akonadi/EntityTreeModel>
 #include <Akonadi/Monitor>
 
 #include <KCalendarCore/Event>
 #include <KCalendarCore/Todo>
 
 #include <KConfigGroup>
 #include <KSharedConfig>
 
 PimCalendarsModel::PimCalendarsModel(QObject *parent)
     : QSortFilterProxyModel(parent)
 {
     setSortRole(Qt::DisplayRole);
     setSortCaseSensitivity(Qt::CaseInsensitive);
     setSortLocaleAware(true);
     setDynamicSortFilter(true);
 
     auto cr = new Akonadi::Monitor(this);
     cr->setMimeTypeMonitored(KCalendarCore::Event::eventMimeType());
     cr->setMimeTypeMonitored(KCalendarCore::Todo::todoMimeType());
     cr->setTypeMonitored(Akonadi::Monitor::Collections);
     cr->collectionFetchScope().setListFilter(Akonadi::CollectionFetchScope::Enabled);
 
     mEtm = new Akonadi::EntityTreeModel(cr, this);
     mEtm->setItemPopulationStrategy(Akonadi::EntityTreeModel::NoItemPopulation);
     mEtm->setListFilter(Akonadi::CollectionFetchScope::Enabled);
     connect(mEtm, &Akonadi::EntityTreeModel::collectionTreeFetched, this, [this]() {
         sort(0, Qt::AscendingOrder);
     });
 
     setSourceModel(mEtm);
 
     auto config = KSharedConfig::openConfig();
     auto group = config->group(QStringLiteral("PIMEventsPlugin"));
     const auto enabledCalendarIds = group.readEntry(QStringLiteral("calendars"), QList<qint64>());
     mEnabledCalendars = QSet<qint64>(enabledCalendarIds.begin(), enabledCalendarIds.end());
 }
 
 PimCalendarsModel::~PimCalendarsModel() = default;
 
 QHash<int, QByteArray> PimCalendarsModel::roleNames() const
 {
     return {
         {CollectionIdRole, "collectionId"},
         {NameRole, "name"},
         {EnabledRole, "enabled"},
         {CheckedRole, "checked"},
         {IconNameRole, "iconName"},
     };
 }
 
 QVariant PimCalendarsModel::data(const QModelIndex &index, int role) const
 {
     Q_ASSERT(checkIndex(index, QAbstractItemModel::CheckIndexOption::IndexIsValid));
 
     if (role == Qt::DisplayRole) {
         return QSortFilterProxyModel::data(index, role);
     }
 
     const auto &col = QSortFilterProxyModel::data(index, Akonadi::EntityTreeModel::CollectionRole).value<Akonadi::Collection>();
 
     if (!col.isValid()) {
         return {};
     }
 
     switch (role) {
     case CollectionIdRole:
         return col.id();
     case NameRole:
         return col.displayName();
     case EnabledRole: {
         const auto mts = col.contentMimeTypes();
         return mts.contains(KCalendarCore::Event::eventMimeType()) || mts.contains(KCalendarCore::Todo::todoMimeType());
     }
     case CheckedRole:
         return mEnabledCalendars.contains(col.id());
     case IconNameRole: {
         const auto attr = col.attribute<Akonadi::EntityDisplayAttribute>();
         const QString icon = attr ? attr->iconName() : QString();
     }
     default:
         return {};
     }
 }
 
 void PimCalendarsModel::setChecked(qint64 collectionId, bool checked)
 {
     bool done = false;
     if (checked) {
         done = !mEnabledCalendars.contains(collectionId);
         mEnabledCalendars.insert(collectionId);
     } else {
         done = mEnabledCalendars.remove(collectionId);
     }
     if (done) {
         const QModelIndex idx = Akonadi::EntityTreeModel::modelIndexForCollection(this, Akonadi::Collection(collectionId));
         Q_EMIT dataChanged(idx, idx);
     }
 }
 
 void PimCalendarsModel::saveConfig()
 {
     auto config = KSharedConfig::openConfig();
     auto group = config->group(QStringLiteral("PIMEventsPlugin"));
     auto savedList = group.readEntry("calendars", QList<qint64>());
     auto currentList = mEnabledCalendars.values();
     std::sort(savedList.begin(), savedList.end());
     std::sort(currentList.begin(), currentList.end());
 
     if (currentList != savedList) {
         group.writeEntry("calendars", currentList);
         SettingsChangeNotifier::self()->notifySettingsChanged();
     }
 }
 
 #include "moc_pimcalendarsmodel.cpp"