// Entity Framework Plus

// Doc: https://entityframework-plus.net/audit

​

// @nuget: Microsoft.EntityFrameworkCore.SqlServer

// @nuget: Z.EntityFramework.Plus.EFCore

​

using System;

using System.Linq;

using Microsoft.EntityFrameworkCore;

using System.Collections.Generic;

using Microsoft.Data.SqlClient;

using Z.EntityFramework.Plus;

using System.ComponentModel.DataAnnotations;

using System.ComponentModel.DataAnnotations.Schema;

​

public class Program

{

    public static void Main()

    {

        using (var context = new EntityContext())

        {

            context.Database.EnsureCreated();

        }

        AuditManager.DefaultConfiguration.ExcludeDataAnnotation();

        AuditManager.DefaultConfiguration.DataAnnotationDisplayName();

        AuditManager.DefaultConfiguration.AutoSavePreAction = (context, audit) =>

        {

            audit.CreatedBy = "Kevin Lam";

            // ADD "Where(x => x.AuditEntryID == 0)" to allow multiple SaveChanges with same Audit

            var customAuditEntries = audit.Entries.Select(x => new DbAuditEntry(x)).ToList();

            (context as EntityContext).AuditEntries.AddRange(customAuditEntries);

            //var customAuditEntries = audit.Entries.ToList();

            //(context as EntityContext).AuditEntries.AddRange(customAuditEntries);

        };

​

        // Audit : Easily tracks changes, exclude/include entity or property and auto save audit entries in the database.

        using (var context = new EntityContext())

        {

            context.Customers.Add(new Customer() { Name = "Customer_C", Description = "Description", IsActive = false}); // add

            context.Customers.Add(new Customer() { Name = "Customer_D", Description = "Description2", IsActive = true}); // add

            //var audit = new Audit();

            //audit.CreatedBy = "lamkek"; // Optional

            //context.SaveChanges(audit);

            context.SaveChanges();

            var c1 = context.Customers.FirstOrDefault();

            c1.Name = "Customer_E";

            context.Update(c1);

            context.SaveChanges();

            // Access to all auditing information

            //var entries = audit.Entries;

            var entries = context.AuditEntries;

            FiddleHelper.WriteTable("All Entry",entries);

            foreach(var entry in entries)

            { 

                FiddleHelper.WriteTable("Properties for One Entry",entry.Properties);

            } 

        }

    }

​

    public class EntityContext : DbContext

    {

        public EntityContext() 

        {

        }

​

        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)

        {

            optionsBuilder.UseSqlServer(new SqlConnection(FiddleHelper.GetConnectionStringSqlServer()));

​

            base.OnConfiguring(optionsBuilder);

        }

        public override int SaveChanges()

        {

            var audit = new Audit();

            audit.CreatedBy="lamkek";

            audit.PreSaveChanges(this);

            var rowAffecteds = base.SaveChanges();

            audit.PostSaveChanges();

​

            if (audit.Configuration.AutoSavePreAction != null)

            {

                audit.Configuration.AutoSavePreAction(this, audit);

                base.SaveChanges();

            }

​

            return rowAffecteds;

        }

​

        public DbSet<DbAuditEntry> AuditEntries { get; set; }

        public DbSet<DbAuditEntryProperty> AuditEntryProperties { get; set; }

​

        //public DbSet<AuditEntry> AuditEntries { get; set; }

        //public DbSet<AuditEntryProperty> AuditEntryProperties { get; set; }

​

        public DbSet<Customer> Customers { get; set; }

    } 

    public class DbAuditEntry

    {

        public DbAuditEntry()

        {

        }

        public DbAuditEntry(AuditEntry entry)

        {

            //EntitySetName = entry.EntitySetName;

            EntityTypeName = entry.EntityTypeName;

            State = entry.State;

            StateName = entry.StateName;

            CreatedBy = entry.CreatedBy;

            CreatedDate = entry.CreatedDate;

            Properties = entry.Properties.Select(x => new DbAuditEntryProperty(x)).ToList();

        }

        [Key]

        public int AuditEntryID { get; set; }

        //public string EntitySetName { get; set; }

        public string EntityTypeName { get; set; }

        public AuditEntryState State { get; set; }

        public string StateName { get; set; }

        public string CreatedBy { get; set; }

​

        public DateTime CreatedDate { get; set; }

        [NotMapped] 

        public List<DbAuditEntryProperty> Properties { get; set; }

    }

    public class DbAuditEntryProperty

    {

        public DbAuditEntryProperty()

        {

        }

        public DbAuditEntryProperty(AuditEntryProperty property)

        {

            RelationName = property.RelationName;

            PropertyName = property.PropertyName;

            OldValue = property.OldValueFormatted;

            NewValue = property.NewValueFormatted;

        }

        [Key]

        public int AuditEntryPropertyID { get; set; }

        public int AuditEntryID { get; set; }

        public string RelationName { get; set; }

        public string PropertyName { get; set; }

        public string OldValue { get; set; }

        public string NewValue { get; set; }

    }

    [AuditDisplay("KevinTable")]

    public class Customer

    {

        [AuditDisplay("Customer Identification")]

        public int CustomerID { get; set; }

        public string Name { get; set; }

        public string Description { get; set; }

        [AuditExclude]  

        public Boolean IsActive { get; set; }

    }

}