Commit
Add source generator for repository-pattern code
- Loading branch information
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,83 @@ | ||
| using Microsoft.CodeAnalysis; | ||
| using Microsoft.CodeAnalysis.CSharp.Syntax; | ||
| using Microsoft.CodeAnalysis.Text; | ||
| using System.Collections.Immutable; | ||
| using System.Text; | ||
|
|
||
| namespace WeihanLi.EntityFramework.SourceGenerator; | ||
|
|
||
| [Generator] | ||
|
Check warning on line 9 in src/WeihanLi.EntityFramework.SourceGenerator/EFRepositorySourceGenerator.cs
|
||
| public sealed class EFRepositorySourceGenerator : IIncrementalGenerator | ||
|
Check warning on line 10 in src/WeihanLi.EntityFramework.SourceGenerator/EFRepositorySourceGenerator.cs
|
||
| { | ||
| public void Initialize(IncrementalGeneratorInitializationContext context) | ||
| { | ||
| var additionalFiles = context.AdditionalTextsProvider | ||
| .Where(file => file.Path.EndsWith(".template")) | ||
| .Select((file, cancellationToken) => file.GetText(cancellationToken)?.ToString()) | ||
| .Where(content => !string.IsNullOrEmpty(content)); | ||
|
|
||
| var dbContextDeclarations = context.SyntaxProvider | ||
| .CreateSyntaxProvider( | ||
| predicate: static (s, _) => IsDbContextDeclaration(s), | ||
| transform: static (ctx, _) => GetDbContextDeclaration(ctx)) | ||
| .Where(static m => m is not null); | ||
|
|
||
| var combined = dbContextDeclarations.Combine(additionalFiles.Collect()); | ||
|
|
||
| context.RegisterSourceOutput(combined, (spc, source) => | ||
| { | ||
| var (dbContextDeclaration, templates) = source; | ||
| var dbContextName = dbContextDeclaration!.Identifier.Text; | ||
| var repositoryNamespace = dbContextDeclaration.Parent!.GetNamespace(); | ||
| var generatedCode = GenerateRepositoryCode(dbContextName, repositoryNamespace, templates); | ||
|
Check warning on line 32 in src/WeihanLi.EntityFramework.SourceGenerator/EFRepositorySourceGenerator.cs
|
||
| spc.AddSource($"{dbContextName}Repository.g.cs", SourceText.From(generatedCode, Encoding.UTF8)); | ||
| }); | ||
| } | ||
|
|
||
| private static bool IsDbContextDeclaration(SyntaxNode node) | ||
| { | ||
| return node is ClassDeclarationSyntax classDeclaration && | ||
| classDeclaration.BaseList?.Types.Any(t => t.ToString().Contains("DbContext")) == true; | ||
| } | ||
|
|
||
| private static ClassDeclarationSyntax? GetDbContextDeclaration(GeneratorSyntaxContext context) | ||
| { | ||
| var classDeclaration = (ClassDeclarationSyntax)context.Node; | ||
| return classDeclaration; | ||
| } | ||
|
|
||
| private static string GenerateRepositoryCode(string dbContextName, string repositoryNamespace, ImmutableArray<string> templates) | ||
| { | ||
| var builder = new StringBuilder(); | ||
| builder.AppendLine("using WeihanLi.EntityFramework;"); | ||
| builder.AppendLine($"namespace {repositoryNamespace}"); | ||
| builder.AppendLine("{"); | ||
|
|
||
| foreach (var template in templates) | ||
| { | ||
| var code = template.Replace("{{DbContextName}}", dbContextName); | ||
| builder.AppendLine(code); | ||
| } | ||
|
|
||
| builder.AppendLine("}"); | ||
| return builder.ToString(); | ||
| } | ||
| } | ||
|
|
||
| internal static class SyntaxNodeExtensions | ||
| { | ||
| public static string GetNamespace(this SyntaxNode node) | ||
| { | ||
| while (node != null) | ||
| { | ||
| if (node is NamespaceDeclarationSyntax namespaceDeclaration) | ||
| { | ||
| return namespaceDeclaration.Name.ToString(); | ||
| } | ||
|
|
||
| node = node.Parent; | ||
|
Check warning on line 78 in src/WeihanLi.EntityFramework.SourceGenerator/EFRepositorySourceGenerator.cs
|
||
| } | ||
|
|
||
| return string.Empty; | ||
| } | ||
| } | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,23 @@ | ||
| <Project Sdk="Microsoft.NET.Sdk"> | ||
|
|
||
| <PropertyGroup> | ||
| <TargetFramework>net8.0</TargetFramework> | ||
| <LangVersion>latest</LangVersion> | ||
| <GeneratePackageOnBuild>true</GeneratePackageOnBuild> | ||
| <Title>WeihanLi.EntityFramework.SourceGenerator</Title> | ||
| <Description>Source generator for WeihanLi.EntityFramework</Description> | ||
| <Authors>WeihanLi</Authors> | ||
| <Company>WeihanLi</Company> | ||
| <Product>WeihanLi.EntityFramework</Product> | ||
| <PackageLicenseExpression>Apache-2.0</PackageLicenseExpression> | ||
| <RepositoryUrl>https://github.com/WeihanLi/WeihanLi.EntityFramework</RepositoryUrl> | ||
| <RepositoryType>git</RepositoryType> | ||
| <Copyright>Copyright 2019-2024 (c) WeihanLi</Copyright> | ||
| </PropertyGroup> | ||
|
|
||
| <ItemGroup> | ||
| <PackageReference Include="Microsoft.CodeAnalysis.CSharp" Version="4.12.0" PrivateAssets="all" /> | ||
| <PackageReference Include="Microsoft.CodeAnalysis.Analyzers" Version="3.11.0" PrivateAssets="all" /> | ||
| </ItemGroup> | ||
|
|
||
| </Project> |