Merge pull request #357 from jetelain/prompt-unsaved-document

Show save prompt when trying to close a "dirty" document
tgjones · Jul 30, 2023 · 4641fc5 · 4641fc5
2 parents 3099ce3 + 78a9013
commit 4641fc5
Show file tree

Hide file tree

Showing 3 changed files with 117 additions and 72 deletions.
diff --git a/src/Gemini/Framework/PersistedDocument.cs b/src/Gemini/Framework/PersistedDocument.cs
@@ -1,6 +1,10 @@
 using System.IO;
 using System.Threading;
 using System.Threading.Tasks;
+using System.Windows;
+using Gemini.Framework.Commands;
+using Gemini.Modules.Shell.Commands;
+using Gemini.Properties;
 
 namespace Gemini.Framework
 {
@@ -37,10 +41,22 @@ public bool IsDirty
             }
         }
 
-        public override Task<bool> CanCloseAsync(CancellationToken cancellationToken)
+        public override async Task<bool> CanCloseAsync(CancellationToken cancellationToken)
         {
-            // TODO: Show save prompt.
-            return Task.FromResult(!IsDirty);
+            if (IsDirty)
+            {
+                var response = MessageBox.Show(string.Format(Resources.DocumentDirtyCloseConfirmText, FileName), Resources.DocumentDirtyCloseConfirmTitle, MessageBoxButton.YesNoCancel);
+                if (response == MessageBoxResult.Cancel)
+                {
+                    return false;
+                }
+                if (response == MessageBoxResult.Yes)
+                {
+                    await Save();
+                    return !IsNew; // User may discard file picker prompt
+                }
+            }
+            return true;
         }
 
         private void UpdateDisplayName()
@@ -93,5 +109,10 @@ public async Task Save(string filePath)
         }
 
         protected abstract Task DoSave(string filePath);
+
+        public Task Save()
+        {
+            return ((ICommandHandler<SaveFileCommandDefinition>)this).Run(null);
+        }
     }
 }