Fixed crash when windows shutting down or rebooting

PixiEditor · Oct 16, 2021 · 8f67a47 · 8f67a47
1 parent fafb0f4
commit 8f67a47
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 3 deletions.
diff --git a/PixiEditor/App.xaml.cs b/PixiEditor/App.xaml.cs
@@ -0,0 +1,25 @@
+using PixiEditor.Models.Dialogs;
+using PixiEditor.Models.Enums;
+using PixiEditor.ViewModels;
+using System.Linq;
+using System.Windows;
+
+namespace PixiEditor
+{
+    /// <summary>
+    ///     Interaction logic for App.xaml.
+    /// </summary>
+    public partial class App : Application
+    {
+        protected override void OnSessionEnding(SessionEndingCancelEventArgs e)
+        {
+            base.OnSessionEnding(e);
+
+            if (ViewModelMain.Current.BitmapManager.Documents.Any(x => !x.ChangesSaved))
+            {
+                ConfirmationType confirmation = ConfirmationDialog.Show($"{e.ReasonSessionEnding} with unsaved data. Are you sure?");
+                e.Cancel = (confirmation == ConfirmationType.Canceled || confirmation == ConfirmationType.No);
+            }
+        }
+    }
+}
diff --git a/PixiEditor/Models/Dialogs/ConfirmationDialog.cs b/PixiEditor/Models/Dialogs/ConfirmationDialog.cs
@@ -1,6 +1,9 @@
 using PixiEditor.Models.Enums;
 using PixiEditor.Views;
-
+using System;
+using System.IO;
+using System.Windows;
+
 namespace PixiEditor.Models.Dialogs
 {
     public static class ConfirmationDialog
@@ -12,12 +15,12 @@ public static ConfirmationType Show(string message)
                 Body = message,
                 Topmost = true
             };
-            if ((bool)popup.ShowDialog())
+            if (popup.ShowDialog().GetValueOrDefault())
             {
                 return popup.Result ? ConfirmationType.Yes : ConfirmationType.No;
             }
 
             return ConfirmationType.Canceled;
-        }
+        }
     }
 }