Terminal.Gui/Analyzers/Terminal.Gui.Analyzers.Internal.Tests/Generators/EnumExtensions/EnumExtensionMethodsIncrementalGeneratorTests.cs

using System.Collections.Concurrent;
using System.Collections.ObjectModel;
using System.Reflection;
using NUnit.Framework.Interfaces;
using NUnit.Framework.Internal;
using Terminal.Gui.Analyzers.Internal.Attributes;
using Terminal.Gui.Analyzers.Internal.Generators.EnumExtensions;
// ReSharper disable InconsistentNaming

namespace Terminal.Gui.Analyzers.Internal.Tests.Generators.EnumExtensions;

[TestFixture]
[Category ("Source Generators")]
[TestOf (typeof (EnumExtensionMethodsIncrementalGenerator))]
[Parallelizable (ParallelScope.Children)]
[SuppressMessage ("ReSharper", "ExceptionNotDocumented")]
public class EnumExtensionMethodsIncrementalGeneratorTests
{
    private static bool _isInitialized;

    /// <summary>All enum types declared in the test assembly.</summary>
    private static readonly ObservableCollection<Type> _allEnumTypes = [];

    /// <summary>
    ///     All enum types without a <see cref="GenerateEnumExtensionMethodsAttribute"/>, <see cref="_allEnumTypes"/>
    /// </summary>
    private static readonly HashSet<Type> _boringEnumTypes = [];

    /// <summary>All extension classes generated for enums with our attribute.</summary>
    private static readonly ObservableCollection<Type> _enumExtensionClasses = [];

    private static readonly ConcurrentDictionary<Type, EnumData> _extendedEnumTypeMappings = [];
    private static IEnumerable<Type> ExtendedEnumTypes => _extendedEnumTypeMappings.Keys;

    private static readonly ReaderWriterLockSlim _initializationLock = new ();

    private static IEnumerable<AssemblyExtendedEnumTypeAttribute> GetAssemblyExtendedEnumTypeAttributes () =>
        Assembly.GetExecutingAssembly ()
                .GetCustomAttributes<AssemblyExtendedEnumTypeAttribute> ();

    private static IEnumerable<TestCaseData> Get_AssemblyExtendedEnumTypeAttribute_EnumHasGeneratorAttribute_Cases ()
    {
        return GetAssemblyExtendedEnumTypeAttributes ()
            .Select (
                     static attr => new TestCaseData (attr)
                     {
                         TestName = $"{nameof (AssemblyExtendedEnumTypeAttribute_EnumHasGeneratorAttribute)}({attr.EnumType.Name},{attr.ExtensionClass.Name})",
                         HasExpectedResult = true,
                         ExpectedResult = true
                     });
    }

    [Test]
    [Category ("Attributes")]
    [TestCaseSource (nameof (Get_AssemblyExtendedEnumTypeAttribute_EnumHasGeneratorAttribute_Cases))]
    public bool AssemblyExtendedEnumTypeAttribute_EnumHasGeneratorAttribute (AssemblyExtendedEnumTypeAttribute attr)
    {
        Assume.That (attr, Is.Not.Null);
        Assume.That (attr.EnumType, Is.Not.Null);
        Assume.That (attr.EnumType.IsEnum);

        return attr.EnumType.IsDefined (typeof (GenerateEnumExtensionMethodsAttribute));
    }

    [Test]
    [Category("Attributes")]
    public void AssemblyExtendedEnumTypeAttribute_ExtensionClassHasExpectedReverseMappingAttribute ([ValueSource(nameof(GetAssemblyExtendedEnumTypeAttributes))]AssemblyExtendedEnumTypeAttribute attr)
    {
        Assume.That (attr, Is.Not.Null);
        Assume.That (attr.ExtensionClass, Is.Not.Null);
        Assume.That (attr.ExtensionClass.IsClass);
        Assume.That (attr.ExtensionClass.IsSealed);

        Assert.That (attr.ExtensionClass.IsDefined (typeof (ExtensionsForEnumTypeAttribute<>)));
    }

    [Test]
    [Category("Attributes")]
    public void ExtendedEnum_AssemblyHasMatchingAttribute ([ValueSource(nameof(GetExtendedEnum_EnumData))]EnumData enumData)
    {
        Assume.That (enumData, Is.Not.Null);
        Assume.That (enumData.EnumType, Is.Not.Null);
        Assume.That (enumData.EnumType.IsEnum);

        Assert.That (enumData.EnumType, Has.Attribute<GenerateEnumExtensionMethodsAttribute> ());
    }

