Qv2ray/src/Qv2rayApplication.hpp

#pragma once

#include "core/handler/ConfigHandler.hpp"
#include "core/handler/RouteHandler.hpp"
#include "libs/QJsonStruct/QJsonStruct.hpp"

#include <QSystemTrayIcon>
#ifdef Q_OS_ANDROID
    #include <QApplication>
#else
    #include <SingleApplication>
#endif

class MainWindow;

namespace Qv2ray
{
    enum Qv2rayExitCode
    {
        QV2RAY_NORMAL = 0,
        QV2RAY_SECONDARY_INSTANCE = 0,
        QV2RAY_PRE_INITIALIZE_FAIL = -1,
        QV2RAY_EARLY_SETUP_FAIL = -2,
        QV2RAY_CONFIG_PATH_FAIL = -3,
        QV2RAY_CONFIG_FILE_FAIL = -4,
        QV2RAY_SSL_FAIL = -5,
        QV2RAY_NEW_VERSION = -6
    };
    struct Qv2rayProcessArguments
    {
        enum Argument
        {
            NORMAL = 0,
            QV2RAY_LINK = 1,
            EXIT = 2,
            RECONNECT = 3,
            DISCONNECT = 4
        };
        QList<Argument> arguments;
        QString version;
        QString data;
        QList<QString> links;
        QList<QString> fullArgs;
        //
        QString _qvNewVersionPath;
        JSONSTRUCT_REGISTER(Qv2rayProcessArguments, F(arguments, version, data, links, fullArgs))
    };

    inline Qv2rayProcessArguments Qv2rayProcessArgument;
#ifdef Q_OS_ANDROID
    class Qv2rayApplication : public QApplication
#else
    class Qv2rayApplication : public SingleApplication
#endif
    {
        Q_OBJECT

        enum commandline_status
        {
            QV2RAY_ERROR,
            QV2RAY_QUIT,
            QV2RAY_CONTINUE
        };

      public:
        enum Qv2raySetupStatus
        {
            NORMAL,
            SINGLE_APPLICATION,
            FAILED
        };
        //
        void QuitApplication(int retCode = 0);
        static bool PreInitialize(int argc, char **argv);
        explicit Qv2rayApplication(int &argc, char *argv[]);
        Qv2raySetupStatus SetupQv2ray();
        bool FindAndCreateInitialConfiguration();
        bool LoadConfiguration();
        void InitializeGlobalVariables();
        Qv2rayExitCode RunQv2ray();

      public:
        QSystemTrayIcon **GetTrayIcon()
        {
            return &hTray;
        }
        void showMessage(const QString &m, const QIcon &icon, int msecs = 10000)
        {
            hTray->showMessage("Qv2ray", m, icon, msecs);
        }
        void showMessage(const QString &m, QSystemTrayIcon::MessageIcon icon = QSystemTrayIcon::Information, int msecs = 10000)
        {
            hTray->showMessage("Qv2ray", m, icon, msecs);
        }

      private slots:
        void aboutToQuitSlot();
        void onMessageReceived(quint32 clientID, QByteArray msg);

      private:
        QSystemTrayIcon *hTray;
        MainWindow *mainWindow;
        static commandline_status ParseCommandLine(QString *errorMessage, const QStringList &args);
        bool initialized = false;
    };
} // namespace Qv2ray

using namespace Qv2ray;

#define qvApp (dynamic_cast<Qv2ray::Qv2rayApplication *>(QCoreApplication::instance()))
#define qvAppTrayIcon (*qvApp->GetTrayIcon())