From 3c24ca08d042782166b0e1a7e1ce7297062e309e Mon Sep 17 00:00:00 2001
From: Craftplacer <craftplacer@outlook.de>
Date: Fri, 8 May 2020 02:48:27 +0200
Subject: [PATCH] Check whether the build is deployed within the public check
 updates method

---
 osu.Desktop/Updater/SquirrelUpdateManager.cs | 15 +++++++++------
 osu.Game/Updater/SimpleUpdateManager.cs      | 12 +++++++++---
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/osu.Desktop/Updater/SquirrelUpdateManager.cs b/osu.Desktop/Updater/SquirrelUpdateManager.cs
index 2834f1f71d..a3b21b4bd9 100644
--- a/osu.Desktop/Updater/SquirrelUpdateManager.cs
+++ b/osu.Desktop/Updater/SquirrelUpdateManager.cs
@@ -26,6 +26,7 @@ namespace osu.Desktop.Updater
 
         private UpdateManager updateManager;
         private NotificationOverlay notificationOverlay;
+        private OsuGameBase gameBase;
 
         public Task PrepareUpdateAsync() => UpdateManager.RestartAppWhenExited();
 
@@ -34,16 +35,18 @@ namespace osu.Desktop.Updater
         [BackgroundDependencyLoader]
         private void load(NotificationOverlay notification, OsuGameBase game)
         {
+            gameBase = game;
             notificationOverlay = notification;
 
-            if (game.IsDeployedBuild)
-            {
-                Splat.Locator.CurrentMutable.Register(() => new SquirrelLogger(), typeof(Splat.ILogger));
-                CheckForUpdate();
-            }
+            Splat.Locator.CurrentMutable.Register(() => new SquirrelLogger(), typeof(Splat.ILogger));
+            CheckForUpdate();
         }
 
-        public override void CheckForUpdate() => Schedule(() => Task.Run(() => checkForUpdateAsync()));
+        public override void CheckForUpdate()
+        {
+            if (gameBase.IsDeployedBuild)
+                Schedule(() => Task.Run(() => checkForUpdateAsync()));
+        }
 
         private async void checkForUpdateAsync(bool useDeltaPatching = true, UpdateProgressNotification notification = null)
         {
diff --git a/osu.Game/Updater/SimpleUpdateManager.cs b/osu.Game/Updater/SimpleUpdateManager.cs
index 5cc42090f4..8513ea94b4 100644
--- a/osu.Game/Updater/SimpleUpdateManager.cs
+++ b/osu.Game/Updater/SimpleUpdateManager.cs
@@ -26,16 +26,22 @@ namespace osu.Game.Updater
         [Resolved]
         private GameHost host { get; set; }
 
+        private OsuGameBase gameBase;
+
         [BackgroundDependencyLoader]
         private void load(OsuGameBase game)
         {
+            gameBase = game;
             version = game.Version;
 
-            if (game.IsDeployedBuild)
-                CheckForUpdate();
+            CheckForUpdate();
         }
 
-        public override void CheckForUpdate() => Schedule(() => Task.Run(checkForUpdateAsync));
+        public override void CheckForUpdate()
+        {
+            if (gameBase.IsDeployedBuild)
+                Schedule(() => Task.Run(checkForUpdateAsync));
+        }
 
         private async void checkForUpdateAsync()
         {