File indexing completed on 2024-11-10 04:41:10 

0001 /*
0002     SPDX-FileCopyrightText: 2006 Volker Krause <vkrause@kde.org>
0003     SPDX-FileCopyrightText: 2007 Robert Zwerus <arzie@dds.nl>
0004 
0005     SPDX-License-Identifier: LGPL-2.0-or-later
0006 */
0007 
0008 #include "itemdumper.h"
0009 #include "collectionpathresolver.h"
0010 
0011 #include "item.h"
0012 
0013 #include <QApplication>
0014 #include <QDebug>
0015 #include <QFile>
0016 
0017 #include <KLocalizedString>
0018 
0019 #include <KAboutData>
0020 #include <QCommandLineOption>
0021 #include <QCommandLineParser>
0022 
0023 #include "itemcreatejob.h"
0024 #include "transactionjobs.h"
0025 
0026 #define GLOBAL_TRANSACTION 1
0027 
0028 using namespace Akonadi;
0029 
0030 ItemDumper::ItemDumper(const QString &path, const QString &filename, const QString &mimetype, int count)
0031     : mJobCount(0)
0032 {
0033     auto resolver = new CollectionPathResolver(path, this);
0034     Q_ASSERT(resolver->exec());
0035     const Collection collection = Collection(resolver->collection());
0036 
0037     QFile f(filename);
0038     Q_ASSERT(f.open(QIODevice::ReadOnly));
0039     QByteArray data = f.readAll();
0040     f.close();
0041     Item item;
0042     item.setMimeType(mimetype);
0043     item.setPayloadFromData(data);
0044     mTime.start();
0045 #ifdef GLOBAL_TRANSACTION
0046     auto begin = new TransactionBeginJob(this);
0047     connect(begin, &TransactionBeginJob::result, this, &ItemDumper::done);
0048     ++mJobCount;
0049 #endif
0050     for (int i = 0; i < count; ++i) {
0051         ++mJobCount;
0052         auto job = new ItemCreateJob(item, collection, this);
0053         connect(job, &ItemCreateJob::result, this, &ItemDumper::done);
0054     }
0055 #ifdef GLOBAL_TRANSACTION
0056     auto commit = new TransactionCommitJob(this);
0057     connect(commit, &TransactionCommitJob::result, this, &ItemDumper::done);
0058     ++mJobCount;
0059 #endif
0060 }
0061 
0062 void ItemDumper::done(KJob *job)
0063 {
0064     --mJobCount;
0065     if (job->error()) {
0066         qWarning() << "Error while creating item: " << job->errorString();
0067     }
0068     if (mJobCount <= 0) {
0069         qDebug() << "Time:" << mTime.elapsed() << "ms";
0070         qApp->quit();
0071     }
0072 }
0073 
0074 int main(int argc, char **argv)
0075 {
0076     QApplication app(argc, argv);
0077     KAboutData aboutData(QStringLiteral("test"), i18n("Test Application"), QStringLiteral("1.0"));
0078 
0079     QCommandLineParser parser;
0080     KAboutData::setApplicationData(aboutData);
0081     parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("path"), i18n("IMAP destination path"), QStringLiteral("argument")));
0082     parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("mimetype"),
0083                                         i18n("Source mimetype"),
0084                                         QStringLiteral("argument"),
0085                                         QStringLiteral("application/octet-stream")));
0086     parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("file"), i18n("Source file"), QStringLiteral("argument")));
0087     parser.addOption(QCommandLineOption(QStringList() << QStringLiteral("count"),
0088                                         i18n("Number of times this file is added"),
0089                                         QStringLiteral("argument"),
0090                                         QStringLiteral("1")));
0091 
0092     aboutData.setupCommandLine(&parser);
0093     parser.process(app);
0094     aboutData.processCommandLine(&parser);
0095 
0096     QString path = parser.value(QStringLiteral("path"));
0097     QString mimetype = parser.value(QStringLiteral("mimetype"));
0098     QString file = parser.value(QStringLiteral("file"));
0099     int count = qMax(1, parser.value(QStringLiteral("count")).toInt());
0100     ItemDumper d(path, file, mimetype, count);
0101     return app.exec();
0102 }
0103 
0104 #include "moc_itemdumper.cpp"