PixiEditor/PixiEditor/Models/Controllers/Shortcuts/ShortcutController.cs

using PixiEditor.Models.Tools;
using PixiEditor.Models.Tools.Tools;
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Linq;
using System.Windows.Documents;
using System.Windows.Input;

namespace PixiEditor.Models.Controllers.Shortcuts
{
    public class ShortcutController
    {
        public ShortcutController(params ShortcutGroup[] shortcutGroups)
        {
            ShortcutGroups = new ObservableCollection<ShortcutGroup>(shortcutGroups);
        }

        public static bool ShortcutExecutionBlocked => _shortcutExecutionBlockers.Count > 0;

        private static List<string> _shortcutExecutionBlockers = new List<string>();

        public ObservableCollection<ShortcutGroup> ShortcutGroups { get; init; }

        public Shortcut LastShortcut { get; private set; }

        public Dictionary<Key, Tool> TransientShortcuts { get; set; } = new Dictionary<Key, Tool>();

        public static void BlockShortcutExection(string blocker)
        {
            if (_shortcutExecutionBlockers.Contains(blocker)) return;
            _shortcutExecutionBlockers.Add(blocker);
        }

        public static void UnblockShortcutExecution(string blocker)
        {
            if (!_shortcutExecutionBlockers.Contains(blocker)) return;
            _shortcutExecutionBlockers.Remove(blocker);
        }

        public static void UnblockShortcutExecutionAll()
        {
            _shortcutExecutionBlockers.Clear();
        }

        public Shortcut GetToolShortcut<T>()
        {
            return GetToolShortcut(typeof(T));
        }

        public Shortcut GetToolShortcut(Type type)
        {
            return ShortcutGroups.SelectMany(x => x.Shortcuts).ToList().Where(i => i.CommandParameter is Type nextType && nextType == type).SingleOrDefault();
        }

        public Key GetToolShortcutKey<T>()
        {
            return GetToolShortcutKey(typeof(T));
        }

        public Key GetToolShortcutKey(Type type)
        {
            var sh = GetToolShortcut(type);
            return sh != null ? sh.ShortcutKey : Key.None;
        }

        public void KeyPressed(Key key, ModifierKeys modifiers)
        {
            if (!ShortcutExecutionBlocked)
            {
                Shortcut[] shortcuts = ShortcutGroups.SelectMany(x => x.Shortcuts).ToList().FindAll(x => x.ShortcutKey == key).ToArray();
                if (shortcuts.Length < 1)
                {
                    return;
                }

                shortcuts = shortcuts.OrderByDescending(x => x.Modifier).ToArray();
                for (int i = 0; i < shortcuts.Length; i++)
                {
                    if (modifiers.HasFlag(shortcuts[i].Modifier))
                    {
                        shortcuts[i].Execute();
                        LastShortcut = shortcuts[i];
                        break;
                    }
                }
            }
        }
    }
}