CodePushNativeModule.cpp

// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

#include "pch.h"

#include "CodePushNativeModule.h"
#include "CodePushUtils.h"
#include "CodePushUpdateUtils.h"
#include "CodePushPackage.h"
#include "CodePushTelemetryManager.h"
#include "CodePushConfig.h"
#include "CodePushUtils.h"

#include <string_view>

#include "miniz/miniz.h"

#include "winrt/Windows.ApplicationModel.h"
#include "winrt/Windows.Data.Json.h"
#include "winrt/Windows.Storage.FileProperties.h"

#include "ReactPackageProvider.h"

namespace Microsoft::CodePush::ReactNative
{
    using namespace winrt;
    using namespace winrt::Microsoft::ReactNative;
    using namespace Windows::Data::Json;
    using namespace Windows::Storage;
    using namespace Windows::Foundation;

    ReactNativeHost CodePushNativeModule::s_host{};
    CodePushNativeModule::CodePushInstallMode CodePushNativeModule::s_installMode{};
    bool CodePushNativeModule::isRunningBinaryVersion{ false };
    bool CodePushNativeModule::needToReportRollback{ false };
    /*static*/ bool CodePushNativeModule::s_initialized{ false };
    /*static*/ hstring CodePushNativeModule::s_javaScriptBundleFileName{ L"index.windows" };

    /*static*/ IAsyncOperation<StorageFile> CodePushNativeModule::GetBinaryBundleAsync()
    {
        auto appXFolder{ Windows::ApplicationModel::Package::Current().InstalledLocation() };
        auto bundleFolder{ (co_await appXFolder.TryGetItemAsync(L"Bundle")).try_as<StorageFolder>() };
        if (bundleFolder == nullptr)
        {
            co_return nullptr;
        }

        auto bundleFile{ (co_await bundleFolder.TryGetItemAsync(s_javaScriptBundleFileName + BundleExtension)).try_as<StorageFile>() };
        co_return bundleFile;
    }

    /*static*/ IAsyncOperation<StorageFile> CodePushNativeModule::GetBundleFileAsync()
    {
        auto bundleFileName{ s_host.InstanceSettings().JavaScriptBundleFile() };
        if (!bundleFileName.empty())
        {
            s_javaScriptBundleFileName = bundleFileName;
        }

        auto packageBundle{ co_await CodePushPackage::GetCurrentPackageBundleAsync() };
        auto binaryBundle{ co_await GetBinaryBundleAsync() };

        if (packageBundle == nullptr)
        {
            CodePushUtils::LogBundleUrl(binaryBundle);
            isRunningBinaryVersion = true;
            co_return binaryBundle;
        }

        auto binaryAppVersion{ CodePushConfig::Current().GetAppVersion() };
        auto currentPackageMetadata{ co_await CodePushPackage::GetCurrentPackageAsync() };
        if (currentPackageMetadata == nullptr)
        {
            CodePushUtils::LogBundleUrl(binaryBundle);
            isRunningBinaryVersion = true;
            co_return binaryBundle;
        }

        auto packageDate{ currentPackageMetadata.GetNamedString(BinaryBundleDateKey, L"") };
        auto packageAppVersion{ currentPackageMetadata.GetNamedString(AppVersionKey, L"") };

        if ((co_await CodePushUpdateUtils::ModifiedDateStringOfFileAsync(binaryBundle)) == packageDate && binaryAppVersion == packageAppVersion)
        {
            // Return package file because it is newer than the JS bundle in the AppX folder
            CodePushUtils::LogBundleUrl(packageBundle);
            isRunningBinaryVersion = false;
            co_return packageBundle;
        }
        else
        {
            auto isRelease{ false };
    #ifndef _DEBUG
            isRelease = true;
    #endif

            if (isRelease || binaryAppVersion != packageAppVersion)
            {
                co_await ClearUpdatesStaticAsync();
            }

            CodePushUtils::LogBundleUrl(binaryBundle);
            isRunningBinaryVersion = true;
            co_return binaryBundle;
        }
    }

