DocSpace-client/common/ASC.Api.Core/Core/CustomEndpointDataSource.cs

namespace ASC.Api.Core.Core;

public class CustomEndpointDataSource : EndpointDataSource
{
    public EndpointDataSource Source { get; }
    public override IReadOnlyList<Endpoint> Endpoints { get; }

    public CustomEndpointDataSource(EndpointDataSource source)
    {
        Source = source;
        var endpoints = Source.Endpoints.Cast<RouteEndpoint>();
        Endpoints = endpoints
            .SelectMany(r =>
            {
                var endpoints = new List<RouteEndpoint>();

                var attr = r.Metadata.OfType<CustomHttpMethodAttribute>().FirstOrDefault();
                var enableFormat = attr == null || !attr.DisableFormat;

                if (enableFormat)
                    endpoints.Add(new RouteEndpoint(r.RequestDelegate, RoutePatternFactory.Parse(r.RoutePattern.RawText + ".{format}"), r.Order, r.Metadata, r.DisplayName));
                else
                {
                    endpoints.Add(new RouteEndpoint(r.RequestDelegate, RoutePatternFactory.Parse(r.RoutePattern.RawText + ".json"), r.Order - 1, r.Metadata, r.DisplayName));
                    endpoints.Add(new RouteEndpoint(r.RequestDelegate, RoutePatternFactory.Parse(r.RoutePattern.RawText + ".xml"), r.Order - 1, r.Metadata, r.DisplayName));
                }

                return endpoints;
            }).ToList();
    }

    public override IChangeToken GetChangeToken() => Source.GetChangeToken();
}

public static class EndpointExtension
{
    public static IEndpointRouteBuilder MapCustom(this IEndpointRouteBuilder endpoints)
    {
        endpoints.DataSources.Add(new CustomEndpointDataSource(endpoints.DataSources.First()));

        return endpoints;
    }
}