main.cpp

#include <iostream>
#include <fstream>
using namespace std;

#include <QtGui/QApplication>
#include "dialogs/main/MainWindow.h"

#ifdef COMPILE_TESTS
#include <gtest/gtest.h>
#endif

#include "Databases.h"
#include "Utils.h"

#include "dialogs/utils/PasswordDialog.h"

#include "CommandLine.h"

#ifdef _WIN32
#include "windowscompat/guicon.h"
#endif

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);


    for (int i = 0; i < argc; ++i)
    {
        #ifdef COMPILE_TESTS
        if ((strcmp(argv[i], "-t") == 0) || (strcmp(argv[i], "--test") == 0))
        {
            char arg1[] = "--gtest_output=xml:test_report.xml";
            char *args[] = { argv[0], arg1 };
            int argCount = 2;
            ::testing::InitGoogleTest(&argCount, args);
            return RUN_ALL_TESTS();
        }
        #endif

        if ((strcmp(argv[i], "-c") == 0) || (strcmp(argv[i], "--cli") == 0))
        {
#ifdef _WIN32
            RedirectIOToConsole();
#endif
            Databases::init();
            CommandLine::run();
            Databases::finalise();
            return 0;
        }
    }

    // Redirect buffer so that it prints to a logfile instead of the console
    ofstream log;
    log.open("logfile.log");
    streambuf *const coutStreamBuffer = cout.rdbuf();
    cout.rdbuf(log.rdbuf());

    Databases::init();
    int exitCode = 1;

    if (PasswordDialog().exec() == PasswordDialog::Accepted)
    {
        MainWindow w;
        w.show();

        try { exitCode = a.exec(); } catch(const std::exception &e) { showFatalDialog(e.what()); }
    }

    Databases::finalise();

    log.close();
    cout.rdbuf(coutStreamBuffer);

    return exitCode;
}