Steven He
5 years ago
9 changed files with 265 additions and 23 deletions
@ -0,0 +1,6 @@ |
|||||
|
<Project DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003"> |
||||
|
<ItemGroup> |
||||
|
<ProjectReference Include="$(MSBuildThisFileDirectory)/../src/Avalonia.SourceGenerator/Avalonia.SourceGenerator.csproj" OutputItemType="Analyzer" ReferenceOutputAssembly="false" /> |
||||
|
<Compile Include="$(MSBuildThisFileDirectory)/../src/Shared/SourceGeneratorAttributes.cs" /> |
||||
|
</ItemGroup> |
||||
|
</Project> |
||||
@ -0,0 +1,17 @@ |
|||||
|
<Project Sdk="Microsoft.NET.Sdk"> |
||||
|
|
||||
|
<PropertyGroup> |
||||
|
<TargetFramework>netstandard2.0</TargetFramework> |
||||
|
<Nullable>enable</Nullable> |
||||
|
</PropertyGroup> |
||||
|
|
||||
|
<ItemGroup> |
||||
|
<PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.3.3"> |
||||
|
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets> |
||||
|
<PrivateAssets>all</PrivateAssets> |
||||
|
</PackageReference> |
||||
|
<PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.1.0" /> |
||||
|
<Compile Include="..\Shared\SourceGeneratorAttributes.cs" /> |
||||
|
</ItemGroup> |
||||
|
|
||||
|
</Project> |
||||
@ -0,0 +1,167 @@ |
|||||
|
using System.Collections.Generic; |
||||
|
using System.Linq; |
||||
|
using Microsoft.CodeAnalysis; |
||||
|
using Microsoft.CodeAnalysis.CSharp; |
||||
|
using Microsoft.CodeAnalysis.CSharp.Syntax; |
||||
|
|
||||
|
namespace Avalonia.SourceGenerator |
||||
|
{ |
||||
|
internal class GenerateSubtypesSyntaxReceiver : ISyntaxReceiver |
||||
|
{ |
||||
|
public List<(MethodDeclarationSyntax, AttributeSyntax)> CandidateMethods { get; } = new(); |
||||
|
public List<SyntaxNode> Types { get; } = new(); |
||||
|
|
||||
|
public void OnVisitSyntaxNode(SyntaxNode syntaxNode) |
||||
|
{ |
||||
|
if (syntaxNode is MethodDeclarationSyntax declarationSyntax) |
||||
|
{ |
||||
|
foreach (var attribute in declarationSyntax.AttributeLists.SelectMany(i => i.Attributes)) |
||||
|
{ |
||||
|
CandidateMethods.Add((declarationSyntax, attribute)); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
if (syntaxNode is ClassDeclarationSyntax or StructDeclarationSyntax) |
||||
|
{ |
||||
|
Types.Add(syntaxNode); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
[Generator] |
||||
|
internal class SubtypesFactoryGenerator : ISourceGenerator |
||||
|
{ |
||||
|
private readonly GenerateSubtypesSyntaxReceiver _receiver = new(); |
||||
|
private static readonly string s_attributeName = typeof(SubtypesFactoryAttribute).FullName; |
||||
|
|
||||
|
public void Execute(GeneratorExecutionContext context) |
||||
|
{ |
||||
|
var methods = new List<(IMethodSymbol, ITypeSymbol, string)>(); |
||||
|
|
||||
|
foreach (var (method, attribute) in _receiver.CandidateMethods) |
||||
|
{ |
||||
|
var semanticModel = context.Compilation.GetSemanticModel(method.SyntaxTree); |
||||
|
var attributeTypeInfo = semanticModel.GetTypeInfo(attribute); |
||||
|
if (attributeTypeInfo.Type is null || |
||||
|
attributeTypeInfo.Type.ToString() != s_attributeName || |
||||
|
attribute.ArgumentList is null) |
||||
|
{ |
||||
|
continue; |
||||
|
} |
||||
|
|
||||
|
var arguments = attribute.ArgumentList.Arguments; |
||||
|
if (arguments.Count != 2) |
||||
|
{ |
||||
|
continue; |
||||
|
} |
||||
|
|
||||
|
if (arguments[0].Expression is not TypeOfExpressionSyntax typeOfExpr || |
||||
|
arguments[1].Expression is not LiteralExpressionSyntax and not IdentifierNameSyntax) |
||||
|
{ |
||||
|
continue; |
||||
|
} |
||||
|
|
||||
|
var type = semanticModel.GetTypeInfo(typeOfExpr.Type); |
||||
|
var ns = semanticModel.GetConstantValue(arguments[1].Expression); |
||||
|
var methodDeclInfo = semanticModel.GetDeclaredSymbol(method); |
||||
|
|
||||
|
if (type.Type is not ITypeSymbol baseType || |
||||
|
ns.HasValue is false || |
||||
|
ns.Value is not string nsValue || |
||||
|
methodDeclInfo is not IMethodSymbol methodSymbol || |
||||
|
methodSymbol.Parameters.Length != 2 || |
||||
|
methodSymbol.Parameters[1].RefKind != RefKind.Out) |
||||
|
{ |
||||
|
continue; |
||||
|
} |
||||
|
|
||||
|
methods.Add((methodSymbol, baseType, nsValue)); |
||||
|
} |
||||
|
|
||||
|
var types = new List<ITypeSymbol>(); |
||||
|
foreach (var type in _receiver.Types) |
||||
|
{ |
||||
|
var semanticModel = context.Compilation.GetSemanticModel(type.SyntaxTree); |
||||
|
var decl = semanticModel.GetDeclaredSymbol(type); |
||||
|
if (decl is ITypeSymbol typeSymbol) |
||||
|
{ |
||||
|
types.Add(typeSymbol); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
GenerateSubTypes(context, methods, types); |
||||
|
} |
||||
|
|
||||
|
private bool IsSubtypeOf(ITypeSymbol type, ITypeSymbol baseType) |
||||
|
{ |
||||
|
if (type.BaseType is null) |
||||
|
{ |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
if (SymbolEqualityComparer.Default.Equals(type.BaseType, baseType)) |
||||
|
{ |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
return IsSubtypeOf(type.BaseType, baseType); |
||||
|
} |
||||
|
|
||||
|
private void GenerateSubTypes( |
||||
|
GeneratorExecutionContext context, |
||||
|
List<(IMethodSymbol Method, ITypeSymbol BaseType, string Namespace)> methods, |
||||
|
List<ITypeSymbol> types) |
||||
|
{ |
||||
|
foreach (var (method, baseType, @namespace) in methods) |
||||
|
{ |
||||
|
var candidateTypes = types.Where(i => IsSubtypeOf(i, baseType)).Where(i => $"{i.ContainingNamespace}.".StartsWith($"{@namespace}.")).ToArray(); |
||||
|
var type = method.ContainingType; |
||||
|
var isGeneric = type.TypeParameters.Length > 0; |
||||
|
var isClass = type.TypeKind == TypeKind.Class; |
||||
|
|
||||
|
if (method.DeclaringSyntaxReferences.FirstOrDefault()?.GetSyntax() is not MethodDeclarationSyntax methodDecl) |
||||
|
{ |
||||
|
continue; |
||||
|
} |
||||
|
|
||||
|
var parameters = new SeparatedSyntaxList<ParameterSyntax>().AddRange(methodDecl.ParameterList.Parameters.Select(i => i.WithAttributeLists(new SyntaxList<AttributeListSyntax>()))); |
||||
|
|
||||
|
var methodDeclText = methodDecl |
||||
|
.WithAttributeLists(new SyntaxList<AttributeListSyntax>()) |
||||
|
.WithParameterList(methodDecl.ParameterList.WithParameters(parameters)) |
||||
|
.WithBody(null) |
||||
|
.WithSemicolonToken(SyntaxFactory.Token(SyntaxKind.None)) |
||||
|
.WithoutTrivia().ToString(); |
||||
|
|
||||
|
var typeDecl = $"partial {(isClass ? "class" : "struct")} {type.Name}{(isGeneric ? $"<{string.Join(", ", type.TypeParameters)}>" : "")}"; |
||||
|
var source = $@"using System;
|
||||
|
using System.Collections.Generic; |
||||
|
|
||||
|
namespace {method.ContainingNamespace} |
||||
|
{{ |
||||
|
{typeDecl} |
||||
|
{{ |
||||
|
{methodDeclText} |
||||
|
{{ |
||||
|
var hasMatch = false; |
||||
|
(hasMatch, {method.Parameters[1].Name}) = {method.Parameters[0].Name} switch |
||||
|
{{ |
||||
|
{string.Join("\n", candidateTypes.Select(i => $" \"{i.Name}\" => (true, ({method.Parameters[1].Type})new {i}()),"))} |
||||
|
_ => (false, default({method.Parameters[1].Type})) |
||||
|
}}; |
||||
|
|
||||
|
return hasMatch; |
||||
|
}} |
||||
|
}} |
||||
|
}}";
|
||||
|
|
||||
|
context.AddSource($"{type}.{method.MetadataName}.gen.cs", source); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public void Initialize(GeneratorInitializationContext context) |
||||
|
{ |
||||
|
context.RegisterForSyntaxNotifications(() => _receiver); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,17 @@ |
|||||
|
using System; |
||||
|
|
||||
|
namespace Avalonia.SourceGenerator |
||||
|
{ |
||||
|
[AttributeUsage(AttributeTargets.Method, Inherited = false, AllowMultiple = true)] |
||||
|
internal sealed class SubtypesFactoryAttribute : Attribute |
||||
|
{ |
||||
|
public SubtypesFactoryAttribute(Type baseType, string @namespace) |
||||
|
{ |
||||
|
BaseType = baseType; |
||||
|
Namespace = @namespace; |
||||
|
} |
||||
|
|
||||
|
public string Namespace { get; } |
||||
|
public Type BaseType { get; } |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue