// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

#nullable enable

using System.ComponentModel;
using System.Diagnostics.CodeAnalysis;
using System.Reflection;
using System.Runtime.ExceptionServices;
using System.Runtime.InteropServices;
using Windows.Win32;

namespace System.Windows.Forms.PropertyGridInternal.TestUtilities;

public class SubPropertyGrid<TSelected> : PropertyGrid where TSelected : new()
{
    private static MessageId WM_DELAYEDEXECUTION { get; } = PInvoke.RegisterWindowMessage("WinFormsSubPropertyGridDelayedExecution");

    internal PropertyGridView GridView => this.TestAccessor.Dynamic._gridView;

    [DisallowNull]
    internal GridEntry? SelectedEntry
    {
        get => GridView.SelectedGridEntry;
        set => SelectedGridItem = value;
    }

    internal GridEntry this[string propertyName]
    {
        get
        {
            string categoryName = SelectedObject!.GetType().GetProperty(propertyName)!
                .GetCustomAttribute<CategoryAttribute>()!.Category;
            return GetCurrentEntries()!
                .Single(entry => entry.PropertyName == categoryName)
                .Children
                .Single(entry => entry.PropertyName == propertyName);
        }
    }

    public SubPropertyGrid() => SelectedObject = new TSelected();

    public void PopupEditorAndClose(Action? onClosingAction = null)
    {
        Exception? exception = null;
        bool called = false;

        var callbackHandle = GCHandle.Alloc(() =>
        {
            try
            {
                onClosingAction?.Invoke();
            }
            catch (Exception ex)
            {
                exception = ex;
            }

            GridView.CloseDropDown();

            called = true;
        });

        try
        {
            PInvokeCore.PostMessage(this, WM_DELAYEDEXECUTION, lParam: GCHandle.ToIntPtr(callbackHandle));
            GridView.PopupEditor(GridView.TestAccessor.Dynamic._selectedRow);
        }
        finally
        {
            callbackHandle.Free();
        }

        Assert.True(called);

        if (exception is not null)
        {
            ExceptionDispatchInfo.Capture(exception).Throw();
        }
    }

    protected override void Dispose(bool disposing)
    {
        object? selectedObject = SelectedObject;

        base.Dispose(disposing);

        if (disposing && selectedObject is IDisposable disposable)
        {
            disposable.Dispose();
        }
    }

    protected override void WndProc(ref Message m)
    {
        if (m.MsgInternal != WM_DELAYEDEXECUTION)
        {
            base.WndProc(ref m);
            return;
        }

        ((Action)GCHandle.FromIntPtr(m.LParamInternal).Target!)();
    }

    protected override void OnGotFocus(EventArgs e)
    {
        base.OnGotFocus(e);

        GridView.Focus();
    }
}
