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

 /*
     SPDX-FileCopyrightText: 2024 g10 Code GmbH
     SPDX-FileContributor: Daniel Vrátil <dvratil@kde.org>
 
     SPDX-License-Identifier: LGPL-2.0-or-later
 */
 
 #include "akonadifull-version.h"
 #include "dbmigrator.h"
 
 #include <KAboutData>
 #include <KAboutLicense>
 #include <KLocalizedString>
 
 #include <QCommandLineOption>
 #include <QCommandLineParser>
 #include <QCoreApplication>
 
 #include <iostream>
 
 using namespace Akonadi::Server;
 
 class CLIDelegate : public UIDelegate
 {
 public:
     Result questionYesNo(const QString &question) override
     {
         std::cout << qUtf8Printable(question)
                   << qUtf8Printable(i18nc("Options for user to input to command line prompt, parenthesis indicate "
                                           "which letter to type, capitalized option is default",
                                           " [(y)es/(N)o] "));
         std::cout.flush();
         std::string answer;
         std::getline(std::cin, answer);
 
         const auto yes = i18nc("Letter for option \"(y)es\" prompted from command line", "y");
 
         if (answer.size() > 0 && std::tolower(answer[0]) == yes[0].toLatin1()) {
             return Result::Yes;
         }
         return Result::No;
     }
 
     Result questionYesNoSkip(const QString &question) override
     {
         std::cout << qUtf8Printable(question)
                   << qUtf8Printable(i18nc("Options for user to input to command line prompt, parenthesis indicate "
                                           "which letter to type, capitalized option is default",
                                           " [(y)es/(N)o/(s)kip] "));
 
         std::cout.flush();
         std::string answer;
         std::getline(std::cin, answer);
 
         const auto yes = i18nc("Letter for option \"(y)es\" prompted from command line", "y");
         const auto skip = i18nc("Letter for option \"(s)kip\" prompted from command line", "s");
 
         if (answer.size() > 0 && std::tolower(answer[0]) == yes[0].toLatin1()) {
             return Result::Yes;
         }
         if (answer.size() > 0 && std::tolower(answer[0]) == skip[0].toLatin1()) {
             return Result::Skip;
         }
         return Result::No;
     }
 };
 
 int main(int argc, char **argv)
 {
     Q_INIT_RESOURCE(akonadidb);
 
     QCoreApplication app(argc, argv);
     KLocalizedString::setApplicationDomain("akonadi-db-migrator");
     KAboutData about(QStringLiteral("akonadi-db-migrator"),
                      i18n("Akonadi DB Migrator"),
                      QStringLiteral(AKONADI_FULL_VERSION),
                      i18n("Akonadi DB Migrator"),
                      KAboutLicense::LGPL,
                      i18n("(c) 2024 g10 Code GmbH"));
     KAboutData::setApplicationData(about);
 
     QCommandLineParser parser;
     about.setupCommandLine(&parser);
     const QCommandLineOption engineOpt{QStringLiteral("newengine"),
                                        i18n("The new DB engine to use. Possible values are  \"sqlite\", \"mysql\" and \"postgres\""),
                                        QStringLiteral("ENGINE")};
     parser.addOption(engineOpt);
     parser.process(app);
 
     const auto targetEngine = parser.value(engineOpt).toLower();
     if (targetEngine != QLatin1StringView("sqlite") && targetEngine != QLatin1StringView("mysql") && targetEngine != QLatin1StringView("postgres")) {
         std::cerr << qUtf8Printable(i18nc("@info:shell", "Invalid target engine: %1.", targetEngine)) << std::endl;
         return 1;
     }
 
     CLIDelegate delegate;
     DbMigrator migrator(targetEngine, &delegate);
     QObject::connect(&migrator, &DbMigrator::info, &app, [](const QString &message) {
         std::cout << qUtf8Printable(message) << std::endl;
     });
     QObject::connect(&migrator, &DbMigrator::error, &app, [](const QString &message) {
         std::cerr << qUtf8Printable(message) << std::endl;
     });
     QObject::connect(&migrator, &DbMigrator::migrationCompleted, &app, [](bool success) {
         if (success) {
             std::cout << qUtf8Printable(i18nc("@info:status", "Migration completed successfully.")) << std::endl;
         } else {
             std::cout << qUtf8Printable(i18nc("@info:status", "Migration failed.")) << std::endl;
         }
         qApp->quit();
     });
     QObject::connect(&migrator, &DbMigrator::progress, &app, [](const QString &table, int tablesDone, int tablesTotal) {
         std::cout << qUtf8Printable(i18nc("@info:progress", "Migrating table %1 (%2/%3)...", table, tablesDone, tablesTotal)) << std::endl;
     });
     QString lastTable;
     int lastPerc = -1;
     QObject::connect(&migrator, &DbMigrator::tableProgress, &app, [&lastTable, &lastPerc](const QString &table, int rowsDone, int rowsTotal) {
         const int perc = rowsDone * 100 / rowsTotal;
         if (lastTable != table) {
             lastPerc = -1;
         }
         if (perc % 10 != 0 || perc == lastPerc) {
             return;
         }
         lastPerc = perc;
         lastTable = table;
         std::cout << qUtf8Printable(i18nc("@info:progress", "%1%...", perc)) << std::endl;
     });
 
     QMetaObject::invokeMethod(&migrator, &DbMigrator::startMigration, Qt::QueuedConnection);
 
     return app.exec();
 }