    /*static*/ IAsyncOperation<StorageFolder> CodePushNativeModule::GetBundleAssetsFolderAsync()
    {
        auto appXFolder{ Windows::ApplicationModel::Package::Current().InstalledLocation() };
        auto bundleFolder{ (co_await appXFolder.TryGetItemAsync(L"Bundle")).try_as<StorageFolder>() };
        if (bundleFolder != nullptr)
        {
            auto bundleAssetsFolder{ (co_await bundleFolder.TryGetItemAsync(CodePushUpdateUtils::AssetsFolderName)).try_as<StorageFolder>() };
            co_return bundleAssetsFolder;
        }
        co_return nullptr;
    }

    // Rather than store files in the library files, CodePush for ReactNativeWindows will use AppData folders.
    /*static*/ StorageFolder CodePushNativeModule::GetLocalStorageFolder()
    {
        return ApplicationData::Current().LocalFolder();
    }

    /*static*/ ApplicationDataContainer CodePushNativeModule::GetLocalSettings()
    {
        return ApplicationData::Current().LocalSettings();
    }

    void CodePushNativeModule::OverrideAppVersion(std::wstring_view appVersion) 
    {
        CodePushConfig::Current().SetAppVersion(appVersion);
    }

    void CodePushNativeModule::SetDeploymentKey(std::wstring_view deploymentKey) 
    {
        CodePushConfig::Current().SetDeploymentKey(deploymentKey);
    }

    /*
     * This method checks to see whether a specific package hash
     * has previously failed installation.
     */
    bool CodePushNativeModule::IsFailedHash(std::wstring_view packageHash) 
    { 
        auto localSettings{ GetLocalSettings() };
        auto failedUpdatesData{ localSettings.Values().TryLookup(FailedUpdatesKey) };
        if (failedUpdatesData == nullptr)
        {
            return false;
        }
        auto failedUpdatesString{ unbox_value<hstring>(failedUpdatesData) };
        JsonArray failedUpdates;
        auto success{ JsonArray::TryParse(failedUpdatesString, failedUpdates) };
        if (!success || packageHash.empty())
        {
            return false;
        }
        else
        {
            for (const auto& failedPackage : failedUpdates)
            {
                // We don't have to worry about backwards compatability, but just to be safe...
                if (failedPackage.ValueType() == JsonValueType::Object)
                {
                    auto failedPackageHash{ failedPackage.GetObject().GetNamedString(PackageHashKey) };
                    if (packageHash == failedPackageHash)
                    {
                        return true;
                    }
                }
            }

            return false;
        }
    }

    /*
     * This method is used to get information about the latest rollback.
     * This information will be used to decide whether the application
     * should ignore the update or not.
     */
    JsonObject CodePushNativeModule::GetRollbackInfo() { return nullptr; }

    /*
     * This method is used to get the count of rollback for the package
     * using the latest rollback information.
     */
    int CodePushNativeModule::GetRollbackCountForPackage(std::wstring_view packageHash, const JsonObject& latestRollbackInfo) 
    {
        auto oldPackageHash{ latestRollbackInfo.GetNamedString(LatestRollbackPackageHashKey, L"null") };
        if (packageHash == oldPackageHash)
        {
            auto oldCount{ latestRollbackInfo.GetNamedNumber(LatestRollbackCountKey, 0) };
            return static_cast<int>(oldCount);
        }
        return 0; 
    }

    /*
     * This method checks to see whether a specific package hash
     * represents a downloaded and installed update, that hasn't
     * been applied yet via an app restart.
     */
    /*static*/ bool CodePushNativeModule::IsPendingUpdate(std::wstring_view packageHash)
    { 
        auto localSettings{ GetLocalSettings() };
        auto pendingUpdateData{ localSettings.Values().TryLookup(PendingUpdateKey) };
        if (pendingUpdateData != nullptr)
        {
            auto pendingUpdateString{ unbox_value<hstring>(pendingUpdateData) };
            JsonObject pendingUpdate;
            auto success{ JsonObject::TryParse(pendingUpdateString, pendingUpdate) };
        
            // If there is a pending update whose "state" isn't loading, then we consider it "pending".
            // Additionally, if a specific hash was provided, we ensure it matches that of the pending update.
            auto updateIsPending{ success &&
                pendingUpdate != nullptr &&
                pendingUpdate.GetNamedBoolean(PendingUpdateIsLoadingKey, false) == false &&
                (packageHash.empty() || pendingUpdate.GetNamedString(PendingUpdateHashKey, L"null") == packageHash) };

            return updateIsPending;
        }
        return false;
    }

