From 046411460b711e269602edeabb69e05a1aa1de10 Mon Sep 17 00:00:00 2001
From: countincognito <countincognito@outlook.com>
Date: Sun, 19 Jan 2025 23:50:55 +0000
Subject: [PATCH] Add ability to embed view models (with corresponding view)
 into MsBoxStandardView.

---
 Exmaple2.0/App.axaml                          |  7 ++
 Exmaple2.0/EmbeddedViewLocator.cs             | 40 +++++++++++
 Exmaple2.0/ViewModels/EmbeddedViewModel.cs    |  6 ++
 Exmaple2.0/Views/EmbeddedView.axaml           | 23 +++++++
 Exmaple2.0/Views/EmbeddedView.axaml.cs        | 11 +++
 Exmaple2.0/Views/MainWindow.axaml             |  1 +
 Exmaple2.0/Views/MainWindow.axaml.cs          | 19 +++++
 .../Controls/MsBoxStandardView.axaml          | 69 +++++++++++--------
 .../Controls/MsBoxStandardView.axaml.cs       | 11 +++
 .../Dto/MessageBoxStandardParams.cs           |  5 ++
 MsBox.Avalonia/MessageBoxManager.cs           |  4 +-
 .../ViewModels/MsBoxStandardViewModel.cs      |  9 +++
 12 files changed, 174 insertions(+), 31 deletions(-)
 create mode 100644 Exmaple2.0/EmbeddedViewLocator.cs
 create mode 100644 Exmaple2.0/ViewModels/EmbeddedViewModel.cs
 create mode 100644 Exmaple2.0/Views/EmbeddedView.axaml
 create mode 100644 Exmaple2.0/Views/EmbeddedView.axaml.cs

diff --git a/Exmaple2.0/App.axaml b/Exmaple2.0/App.axaml
index b8dbed0..5b7f261 100644
--- a/Exmaple2.0/App.axaml
+++ b/Exmaple2.0/App.axaml
@@ -11,4 +11,11 @@
        <SimpleTheme></SimpleTheme>
         <StyleInclude Source="avares://MsBox.Avalonia.Markdown/Controls/MarkdownView.axaml" />
     </Application.Styles>
+
+    <Application.Resources>
+        <ResourceDictionary>
+            <!--This is only necessary if you want to override the default ViewLocator above-->
+            <local:EmbeddedViewLocator x:Key="EmbeddedMessageBoxViewLocator" />
+        </ResourceDictionary>
+    </Application.Resources>   
 </Application>
