File indexing completed on 2025-01-05 04:46:24

 /*
     SPDX-FileCopyrightText: 2009 Volker Krause <vkrause@kde.org>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "agentsearchinterface.h"
 #include "agentbase.h"
 #include "agentsearchinterface_p.h"
 #include "akonadiagentbase_debug.h"
 #include "collection.h"
 #include "collectionfetchjob.h"
 #include "collectionfetchscope.h"
 #include "private/imapset_p.h"
 #include "searchadaptor.h"
 #include "searchresultjob_p.h"
 #include "servermanager.h"
 #include <QDBusConnection>
 
 using namespace Akonadi;
 using namespace std::chrono_literals;
 AgentSearchInterfacePrivate::AgentSearchInterfacePrivate(AgentSearchInterface *qq)
     : q(qq)
 {
     new Akonadi__SearchAdaptor(this);
     QDBusConnection::sessionBus().registerObject(QStringLiteral("/Search"), this, QDBusConnection::ExportAdaptors);
 
     QTimer::singleShot(0s, this, &AgentSearchInterfacePrivate::delayedInit);
 }
 
 void AgentSearchInterfacePrivate::delayedInit()
 {
     QDBusInterface iface(ServerManager::serviceName(ServerManager::Server),
                          QStringLiteral("/SearchManager"),
                          QStringLiteral("org.freedesktop.Akonadi.SearchManager"),
                          QDBusConnection::sessionBus(),
                          this);
     QDBusMessage msg = iface.call(QStringLiteral("registerInstance"), dynamic_cast<AgentBase *>(q)->identifier());
     // TODO ?
 }
 
 void AgentSearchInterfacePrivate::addSearch(const QString &query, const QString &queryLanguage, quint64 resultCollectionId)
 {
     q->addSearch(query, queryLanguage, Collection(resultCollectionId));
 }
 
 void AgentSearchInterfacePrivate::removeSearch(quint64 resultCollectionId)
 {
     q->removeSearch(Collection(resultCollectionId));
 }
 
 void AgentSearchInterfacePrivate::search(const QByteArray &searchId, const QString &query, quint64 collectionId)
 {
     mSearchId = searchId;
     mCollectionId = collectionId;
 
     auto fetchJob = new CollectionFetchJob(Collection(mCollectionId), CollectionFetchJob::Base, this);
     fetchJob->fetchScope().setAncestorRetrieval(CollectionFetchScope::All);
     fetchJob->setProperty("query", query);
     connect(fetchJob, &KJob::finished, this, &AgentSearchInterfacePrivate::collectionReceived);
 }
 
 void AgentSearchInterfacePrivate::collectionReceived(KJob *job)
 {
     auto fetchJob = qobject_cast<CollectionFetchJob *>(job);
     if (fetchJob->error()) {
         qCCritical(AKONADIAGENTBASE_LOG) << fetchJob->errorString();
         new SearchResultJob(fetchJob->property("searchId").toByteArray(), Collection(mCollectionId), this);
         return;
     }
 
     if (fetchJob->collections().count() != 1) {
         qCDebug(AKONADIAGENTBASE_LOG) << "Server requested search in invalid collection, or collection was removed in the meanwhile";
         // Tell server we are done
         new SearchResultJob(fetchJob->property("searchId").toByteArray(), Collection(mCollectionId), this);
         return;
     }
 
     const Collection collection = fetchJob->collections().at(0);
     q->search(fetchJob->property("query").toString(), collection);
 }
 
 AgentSearchInterface::AgentSearchInterface()
     : d(new AgentSearchInterfacePrivate(this))
 {
 }
 
 AgentSearchInterface::~AgentSearchInterface() = default;
 
 void AgentSearchInterface::searchFinished(const QList<qint64> &result, ResultScope scope)
 {
     if (scope == Akonadi::AgentSearchInterface::Rid) {
         QList<QByteArray> rids;
         rids.reserve(result.size());
         for (qint64 rid : result) {
             rids << QByteArray::number(rid);
         }
 
         searchFinished(rids);
         return;
     }
 
     auto resultJob = new SearchResultJob(d->mSearchId, Collection(d->mCollectionId), d.get());
     resultJob->setResult(result);
 }
 
 void AgentSearchInterface::searchFinished(const ImapSet &result, ResultScope scope)
 {
     if (scope == Akonadi::AgentSearchInterface::Rid) {
         QList<QByteArray> rids;
         const ImapInterval::List lstInterval = result.intervals();
         for (const ImapInterval &interval : lstInterval) {
             const int endInterval(interval.end());
             for (int i = interval.begin(); i <= endInterval; ++i) {
                 rids << QByteArray::number(i);
             }
         }
 
         searchFinished(rids);
         return;
     }
 
     auto resultJob = new SearchResultJob(d->mSearchId, Collection(d->mCollectionId), d.get());
     resultJob->setResult(result);
 }
 
 void AgentSearchInterface::searchFinished(const QList<QByteArray> &result)
 {
     auto resultJob = new SearchResultJob(d->mSearchId, Collection(d->mCollectionId), d.get());
     resultJob->setResult(result);
 }
 
 #include "moc_agentsearchinterface_p.cpp"