Files
election-system/src/ElectionSystem.Api/Infrastructure/Persistence/ElectionDbContext.cs
T
2026-08-29 17:26:59 +08:00

108 lines
6.1 KiB
C#

using ElectionSystem.Api.Infrastructure.Auditing;
using ElectionSystem.Api.Modules.Identity.Domain;
using ElectionSystem.Api.Modules.Organization.Domain;
using ElectionSystem.Api.Modules.Voters.Domain;
using ElectionSystem.Api.Modules.Candidates.Domain;
using Microsoft.EntityFrameworkCore;
namespace ElectionSystem.Api.Infrastructure.Persistence;
public sealed class ElectionDbContext(DbContextOptions<ElectionDbContext> options) : DbContext(options)
{
public DbSet<ApplicationUser> Users => Set<ApplicationUser>();
public DbSet<OrganizationUnit> OrganizationUnits => Set<OrganizationUnit>();
public DbSet<AdministrativeRegion> AdministrativeRegions => Set<AdministrativeRegion>();
public DbSet<UserRegionalRoleAssignment> UserRegionalRoleAssignments => Set<UserRegionalRoleAssignment>();
public DbSet<CountingStation> CountingStations => Set<CountingStation>();
public DbSet<UserCountingStationAssignment> UserCountingStationAssignments => Set<UserCountingStationAssignment>();
public DbSet<VoterProfile> Voters => Set<VoterProfile>();
public DbSet<CandidateProfile> Candidates => Set<CandidateProfile>();
public DbSet<AuditEntry> AuditEntries => Set<AuditEntry>();
protected override void OnModelCreating(ModelBuilder modelBuilder)
{
modelBuilder.Entity<ApplicationUser>(entity =>
{
entity.ToTable("identity_users");
entity.Property(x => x.UserName).HasMaxLength(128).IsRequired();
entity.Property(x => x.DisplayName).HasMaxLength(128).IsRequired();
entity.Property(x => x.ExternalSubject).HasMaxLength(256);
entity.HasIndex(x => x.UserName).IsUnique();
entity.HasIndex(x => x.ExternalSubject).IsUnique();
});
modelBuilder.Entity<AdministrativeRegion>(entity =>
{
entity.ToTable("administrative_regions");
entity.Property(x => x.Code).HasMaxLength(32).IsRequired();
entity.Property(x => x.Name).HasMaxLength(200).IsRequired();
entity.Property(x => x.Source).HasMaxLength(128).IsRequired();
entity.Property(x => x.SourceVersion).HasMaxLength(64).IsRequired();
entity.HasIndex(x => new { x.Code, x.SourceVersion }).IsUnique();
entity.HasIndex(x => new { x.ParentId, x.Level, x.IsActive });
entity.HasOne<AdministrativeRegion>().WithMany().HasForeignKey(x => x.ParentId).OnDelete(DeleteBehavior.Restrict);
});
modelBuilder.Entity<UserRegionalRoleAssignment>(entity =>
{
entity.ToTable("identity_user_regional_roles");
entity.HasIndex(x => new { x.UserId, x.Role, x.RegionId }).IsUnique();
entity.HasOne<ApplicationUser>().WithMany().HasForeignKey(x => x.UserId).OnDelete(DeleteBehavior.Cascade);
entity.HasOne<AdministrativeRegion>().WithMany().HasForeignKey(x => x.RegionId).OnDelete(DeleteBehavior.Restrict);
});
modelBuilder.Entity<CountingStation>(entity =>
{
entity.ToTable("counting_stations");
entity.Property(x => x.Name).HasMaxLength(200).IsRequired();
entity.Property(x => x.Code).HasMaxLength(64).IsRequired();
entity.HasIndex(x => x.Code).IsUnique();
entity.HasIndex(x => x.RegionId);
entity.HasOne<AdministrativeRegion>().WithMany().HasForeignKey(x => x.RegionId).OnDelete(DeleteBehavior.Restrict);
});
modelBuilder.Entity<UserCountingStationAssignment>(entity =>
{
entity.ToTable("identity_user_counting_stations");
entity.HasIndex(x => new { x.UserId, x.CountingStationId }).IsUnique();
entity.HasOne<ApplicationUser>().WithMany().HasForeignKey(x => x.UserId).OnDelete(DeleteBehavior.Cascade);
entity.HasOne<CountingStation>().WithMany().HasForeignKey(x => x.CountingStationId).OnDelete(DeleteBehavior.Cascade);
});
modelBuilder.Entity<VoterProfile>(entity =>
{
entity.ToTable("voter_profiles");
entity.Property(x => x.VoterNumber).HasMaxLength(64).IsRequired();
entity.Property(x => x.DisplayName).HasMaxLength(128).IsRequired();
entity.Property(x => x.DisabledReason).HasMaxLength(500);
entity.HasIndex(x => x.VoterNumber).IsUnique();
entity.HasIndex(x => new { x.RegionId, x.Status, x.DisplayName });
entity.HasIndex(x => x.UserId).IsUnique();
entity.HasOne<AdministrativeRegion>().WithMany().HasForeignKey(x => x.RegionId).OnDelete(DeleteBehavior.Restrict);
entity.HasOne<ApplicationUser>().WithMany().HasForeignKey(x => x.UserId).OnDelete(DeleteBehavior.Restrict);
});
modelBuilder.Entity<CandidateProfile>(entity =>
{
entity.ToTable("candidate_profiles");
entity.Property(x => x.DisplayName).HasMaxLength(128).IsRequired();
entity.Property(x => x.NominationSummary).HasMaxLength(4000);
entity.Property(x => x.StatusReason).HasMaxLength(500);
entity.HasIndex(x => new { x.RegionId, x.Status, x.DisplayName });
entity.HasOne<AdministrativeRegion>().WithMany().HasForeignKey(x => x.RegionId).OnDelete(DeleteBehavior.Restrict);
});
modelBuilder.Entity<OrganizationUnit>(entity =>
{
entity.ToTable("organization_units");
entity.Property(x => x.Name).HasMaxLength(200).IsRequired();
entity.Property(x => x.Code).HasMaxLength(64).IsRequired();
entity.HasIndex(x => x.Code).IsUnique();
entity.HasOne<OrganizationUnit>().WithMany().HasForeignKey(x => x.ParentId).OnDelete(DeleteBehavior.Restrict);
});
modelBuilder.Entity<AuditEntry>(entity =>
{
entity.ToTable("audit_entries");
entity.Property(x => x.Action).HasMaxLength(128).IsRequired();
entity.Property(x => x.ResourceType).HasMaxLength(128).IsRequired();
entity.Property(x => x.TraceId).HasMaxLength(128);
entity.Property(x => x.ActorId).HasMaxLength(128);
entity.Property(x => x.MetadataJson).HasColumnType("json");
entity.HasIndex(x => new { x.OccurredAtUtc, x.ResourceType });
});
}
}