using Avalonia;
using Avalonia.Controls;
using Avalonia.Controls.Primitives;
using Avalonia.Input;
using Avalonia.Interactivity;
using Avalonia.LogicalTree;
using Ryujinx.Ava.UI.Helpers;
using Ryujinx.Ava.UI.ViewModels.Input;
using Ryujinx.Input;
using Ryujinx.Input.Assigner;
using System.Collections.Generic;
using System;
using Button = Ryujinx.Input.Button;
using Key = Ryujinx.Common.Configuration.Hid.Key;

namespace Ryujinx.Ava.UI.Views.Input
{
    public partial class KeyboardInputView : UserControl
    {
        private ButtonKeyAssigner _currentAssigner;

        public KeyboardInputView()
        {
            InitializeComponent();

            foreach (ILogical visual in SettingButtons.GetLogicalDescendants())
            {
                if (visual is ToggleButton button and not CheckBox)
                {
                    button.IsCheckedChanged += Button_IsCheckedChanged;
                }
            }
        }

        protected override void OnPointerReleased(PointerReleasedEventArgs e)
        {
            base.OnPointerReleased(e);

            if (_currentAssigner is { ToggledButton.IsPointerOver: false })
            {
                _currentAssigner.Cancel();
            }
        }

        private void Button_IsCheckedChanged(object sender, RoutedEventArgs e)
        {
            if (sender is not ToggleButton button) 
                return;
            
            if (button.IsChecked is true)
            {
                if (_currentAssigner != null && button == _currentAssigner.ToggledButton)
                {
                    return;
                }

                if (_currentAssigner == null)
                {
                    _currentAssigner = new ButtonKeyAssigner(button);

                    Focus(NavigationMethod.Pointer);

                    PointerPressed += MouseClick;

                    if (DataContext is not KeyboardInputViewModel viewModel)
                        return;

                    IKeyboard keyboard =
                        (IKeyboard)viewModel.ParentModel.AvaloniaKeyboardDriver.GetGamepad("0"); // Open Avalonia keyboard for cancel operations.
                    IButtonAssigner assigner = 
                        new KeyboardKeyAssigner((IKeyboard)viewModel.ParentModel.SelectedGamepad);

                    _currentAssigner.ButtonAssigned += (_, e) =>
                    {
                        if (e.ButtonValue.HasValue)
                        {
                            Button buttonValue = e.ButtonValue.Value;
                            FlagInputConfigChanged();

                            switch (button.Name)
                            {
                                case "ButtonZl":
                                    viewModel.Config.ButtonZl = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonL":
                                    viewModel.Config.ButtonL = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonMinus":
                                    viewModel.Config.ButtonMinus = buttonValue.AsHidType<Key>();
                                    break;
                                case "LeftStickButton":
                                    viewModel.Config.LeftStickButton = buttonValue.AsHidType<Key>();
                                    break;
                                case "LeftStickUp":
                                    viewModel.Config.LeftStickUp = buttonValue.AsHidType<Key>();
                                    break;
                                case "LeftStickDown":
                                    viewModel.Config.LeftStickDown = buttonValue.AsHidType<Key>();
                                    break;
                                case "LeftStickRight":
                                    viewModel.Config.LeftStickRight = buttonValue.AsHidType<Key>();
                                    break;
                                case "LeftStickLeft":
                                    viewModel.Config.LeftStickLeft = buttonValue.AsHidType<Key>();
                                    break;
                                case "DpadUp":
                                    viewModel.Config.DpadUp = buttonValue.AsHidType<Key>();
                                    break;
                                case "DpadDown":
                                    viewModel.Config.DpadDown = buttonValue.AsHidType<Key>();
                                    break;
                                case "DpadLeft":
                                    viewModel.Config.DpadLeft = buttonValue.AsHidType<Key>();
                                    break;
                                case "DpadRight":
                                    viewModel.Config.DpadRight = buttonValue.AsHidType<Key>();
                                    break;
                                case "LeftButtonSr":
                                    viewModel.Config.LeftButtonSr = buttonValue.AsHidType<Key>();
                                    break;
                                case "LeftButtonSl":
                                    viewModel.Config.LeftButtonSl = buttonValue.AsHidType<Key>();
                                    break;
                                case "RightButtonSr":
                                    viewModel.Config.RightButtonSr = buttonValue.AsHidType<Key>();
                                    break;
                                case "RightButtonSl":
                                    viewModel.Config.RightButtonSl = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonZr":
                                    viewModel.Config.ButtonZr = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonR":
                                    viewModel.Config.ButtonR = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonPlus":
                                    viewModel.Config.ButtonPlus = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonA":
                                    viewModel.Config.ButtonA = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonB":
                                    viewModel.Config.ButtonB = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonX":
                                    viewModel.Config.ButtonX = buttonValue.AsHidType<Key>();
                                    break;
                                case "ButtonY":
                                    viewModel.Config.ButtonY = buttonValue.AsHidType<Key>();
                                    break;
                                case "RightStickButton":
                                    viewModel.Config.RightStickButton = buttonValue.AsHidType<Key>();
                                    break;
                                case "RightStickUp":
                                    viewModel.Config.RightStickUp = buttonValue.AsHidType<Key>();
                                    break;
                                case "RightStickDown":
                                    viewModel.Config.RightStickDown = buttonValue.AsHidType<Key>();
                                    break;
                                case "RightStickRight":
                                    viewModel.Config.RightStickRight = buttonValue.AsHidType<Key>();
                                    break;
                                case "RightStickLeft":
                                    viewModel.Config.RightStickLeft = buttonValue.AsHidType<Key>();
                                    break;
                            }
                        }
                    };

                    _currentAssigner.GetInputAndAssign(assigner, keyboard);
                }
                else
                {
                    if (_currentAssigner != null)
                    {
                        _currentAssigner.Cancel();
                        _currentAssigner = null;
                        button.IsChecked = false;
                    }
                }
            }
            else
            {
                _currentAssigner?.Cancel();
                _currentAssigner = null;
            }
        }

