using Avalonia;
using Avalonia.Controls;
using Avalonia.Input;
using Ryujinx.Input;
using System;
using System.Numerics;
using MouseButton = Ryujinx.Input.MouseButton;
using Size = System.Drawing.Size;

namespace Ryujinx.Ava.Input
{
    internal class AvaloniaMouseDriver : IGamepadDriver
    {
        private Control _widget;
        private bool _isDisposed;
        private Size _size;
        private readonly Window _window;

        public bool[] PressedButtons { get; }

        public Vector2 CurrentPosition { get; private set; }
        public Vector2 Scroll { get; private set; }

        public AvaloniaMouseDriver(Window window, Control parent)
        {
            _widget = parent;
            _window = window;

            _widget.PointerMoved += Parent_PointerMovedEvent;
            _widget.PointerPressed += Parent_PointerPressEvent;
            _widget.PointerReleased += Parent_PointerReleaseEvent;
            _widget.PointerWheelChanged += Parent_ScrollEvent;
            
            _window.PointerMoved += Parent_PointerMovedEvent;
            _window.PointerPressed += Parent_PointerPressEvent;
            _window.PointerReleased += Parent_PointerReleaseEvent;
            _window.PointerWheelChanged += Parent_ScrollEvent;

            PressedButtons = new bool[(int)MouseButton.Count];

            _size = new Size((int)parent.Bounds.Width, (int)parent.Bounds.Height);
            parent.GetObservable(Control.BoundsProperty).Subscribe(Resized);
        }

        private void Resized(Rect rect)
        {
            _size = new Size((int)rect.Width, (int)rect.Height);
        }

        private void Parent_ScrollEvent(object o, PointerWheelEventArgs args)
        {
            Scroll = new Vector2((float)args.Delta.X, (float)args.Delta.Y);
        }

        private void Parent_PointerReleaseEvent(object o, PointerReleasedEventArgs args)
        {
            PressedButtons[(int)args.InitialPressMouseButton - 1] = false;
        }

        private void Parent_PointerPressEvent(object o, PointerPressedEventArgs args)
        {
            var pointerProperties = args.GetCurrentPoint(_widget).Properties;

            PressedButtons[(int)pointerProperties.PointerUpdateKind] = true;
        }

        private void Parent_PointerMovedEvent(object o, PointerEventArgs args)
        {
            var position = args.GetPosition(_widget);

            CurrentPosition = new Vector2((float)position.X, (float)position.Y);
        }

        public void SetMousePressed(MouseButton button)
        {
            PressedButtons[(int)button] = true;
        }

        public void SetMouseReleased(MouseButton button)
        {
            PressedButtons[(int)button] = false;
        }

        public void SetPosition(double x, double y)
        {
            CurrentPosition = new Vector2((float)x, (float)y);
        }

        public bool IsButtonPressed(MouseButton button)
        {
            return PressedButtons[(int)button];
        }

        public Size GetClientSize()
        {
            return _size;
        }

        public string DriverName => "Avalonia";

        public event Action<string> OnGamepadConnected
        {
            add { }
            remove { }
        }

        public event Action<string> OnGamepadDisconnected
        {
            add { }
            remove { }
        }

        public ReadOnlySpan<string> GamepadsIds => new[] { "0" };

        public IGamepad GetGamepad(string id)
        {
            return new AvaloniaMouse(this);
        }

        public void Dispose()
        {
            if (_isDisposed)
            {
                return;
            }

            _isDisposed = true;

            _widget.PointerMoved -= Parent_PointerMovedEvent;
            _widget.PointerPressed -= Parent_PointerPressEvent;
            _widget.PointerReleased -= Parent_PointerReleaseEvent;
            _widget.PointerWheelChanged -= Parent_ScrollEvent;

            _window.PointerMoved -= Parent_PointerMovedEvent;
            _window.PointerPressed -= Parent_PointerPressEvent;
            _window.PointerReleased -= Parent_PointerReleaseEvent;
            _window.PointerWheelChanged -= Parent_ScrollEvent;

            _widget = null;
        }
    }
}