    [Test]
    public void BoringEnum_DoesNotHaveExtensions ([ValueSource (nameof (_boringEnumTypes))] Type enumType)
    {
        Assume.That (enumType.IsEnum);

        Assert.That (enumType, Has.No.Attribute<GenerateEnumExtensionMethodsAttribute> ());
    }

    [Test]
    public void ExtendedEnum_FastIsDefinedFalse_DoesNotHaveFastIsDefined ([ValueSource (nameof (GetExtendedEnumTypes_FastIsDefinedFalse))] EnumData enumData)
    {
        Assume.That (enumData.EnumType.IsEnum);
        Assume.That (enumData.EnumType, Has.Attribute<GenerateEnumExtensionMethodsAttribute> ());
        Assume.That (enumData.GeneratorAttribute, Is.Not.Null);
        Assume.That (enumData.GeneratorAttribute, Is.EqualTo (enumData.EnumType.GetCustomAttribute<GenerateEnumExtensionMethodsAttribute> ()));
        Assume.That (enumData.GeneratorAttribute, Has.Property ("FastIsDefined").False);
        Assume.That (enumData.ExtensionClass, Is.Not.Null);

        Assert.That (enumData.ExtensionClass!.GetMethod ("FastIsDefined"), Is.Null);
    }

    [Test]
    public void ExtendedEnum_StaticExtensionClassExists ([ValueSource (nameof (ExtendedEnumTypes))] Type enumType)
    {
        Assume.That (enumType.IsEnum);
        Assume.That (enumType, Has.Attribute<GenerateEnumExtensionMethodsAttribute> ());
        Assume.That (enumType, Has.Attribute<GenerateEnumExtensionMethodsAttribute> ());
    }

    [Test]
    public void ExtendedEnum_FastIsDefinedTrue_HasFastIsDefined ([ValueSource (nameof (GetExtendedEnumTypes_FastIsDefinedTrue))] EnumData enumData)
    {
        Assume.That (enumData.EnumType, Is.Not.Null);
        Assume.That (enumData.EnumType.IsEnum);
        Assume.That (enumData.EnumType, Has.Attribute<GenerateEnumExtensionMethodsAttribute> ());
        Assume.That (enumData.ExtensionClass, Is.Not.Null);
        TypeWrapper extensionClassTypeInfo = new(enumData.ExtensionClass!);
        Assume.That (extensionClassTypeInfo.IsStaticClass);
        Assume.That (enumData.GeneratorAttribute, Is.Not.Null);
        Assume.That (enumData.GeneratorAttribute, Is.EqualTo (enumData.EnumType.GetCustomAttribute<GenerateEnumExtensionMethodsAttribute> ()));
        Assume.That (enumData.GeneratorAttribute, Has.Property ("FastIsDefined").True);

        MethodInfo? fastIsDefinedMethod = enumData.ExtensionClass!.GetMethod ("FastIsDefined");

        Assert.That (fastIsDefinedMethod, Is.Not.Null);
        Assert.That (fastIsDefinedMethod, Has.Attribute<ExtensionAttribute> ());
        extensionClassTypeInfo.GetMethodsWithAttribute<ExtensionAttribute> (false);


    }

    private static IEnumerable<EnumData> GetExtendedEnum_EnumData ()
    {
        _initializationLock.EnterUpgradeableReadLock ();

        try
        {
            if (!_isInitialized)
            {
                Initialize ();
            }

            return _extendedEnumTypeMappings.Values;
        }
        finally
        {
            _initializationLock.ExitUpgradeableReadLock ();
        }
    }

    private static IEnumerable<EnumData> GetExtendedEnumTypes_FastIsDefinedFalse ()
    {
        _initializationLock.EnterUpgradeableReadLock ();

        try
        {
            if (!_isInitialized)
            {
                Initialize ();
            }

            return _extendedEnumTypeMappings.Values.Where (static t => t.GeneratorAttribute?.FastIsDefined is false);
        }
        finally
        {
            _initializationLock.ExitUpgradeableReadLock ();
        }
    }