    /*
     * This method is used to clear updates that are installed
     * under a different app version and hence don't apply anymore,
     * during a debug run configuration and when React Native Windows is
     * running the JS bundle from the dev server.
     */
    IAsyncAction CodePushNativeModule::ClearDebugUpdates()
    {
    #ifndef BUNDLE
        auto binaryAppVersion{ CodePushConfig::Current().GetAppVersion() };
        auto currentPackageMetadata{ co_await CodePushPackage::GetCurrentPackageAsync() };
        if (currentPackageMetadata != nullptr)
        {
            auto packageAppVersion{ currentPackageMetadata.GetNamedString(AppVersionKey, L"") };
            if (binaryAppVersion.empty() || binaryAppVersion != packageAppVersion)
            {
                co_await ClearUpdatesStaticAsync();
            }
        }
    #endif
        co_return;
    }

    /*static*/ IAsyncAction CodePushNativeModule::ClearUpdatesStaticAsync()
    {
        co_await CodePushPackage::ClearUpdatesAsync();
        RemovePendingUpdate();
        RemoveFailedUpdates();
    }

    void CodePushNativeModule::DispatchDownloadProgressEvent()
    {
        // Notify the script-side about the progress
        m_context.CallJSFunction(
            L"RCTDeviceEventEmitter",
            L"emit",
            L"CodePushDownloadProgress",
            JSValueObject{
                {"totalBytes", m_latestExpectedContentLength },
                {"receivedBytes", m_latestReceivedContentLength } });
    }

    /*static*/ IAsyncAction CodePushNativeModule::LoadBundle()
    {
        if (!s_host.InstanceSettings().UseWebDebugger())
        {
            auto bundleFile{ co_await GetBundleFileAsync() };
            if (bundleFile != nullptr)
            {
                std::wstring_view bundlePath{ bundleFile.Path() };
                hstring bundleRootPath{ bundlePath.substr(0, bundlePath.rfind('\\')) };
                s_host.InstanceSettings().BundleRootPath(bundleRootPath);
            }
        }

        s_host.ReloadInstance();
        // The instance will call Initialize() upon reloading this module
    }

    /*
     * This method is used when an update has failed installation
     * and the app needs to be rolled back to the previous bundle.
     * This method is automatically called when the rollback timer
     * expires without the app indicating whether the update succeeded,
     * and therefore, it shouldn't be called directly.
     */
    IAsyncAction CodePushNativeModule::RollbackPackage()
    {
        auto failedPackage{ co_await CodePushPackage::GetCurrentPackageAsync() };
        if (failedPackage == nullptr)
        {
            CodePushUtils::Log(L"Attempted to perform a rollback when there is no current update.");
        }
        else
        {
            SaveFailedUpdate(failedPackage);
        }

        // Rollback to the previous version and de-register the new update
        co_await CodePushPackage::RollbackPackage();
        RemovePendingUpdate();
        co_await LoadBundle();
    }

    /*
     * This method is used to clear away failed updates in the event that
     * a new app store binary is installed.
     */

    /*static*/ void CodePushNativeModule::RemoveFailedUpdates()
    {
        auto localSettings{ GetLocalSettings() };
        localSettings.Values().TryRemove(FailedUpdatesKey);
    }

    /*
     * This method is used to register the fact that a pending
     * update succeeded and therefore can be removed.
     */
    /*static*/ void CodePushNativeModule::RemovePendingUpdate()
    {
        // remove pending update from LocalSettings
        auto localSettings{ GetLocalSettings() };
        localSettings.Values().TryRemove(PendingUpdateKey);
    }

    IAsyncAction CodePushNativeModule::RestartAppInternal(bool onlyIfUpdateIsPending)
    {
        if (m_restartInProgress)
        {
            CodePushUtils::Log(L"Restart request queued until the current restart is completed.");
            m_restartQueue.push_back(onlyIfUpdateIsPending);
        }
        else if (!m_allowed)
        {
            CodePushUtils::Log(L"Restart request queued until restarts are re-allowed.");
            m_restartQueue.push_back(onlyIfUpdateIsPending);
            co_return;
        }

        m_restartInProgress = true;
        if (!onlyIfUpdateIsPending || IsPendingUpdate(L""))
        {
            co_await LoadBundle();
            CodePushUtils::Log(L"Restarting app.");
            co_return;
        }

        m_restartInProgress = false;
        if (m_restartQueue.size() > 0)
        {
            auto buf{ m_restartQueue[0] };
            m_restartQueue.erase(m_restartQueue.begin());
            co_await RestartAppInternal(buf);
        }
    }

