looqs/cli/main.cpp

#include <QCoreApplication>
#include <QFile>
#include <QFileInfo>
#include <QDateTime>
#include <QDataStream>
#include <QDebug>
#include <QProcessEnvironment>
#include <QMap>
#include <QDebug>
#include <QSettings>
#include <functional>
#include <QTimer>
#include <exception>
#include "encodingdetector.h"
#include "pdfprocessor.h"
#include "defaulttextprocessor.h"
#include "utils.h"
#include "command.h"
#include "commandadd.h"
#include "commanddelete.h"
#include "commandupdate.h"
#include "commandsearch.h"
#include "commandlist.h"
#include "commandtag.h"
#include "databasefactory.h"
#include "logger.h"
#include "sandboxedprocessor.h"
#include "../shared/common.h"
#include "../shared/filescanworker.h"
#include "../shared/dbmigrator.h"

void printUsage(QString argv0)
{
	qInfo() << "Usage:" << argv0 << "command";
	qInfo() << "Valid commands: add, update, search, delete, tag, list. Each command has a --help option.";
}

Command *commandFromName(QString name, SqliteDbService &dbService)
{
	if(name == "add")
	{
		return new CommandAdd(dbService);
	}
	if(name == "delete")
	{
		return new CommandDelete(dbService);
	}
	if(name == "update")
	{
		return new CommandUpdate(dbService);
	}
	if(name == "search")
	{
		return new CommandSearch(dbService);
	}
	if(name == "list")
	{
		return new CommandList(dbService);
	}
	if(name == "tag")
	{
		return new CommandTag(dbService);
	}

	return nullptr;
}

int main(int argc, char *argv[])
{
	Common::setupAppInfo();
	QCoreApplication app(argc, argv);
	QStringList args = app.arguments();
	QString argv0 = args.takeFirst();

	if(args.length() < 1)
	{
		printUsage(argv0);
		exit(1);
	}

	try
	{
		Common::ensureConfigured();

		DatabaseFactory factory{Common::databasePath()};
		DBMigrator migrator{factory};
		if(migrator.migrationNeeded())
		{
			Logger::info() << "Database is being upgraded, please be patient..." << Qt::endl;

			QObject::connect(&migrator, &DBMigrator::migrationDone,
							 [&](uint32_t migration)
							 { Logger::info() << "Progress: Successfully migrated to: " << migration << Qt::endl; });
			QObject::connect(&migrator, &DBMigrator::done,
							 []() { Logger::info() << "Database upgrade successful" << Qt::endl; });
			QObject::connect(&migrator, &DBMigrator::error,
							 [&](QString error)
							 {
								 Logger::error() << error << Qt::endl;
								 qApp->quit();
							 });

			migrator.performMigrations();
		}
	}
	catch(LooqsGeneralException &e)
	{
		Logger::error() << "Error: " << e.message;
		return 1;
	}
	qRegisterMetaType<PageData>();
	qRegisterMetaType<FileScanResult>("FileScanResult");

	QString connectionString = Common::databasePath();
	DatabaseFactory dbFactory(connectionString);
	SqliteDbService dbService(dbFactory);
	QString commandName = args.first();
	if(commandName == "process")
	{
		if(args.length() < 1)
		{
			qDebug() << "Filename is required";
			return 1;
		}

		QString file = args.at(1);
		SandboxedProcessor processor(file);
		return static_cast<int>(processor.process());
	}
	Command *cmd = commandFromName(commandName, dbService);
	if(cmd != nullptr)
	{
		try
		{
			QObject::connect(cmd, &Command::finishedCmd, [](int retval) { QCoreApplication::exit(retval); });
			cmd->setArguments(args);
			QTimer::singleShot(0, cmd, &Command::execute);
		}
		catch(const LooqsGeneralException &e)
		{
			Logger::error() << "Exception caught, message:" << e.message << Qt::endl;
			return 1;
		}
	}
	else
	{
		printUsage(argv0);
		return 1;
	}

	return app.exec();
}