Adjust useRoutes generation

This commit is contained in:
2024-09-23 18:05:22 +02:00
parent f797a613b3
commit 92752ff2c3
3 changed files with 7 additions and 5 deletions
@@ -74,9 +74,7 @@ public class MapEndpointGenerator : IIncrementalGenerator
var useRoutesMethodBuilder = SyntaxBuilder.CreateMethod(Constants.IEndpointRouteBuilderTypeName, Constants.UseRoutesMethodName)
.WithThisParameter(Constants.IEndpointRouteBuilderTypeName, Constants.BuilderParameterName)
.WithModifiers($"{Constants.Public} {Constants.Static}");
extensionClassBuilder.WithMethod(useRoutesMethodBuilder);
var useRoutesBody = new StringBuilder();
foreach (var classDeclarationSyntax in classes)
{
if (classDeclarationSyntax is null)
@@ -137,7 +135,8 @@ public class MapEndpointGenerator : IIncrementalGenerator
useRoutesBody.AppendLine("return builder;");
useRoutesMethodBuilder.WithBody(useRoutesBody.ToString());
foreach(var classUsing in routeUsings)
extensionClassBuilder.WithMethod(useRoutesMethodBuilder);
foreach (var classUsing in routeUsings)
{
builder.WithUsing(classUsing);
}