diff --git a/Net.Sdk.Web.Extensions.SourceGenerators/Constants.cs b/Net.Sdk.Web.Extensions.SourceGenerators/Constants.cs index fbb38b5..2ae0288 100644 --- a/Net.Sdk.Web.Extensions.SourceGenerators/Constants.cs +++ b/Net.Sdk.Web.Extensions.SourceGenerators/Constants.cs @@ -67,6 +67,7 @@ public static class Constants public const string MapAllRoutes = "MapAllRoutes"; public const string UseRoutesMethodName = "UseRoutes"; + public const string RegisterRoutesMethodName = "RegisterRoutes"; public const string EndpointRouteBuilderExtensionsName = "EndpointRouteBuilderExtensions"; public const string IEndpointRouteBuilderTypeName = "IEndpointRouteBuilder"; public const string WebApplicationBuilderExtensionsName = "WebApplicationBuilderExtension"; diff --git a/Net.Sdk.Web.Extensions.SourceGenerators/Net.Sdk.Web.Extensions.SourceGenerators.csproj b/Net.Sdk.Web.Extensions.SourceGenerators/Net.Sdk.Web.Extensions.SourceGenerators.csproj index 3c1a2cc..c03a4b6 100644 --- a/Net.Sdk.Web.Extensions.SourceGenerators/Net.Sdk.Web.Extensions.SourceGenerators.csproj +++ b/Net.Sdk.Web.Extensions.SourceGenerators/Net.Sdk.Web.Extensions.SourceGenerators.csproj @@ -2,7 +2,7 @@ netstandard2.0 - 0.8.20 + 0.8.21 latest true Alexandru Macocian diff --git a/Net.Sdk.Web.Extensions.SourceGenerators/UseRoutesGenerator.cs b/Net.Sdk.Web.Extensions.SourceGenerators/UseRoutesGenerator.cs index 3c9a61c..a4c9394 100644 --- a/Net.Sdk.Web.Extensions.SourceGenerators/UseRoutesGenerator.cs +++ b/Net.Sdk.Web.Extensions.SourceGenerators/UseRoutesGenerator.cs @@ -88,7 +88,12 @@ public class UseRoutesGenerator : IIncrementalGenerator .WithModifiers($"{Constants.Public} {Constants.Static}") .WithAttribute(SyntaxBuilder.CreateAttribute(Constants.MethodImplAttribute) .WithRawArgument(Constants.MethodImplArgument)); + var registerWebAppMethodBuilder = SyntaxBuilder.CreateMethod(Constants.WebApplicationTypeName, Constants.RegisterRoutesMethodName) + .WithModifiers($"{Constants.Public} {Constants.Static}") + .WithAttribute(SyntaxBuilder.CreateAttribute(Constants.MethodImplAttribute) + .WithRawArgument(Constants.MethodImplArgument)); webAppBuilder.WithMethod(useRoutesWebAppMethodBuilder); + webAppBuilder.WithMethod(registerWebAppMethodBuilder); var useRoutesWebAppBody = new StringBuilder(); foreach (var classToMethodMap in classToMethodMapping) @@ -183,6 +188,7 @@ public class UseRoutesGenerator : IIncrementalGenerator useRoutesWebAppBody.AppendLine("return builder;"); useRoutesWebAppMethodBuilder.WithBody(useRoutesWebAppBody.ToString()); + registerWebAppMethodBuilder.WithBody(useRoutesWebAppBody.ToString()); foreach (var classUsing in routeUsings) { builder.WithUsing(classUsing);