1 files changed, 43 insertions, 0 deletions
diff --git a/code/api/src/Utilities/SwaggerGenOptionsExtensions.cs b/code/api/src/Utilities/SwaggerGenOptionsExtensions.cs
new file mode 100644
index 0000000..a3d9036
--- /dev/null
+++ b/code/api/src/Utilities/SwaggerGenOptionsExtensions.cs
@@ -0,0 +1,43 @@
+#nullable enable
+using IOL.GreatOffice.Api.Endpoints;
+using Microsoft.AspNetCore.Mvc.ApiExplorer;
+using Microsoft.AspNetCore.Mvc.Controllers;
+using Swashbuckle.AspNetCore.SwaggerGen;
+
+namespace IOL.GreatOffice.Api.Utilities;
+
+public static class SwaggerGenOptionsExtensions
+{
+    /// <summary>
+    /// Updates Swagger document to support ApiEndpoints.<br/><br/>
+    /// For controllers inherited from <see cref="EndpointBase"/>:<br/>
+    /// - Replaces action Tag with <c>[namespace]</c><br/>
+    /// </summary>
+    public static void UseApiEndpoints(this SwaggerGenOptions options) {
+        options.TagActionsBy(EndpointNamespaceOrDefault);
+    }
+
+    private static IList<string?> EndpointNamespaceOrDefault(ApiDescription api) {
+        if (api.ActionDescriptor is not ControllerActionDescriptor actionDescriptor) {
+            throw new InvalidOperationException($"Unable to determine tag for endpoint: {api.ActionDescriptor.DisplayName}");
+        }
+
+        if (actionDescriptor.ControllerTypeInfo.GetBaseTypesAndThis().Any(t => t == typeof(EndpointBase))) {
+            return new[] {
+                actionDescriptor.ControllerTypeInfo.Namespace?.Split('.').Last()
+            };
+        }
+
+        return new[] {
+            actionDescriptor.ControllerName
+        };
+    }
+
+    private static IEnumerable<Type> GetBaseTypesAndThis(this Type type) {
+        var current = type;
+        while (current != null) {
+            yield return current;
+            current = current.BaseType;
+        }
+    }
+}
+\ No newline at end of file