Initial code commit.

Sufi.Demo.PeopleDirectory.Persistence/Contexts/ApplicationDbContext.cs

@@ -0,0 +1,88 @@
+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;
+				}
+			}
+		}
+	}
+}