-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
2746c54
commit e7d298f
Showing
3 changed files
with
128 additions
and
173 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,118 @@ | ||
| namespace StrongTypeIdGenerator.Analyzer | ||
| { | ||
| using Microsoft.CodeAnalysis; | ||
| using Microsoft.CodeAnalysis.CSharp; | ||
| using Microsoft.CodeAnalysis.CSharp.Syntax; | ||
| using System; | ||
| using System.Collections.Immutable; | ||
|
|
||
| public abstract class BaseIdGenerator : IIncrementalGenerator | ||
| { | ||
| public void Initialize(IncrementalGeneratorInitializationContext context) | ||
| { | ||
| IncrementalValuesProvider<ClassDeclarationSyntax> classDeclarations = context.SyntaxProvider | ||
| .CreateSyntaxProvider( | ||
| predicate: static (s, _) => IsSyntaxTargetForGeneration(s), | ||
| transform: (ctx, _) => GetSemanticTargetForGeneration(ctx)) | ||
| .Where(static m => m is not null)!; | ||
|
|
||
| IncrementalValueProvider<(Compilation, ImmutableArray<ClassDeclarationSyntax>)> compilationAndClasses = | ||
| context.CompilationProvider.Combine(classDeclarations.Collect()); | ||
|
|
||
| context.RegisterSourceOutput(compilationAndClasses, | ||
| (spc, source) => Execute(source.Item1, source.Item2, spc)); | ||
| } | ||
|
|
||
| protected static bool IsSyntaxTargetForGeneration(SyntaxNode node) => | ||
| node is ClassDeclarationSyntax classDeclaration && | ||
| classDeclaration.AttributeLists.Count > 0; | ||
|
|
||
| private ClassDeclarationSyntax? GetSemanticTargetForGeneration(GeneratorSyntaxContext context) | ||
| { | ||
| var classDeclarationSyntax = (ClassDeclarationSyntax)context.Node; | ||
|
|
||
| foreach (var attributeListSyntax in classDeclarationSyntax.AttributeLists) | ||
| { | ||
| foreach (var attributeSyntax in attributeListSyntax.Attributes) | ||
| { | ||
| if (context.SemanticModel.GetSymbolInfo(attributeSyntax).Symbol is IMethodSymbol attributeSymbol) | ||
| { | ||
| if (attributeSymbol.ContainingType.ToDisplayString() == MarkerAttributeFullName) | ||
| { | ||
| return classDeclarationSyntax; | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| return null; | ||
| } | ||
|
|
||
| protected abstract string MarkerAttributeFullName { get; } | ||
|
|
||
| protected abstract INamedTypeSymbol GetIdTypeSymbol(Compilation compilation); | ||
|
|
||
| protected abstract void Execute(Compilation compilation, ImmutableArray<ClassDeclarationSyntax> classes, SourceProductionContext context); | ||
|
|
||
| protected static string? GetNamespace(ClassDeclarationSyntax classDeclaration) | ||
| { | ||
| if (classDeclaration is null) | ||
| { | ||
| throw new ArgumentNullException(nameof(classDeclaration)); | ||
| } | ||
|
|
||
| // Walk upwards until we find the namespace declaration | ||
| SyntaxNode? potentialNamespaceParent = classDeclaration.Parent; | ||
| while (potentialNamespaceParent != null && | ||
| potentialNamespaceParent is not NamespaceDeclarationSyntax && | ||
| potentialNamespaceParent is not FileScopedNamespaceDeclarationSyntax) | ||
| { | ||
| potentialNamespaceParent = potentialNamespaceParent.Parent; | ||
| } | ||
|
|
||
| // Return the namespace name if it was found, otherwise null | ||
| if (potentialNamespaceParent is BaseNamespaceDeclarationSyntax namespaceDeclaration) | ||
| { | ||
| return namespaceDeclaration.Name.ToString(); | ||
| } | ||
|
|
||
| return null; | ||
| } | ||
|
|
||
| protected bool HasCheckValueMethod(Compilation compilation, ClassDeclarationSyntax classDeclaration) | ||
| { | ||
| if (compilation is null) | ||
| { | ||
| throw new ArgumentNullException(nameof(compilation)); | ||
| } | ||
|
|
||
| if (classDeclaration is null) | ||
| { | ||
| throw new ArgumentNullException(nameof(classDeclaration)); | ||
| } | ||
|
|
||
| var semanticModel = compilation.GetSemanticModel(classDeclaration.SyntaxTree); | ||
| var classSymbol = semanticModel.GetDeclaredSymbol(classDeclaration) as INamedTypeSymbol; | ||
| if (classSymbol is null) | ||
| { | ||
| return false; | ||
| } | ||
|
|
||
| var idTypeSymbol = GetIdTypeSymbol(compilation); | ||
|
|
||
| foreach (var member in classSymbol.GetMembers("CheckValue")) | ||
| { | ||
| if (member is IMethodSymbol methodSymbol && | ||
| methodSymbol.IsStatic && | ||
| methodSymbol.DeclaredAccessibility == Accessibility.Private && | ||
| methodSymbol.Parameters.Length == 1 && | ||
| SymbolEqualityComparer.Default.Equals(methodSymbol.Parameters[0].Type, idTypeSymbol)) | ||
| { | ||
| return true; | ||
| } | ||
| } | ||
|
|
||
| return false; | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters