RelayCommand.cs

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

using System.Windows.Input;
 
namespace Dreamine.MVVM.ViewModels
{
    public sealed class RelayCommand : ICommand
    {
        private readonly Action _execute;
        private readonly Func<bool>? _canExecute;

        private readonly SynchronizationContext? _syncContext;

        public RelayCommand(Action execute, Func<bool>? canExecute = null)
        {
            // Argument 검증
            _execute = execute ?? throw new ArgumentNullException(nameof(execute));
            _canExecute = canExecute;
 
            // 생성 시점의 SynchronizationContext를 캡처하여 UI 스레드 마샬링에 사용
            _syncContext = SynchronizationContext.Current;
        }

        public bool CanExecute(object? parameter)
        {
            // canExecute 미지정 시 항상 실행 가능
            return _canExecute?.Invoke() ?? true;
        }

        public void Execute(object? parameter)
        {
            if (CanExecute(parameter))
                _execute();
        }

        public event EventHandler? CanExecuteChanged;

        public void RaiseCanExecuteChanged()
        {
            // 이벤트 핸들러가 없으면 빠른 종료
            var handler = CanExecuteChanged;
            if (handler is null)
                return;
 
            // 캡처한 컨텍스트가 있고 현재 컨텍스트가 다르면 Post로 UI 스레드 호출
            if (_syncContext != null && !ReferenceEquals(SynchronizationContext.Current, _syncContext))
            {
                _syncContext.Post(static state =>
                {
                    // state는 (RelayCommand, EventHandler) 튜플
                    var tuple = ((RelayCommand cmd, EventHandler evt))state!;
                    tuple.evt.Invoke(tuple.cmd, EventArgs.Empty);
                }, (this, handler));
 
                return;
            }
 
            // 동일 컨텍스트(보통 UI)에서 즉시 호출
            handler.Invoke(this, EventArgs.Empty);
        }
    }

    public sealed class RelayCommand<T> : ICommand
    {
        private readonly Action<T> _execute;
        private readonly Func<T, bool>? _canExecute;

        private readonly SynchronizationContext? _syncContext;

        public RelayCommand(Action<T> execute, Func<T, bool>? canExecute = null)
        {
            // Argument 검증
            _execute = execute ?? throw new ArgumentNullException(nameof(execute));
            _canExecute = canExecute;
 
            // 생성 시점 SynchronizationContext 캡처
            _syncContext = SynchronizationContext.Current;
        }

        public bool CanExecute(object? parameter)
        {
            // canExecute 미지정 시 항상 실행 가능
            if (_canExecute is null)
                return true;
 
            // 파라미터 변환 실패 시 실행 불가 처리(예외 대신 false)
            if (!TryGetParameter(parameter, out var value))
                return false;
 
            return _canExecute.Invoke(value);
        }

        public void Execute(object? parameter)
        {
            if (!CanExecute(parameter))
                return;
 
            if (!TryGetParameter(parameter, out var value))
                throw new ArgumentException($"Command parameter is not assignable to {typeof(T).FullName}.", nameof(parameter));
 
            _execute(value);
        }

        public event EventHandler? CanExecuteChanged;

        public void RaiseCanExecuteChanged()
        {
            // 핸들러 스냅샷
            var handler = CanExecuteChanged;
            if (handler is null)
                return;
 
            // 다른 컨텍스트라면 Post로 이벤트 발생
            if (_syncContext != null && !ReferenceEquals(SynchronizationContext.Current, _syncContext))
            {
                _syncContext.Post(static state =>
                {
                    // state는 (RelayCommand<T>, EventHandler) 튜플
                    var tuple = ((RelayCommand<T> cmd, EventHandler evt))state!;
                    tuple.evt.Invoke(tuple.cmd, EventArgs.Empty);
                }, (this, handler));
 
                return;
            }
 
            handler.Invoke(this, EventArgs.Empty);
        }

        private static bool TryGetParameter(object? parameter, out T value)
        {
            // null 처리
            if (parameter is null)
            {
                // 참조 타입 또는 Nullable이면 default 허용
                if (default(T) is null)
                {
                    value = default!;
                    return true;
                }
 
                // non-nullable value type은 null 불가
                value = default!;
                return false;
            }
 
            // 이미 T면 그대로 사용
            if (parameter is T t)
            {
                value = t;
                return true;
            }
 
            // 타입 불일치
            value = default!;
            return false;
        }
    }
}