#include <QApplication>
#include <QFontDatabase>
#include <QCommandLineParser>
#include <QMessageBox>
#include <QTranslator>
#include <QTimer>
#include <QLoggingCategory>
#include <QGuiApplication>
#include <QQmlApplicationEngine>
#include <QQmlContext>
#include "ui/uilogic.h"
#include <QDebug>

#include "ui/pages.h"

#include "ui/pages_logic/AppSettingsLogic.h"
#include "ui/pages_logic/GeneralSettingsLogic.h"
#include "ui/pages_logic/NetworkSettingsLogic.h"
#include "ui/pages_logic/NewServerLogic.h"
#include "ui/pages_logic/ProtocolSettingsLogic.h"
#include "ui/pages_logic/ServerListLogic.h"
#include "ui/pages_logic/ServerSettingsLogic.h"
#include "ui/pages_logic/ServerVpnProtocolsLogic.h"
#include "ui/pages_logic/ShareConnectionLogic.h"
#include "ui/pages_logic/SitesLogic.h"
#include "ui/pages_logic/StartPageLogic.h"
#include "ui/pages_logic/VpnLogic.h"
#include "ui/pages_logic/WizardLogic.h"


#include "debug.h"
#include "defines.h"

#if !defined(Q_OS_ANDROID) && !defined(Q_OS_IOS)
#define QAPPLICATION_CLASS QGuiApplication
#include "singleapplication.h"
#undef QAPPLICATION_CLASS
#endif


#ifdef Q_OS_WIN
#include "Windows.h"
#endif

static void loadTranslator()
{
    QTranslator* translator = new QTranslator;
    if (translator->load(QLocale(), QString("amneziavpn"), QLatin1String("_"), QLatin1String(":/translations"))) {
        qApp->installTranslator(translator);
    }
}

int main(int argc, char *argv[])
{
    QLoggingCategory::setFilterRules(QStringLiteral("qtc.ssh=false"));

    QGuiApplication::setAttribute(Qt::AA_EnableHighDpiScaling, true);

#ifdef Q_OS_WIN
    AllowSetForegroundWindow(ASFW_ANY);
#endif

#if !defined(Q_OS_ANDROID) && !defined(Q_OS_IOS)
    SingleApplication app(argc, argv, true, SingleApplication::Mode::User | SingleApplication::Mode::SecondaryNotification);

    if (!app.isPrimary()) {
        QTimer::singleShot(1000, &app, [&](){
            app.quit();
        });
        return app.exec();
    }
#else
    QApplication app(argc, argv);
#endif

#ifdef Q_OS_WIN
    AllowSetForegroundWindow(0);
#endif

    loadTranslator();

    QFontDatabase::addApplicationFont(":/fonts/Lato-Black.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-BlackItalic.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-Bold.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-BoldItalic.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-Italic.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-Light.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-LightItalic.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-Regular.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-Thin.ttf");
    QFontDatabase::addApplicationFont(":/fonts/Lato-ThinItalic.ttf");

    app.setApplicationName(APPLICATION_NAME);
    app.setOrganizationName(ORGANIZATION_NAME);
    app.setApplicationDisplayName(APPLICATION_NAME);

    QCommandLineParser parser;
    parser.setApplicationDescription(APPLICATION_NAME);
    parser.addHelpOption();
    parser.addVersionOption();

    QCommandLineOption c_autostart {{"a", "autostart"}, "System autostart"};
    parser.addOption(c_autostart);

    parser.process(app);

    if (!Debug::init()) {
        qWarning() << "Initialization of debug subsystem failed";
    }

    app.setQuitOnLastWindowClosed(false);

    UiLogic *uiLogic = new UiLogic;

    AppSettingsLogic *appSettingsLogic = new AppSettingsLogic(uiLogic);
    GeneralSettingsLogic *generalSettingsLogic = new GeneralSettingsLogic(uiLogic);
    NetworkSettingsLogic *networkSettingsLogic = new NetworkSettingsLogic(uiLogic);
    NewServerLogic *newServerLogic = new NewServerLogic(uiLogic);
    ProtocolSettingsLogic *protocolSettingsLogic = new ProtocolSettingsLogic(uiLogic);
    ServerListLogic *serverListLogic = new ServerListLogic(uiLogic);
    ServerSettingsLogic *serverSettingsLogic = new ServerSettingsLogic(uiLogic);
    ServerVpnProtocolsLogic *serverVpnProtocolsLogic = new ServerVpnProtocolsLogic(uiLogic);
    ShareConnectionLogic *shareConnectionLogic = new ShareConnectionLogic(uiLogic);
    SitesLogic *sitesLogic = new SitesLogic(uiLogic);
    StartPageLogic *startPageLogic = new StartPageLogic(uiLogic);
    VpnLogic *vpnLogic = new VpnLogic(uiLogic);
    WizardLogic *wizardLogic = new WizardLogic(uiLogic);

    QQmlApplicationEngine engine;
    PageEnumNS::declareQML();
    const QUrl url(QStringLiteral("qrc:/ui/qml/main.qml"));
    QObject::connect(&engine, &QQmlApplicationEngine::objectCreated,
                     &app, [url](QObject *obj, const QUrl &objUrl) {
        if (!obj && url == objUrl)
            QCoreApplication::exit(-1);
    }, Qt::QueuedConnection);

    engine.rootContext()->setContextProperty("UiLogic", uiLogic);

    engine.rootContext()->setContextProperty("AppSettingsLogic", appSettingsLogic);
    engine.rootContext()->setContextProperty("GeneralSettingsLogic", generalSettingsLogic);
    engine.rootContext()->setContextProperty("NetworkSettingsLogic", networkSettingsLogic);
    engine.rootContext()->setContextProperty("NewServerLogic", newServerLogic);
    engine.rootContext()->setContextProperty("ProtocolSettingsLogic", protocolSettingsLogic);
    engine.rootContext()->setContextProperty("ServerListLogic", serverListLogic);
    engine.rootContext()->setContextProperty("ServerSettingsLogic", serverSettingsLogic);
    engine.rootContext()->setContextProperty("ServerVpnProtocolsLogic", serverVpnProtocolsLogic);
    engine.rootContext()->setContextProperty("ShareConnectionLogic", shareConnectionLogic);
    engine.rootContext()->setContextProperty("SitesLogic", sitesLogic);
    engine.rootContext()->setContextProperty("StartPageLogic", startPageLogic);
    engine.rootContext()->setContextProperty("VpnLogic", vpnLogic);
    engine.rootContext()->setContextProperty("WizardLogic", wizardLogic);

    engine.load(url);

    // TODO - fix
//#ifdef Q_OS_WIN
//    if (parser.isSet("a")) mainWindow.showOnStartup();
//    else mainWindow.show();
//#else
//    mainWindow.showOnStartup();
//#endif


     // TODO - fix
//#if !defined(Q_OS_ANDROID) && !defined(Q_OS_IOS)
//    if (app.isPrimary()) {
//        QObject::connect(&app, &SingleApplication::instanceStarted, &mainWindow, [&](){
//            qDebug() << "Secondary instance started, showing this window instead";
//            mainWindow.show();
//            mainWindow.showNormal();
//            mainWindow.raise();
//        });
//    }
//#endif

    return app.exec();
}