DreamineMessageBoxWindow.xaml.cs

이 파일의 문서화 페이지로 가기

using System.Threading;
using System.Windows;
using System.Windows.Controls;
 
namespace Dreamine.UI.Wpf.Controls.MessageBox
{
    public partial class DreamineMessageBoxWindow : Window
    {
        public MessageBoxResult Result { get; private set; } = MessageBoxResult.None;

        private readonly CancellationTokenSource _cts = new();

        public DreamineMessageBoxWindow(string title, string message, MessageBoxImage icon, MessageBoxButton buttons, MessageBoxResult autoClickTarget = MessageBoxResult.None, int autoClickDelaySeconds = 0, int enableDelaySeconds = 0)
        {
            InitializeComponent();
 
            TxtTitle.Text = title;
            TxtMessage.Text = message;
            TxtIcon.Text = GetEmoji(icon);
 
            // 버튼 구성
            SetupButtons(buttons);
 
            if (enableDelaySeconds > 0)
                _ = DisableButtonsAndEnableLaterAsync(enableDelaySeconds, _cts.Token);
 
            if (autoClickTarget != MessageBoxResult.None && autoClickDelaySeconds > 0)
                _ = StartAutoClickAsync(autoClickTarget, autoClickDelaySeconds, _cts.Token);
        }

        protected override void OnClosed(EventArgs e)
        {
            _cts.Cancel();
            _cts.Dispose();
            base.OnClosed(e);
        }

        private async Task DisableButtonsAndEnableLaterAsync(int delaySeconds, CancellationToken ct = default)
        {
            var buttons = new[] { BtnOk, BtnYes, BtnNo, BtnCancel };
 
            // 버튼별 원래 텍스트 저장
            var originalTexts = buttons.ToDictionary(btn => btn, btn => btn.Content?.ToString());
 
            foreach (var btn in buttons)
                btn.IsEnabled = false;
 
            for (int i = delaySeconds; i > 0; i--)
            {
                if (ct.IsCancellationRequested) return;
 
                foreach (var btn in buttons.Where(b => b.Visibility == Visibility.Visible))
                    btn.Content = $"{originalTexts[btn]} ({i})";
 
                try { await Task.Delay(1000, ct); }
                catch (OperationCanceledException) { return; }
            }
 
            foreach (var btn in buttons)
            {
                if (btn.Visibility == Visibility.Visible)
                {
                    btn.IsEnabled = true;
                    btn.Content = originalTexts[btn];
                }
            }
        }

        private async Task StartAutoClickAsync(MessageBoxResult target, int delaySeconds, CancellationToken ct = default)
        {
            var button = GetTargetButton(target);
            if (button == null) return;
 
            string originalText = button.Content?.ToString() ?? "";
 
            for (int i = delaySeconds; i > 0; i--)
            {
                if (ct.IsCancellationRequested) return;
 
                button.Content = $"{originalText} ({i})";
 
                try { await Task.Delay(1000, ct); }
                catch (OperationCanceledException) { return; }
            }
 
            if (ct.IsCancellationRequested) return;
            button.Content = originalText;
 
            switch (target)
            {
                case MessageBoxResult.OK: BtnOk_Click(null!, null!); break;
                case MessageBoxResult.Yes: BtnYes_Click(null!, null!); break;
                case MessageBoxResult.No: BtnNo_Click(null!, null!); break;
                case MessageBoxResult.Cancel: BtnCancel_Click(null!, null!); break;
            }
        }

        private Button? GetTargetButton(MessageBoxResult result) => result switch
        {
            MessageBoxResult.OK => BtnOk,
            MessageBoxResult.Cancel => BtnCancel,
            MessageBoxResult.Yes => BtnYes,
            MessageBoxResult.No => BtnNo,
            _ => null
        };

        private string GetEmoji(MessageBoxImage icon)
        {
            return icon switch
            {
                MessageBoxImage.Error => "❌",
                MessageBoxImage.Warning => "⚠️",
                MessageBoxImage.Information => "ℹ️",
                MessageBoxImage.Question => "❓",
                _ => "🔔"
            };
        }

        private void SetupButtons(MessageBoxButton buttons)
        {
            BtnOk.Visibility = Visibility.Collapsed;
            BtnYes.Visibility = Visibility.Collapsed;
            BtnNo.Visibility = Visibility.Collapsed;
            BtnCancel.Visibility = Visibility.Collapsed;
 
            switch (buttons)
            {
                case MessageBoxButton.OK:
                    BtnOk.Focus();
                    BtnOk.Visibility = Visibility.Visible;
                    break;
                case MessageBoxButton.OKCancel:
                    BtnOk.Focus();
                    BtnOk.Visibility = Visibility.Visible;
                    BtnCancel.Visibility = Visibility.Visible;
                    break;
                case MessageBoxButton.YesNo:
                    BtnYes.Focus();
                    BtnYes.Visibility = Visibility.Visible;
                    BtnNo.Visibility = Visibility.Visible;
                    break;
                case MessageBoxButton.YesNoCancel:
                    BtnYes.Focus();
                    BtnYes.Visibility = Visibility.Visible;
                    BtnNo.Visibility = Visibility.Visible;
                    BtnCancel.Visibility = Visibility.Visible;
                    break;
            }
        }

        private void BtnOk_Click(object sender, RoutedEventArgs e)
        {
            Result = MessageBoxResult.OK;
            Close();
        }

        private void BtnCancel_Click(object sender, RoutedEventArgs e)
        {
            Result = MessageBoxResult.Cancel;
            Close();
        }

        private void BtnYes_Click(object sender, RoutedEventArgs e)
        {
            Result = MessageBoxResult.Yes;
            Close();
        }

        private void BtnNo_Click(object sender, RoutedEventArgs e)
        {
            Result = MessageBoxResult.No;
            Close();
        }
    }
}