Updated to incremental source generator

src/EnumsEnhanced/EnumsEnhanced.cs

@@ -1,4 +1,5 @@
-using System.Globalization;
+using System.Collections.Immutable;
+using System.Globalization;
 using System.Runtime.CompilerServices;
 using System.Text;
 using Microsoft.CodeAnalysis;
@@ -7,7 +8,7 @@
 namespace EnumsEnhanced;
 
 [Generator]
-internal class EnumsEnhanced : ISourceGenerator
+internal class EnumsEnhanced : IIncrementalGenerator
 {
     public static readonly DiagnosticDescriptor UnderylingEnumerationTypeNotFound
       = new("EE001",
@@ -17,7 +18,43 @@ public static readonly DiagnosticDescriptor UnderylingEnumerationTypeNotFound
             DiagnosticSeverity.Error,
             true);
 
-    private void GenerateEnumMethods(GeneratorExecutionContext context, EnumDeclarationSyntax enumDeclarationSyntax, INamedTypeSymbol enumSymbol, ISymbol[] memberSymbols, StringBuilder methodSb)
+    /// <inheritdoc/>
+    public void Initialize(IncrementalGeneratorInitializationContext context)
+    {
+        var enumDeclarationsProvider = context.SyntaxProvider.CreateSyntaxProvider(
+            static (n, _) => n is EnumDeclarationSyntax,
+            static (n, _) => ((INamedTypeSymbol)n.SemanticModel.GetDeclaredSymbol(n.Node)!, (EnumDeclarationSyntax)n.Node)
+        );
+
+        context.RegisterSourceOutput(enumDeclarationsProvider.Collect(), static (sourceProductionContext, enumDeclarations) =>
+        {
+            foreach (var pair in enumDeclarations)
+            {
+                var symbol = pair.Item1;
+                var eds = pair.Item2;
+
+                if (symbol.ContainingType != null)
+                    continue;
+
+                var membersSymbols = symbol
+                    .GetMembers()
+                    .Where(x => x is IFieldSymbol)
+                    .Cast<IFieldSymbol>()
+                    .ToImmutableArray();
+
+                var sb = new StringBuilder();
+
+                GenerateEnumMethods(sourceProductionContext, eds, symbol, membersSymbols, sb);
+
+                string classCode = GetClassTemplate(eds, symbol, out string? className)
+                    .Replace("{CLASS_BODY}", sb.ToString());
+
+                sourceProductionContext.AddSource($"{className}.g.cs", classCode.Trim());
+            }
+        });
+    }
+
+    private static void GenerateEnumMethods(SourceProductionContext sourceProductionContext, EnumDeclarationSyntax enumDeclarationSyntax, INamedTypeSymbol enumSymbol, IList<IFieldSymbol> memberSymbols, StringBuilder methodSb)
     {
         var methodImplAttributeText = new StringBuilder();
         methodImplAttributeText.AppendLine("#if NETCOREAPP3_0_OR_GREATER");
@@ -32,7 +69,7 @@ private void GenerateEnumMethods(GeneratorExecutionContext context, EnumDeclarat
 
         if (enumUnderlyingType == null)
         {
-            context.ReportDiagnostic(
+            sourceProductionContext.ReportDiagnostic(
                 Diagnostic.Create(UnderylingEnumerationTypeNotFound,
                 enumDeclarationSyntax.Identifier.GetLocation(),
                 enumDeclarationSyntax.Identifier.ToString()));
@@ -149,7 +186,7 @@ string GetUnsafeHasFlagReturnStatement()
 
             var switchCasesValue = new StringBuilder();
 
-            foreach (var member in memberSymbols.OrderBy(x => x.Name.Length).Cast<IFieldSymbol>())
+            foreach (var member in memberSymbols.OrderBy(x => x.Name.Length))
             {
                 string memberRef = $"{enumSymbol.Name}.{member.Name}";
 
@@ -223,9 +260,9 @@ string GetUnsafeHasFlagReturnStatement()
 
             const string separator = $", ";
 
-            for (int i = 0; i < memberSymbols.Length; i++)
+            for (int i = 0; i < memberSymbols.Count; i++)
             {
-                var member = (IFieldSymbol)memberSymbols[i];
+                var member = memberSymbols[i];
                 string memberRef = $"{enumSymbol.Name}.{member.Name}";
 
                 if (member.HasConstantValue)
@@ -450,7 +487,7 @@ string GetUnsafeHasFlagReturnStatement()
         }
     }
 
-    private string GetClassTemplate(EnumDeclarationSyntax eds, ISymbol enumSymbol, out string className)
+    private static string GetClassTemplate(EnumDeclarationSyntax eds, ISymbol enumSymbol, out string className)
     {
         className = $"{enumSymbol.Name}Enhanced";
 
@@ -488,55 +525,6 @@ static string AccessibilityToAccessModifier(Accessibility accessibility)
             };
         }
     }
-
-    /// <inheritdoc/>
-    public void Execute(GeneratorExecutionContext context)
-    {
-        if (context.SyntaxReceiver is not ServiceNotifications receiver || receiver.DeclaredEnums.Count == 0)
-            return;
-
-        for (int i = 0; i < receiver.DeclaredEnums.Count; i++)
-        {
-            var eds = receiver.DeclaredEnums[i];
-
-            try
-            {
-                var semanticModel = context.Compilation.GetSemanticModel(eds.SyntaxTree);
-
-                if (semanticModel == null)
-                    continue;
-
-                if (semanticModel.GetDeclaredSymbol(eds) is not INamedTypeSymbol symbol)
-                    return;
-
-                if (symbol.ContainingType != null)
-                    continue;
-
-                var membersSymbols = new ISymbol[eds.Members.Count];
-
-                for (int j = 0; j < eds.Members.Count; j++)
-                    membersSymbols[j] = semanticModel.GetDeclaredSymbol(eds.Members[j])!;
-
-                var sb = new StringBuilder();
-
-                GenerateEnumMethods(context, eds, symbol, membersSymbols, sb);
-
-                string classCode = GetClassTemplate(eds, symbol, out string? className)
-                    .Replace("{CLASS_BODY}", sb.ToString());
-
-                context.AddSource($"{className}.g.cs", classCode.Trim());
-            }
-            catch
-            {
-            }
-        }
-    }
-
-    /// <inheritdoc/>
-    public void Initialize(GeneratorInitializationContext context)
-    {
-        context.RegisterForSyntaxNotifications(() => new ServiceNotifications());
-    }
 }
 
 internal class ServiceNotifications : ISyntaxReceiver

src/EnumsEnhanced/EnumsEnhanced.csproj

@@ -40,7 +40,7 @@
     </ItemGroup>
 
     <ItemGroup>
-        <PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.3.4">
+        <PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.11.0">
             <PrivateAssets>all</PrivateAssets>
             <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
         </PackageReference>

src/Tests/EnumsEnhancedTest/EnumsEnhancedTest.csproj

@@ -17,8 +17,8 @@
     </ItemGroup>
 
     <ItemGroup>
-        <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.11.0" />
-        <PackageReference Include="xunit" Version="2.9.0" />
+        <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.11.1" />
+        <PackageReference Include="xunit" Version="2.9.2" />
         <PackageReference Include="xunit.runner.visualstudio" Version="2.8.2">
             <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
             <PrivateAssets>all</PrivateAssets>