Notes.SqliteRepository/DataContext.cs - master - hlebusheck/notes

Notes.SqliteRepository/DataContext.cs
89 строк3 KB
Hlebusheck
Исправлен баг с доступом только для чтения поиска в TextBox
31 дек 2022, 19:10
31 дек 2022, 19:10f26f59b
using Microsoft.EntityFrameworkCore;
using Notes.Model;

namespace Notes.SqliteRepository
{
    public class DataContext : DbContext
    {
        public DataContext()
        {
            ConfigureDefaultDbPath();
        }
        public DataContext(DbContextOptions<DataContext> options) : this(options, null) { }
        public DataContext(DbContextOptions<DataContext> options, string? path)
            : base(options)
        {
            DbPath = string.IsNullOrWhiteSpace(path)
                ? ConfigureDefaultDbPath()
                : path;

            Database.Migrate();
        }

        public string DbPath { get; set; } = string.Empty;

        public bool AutoSetInsertedDate { get; set; } = true;

        public bool AutoSetUpdatedDate { get; set; } = true;

        public DbSet<Memo> Memos => Set<Memo>();

        public override Task<int> SaveChangesAsync(CancellationToken cancellationToken = default)
        {
            if (AutoSetInsertedDate)
            {
                var insertedEntries = ChangeTracker.Entries()
                                   .Where(x => x.State == EntityState.Added)
                                   .Select(x => x.Entity);

                foreach (var insertedEntry in insertedEntries)
                {
                    var auditableEntity = insertedEntry as IAuditableEntity;
                    if (auditableEntity != null)
                        auditableEntity.InsertedDate = DateTime.Now;
                }
            }

            if (AutoSetUpdatedDate)
            {
                var modifiedEntries = ChangeTracker.Entries()
                       .Where(x => x.State == EntityState.Modified)
                       .Select(x => x.Entity);

                foreach (var modifiedEntry in modifiedEntries)
                {
                    var auditableEntity = modifiedEntry as IAuditableEntity;
                    if (auditableEntity != null)
                        auditableEntity.UpdatedDate = DateTime.Now;
                }
            }

            return base.SaveChangesAsync(cancellationToken);
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            modelBuilder.Entity<Memo>().OwnsOne(memo => memo.BodyProperties);

            base.OnModelCreating(modelBuilder);
        }

        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
        {
            base.OnConfiguring(optionsBuilder);
            optionsBuilder.UseSqlite(@$"Data Source={DbPath}");
        }

        private string ConfigureDefaultDbPath()
        {
            var folder = Environment.SpecialFolder.LocalApplicationData;
            var path = Environment.GetFolderPath(folder);
#if DEBUG
            var dbName = "notes.test.db";
#else
            var dbName = "notes.db";
#endif
            return Path.Join(path, dbName);
        }
    }
}