using DigitalData.MessagingService.Domain.Entities;
using Microsoft.EntityFrameworkCore;

namespace DigitalData.MessagingService.Infrastructure.Persistence;

/// <summary>
/// Entity Framework Core DbContext for MessagingService.
/// </summary>
public class MessagingServiceDbContext(DbContextOptions<MessagingServiceDbContext> options) : DbContext(options)
{
    public DbSet<EmailAccount> EmailAccounts => Set<EmailAccount>();
    public DbSet<ReceivedEmail> ReceivedEmails => Set<ReceivedEmail>();
    public DbSet<EmailAttachment> EmailAttachments => Set<EmailAttachment>();

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);

        modelBuilder.Entity<EmailAccount>(entity =>
        {
            entity.Property(e => e.Username)
                  .UseCollation("SQL_Latin1_General_CP1_CI_AS");
        });

        modelBuilder.Entity<ReceivedEmail>(entity =>
        {
            entity.HasMany(e => e.Attachments)
                  .WithOne(a => a.Email)
                  .HasForeignKey(a => a.EmailId)
                  .OnDelete(DeleteBehavior.Cascade);

            entity.HasOne(e => e.Account)
                  .WithMany()
                  .HasForeignKey(e => e.AccountId)
                  .OnDelete(DeleteBehavior.Restrict);
        });
    }
}