    /*
     * When an update failed to apply, this method can be called
     * to store its hash so that it can be ignored on future
     * attempts to check the server for an update.
     */
    void CodePushNativeModule::SaveFailedUpdate(JsonObject& failedPackage)
    {
        if (IsFailedHash(failedPackage.GetNamedString(PackageHashKey)))
        {
            return;
        }

        auto localSettings{ GetLocalSettings() };
        auto failedUpdates{ localSettings.Values().TryLookup(FailedUpdatesKey).try_as<JsonArray>() };
        if (failedUpdates == nullptr)
        {
            failedUpdates = JsonArray{};
        }

        failedUpdates.Append(failedPackage);
        localSettings.Values().Insert(FailedUpdatesKey, box_value(failedUpdates.Stringify()));
    }

    /*
     * When an update is installed whose mode isn't Immediate, this method
     * can be called to store the pending update's metadata (e.g. packageHash)
     * so that it can be used when the actual update application occurs at a later point.
     */
    void CodePushNativeModule::SavePendingUpdate(std::wstring_view packageHash, bool isLoading)
    {
        // Since we're not restarting, we need to store the fact that the update
        // was installed, but hasn't yet become "active".
        auto localSettings{ GetLocalSettings() };
        JsonObject pendingUpdate{};
        pendingUpdate.Insert(PendingUpdateHashKey, JsonValue::CreateStringValue(packageHash));
        pendingUpdate.Insert(PendingUpdateIsLoadingKey, JsonValue::CreateBooleanValue(isLoading));
        localSettings.Values().Insert(PendingUpdateKey, box_value(pendingUpdate.Stringify()));
    }

    /*static*/ void CodePushNativeModule::SetHost(const ReactNativeHost& host)
    {
        s_host = host;
    }

    void CodePushNativeModule::Initialize(ReactContext const& reactContext) noexcept
    {
        m_context = reactContext;
        InitializeUpdateAfterRestart();
    }

    void CodePushNativeModule::GetConstants(winrt::Microsoft::ReactNative::ReactConstantProvider& constants) noexcept
    {
        constants.Add(L"codePushInstallModeImmediate", CodePushInstallMode::Immediate);
        constants.Add(L"codePushInstallModeOnNextRestart", CodePushInstallMode::OnNextRestart);
        constants.Add(L"codePushInstallModeOnNextResume", CodePushInstallMode::OnNextResume);
        constants.Add(L"codePushInstallModeOnNextSuspend", CodePushInstallMode::OnNextSuspend);

        constants.Add(L"codePushUpdateStateRunning", CodePushUpdateState::Running);
        constants.Add(L"codePushUpdateStatePending", CodePushUpdateState::Pending);
        constants.Add(L"codePushUpdateStateLatest", CodePushUpdateState::Latest);
    }

    /*
     * This is native-side of the RemotePackage.download method
     */
    fire_and_forget CodePushNativeModule::DownloadUpdateAsync(JsonObject updatePackage, bool notifyProgress, ReactPromise<IJsonValue> promise) noexcept
    {
        auto binaryBundle{ co_await GetBinaryBundleAsync() };
        if (binaryBundle != nullptr)
        {
            auto modifiedDate{ co_await CodePushUpdateUtils::ModifiedDateStringOfFileAsync(binaryBundle) };
            updatePackage.Insert(BinaryBundleDateKey, JsonValue::CreateStringValue(modifiedDate));
        }

        auto publicKey{ CodePushConfig::Current().GetPublicKey() };

        try
        {
            co_await CodePushPackage::DownloadPackageAsync(
                updatePackage,
                s_javaScriptBundleFileName + BundleExtension,
                /* publicKey */ publicKey,
                /* progressCallback */ [=](int64_t expectedContentLength, int64_t receivedContentLength) {
                    // React-Native-Windows doesn't have a frame observer to my knowledge.
                    if (notifyProgress)
                    {
                        m_latestExpectedContentLength = expectedContentLength;
                        m_latestReceivedContentLength = receivedContentLength;
                        DispatchDownloadProgressEvent();
                    }
                });
        }
        catch (const hresult_error& ex)
        {
            SaveFailedUpdate(updatePackage);

            m_didUpdateProgress = false;
            promise.Reject(ex.message().c_str());
        }

        auto newPackage{ co_await CodePushPackage::GetPackageAsync(updatePackage.GetNamedString(PackageHashKey)) };
        if (newPackage == nullptr)
        {
            promise.Reject(L"An error has occurred retreiving the downloaded package.");
        }
        else
        {
            promise.Resolve(newPackage);
        }

        co_return;
    }

