WPF - Allow processing a window after init

#467

Author: Deadpikle

src/NetSparkle.UI.WPF/UIFactory.cs

@@ -91,6 +91,19 @@ public UIFactory(ImageSource applicationIcon) : this()
         /// </summary>
         public System.Windows.Media.Brush UpdateWindowGridBackgroundBrush { get; set; }
 
+        /// <summary>
+        /// A delegate for handling windows that are created by a <see cref="UIFactory"/>
+        /// </summary>
+        /// <param name="window"><see cref="Window"/> that has been created and initialized (with view model, if applicable)</param>
+        /// <param name="factory"><see cref="UIFactory"/> that created the given <see cref="Window"/></param>
+        public delegate void WindowHandler(Window window, UIFactory factory);
+
+        /// <summary>
+        /// Set this property to manually do any other setup on a <see cref="Window"/> after it has been created by the <see cref="UIFactory"/>.
+        /// Can be used to tweak view models, change styles on the <see cref="Window"/>, etc.
+        /// </summary>
+        public WindowHandler ProcessWindowAfterInit { get; set; }
+
         /// <inheritdoc/>
         public virtual IUpdateAvailable CreateUpdateAvailableWindow(SparkleUpdater sparkle, List<AppCastItem> updates, bool isUpdateAlreadyDownloaded = false)
         {
@@ -120,6 +133,7 @@ public virtual IUpdateAvailable CreateUpdateAvailableWindow(SparkleUpdater spark
                 viewModel.ReleaseNotesGrabber = ReleaseNotesGrabberOverride;
             }
             viewModel.Initialize(sparkle, updates, isUpdateAlreadyDownloaded, ReleaseNotesHTMLTemplate, AdditionalReleaseNotesHeaderHTML, ReleaseNotesDateTimeFormat);
+            ProcessWindowAfterInit?.Invoke(window, this);
             return window;
         }
 
@@ -131,16 +145,20 @@ public virtual IDownloadProgress CreateProgressWindow(SparkleUpdater sparkle, Ap
                 ItemToDownload = item,
                 SoftwareWillRelaunchAfterUpdateInstalled = sparkle.RelaunchAfterUpdate
             };
-            return new DownloadProgressWindow(viewModel)
+            var window = new DownloadProgressWindow(viewModel)
             {
                 Icon = _applicationIcon
             };
+            ProcessWindowAfterInit?.Invoke(window, this);
+            return window;
         }
 
         /// <inheritdoc/>
         public virtual ICheckingForUpdates ShowCheckingForUpdates(SparkleUpdater sparkle)
         {
-            return new CheckingForUpdatesWindow { Icon = _applicationIcon };
+            var window = new CheckingForUpdatesWindow() { Icon = _applicationIcon };
+            ProcessWindowAfterInit?.Invoke(window, this);
+            return window;
         }
 
         /// <inheritdoc/>
@@ -192,6 +210,7 @@ public virtual void ShowToast(SparkleUpdater sparkle, List<AppCastItem> updates,
                 };
                 try
                 {
+                    ProcessWindowAfterInit?.Invoke(toast, this);
                     toast.Show(Resources.DefaultUIFactory_ToastMessage, Resources.DefaultUIFactory_ToastCallToAction, 5);
                     System.Windows.Threading.Dispatcher.Run();
                 }
@@ -218,6 +237,7 @@ private void ShowMessage(string title, string message)
                 Icon = _applicationIcon
             };
             messageWindow.WindowStartupLocation = WindowStartupLocation.CenterScreen;
+            ProcessWindowAfterInit?.Invoke(messageWindow, this);
             messageWindow.ShowDialog();
         }