1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
|
/*
Copyright (c) 2009 Volker Krause <vkrause@kde.org>
This library is free software; you can redistribute it and/or modify it
under the terms of the GNU Library General Public License as published by
the Free Software Foundation; either version 2 of the License, or (at your
option) any later version.
This library is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public
License for more details.
You should have received a copy of the GNU Library General Public License
along with this library; see the file COPYING.LIB. If not, write to the
Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
02110-1301, USA.
*/
#include "agentsearchinterface.h"
#include "agentsearchinterface_p.h"
#include "collection.h"
#include "dbusconnectionpool.h"
#include "searchresultjob_p.h"
#include "searchadaptor.h"
#include "collectionfetchjob.h"
#include "collectionfetchscope.h"
#include "servermanager.h"
#include "agentbase.h"
using namespace Akonadi;
AgentSearchInterfacePrivate::AgentSearchInterfacePrivate(AgentSearchInterface *qq)
: q(qq)
{
new Akonadi__SearchAdaptor(this);
DBusConnectionPool::threadConnection().registerObject(QLatin1String("/Search"),
this, QDBusConnection::ExportAdaptors);
QTimer::singleShot(0, this, SLOT(delayedInit()));
}
void AgentSearchInterfacePrivate::delayedInit()
{
QDBusInterface iface(ServerManager::serviceName(ServerManager::Server),
QLatin1String("/SearchManager"),
QLatin1String("org.freedesktop.Akonadi.SearchManager"),
QDBusConnection::sessionBus(), this);
QDBusMessage msg = iface.call(QLatin1String("registerInstance"), dynamic_cast<AgentBase *>(q)->identifier());
}
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;
CollectionFetchJob *fetchJob = new CollectionFetchJob(Collection(mCollectionId), CollectionFetchJob::Base, this);
fetchJob->fetchScope().setAncestorRetrieval(CollectionFetchScope::All);
fetchJob->setProperty("query", query);
connect(fetchJob, SIGNAL(finished(KJob*)), this, SLOT(collectionReceived(KJob*)));
}
void AgentSearchInterfacePrivate::collectionReceived(KJob *job)
{
CollectionFetchJob *fetchJob = qobject_cast<CollectionFetchJob *>(job);
if (fetchJob->error()) {
kError() << fetchJob->errorString();
new SearchResultJob(fetchJob->property("searchId").toByteArray(), Collection(mCollectionId), this);
return;
}
if (fetchJob->collections().count() != 1) {
kDebug() << "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().first();
q->search(fetchJob->property("query").toString(),
collection);
}
AgentSearchInterface::AgentSearchInterface()
: d(new AgentSearchInterfacePrivate(this))
{
}
AgentSearchInterface::~AgentSearchInterface()
{
delete d;
}
void AgentSearchInterface::searchFinished(const QVector<qint64> result, ResultScope scope)
{
if (scope == Akonadi::AgentSearchInterface::Rid) {
QVector<QByteArray> rids;
rids.reserve(result.size());
Q_FOREACH (qint64 rid, result) {
rids << QByteArray::number(rid);
}
searchFinished(rids);
return;
}
SearchResultJob *resultJob = new SearchResultJob(d->mSearchId, Collection(d->mCollectionId), d);
resultJob->setResult(result);
}
void AgentSearchInterface::searchFinished(const ImapSet &result, ResultScope scope)
{
if (scope == Akonadi::AgentSearchInterface::Rid) {
QVector<QByteArray> rids;
Q_FOREACH (const ImapInterval &interval, result.intervals()) {
for (int i = interval.begin(); i <= interval.end(); ++i) {
rids << QByteArray::number(i);
}
}
searchFinished(rids);
return;
}
SearchResultJob *resultJob = new SearchResultJob(d->mSearchId, Collection(d->mCollectionId), d);
resultJob->setResult(result);
}
void AgentSearchInterface::searchFinished(const QVector<QByteArray> &result)
{
SearchResultJob *resultJob = new SearchResultJob(d->mSearchId, Collection(d->mCollectionId), d);
resultJob->setResult(result);
}
#include "moc_agentsearchinterface_p.cpp"
|