Imported Upstream version 2.99.6upstream/2.99.6

1 files changed, 137 insertions, 0 deletions

diff --git a/SQLiteStudio3/coreSQLiteStudio/services/updatemanager.h b/SQLiteStudio3/coreSQLiteStudio/services/updatemanager.h
new file mode 100644
index 0000000..b8e6006
--- /dev/null
+++ b/SQLiteStudio3/coreSQLiteStudio/services/updatemanager.h
@@ -0,0 +1,137 @@
+#ifndef UPDATEMANAGER_H
+#define UPDATEMANAGER_H
+
+#include "common/global.h"
+#include "sqlitestudio.h"
+#include <QObject>
+#include <functional>
+#include <QProcess>
+
+class QNetworkAccessManager;
+class QNetworkReply;
+class QTemporaryDir;
+class QFile;
+
+class API_EXPORT UpdateManager : public QObject
+{
+        Q_OBJECT
+    public:
+        typedef std::function<bool(const QString& msg)> RetryFunction;
+
+        struct UpdateEntry
+        {
+            QString compontent;
+            QString version;
+            QString url;
+        };
+
+        explicit UpdateManager(QObject *parent = 0);
+        ~UpdateManager();
+
+        void checkForUpdates();
+        void update();
+        bool isPlatformEligibleForUpdate() const;
+        static bool executeFinalStep(const QString& tempDir, const QString& backupDir, const QString& appDir);
+        static bool handleUpdateOptions(const QStringList& argList, int& returnCode);
+        static QString getStaticErrorMessage();
+
+        static void setRetryFunction(const RetryFunction &value);
+
+        static_char* UPDATE_OPTION_NAME = "--update-final-step";
+        static_char* WIN_INSTALL_FILE = "install.dat";
+        static_char* WIN_UPDATE_DONE_FILE = "UpdateFinished.lck";
+
+    private:
+        enum class LinuxPermElevator
+        {
+            KDESU,
+            GKSU,
+            PKEXEC,
+            NONE
+        };
+
+        QString getPlatformForUpdate() const;
+        QString getCurrentVersions() const;
+        void handleAvailableUpdatesReply(QNetworkReply* reply);
+        void handleDownloadReply(QNetworkReply* reply);
+        void getUpdatesMetadata(QNetworkReply*& replyStoragePointer);
+        void handleUpdatesMetadata(QNetworkReply* reply);
+        QList<UpdateEntry> readMetadata(const QJsonDocument& doc);
+        void downloadUpdates();
+        void updatingFailed(const QString& errMsg);
+        void installUpdates();
+        bool installComponent(const QString& component, const QString& tempDir);
+        bool executeFinalStep(const QString& tempDir);
+        bool executeFinalStepAsRoot(const QString& tempDir, const QString& backupDir, const QString& appDir);
+#if defined(Q_OS_LINUX)
+        bool executeFinalStepAsRootLinux(const QString& tempDir, const QString& backupDir, const QString& appDir);
+        bool unpackToDirLinux(const QString& packagePath, const QString& outputDir);
+#elif defined(Q_OS_MACX)
+        bool unpackToDirMac(const QString& packagePath, const QString& outputDir);
+        bool executeFinalStepAsRootMac(const QString& tempDir, const QString& backupDir, const QString& appDir);
+#elif defined(Q_OS_WIN32)
+        bool runAnotherInstanceForUpdate(const QString& tempDir, const QString& backupDir, const QString& appDir, bool reqAdmin);
+        bool unpackToDirWin(const QString& packagePath, const QString& outputDir);
+#endif
+        bool doRequireAdminPrivileges();
+        bool unpackToDir(const QString& packagePath, const QString& outputDir);
+        void handleStaticFail(const QString& errMsg);
+        QString getAppDirPath() const;
+        void cleanup();
+
+        static bool moveDir(const QString& src, const QString& dst, bool contentsOnly = false);
+        static bool deleteDir(const QString& path);
+        static bool execCmd(const QString& cmd, const QStringList& args, QString* errorMsg = nullptr);
+        static bool waitForProcess(QProcess& proc);
+        static QString readError(QProcess& proc, bool reverseOrder = false);
+        static void staticUpdatingFailed(const QString& errMsg);
+        static LinuxPermElevator findPermElevatorForLinux();
+        static QString wrapCmdLineArgument(const QString& arg);
+        static QString escapeCmdLineArgument(const QString& arg);
+        static QString getBackupDir(const QString& appDir);
+#if defined(Q_OS_WIN32)
+        static bool executePreFinalStepWin(const QString& tempDir, const QString& backupDir, const QString& appDir, bool reqAdmin);
+        static bool executeFinalStepAsRootWin(const QString& tempDir, const QString& backupDir, const QString& appDir);
+        static bool executePostFinalStepWin(const QString& tempDir);
+        static bool waitForFileToDisappear(const QString& filePath, int seconds);
+        static bool waitForFileToAppear(const QString& filePath, int seconds);
+#endif
+
+        QNetworkAccessManager* networkManager = nullptr;
+        QNetworkReply* updatesCheckReply = nullptr;
+        QNetworkReply* updatesGetUrlsReply = nullptr;
+        QNetworkReply* updatesGetReply = nullptr;
+        bool updatesInProgress = false;
+        QList<UpdateEntry> updatesToDownload;
+        QHash<QString,QString> updatesToInstall;
+        QTemporaryDir* tempDir = nullptr;
+        QFile* currentDownloadFile = nullptr;
+        int totalPercent = 0;
+        int totalDownloadsCount = 0;
+        QString currentJobTitle;
+        bool requireAdmin = false;
+        static RetryFunction retryFunction;
+
+        static QString staticErrorMessage;
+        static_char* WIN_PRE_FINAL_UPDATE_OPTION_NAME = "--update-pre-final-step";
+        static_char* WIN_POST_FINAL_UPDATE_OPTION_NAME = "--update-post-final-step";
+        static_char* WIN_UPDATER_BINARY = "UpdateSQLiteStudio.exe";
+        static_char* updateServiceUrl = "http://sqlitestudio.pl/updates3.rvt";
+        static_char* manualUpdatesHelpUrl = "http://wiki.sqlitestudio.pl/index.php/User_Manual#Manual";
+
+    private slots:
+        void finished(QNetworkReply* reply);
+        void downloadProgress(qint64 bytesReceived, qint64 totalBytes);
+        void readDownload();
+
+    signals:
+        void updatesAvailable(const QList<UpdateManager::UpdateEntry>& updates);
+        void noUpdatesAvailable();
+        void updatingProgress(const QString& jobTitle, int jobPercent, int totalPercent);
+        void updatingFinished();
+        void updatingError(const QString& errorMessage);
+};
+
+#define UPDATES SQLITESTUDIO->getUpdateManager()
+
+#endif // UPDATEMANAGER_H