FrmUpdater.h

 
#ifndef RABBITCOMMON_FRMUPDATER_H
#define RABBITCOMMON_FRMUPDATER_H
 
#include <QSystemTrayIcon>
#include <QWidget>
#include <QFile>
#include <QPixmap>
#include <QApplication>
 
#include <QUrl>
#include <QButtonGroup>
#include <QCommandLineParser>
#include "rabbitcommon_export.h"
 
namespace RabbitCommon {
class CDownload;
}
 
namespace Ui {
class CFrmUpdater;
}
 
class QStateMachine;
class QState;
 
typedef int (*pUpdateCallback)(const QString szFile);
 
class RABBITCOMMON_EXPORT CFrmUpdater : public QWidget
{
    Q_OBJECT
 
public:
    explicit CFrmUpdater(QVector<QUrl> urls = QVector<QUrl>(), QWidget *parent = nullptr);
    virtual ~CFrmUpdater() override;
 
    int SetVersion(const QString &szVersion);
    int SetTitle(QImage icon = QImage(), const QString &szTitle = QString());
 
    Q_DECL_DEPRECATED
    int GenerateUpdateXml();
    Q_DECL_DEPRECATED
    int GenerateUpdateXml(QCommandLineParser &parser);
    int GenerateUpdateJson();
    int GenerateUpdateJson(QCommandLineParser &parser);
 
    int SetInstallAutoStartup(bool bAutoStart = true);
 
    int SetUpdateCallback(pUpdateCallback pCb);
 
protected Q_SLOTS:
    // [Add the slot functions of RabbitCommon::CDownload]
    void slotDownloadError(int nErr, const QString szError);
    void slotDownloadFileFinished(const QString szFile);
    void slotDownloadProgress(qint64 bytesReceived, qint64 bytesTotal);
    // [Add the slot functions of RabbitCommon::CDownload]
 
    void slotButtonClickd(int id);
    void slotCheck();
    void slotDownloadFile();
    virtual void slotCheckConfigFile();
    void slotDownloadSetupFile();
    void slotUpdate();
    void slotStateFinished();
 
    void on_pbOK_clicked();
    void on_pbClose_clicked();
    void slotShowWindow(QSystemTrayIcon::ActivationReason reason);
 
Q_SIGNALS:
    void sigError();
    void sigFinished();
    void sigDownLoadRedire();
 
private:
    CFrmUpdater(QWidget *parent);
    int CompareVersion(const QString &newVersion, const QString &currentVersion);
    int InitStateMachine();
    bool IsDownLoad();
    int CheckRedirectConfigFile();
    int CheckUpdateConfigFile();
    bool CheckPrompt(const QString &szVersion);
    QString InstallScript(const QString szDownLoadFile,
                          const QString szApplicationName);
    Q_INVOKABLE int Execute(const QString szFile);
 
private:
    Ui::CFrmUpdater *ui;
    bool m_InstallAutoStartupType;
 
    QSystemTrayIcon m_TrayIcon;
 
    QButtonGroup m_ButtonGroup;
 
    QString m_szCurrentVersion;
 
    QFile m_DownloadFile;
    bool m_bDownload;
    QVector<QUrl> m_Urls;
    // [Instantiate the object of RabbitCommon::CDownload]
    QSharedPointer<RabbitCommon::CDownload> m_Download;
    // [Instantiate the object of RabbitCommon::CDownload]
 
    QStateMachine* m_StateMachine;
    QState *m_pStateDownloadSetupFile;
 
    enum class CONFIG_TYPE{
        VERSION,
        FILE,
        VERSION_FILE
    };
 
    struct CONFIG_VERSION {
        QString szVerion;
        QString szMinUpdateVersion;
        QString szInfomation;
        QString szTime;
        bool bForce;
        QString szHome;
    };
 
    struct CONFIG_FILE {
        QString szSystem;
        QString szSystemMinVersion;
        QString szArchitecture;
        QString szArchitectureMinVersion;
        QString szFileName;
        QVector<QUrl> urls;
        QString szMd5sum;
    };
    
    struct CONFIG_REDIRECT {
        QString szVersion;
        QString szMinUpdateVersion;
        QVector<CONFIG_FILE> files;
    };
 
    struct CONFIG_INFO {
        CONFIG_VERSION version;
        QVector<CONFIG_FILE> files;
    } m_Info;
    
    CONFIG_FILE m_ConfigFile;
    
    int GetRedirectFromFile(const QString& szFile, QVector<CONFIG_REDIRECT> &conf);
    int GetConfigFromFile(const QString& szFile, CONFIG_INFO &conf);
    int GetConfigFromCommandLine(/*[in]*/QCommandLineParser &parser,
                /*[out]*/QString &szFile,
                /*[out]*/CONFIG_INFO &info,
                /*[out]*/CONFIG_TYPE &type);
    int GenerateUpdateXmlFile(const QString &szFile,
                              const CONFIG_INFO &info,
                              CONFIG_TYPE &type);
    int GenerateJsonFile(const QString &szFile,
                         const CONFIG_INFO &info,
                         CONFIG_TYPE type);
    
    pUpdateCallback m_pcbUpdate;
 
    // QWidget interface
protected:
    virtual void showEvent(QShowEvent *event) override;
private Q_SLOTS:
    void on_cbPrompt_clicked(bool checked);
    void on_cbHomePage_clicked(bool checked);
 
#if defined(HAVE_TEST)
    friend class CUnitTests;
    friend class CTestGenerateJsonFile;
    friend class CTestGenerateDefaultJsonFile;
    friend class CTestUpdater;
#endif
};
 
#endif // RABBITCOMMON_FRMUPDATER_H