using Avalonia;
using Avalonia.Interactivity;
using Avalonia.LogicalTree;
using Toolkit.UI.Controls.Avalonia;

namespace Toolkit.UI.Avalonia;

public class NavigationViewItemExtension
{
    public static readonly AttachedProperty<bool> IsItemClickEnabledProperty =
        AvaloniaProperty.RegisterAttached<NavigationViewItem, bool>("IsItemClickEnabled",
            typeof(NavigationViewItemExtension), false);

    public static readonly RoutedEvent<ItemInvokedEventArgs> ItemClickEvent = 
        RoutedEvent.Register<ItemInvokedEventArgs>("ItemClick", 
            RoutingStrategies.Bubble, typeof(NavigationViewItemExtension));

    static NavigationViewItemExtension()
    {
        IsItemClickEnabledProperty.Changed.AddClassHandler<NavigationViewItem>(OnIsItemClickEnabledPropertyChanged);
    }

    private static void OnIsItemClickEnabledPropertyChanged(NavigationViewItem sender, 
        AvaloniaPropertyChangedEventArgs args)
    {
        bool TrySetupNavigationView()
        {
            if (sender.GetLogicalAncestors().OfType<NavigationView>().FirstOrDefault() is NavigationView navigationView)
            {
                void OnItemInvoked(object? _, FluentAvalonia.UI.Controls.NavigationViewItemInvokedEventArgs args)
                {
                    if (args.InvokedItemContainer == sender)
                    {
                        sender.RaiseEvent(new ItemInvokedEventArgs { RoutedEvent = ItemClickEvent });
                    }
                }

                navigationView.ItemInvoked += OnItemInvoked;
                return true;
            }

            return false;
        }

        if (!TrySetupNavigationView())
        { 
            void OnAttachedToVisualTree(object? _, VisualTreeAttachmentEventArgs __)
            {
                sender.AttachedToVisualTree -= OnAttachedToVisualTree;
                TrySetupNavigationView();
            }

            sender.AttachedToVisualTree += OnAttachedToVisualTree;
        }
    }

    public static bool GetIsItemClickEnabled(NavigationViewItem element) =>
        element.GetValue(IsItemClickEnabledProperty);

    public static void SetIsItemClickEnabled(NavigationViewItem element, bool value) => 
        element.SetValue(IsItemClickEnabledProperty, value);

    public static void AddItemClickHandler(NavigationViewItem element, EventHandler<ItemInvokedEventArgs> handler) => 
        element.AddHandler(ItemClickEvent, handler);

    public static void RemoveItemClickHandler(NavigationViewItem element, EventHandler<ItemInvokedEventArgs> handler) =>
        element.RemoveHandler(ItemClickEvent, handler);
}