Initial code commit.

2026-02-03 10:44:31 +08:00
parent 8927c5ae0e
commit d69fe2cc1f
99 changed files with 10839 additions and 0 deletions
--- a/Sufi.Demo.PeopleDirectory.Persistence/Contexts/AuditableContext.cs
+++ b/Sufi.Demo.PeopleDirectory.Persistence/Contexts/AuditableContext.cs
@@ -0,0 +1,117 @@
+using Microsoft.AspNetCore.Identity;
+using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.EntityFrameworkCore.ChangeTracking;
+using Sufi.Demo.PeopleDirectory.Application.Enums;
+using Sufi.Demo.PeopleDirectory.Persistence.Models.Audit;
+using Sufi.Demo.PeopleDirectory.Persistence.Models.Identity;
+
+namespace Sufi.Demo.PeopleDirectory.Persistence.Contexts
+{
+	public abstract class AuditableContext(
+		DbContextOptions options
+		) : IdentityDbContext<AppUser, AppRole, string, IdentityUserClaim<string>, IdentityUserRole<string>, 
+			IdentityUserLogin<string>, AppRoleClaim, IdentityUserToken<string>>(options)
+	{
+		public DbSet<Audit> AuditTrails { get; set; } = null!;
+
+		protected override void OnModelCreating(ModelBuilder builder)
+		{
+			base.OnModelCreating(builder);
+
+			builder.Entity<Audit>().ToTable("AuditTrails", "Audit");
+		}
+
+		public virtual async Task<int> SaveChangesAsync(string? userId = null, CancellationToken cancellationToken = new())
+		{
+			var auditEntries = OnBeforeSaveChanges(userId);
+			var result = await base.SaveChangesAsync(cancellationToken);
+			await OnAfterSaveChanges(auditEntries, cancellationToken);
+			return result;
+		}
+
+		private List<AuditEntry> OnBeforeSaveChanges(string? userId)
+		{
+			ChangeTracker.DetectChanges();
+			var auditEntries = new List<AuditEntry>();
+			foreach (var entry in ChangeTracker.Entries())
+			{
+				if (entry.Entity is Audit || entry.State == EntityState.Detached || entry.State == EntityState.Unchanged)
+					continue;
+
+				var auditEntry = new AuditEntry(entry)
+				{
+					TableName = entry.Entity.GetType().Name,
+					UserId = userId
+				};
+				auditEntries.Add(auditEntry);
+				foreach (var property in entry.Properties)
+				{
+					if (property.IsTemporary)
+					{
+						auditEntry.TemporaryProperties.Add(property);
+						continue;
+					}
+
+					string propertyName = property.Metadata.Name;
+					if (property.Metadata.IsPrimaryKey())
+					{
+						auditEntry.KeyValues[propertyName] = property.CurrentValue;
+						continue;
+					}
+
+					switch (entry.State)
+					{
+						case EntityState.Added:
+							auditEntry.AuditType = AuditType.Create;
+							auditEntry.NewValues[propertyName] = property.CurrentValue;
+							break;
+
+						case EntityState.Deleted:
+							auditEntry.AuditType = AuditType.Delete;
+							auditEntry.OldValues[propertyName] = property.OriginalValue;
+							break;
+
+						case EntityState.Modified:
+							if (property.IsModified && property.OriginalValue?.Equals(property.CurrentValue) == false)
+							{
+								auditEntry.ChangedColumns.Add(propertyName);
+								auditEntry.AuditType = AuditType.Update;
+								auditEntry.OldValues[propertyName] = property.OriginalValue;
+								auditEntry.NewValues[propertyName] = property.CurrentValue;
+							}
+							break;
+					}
+				}
+			}
+			foreach (var auditEntry in auditEntries.Where(_ => !_.HasTemporaryProperties))
+			{
+				AuditTrails.Add(auditEntry.ToAudit());
+			}
+			return [.. auditEntries.Where(_ => _.HasTemporaryProperties)];
+		}
+
+		private Task OnAfterSaveChanges(List<AuditEntry> auditEntries, CancellationToken cancellationToken = new())
+		{
+			if (auditEntries == null || auditEntries.Count == 0)
+				return Task.CompletedTask;
+
+			foreach (var auditEntry in auditEntries)
+			{
+				foreach (var prop in auditEntry.TemporaryProperties)
+				{
+					if (prop.Metadata.IsPrimaryKey())
+					{
+						auditEntry.KeyValues[prop.Metadata.Name] = prop.CurrentValue;
+					}
+					else
+					{
+						auditEntry.NewValues[prop.Metadata.Name] = prop.CurrentValue;
+					}
+				}
+				AuditTrails.Add(auditEntry.ToAudit());
+			}
+			return SaveChangesAsync(cancellationToken);
+		}
+	}
+}