    /*
     * This is the native side of the CodePush.getConfiguration method. It isn't
     * currently exposed via the "react-native-code-push" module, and is used
     * internally only by the CodePush.checkForUpdate method in order to get the
     * app version, as well as the deployment key that was configured in App.cpp.
     */
    fire_and_forget CodePushNativeModule::GetConfiguration(ReactPromise<IJsonValue> promise) noexcept 
    {
        auto configuration{ CodePushConfig::Current().GetConfiguration() };
        if (isRunningBinaryVersion)
        {
            auto errorMessage{ L"Error: Package hashing is currently unimplemented. Binary hash was not obtained." };
            auto error{ hresult_error(E_NOTIMPL, errorMessage) };
            CodePushUtils::Log(error);
            CodePushUtils::Log(L"Error obtaining hash for binary contents.");
            promise.Resolve(configuration);
            co_return;
        }
        promise.Resolve(configuration);
    }

    /*
     * This method is the native side of the CodePush.getUpdateMetadata method.
     */
    fire_and_forget CodePushNativeModule::GetUpdateMetadataAsync(CodePushUpdateState updateState, ReactPromise<IJsonValue> promise) noexcept 
    {
        auto package{ co_await CodePushPackage::GetCurrentPackageAsync() };
        if (package == nullptr)
        {
            // The app hasn't downloaded any CodePush updates yet,
            // so we simply return nil regardless if the user
            // wanted to retrieve the pending or running update.
            promise.Resolve(JsonValue::CreateNullValue());
            co_return;
        }

        // We have a CodePush update, so let's see if it's currently in a pending state.
        bool currentUpdateIsPending{ IsPendingUpdate(package.GetNamedString(PackageHashKey)) };

        if (updateState == CodePushUpdateState::Pending && !currentUpdateIsPending) {
            // The caller wanted a pending update
            // but there isn't currently one.
            promise.Resolve(JsonValue::CreateNullValue());
        }
        else if (updateState == CodePushUpdateState::Running && currentUpdateIsPending) {
            // The caller wants the running update, but the current
            // one is pending, so we need to grab the previous.
            promise.Resolve(co_await CodePushPackage::GetPreviousPackageAsync());
        }
        else {
            // The current package satisfies the request:
            // 1) Caller wanted a pending, and there is a pending update
            // 2) Caller wanted the running update, and there isn't a pending
            // 3) Caller wants the latest update, regardless if it's pending or not
            if (isRunningBinaryVersion) {
                // This only matters in Debug builds. Since we do not clear "outdated" updates,
                // we need to indicate to the JS side that somehow we have a current update on
                // disk that is not actually running.
                package.Insert(L"_isDebugOnly", JsonValue::CreateBooleanValue(true));
            }

            // Enable differentiating pending vs. non-pending updates
            package.Insert(PackageIsPendingKey, JsonValue::CreateBooleanValue(currentUpdateIsPending));
            promise.Resolve(package);
        }
        co_return; 
    }

    /*
     * This method is the native side of the LocalPackage.install method.
     */
    fire_and_forget CodePushNativeModule::InstallUpdateAsync(JsonObject updatePackage, CodePushInstallMode installMode, int minimumBackgroundDuration, ReactPromise<void> promise) noexcept 
    { 
        try
        {
            co_await CodePushPackage::InstallPackageAsync(updatePackage, IsPendingUpdate(L""));
        }
        catch (const hresult_error& ex)
        {
            promise.Reject(ex.message().c_str());
            co_return;
        }
        SavePendingUpdate(updatePackage.GetNamedString(PackageHashKey), false);
        s_installMode = installMode;
        if (s_installMode == CodePushInstallMode::OnNextResume || s_installMode == CodePushInstallMode::OnNextSuspend) {
            // Essentially, for RNW, InstallMode is currently always Immediate
            auto errorMessage{ L"Error: ON_NEXT_RESUME and ON_NEXT_SUSPEND install modes are not currently supported." };
            hresult_error error{ E_NOTIMPL, errorMessage };
            CodePushUtils::Log(error);
            throw error;
        }

        // Signal to JS that the update has been applied.
        promise.Resolve();
        co_return; 
    }

