using System.Linq;
using Avalonia;
using Avalonia.Controls;
using Avalonia.Input;
using Avalonia.Interactivity;
using Avalonia.VisualTree;
using JetBrains.Annotations;

namespace StabilityMatrix.Avalonia.Controls.Extensions;

/// <summary>
/// Show tooltip on Controls with IsEffectivelyEnabled = false
/// https://github.com/AvaloniaUI/Avalonia/issues/3847#issuecomment-1618790059
/// </summary>
[PublicAPI]
public static class ShowDisabledTooltipExtension
{
    static ShowDisabledTooltipExtension()
    {
        ShowOnDisabledProperty.Changed.AddClassHandler<Control>(HandleShowOnDisabledChanged);
    }

    public static bool GetShowOnDisabled(AvaloniaObject obj)
    {
        return obj.GetValue(ShowOnDisabledProperty);
    }

    public static void SetShowOnDisabled(AvaloniaObject obj, bool value)
    {
        obj.SetValue(ShowOnDisabledProperty, value);
    }

    public static readonly AttachedProperty<bool> ShowOnDisabledProperty =
        AvaloniaProperty.RegisterAttached<object, Control, bool>("ShowOnDisabled");

    private static void HandleShowOnDisabledChanged(
        Control control,
        AvaloniaPropertyChangedEventArgs e
    )
    {
        if (e.NewValue is bool isEnabledVal && isEnabledVal)
        {
            control.DetachedFromVisualTree += AttachedControl_DetachedFromVisualOrExtension;
            control.AttachedToVisualTree += AttachedControl_AttachedToVisualTree;
            if (control.IsInitialized)
            {
                // enabled after visual attached
                AttachedControl_AttachedToVisualTree(control, null!);
            }
        }
        else
        {
            AttachedControl_DetachedFromVisualOrExtension(control, null!);
        }
    }

    private static void AttachedControl_AttachedToVisualTree(
        object? sender,
        VisualTreeAttachmentEventArgs e
    )
    {
        if (sender is not Control control || TopLevel.GetTopLevel(control) is not { } tl)
        {
            return;
        }
        // NOTE pointermove needed to be tunneled for me but you may not need to...
        tl.AddHandler(
            InputElement.PointerMovedEvent,
            TopLevel_PointerMoved,
            RoutingStrategies.Tunnel
        );
    }

    private static void AttachedControl_DetachedFromVisualOrExtension(
        object? s,
        VisualTreeAttachmentEventArgs e
    )
    {
        if (s is not Control control)
        {
            return;
        }
        control.DetachedFromVisualTree -= AttachedControl_DetachedFromVisualOrExtension;
        control.AttachedToVisualTree -= AttachedControl_AttachedToVisualTree;
        if (TopLevel.GetTopLevel(control) is not { } tl)
        {
            return;
        }
        tl.RemoveHandler(InputElement.PointerMovedEvent, TopLevel_PointerMoved);
    }

    private static void TopLevel_PointerMoved(object? sender, PointerEventArgs e)
    {
        if (sender is not Control tl)
        {
            return;
        }

        var attachedControls = tl.GetVisualDescendants()
            .Where(GetShowOnDisabled)
            .Cast<Control>()
            .ToList();

        // find disabled children under pointer w/ this extension enabled
        var disabledChildUnderPointer = attachedControls.FirstOrDefault(
            x =>
                x.Bounds.Contains(e.GetPosition(x.Parent as Visual))
                && x is { IsEffectivelyVisible: true, IsEffectivelyEnabled: false }
        );

        if (disabledChildUnderPointer != null)
        {
            // manually show tooltip
            ToolTip.SetIsOpen(disabledChildUnderPointer, true);
        }

        var disabledTooltipsToHide = attachedControls.Where(
            x => ToolTip.GetIsOpen(x) && x != disabledChildUnderPointer && !x.IsEffectivelyEnabled
        );

        foreach (var control in disabledTooltipsToHide)
        {
            ToolTip.SetIsOpen(control, false);
        }
    }
}