File indexing completed on 2024-05-12 05:10:49

 /*
   This file is part of KOrganizer.
 
   SPDX-FileCopyrightText: 2015 Sandro Knauß <knauss@kolabsys.com>
 
   SPDX-License-Identifier: GPL-2.0-or-later WITH Qt-Commercial-exception-1.0
 */
 
 #include "searchcollectionhelper.h"
 #include "akonadicalendar_debug.h"
 
 #include <Akonadi/CollectionFetchJob>
 #include <Akonadi/CollectionFetchScope>
 #include <Akonadi/CollectionModifyJob>
 #include <Akonadi/EntityDisplayAttribute>
 #include <Akonadi/PersistentSearchAttribute>
 #include <Akonadi/SearchCreateJob>
 #include <Akonadi/SearchQuery>
 
 #include <KCalendarCore/Event>
 #include <KCalendarCore/Journal>
 #include <KCalendarCore/Todo>
 
 #include <KLocalizedString>
 
 using namespace Akonadi;
 
 SearchCollectionHelper::SearchCollectionHelper(QObject *parent)
     : QObject(parent)
     , mIdentityManager(KIdentityManagementCore::IdentityManager::self())
 {
     setupSearchCollections();
     connect(mIdentityManager, qOverload<>(&KIdentityManagementCore::IdentityManager::changed), this, &SearchCollectionHelper::updateOpenInvitation);
     connect(mIdentityManager, qOverload<>(&KIdentityManagementCore::IdentityManager::changed), this, &SearchCollectionHelper::updateDeclinedInvitation);
 
     updateOpenInvitation();
     updateDeclinedInvitation();
 }
 
 void SearchCollectionHelper::setupSearchCollections()
 {
     // Collection "Search", has always ID 1
     auto fetchJob = new Akonadi::CollectionFetchJob(Akonadi::Collection(1), Akonadi::CollectionFetchJob::FirstLevel);
     fetchJob->fetchScope().setListFilter(Akonadi::CollectionFetchScope::NoFilter);
     connect(fetchJob, &Akonadi::CollectionFetchJob::result, this, &SearchCollectionHelper::onSearchCollectionsFetched);
 }
 
 void SearchCollectionHelper::onSearchCollectionsFetched(KJob *job)
 {
     if (job->error()) {
         qCWarning(AKONADICALENDAR_LOG) << "Search failed: " << job->errorString();
     } else {
         auto fetchJob = static_cast<Akonadi::CollectionFetchJob *>(job);
         const Akonadi::Collection::List lstCols = fetchJob->collections();
         for (const Akonadi::Collection &col : lstCols) {
             const QString collectionName = col.name();
             if (collectionName == QLatin1StringView("OpenInvitations")) {
                 mOpenInvitationCollection = col;
             } else if (collectionName == QLatin1StringView("DeclinedInvitations")) {
                 mDeclineCollection = col;
             }
         }
     }
     updateOpenInvitation();
     updateDeclinedInvitation();
 }
 
 void SearchCollectionHelper::updateSearchCollection(Akonadi::Collection col,
                                                     KCalendarCore::Attendee::PartStat status,
                                                     const QString &name,
                                                     const QString &displayName)
 {
     // Update or create search collections
 
     Akonadi::SearchQuery query(Akonadi::SearchTerm::RelOr);
     const QStringList lstEmails = mIdentityManager->allEmails();
     for (const QString &email : lstEmails) {
         if (!email.isEmpty()) {
             query.addTerm(Akonadi::IncidenceSearchTerm(Akonadi::IncidenceSearchTerm::PartStatus, QString(email + QString::number(status))));
         }
     }
 
     if (!col.isValid()) {
         auto job = new Akonadi::SearchCreateJob(name, query);
         job->setRemoteSearchEnabled(false);
         job->setSearchMimeTypes({
             KCalendarCore::Event::eventMimeType(),
             KCalendarCore::Todo::todoMimeType(),
             KCalendarCore::Journal::journalMimeType(),
         });
         connect(job, &Akonadi::SearchCreateJob::result, this, &SearchCollectionHelper::createSearchJobFinished);
         qCDebug(AKONADICALENDAR_LOG) << "We have to create a " << name << " virtual Collection";
     } else {
         auto attribute = col.attribute<Akonadi::PersistentSearchAttribute>(Akonadi::Collection::AddIfMissing);
         auto displayname = col.attribute<Akonadi::EntityDisplayAttribute>(Akonadi::Collection::AddIfMissing);
         attribute->setQueryString(QString::fromLatin1(query.toJSON()));
         attribute->setRemoteSearchEnabled(false);
         attribute->setRecursive(true);
         displayname->setDisplayName(displayName);
         col.setEnabled(true);
         auto job = new Akonadi::CollectionModifyJob(col, this);
         connect(job, &Akonadi::CollectionModifyJob::result, this, &SearchCollectionHelper::modifyResult);
         qCDebug(AKONADICALENDAR_LOG) << "updating " << name << " (" << col.id() << ") virtual Collection";
         qCDebug(AKONADICALENDAR_LOG) << "query" << query.toJSON();
     }
 }
 
 void SearchCollectionHelper::updateDeclinedInvitation()
 {
     updateSearchCollection(mDeclineCollection,
                            KCalendarCore::Attendee::Declined,
                            QStringLiteral("DeclinedInvitations"),
                            i18nc("A collection of all declined invidations.", "Declined Invitations"));
 }
 
 void SearchCollectionHelper::updateOpenInvitation()
 {
     updateSearchCollection(mOpenInvitationCollection,
                            KCalendarCore::Attendee::NeedsAction,
                            QStringLiteral("OpenInvitations"),
                            i18nc("A collection of all open invidations.", "Open Invitations"));
 }
 
 void SearchCollectionHelper::createSearchJobFinished(KJob *job)
 {
     auto createJob = qobject_cast<Akonadi::SearchCreateJob *>(job);
     const Akonadi::Collection searchCollection = createJob->createdCollection();
     if (job->error()) {
         qCWarning(AKONADICALENDAR_LOG) << "Error occurred " << searchCollection.name() << job->errorString();
         return;
     }
     qCDebug(AKONADICALENDAR_LOG) << "Created search folder successfully " << searchCollection.name();
 
     const QString searchCollectionName = searchCollection.name();
     if (searchCollectionName == QLatin1StringView("OpenInvitations")) {
         mOpenInvitationCollection = searchCollection;
         updateOpenInvitation();
     } else if (searchCollectionName == QLatin1StringView("DeclinedInvitations")) {
         mDeclineCollection = searchCollection;
         updateDeclinedInvitation();
     }
 }
 
 void SearchCollectionHelper::modifyResult(KJob *job)
 {
     if (job->error()) {
         qCWarning(AKONADICALENDAR_LOG) << "Error occurred " << job->errorString();
     } else {
         qCDebug(AKONADICALENDAR_LOG) << "modify was successful";
     }
 }
 
 #include "moc_searchcollectionhelper.cpp"