    /*
     * This method isn't publicly exposed via the "react-native-code-push"
     * module, and is only used internally to populate the RemotePackage.failedInstall property.
     */
    void CodePushNativeModule::IsFailedUpdate(std::wstring packageHash, ReactPromise<bool> promise) noexcept
    {
        auto isFailedHash{ IsFailedHash(packageHash) };
        promise.Resolve(isFailedHash);
    }

    /*
     * This method is used to save information about the latest rollback.
     * This information will be used to decide whether the application
     * should ignore the update or not.
     */
    void CodePushNativeModule::SetLatestRollbackInfo(std::wstring packageHash) noexcept
    {
        if (packageHash.empty())
        {
            return;
        }

        auto localSettings{ GetLocalSettings() };
        JsonObject latestRollbackInfo;
        auto res{ localSettings.Values().TryLookup(LatestRollbackInfoKey) };
        if (res != nullptr)
        {
            auto infoString{ unbox_value<hstring>(res) };
            JsonObject::TryParse(infoString, latestRollbackInfo);
        }

        auto initialRollbackCount{ GetRollbackCountForPackage(packageHash, latestRollbackInfo) };
        auto count{ initialRollbackCount + 1 };
        auto currentTimeMillis{ clock::to_time_t(clock::now()) * 1000 };

        latestRollbackInfo.Insert(LatestRollbackCountKey, JsonValue::CreateNumberValue(count));
        latestRollbackInfo.Insert(LatestRollbackTimeKey, JsonValue::CreateNumberValue(static_cast<double>(currentTimeMillis)));
        latestRollbackInfo.Insert(LatestRollbackPackageHashKey, JsonValue::CreateStringValue(packageHash));

        localSettings.Values().Insert(LatestRollbackInfoKey, box_value(latestRollbackInfo.Stringify()));
    }

    /*
     * This method is used when the app is started to either
     * initialize a pending update or rollback a faulty update
     * to the previous version.
     */
    IAsyncAction CodePushNativeModule::InitializeUpdateAfterRestart()
    {
        if (s_host.InstanceSettings().UseWebDebugger())
        {
            co_await ClearDebugUpdates();
        }

        auto localSettings{ GetLocalSettings() };
        auto pendingUpdateData{ localSettings.Values().TryLookup(PendingUpdateKey) };
        if (pendingUpdateData != nullptr)
        {
            auto pendingUpdateString{ unbox_value<hstring>(pendingUpdateData) };
            JsonObject pendingUpdate;
            auto success{ JsonObject::TryParse(pendingUpdateString, pendingUpdate) };
            if (success)
            {
                m_isFirstRunAfterUpdate = true;
                auto updateIsLoading{ pendingUpdate.GetNamedBoolean(PendingUpdateIsLoadingKey, false) };
                if (updateIsLoading)
                {
                    // Pending update was initialized, but notifyApplicationReady was not called.
                    // Therefore, deduce that it is a broken update and rollback.
                    CodePushUtils::Log(L"Update did not finish loading the last time, rolling back to a previous version.");
                    needToReportRollback = true;
                    co_await RollbackPackage();
                }
                else
                {
                    // Mark that we tried to initialize the new update, so that if it crashes,
                    // we will know that we need to rollback when the app next starts.
                    SavePendingUpdate(pendingUpdate.GetNamedString(PendingUpdateHashKey, L""), true);
                }
            }
        }
    }

    /*
     * This method is used to get information about the latest rollback.
     * This information will be used to decide whether the application
     * should ignore the update or not.
     */
    void CodePushNativeModule::GetLatestRollbackInfo(ReactPromise<IJsonValue> promise) noexcept 
    {
        auto localSettings{ GetLocalSettings() };
        auto res{ localSettings.Values().TryLookup(LatestRollbackInfoKey) };
        auto infoString{ unbox_value<hstring>(res) };
        JsonObject latestRollbackInfo;
        auto success{ JsonObject::TryParse(infoString, latestRollbackInfo) };
        if (success)
        {
            promise.Resolve(latestRollbackInfo);
        }
        else
        {
            promise.Resolve(JsonValue::CreateNullValue());
        }
    }

