using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using StellaOps.Concelier.Storage.Postgres.Repositories;
using StellaOps.Infrastructure.Postgres;
using StellaOps.Infrastructure.Postgres.Options;

namespace StellaOps.Concelier.Storage.Postgres;

/// <summary>
/// Extension methods for configuring Concelier PostgreSQL storage services.
/// </summary>
public static class ServiceCollectionExtensions
{
    /// <summary>
    /// Adds Concelier PostgreSQL storage services.
    /// </summary>
    /// <param name="services">Service collection.</param>
    /// <param name="configuration">Configuration root.</param>
    /// <param name="sectionName">Configuration section name for PostgreSQL options.</param>
    /// <returns>Service collection for chaining.</returns>
    public static IServiceCollection AddConcelierPostgresStorage(
        this IServiceCollection services,
        IConfiguration configuration,
        string sectionName = "Postgres:Concelier")
    {
        services.Configure<PostgresOptions>(sectionName, configuration.GetSection(sectionName));
        services.AddSingleton<ConcelierDataSource>();

        // Register repositories
        services.AddScoped<IAdvisoryRepository, AdvisoryRepository>();
        services.AddScoped<ISourceRepository, SourceRepository>();
        services.AddScoped<IAdvisoryAliasRepository, AdvisoryAliasRepository>();
        services.AddScoped<IAdvisoryCvssRepository, AdvisoryCvssRepository>();
        services.AddScoped<IAdvisoryAffectedRepository, AdvisoryAffectedRepository>();
        services.AddScoped<IAdvisoryReferenceRepository, AdvisoryReferenceRepository>();
        services.AddScoped<IAdvisoryCreditRepository, AdvisoryCreditRepository>();
        services.AddScoped<IAdvisoryWeaknessRepository, AdvisoryWeaknessRepository>();
        services.AddScoped<IKevFlagRepository, KevFlagRepository>();
        services.AddScoped<ISourceStateRepository, SourceStateRepository>();
        services.AddScoped<IFeedSnapshotRepository, FeedSnapshotRepository>();
        services.AddScoped<IAdvisorySnapshotRepository, AdvisorySnapshotRepository>();
        services.AddScoped<IMergeEventRepository, MergeEventRepository>();

        return services;
    }

    /// <summary>
    /// Adds Concelier PostgreSQL storage services with explicit options.
    /// </summary>
    /// <param name="services">Service collection.</param>
    /// <param name="configureOptions">Options configuration action.</param>
    /// <returns>Service collection for chaining.</returns>
    public static IServiceCollection AddConcelierPostgresStorage(
        this IServiceCollection services,
        Action<PostgresOptions> configureOptions)
    {
        services.Configure(configureOptions);
        services.AddSingleton<ConcelierDataSource>();

        // Register repositories
        services.AddScoped<IAdvisoryRepository, AdvisoryRepository>();
        services.AddScoped<ISourceRepository, SourceRepository>();
        services.AddScoped<IAdvisoryAliasRepository, AdvisoryAliasRepository>();
        services.AddScoped<IAdvisoryCvssRepository, AdvisoryCvssRepository>();
        services.AddScoped<IAdvisoryAffectedRepository, AdvisoryAffectedRepository>();
        services.AddScoped<IAdvisoryReferenceRepository, AdvisoryReferenceRepository>();
        services.AddScoped<IAdvisoryCreditRepository, AdvisoryCreditRepository>();
        services.AddScoped<IAdvisoryWeaknessRepository, AdvisoryWeaknessRepository>();
        services.AddScoped<IKevFlagRepository, KevFlagRepository>();
        services.AddScoped<ISourceStateRepository, SourceStateRepository>();
        services.AddScoped<IFeedSnapshotRepository, FeedSnapshotRepository>();
        services.AddScoped<IAdvisorySnapshotRepository, AdvisorySnapshotRepository>();
        services.AddScoped<IMergeEventRepository, MergeEventRepository>();

        return services;
    }
}