From 66894aa56924ed18cdcaa8a75ff15dc5239e724d Mon Sep 17 00:00:00 2001
From: rabbitism <dongbin@irihi.tech>
Date: Thu, 11 Jan 2024 02:41:58 +0800
Subject: [PATCH] feat: add default buttons.

---
 .../ViewModels/MessageBoxDemoViewModel.cs     |   2 +-
 demo/Ursa.Demo/Views/MainWindow.axaml         |   2 -
 .../Controls/MessageBoxWindow.axaml           |  63 ++++++----
 src/Ursa/Controls/MessageBox/MessageBox.cs    |  10 +-
 .../Controls/MessageBox/MessageBoxButton.cs   |   9 ++
 .../Controls/MessageBox/MessageBoxResult.cs   |  10 ++
 .../Controls/MessageBox/MessageBoxWindow.cs   | 117 ++++++++++++++++++
 7 files changed, 183 insertions(+), 30 deletions(-)
 create mode 100644 src/Ursa/Controls/MessageBox/MessageBoxButton.cs
 create mode 100644 src/Ursa/Controls/MessageBox/MessageBoxResult.cs

diff --git a/demo/Ursa.Demo/ViewModels/MessageBoxDemoViewModel.cs b/demo/Ursa.Demo/ViewModels/MessageBoxDemoViewModel.cs
index 809b95a..3d95040 100644
--- a/demo/Ursa.Demo/ViewModels/MessageBoxDemoViewModel.cs
+++ b/demo/Ursa.Demo/ViewModels/MessageBoxDemoViewModel.cs
@@ -17,6 +17,6 @@ public class MessageBoxDemoViewModel: ObservableObject
 
     private async Task OnDefaultMessageAsync()
     {
-        await MessageBox.ShowAsync("Hello Message Box");
+        var result = await MessageBox.ShowAsync("Hello Message Box");
     }
 }
\ No newline at end of file
diff --git a/demo/Ursa.Demo/Views/MainWindow.axaml b/demo/Ursa.Demo/Views/MainWindow.axaml
index 5504e4a..054e043 100644
--- a/demo/Ursa.Demo/Views/MainWindow.axaml
+++ b/demo/Ursa.Demo/Views/MainWindow.axaml
@@ -10,8 +10,6 @@
     d:DesignHeight="450"
     d:DesignWidth="800"
     x:CompileBindings="True"
-    ExtendClientAreaTitleBarHeightHint="48"
-    ExtendClientAreaToDecorationsHint="True"
     x:DataType="viewModels:MainWindowViewModel"
     Icon="/Assets/Ursa.ico"
     mc:Ignorable="d">
diff --git a/src/Ursa.Themes.Semi/Controls/MessageBoxWindow.axaml b/src/Ursa.Themes.Semi/Controls/MessageBoxWindow.axaml
index 82ff018..a2c496f 100644
--- a/src/Ursa.Themes.Semi/Controls/MessageBoxWindow.axaml
+++ b/src/Ursa.Themes.Semi/Controls/MessageBoxWindow.axaml
@@ -1,39 +1,52 @@
-<ResourceDictionary xmlns="https://github.com/avaloniaui"
-                    xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
-                    xmlns:u="https://irihi.tech/ursa">
-    <!-- Add Resources Here -->
-    <ControlTheme TargetType="u:MessageBoxWindow" x:Key="{x:Type u:MessageBoxWindow}">
+<ResourceDictionary
+    xmlns="https://github.com/avaloniaui"
+    xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+    xmlns:u="https://irihi.tech/ursa">
+    <!--  Add Resources Here  -->
+    <ControlTheme x:Key="{x:Type u:MessageBoxWindow}" TargetType="u:MessageBoxWindow">
+        <Setter Property="Title" Value="{x:Null}"></Setter>
         <Setter Property="Background" Value="{DynamicResource WindowDefaultBackground}" />
         <Setter Property="TransparencyBackgroundFallback" Value="{DynamicResource WindowDefaultBackground}" />
         <Setter Property="Foreground" Value="{DynamicResource WindowDefaultForeground}" />
         <Setter Property="FontSize" Value="{DynamicResource DefaultFontSize}" />
         <Setter Property="FontFamily" Value="{DynamicResource DefaultFontFamily}" />
