HRCenter/Diligent.WebAPI.Data/DatabaseContext.cs

using Diligent.WebAPI.Data.Configurations;

namespace Diligent.WebAPI.Data;

public class DatabaseContext : IdentityDbContext<User, AppRole, int>
{
    public DbSet<InsuranceCompany> InsuranceCompanies { get; set; }
    public DbSet<Insurer> Insurers { get; set; }
    public DbSet<InsurancePolicy> InsurancePolicies { get; set; }
    public DbSet<WebhookSubscription> WebhookSubscriptions { get; set; }
    public DbSet<WebhookDefinition> WebhookDefinitions { get; set; }
    public DbSet<RefreshToken> RefreshTokens { get; set; }
    public DbSet<Ad> Ads { get; set; }
    public DbSet<Applicant> Applicants { get; set; }
    public DbSet<Technology> Technologies { get; set; }
    public DbSet<TechnologyApplicant> ApplicantTechnologies { get; set; }
    public DbSet<Comment> Comments { get; set; }
    public DbSet<SelectionLevel> SelectionLevels { get; set; }
    public DbSet<SelectionProcess> SelectionProcesses { get; set; }


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

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);
        modelBuilder.Entity<SelectionLevel>().HasData(
        new List<SelectionLevel>{
            new SelectionLevel{ Id = 1, Name = "HR intervju"},
            new SelectionLevel{ Id = 2, Name = "Screening test"},
            new SelectionLevel{ Id = 3, Name = "Tehnicki intervju"},
            new SelectionLevel{ Id = 4, Name = "Konacna odluka"},
        });
        modelBuilder.ApplyConfiguration(new ApplicantConfiguration());
        modelBuilder.ApplyConfiguration(new TechnologyConfiguration());
        modelBuilder.ApplyConfiguration(new CommentConfiguration());
        modelBuilder.ApplyConfiguration(new UserConfiguration());
        modelBuilder.ApplyConfiguration(new AdConfiguration());
    }
}