    private static IEnumerable<EnumData> GetExtendedEnumTypes_FastIsDefinedTrue ()
    {
        _initializationLock.EnterUpgradeableReadLock ();

        try
        {
            if (!_isInitialized)
            {
                Initialize ();
            }

            return _extendedEnumTypeMappings.Values.Where (static t => t.GeneratorAttribute?.FastIsDefined is true);
        }
        finally
        {
            _initializationLock.ExitUpgradeableReadLock ();
        }
    }

    private static void Initialize ()
    {
        if (!_initializationLock.IsUpgradeableReadLockHeld || !_initializationLock.TryEnterWriteLock (5000))
        {
            return;
        }

        try
        {
            if (_isInitialized)
            {
                return;
            }

            _allEnumTypes.CollectionChanged += AllEnumTypes_CollectionChanged;
            _enumExtensionClasses.CollectionChanged += EnumExtensionClasses_OnCollectionChanged;

            Type [] allAssemblyTypes = Assembly
                                       .GetExecutingAssembly ()
                                       .GetTypes ();

            foreach (Type type in allAssemblyTypes.Where (IsDefinedEnum))
            {
                _allEnumTypes.Add (type);
            }

            foreach (Type type in allAssemblyTypes.Where (HasExtensionForEnumTypeAttribute))
            {
                _enumExtensionClasses.Add (type);
            }

            _isInitialized = true;
        }
        finally
        {
            _initializationLock.ExitWriteLock ();
        }

        return;

        static bool IsDefinedEnum (Type t) { return t is { IsEnum: true, IsGenericType: false, IsConstructedGenericType: false, IsTypeDefinition: true }; }

        static void AllEnumTypes_CollectionChanged (object? sender, NotifyCollectionChangedEventArgs e)
        {
            if (e.Action is not NotifyCollectionChangedAction.Add and not NotifyCollectionChangedAction.Replace || e.NewItems is null)
            {
                return;
            }

            foreach (Type enumType in e.NewItems.OfType<Type> ())
            {
                if (enumType.GetCustomAttribute<GenerateEnumExtensionMethodsAttribute> () is not { } generatorAttribute)
                {
                    _boringEnumTypes.Add (enumType);

                    continue;
                }

                _extendedEnumTypeMappings.AddOrUpdate (
                                               enumType,
                                               CreateNewEnumData,
                                               UpdateGeneratorAttributeProperty,
                                               generatorAttribute);
            }
        }

        static EnumData CreateNewEnumData (Type tEnum, GenerateEnumExtensionMethodsAttribute attr) { return new (tEnum, attr); }

        static EnumData UpdateGeneratorAttributeProperty (Type tEnum, EnumData data, GenerateEnumExtensionMethodsAttribute attr)
        {
            data.GeneratorAttribute ??= attr;

            return data;
        }

        static void EnumExtensionClasses_OnCollectionChanged (object? sender, NotifyCollectionChangedEventArgs e)
        {
            if (e.Action != NotifyCollectionChangedAction.Add)
            {
                return;
            }

            foreach (Type extensionClassType in e.NewItems!.OfType<Type> ())
            {
                if (extensionClassType.GetCustomAttribute (typeof (ExtensionsForEnumTypeAttribute<>), false) is not IExtensionsForEnumTypeAttributes
                        {
                            EnumType.IsEnum: true
                        } extensionForAttribute)
                {
                    continue;
                }

                _extendedEnumTypeMappings [extensionForAttribute.EnumType].ExtensionClass ??= extensionClassType;
            }
        }
    }

    private static bool HasExtensionForEnumTypeAttribute (Type t) => t.IsClass && t.IsDefined (typeof (ExtensionsForEnumTypeAttribute<>));

    public sealed record EnumData (
        Type EnumType,
        GenerateEnumExtensionMethodsAttribute? GeneratorAttribute = null,
        Type? ExtensionClass = null,
        IExtensionsForEnumTypeAttributes? ExtensionForEnumTypeAttribute = null)
    {
        public Type? ExtensionClass { get; set; } = ExtensionClass;

        public IExtensionsForEnumTypeAttributes? ExtensionForEnumTypeAttribute { get; set; } = ExtensionForEnumTypeAttribute;
        public GenerateEnumExtensionMethodsAttribute? GeneratorAttribute { get; set; } = GeneratorAttribute;
    }
}