tools/baloosearch/main.cpp

/*
    This file is part of the KDE Baloo Project
    SPDX-FileCopyrightText: 2013-2015 Vishesh Handa <vhanda@kde.org>
 
    SPDX-License-Identifier: LGPL-2.1-only OR LGPL-3.0-only OR LicenseRef-KDE-Accepted-LGPL
*/
 
#include <QCoreApplication>
#include <QCommandLineParser>
#include <QCommandLineOption>
#include <QFileInfo>
#include <QTextStream>
#include <QElapsedTimer>
 
#include <KAboutData>
#include <KLocalizedString>
 
#include "query.h"
 
#include <iostream>
 
int main(int argc, char* argv[])
{
    QCoreApplication app(argc, argv);
 
    KAboutData aboutData(QStringLiteral("Baloo"),
                         i18n("Baloo Search"),
                         QStringLiteral(PROJECT_VERSION),
                         i18n("A tool to search through the files indexed by Baloo"),
                         KAboutLicense::GPL);
    aboutData.addAuthor(i18n("Vishesh Handa"), QString(), QStringLiteral("vhanda@kde.org"));
 
    KAboutData::setApplicationData(aboutData);
 
    QCommandLineParser parser;
    parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("l") << QStringLiteral("limit"),
                                        i18n("The maximum number of results"),
                                        i18n("limit")));
    parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("o") << QStringLiteral("offset"),
                                        i18n("Offset from which to start the search"),
                                        i18n("offset")));
    parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("t") << QStringLiteral("type"),
                                        i18n("Type of data to be searched"),
                                        i18n("typeStr")));
    parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("d") << QStringLiteral("directory"),
                                        i18n("Limit search to specified directory"),
                                        i18n("directory")));
    parser.addOption(QCommandLineOption({QStringLiteral("i"), QStringLiteral("id")},
                                        i18n("Show document IDs")));
    parser.addOption(QCommandLineOption({QStringLiteral("s"), QStringLiteral("sort")},
                                        i18n("Sorting criteria"), QStringLiteral("auto|time|none"), QStringLiteral("auto")));
    parser.addPositionalArgument(i18n("query"), i18n("List of words to query for"));
    parser.addHelpOption();
    parser.addVersionOption();
    parser.process(app);
 
    int queryLimit = -1;
    int offset = 0;
    QString typeStr;
    bool showDocumentId = parser.isSet(QStringLiteral("id"));
 
    QStringList args = parser.positionalArguments();
    if (args.isEmpty()) {
        parser.showHelp(1);
    }
 
    if (parser.isSet(QStringLiteral("type"))) {
        typeStr = parser.value(QStringLiteral("type"));
    }
    if (parser.isSet(QStringLiteral("limit"))) {
        queryLimit = parser.value(QStringLiteral("limit")).toInt();
    }
    if (parser.isSet(QStringLiteral("offset"))) {
        offset = parser.value(QStringLiteral("offset")).toInt();
    }
    const Baloo::Query::SortingOption orderBy = [&parser]() {
       auto val = parser.value(QStringLiteral("sort"));
       if (val == QStringLiteral("auto")) {
           return Baloo::Query::SortAuto;
       } else if (val == QStringLiteral("time")) {
           return Baloo::Query::SortAuto;
       } else if (val == QStringLiteral("none")) {
           return Baloo::Query::SortNone;
       } else {
           parser.showHelp(1);
       }
    }();
 
    QString queryStr = args.join(QLatin1Char(' '));
 
    Baloo::Query query;
    query.addType(typeStr);
    query.setSearchString(queryStr);
    query.setLimit(queryLimit);
    query.setOffset(offset);
    query.setSortingOption(orderBy);
 
    if (parser.isSet(QStringLiteral("directory"))) {
        QString folderName = parser.value(QStringLiteral("directory"));
        const QFileInfo fi(folderName);
        if (!fi.isDir()) {
            std::cerr << qPrintable(i18n("%1 is not a valid directory", folderName)) << std::endl;
            return 1;
        }
        while (folderName.endsWith(QLatin1Char('/')) && (folderName.size() > 1)) {
            folderName.chop(1);
        }
        auto canonicalPath = fi.canonicalFilePath();
        if (canonicalPath != folderName) {
            std::cerr << qPrintable(i18n("Using canonical path '%1' for '%2'", canonicalPath, folderName)) << std::endl;
        }
        query.setIncludeFolder(canonicalPath);
    }
 
    QElapsedTimer timer;
    timer.start();
 
    Baloo::ResultIterator iter = query.exec();
    while (iter.next()) {
        const QString filePath = iter.filePath();
        if (showDocumentId) {
            std::cout << iter.documentId().constData() << " ";
        }
        std::cout << qPrintable(filePath) << std::endl;
    }
    std::cerr << qPrintable(i18n("Elapsed: %1 msecs", timer.nsecsElapsed() / 1000000.0)) << std::endl;
 
    return 0;
}