[Feature] Prevent Windows Log off/Restart/Shutdown when running critical task (#758)

# Main Goal
Preventing Windows from ending it's ~~life~~ session when Collapse is
running critical task. Should prevent accidental shutting down when
Collapse is installing/updating/repairing games.


![image](https://github.com/user-attachments/assets/97f8d356-e6a3-43ed-a0d3-9c1f2ed9675e)

Thanks @neon-nyan for the screenshot.

P.S. This doesn't work when you run Collapse inside a debugger, because
the debugger will kill its target process when Windows is ending the
session.

## PR Status :
- Overall Status : Done
- Commits : Done
- Synced to base (Collapse:main) : Yes
- Build status : OK
- Crashing : No
- Bug found caused by PR : 0

### Templates

<details>
  <summary>Changelog Prefixes</summary>
  
  ```
    **[New]**
    **[Imp]**
    **[Fix]**
    **[Loc]**
    **[Doc]**
  ```

</details>

Author: neon-nyan

CollapseLauncher/Classes/Helper/WindowUtility.cs

@@ -451,12 +451,14 @@ private static IntPtr InstallWndProcCallback(IntPtr hwnd, WndProcDelegate wndPro
 
         private static IntPtr MainWndProc(IntPtr hwnd, uint msg, UIntPtr wParam, IntPtr lParam)
         {
-            const uint WM_SYSCOMMAND        = 0x0112;
-            const uint WM_SHOWWINDOW        = 0x0018;
-            const uint WM_NCHITTEST         = 0x0084;
-            const uint WM_NCCALCSIZE        = 0x0083;
-            const uint WM_SETTINGCHANGE     = 0x001A;
-            const uint WM_ACTIVATE          = 0x0006;
+            const uint WM_SYSCOMMAND      = 0x0112;
+            const uint WM_SHOWWINDOW      = 0x0018;
+            const uint WM_NCHITTEST       = 0x0084;
+            const uint WM_NCCALCSIZE      = 0x0083;
+            const uint WM_SETTINGCHANGE   = 0x001A;
+            const uint WM_ACTIVATE        = 0x0006;
+            const uint WM_QUERYENDSESSION = 0x0011;
+            const uint WM_ENDSESSION      = 0x0016;
 
             switch (msg)
             {
@@ -592,6 +594,29 @@ private static IntPtr MainWndProc(IntPtr hwnd, uint msg, UIntPtr wParam, IntPtr
 
                         break;
                     }
+                case WM_QUERYENDSESSION:
+                    // Return FALSE (0) to prevent shutdown if critical operation is in progress
+                    if (MainWindow.IsCriticalOpInProgress)
+                    {
+                        return 0;
+                    }
+                    // Let Windows continue shutdown if no critical operation
+                    break;
+            
+                case WM_ENDSESSION:
+                    // wParam is TRUE if session is ending
+                    if (wParam != UIntPtr.Zero)
+                    {
+                        if (MainWindow.IsCriticalOpInProgress)
+                        {
+                            // Still try to block it at this stage
+                            return 0;
+                        }
+                
+                        // No critical operation, calling app close method
+                        (CurrentWindow as MainWindow)?.CloseApp();
+                    }
+                    break;
             }
 
             return PInvoke.CallWindowProc(_oldMainWndProcPtr, hwnd, msg, wParam, lParam);

CollapseLauncher/XAMLs/MainApp/MainWindow.xaml.cs

@@ -11,6 +11,7 @@
 using Hi3Helper;
 using Hi3Helper.SentryHelper;
 using Hi3Helper.Shared.Region;
+using Hi3Helper.Win32.ManagedTools;
 using Hi3Helper.Win32.Native.LibraryImport;
 using Microsoft.UI.Composition;
 using Microsoft.UI.Input;
@@ -19,6 +20,7 @@
 using Microsoft.UI.Xaml.Controls;
 using Microsoft.UI.Xaml.Media.Animation;
 using System;
+using System.Threading;
 using System.Threading.Tasks;
 using Windows.UI;
 using static CollapseLauncher.Dialogs.SimpleDialogs;
@@ -34,8 +36,31 @@ namespace CollapseLauncher
     public sealed partial class MainWindow : Window
     {
         private static bool _isForceDisableIntro;
-        
-        public static bool IsCriticalOpInProgress { get; set; }
+
+        private static readonly Lock CriticalOpLock = new Lock();
+        public static bool IsCriticalOpInProgress
+        {
+            get;
+            set
+            {
+                lock (CriticalOpLock)
+                {
+                    var lastValue = field;
+                    field = value;
+                    
+                    if (value)
+                    {
+                        ShutdownBlocker.StartBlocking(WindowUtility.CurrentWindowPtr, Locale.Lang._Dialogs.EnsureExitSubtitle,
+                                                      ILoggerHelper.GetILogger("ShutdownBlocker"));
+                    }
+                    else if (lastValue)
+                    {
+                        ShutdownBlocker.StopBlocking(WindowUtility.CurrentWindowPtr,
+                                                     ILoggerHelper.GetILogger("ShutdownBlocker"));
+                    }
+                }
+            }
+        }
 
         public void InitializeWindowProperties(bool startOobe = false)
         {

CollapseLauncher/XAMLs/MainApp/Pages/Dialogs/SimpleDialogs.cs

@@ -994,7 +994,7 @@ public static Task<ContentDialogResult> Dialog_UninstallGame(string gameLocation
         public static Task<ContentDialogResult> Dialog_EnsureExit()
         {
             return SpawnDialog(Lang._Dialogs.EnsureExitTitle,
-                               Lang._Dialogs.EnsureExitSubtitle,
+                               $"{Lang._Dialogs.EnsureExitSubtitle} {Lang._Dialogs.EnsureExitSubtitle2}",
                                null,
                                Lang._Misc.NoCancel,
                                Lang._Misc.Yes,

CollapseLauncher/packages.lock.json

@@ -105,12 +105,6 @@
         "resolved": "0.41.2",
         "contentHash": "oso8cno80OzCgAQOygtJXT1UhlVr2eSjwzXLqSLHyfKmSJ+3OcwDbrpzNwMj3/1SgncvQxP7OsNQ3bRfE3HvjQ=="
       },
-      "Microsoft.DotNet.ILCompiler": {
-        "type": "Direct",
-        "requested": "[9.0.5, )",
-        "resolved": "9.0.5",
-        "contentHash": "rfDodV68TLSUgVM6ZQusuUXD7GAIqwCKJWlPtfsd99iLlLbeGfqUQE7Xo8xLa1ju5GCdSBO0q35suzZ8ppclpA=="
-      },
       "Microsoft.Extensions.DependencyInjection.Abstractions": {
         "type": "Direct",
         "requested": "[9.0.5, )",
@@ -630,15 +624,6 @@
         "resolved": "0.5.0",
         "contentHash": "5Tw6O9sBDAN1aV+kpOSVvqvFk6Ahk6bYz0TTx3808Dp40M45gKTtzTzI9SS1VeAkljE6BAOeKaykpPnG36oOgw=="
       },
-      "Microsoft.DotNet.ILCompiler": {
-        "type": "Direct",
-        "requested": "[9.0.5, )",
-        "resolved": "9.0.5",
-        "contentHash": "rfDodV68TLSUgVM6ZQusuUXD7GAIqwCKJWlPtfsd99iLlLbeGfqUQE7Xo8xLa1ju5GCdSBO0q35suzZ8ppclpA==",
-        "dependencies": {
-          "runtime.win-x64.Microsoft.DotNet.ILCompiler": "9.0.5"
-        }
-      },
       "Microsoft.Graphics.Win2D": {
         "type": "Direct",
         "requested": "[1.3.2, )",
@@ -688,11 +673,6 @@
           "Microsoft.WindowsAppSDK.Base": "1.8.250501001-experimental",
           "Microsoft.WindowsAppSDK.InteractiveExperiences": "1.8.250506001-experimental"
         }
-      },
-      "runtime.win-x64.Microsoft.DotNet.ILCompiler": {
-        "type": "Transitive",
-        "resolved": "9.0.5",
-        "contentHash": "/x+iKUhSOhYrbaEr9x78wZmBk7Le8FlY9Xwx7uENRFCZ7K5sGrTffkJiig22EzYuALh8SeixjSzVFS0nm5+NgQ=="
       }
     }
   }

Hi3Helper.Core/Lang/Locale/LangDialogs.cs

@@ -212,6 +212,7 @@ public sealed partial class LangDialogs
                 public string UACWarningDontShowAgain { get; set; } = LangFallback?._Dialogs.UACWarningDontShowAgain;
                 public string EnsureExitTitle { get; set; } = LangFallback?._Dialogs.EnsureExitTitle;
                 public string EnsureExitSubtitle { get; set; } = LangFallback?._Dialogs.EnsureExitSubtitle;
+                public string EnsureExitSubtitle2 { get; set; } = LangFallback?._Dialogs.EnsureExitSubtitle2;
 
                 public string UserFeedback_DialogTitle { get; set; } = LangFallback?._Dialogs.UserFeedback_DialogTitle;
                 public string UserFeedback_TextFieldTitleHeader { get; set; } = LangFallback?._Dialogs.UserFeedback_TextFieldTitleHeader;

Hi3Helper.Core/Lang/en_US.json

@@ -1080,7 +1080,8 @@
     "UACWarningDontShowAgain": "Don't Show Again",
 
     "EnsureExitTitle": "Exiting Application",
-    "EnsureExitSubtitle": "There are critical operations running in the background. Are you sure you want to exit?",
+    "EnsureExitSubtitle": "There are critical operations running in the background.",
+    "EnsureExitSubtitle2": "Are you sure you want to exit?",
 
     "UserFeedback_DialogTitle": "Share Your Thoughts",
     "UserFeedback_TextFieldTitleHeader": "Feedback Title",