82 lines
4.1 KiB
C#
82 lines
4.1 KiB
C#
// Licensed to the .NET Foundation under one or more agreements.
|
|
// The .NET Foundation licenses this file to you under the MIT license.
|
|
// See the LICENSE file in the project root for more information.
|
|
|
|
using System.Collections.Immutable;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using CommunityToolkit.Mvvm.SourceGenerators.Extensions;
|
|
using CommunityToolkit.Mvvm.SourceGenerators.Input.Models;
|
|
using Microsoft.CodeAnalysis;
|
|
using Microsoft.CodeAnalysis.CSharp;
|
|
using Microsoft.CodeAnalysis.CSharp.Syntax;
|
|
|
|
namespace CommunityToolkit.Mvvm.SourceGenerators;
|
|
|
|
/// <summary>
|
|
/// A source generator for message registration without relying on compiled LINQ expressions.
|
|
/// </summary>
|
|
[Generator(LanguageNames.CSharp)]
|
|
public sealed partial class IMessengerRegisterAllGenerator : IIncrementalGenerator
|
|
{
|
|
/// <inheritdoc/>
|
|
public void Initialize(IncrementalGeneratorInitializationContext context)
|
|
{
|
|
// Get all class declarations. This pipeline step also needs to filter out duplicate recipient
|
|
// definitions (it might happen if a recipient has partial declarations). To do this, all pairs
|
|
// of class declarations and associated symbols are gathered, and then only the pair where the
|
|
// class declaration is the first syntax reference for the associated symbol is kept.
|
|
// Just like with the ObservableValidator generator, we also intentionally skip abstract types.
|
|
IncrementalValuesProvider<INamedTypeSymbol> typeSymbols =
|
|
context.SyntaxProvider
|
|
.CreateSyntaxProvider(
|
|
static (node, _) => node is ClassDeclarationSyntax,
|
|
static (context, _) => (context.Node, Symbol: (INamedTypeSymbol)context.SemanticModel.GetDeclaredSymbol(context.Node)!))
|
|
.Where(static item => item.Symbol is { IsAbstract: false, IsGenericType: false } && item.Node.IsFirstSyntaxDeclarationForSymbol(item.Symbol))
|
|
.Select(static (item, _) => item.Symbol);
|
|
|
|
// Get the target IRecipient<TMessage> interfaces and filter out other types
|
|
IncrementalValuesProvider<(INamedTypeSymbol Type, ImmutableArray<INamedTypeSymbol> Interfaces)> typeAndInterfaceSymbols =
|
|
typeSymbols
|
|
.Select(static (item, _) => (item, Interfaces: Execute.GetInterfaces(item)))
|
|
.Where(static item => !item.Interfaces.IsEmpty);
|
|
|
|
// Get the recipient info for all target types
|
|
IncrementalValuesProvider<RecipientInfo> recipientInfo =
|
|
typeAndInterfaceSymbols
|
|
.Select(static (item, _) => Execute.GetInfo(item.Type, item.Interfaces))
|
|
.WithComparer(RecipientInfo.Comparer.Default);
|
|
|
|
// Check whether the header file is needed
|
|
IncrementalValueProvider<bool> isHeaderFileNeeded =
|
|
recipientInfo
|
|
.Collect()
|
|
.Select(static (item, _) => item.Length > 0);
|
|
|
|
// Check whether [DynamicallyAccessedMembers] is available
|
|
IncrementalValueProvider<bool> isDynamicallyAccessedMembersAttributeAvailable =
|
|
context.CompilationProvider
|
|
.Select(static (item, _) => item.HasAccessibleTypeWithMetadataName("System.Diagnostics.CodeAnalysis.DynamicallyAccessedMembersAttribute"));
|
|
|
|
// Gather the conditional flag and attribute availability
|
|
IncrementalValueProvider<(bool IsHeaderFileNeeded, bool IsDynamicallyAccessedMembersAttributeAvailable)> headerFileInfo =
|
|
isHeaderFileNeeded.Combine(isDynamicallyAccessedMembersAttributeAvailable);
|
|
|
|
// Generate the header file with the attributes
|
|
context.RegisterConditionalImplementationSourceOutput(headerFileInfo, static (context, item) =>
|
|
{
|
|
CompilationUnitSyntax compilationUnit = Execute.GetSyntax(item);
|
|
|
|
context.AddSource("__IMessengerExtensions.g.cs", compilationUnit.GetText(Encoding.UTF8));
|
|
});
|
|
|
|
// Generate the class with all registration methods
|
|
context.RegisterImplementationSourceOutput(recipientInfo, static (context, item) =>
|
|
{
|
|
CompilationUnitSyntax compilationUnit = Execute.GetSyntax(item);
|
|
|
|
context.AddSource($"{item.FilenameHint}.g.cs", compilationUnit.GetText(Encoding.UTF8));
|
|
});
|
|
}
|
|
}
|