using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using SVSim.Database.Common;
using SVSim.Database.DataSeeders;
using SVSim.Database.Models;

namespace SVSim.Database;

public class SVSimDbContext : DbContext
{
    private readonly ILogger<SVSimDbContext> _logger;

    public SVSimDbContext(ILogger<SVSimDbContext> logger, DbContextOptions<SVSimDbContext> options) : base(options)
    {
        _logger = logger;
    }

    #region DbSets

    public DbSet<Viewer> Viewers => Set<Viewer>();

    public DbSet<ShadowverseCardEntry> Cards => Set<ShadowverseCardEntry>();
    public DbSet<ShadowverseCardSetEntry> CardSets => Set<ShadowverseCardSetEntry>();
    public DbSet<ShadowverseDeckEntry> Decks => Set<ShadowverseDeckEntry>();

    public DbSet<ClassEntry> Classes => Set<ClassEntry>();
    public DbSet<ClassExpEntry> ClassExpCurve => Set<ClassExpEntry>();
    public DbSet<LeaderSkinEntry> LeaderSkins => Set<LeaderSkinEntry>();
    public DbSet<SleeveEntry> Sleeves => Set<SleeveEntry>();
    public DbSet<EmblemEntry> Emblems => Set<EmblemEntry>();
    public DbSet<DegreeEntry> Degrees => Set<DegreeEntry>();
    public DbSet<MyPageBackgroundEntry> MyPageBackgrounds => Set<MyPageBackgroundEntry>();
    public DbSet<BattlefieldEntry> Battlefields => Set<BattlefieldEntry>();
    public DbSet<RankInfoEntry> RankInfo => Set<RankInfoEntry>();
    public DbSet<ItemEntry> Items => Set<ItemEntry>();

    public DbSet<GameConfiguration> GameConfigurations => Set<GameConfiguration>();

    // Prod-captured globals — populated by SVSim.Bootstrap, not HasData. See
    // docs/audits/prod-data-capture-strategy-2026-05-23.md.
    public DbSet<MyRotationSettingEntry> MyRotationSettings => Set<MyRotationSettingEntry>();
    public DbSet<MyRotationAbilityEntry> MyRotationAbilities => Set<MyRotationAbilityEntry>();
    public DbSet<AvatarAbilityEntry> AvatarAbilities => Set<AvatarAbilityEntry>();
    public DbSet<DefaultDeckEntry> DefaultDecks => Set<DefaultDeckEntry>();
    public DbSet<DefaultLeaderSkinSettingEntry> DefaultLeaderSkinSettings => Set<DefaultLeaderSkinSettingEntry>();
    public DbSet<ArenaSeasonConfig> ArenaSeasons => Set<ArenaSeasonConfig>();
    public DbSet<SpotCardEntry> SpotCards => Set<SpotCardEntry>();
    public DbSet<ReprintedCardEntry> ReprintedCards => Set<ReprintedCardEntry>();
    public DbSet<UnlimitedRestrictionEntry> UnlimitedRestrictions => Set<UnlimitedRestrictionEntry>();
    public DbSet<LoadingExclusionCardEntry> LoadingExclusionCards => Set<LoadingExclusionCardEntry>();
    public DbSet<BattlePassLevelEntry> BattlePassLevels => Set<BattlePassLevelEntry>();
    public DbSet<DailyLoginBonusEntry> DailyLoginBonuses => Set<DailyLoginBonusEntry>();
    public DbSet<BannerEntry> Banners => Set<BannerEntry>();
    public DbSet<ColosseumConfig> Colosseums => Set<ColosseumConfig>();
    public DbSet<SealedConfig> SealedSeasons => Set<SealedConfig>();
    public DbSet<MasterPointRankingPeriodEntry> MasterPointRankingPeriods => Set<MasterPointRankingPeriodEntry>();
    public DbSet<MaintenanceCardEntry> MaintenanceCards => Set<MaintenanceCardEntry>();
    public DbSet<FeatureMaintenanceEntry> FeatureMaintenances => Set<FeatureMaintenanceEntry>();
    public DbSet<PreReleaseInfo> PreReleaseInfos => Set<PreReleaseInfo>();

    #endregion

    public override async Task<int> SaveChangesAsync(CancellationToken cancellationToken = default)
    {
        foreach (var entityEntry in ChangeTracker.Entries())
        {
            if (entityEntry.Entity is ITimeTrackedEntity timeTrackedEntity)
            {
                if (entityEntry.State is EntityState.Added && timeTrackedEntity.DateCreated == DateTime.MinValue)
                {
                    timeTrackedEntity.DateCreated = DateTime.UtcNow;
                }
                if (entityEntry.State is EntityState.Modified or EntityState.Added)
                {
                    timeTrackedEntity.DateUpdated = DateTime.UtcNow;
                }
            }
        }

        return await base.SaveChangesAsync(cancellationToken);
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<ShadowverseDeckEntry>()
            .OwnsMany(de => de.Cards);

        // BaseEntity<TKey> annotates Id with [DatabaseGenerated(None)] for the integer-PK
        // entities seeded via HasData. ShadowverseDeckEntry uses Guid and is created at
        // runtime — without client-side generation every new deck gets Guid.Empty and the
        // second deck insert collides on PK. (DDL has no column default; this only works
        // because EF generates a sequential Guid before INSERT.)
        modelBuilder.Entity<ShadowverseDeckEntry>()
            .Property(d => d.Id)
            .ValueGeneratedOnAdd();

        // EF can't figure this many-to-many out on its own
        modelBuilder.Entity<SleeveEntry>()
            .HasMany(se => se.Viewers)
            .WithMany(v => v.Sleeves);

        modelBuilder.HasSequence<long>("ShortUdidSequence").StartsAt(400000000);
        modelBuilder.Entity<Viewer>()
            .Property(v => v.ShortUdid)
            .UseSequence("ShortUdidSequence");

        new BaseDataSeeder().Seed(modelBuilder);
        new DefaultSettingsSeeder().Seed(modelBuilder);

        base.OnModelCreating(modelBuilder);
    }

    public void UpdateDatabase()
    {
        IEnumerable<string> pendingMigrations = Database.GetPendingMigrations();
        if (!pendingMigrations.Any())
        {
            _logger.LogDebug("No pending migrations found, continuing.");
            return;
        }

        foreach (string migration in pendingMigrations)
        {
            _logger.LogInformation("Found pending migration with name {migrationName}.", migration);
        }
        _logger.LogInformation("Attempting to apply pending migrations...");
        Database.Migrate();
        _logger.LogInformation("Migrations applied.");
    }
}