DocSpace-buildtools/common/ASC.Core.Common/EF/Context/TenantDbContext.cs

namespace ASC.Core.Common.EF.Context;

public class MySqlTenantDbContext : TenantDbContext { }
public class PostgreSqlTenantDbContext : TenantDbContext { }
public class TenantDbContext : BaseDbContext
{
    public DbSet<DbTenant> Tenants { get; set; }
    public DbSet<DbTenantVersion> TenantVersion { get; set; }
    public DbSet<DbTenantPartner> TenantPartner { get; set; }
    public DbSet<DbTenantForbiden> TenantForbiden { get; set; }
    public DbSet<TenantIpRestrictions> TenantIpRestrictions { get; set; }
    public DbSet<DbCoreSettings> CoreSettings { get; set; }

    public TenantDbContext() { }
    public TenantDbContext(DbContextOptions<TenantDbContext> options)
        : base(options)
    {
    }

    protected override Dictionary<Provider, Func<BaseDbContext>> ProviderContext
    {
        get
        {
            return new Dictionary<Provider, Func<BaseDbContext>>()
            {
                { Provider.MySql, () => new MySqlTenantDbContext() } ,
                { Provider.PostgreSql, () => new PostgreSqlTenantDbContext() } ,
            };
        }
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        ModelBuilderWrapper
            .From(modelBuilder, Provider)
            .AddDbTenant()
            .AddCoreSettings()
            .AddDbTenantForbiden()
            .AddTenantIpRestrictions()
            .AddDbTenantPartner()
            .AddDbTenantVersion();
    }
}

public static class TenantDbExtension
{
    public static DIHelper AddTenantDbContextService(this DIHelper services)
    {
        return services.AddDbContextManagerService<TenantDbContext>();
    }
}