    /*
     * This method isn't publicly exposed via the "react-native-code-push"
     * module, and is only used internally to populate the LocalPackage.isFirstRun property.
     */
    fire_and_forget CodePushNativeModule::IsFirstRun(std::wstring packageHash, ReactPromise<bool> promise) noexcept
    {
        auto isFirstRun = m_isFirstRunAfterUpdate
            && !packageHash.empty()
            && packageHash == co_await CodePushPackage::GetCurrentPackageHashAsync();

        promise.Resolve(isFirstRun);
    }

    /*
     * This method is the native side of the CodePush.notifyApplicationReady() method.
     */
    void CodePushNativeModule::NotifyApplicationReady(ReactPromise<IJsonValue> promise) noexcept
    {
        RemovePendingUpdate();
        promise.Resolve(JsonValue::CreateNullValue());
    }

    void CodePushNativeModule::Allow(ReactPromise<JSValue> promise) noexcept 
    {
        CodePushUtils::Log(L"Re-allowing restarts.");
        m_allowed = true;

        if (m_restartQueue.size() > 0)
        {
            CodePushUtils::Log(L"Executing pending restart.");
            auto buf{ m_restartQueue[0] };
            m_restartQueue.erase(m_restartQueue.begin());
            RestartAppInternal(buf);
        }

        promise.Resolve(JSValue::Null);
    }

    void CodePushNativeModule::ClearPendingRestart() noexcept 
    {
        m_restartQueue.clear();
    }

    void CodePushNativeModule::Disallow(ReactPromise<JSValue> promise) noexcept
    {
        CodePushUtils::Log(L"Disallowing restarts.");
        m_allowed = false;
        promise.Resolve(JSValue::Null);
    }

    /*
     * This method is the native side of the CodePush.restartApp() method.
     */
    fire_and_forget CodePushNativeModule::RestartApp(bool onlyIfUpdateIsPending, ReactPromise<JSValue> promise) noexcept 
    {
        co_await RestartAppInternal(onlyIfUpdateIsPending);
        promise.Resolve(JSValue::Null);
    }

    /*
     * This method clears CodePush's downloaded updates.
     * It is needed to switch to a different deployment if the current deployment is more recent.
     * Note: we don’t recommend to use this method in scenarios other than that (CodePush will call this method
     * automatically when needed in other cases) as it could lead to unpredictable behavior.
     */
    fire_and_forget CodePushNativeModule::ClearUpdates() noexcept 
    {
        co_await ClearUpdatesStaticAsync();
    }

    /*
     * This method is the native side of the CodePush.downloadAndReplaceCurrentBundle()
     * method, which replaces the current bundle with the one downloaded from
     * removeBundleUrl. It is only to be used during tests and no-ops if the test
     * configuration flag is not set.
     */
    fire_and_forget CodePushNativeModule::DownloadAndReplaceCurrentBundle(std::wstring remoteBundleUrl) noexcept 
    {
        auto errorMessage{ L"Error: DownloadAndReplaceCurrentBundle is not currently implmented" };
        hresult_error error{ E_NOTIMPL, errorMessage };
        CodePushUtils::Log(error);
        throw error;
    }

