using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using StellaOps.Infrastructure.Postgres.Options;
using StellaOps.PacksRegistry.Core.Contracts;
using StellaOps.PacksRegistry.Persistence.Postgres;
using StellaOps.PacksRegistry.Persistence.Postgres.Repositories;

namespace StellaOps.PacksRegistry.Persistence.Extensions;

/// <summary>
/// Extension methods for configuring PacksRegistry persistence services.
/// </summary>
public static class PacksRegistryPersistenceExtensions
{
    /// <summary>
    /// Adds PacksRegistry PostgreSQL persistence services.
    /// </summary>
    public static IServiceCollection AddPacksRegistryPersistence(
        this IServiceCollection services,
        IConfiguration configuration,
        string sectionName = "Postgres:PacksRegistry")
    {
        services.Configure<PostgresOptions>(configuration.GetSection(sectionName));
        services.AddSingleton<PacksRegistryDataSource>();

        // Register repositories
        services.AddSingleton<IPackRepository, PostgresPackRepository>();
        services.AddSingleton<IAttestationRepository, PostgresAttestationRepository>();
        services.AddSingleton<IAuditRepository, PostgresAuditRepository>();
        services.AddSingleton<ILifecycleRepository, PostgresLifecycleRepository>();
        services.AddSingleton<IMirrorRepository, PostgresMirrorRepository>();
        services.AddSingleton<IParityRepository, PostgresParityRepository>();

        return services;
    }

    /// <summary>
    /// Adds PacksRegistry PostgreSQL persistence services with explicit options.
    /// </summary>
    public static IServiceCollection AddPacksRegistryPersistence(
        this IServiceCollection services,
        Action<PostgresOptions> configureOptions)
    {
        services.Configure(configureOptions);
        services.AddSingleton<PacksRegistryDataSource>();

        // Register repositories
        services.AddSingleton<IPackRepository, PostgresPackRepository>();
        services.AddSingleton<IAttestationRepository, PostgresAttestationRepository>();
        services.AddSingleton<IAuditRepository, PostgresAuditRepository>();
        services.AddSingleton<ILifecycleRepository, PostgresLifecycleRepository>();
        services.AddSingleton<IMirrorRepository, PostgresMirrorRepository>();
        services.AddSingleton<IParityRepository, PostgresParityRepository>();

        return services;
    }
}