Pass through correct update to apply when calling `WaitExitThenApplyU…

…pdates`
ppy · Sep 2, 2024 · cd9b822 · cd9b822
1 parent 68e6fa2
commit cd9b822
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/osu.Desktop/Updater/VelopackUpdateManager.cs b/osu.Desktop/Updater/VelopackUpdateManager.cs
@@ -52,7 +52,7 @@ private async Task<bool> checkForUpdateAsync(UpdateProgressNotification? notific
                 if (localUserInfo?.IsPlaying.Value == true)
                     return false;
 
-                var info = await updateManager.CheckForUpdatesAsync().ConfigureAwait(false);
+                UpdateInfo? info = await updateManager.CheckForUpdatesAsync().ConfigureAwait(false);
 
                 // Handle no updates available.
                 if (info == null)
@@ -65,7 +65,7 @@ private async Task<bool> checkForUpdateAsync(UpdateProgressNotification? notific
                     {
                         Activated = () =>
                         {
-                            restartToApplyUpdate();
+                            restartToApplyUpdate(null);
                             return true;
                         }
                     });
@@ -78,7 +78,7 @@ private async Task<bool> checkForUpdateAsync(UpdateProgressNotification? notific
                 {
                     notification = new UpdateProgressNotification
                     {
-                        CompletionClickAction = restartToApplyUpdate,
+                        CompletionClickAction = () => restartToApplyUpdate(info),
                     };
 
                     Schedule(() => notificationOverlay.Post(notification));
@@ -117,9 +117,9 @@ private async Task<bool> checkForUpdateAsync(UpdateProgressNotification? notific
             return true;
         }
 
-        private bool restartToApplyUpdate()
+        private bool restartToApplyUpdate(UpdateInfo? info)
         {
-            updateManager.WaitExitThenApplyUpdates(null);
+            updateManager.WaitExitThenApplyUpdates(info?.TargetFullRelease);
             Schedule(() => game.AttemptExit());
             return true;
         }