-        <Setter Property="Padding" Value="48 24"></Setter>
-        <Setter Property="MinHeight" Value="300"></Setter>
-        <Setter Property="SizeToContent" Value="WidthAndHeight"></Setter>
-        <Setter Property="WindowStartupLocation" Value="CenterOwner"></Setter>
-        <Setter Property="ExtendClientAreaTitleBarHeightHint" Value="48"></Setter>
-        <Setter Property="ExtendClientAreaToDecorationsHint" Value="True"></Setter>
-        <Setter Property="SystemDecorations" Value="BorderOnly"></Setter>
-        <Setter Property="CanResize" Value="True"></Setter>
+        <Setter Property="Padding" Value="48 24" />
+        <Setter Property="SizeToContent" Value="WidthAndHeight" />
+        <Setter Property="WindowStartupLocation" Value="CenterOwner" />
+        <Setter Property="ExtendClientAreaTitleBarHeightHint" Value="0" />
+        <Setter Property="ExtendClientAreaToDecorationsHint" Value="True" />
+        <Setter Property="SystemDecorations" Value="BorderOnly" />
+        <Setter Property="CanResize" Value="True" />
         <Setter Property="Template">
             <ControlTemplate TargetType="u:MessageBoxWindow">
                 <Panel>
                     <Border Name="PART_TransparencyFallback" IsHitTestVisible="False" />
                     <Border Background="{TemplateBinding Background}" IsHitTestVisible="False" />
                     <Panel Margin="{TemplateBinding WindowDecorationMargin}" Background="Transparent" />
-                    <VisualLayerManager>
-                        <VisualLayerManager.ChromeOverlayLayer>
-                            <TitleBar />
-                        </VisualLayerManager.ChromeOverlayLayer>
-                        <ContentPresenter Grid.Row="0"
-                                          Name="PART_ContentPresenter"
-                                          Margin="{TemplateBinding Padding}"
-                                          HorizontalContentAlignment="{TemplateBinding HorizontalContentAlignment}"
-                                          VerticalContentAlignment="{TemplateBinding VerticalContentAlignment}"
-                                          Content="{TemplateBinding Content}"
-                                          ContentTemplate="{TemplateBinding ContentTemplate}" />
-                    </VisualLayerManager>
+                    <Grid RowDefinitions="Auto, *, Auto">
+                        <Grid ColumnDefinitions="*, Auto" Grid.Row="0">
+                            <TextBlock Grid.Column="0" Text="{TemplateBinding Title}" />
+                            <Button
+                                Grid.Column="1"
+                                Name="{x:Static u:MessageBoxWindow.PART_CloseButton}"
+                                Content="X"
+                                />
+                        </Grid>
+                        <ContentPresenter
+                            Name="PART_ContentPresenter"
+                            Grid.Row="1"
+                            Margin="{TemplateBinding Padding}"
+                            HorizontalContentAlignment="{TemplateBinding HorizontalContentAlignment}"
+                            VerticalContentAlignment="{TemplateBinding VerticalContentAlignment}"
+                            Content="{TemplateBinding Content}"
+                            ContentTemplate="{TemplateBinding ContentTemplate}" />
+                        <StackPanel Grid.Row="2" Orientation="Horizontal">
+                            <Button Name="{x:Static u:MessageBoxWindow.PART_YesButton}" Classes="Primary" Content="Yes" />
+                            <Button Name="{x:Static u:MessageBoxWindow.PART_NoButton}" Classes="Danger" Content="No" />
+                            <Button Name="{x:Static u:MessageBoxWindow.PART_OKButton}" Classes="Primary" Content="OK" />
+                            <Button Name="{x:Static u:MessageBoxWindow.PART_CancelButton}" Classes="Tertiary" Content="Cancel" />
+                        </StackPanel>
+                    </Grid>
                 </Panel>
             </ControlTemplate>
         </Setter>
diff --git a/src/Ursa/Controls/MessageBox/MessageBox.cs b/src/Ursa/Controls/MessageBox/MessageBox.cs
index eccf256..9fa9c02 100644
--- a/src/Ursa/Controls/MessageBox/MessageBox.cs
+++ b/src/Ursa/Controls/MessageBox/MessageBox.cs
@@ -5,7 +5,7 @@ namespace Ursa.Controls;
 
 public static class MessageBox
 {
-    public static async Task ShowAsync(string message)
+    public static async Task<MessageBoxResult> ShowAsync(string message)
     {
         var messageWindow = new MessageBoxWindow()
         {
@@ -18,11 +18,17 @@ public static class MessageBox
             if (main is null)
             {
                 messageWindow.Show();
+                return MessageBoxResult.None;
             }
             else
             {
-                await messageWindow.ShowDialog(main);
+                var result = await messageWindow.ShowDialog<MessageBoxResult>(main);
+                return result;
             }
         }
+        else
+        {
+            return MessageBoxResult.None;
+        }
     }
 }
\ No newline at end of file
diff --git a/src/Ursa/Controls/MessageBox/MessageBoxButton.cs b/src/Ursa/Controls/MessageBox/MessageBoxButton.cs
new file mode 100644
index 0000000..91fd081
--- /dev/null
+++ b/src/Ursa/Controls/MessageBox/MessageBoxButton.cs
@@ -0,0 +1,9 @@
+namespace Ursa.Controls;
+
+public enum MessageBoxButton
+{
+    OK,
+    OKCancel,
+    YesNo,
+    YesNoCancel,
+}
\ No newline at end of file
diff --git a/src/Ursa/Controls/MessageBox/MessageBoxResult.cs b/src/Ursa/Controls/MessageBox/MessageBoxResult.cs
new file mode 100644
index 0000000..ad2491b
--- /dev/null
+++ b/src/Ursa/Controls/MessageBox/MessageBoxResult.cs
@@ -0,0 +1,10 @@
+namespace Ursa.Controls;
+
+public enum MessageBoxResult
+{
+    Cancel,
+    No,
+    None,
+    OK,
+    Yes,
+}
\ No newline at end of file
diff --git a/src/Ursa/Controls/MessageBox/MessageBoxWindow.cs b/src/Ursa/Controls/MessageBox/MessageBoxWindow.cs
index 5d2dbef..2e111ae 100644
--- a/src/Ursa/Controls/MessageBox/MessageBoxWindow.cs
+++ b/src/Ursa/Controls/MessageBox/MessageBoxWindow.cs
@@ -1,14 +1,131 @@
 using Avalonia.Controls;
