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

namespace StellaOps.Router.Transport.Udp;

/// <summary>
/// Extension methods for registering UDP transport services.
/// </summary>
public static class ServiceCollectionExtensions
{
    /// <summary>
    /// Adds UDP 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 AddUdpTransportServer(
        this IServiceCollection services,
        Action<UdpTransportOptions>? configure = null)
    {
        if (configure is not null)
        {
            services.Configure(configure);
        }

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

        return services;
    }

    /// <summary>
    /// Adds UDP 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 AddUdpTransportClient(
        this IServiceCollection services,
        Action<UdpTransportOptions>? configure = null)
    {
        if (configure is not null)
        {
            services.Configure(configure);
        }

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

        return services;
    }
}