using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using StellaOps.Infrastructure.Postgres.Migrations;
using StellaOps.Infrastructure.Postgres.Options;
using StellaOps.VexHub.Core;
using StellaOps.VexHub.Persistence.Postgres;
using StellaOps.VexHub.Persistence.Postgres.Repositories;

namespace StellaOps.VexHub.Persistence.Extensions;

/// <summary>
/// Service collection extensions for VexHub persistence.
/// </summary>
public static class VexHubPersistenceExtensions
{
    /// <summary>
    /// Adds VexHub PostgreSQL persistence services to the service collection.
    /// </summary>
    public static IServiceCollection AddVexHubPersistence(
        this IServiceCollection services,
        IConfiguration configuration)
    {
        services.Configure<PostgresOptions>(configuration.GetSection("Postgres"));

        services.AddSingleton<VexHubDataSource>();
        services.AddStartupMigrations(
            VexHubDataSource.DefaultSchemaName,
            "VexHub.Persistence",
            typeof(VexHubDataSource).Assembly);
        services.AddScoped<IVexSourceRepository, PostgresVexSourceRepository>();
        services.AddScoped<IVexConflictRepository, PostgresVexConflictRepository>();
        services.AddScoped<IVexIngestionJobRepository, PostgresVexIngestionJobRepository>();
        services.AddScoped<IVexStatementRepository, PostgresVexStatementRepository>();
        services.AddScoped<IVexProvenanceRepository, PostgresVexProvenanceRepository>();

        return services;
    }

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

        services.AddSingleton<VexHubDataSource>();
        services.AddStartupMigrations(
            VexHubDataSource.DefaultSchemaName,
            "VexHub.Persistence",
            typeof(VexHubDataSource).Assembly);
        services.AddScoped<IVexSourceRepository, PostgresVexSourceRepository>();
        services.AddScoped<IVexConflictRepository, PostgresVexConflictRepository>();
        services.AddScoped<IVexIngestionJobRepository, PostgresVexIngestionJobRepository>();
        services.AddScoped<IVexStatementRepository, PostgresVexStatementRepository>();
        services.AddScoped<IVexProvenanceRepository, PostgresVexProvenanceRepository>();

        return services;
    }
}