IdentityShroud/IdentityShroud.Core/EFCore/Db.cs

using IdentityShroud.Core.Model;
using IdentityShroud.Core.Security;
using IdentityShroud.Core.Security.Keys;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;

namespace IdentityShroud.Core.EFCore;

public class DbConfiguration
{
    public string ConnectionString { get; set; } = "";
    public bool LogSensitiveData { get; set; } = false;
}

public class Db(
    IOptions<DbConfiguration> configuration, 
    ILoggerFactory? loggerFactory) 
    : DbContext
{
    public virtual DbSet<Client> Clients { get; set; }
    public virtual DbSet<Realm> Realms { get; set; }
    public virtual DbSet<RealmSigningKey> Keys { get; set; }
    public virtual DbSet<RealmDek> Deks { get; set; }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseNpgsql("<connection string>");
        optionsBuilder.UseNpgsql(
            configuration.Value.ConnectionString,
            o => o.MigrationsAssembly("IdentityShroud.Migrations")); // , o => o.UseNodaTime().UseVector().MigrationsAssembly("Migrations.KnowledgeBaseDB"));
        optionsBuilder.UseSnakeCaseNamingConvention();
        
        if (configuration.Value.LogSensitiveData)
            optionsBuilder.EnableSensitiveDataLogging();

        if (loggerFactory is { } )
        {
            optionsBuilder.UseLoggerFactory(loggerFactory);
        }
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.ApplyConfigurationsFromAssembly(typeof(Db).Assembly);
    }

    protected override void ConfigureConventions(ModelConfigurationBuilder b)
    {
        base.ConfigureConventions(b);
        
        b.Properties<DekId>().HaveConversion<DekIdConverter>();
        b.Properties<Dictionary<string, string>>().HaveConversion<DictionaryToJsonConverter<string, string>>();
        b.Properties<JwtSigAlgName>().HaveConversion<JwtSigAlgNameConverter>();
        b.Properties<KekId>().HaveConversion<KekIdConverter>();
        b.Properties<KeyType>().HaveConversion<KeyTypeConverter>();
        b.Properties<RealmSigningKeyId>().HaveConversion<RealmSigningKeyIdConverter>();
    }
}