        private void FlagInputConfigChanged()
        {
            (DataContext as KeyboardInputViewModel)!.ParentModel.IsModified = true;
        }

        private void MouseClick(object sender, PointerPressedEventArgs e)
        {
            bool shouldUnbind = e.GetCurrentPoint(this).Properties.IsMiddleButtonPressed;

            bool shouldRemoveBinding = e.GetCurrentPoint(this).Properties.IsRightButtonPressed;

            if (shouldRemoveBinding)
            {
                DeleteBind();
            }

            _currentAssigner?.Cancel(shouldUnbind);

            PointerPressed -= MouseClick;
        }

        private void DeleteBind()
        {
            if (DataContext is not KeyboardInputViewModel viewModel)
                return;

            if (_currentAssigner != null)
            {
                Dictionary<string, Action> buttonActions = new Dictionary<string, Action>
                {
                    { "ButtonZl", () => viewModel.Config.ButtonZl = Key.Unbound },
                    { "ButtonL", () => viewModel.Config.ButtonL = Key.Unbound },
                    { "ButtonMinus", () => viewModel.Config.ButtonMinus = Key.Unbound },
                    { "LeftStickButton", () => viewModel.Config.LeftStickButton = Key.Unbound },
                    { "LeftStickUp", () => viewModel.Config.LeftStickUp = Key.Unbound },
                    { "LeftStickDown", () => viewModel.Config.LeftStickDown = Key.Unbound },
                    { "LeftStickRight", () => viewModel.Config.LeftStickRight = Key.Unbound },
                    { "LeftStickLeft", () => viewModel.Config.LeftStickLeft = Key.Unbound },
                    { "DpadUp", () => viewModel.Config.DpadUp = Key.Unbound },
                    { "DpadDown", () => viewModel.Config.DpadDown = Key.Unbound },
                    { "DpadLeft", () => viewModel.Config.DpadLeft = Key.Unbound },
                    { "DpadRight", () => viewModel.Config.DpadRight = Key.Unbound },
                    { "LeftButtonSr", () => viewModel.Config.LeftButtonSr = Key.Unbound },
                    { "LeftButtonSl", () => viewModel.Config.LeftButtonSl = Key.Unbound },
                    { "RightButtonSr", () => viewModel.Config.RightButtonSr = Key.Unbound },
                    { "RightButtonSl", () => viewModel.Config.RightButtonSl = Key.Unbound },
                    { "ButtonZr", () => viewModel.Config.ButtonZr = Key.Unbound },
                    { "ButtonR", () => viewModel.Config.ButtonR = Key.Unbound },
                    { "ButtonPlus", () => viewModel.Config.ButtonPlus = Key.Unbound },
                    { "ButtonA", () => viewModel.Config.ButtonA = Key.Unbound },
                    { "ButtonB", () => viewModel.Config.ButtonB = Key.Unbound },
                    { "ButtonX", () => viewModel.Config.ButtonX = Key.Unbound },
                    { "ButtonY", () => viewModel.Config.ButtonY = Key.Unbound },
                    { "RightStickButton", () => viewModel.Config.RightStickButton = Key.Unbound },
                    { "RightStickUp", () => viewModel.Config.RightStickUp = Key.Unbound },
                    { "RightStickDown", () => viewModel.Config.RightStickDown = Key.Unbound },
                    { "RightStickRight", () => viewModel.Config.RightStickRight = Key.Unbound },
                    { "RightStickLeft", () => viewModel.Config.RightStickLeft = Key.Unbound }
                };

                if (buttonActions.TryGetValue(_currentAssigner.ToggledButton.Name, out Action action))
                {
                    action();
                    FlagInputConfigChanged();
                }
            }
        }

        protected override void OnDetachedFromVisualTree(VisualTreeAttachmentEventArgs e)
        {
            base.OnDetachedFromVisualTree(e);
            _currentAssigner?.Cancel();
            _currentAssigner = null;
        }
    }
}