main.cpp

#include <QDebug>
#include <QTextCodec>
#include <QSslSocket>
#include <QLocale>
#include <QTranslator>
#include <QScopedPointer>
#include <QOpenGLContext>
#include <QDateTime>
#include <QSysInfo>
#include <QProcess>
#include <QWebEngineSettings>
#include <QWindow>
#include <QAccessible>

#include <core/configmgr.h>
#include <core/mainconfig.h>
#include <core/coreconfig.h>
#include <core/sessionconfig.h>
#include <core/singleinstanceguard.h>
#include <core/vnotex.h>
#include <core/logger.h>
#include <widgets/mainwindow.h>
#include <core/exception.h>
#include <utils/widgetutils.h>
#include <widgets/messageboxhelper.h>
#include "commandlineoptions.h"
#include "application.h"
#include "fakeaccessible.h"

using namespace vnotex;

void loadTranslators(QApplication &p_app);

void initWebEngineSettings();

void showMessageOnCommandLineIfAvailable(const QString &p_msg);

int main(int argc, char *argv[])
{
    QTextCodec *codec = QTextCodec::codecForName("UTF8");
    if (codec) {
        QTextCodec::setCodecForLocale(codec);
    }

    QCoreApplication::setAttribute(Qt::AA_UseHighDpiPixmaps);

    // This only takes effect on Win, X11 and Android.
    QCoreApplication::setAttribute(Qt::AA_EnableHighDpiScaling);

    // Set OpenGL option on Windows.
    // Set environment QT_OPENGL to "angle/desktop/software".
#if defined(Q_OS_WIN)
    {
        auto option = SessionConfig::getOpenGLAtBootstrap();
        qDebug() << "OpenGL option" << SessionConfig::openGLToString(option);
        switch (option) {
        case SessionConfig::OpenGL::Desktop:
            QCoreApplication::setAttribute(Qt::AA_UseDesktopOpenGL);
            break;

        case SessionConfig::OpenGL::Angle:
            QCoreApplication::setAttribute(Qt::AA_UseOpenGLES);
            break;

        case SessionConfig::OpenGL::Software:
            QCoreApplication::setAttribute(Qt::AA_UseSoftwareOpenGL);
            break;

        default:
            break;
        }
    }
#endif

    Application app(argc, argv);

    initWebEngineSettings();

    QAccessible::installFactory(&FakeAccessible::accessibleFactory);

    {
        const QString iconPath = ":/vnotex/data/core/icons/vnote.ico";
        // Make sense only on Windows.
        app.setWindowIcon(QIcon(iconPath));

        app.setApplicationName(ConfigMgr::c_appName);
        app.setOrganizationName(ConfigMgr::c_orgName);

        app.setApplicationVersion(ConfigMgr::getApplicationVersion());
    }

    CommandLineOptions cmdOptions;
    switch (cmdOptions.parse(app.arguments())) {
    case CommandLineOptions::Ok:
        break;

    case CommandLineOptions::Error:
        fprintf(stderr, "%s\n", qPrintable(cmdOptions.m_errorMsg));
        // Arguments to WebEngineView will be unknown ones. So just let it go.
        break;

    case CommandLineOptions::VersionRequested:
    {
        auto versionStr = QString("%1 %2").arg(app.applicationName()).arg(app.applicationVersion());
        showMessageOnCommandLineIfAvailable(versionStr);
        return 0;
    }

    case CommandLineOptions::HelpRequested:
        Q_FALLTHROUGH();
    default:
        showMessageOnCommandLineIfAvailable(cmdOptions.m_helpText);
        return 0;
    }

    // Guarding.
    SingleInstanceGuard guard;
    bool canRun = guard.tryRun();
    if (!canRun) {
        guard.requestOpenFiles(cmdOptions.m_pathsToOpen);
        guard.requestShow();
        return 0;
    }

    try {
        ConfigMgr::getInst();
    } catch (Exception &e) {
        MessageBoxHelper::notify(MessageBoxHelper::Critical,
                                 MainWindow::tr("%1 failed to start.").arg(ConfigMgr::c_appName),
                                 MainWindow::tr("Failed to initialize configuration manager. "
                                                "Please check if all the files are intact or reinstall the application."),
                                 e.what());
        return -1;
    }

    // Init logger after app info is set.
    Logger::init(cmdOptions.m_verbose);

    qInfo() << QString("%1 (v%2) started at %3 (%4)").arg(ConfigMgr::c_appName,
                                                          app.applicationVersion(),
                                                          QDateTime::currentDateTime().toString(),
                                                          QSysInfo::productType());

    qInfo() << "OpenSSL build version:" << QSslSocket::sslLibraryBuildVersionString()
            << "link version:" << QSslSocket::sslLibraryVersionString();

    if (QSslSocket::sslLibraryBuildVersionNumber() != QSslSocket::sslLibraryVersionNumber()) {
        qWarning() << "versions of the built and linked OpenSSL mismatch, network may not work";
    }

    // Should set the correct locale before VNoteX::getInst().
    loadTranslators(app);

    MainWindow window;
    window.show();

    QObject::connect(&guard, &SingleInstanceGuard::showRequested,
                     &window, &MainWindow::showMainWindow);
    QObject::connect(&guard, &SingleInstanceGuard::openFilesRequested,
                     &window, &MainWindow::openFiles);

    QObject::connect(&app, &Application::openFileRequested,
                     &window, [&window](const QString &p_filePath) {
                         window.openFiles(QStringList() << p_filePath);
                     });

    // Let MainWindow show first to decide the screen on which app is running.
    WidgetUtils::calculateScaleFactor(window.windowHandle()->screen());

    if (app.styleSheet().isEmpty()) {
        auto style = VNoteX::getInst().getThemeMgr().fetchQtStyleSheet();
        if (!style.isEmpty()) {
            app.setStyleSheet(style);
            WidgetUtils::updateStyle(&window);
        }
    }

    VNoteX::getInst().getThemeMgr().setBaseBackground(window.palette().color(QPalette::Base));

    window.kickOffOnStart(cmdOptions.m_pathsToOpen);

    int ret = app.exec();
    if (ret == RESTART_EXIT_CODE) {
        // Asked to restart VNote.
        guard.exit();
        QProcess::startDetached(QCoreApplication::applicationFilePath(), QStringList());
        // Must use exit() in Linux to quit the parent process in Qt 5.12.
        // Thanks to @ygcaicn.
        exit(0);
        return 0;
    }

    return ret;
}

