mitrage to IIncrementalGenerator

Uses IIncrementalGenerator (Source Generator V2).
Now the generator can run in VS 17.0/.NET 6.0 or later.

Author: ufcpp

samples/StringLiteralCodeAnalysisSample/Program.cs

@@ -93,7 +93,7 @@ private static Compilation Compile(string source)
             options: copt);
 
         // apply the source generator
-        var driver = CSharpGeneratorDriver.Create(new[] { new Utf8StringLiteralGenerator() }, parseOptions: opt);
+        var driver = CSharpGeneratorDriver.Create(new Utf8StringLiteralGenerator());
         driver.RunGeneratorsAndUpdateCompilation(compilation, out var resultCompilation, out _);
 
         return resultCompilation;

src/StringLiteralGenerator/Utf8StringLiteralGenerator.Emitter.cs

@@ -2,14 +2,15 @@
 using Microsoft.CodeAnalysis.Text;
 using System;
 using System.Collections.Generic;
+using System.Collections.Immutable;
 using System.Linq;
 using System.Text;
 
 namespace StringLiteralGenerator;
 
 public partial class Utf8StringLiteralGenerator
 {
-    private static void Emit(GeneratorExecutionContext context, IEnumerable<Utf8LiteralMethod> methods)
+    private static void Emit(SourceProductionContext context, ImmutableArray<Utf8LiteralMethod> methods)
     {
         var buffer = new StringBuilder();
 

src/StringLiteralGenerator/Utf8StringLiteralGenerator.Init.cs

@@ -19,7 +19,7 @@ public Utf8Attribute(string s) { }
 }
 ";
 
-    private static void AddAttribute(GeneratorPostInitializationContext context)
+    private static void AddAttribute(IncrementalGeneratorPostInitializationContext context)
     {
         context.AddSource("Utf8Attribute", SourceText.From(attributeText, Encoding.UTF8));
     }

src/StringLiteralGenerator/Utf8StringLiteralGenerator.cs

@@ -1,49 +1,23 @@
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.CSharp.Syntax;
-using System.Collections.Generic;
 
 namespace StringLiteralGenerator;
 
 [Generator]
-public partial class Utf8StringLiteralGenerator : ISourceGenerator
+public partial class Utf8StringLiteralGenerator : IIncrementalGenerator
 {
-    public void Execute(GeneratorExecutionContext context)
+    public void Initialize(IncrementalGeneratorInitializationContext context)
     {
-        if (context.SyntaxReceiver is not SyntaxReceiver receiver) return;
+        context.RegisterPostInitializationOutput(AddAttribute);
 
-        var compilation = context.Compilation;
+        var provider = context.SyntaxProvider
+            .CreateSyntaxProvider(
+                static (node, _) => IsSyntaxTargetForGeneration(node),
+                static (context, _) => GetSemanticTargetForGeneration(context.SemanticModel, (MethodDeclarationSyntax)context.Node)!
+                )
+            .Where(x => x is not null)
+            .Collect();
 
-        Emit(context, enumerate());
-
-        IEnumerable<Utf8LiteralMethod> enumerate()
-        {
-            foreach (var m in receiver.CandidateMethods)
-            {
-                var model = compilation.GetSemanticModel(m.SyntaxTree);
-
-                if (GetSemanticTargetForGeneration(model, m) is { } t)
-                    yield return t;
-            }
-        }
-    }
-
-    public void Initialize(GeneratorInitializationContext context)
-    {
-        context.RegisterForPostInitialization(AddAttribute);
-        context.RegisterForSyntaxNotifications(() => new SyntaxReceiver());
-    }
-
-    class SyntaxReceiver : ISyntaxReceiver
-    {
-        public List<MethodDeclarationSyntax> CandidateMethods { get; } = new List<MethodDeclarationSyntax>();
-
-        public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
-        {
-            // any field with at least one attribute is a candidate for property generation
-            if (IsSyntaxTargetForGeneration(syntaxNode))
-            {
-                CandidateMethods.Add((MethodDeclarationSyntax)syntaxNode);
-            }
-        }
+        context.RegisterSourceOutput(provider, Emit);
     }
 }