ShrlAlgoToolkit/WPFDark/Controls/BiaFilteringComboBox.xaml.cs

using System;
using System.Diagnostics;
using System.Linq;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Input;

using WPFDark.Internals;

namespace WPFDark.Controls
{
    public class BiaFilteringComboBox : ComboBox
    {
        #region FilterWords

        public string? FilterWords
        {
            get => _filterWords;
            set
            {
                if (value != _filterWords)
                    SetValue(FilterWordsProperty, value);
            }
        }

        private string? _filterWords;

        public static readonly DependencyProperty FilterWordsProperty =
            DependencyProperty.Register(
                nameof(FilterWords),
                typeof(string),
                typeof(BiaFilteringComboBox),
                new PropertyMetadata(
                    default,
                    (s, e) =>
                    {
                        var self = (BiaFilteringComboBox) s;
                        self._filterWords = e.NewValue?.ToString() ?? "";
                    }));

        #endregion

        #region IsEnableMouseWheel

        public bool IsMouseWheelEnabled
        {
            get => _isMouseWheelEnabled;
            set
            {
                if (value != _isMouseWheelEnabled)
                    SetValue(IsMouseWheelEnabledProperty, value);
            }
        }

        private bool _isMouseWheelEnabled = true;

        public static readonly DependencyProperty IsMouseWheelEnabledProperty =
            DependencyProperty.Register(
                nameof(IsMouseWheelEnabled),
                typeof(bool),
                typeof(BiaFilteringComboBox),
                new PropertyMetadata(
                    true,
                    (s, e) =>
                    {
                        var self = (BiaFilteringComboBox) s;
                        self._isMouseWheelEnabled = (bool) e.NewValue;
                    }));

        #endregion

        #region StartedContinuousEditingCommand

        public ICommand? StartedContinuousEditingCommand
        {
            get => _StartedContinuousEditingCommand;
            set
            {
                if (value != _StartedContinuousEditingCommand)
                    SetValue(StartedContinuousEditingCommandProperty, value);
            }
        }

        private ICommand? _StartedContinuousEditingCommand;

        public static readonly DependencyProperty StartedContinuousEditingCommandProperty =
            DependencyProperty.Register(
                nameof(StartedContinuousEditingCommand),
                typeof(ICommand),
                typeof(BiaFilteringComboBox),
                new PropertyMetadata(
                    default(ICommand),
                    (s, e) =>
                    {
                        var self = (BiaFilteringComboBox) s;
                        self._StartedContinuousEditingCommand = (ICommand) e.NewValue;
                    }));

        #endregion

        #region EndContinuousEditingCommand

        public ICommand? EndContinuousEditingCommand
        {
            get => _EndContinuousEditingCommand;
            set
            {
                if (value != _EndContinuousEditingCommand)
                    SetValue(EndContinuousEditingCommandProperty, value);
            }
        }

        private ICommand? _EndContinuousEditingCommand;

        public static readonly DependencyProperty EndContinuousEditingCommandProperty =
            DependencyProperty.Register(
                nameof(EndContinuousEditingCommand),
                typeof(ICommand),
                typeof(BiaFilteringComboBox),
                new PropertyMetadata(
                    default(ICommand),
                    (s, e) =>
                    {
                        var self = (BiaFilteringComboBox) s;
                        self._EndContinuousEditingCommand = (ICommand) e.NewValue;
                    }));

        #endregion

        static BiaFilteringComboBox()
        {
            DefaultStyleKeyProperty.OverrideMetadata(typeof(BiaFilteringComboBox),
                new FrameworkPropertyMetadata(typeof(BiaFilteringComboBox)));
        }

        private bool _isOpen;
        private bool _isDoneSetValue;
        private bool _isDoneDiscard;
        private object? _ContinuousEditingStartValue;
        private FrameworkElement? _dropDown;

        protected override void OnDropDownOpened(EventArgs e)
        {
            base.OnDropDownOpened(e);

            if (_isOpen)
                return;

            _isOpen = true;
            _isDoneSetValue = false;
            _isDoneDiscard = false;
            _ContinuousEditingStartValue = SelectedItem;
            _dropDown = FindDropDown();

            StartedContinuousEditingCommand?.ExecuteIfCan(null);

            _dropDown.PreviewMouseDown += DropDownOnPreviewMouseDown;
            _dropDown.PreviewKeyDown += DropDownOnPreviewKeyDown;
        }

        protected override void OnDropDownClosed(EventArgs e)
        {
            base.OnDropDownClosed(e);

            Debug.Assert(_dropDown != null);

            _dropDown.PreviewMouseDown -= DropDownOnPreviewMouseDown;
            _dropDown.PreviewKeyDown -= DropDownOnPreviewKeyDown;

            if (_isDoneSetValue == false)
                Discard();

            _isOpen = false;
        }

        private void DropDownOnPreviewMouseDown(object sender, MouseButtonEventArgs e)
        {
            SetValue();
        }

        private void DropDownOnPreviewKeyDown(object sender, KeyEventArgs e)
        {
            if (e.Key == Key.Return)
                SetValue();
        }

        private void SetValue()
        {
            if (_isDoneSetValue)
                return;

            _isDoneSetValue = true;

            if (EndContinuousEditingCommand != null)
            {
                if (EndContinuousEditingCommand.CanExecute(null))
                {
                    var changedValue = SelectedItem;
                    SelectedItem = _ContinuousEditingStartValue;

                    EndContinuousEditingCommand.Execute(null);

                    SelectedItem = changedValue;
                }
            }
        }

        private void Discard()
        {
            if (_isDoneDiscard)
                return;

            _isDoneDiscard = true;

            var done = false;

            if (EndContinuousEditingCommand != null)
            {
                if (EndContinuousEditingCommand.CanExecute(null))
                {
                    SelectedItem = _ContinuousEditingStartValue;

                    EndContinuousEditingCommand.Execute(null);

                    done = true;
                }
            }

            if (done == false)
                SelectedItem = _ContinuousEditingStartValue;
        }

        private FrameworkElement FindDropDown() =>
            this.Descendants().OfType<FrameworkElement>()
                .First(x => x.Name == "dropdown");
    }
}