|
| 1 | +using Microsoft.CodeAnalysis; |
| 2 | + |
| 3 | +namespace AutoCtor; |
| 4 | + |
| 5 | +public sealed partial class AutoConstructSourceGenerator : IIncrementalGenerator |
| 6 | +{ |
| 7 | + public void Initialize(IncrementalGeneratorInitializationContext context) |
| 8 | + { |
| 9 | + var properties = context.AnalyzerConfigOptionsProvider |
| 10 | + .Select(static (p, ct) => |
| 11 | + { |
| 12 | + return p.GlobalOptions.TryGetValue("build_property.AutoCtorGuards", out var value) |
| 13 | + && (value.Equals("true", StringComparison.OrdinalIgnoreCase) |
| 14 | + || value.Equals("enable", StringComparison.OrdinalIgnoreCase)); |
| 15 | + }) |
| 16 | + .WithTrackingName(TrackingNames.BuildProperties); |
| 17 | + |
| 18 | + var types = context.SyntaxProvider.ForAttributeWithMetadataName( |
| 19 | + AttributeNames.AutoConstruct, |
| 20 | + GeneratorUtilities.IsTypeDeclarationWithAttributes, |
| 21 | + static (c, ct) => TypeModel.Create((INamedTypeSymbol)c.TargetSymbol)) |
| 22 | + .WithTrackingName(TrackingNames.TypeModels) |
| 23 | + .Collect(); |
| 24 | + |
| 25 | + var postCtorMethods = context.SyntaxProvider.ForAttributeWithMetadataName( |
| 26 | + AttributeNames.AutoPostConstruct, |
| 27 | + GeneratorUtilities.IsMethodDeclarationWithAttributes, |
| 28 | + static (c, ct) => PostCtorModel.Create((IMethodSymbol)c.TargetSymbol)) |
| 29 | + .WithTrackingName(TrackingNames.PostCtorMethods) |
| 30 | + .Collect(); |
| 31 | + |
| 32 | + context.RegisterSourceOutput( |
| 33 | + types.Combine(postCtorMethods).Combine(properties), |
| 34 | + Emitter.GenerateSource); |
| 35 | + } |
| 36 | +} |
0 commit comments