up

src/__Libraries/StellaOps.Router.Gateway/ApplicationBuilderExtensions.cs

@@ -0,0 +1,62 @@
+using StellaOps.Router.Gateway.Middleware;
+using StellaOps.Router.Gateway.OpenApi;
+
+namespace StellaOps.Router.Gateway;
+
+/// <summary>
+/// Extension methods for configuring the router gateway middleware pipeline.
+/// </summary>
+public static class ApplicationBuilderExtensions
+{
+    /// <summary>
+    /// Adds the router gateway middleware pipeline.
+    /// </summary>
+    /// <param name="app">The application builder.</param>
+    /// <returns>The application builder for chaining.</returns>
+    public static IApplicationBuilder UseRouterGateway(this IApplicationBuilder app)
+    {
+        // Enforce payload limits first
+        app.UseMiddleware<PayloadLimitsMiddleware>();
+
+        // Resolve endpoints from routing state
+        app.UseMiddleware<EndpointResolutionMiddleware>();
+
+        // Make routing decisions (select instance)
+        app.UseMiddleware<RoutingDecisionMiddleware>();
+
+        // Dispatch to transport and return response
+        app.UseMiddleware<TransportDispatchMiddleware>();
+
+        return app;
+    }
+
+    /// <summary>
+    /// Adds the router gateway middleware pipeline without payload limiting.
+    /// </summary>
+    /// <param name="app">The application builder.</param>
+    /// <returns>The application builder for chaining.</returns>
+    public static IApplicationBuilder UseRouterGatewayCore(this IApplicationBuilder app)
+    {
+        // Resolve endpoints from routing state
+        app.UseMiddleware<EndpointResolutionMiddleware>();
+
+        // Make routing decisions (select instance)
+        app.UseMiddleware<RoutingDecisionMiddleware>();
+
+        // Dispatch to transport and return response
+        app.UseMiddleware<TransportDispatchMiddleware>();
+
+        return app;
+    }
+
+    /// <summary>
+    /// Maps OpenAPI endpoints to the application.
+    /// Should be called before UseRouterGateway so OpenAPI requests are handled first.
+    /// </summary>
+    /// <param name="endpoints">The endpoint route builder.</param>
+    /// <returns>The endpoint route builder for chaining.</returns>
+    public static IEndpointRouteBuilder MapRouterOpenApi(this IEndpointRouteBuilder endpoints)
+    {
+        return endpoints.MapRouterOpenApiEndpoints();
+    }
+}