using backend.Auth;
using backend.Conversations;
using backend.Devices;
using backend.Pairing;
using backend.Settings;
using Microsoft.AspNetCore.Identity;
using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore;

namespace backend.Data;

public class AppDbContext(DbContextOptions<AppDbContext> options)
    : IdentityDbContext<ApplicationUser, IdentityRole<Guid>, Guid>(options)
{
    public DbSet<Device> Devices => Set<Device>();
    public DbSet<DeviceConfig> DeviceConfigs => Set<DeviceConfig>();
    public DbSet<SystemSettings> SystemSettings => Set<SystemSettings>();
    public DbSet<PairingCode> PairingCodes => Set<PairingCode>();
    public DbSet<Conversation> Conversations => Set<Conversation>();
    public DbSet<Turn> Turns => Set<Turn>();

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

        b.Entity<Device>()
            .HasOne<ApplicationUser>()
            .WithMany()
            .HasForeignKey(d => d.OwnerUserId)
            .OnDelete(DeleteBehavior.Cascade);
        b.Entity<Device>().HasIndex(d => d.TokenHash).IsUnique();
        b.Entity<Device>()
            .HasOne(d => d.Config)
            .WithOne()
            .HasForeignKey<DeviceConfig>(c => c.DeviceId)
            .OnDelete(DeleteBehavior.Cascade);
        b.Entity<DeviceConfig>().HasKey(c => c.DeviceId);
        b.Entity<SystemSettings>().HasKey(s => s.Id);
        b.Entity<SystemSettings>().Property(s => s.Id).ValueGeneratedNever();
        b.Entity<PairingCode>().HasKey(p => p.Code);
        b.Entity<PairingCode>().HasIndex(p => p.ExpiresAt);

        b.Entity<Conversation>().HasIndex(c => c.DeviceId);
        b.Entity<Conversation>()
            .HasMany(c => c.Turns)
            .WithOne()
            .HasForeignKey(t => t.ConversationId)
            .OnDelete(DeleteBehavior.Cascade);
        b.Entity<Turn>().HasIndex(t => new { t.ConversationId, t.Index }).IsUnique();
    }
}