Allowed to override the ClaimsToHeaders middleware

kesskalli · Dec 28, 2020 · 4fc2b4e · 4fc2b4e
1 parent 3ef6abd
commit 4fc2b4e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/Ocelot/Middleware/OcelotPipelineConfiguration.cs b/src/Ocelot/Middleware/OcelotPipelineConfiguration.cs
@@ -56,6 +56,14 @@ public class OcelotPipelineConfiguration
         /// </value>
         public Func<HttpContext, Func<Task>, Task> AuthorizationMiddleware { get; set; }
 
+        /// <summary>
+        /// This allows the user to completely override the ocelot claims to headers middleware
+        /// </summary>
+        /// <value>
+        /// <placeholder>This allows the user to completely override the ocelot claims to headers middleware</placeholder>
+        /// </value>
+        public Func<HttpContext, Func<Task>, Task> ClaimsToHeadersMiddleware { get; set; }
+
         /// <summary>
         /// This allows the user to implement there own query string manipulation logic
         /// </summary>

diff --git a/src/Ocelot/Middleware/OcelotPipelineExtensions.cs b/src/Ocelot/Middleware/OcelotPipelineExtensions.cs
@@ -121,8 +121,17 @@ public static class OcelotPipelineExtensions
                 app.Use(pipelineConfiguration.AuthorizationMiddleware);
             }
 
-            // Now we can run the claims to headers transformation middleware
-            app.UseClaimsToHeadersMiddleware();
+            // Now we can run the claims to headers transformation middleware.
+            // We allow the ocelot middleware to be overriden by whatever the
+            // user wants
+            if (pipelineConfiguration.ClaimsToHeadersMiddleware == null)
+            {
+                app.UseClaimsToHeadersMiddleware();
+            }
+            else
+            {
+                app.Use(pipelineConfiguration.ClaimsToHeadersMiddleware);
+            }
 
             // Allow the user to implement their own query string manipulation logic
             app.UseIfNotNull(pipelineConfiguration.PreQueryStringBuilderMiddleware);