From f2f0ec7241ae25dc3a0c774f66c93ced6fb67abb Mon Sep 17 00:00:00 2001
From: bluepilledgreat <97983689+bluepilledgreat@users.noreply.github.com>
Date: Thu, 23 Jan 2025 23:03:22 +0000
Subject: [PATCH] add close confirmation message

---
 .../Editor/BootstrapperEditorWindow.xaml      |  4 +-
 .../Editor/BootstrapperEditorWindow.xaml.cs   | 40 ++++++++++++++-----
 .../BootstrapperEditorWindowViewModel.cs      |  3 ++
 3 files changed, 35 insertions(+), 12 deletions(-)

diff --git a/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml b/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml
index 93952ac7a..be25dc018 100644
--- a/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml
+++ b/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml
@@ -15,6 +15,7 @@
     d:DataContext="{d:DesignInstance dmodels:BootstrapperEditorWindowViewModel,
                                      IsDesignTimeCreatable=True}"
     Background="{ui:ThemeResource ApplicationBackgroundBrush}"
+    Closing="OnClosing"
     ExtendsContentIntoTitleBar="True"
     mc:Ignorable="d">
     <Grid>
@@ -40,8 +41,7 @@
             Margin="10,10,10,0"
             ShowLineNumbers="True"
             Style="{StaticResource NewTextEditor}"
-            SyntaxHighlighting="XML"
-            TextChanged="OnCodeChanged" />
+            SyntaxHighlighting="XML" />
 
         <ui:Button
             Grid.Row="2"
diff --git a/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml.cs b/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml.cs
index 38aab7ac4..bef12a082 100644
--- a/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml.cs
+++ b/Bloxstrap/UI/Elements/Editor/BootstrapperEditorWindow.xaml.cs
@@ -9,6 +9,7 @@
 
 using Bloxstrap.UI.Elements.Base;
 using Bloxstrap.UI.ViewModels.Editor;
+using System.Windows;
 
 namespace Bloxstrap.UI.Elements.Editor
 {
@@ -130,7 +131,8 @@ private static void PopulateElementInfo()
             }
         }
 
-        CompletionWindow? _completionWindow = null;
+        private BootstrapperEditorWindowViewModel _viewModel;
+        private CompletionWindow? _completionWindow = null;
 
         public BootstrapperEditorWindow(string name)
         {
@@ -141,17 +143,18 @@ public BootstrapperEditorWindow(string name)
             string themeContents = File.ReadAllText(Path.Combine(directory, "Theme.xml"));
             themeContents = ToCRLF(themeContents); // make sure the theme is in CRLF. a function expects CRLF.
 
-            var viewModel = new BootstrapperEditorWindowViewModel();
-            viewModel.ThemeSavedCallback = ThemeSavedCallback;
-            viewModel.Directory = directory;
-            viewModel.Name = name;
-            viewModel.Title = $"Editing \"{name}\"";
-            viewModel.Code = themeContents;
+            _viewModel = new BootstrapperEditorWindowViewModel();
+            _viewModel.ThemeSavedCallback = ThemeSavedCallback;
+            _viewModel.Directory = directory;
+            _viewModel.Name = name;
+            _viewModel.Title = $"Editing \"{name}\"";
+            _viewModel.Code = themeContents;
 
-            DataContext = viewModel;
+            DataContext = _viewModel;
             InitializeComponent();
 
-            UIXML.Text = viewModel.Code;
+            UIXML.Text = _viewModel.Code;
+            UIXML.TextChanged += OnCodeChanged;
             UIXML.TextArea.TextEntered += OnTextAreaTextEntered;
 
             LoadHighlightingTheme();
@@ -180,13 +183,30 @@ private static string ToCRLF(string text)
             return text.Replace("\r\n", "\n").Replace("\r", "\n").Replace("\n", "\r\n");
         }
 
-        private void OnCodeChanged(object sender, EventArgs e)
+        private void OnCodeChanged(object? sender, EventArgs e)
         {
             BootstrapperEditorWindowViewModel viewModel = (BootstrapperEditorWindowViewModel)DataContext;
             viewModel.Code = UIXML.Text;
+            viewModel.CodeChanged = true;
             viewModel.OnPropertyChanged(nameof(viewModel.Code));
         }
 
+        private void OnClosing(object sender, System.ComponentModel.CancelEventArgs e)
+        {
+            if (!_viewModel.CodeChanged)
+                return;
+
+            var result = Frontend.ShowMessageBox($"Save changes to {_viewModel.Name}?", MessageBoxImage.Information, MessageBoxButton.YesNoCancel);
+            if (result == MessageBoxResult.Cancel)
+            {
+                e.Cancel = true;
+            }
+            else if (result == MessageBoxResult.Yes)
+            {
+                _viewModel.SaveCommand.Execute(null);
+            }
+        }
+
         private void OnTextAreaTextEntered(object sender, TextCompositionEventArgs e)
         {
             switch (e.Text)
diff --git a/Bloxstrap/UI/ViewModels/Editor/BootstrapperEditorWindowViewModel.cs b/Bloxstrap/UI/ViewModels/Editor/BootstrapperEditorWindowViewModel.cs
index 7fc602435..2e0bcb4af 100644
--- a/Bloxstrap/UI/ViewModels/Editor/BootstrapperEditorWindowViewModel.cs
+++ b/Bloxstrap/UI/ViewModels/Editor/BootstrapperEditorWindowViewModel.cs
@@ -26,6 +26,8 @@ public class BootstrapperEditorWindowViewModel : NotifyPropertyChangedViewModel
         public string Title { get; set; } = "Editing \"Custom Theme\"";
         public string Code { get; set; } = "";
 
+        public bool CodeChanged { get; set; } = false;
+
         private void Preview()
         {
             const string LOG_IDENT = "BootstrapperEditorWindowViewModel::Preview";
@@ -61,6 +63,7 @@ private void Save()
             try
             {
                 File.WriteAllText(path, Code);
+                CodeChanged = false;
                 ThemeSavedCallback.Invoke(true, "Your theme has been saved!");
             }
             catch (Exception ex)