using InboxIntel.Application.Abstractions;
using InboxIntel.Domain.Common;
using InboxIntel.Domain.Entities;
using Microsoft.EntityFrameworkCore;
using System.Reflection;

namespace InboxIntel.Infrastructure.Persistence;

public class AppDbContext : DbContext, IAppDbContext
{
    public AppDbContext(DbContextOptions<AppDbContext> options) : base(options) { }

    public DbSet<User> Users => Set<User>();
    public DbSet<Email> Emails => Set<Email>();
    public DbSet<MailThread> Threads => Set<MailThread>();
    public DbSet<Sender> Senders => Set<Sender>();
    public DbSet<MailDomain> Domains => Set<MailDomain>();
    public DbSet<Attachment> Attachments => Set<Attachment>();
    public DbSet<Label> Labels => Set<Label>();
    public DbSet<EmailLabel> EmailLabels => Set<EmailLabel>();
    public DbSet<SyncState> SyncStates => Set<SyncState>();
    public DbSet<AnalyticsAggregate> AnalyticsAggregates => Set<AnalyticsAggregate>();
    public DbSet<WidgetLayout> WidgetLayouts => Set<WidgetLayout>();
    public DbSet<UnsubscribeItem> UnsubscribeItems => Set<UnsubscribeItem>();

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.ApplyConfigurationsFromAssembly(Assembly.GetExecutingAssembly());
        base.OnModelCreating(modelBuilder);
    }

    public override Task<int> SaveChangesAsync(CancellationToken ct = default)
    {
        foreach (var entry in ChangeTracker.Entries<AuditableEntity>())
        {
            if (entry.State == EntityState.Modified)
                entry.Entity.UpdatedAtUtc = DateTimeOffset.UtcNow;
        }
        return base.SaveChangesAsync(ct);
    }
}