Warn about unsaved changes on open and exit.

int19h · Sep 15, 2018 · f8c04a8 · f8c04a8
1 parent 9e6c516
commit f8c04a8
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 1 deletion.
diff --git a/WarBender.UI/MainForm.Designer.cs b/WarBender.UI/MainForm.Designer.cs
diff --git a/WarBender.UI/MainForm.cs b/WarBender.UI/MainForm.cs
@@ -1,6 +1,8 @@
 using System;
 using System.Collections;
 using System.Collections.Generic;
+using System.ComponentModel;
+using System.ComponentModel.Design;
 using System.Data;
 using System.Diagnostics;
 using System.IO;
@@ -17,7 +19,6 @@ namespace WarBender.UI {
     public partial class MainForm : FormBase {
         public static MainForm Instance;
 
-        public Module module;
         private string _initialText;
         private readonly ModelGetters _modelGetters = new ModelGetters();
 
@@ -57,10 +58,27 @@ public MainForm() {
 
         internal GameDesignerHost DesignerHost { get; private set; }
 
+        public string FileName { get; private set; }
+
+        public bool IsDirty { get; private set; }
+
         public Game Game {
             get => DesignerHost?.Game;
             set {
+                if (DesignerHost != null) {
+                    DesignerHost.ComponentAdded -= DesignerHost_ComponentAdded;
+                    DesignerHost.ComponentChanged -= DesignerHost_ComponentChanged;
+                    DesignerHost.ComponentRemoved -= DesignerHost_ComponentRemoved;
+                }
+
                 DesignerHost = value == null ? null : new GameDesignerHost(value);
+
+                if (DesignerHost != null) {
+                    DesignerHost.ComponentAdded += DesignerHost_ComponentAdded;
+                    DesignerHost.ComponentChanged += DesignerHost_ComponentChanged;
+                    DesignerHost.ComponentRemoved += DesignerHost_ComponentRemoved;
+                }
+
                 _modelGetters.Game = value;
             }
         }
@@ -250,6 +268,10 @@ private async Task WithModuleFor(Stopwatch sw, string fileName, Func<Module, Tas
         }
 
         public async Task OpenAsync(string fileName) {
+            if (!ConfirmDiscardingChanges()) {
+                return;
+            }
+
             UpdateFileDialogs(fileName);
             UseWaitCursor = true;
             openToolStripMenuItem.Enabled = false;
@@ -266,6 +288,7 @@ public async Task OpenAsync(string fileName) {
                     mdiChild.Close();
                 }
                 Game = null;
+                IsDirty = false;
                 Text = _initialText;
 
                 Game game = null;
@@ -295,6 +318,7 @@ await Task.Run(() => {
                 treeListView.Expand(Game.Data);
                 ShowPropertyGrid(treeListView.Objects);
 
+                FileName = fileName;
                 Text = Path.GetFileName(fileName) + " - " + _initialText;
 
                 if (roundtripError != null) {
@@ -394,6 +418,8 @@ await Task.Run(() => {
                     }
                 }
 
+                IsDirty = false;
+                FileName = fileName;
                 Text = Path.GetFileName(fileName) + " - " + _initialText;
 
                 sw.Stop();
@@ -453,6 +479,23 @@ protected override bool ProcessCmdKey(ref Message msg, Keys keyData) {
             return base.ProcessCmdKey(ref msg, keyData);
         }
 
+        private void ProcessComponentChange(object component) {
+            if (component is IDataObjectChild) {
+                IsDirty = true;
+            }
+        }
+
+        private bool ConfirmDiscardingChanges() {
+            if (!IsDirty) {
+                return true;
+            }
+            var dr = MessageBox.Show(this,
+                "You have unsaved changes. If you continue with this operation, " +
+                "they will be lost. Are you sure you want to proceed?",
+                null, MessageBoxButtons.YesNo, MessageBoxIcon.Warning);
+            return dr == DialogResult.Yes;
+        }
+
         private void menuStrip_ItemAdded(object sender, ToolStripItemEventArgs e) {
             // Hide MDI child icon.
             if (e.Item.GetType().ToString() == "System.Windows.Forms.MdiControlStrip+SystemMenuItem") {
@@ -570,5 +613,23 @@ private void showConsoleToolStripMenuItem_Click(object sender, EventArgs e) {
                 mdiChild.WindowState = FormWindowState.Minimized;
             }
         }
+
+        private void DesignerHost_ComponentRemoved(object sender, ComponentEventArgs e) {
+            ProcessComponentChange(e.Component);
+        }
+
+        private void DesignerHost_ComponentChanged(object sender, ComponentChangedEventArgs e) {
+            ProcessComponentChange(e.Component);
+        }
+
+        private void DesignerHost_ComponentAdded(object sender, ComponentEventArgs e) {
+            ProcessComponentChange(e.Component);
+        }
+
+        private void MainForm_FormClosing(object sender, FormClosingEventArgs e) {
+            if (!ConfirmDiscardingChanges()) {
+                e.Cancel = true;
+            }
+        }
     }
 }