kio_search.cpp

/*
    SPDX-FileCopyrightText: 2008-2010 Sebastian Trueg <trueg at kde.org>
    SPDX-FileCopyrightText: 2012-2014 Vishesh Handa <me@vhanda.in>
    SPDX-FileCopyrightText: 2020 Stefan Brüns <bruns@kde.org>
 
    SPDX-License-Identifier: GPL-2.0-or-later
*/
 
#include "kio_search.h"
#include "../common/udstools.h"
 
#include "query.h"
#include "resultiterator.h"
#include <sys/stat.h>
 
#include <KLocalizedString>
#include <KUser>
#include <QCoreApplication>
#include <QFileInfo>
#include <QUrl>
#include <QUrlQuery>
 
using namespace Baloo;
 
// Pseudo plugin class to embed meta data
class KIOPluginForMetaData : public QObject
{
    Q_OBJECT
    Q_PLUGIN_METADATA(IID "org.kde.kio.worker.baloosearch" FILE "baloosearch.json")
};
 
namespace
{
 
KIO::UDSEntry statSearchFolder(const QUrl& url)
{
    KIO::UDSEntry uds;
    uds.reserve(9);
#ifdef Q_OS_WIN
    uds.fastInsert(KIO::UDSEntry::UDS_ACCESS, _S_IREAD );
#else
    uds.fastInsert(KIO::UDSEntry::UDS_ACCESS, S_IRUSR | S_IXUSR);
#endif
    uds.fastInsert(KIO::UDSEntry::UDS_USER, KUser().loginName());
    uds.fastInsert(KIO::UDSEntry::UDS_FILE_TYPE, S_IFDIR);
    uds.fastInsert(KIO::UDSEntry::UDS_MIME_TYPE, QStringLiteral("inode/directory"));
    uds.fastInsert(KIO::UDSEntry::UDS_ICON_OVERLAY_NAMES, QStringLiteral("baloo"));
    uds.fastInsert(KIO::UDSEntry::UDS_DISPLAY_TYPE, i18n("Search Folder"));
    uds.fastInsert(KIO::UDSEntry::UDS_URL, url.url());
 
    QUrlQuery query(url);
    QString title = query.queryItemValue(QStringLiteral("title"), QUrl::FullyDecoded);
    if (!title.isEmpty()) {
        uds.fastInsert(KIO::UDSEntry::UDS_NAME, title);
        uds.fastInsert(KIO::UDSEntry::UDS_DISPLAY_NAME, title);
    }
 
    return uds;
}
 
}
 
SearchProtocol::SearchProtocol(const QByteArray& poolSocket, const QByteArray& appSocket)
    : KIO::WorkerBase("baloosearch", poolSocket, appSocket)
{
}
 
SearchProtocol::~SearchProtocol()
{
}
 
KIO::WorkerResult SearchProtocol::listDir(const QUrl& url)
{
    Query q = Query::fromSearchUrl(url);
 
    // Canonicalise the search base directory
    const QString folder = q.includeFolder();
    if (!folder.isEmpty()) {
        const QString canonicalPath = QFileInfo(folder).canonicalFilePath();
        if (!canonicalPath.isEmpty()) {
            q.setIncludeFolder(canonicalPath);
        }
    }
 
    q.setSortingOption(Query::SortNone);
    ResultIterator it = q.exec();
 
    UdsFactory udsf;
 
    while (it.next()) {
        KIO::UDSEntry uds = udsf.createUdsEntry(it.filePath());
        if (uds.count()) {
        listEntry(uds);
        }
    }
 
    KIO::UDSEntry uds;
    uds.reserve(5);
    uds.fastInsert(KIO::UDSEntry::UDS_NAME, QStringLiteral("."));
    uds.fastInsert(KIO::UDSEntry::UDS_FILE_TYPE, S_IFDIR);
    uds.fastInsert(KIO::UDSEntry::UDS_MIME_TYPE, QStringLiteral("inode/directory"));
#ifdef Q_OS_WIN
    uds.fastInsert(KIO::UDSEntry::UDS_ACCESS, _S_IREAD );
#else
    uds.fastInsert(KIO::UDSEntry::UDS_ACCESS, S_IRUSR | S_IXUSR);
#endif
    uds.fastInsert(KIO::UDSEntry::UDS_USER, KUser().loginName());
    listEntry(uds);
 
    return KIO::WorkerResult::pass();
}
 
KIO::WorkerResult SearchProtocol::mimetype(const QUrl&)
{
    mimeType(QStringLiteral("inode/directory"));
    return KIO::WorkerResult::pass();
}
 
KIO::WorkerResult SearchProtocol::stat(const QUrl& url)
{
    statEntry(statSearchFolder(url));
    return KIO::WorkerResult::pass();
}
 
extern "C"
{
    Q_DECL_EXPORT int kdemain(int argc, char** argv)
    {
        QCoreApplication app(argc, argv);
        app.setApplicationName(QStringLiteral("kio_baloosearch"));
        Baloo::SearchProtocol worker(argv[2], argv[3]);
        worker.dispatchLoop();
        return 0;
    }
}
 
#include "kio_search.moc"