Fix Install button being disabled after cancelling install

CriticalFlaw · May 5, 2023 · 2eab19c · 2eab19c
1 parent 964225f
commit 2eab19c
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/src/TF2HUD.Editor/ViewModels/MainWindowViewModel.cs b/src/TF2HUD.Editor/ViewModels/MainWindowViewModel.cs
@@ -195,7 +195,11 @@ public async void BtnInstall_Click()
                     MainWindow.SetupDirectory(true);
 
                 // Stop the process if Team Fortress 2 is still running.
-                if (Utilities.CheckIsGameRunning()) return;
+                if (Utilities.CheckIsGameRunning())
+                {
+                    Installing = false;
+                    return;
+                }
 
                 // Clear tf/custom directory of other installed HUDs.
                 MainWindow.Logger.Info("Preparing directories for extraction.");
@@ -208,7 +212,11 @@ public async void BtnInstall_Click()
                 foreach (var foundHud in Directory.GetDirectories(MainWindow.HudPath))
                 {
                     if (!foundHud.Remove(0, MainWindow.HudPath.Length).ToLowerInvariant().Contains("hud") || !File.Exists($"{foundHud}\\info.vdf")) continue;
-                    if (MainWindow.ShowMessageBox(MessageBoxImage.Warning, Resources.info_unsupported_hud_found, MessageBoxButton.YesNoCancel) != MessageBoxResult.Yes) return;
+                    if (MainWindow.ShowMessageBox(MessageBoxImage.Warning, Resources.info_unsupported_hud_found, MessageBoxButton.YesNoCancel) != MessageBoxResult.Yes)
+                    {
+                        Installing = false;
+                        return;
+                    }
                     Directory.Delete(foundHud, true);
                 }
 
@@ -261,7 +269,11 @@ public async void BtnInstall_Click()
                 }
 
                 // Update the page view.
-                if (string.IsNullOrWhiteSpace(SelectedHud.Name)) return;
+                if (string.IsNullOrWhiteSpace(SelectedHud.Name))
+                {
+                    Installing = false;
+                    return;
+                }
                 SelectedHud.Settings.SaveSettings();
                 SelectedHud.ApplyCustomizations();