+using Avalonia.Controls.Metadata;
+using Avalonia.Controls.Primitives;
+using Avalonia.Input;
+using Avalonia.Interactivity;
 using Avalonia.Platform;
 
 namespace Ursa.Controls;
 
+[TemplatePart(PART_CloseButton, typeof(Button))]
+[TemplatePart(PART_NoButton, typeof(Button))]
+[TemplatePart(PART_OKButton, typeof(Button))]
+[TemplatePart(PART_CancelButton, typeof(Button))]
+[TemplatePart(PART_YesButton, typeof(Button))]
 public class MessageBoxWindow: Window
 {
+    public const string PART_CloseButton = "PART_CloseButton";
+    public const string PART_YesButton = "PART_YesButton";
+    public const string PART_NoButton = "PART_NoButton";
+    public const string PART_OKButton = "PART_OKButton";
+    public const string PART_CancelButton = "PART_CancelButton";
+
+    private MessageBoxButton _buttonConfigs;
+    
+    private Button? _closeButton;
+    private Button? _yesButton;
+    private Button? _noButton;
+    private Button? _okButton;
+    private Button? _cancelButton;
+    
     protected override Type StyleKeyOverride => typeof(MessageBoxWindow);
 
     static MessageBoxWindow()
     {
         
     }
+
+    public MessageBoxWindow()
+    {
+        _buttonConfigs = MessageBoxButton.OK;
+    }
+
+    public MessageBoxWindow(MessageBoxButton buttons)
+    {
+        _buttonConfigs = buttons;
+    }
+
+    protected override void OnApplyTemplate(TemplateAppliedEventArgs e)
+    {
+        base.OnApplyTemplate(e);
+        if (_closeButton != null)
+        {
+            _closeButton.Click -= OnCloseButtonClick;
+        }
+        if (_yesButton != null)
+        {
+            _yesButton.Click -= OnYesButtonClick;
+        }
+        if (_noButton != null)
+        {
+            _noButton.Click -= OnNoButtonClick;
+        }
+        if (_okButton != null)
+        {
+            _okButton.Click -= OnOKButtonClick;
+        }
+        if (_cancelButton != null)
+        {
+            _cancelButton.Click -= OnCancelButtonClick;
+        }
+        _yesButton = e.NameScope.Find<Button>(PART_YesButton);
+        _noButton = e.NameScope.Find<Button>(PART_NoButton);
+        _okButton = e.NameScope.Find<Button>(PART_OKButton);
+        _cancelButton = e.NameScope.Find<Button>(PART_CancelButton);
+        _closeButton = e.NameScope.Find<Button>(PART_CloseButton);
+        if (_closeButton is not null)
+        {
+            _closeButton.Click += OnCloseButtonClick;
+        }
+        if (_yesButton is not null)
+        {
+            _yesButton.Click += OnYesButtonClick;
+        }
+        if (_noButton is not null)
+        {
+            _noButton.Click += OnNoButtonClick;
+        }
+        if (_okButton is not null)
+        {
+            _okButton.Click += OnOKButtonClick;
+        }
+        if (_cancelButton is not null)
+        {
+            _cancelButton.Click += OnCancelButtonClick;
+        }
+    }
+
+    private void OnCloseButtonClick(object sender, RoutedEventArgs e)
+    {
+        this.Close(MessageBoxResult.None);
+    }
+    
+    private void OnYesButtonClick(object sender, RoutedEventArgs e)
+    {
+        this.Close(MessageBoxResult.Yes);
+    }
+    
+    private void OnNoButtonClick(object sender, RoutedEventArgs e)
+    {
+        this.Close(MessageBoxResult.No);
+    }
+    
+    private void OnOKButtonClick(object sender, RoutedEventArgs e)
+    {
+        this.Close(MessageBoxResult.OK);
+    }
+    
+    private void OnCancelButtonClick(object sender, RoutedEventArgs e)
+    {
+        this.Close(MessageBoxResult.Cancel);
+    }
+
+    protected override void OnKeyUp(KeyEventArgs e)
+    {
+        base.OnKeyUp(e);
+        if (e.Key == Key.Escape)
+        {
+            this.Close(MessageBoxResult.None);
+        }
+    }
 }
\ No newline at end of file