| | | 1 | | using System.Text; |
| | | 2 | | |
| | | 3 | | using Microsoft.CodeAnalysis; |
| | | 4 | | using Microsoft.CodeAnalysis.Text; |
| | | 5 | | |
| | | 6 | | using NexusLabs.Needlr.Generators.CodeGen; |
| | | 7 | | using NexusLabs.Needlr.Generators.Models; |
| | | 8 | | |
| | | 9 | | namespace NexusLabs.Needlr.Generators; |
| | | 10 | | |
| | | 11 | | /// <summary> |
| | | 12 | | /// Incremental source generator that emits one additional public forwarding |
| | | 13 | | /// constructor for each eligible positional record. |
| | | 14 | | /// </summary> |
| | | 15 | | [Generator(LanguageNames.CSharp)] |
| | | 16 | | public sealed class RecordConstructorOverloadGenerator : IIncrementalGenerator |
| | | 17 | | { |
| | | 18 | | public void Initialize(IncrementalGeneratorInitializationContext context) |
| | | 19 | | { |
| | 17 | 20 | | var candidateModels = context.SyntaxProvider.CreateSyntaxProvider( |
| | 17 | 21 | | predicate: static (node, _) => |
| | 701 | 22 | | RecordConstructorOverloadDiscoveryHelper |
| | 701 | 23 | | .IsCandidateRecordDeclaration(node), |
| | 17 | 24 | | transform: static (generatorContext, _) => |
| | 23 | 25 | | RecordConstructorOverloadDiscoveryHelper |
| | 23 | 26 | | .TryCreateCanonicalModel(generatorContext)) |
| | 17 | 27 | | .WithTrackingName( |
| | 17 | 28 | | RecordConstructorOverloadTrackingNames.Candidates); |
| | | 29 | | |
| | 17 | 30 | | var models = candidateModels |
| | 21 | 31 | | .Where(static model => model is not null) |
| | 16 | 32 | | .Select(static (model, _) => model!.Value) |
| | 17 | 33 | | .WithTrackingName(RecordConstructorOverloadTrackingNames.Models); |
| | | 34 | | |
| | 17 | 35 | | var assemblyName = context.CompilationProvider |
| | 17 | 36 | | .Select(static (compilation, _) => |
| | 19 | 37 | | compilation.AssemblyName ?? "Generated") |
| | 17 | 38 | | .WithTrackingName( |
| | 17 | 39 | | RecordConstructorOverloadTrackingNames.AssemblyName); |
| | | 40 | | |
| | 17 | 41 | | var breadcrumbLevel = context.AnalyzerConfigOptionsProvider |
| | 17 | 42 | | .Select(static (options, _) => |
| | 17 | 43 | | TypeRegistryGenerator.GetBreadcrumbLevel(options)) |
| | 17 | 44 | | .WithTrackingName( |
| | 17 | 45 | | RecordConstructorOverloadTrackingNames.BreadcrumbLevel); |
| | | 46 | | |
| | 17 | 47 | | var emitContext = assemblyName |
| | 17 | 48 | | .Combine(breadcrumbLevel) |
| | 17 | 49 | | .Select(static (pair, _) => |
| | 17 | 50 | | new RecordConstructorOverloadEmitContext( |
| | 17 | 51 | | pair.Left, |
| | 17 | 52 | | pair.Right)) |
| | 17 | 53 | | .WithTrackingName( |
| | 17 | 54 | | RecordConstructorOverloadTrackingNames.EmitContext); |
| | | 55 | | |
| | 17 | 56 | | var output = models |
| | 17 | 57 | | .Combine(emitContext) |
| | 17 | 58 | | .WithTrackingName( |
| | 17 | 59 | | RecordConstructorOverloadTrackingNames.Output); |
| | | 60 | | |
| | 17 | 61 | | context.RegisterSourceOutput(output, static (sourceContext, source) => |
| | 17 | 62 | | { |
| | 16 | 63 | | var (model, modelContext) = source; |
| | 16 | 64 | | var breadcrumbs = |
| | 16 | 65 | | new BreadcrumbWriter(modelContext.BreadcrumbLevel); |
| | 16 | 66 | | var generatedSource = |
| | 16 | 67 | | RecordConstructorOverloadCodeGenerator.GenerateSource( |
| | 16 | 68 | | model, |
| | 16 | 69 | | modelContext.AssemblyName, |
| | 16 | 70 | | breadcrumbs); |
| | 16 | 71 | | sourceContext.AddSource( |
| | 16 | 72 | | RecordConstructorOverloadCodeGenerator.BuildHintName(model), |
| | 16 | 73 | | SourceText.From(generatedSource, Encoding.UTF8)); |
| | 33 | 74 | | }); |
| | 17 | 75 | | } |
| | | 76 | | } |