// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System;
using System.Diagnostics.CodeAnalysis;
using System.Net;
using Microsoft.AspNetCore.Connections;
using Microsoft.AspNetCore.Http.Connections;
using Microsoft.AspNetCore.Http.Connections.Client;
using Microsoft.AspNetCore.Shared;
using Microsoft.AspNetCore.SignalR.Protocol;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Options;
 
namespace Microsoft.AspNetCore.SignalR.Client;
 
/// <summary>
/// Extension methods for <see cref="IHubConnectionBuilder"/>.
/// </summary>
public static class HubConnectionBuilderHttpExtensions
{
    /// <summary>
    /// Configures the <see cref="HttpConnectionOptions"/> to negotiate stateful reconnect with the server.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithStatefulReconnect(this IHubConnectionBuilder hubConnectionBuilder)
    {
        hubConnectionBuilder.Services.Configure<HttpConnectionOptions>(options => options.UseStatefulReconnect = true);
 
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, [StringSyntax(StringSyntaxAttribute.Uri)] string url)
    {
        hubConnectionBuilder.WithUrlCore(new Uri(url), null, null);
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <param name="configureHttpConnection">The delegate that configures the <see cref="HttpConnection"/>.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, [StringSyntax(StringSyntaxAttribute.Uri)] string url, Action<HttpConnectionOptions> configureHttpConnection)
    {
        hubConnectionBuilder.WithUrlCore(new Uri(url), null, configureHttpConnection);
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL and transports.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <param name="transports">A bitmask combining one or more <see cref="HttpTransportType"/> values that specify what transports the client should use.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, [StringSyntax(StringSyntaxAttribute.Uri)] string url, HttpTransportType transports)
    {
        hubConnectionBuilder.WithUrlCore(new Uri(url), transports, null);
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL and transports.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <param name="transports">A bitmask combining one or more <see cref="HttpTransportType"/> values that specify what transports the client should use.</param>
    /// <param name="configureHttpConnection">The delegate that configures the <see cref="HttpConnection"/>.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, [StringSyntax(StringSyntaxAttribute.Uri)] string url, HttpTransportType transports, Action<HttpConnectionOptions> configureHttpConnection)
    {
        hubConnectionBuilder.WithUrlCore(new Uri(url), transports, configureHttpConnection);
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, Uri url)
    {
        hubConnectionBuilder.WithUrlCore(url, null, null);
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <param name="configureHttpConnection">The delegate that configures the <see cref="HttpConnection"/>.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, Uri url, Action<HttpConnectionOptions> configureHttpConnection)
    {
        hubConnectionBuilder.WithUrlCore(url, null, configureHttpConnection);
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL and transports.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <param name="transports">A bitmask combining one or more <see cref="HttpTransportType"/> values that specify what transports the client should use.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, Uri url, HttpTransportType transports)
    {
        hubConnectionBuilder.WithUrlCore(url, transports, null);
        return hubConnectionBuilder;
    }
 
    /// <summary>
    /// Configures the <see cref="HubConnection" /> to use HTTP-based transports to connect to the specified URL and transports.
    /// </summary>
    /// <param name="hubConnectionBuilder">The <see cref="IHubConnectionBuilder" /> to configure.</param>
    /// <param name="url">The URL the <see cref="HttpConnection"/> will use.</param>
    /// <param name="transports">A bitmask combining one or more <see cref="HttpTransportType"/> values that specify what transports the client should use.</param>
    /// <param name="configureHttpConnection">The delegate that configures the <see cref="HttpConnection"/>.</param>
    /// <returns>The same instance of the <see cref="IHubConnectionBuilder"/> for chaining.</returns>
    public static IHubConnectionBuilder WithUrl(this IHubConnectionBuilder hubConnectionBuilder, Uri url, HttpTransportType transports, Action<HttpConnectionOptions> configureHttpConnection)
    {
        hubConnectionBuilder.WithUrlCore(url, transports, configureHttpConnection);
        return hubConnectionBuilder;
    }
 
    private static IHubConnectionBuilder WithUrlCore(this IHubConnectionBuilder hubConnectionBuilder, Uri url, HttpTransportType? transports, Action<HttpConnectionOptions>? configureHttpConnection)
    {
        ArgumentNullThrowHelper.ThrowIfNull(hubConnectionBuilder);
 
        hubConnectionBuilder.Services.Configure<HttpConnectionOptions>(o =>
        {
            o.Url = url;
            if (transports != null)
            {
                o.Transports = transports.Value;
            }
        });
 
        if (configureHttpConnection != null)
        {
            hubConnectionBuilder.Services.Configure(configureHttpConnection);
        }
 
        // Add HttpConnectionOptionsDerivedHttpEndPoint so HubConnection can read the Url from HttpConnectionOptions
        // without the Signal.Client.Core project taking a new dependency on Http.Connections.Client.
        hubConnectionBuilder.Services.AddSingleton<EndPoint, HttpConnectionOptionsDerivedHttpEndPoint>();
 
        // Configure the HttpConnection so that it uses the correct transfer format for the configured IHubProtocol.
        hubConnectionBuilder.Services.AddSingleton<IConfigureOptions<HttpConnectionOptions>, HubProtocolDerivedHttpOptionsConfigurer>();
 
        // If and when HttpConnectionFactory is made public, it can be moved out of this assembly and into Http.Connections.Client.
        hubConnectionBuilder.Services.AddSingleton<IConnectionFactory, HttpConnectionFactory>();
        return hubConnectionBuilder;
    }
 
    private sealed class HttpConnectionOptionsDerivedHttpEndPoint : UriEndPoint
    {
        public HttpConnectionOptionsDerivedHttpEndPoint(IOptions<HttpConnectionOptions> httpConnectionOptions)
            : base(httpConnectionOptions.Value.Url!)
        {
        }
    }
 
    private sealed class HubProtocolDerivedHttpOptionsConfigurer : IConfigureNamedOptions<HttpConnectionOptions>
    {
        private readonly TransferFormat _defaultTransferFormat;
 
        public HubProtocolDerivedHttpOptionsConfigurer(IHubProtocol hubProtocol)
        {
            _defaultTransferFormat = hubProtocol.TransferFormat;
        }
 
        public void Configure(string? name, HttpConnectionOptions options)
        {
            Configure(options);
        }
 
        public void Configure(HttpConnectionOptions options)
        {
            options.DefaultTransferFormat = _defaultTransferFormat;
        }
    }
}