BimAI/BimAI.Infrastructure/Data/BimixDbContext.cs

using Microsoft.EntityFrameworkCore;
using BimAI.Domain.Entities;

namespace BimAI.Infrastructure.Data;

public class BimAIDbContext(DbContextOptions<BimAIDbContext> options) : DbContext(options)
{
    public DbSet<Product> Products { get; set; }
    public DbSet<SyncState> SyncStates { get; set; }
    public DbSet<User> Users { get; set; }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);
        
        // Product properties
        modelBuilder.Entity<Product>().HasKey(x => x.Id);
        modelBuilder.Entity<Product>().Property(x => x.Name).IsRequired().HasMaxLength(512);
        modelBuilder.Entity<Product>().Property(x => x.Code).IsRequired().HasMaxLength(40);
        modelBuilder.Entity<Product>().Property(x => x.Ean).IsRequired().HasMaxLength(50);
        modelBuilder.Entity<Product>().Property(x => x.StockAddresses).IsRequired().HasMaxLength(512);

        // SyncState properties
        modelBuilder.Entity<SyncState>().HasKey((x => x.Entity));
        
        // User properties
        modelBuilder.Entity<User>().HasKey(x => x.Id);
        modelBuilder.Entity<User>().Property(x => x.GoogleId).IsRequired().HasMaxLength(100);
        modelBuilder.Entity<User>().Property(x => x.Email).IsRequired().HasMaxLength(255);
        modelBuilder.Entity<User>().Property(x => x.FullName).IsRequired().HasMaxLength(255);
        modelBuilder.Entity<User>().Property(x => x.IsActive).IsRequired().HasDefaultValue(false);
        modelBuilder.Entity<User>().Property(x => x.LastLoginAt).IsRequired(false);
        
        // User indexes
        modelBuilder.Entity<User>().HasIndex(x => x.GoogleId).IsUnique().HasDatabaseName("IX_Users_GoogleId");
        modelBuilder.Entity<User>().HasIndex(x => x.Email).IsUnique().HasDatabaseName("IX_Users_Email");
        
        // Configure defaults for all CreatedAt and UpdatedAt in entities
        ConfigureBaseEntity(modelBuilder);
    }

    private void ConfigureBaseEntity(ModelBuilder modelBuilder)
    {
        foreach (var entityType in modelBuilder.Model.GetEntityTypes())
        {
            if (typeof(BaseEntity).IsAssignableFrom(entityType.ClrType))
            {
                modelBuilder.Entity(entityType.ClrType)
                    .Property(nameof(BaseEntity.CreatedAt))
                    .HasDefaultValueSql("GETUTCDATE()");
                
                modelBuilder.Entity(entityType.ClrType)
                    .Property(nameof(BaseEntity.UpdatedAt))
                    .HasDefaultValueSql("GETUTCDATE()");
            }
        }
    }
    
    public override int SaveChanges()
    {
        UpdateTimestamps();
        return base.SaveChanges();
    }

    public override Task<int> SaveChangesAsync(CancellationToken cancellationToken = default)
    {
        UpdateTimestamps();
        return base.SaveChangesAsync(cancellationToken);
    }

    private void UpdateTimestamps()
    {
        var entities = ChangeTracker.Entries<BaseEntity>();
        
        foreach (var entity in entities)
        {
            if (entity.State == EntityState.Added)
            {
                entity.Entity.CreatedAt = DateTime.UtcNow;
                entity.Entity.UpdatedAt = DateTime.UtcNow;
                break;
            }
            else if (entity.State == EntityState.Modified)
            {
                entity.Entity.UpdatedAt = DateTime.UtcNow;
                break;
            }
        }
    }
}