using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using StellaOps.Infrastructure.Postgres.Options;
using StellaOps.Scheduler.Models;
using StellaOps.Scheduler.Persistence.Postgres;
using StellaOps.Scheduler.Persistence.Postgres.Repositories;

namespace StellaOps.Scheduler.Persistence.Extensions;

/// <summary>
/// Extension methods for configuring Scheduler persistence services.
/// </summary>
public static class SchedulerPersistenceExtensions
{
    /// <summary>
    /// Adds Scheduler PostgreSQL persistence services using configuration section.
    /// </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 AddSchedulerPersistence(
        this IServiceCollection services,
        IConfiguration configuration,
        string sectionName = "Postgres:Scheduler")
    {
        services.Configure<PostgresOptions>(sectionName, configuration.GetSection(sectionName));
        services.AddSingleton<SchedulerDataSource>();

        // Register repositories
        services.AddScoped<IJobRepository, JobRepository>();
        services.AddScoped<ITriggerRepository, TriggerRepository>();
        services.AddScoped<IWorkerRepository, WorkerRepository>();
        services.AddScoped<IDistributedLockRepository, DistributedLockRepository>();
        services.AddScoped<IJobHistoryRepository, JobHistoryRepository>();
        services.AddScoped<IMetricsRepository, MetricsRepository>();
        services.AddScoped<IGraphJobRepository, GraphJobRepository>();
        services.AddScoped<IRunRepository, RunRepository>();
        services.AddScoped<IScheduleRepository, ScheduleRepository>();
        services.AddScoped<IImpactSnapshotRepository, ImpactSnapshotRepository>();
        services.AddScoped<IPolicyRunJobRepository, PolicyRunJobRepository>();
        services.AddScoped<IFailureSignatureRepository, FailureSignatureRepository>();
        services.AddSingleton<IRunSummaryService, RunSummaryService>();

        return services;
    }

    /// <summary>
    /// Adds Scheduler PostgreSQL persistence 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 AddSchedulerPersistence(
        this IServiceCollection services,
        Action<PostgresOptions> configureOptions)
    {
        services.Configure(configureOptions);
        services.AddSingleton<SchedulerDataSource>();

        // Register repositories
        services.AddScoped<IJobRepository, JobRepository>();
        services.AddScoped<ITriggerRepository, TriggerRepository>();
        services.AddScoped<IWorkerRepository, WorkerRepository>();
        services.AddScoped<IDistributedLockRepository, DistributedLockRepository>();
        services.AddScoped<IJobHistoryRepository, JobHistoryRepository>();
        services.AddScoped<IMetricsRepository, MetricsRepository>();
        services.AddScoped<IGraphJobRepository, GraphJobRepository>();
        services.AddScoped<IFailureSignatureRepository, FailureSignatureRepository>();

        return services;
    }
}