Add better support for customizing SwaggerUI

README.md

@@ -145,6 +145,16 @@ app.UseSwaggerForOcelotUI(opt => {
 })
   ```
 
+You can optionally customize SwaggerUI:
+```CSharp
+app.UseSwaggerForOcelotUI(opt => {
+    // swaggerForOcelot options
+}, uiOpt => {
+    //swaggerUI options
+    uiOpt.DocumentTitle = "Gateway documentation";
+})
+  ```
+
 6. Show your microservices interactive documentation.
 
    > `http://ocelotserviceurl/swagger`

src/MMLib.SwaggerForOcelot/Configuration/SwaggerForOcelotUIOptions.cs

@@ -10,7 +10,7 @@ namespace MMLib.SwaggerForOcelot.Configuration
     /// Configuration for Swagger UI.
     /// </summary>
     /// <seealso cref="Swashbuckle.AspNetCore.SwaggerUI.SwaggerUIOptions" />
-    public class SwaggerForOcelotUIOptions : SwaggerUIOptions
+    public class SwaggerForOcelotUIOptions
     {
         /// <summary>
         /// The relative path to gateway swagger generator.

src/MMLib.SwaggerForOcelot/Middleware/BuilderExtensions.cs

@@ -20,20 +20,22 @@ public static class BuilderExtensions
         /// </summary>
         /// <param name="app">The application builder.</param>
         /// <param name="setupAction">Setup <see cref="SwaggerForOcelotUIOptions"/></param>
+        /// <param name="setupUiAction">Setup SwaggerUI</param>
         /// <returns>
         /// <see cref="IApplicationBuilder"/>.
         /// </returns>
         public static IApplicationBuilder UseSwaggerForOcelotUI(
             this IApplicationBuilder app,
-            Action<SwaggerForOcelotUIOptions> setupAction = null)
+            Action<SwaggerForOcelotUIOptions> setupAction = null,
+            Action<SwaggerUIOptions> setupUiAction = null)
         {
             SwaggerForOcelotUIOptions options = app.ApplicationServices.GetService<IOptions<SwaggerForOcelotUIOptions>>().Value;
             setupAction?.Invoke(options);
             UseSwaggerForOcelot(app, options);
 
             app.UseSwaggerUI(c =>
             {
-                InitUIOption(c, options);
+                setupUiAction?.Invoke(c);
                 IReadOnlyList<SwaggerEndPointOptions> endPoints = app
                     .ApplicationServices.GetService<ISwaggerEndPointProvider>().GetAll();
 
@@ -95,15 +97,5 @@ static string GetDescription(SwaggerEndPointConfig config)
                 }
             }
         }
-
-        private static void InitUIOption(SwaggerUIOptions c, SwaggerForOcelotUIOptions options)
-        {
-            c.ConfigObject = options.ConfigObject;
-            c.DocumentTitle = options.DocumentTitle;
-            c.HeadContent = options.HeadContent;
-            c.IndexStream = options.IndexStream;
-            c.OAuthConfigObject = options.OAuthConfigObject;
-            c.RoutePrefix = options.RoutePrefix;
-        }
     }
 }