\ No newline at end of file
diff --git a/Exmaple2.0/EmbeddedViewLocator.cs b/Exmaple2.0/EmbeddedViewLocator.cs
new file mode 100644
index 0000000..cb6e17a
--- /dev/null
+++ b/Exmaple2.0/EmbeddedViewLocator.cs
@@ -0,0 +1,40 @@
+using Avalonia.Controls;
+using Avalonia.Controls.Templates;
+
+namespace Exmaple2._0;
+
+public class EmbeddedViewLocator
+    : IDataTemplate
+{
+    public static Control BuildView(object? data)
+    {
+        if (data is not null)
+        {
+            var name = data.GetType().AssemblyQualifiedName!.Replace("ViewModel", "View");
+            var type = Type.GetType(name);
+
+            if (type != null)
+            {
+                return (Control)Activator.CreateInstance(type)!;
+            }
+            else
+            {
+                return new TextBlock { Text = "Not Found: " + name };
+            }
+        }
+        else
+        {
+            return new TextBlock { Text = "Data object is null" };
+        }
+    }
+
+    public Control Build(object? data)
+    {
+        return BuildView(data);
+    }
+
+    public bool Match(object? data)
+    {
+        return data is not null;
+    }
+}
diff --git a/Exmaple2.0/ViewModels/EmbeddedViewModel.cs b/Exmaple2.0/ViewModels/EmbeddedViewModel.cs
new file mode 100644
index 0000000..909f575
--- /dev/null
+++ b/Exmaple2.0/ViewModels/EmbeddedViewModel.cs
@@ -0,0 +1,6 @@
+namespace Exmaple2._0.ViewModels;
+
+public class EmbeddedViewModel : ViewModelBase
+{
+    public string Message => "This is an embedded message!";
+}
diff --git a/Exmaple2.0/Views/EmbeddedView.axaml b/Exmaple2.0/Views/EmbeddedView.axaml
new file mode 100644
index 0000000..4d0e202
--- /dev/null
+++ b/Exmaple2.0/Views/EmbeddedView.axaml
@@ -0,0 +1,23 @@
+<UserControl xmlns="https://github.com/avaloniaui"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             xmlns:vm="using:Exmaple2._0.ViewModels"
+             mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450"
+             x:Class="Exmaple2._0.Views.EmbeddedView"
+             x:DataType="vm:EmbeddedViewModel">
+    <StackPanel>
+        <TextBlock Text="{Binding Message}"/>
+        <ComboBox>
+            <ComboBoxItem>
+                <TextBlock Text="Item 1"/>
+            </ComboBoxItem>
+            <ComboBoxItem>
+                <TextBlock Text="Item 2"/>
+            </ComboBoxItem>
+            <ComboBoxItem>
+                <TextBlock Text="Item 3"/>
+            </ComboBoxItem>
+        </ComboBox>
+    </StackPanel>
+</UserControl>
diff --git a/Exmaple2.0/Views/EmbeddedView.axaml.cs b/Exmaple2.0/Views/EmbeddedView.axaml.cs
new file mode 100644
index 0000000..5f80cb5
--- /dev/null
+++ b/Exmaple2.0/Views/EmbeddedView.axaml.cs
@@ -0,0 +1,11 @@
+using Avalonia.Controls;
+
+namespace Exmaple2._0.Views;
+
+public partial class EmbeddedView : UserControl
+{
+    public EmbeddedView()
+    {
+        InitializeComponent();
+    }
+}
\ No newline at end of file
diff --git a/Exmaple2.0/Views/MainWindow.axaml b/Exmaple2.0/Views/MainWindow.axaml
index 2f1a228..737840b 100644
--- a/Exmaple2.0/Views/MainWindow.axaml
+++ b/Exmaple2.0/Views/MainWindow.axaml
@@ -26,6 +26,7 @@
   <ScrollViewer>
     <StackPanel>
       <Button Name="Standard_Show"  Classes="View" Click="Standard_Show_OnClick" Content="Standard Show"/>
+      <Button Name="Standard_Context_Show"  Classes="View" Click="Standard_Context_Show_OnClick" Content="Standard Context Show"/>
       <Button Name="Standard_Dialog"  Classes="View" Click="Standard_Dialog_OnClick" Content="Standard Dialog"/>
       <Button Name="Standard_Popup"  Classes="View" Click="Standard_Popup_OnClick" Content="Custom PopUp"/>
       <Button Name="Custom_Show"  Classes="View" Click="Custom_Dialog_OnClick" Content="Custom Show"/>
diff --git a/Exmaple2.0/Views/MainWindow.axaml.cs b/Exmaple2.0/Views/MainWindow.axaml.cs
index d15a6be..abf210a 100644
--- a/Exmaple2.0/Views/MainWindow.axaml.cs
+++ b/Exmaple2.0/Views/MainWindow.axaml.cs
@@ -1,3 +1,4 @@
+using System;
 using System.Diagnostics;
 using System.Runtime.InteropServices;
 
@@ -5,10 +6,16 @@
 using Avalonia.Controls;
 using Avalonia.Interactivity;
 
+using Exmaple2._0.ViewModels;
+
 using MsBox.Avalonia;
+using MsBox.Avalonia.Controls;
 using MsBox.Avalonia.Dto;
 using MsBox.Avalonia.Enums;
 using MsBox.Avalonia.Models;
+using MsBox.Avalonia.ViewModels;
+
+using static System.Net.Mime.MediaTypeNames;
 
 namespace Exmaple2._0.Views;
 
@@ -32,6 +39,18 @@ private async void Standard_Show_OnClick(object sender, RoutedEventArgs e)
         var result = await box.ShowAsPopupAsync(this);
     }
 