    /*
     * This method is checks if a new status update exists (new version was installed,
     * or an update failed) and return its details (version label, status).
     */
    fire_and_forget CodePushNativeModule::GetNewStatusReportAsync(ReactPromise<IJsonValue> promise) noexcept 
    {
        if (needToReportRollback)
        {
            needToReportRollback = false;
            auto localSettings{ GetLocalSettings() };
            auto failedUpdatesData{ localSettings.Values().TryLookup(FailedUpdatesKey) };
            if (failedUpdatesData != nullptr)
            {
                auto failedUpdatesString{ unbox_value<hstring>(failedUpdatesData) };
                JsonArray failedUpdates;
                auto success{ JsonArray::TryParse(failedUpdatesString, failedUpdates) };
                if (success)
                {
                    auto lastFailedPackage{ failedUpdates.GetObjectAt(failedUpdates.Size() - 1) };
                    if (lastFailedPackage != nullptr)
                    {
                        promise.Resolve(CodePushTelemetryManager::GetRollbackReport(lastFailedPackage));
                        co_return;
                    }
                }
            }
        }
        else if (m_isFirstRunAfterUpdate)
        {
            auto currentPackage = co_await CodePushPackage::GetCurrentPackageAsync();
            if (currentPackage != nullptr)
            {
                promise.Resolve(CodePushTelemetryManager::GetUpdateReport(currentPackage));
                co_return;
            }
        }
        else if (isRunningBinaryVersion)
        {
            auto appVersionString{ CodePushConfig::Current().GetAppVersion() };
            promise.Resolve(CodePushTelemetryManager::GetBinaryUpdateReport(appVersionString));
            co_return;
        }
        else
        {
            auto retryStatusReport{ CodePushTelemetryManager::GetRetryStatusReport() };
            if (retryStatusReport != nullptr)
            {
                promise.Resolve(retryStatusReport);
                co_return;
            }
        }

        promise.Resolve(JsonValue::CreateNullValue());
        co_return;
    }

    void CodePushNativeModule::RecordStatusReported(JsonObject statusReport) noexcept 
    {
        CodePushTelemetryManager::RecordStatusReported(statusReport);
    }

    void CodePushNativeModule::SaveStatusReportForRetry(JsonObject statusReport) noexcept 
    {
        CodePushTelemetryManager::SaveStatusReportForRetry(statusReport);
    }

} // namespace CodePush

// Helper functions for reading and sending JsonValues to and from JavaScript
namespace winrt::Microsoft::ReactNative
{
    using namespace winrt::Windows::Data::Json;

    void WriteValue(IJSValueWriter const& writer, IJsonValue const& value) noexcept
    {
        if (value == nullptr)
        {
            writer.WriteNull();
        }
        else
        {
            switch (value.ValueType())
            {
            case JsonValueType::Object:
                writer.WriteObjectBegin();
                for (const auto& pair : value.GetObject())
                {
                    writer.WritePropertyName(pair.Key());
                    WriteValue(writer, pair.Value());
                }
                writer.WriteObjectEnd();
                break;
            case JsonValueType::Array:
                writer.WriteArrayBegin();
                for (const auto& elem : value.GetArray())
                {
                    WriteValue(writer, elem);
                }
                writer.WriteArrayEnd();
                break;
            case JsonValueType::Boolean:
                writer.WriteBoolean(value.GetBoolean());
                break;
            case JsonValueType::Number:
                writer.WriteDouble(value.GetNumber());
                break;
            case JsonValueType::String:
                writer.WriteString(value.GetString());
                break;
            case JsonValueType::Null:
                writer.WriteNull();
                break;
            }
        }
    }

    void ReadValue(IJSValueReader const& reader, /*out*/ JsonObject& value) noexcept
    {
        if (reader.ValueType() == JSValueType::Object)
        {
            hstring propertyName;
            while (reader.GetNextObjectProperty(propertyName))
            {
                value.Insert(propertyName, ReadValue<IJsonValue>(reader));
            }
        }
    }

    void ReadValue(IJSValueReader const& reader, /*out*/ IJsonValue& value) noexcept
    {
        if (reader.ValueType() == JSValueType::Object)
        {
            JsonObject valueObject;
            hstring propertyName;
            while (reader.GetNextObjectProperty(propertyName))
            {
                valueObject.Insert(propertyName, ReadValue<IJsonValue>(reader));
            }
            value = valueObject;
        }
        else if (reader.ValueType() == JSValueType::Array)
        {
            JsonArray valueArray;
            while (reader.GetNextArrayItem())
            {
                valueArray.Append(ReadValue<IJsonValue>(reader));
            }
            value = valueArray;
        }
        else
        {
            switch (reader.ValueType())
            {
            case JSValueType::Boolean:
                value = JsonValue::CreateBooleanValue(reader.GetBoolean());
                break;
            case JSValueType::Double:
                value = JsonValue::CreateNumberValue(reader.GetDouble());
                break;
            case JSValueType::Int64:
                value = JsonValue::CreateNumberValue(static_cast<double>(reader.GetInt64()));
                break;
            case JSValueType::String:
                value = JsonValue::CreateStringValue(reader.GetString());
                break;
            case JSValueType::Null:
                value = JsonValue::CreateNullValue();
                break;
            }
        }
    }
}