Fixed logical tree, added ContentPropertyAttribute

MahApps · Mar 4, 2014 · f011c3c · f011c3c
1 parent 6348581
commit f011c3c
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/MahApps.Metro/Controls/Dialogs/BaseMetroDialog.cs b/MahApps.Metro/Controls/Dialogs/BaseMetroDialog.cs
@@ -9,10 +9,21 @@
 
 namespace MahApps.Metro.Controls.Dialogs
 {
+    [System.Windows.Markup.ContentProperty("DialogBody")]
     public abstract class BaseMetroDialog : Control
     {
         public static readonly DependencyProperty TitleProperty = DependencyProperty.Register("Title", typeof(string), typeof(BaseMetroDialog), new PropertyMetadata(default(string)));
-        public static readonly DependencyProperty DialogBodyProperty = DependencyProperty.Register("DialogBody", typeof(object), typeof(BaseMetroDialog), new PropertyMetadata(null));
+        public static readonly DependencyProperty DialogBodyProperty = DependencyProperty.Register("DialogBody", typeof(object), typeof(BaseMetroDialog), new PropertyMetadata(null, (o, e) => {
+            BaseMetroDialog dialog = (o as BaseMetroDialog);
+            if (dialog != null) {
+                if (e.OldValue != null) {
+                    dialog.RemoveLogicalChild(e.OldValue);
+                }
+                if (e.NewValue != null) {
+                    dialog.AddLogicalChild(e.NewValue);
+                }
+            }
+        }));
         public static readonly DependencyProperty DialogTopProperty = DependencyProperty.Register("DialogTop", typeof(object), typeof(BaseMetroDialog), new PropertyMetadata(null));
         public static readonly DependencyProperty DialogBottomProperty = DependencyProperty.Register("DialogBottom", typeof(object), typeof(BaseMetroDialog), new PropertyMetadata(null));