+    private async void Standard_Context_Show_OnClick(object sender, RoutedEventArgs e)
+    {
+        var box = MessageBoxManager
+            .GetMessageBoxStandard(
+                "Caption",
+                "Are you sure you would like to delete appender_replace_page_1?",
+                ButtonEnum.YesNo,
+                context: new EmbeddedViewModel());
+
+        var result = await box.ShowAsPopupAsync(this);
+    }
+
     private async void Standard_Dialog_OnClick(object sender, RoutedEventArgs e)
     {
         int maxWidth = 500;
diff --git a/MsBox.Avalonia/Controls/MsBoxStandardView.axaml b/MsBox.Avalonia/Controls/MsBoxStandardView.axaml
index afb834a..7bb20dc 100644
--- a/MsBox.Avalonia/Controls/MsBoxStandardView.axaml
+++ b/MsBox.Avalonia/Controls/MsBoxStandardView.axaml
@@ -10,6 +10,7 @@
              d:DataContext="{x:Static viewModels:DesignDataContexts.StandardInputViewModel}"
              x:Class="MsBox.Avalonia.Controls.MsBoxStandardView" MinWidth="{Binding MinWidth}"
              x:DataType="viewModels:MsBoxStandardViewModel"
+             xmlns:local="clr-namespace:MsBox.Avalonia"
              MaxWidth="{Binding MaxWidth}"
              Width="{Binding Width}"
              MinHeight="{Binding MinHeight}"
@@ -24,23 +25,23 @@
         <KeyBinding Gesture="Ctrl+C" Command="{Binding Copy}" />
     </UserControl.KeyBindings>
     <UserControl.Styles>
-      <Style Selector="Image">
-          <Setter Property="Margin" Value="0,0,15,0" />
-          <Setter Property="MaxHeight" Value="50" />
-          <Setter Property="MaxWidth" Value="50" />
-          <Setter Property="VerticalAlignment" Value="Top" />
-      </Style>
-      <Style Selector="TextBox.styled">
-          <Setter Property="BorderThickness" Value="0" />
-          <Setter Property="Background" Value="Transparent" />
-          <Setter Property="TextWrapping" Value="Wrap" />
-          <Setter Property="AcceptsReturn" Value="True" />
-          <Setter Property="IsReadOnly" Value="True" />
-          <Setter Property="CaretBrush" Value="Transparent" />
-          <Setter Property="ScrollViewer.VerticalScrollBarVisibility" Value="Auto" />
-          <Setter Property="ScrollViewer.HorizontalScrollBarVisibility" Value="Disabled" />
-      </Style>
-      <Style Selector="TextBox.styled:pointerover /template/ Border#border">
+        <Style Selector="Image">
+            <Setter Property="Margin" Value="0,0,15,0" />
+            <Setter Property="MaxHeight" Value="50" />
+            <Setter Property="MaxWidth" Value="50" />
+            <Setter Property="VerticalAlignment" Value="Top" />
+        </Style>
+        <Style Selector="TextBox.styled">
+            <Setter Property="BorderThickness" Value="0" />
+            <Setter Property="Background" Value="Transparent" />
+            <Setter Property="TextWrapping" Value="Wrap" />
+            <Setter Property="AcceptsReturn" Value="True" />
+            <Setter Property="IsReadOnly" Value="True" />
+            <Setter Property="CaretBrush" Value="Transparent" />
+            <Setter Property="ScrollViewer.VerticalScrollBarVisibility" Value="Auto" />
+            <Setter Property="ScrollViewer.HorizontalScrollBarVisibility" Value="Disabled" />
+        </Style>
+        <Style Selector="TextBox.styled:pointerover /template/ Border#border">
             <Setter Property="Background" Value="Transparent" />
         </Style>
         <Style Selector="TextBox.styled:pointerover /template/ Border#PART_BorderElement">
@@ -67,12 +68,12 @@
             <Setter Property="MinWidth" Value="75" />
         </Style>
         <Style Selector="TextBlock.hyperlink">
-          <Setter Property="Foreground" Value="#3483eb" />
+            <Setter Property="Foreground" Value="#3483eb" />
         </Style>
         <Style Selector="TextBlock.hyperlink:pointerover">
-          <Setter Property="Cursor" Value="Hand" />
-          <Setter Property="TextDecorations" Value="Underline" />
-          <Setter Property="Foreground" Value="#3496eb" />
+            <Setter Property="Cursor" Value="Hand" />
+            <Setter Property="TextDecorations" Value="Underline" />
+            <Setter Property="Foreground" Value="#3496eb" />
         </Style>
     </UserControl.Styles>
     <Grid Classes="MsBoxStandardContainer">
@@ -95,6 +96,7 @@
             <Grid.RowDefinitions>
                 <RowDefinition Height="Auto" />
                 <RowDefinition Height="*" />
+                <RowDefinition Height="*" />
                 <RowDefinition Height="Auto" />
             </Grid.RowDefinitions>
             <!--Bold text-->
@@ -105,22 +107,29 @@
                      Text="{Binding ContentMessage}" IsVisible="{Binding !Markdown}" />
             <controls:MarkdownView Grid.Row="1"
                                    Markdown="{Binding ContentMessage}"
-                                   IsVisible="{Binding Markdown}" /> 
+                                   IsVisible="{Binding Markdown}" />
+
+            <!--Embedded View-->
+            <ContentControl Grid.Row="2"
+                            Name="embeddedView"
+                            Content="{Binding Path=Context, Mode=OneWay}"
+                            IsVisible="{Binding IsContextVisible}"/>
+
             <!--HyperLink-->
             <TextBlock Grid.Row="3" IsVisible="{Binding IsHyperLinkVisible }" Name="HyperLink" Classes="hyperlink"
                        Text="{Binding HyperLinkText}"
                        attachadProperty:HyperLinkCommand.Command="{Binding HyperLinkCommand}" Margin="0,0,0,15">
-          </TextBlock>
+            </TextBlock>
         </Grid>
         <!--Input-->
         <Grid Grid.Row="3" Grid.Column="0" Grid.ColumnSpan="2" IsVisible="{Binding IsInputVisible}" Margin="0,0,0,15">
-          <Grid.ColumnDefinitions>
-            <ColumnDefinition Width="Auto" />
-            <ColumnDefinition Width="*" />
-          </Grid.ColumnDefinitions>
-          <TextBlock FontFamily="{Binding FontFamily}" IsVisible="{Binding IsInputVisible}" Text="{Binding InputLabel}" VerticalAlignment="Center" Margin="0,0,15,0" />
-          <TextBox Grid.Column="1" Grid.ColumnSpan="2" Text="{Binding InputValue, Mode=TwoWay}" FontFamily="{Binding FontFamily}" IsVisible="{Binding IsInputVisible}"
-                   AcceptsReturn="{Binding IsInputMultiline}" />
+            <Grid.ColumnDefinitions>
+                <ColumnDefinition Width="Auto" />
+                <ColumnDefinition Width="*" />
+            </Grid.ColumnDefinitions>
+            <TextBlock FontFamily="{Binding FontFamily}" IsVisible="{Binding IsInputVisible}" Text="{Binding InputLabel}" VerticalAlignment="Center" Margin="0,0,15,0" />
+            <TextBox Grid.Column="1" Grid.ColumnSpan="2" Text="{Binding InputValue, Mode=TwoWay}" FontFamily="{Binding FontFamily}" IsVisible="{Binding IsInputVisible}"
+                     AcceptsReturn="{Binding IsInputMultiline}" />
         </Grid>
         <!--Buttons-->
         <StackPanel Orientation="Horizontal" VerticalAlignment="Stretch"  Grid.Row="4" Grid.Column="0" Grid.ColumnSpan="2"
diff --git a/MsBox.Avalonia/Controls/MsBoxStandardView.axaml.cs b/MsBox.Avalonia/Controls/MsBoxStandardView.axaml.cs
index 944ee98..f076a51 100644
--- a/MsBox.Avalonia/Controls/MsBoxStandardView.axaml.cs
+++ b/MsBox.Avalonia/Controls/MsBoxStandardView.axaml.cs
@@ -1,4 +1,6 @@
+using Avalonia;
 using Avalonia.Controls;
+using Avalonia.Controls.Templates;
 
 using MsBox.Avalonia.Base;
 using MsBox.Avalonia.Enums;
@@ -14,6 +16,15 @@ public partial class MsBoxStandardView : UserControl, IFullApi<ButtonResult>, IS
     public MsBoxStandardView()
     {
         InitializeComponent();
+
+        if (Application.Current.TryGetResource("EmbeddedMessageBoxViewLocator", out object? embeddedViewLocator))
+        {
+            var viewLocator = embeddedViewLocator as IDataTemplate;
+            if (viewLocator is not null)
+            {
+                embeddedView.DataTemplates.Add(viewLocator);
+            }
+        }
     }
 
     public void SetButtonResult(ButtonResult bdName)
diff --git a/MsBox.Avalonia/Dto/MessageBoxStandardParams.cs b/MsBox.Avalonia/Dto/MessageBoxStandardParams.cs
index 7429578..836bb18 100644
--- a/MsBox.Avalonia/Dto/MessageBoxStandardParams.cs
+++ b/MsBox.Avalonia/Dto/MessageBoxStandardParams.cs
@@ -9,6 +9,11 @@ public class MessageBoxStandardParams : AbstractMessageBoxParams
     /// </summary>
     public Icon Icon { get; set; } = Icon.None;
 
+    /// <summary>
+    /// View model for embedded context.
+    /// </summary>
+    public object? Context { get; set; } = null;
+
     /// <summary>
     /// Default buttons
     /// </summary>
diff --git a/MsBox.Avalonia/MessageBoxManager.cs b/MsBox.Avalonia/MessageBoxManager.cs
index d2b8de0..69738ec 100644
--- a/MsBox.Avalonia/MessageBoxManager.cs
+++ b/MsBox.Avalonia/MessageBoxManager.cs
@@ -38,6 +38,7 @@ public static IMsBox<ButtonResult> GetMessageBoxStandard(MessageBoxStandardParam
     /// <param name="text"> Text of messagebox body </param>
     /// <param name="enum"> Buttons of messagebox (default OK) </param>
     /// <param name="icon"> Icon of messagebox (default no icon) </param>
+    /// <param name="context"> Embedded view model in messagebox (default null) </param>
     /// <param name="windowStartupLocation"> Startup location of messagebox (default center screen) </param>
     /// <param name="style"></param>
     /// <returns></returns>
@@ -45,7 +46,7 @@ public static IMsBox<ButtonResult> GetMessageBoxStandard(MessageBoxStandardParam
     /// Recommended method for message box
     /// </remarks>
     public static IMsBox<ButtonResult> GetMessageBoxStandard(string title, string text,
-        ButtonEnum @enum = ButtonEnum.Ok, Icon icon = Icon.None,
+        ButtonEnum @enum = ButtonEnum.Ok, Icon icon = Icon.None, object? context = null,
         WindowStartupLocation windowStartupLocation = WindowStartupLocation.CenterScreen) =>
         GetMessageBoxStandard(new MessageBoxStandardParams
         {
@@ -53,6 +54,7 @@ public static IMsBox<ButtonResult> GetMessageBoxStandard(string title, string te
             ContentMessage = text,
             ButtonDefinitions = @enum,
             Icon = icon,
+            Context = context,
             WindowStartupLocation = windowStartupLocation
         });
 }
\ No newline at end of file
diff --git a/MsBox.Avalonia/ViewModels/MsBoxStandardViewModel.cs b/MsBox.Avalonia/ViewModels/MsBoxStandardViewModel.cs
index 71580e1..c027b10 100644
--- a/MsBox.Avalonia/ViewModels/MsBoxStandardViewModel.cs
+++ b/MsBox.Avalonia/ViewModels/MsBoxStandardViewModel.cs
@@ -19,6 +19,12 @@ public MsBoxStandardViewModel(MessageBoxStandardParams @params) :
         _enterDefaultButton = @params.EnterDefaultButton;
         _escDefaultButton = @params.EscDefaultButton;
 
+        Context = @params.Context;
+        if (Context is not null)
+        {
+            IsContextVisible = true;
+        }
+
         SetButtons(@params.ButtonDefinitions);
         ButtonClickCommand = new RelayCommand(o => ButtonClick(o.ToString()));
         EnterClickCommand = new RelayCommand(_ => EnterClick());
@@ -48,6 +54,9 @@ public void SetFullApi(IFullApi<ButtonResult> fullApi)
     public override string InputValue { get; set; }
     public override bool IsInputMultiline { get; internal set; }
     public override bool IsInputVisible { get; internal set; }
+    public virtual object? Context { get; internal set; } = null;
+    public virtual bool IsContextVisible { get; internal set; } = false;
+
     #endregion
 
     public RelayCommand ButtonClickCommand { get; }