void loadTranslators(QApplication &p_app)
{
    auto localeName = ConfigMgr::getInst().getCoreConfig().getLocale();
    if (!localeName.isEmpty()) {
        QLocale::setDefault(QLocale(localeName));
    }

    QLocale locale;
    qInfo() << "locale:" << locale.name();

    const QString resourceTranslationFolder(QStringLiteral(":/vnotex/data/core/translations"));
    const QString envTranslationFolder(QStringLiteral("translations"));

    // For QTextEdit/QTextBrowser and other basic widgets.
    QScopedPointer<QTranslator> qtbaseTranslator(new QTranslator(&p_app));
    if (qtbaseTranslator->load(locale, "qtbase", "_", resourceTranslationFolder)) {
        p_app.installTranslator(qtbaseTranslator.take());
    }

    // qt_zh_CN.ts does not cover the real QDialogButtonBox which uses QPlatformTheme.
    QScopedPointer<QTranslator> dialogButtonBoxTranslator(new QTranslator(&p_app));
    if (dialogButtonBoxTranslator->load(locale, "qdialogbuttonbox", "_", resourceTranslationFolder)) {
        p_app.installTranslator(dialogButtonBoxTranslator.take());
    }

    QScopedPointer<QTranslator> webengineTranslator(new QTranslator(&p_app));
    if (webengineTranslator->load(locale, "qwebengine", "_", resourceTranslationFolder)) {
        p_app.installTranslator(webengineTranslator.take());
    }

    // Load translation for Qt from resource.
    QScopedPointer<QTranslator> qtTranslator(new QTranslator(&p_app));
    if (qtTranslator->load(locale, "qt", "_", resourceTranslationFolder)) {
        p_app.installTranslator(qtTranslator.take());
    }

    // Load translation for Qt from env.
    QScopedPointer<QTranslator> qtEnvTranslator(new QTranslator(&p_app));
    if (qtEnvTranslator->load(locale, "qt", "_", envTranslationFolder)) {
        p_app.installTranslator(qtEnvTranslator.take());
    }

    // Load translation for vnote from resource.
    QScopedPointer<QTranslator> vnoteTranslator(new QTranslator(&p_app));
    if (vnoteTranslator->load(locale, "vnote", "_", resourceTranslationFolder)) {
        p_app.installTranslator(vnoteTranslator.take());
    }

    // Load translation for vtextedit from resource.
    QScopedPointer<QTranslator> vtexteditTranslator(new QTranslator(&p_app));
    if (vtexteditTranslator->load(locale, "vtextedit", "_", resourceTranslationFolder)) {
        p_app.installTranslator(vtexteditTranslator.take());
    }
}

void initWebEngineSettings()
{
    auto settings = QWebEngineSettings::defaultSettings();
    settings->setAttribute(QWebEngineSettings::LocalContentCanAccessRemoteUrls, true);
}

void showMessageOnCommandLineIfAvailable(const QString &p_msg)
{
#if defined(Q_OS_WIN)
    MessageBoxHelper::notify(MessageBoxHelper::Information,
                             QString("<pre>%1</pre>").arg(p_msg));
#else
    fprintf(stderr, "%s\n", qPrintable(p_msg));
#endif
}