PrettyScreenSHOT/Helpers/MessageBoxHelper.cs at 195dbdd127552fcfc15b880faf9f3e3eb84fe32a · Co0ob1iee/PrettyScreenSHOT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
using System.Windows;
using Wpf.Ui.Controls;

namespace PrettyScreenSHOT.Helpers
{
    /// <summary>
    /// Helper for showing Wpf.Ui MessageBoxes with a simplified API
    /// </summary>
    public static class MessageBoxHelper
    {
        public static MessageBoxResult Show(
            string message,
            string title,
            MessageBoxButton button = MessageBoxButton.OK)
        {
            return Application.Current.Dispatcher.Invoke(() =>
            {
                var messageBox = new Wpf.Ui.Controls.MessageBox
                {
                    Title = title,
                    Content = message,
                    ButtonLeftName = GetButtonName(button, true),
                    ButtonRightName = GetButtonName(button, false)
                };

                var result = messageBox.ShowDialogAsync().GetAwaiter().GetResult();
                return MapToMessageBoxResult(result, button);
            });
        }

        private static string GetButtonName(MessageBoxButton button, bool isLeft)
        {
            return button switch
            {
                MessageBoxButton.OK => isLeft ? "OK" : string.Empty,
                MessageBoxButton.OKCancel => isLeft ? "OK" : "Cancel",
                MessageBoxButton.YesNo => isLeft ? "Yes" : "No",
                MessageBoxButton.YesNoCancel => isLeft ? "Yes" : (isLeft ? "No" : "Cancel"),
                _ => isLeft ? "OK" : string.Empty
            };
        }

        private static MessageBoxResult MapToMessageBoxResult(
            Wpf.Ui.Controls.MessageBoxResult wpfUiResult,
            MessageBoxButton buttonType)
        {
            return buttonType switch
            {
                MessageBoxButton.OK => MessageBoxResult.OK,
                MessageBoxButton.OKCancel => wpfUiResult == Wpf.Ui.Controls.MessageBoxResult.Primary
                    ? MessageBoxResult.OK
                    : MessageBoxResult.Cancel,
                MessageBoxButton.YesNo => wpfUiResult == Wpf.Ui.Controls.MessageBoxResult.Primary
                    ? MessageBoxResult.Yes
                    : MessageBoxResult.No,
                MessageBoxButton.YesNoCancel => wpfUiResult switch
                {
                    Wpf.Ui.Controls.MessageBoxResult.Primary => MessageBoxResult.Yes,
                    Wpf.Ui.Controls.MessageBoxResult.Secondary => MessageBoxResult.No,
                    _ => MessageBoxResult.Cancel
                },
                _ => MessageBoxResult.None
            };
        }
    }
}