using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.Extensions.Configuration;

namespace RAIC.Extensions.Configuration.EntityFrameworkCore.Extensions;

public static class ConfigurationBuilderExtensions
{
    public delegate DbContextOptionsBuilder<T> DbContextOptionsTransformer<T>(DbContextOptionsBuilder<T> transform) where T : DbContext;

    /// <summary>
    /// Adds a <see cref="DbSet{}"/> off <typeparamref name="TDbContext"/> as a configuration provider to the <see cref="IConfigurationBuilder"/>.
    /// </summary>
    /// <typeparam name="TDbContext">Type of the <see cref="DbContext"/> which implements <see cref="ISettingsDbContext{,}"/></typeparam>
    /// <typeparam name="TSetting">Concrete type which implements <see cref="ISetting"/></typeparam>
    /// <param name="optionsTransformer">
    /// a <see cref="DbContextOptionsTransformer{}"/> which configures your <see cref="DbContextOptions{}"/>. eg.
    /// <code>
    ///   dbContextOptions => dbContextOptions.UseNpgsql(builder.Configuration.GetConnectionString("Default"))
    /// </code>
    /// </param>
    /// <returns>The <see cref="IConfigurationBuilder"/></returns>
    public static IConfigurationBuilder AddDbSet<TDbContext, TSetting>(this IConfigurationBuilder builder, DbContextOptionsTransformer<TDbContext> optionsTransformer)
        where TDbContext : DbContext, ISettingsDbContext<DbSet<TSetting>, TSetting>
        where TSetting : class, ISetting
    {
        // DEBT: Find way to create non-pooled DbContextFactory since this is only a short lived usage
        var configurationSource = new EntityFrameworkCoreDbSetConfigurationSource<TDbContext, TSetting>()
        {
            DbContextFactory = new PooledDbContextFactory<TDbContext>(optionsTransformer(new DbContextOptionsBuilder<TDbContext>()).Options, poolSize: 1)
        };

        return builder.Add(configurationSource);
    }
}