AUpdater.h

// AUI Framework - Declarative UI toolkit for modern C++20
// Copyright (C) 2020-2025 Alex2772 and Contributors
//
// SPDX-License-Identifier: MPL-2.0
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#pragma once
 
#include <AUI/Common/AObject.h>
#include <AUI/Common/AStringVector.h>
#include <AUI/Common/AProperty.h>
#include <AUI/Thread/AAsyncHolder.h>
#include "AUI/IO/APath.h"
#include "AUI/Thread/AFuture.h"
#include <AUI/Platform/AProcess.h>

class API_AUI_UPDATER AUpdater : public AObject {
public:
   struct InstallCmdline {
       APath installerExecutable;

       AVector<AString> installerArguments;
   };

   struct GetInstallationDirectoryContext {
       APath selfProcessExePath = AProcess::self()->getPathToExecutable();

       APath updaterDir;

       APath originExe;
   };
 
   AUpdater();
   ~AUpdater() override = default;

   virtual void handlePostUpdateCleanup();

   virtual void handleStartup(const AStringVector& applicationArguments);

   virtual void applyUpdateAndRestart();

   struct StatusIdle {};

   struct StatusCheckingForUpdates {};

   struct StatusDownloading {
       mutable AProperty<aui::float_within_0_1> progress;
   };

   struct StatusNotAvailable {};

   struct StatusWaitingForApplyAndRestart {
       InstallCmdline installCmdline;
   };

   AProperty<std::any> status;

   void checkForUpdates();

   void downloadUpdate();
 
   [[nodiscard]]
   const AOptional<AString>& getLastDeploymentError() const noexcept { return mLastDeploymentError; }

   static bool isAvailable();
 
protected:
   AAsyncHolder mAsync;

   InstallCmdline makeDefaultInstallationCmdline() const;

   virtual void triggerUpdateOnStartup();

   virtual AFuture<void> downloadUpdateImpl(const APath& unpackedUpdateDir) = 0;

   virtual AFuture<void> checkForUpdatesImpl() = 0;

   void downloadAndUnpack(AString downloadUrl, const APath& unpackedUpdateDir);

   void reportDownloadedPercentage(aui::float_within_0_1 progress);

   virtual AString getModuleName() const;

   void reportReadyToApplyAndRestart(InstallCmdline cmdline);

   virtual APath getTempWorkDir() const;

   virtual APath getUnpackedUpdateDir() const;

   virtual AOptional<InstallCmdline> loadInstallCmdline() const;

   virtual void saveCmdline(const InstallCmdline& cmdline) const;

   virtual void handleWaitForProcess(uint32_t pid);

   virtual AVector<AString> injectWaitForMyPid(AVector<AString> args);

   virtual void deployUpdate(const APath& source, const APath& destination);

   virtual APath getInstallationDirectory(const GetInstallationDirectoryContext& context);

   virtual void cleanupUnpackedUpdateDirBeforeDownloading();
 
private:
   AOptional<AString> mLastDeploymentError;
};

© 2025 AUI Framework Contributors. Documentation contributions included herein are the copyrights of their respective owners. The documentation provided herein is licensed under the terms of the GNU Free Documentation License version 1.3 as published by the Free Software Foundation. All trademarks are property of their respective owners.