using Microsoft.Extensions.DependencyInjection;
using StellaOps.Router.Common.Abstractions;

namespace StellaOps.Router.Transport.Tcp;

/// <summary>
/// Extension methods for registering TCP transport services.
/// </summary>
public static class ServiceCollectionExtensions
{
    /// <summary>
    /// Adds TCP transport server services to the service collection.
    /// </summary>
    /// <param name="services">The service collection.</param>
    /// <param name="configure">Optional configuration action.</param>
    /// <returns>The service collection.</returns>
    public static IServiceCollection AddTcpTransportServer(
        this IServiceCollection services,
        Action<TcpTransportOptions>? configure = null)
    {
        if (configure is not null)
        {
            services.Configure(configure);
        }

        services.AddSingleton<TcpTransportServer>();
        services.AddSingleton<ITransportServer>(sp => sp.GetRequiredService<TcpTransportServer>());

        return services;
    }

    /// <summary>
    /// Adds TCP transport client services to the service collection.
    /// </summary>
    /// <param name="services">The service collection.</param>
    /// <param name="configure">Optional configuration action.</param>
    /// <returns>The service collection.</returns>
    public static IServiceCollection AddTcpTransportClient(
        this IServiceCollection services,
        Action<TcpTransportOptions>? configure = null)
    {
        if (configure is not null)
        {
            services.Configure(configure);
        }

        services.AddSingleton<TcpTransportClient>();
        services.AddSingleton<ITransportClient>(sp => sp.GetRequiredService<TcpTransportClient>());
        services.AddSingleton<IMicroserviceTransport>(sp => sp.GetRequiredService<TcpTransportClient>());

        return services;
    }
}