89 lines
2.8 KiB
C#
89 lines
2.8 KiB
C#
using Microsoft.AspNetCore.Identity;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Sufi.Demo.PeopleDirectory.Application.Contracts.Services;
|
|
using Sufi.Demo.PeopleDirectory.Domain.Common;
|
|
using Sufi.Demo.PeopleDirectory.Domain.Entities.Misc;
|
|
using Sufi.Demo.PeopleDirectory.Persistence.Models.Identity;
|
|
|
|
namespace Sufi.Demo.PeopleDirectory.Persistence.Contexts
|
|
{
|
|
public class ApplicationDbContext(
|
|
DbContextOptions<ApplicationDbContext> options,
|
|
ICurrentUserService currentUserService
|
|
) : AuditableContext(options)
|
|
{
|
|
public virtual DbSet<Contact> Contacts { get; set; } = null!;
|
|
public virtual DbSet<ServerInfo> ServerInfos { get; set; } = null!;
|
|
|
|
public override Task<int> SaveChangesAsync(string? userId = null, CancellationToken cancellationToken = default)
|
|
{
|
|
PopulateAuditRecords();
|
|
|
|
return base.SaveChangesAsync(userId, cancellationToken);
|
|
}
|
|
|
|
public override async Task<int> SaveChangesAsync(CancellationToken cancellationToken = default)
|
|
{
|
|
PopulateAuditRecords();
|
|
|
|
if (currentUserService.UserId == null)
|
|
{
|
|
return await base.SaveChangesAsync(cancellationToken);
|
|
}
|
|
|
|
return await base.SaveChangesAsync(currentUserService.UserId, cancellationToken);
|
|
}
|
|
|
|
protected override void OnModelCreating(ModelBuilder builder)
|
|
{
|
|
foreach (var property in builder.Model.GetEntityTypes()
|
|
.SelectMany(t => t.GetProperties())
|
|
.Where(p => p.ClrType == typeof(decimal) || p.ClrType == typeof(decimal?)))
|
|
{
|
|
property.SetColumnType("decimal(18,2)");
|
|
}
|
|
|
|
foreach (var property in builder.Model.GetEntityTypes()
|
|
.SelectMany(t => t.GetProperties())
|
|
.Where(p => p.Name is "LastModifiedBy" or "CreatedBy"))
|
|
{
|
|
property.SetColumnType("character varying(100)");
|
|
}
|
|
|
|
base.OnModelCreating(builder);
|
|
|
|
var assembly = typeof(ApplicationDbContext).Assembly;
|
|
builder.ApplyConfigurationsFromAssembly(assembly);
|
|
|
|
builder.Entity<IdentityUserRole<string>>(entity => entity.ToTable("UserRoles", "Identity"));
|
|
|
|
builder.Entity<IdentityUserClaim<string>>(entity => entity.ToTable("UserClaims", "Identity"));
|
|
|
|
builder.Entity<IdentityUserLogin<string>>(entity => entity.ToTable("UserLogins", "Identity"));
|
|
|
|
builder.Entity<AppRoleClaim>(entity => entity.ToTable("RoleClaims", "Identity"));
|
|
|
|
builder.Entity<IdentityUserToken<string>>(entity => entity.ToTable("UserTokens", "Identity"));
|
|
}
|
|
|
|
private void PopulateAuditRecords()
|
|
{
|
|
foreach (var entry in ChangeTracker.Entries<IAuditableEntity>().ToList())
|
|
{
|
|
switch (entry.State)
|
|
{
|
|
case EntityState.Added:
|
|
entry.Entity.CreatedOn = DateTime.UtcNow;
|
|
entry.Entity.CreatedBy = currentUserService.UserId;
|
|
break;
|
|
|
|
case EntityState.Modified:
|
|
entry.Entity.LastModifiedOn = DateTime.UtcNow;
|
|
entry.Entity.LastModifiedBy = currentUserService.UserId;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|