SystemExtensions/SystemExtensions.NetStandard.DependencyInjection/Http/HttpClientBuilder.cs

using Slim;
using System.Extensions;
using System.Net.Http;
using System.Net.Http.Headers;

namespace System.DependencyInjection.Http;

public sealed class HttpClientBuilder<T>
{
    private readonly IServiceProducer serviceProducer;

    private Uri baseAddress;
    private Func<IServiceProvider, HttpMessageHandler> httpMessageHandlerFactory;
    private bool disposeMessageHandler;
    private Action<HttpRequestHeaders> defaultRequestHeadersSetup;
    private long maxResponseBufferSize = 2147483647L; //2GB default HttpClient value [https://docs.microsoft.com/en-us/dotnet/api/system.net.http.httpclient.maxresponsecontentbuffersize?view=net-6.0]
    private TimeSpan timeout = TimeSpan.FromSeconds(100); //100 seconds default HttpClient value [https://docs.microsoft.com/en-us/dotnet/api/system.net.http.httpclient.timeout?view=net-6.0]

    internal HttpClientBuilder(IServiceProducer serviceProducer)
    {
        serviceProducer.ThrowIfNull(nameof(serviceProducer));

        this.serviceProducer = serviceProducer;
    }

    public HttpClientBuilder<T> WithMessageHandler(Func<IServiceProvider, HttpMessageHandler> httpMessageHandlerFactory)
    {
        httpMessageHandlerFactory.ThrowIfNull(nameof(httpMessageHandlerFactory));

        this.httpMessageHandlerFactory = httpMessageHandlerFactory;
        return this;
    }

    public HttpClientBuilder<T> WithDisposeMessageHandler(bool disposeMessageHandler)
    {
        this.disposeMessageHandler = disposeMessageHandler;
        return this;
    }

    public HttpClientBuilder<T> WithBaseAddress(Uri baseAddress)
    {
        baseAddress.ThrowIfNull(nameof(baseAddress));

        this.baseAddress = baseAddress;
        return this;
    }

    public HttpClientBuilder<T> WithDefaultRequestHeadersSetup(Action<HttpRequestHeaders> setup)
    {
        setup.ThrowIfNull(nameof(setup));

        this.defaultRequestHeadersSetup = setup;
        return this;
    }

    public HttpClientBuilder<T> WithMaxResponseBufferSize(long responseBufferSize)
    {
        this.maxResponseBufferSize = responseBufferSize;
        return this;
    }

    public HttpClientBuilder<T> WithTimeout(TimeSpan timeout)
    {
        this.timeout = timeout;
        return this;
    }

    public IServiceProducer Build()
    {
        this.serviceProducer.RegisterScoped<IHttpClient<T>>(sp =>
        {
            var client = this.httpMessageHandlerFactory is not null ?
                new HttpClient<T>(this.httpMessageHandlerFactory(sp), this.disposeMessageHandler) :
                new HttpClient<T>();

            client.BaseAddress = this.baseAddress;
            this.defaultRequestHeadersSetup?.Invoke(client.DefaultRequestHeaders);
            client.MaxResponseContentBufferSize = this.maxResponseBufferSize;
            client.Timeout = this.timeout;
            return client;
        });

        return this.serviceProducer;
    }
}