Update RegisterConditionalImplementationSourceOutput extension

Author: Sergio0694

CommunityToolkit.Mvvm.SourceGenerators/ComponentModel/ObservableValidatorValidateAllPropertiesGenerator.cs

@@ -47,17 +47,18 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
             context.CompilationProvider
             .Select(static (item, _) => item.GetTypeByMetadataName("System.Diagnostics.CodeAnalysis.DynamicallyAccessedMembersAttribute") is { DeclaredAccessibility: Accessibility.Public });
 
+        // Gather the conditional flag and attribute availability
+        IncrementalValueProvider<(bool IsHeaderFileNeeded, bool IsDynamicallyAccessedMembersAttributeAvailable)> headerFileInfo =
+            isHeaderFileNeeded.Combine(isDynamicallyAccessedMembersAttributeAvailable);
+
         // Generate the header file with the attributes
-        context.RegisterImplementationSourceOutput(isHeaderFileNeeded.Combine(isDynamicallyAccessedMembersAttributeAvailable), static (context, item) =>
+        context.RegisterConditionalImplementationSourceOutput(headerFileInfo, static (context, item) =>
         {
-            if (item.Left)
-            {
-                CompilationUnitSyntax compilationUnit = Execute.GetSyntax(item.Right);
+            CompilationUnitSyntax compilationUnit = Execute.GetSyntax(item);
 
-                context.AddSource(
-                    hintName: "__ObservableValidatorExtensions.cs",
-                    sourceText: SourceText.From(compilationUnit.ToFullString(), Encoding.UTF8));
-            }
+            context.AddSource(
+                hintName: "__ObservableValidatorExtensions.cs",
+                sourceText: SourceText.From(compilationUnit.ToFullString(), Encoding.UTF8));
         });
 
         // Generate the class with all validation methods

CommunityToolkit.Mvvm.SourceGenerators/Extensions/IncrementalGeneratorInitializationContextExtensions.cs

@@ -65,22 +65,22 @@ public static void FilterWithLanguageVersion<T>(
     }
 
     /// <summary>
-    /// Conditionally invokes <see cref="IncrementalGeneratorInitializationContext.RegisterImplementationSourceOutput{TSource}(IncrementalValueProvider{TSource}, System.Action{SourceProductionContext, TSource})"/>
-    /// if the value produced by the input <see cref="IncrementalValueProvider{TValue}"/> is <see langword="true"/>.
+    /// Conditionally invokes <see cref="IncrementalGeneratorInitializationContext.RegisterImplementationSourceOutput{TSource}(IncrementalValueProvider{TSource}, Action{SourceProductionContext, TSource})"/>
+    /// if the value produced by the input <see cref="IncrementalValueProvider{TValue}"/> is <see langword="true"/>, and also supplying a given input state.
     /// </summary>
     /// <param name="context">The input <see cref="IncrementalGeneratorInitializationContext"/> value being used.</param>
     /// <param name="source">The source <see cref="IncrementalValueProvider{TValues}"/> instance.</param>
     /// <param name="action">The conditional <see cref="Action{T}"/> to invoke.</param>
-    public static void RegisterConditionalImplementationSourceOutput(
+    public static void RegisterConditionalImplementationSourceOutput<T>(
         this IncrementalGeneratorInitializationContext context,
-        IncrementalValueProvider<bool> source,
-        Action<SourceProductionContext> action)
+        IncrementalValueProvider<(bool Condition, T State)> source,
+        Action<SourceProductionContext, T> action)
     {
-        context.RegisterImplementationSourceOutput(source, (context, flag) =>
+        context.RegisterImplementationSourceOutput(source, (context, item) =>
         {
-            if (flag)
+            if (item.Condition)
             {
-                action(context);
+                action(context, item.State);
             }
         });
     }

CommunityToolkit.Mvvm.SourceGenerators/Messaging/IMessengerRegisterAllGenerator.cs

@@ -62,17 +62,18 @@ item.Symbol.DeclaringSyntaxReferences[0] is SyntaxReference syntaxReference &&
             context.CompilationProvider
             .Select(static (item, _) => item.GetTypeByMetadataName("System.Diagnostics.CodeAnalysis.DynamicallyAccessedMembersAttribute") is { DeclaredAccessibility: Accessibility.Public });
 
+        // Gather the conditional flag and attribute availability
+        IncrementalValueProvider<(bool IsHeaderFileNeeded, bool IsDynamicallyAccessedMembersAttributeAvailable)> headerFileInfo =
+            isHeaderFileNeeded.Combine(isDynamicallyAccessedMembersAttributeAvailable);
+
         // Generate the header file with the attributes
-        context.RegisterImplementationSourceOutput(isHeaderFileNeeded.Combine(isDynamicallyAccessedMembersAttributeAvailable), static (context, item) =>
+        context.RegisterConditionalImplementationSourceOutput(headerFileInfo, static (context, item) =>
         {
-            if (item.Left)
-            {
-                CompilationUnitSyntax compilationUnit = Execute.GetSyntax(item.Right);
+            CompilationUnitSyntax compilationUnit = Execute.GetSyntax(item);
 
-                context.AddSource(
-                    hintName: "__IMessengerExtensions.cs",
-                    sourceText: SourceText.From(compilationUnit.ToFullString(), Encoding.UTF8));
-            }
+            context.AddSource(
+                hintName: "__IMessengerExtensions.cs",
+                sourceText: SourceText.From(compilationUnit.ToFullString(), Encoding.UTF8));
         });
 
         // Generate the class with all registration methods