using FinlyticAnalyzer.Entities;
using Microsoft.EntityFrameworkCore;

namespace FinlyticAnalyzer.Database;

public class AnalyzerDbContext : DbContext
{
    public AnalyzerDbContext(DbContextOptions<AnalyzerDbContext> options) : base(options) { }

    public DbSet<AnalysisEntity> Analyses => Set<AnalysisEntity>();
    public DbSet<AnalyzerSettingsEntity> Settings => Set<AnalyzerSettingsEntity>();
    public DbSet<TradeProposalEntity> TradeProposals => Set<TradeProposalEntity>();

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

        modelBuilder.Entity<AnalysisEntity>(entity =>
        {
            entity.HasIndex(e => e.AnalysisId).IsUnique();
            entity.HasIndex(e => e.EventId);
            entity.HasIndex(e => e.Isin);
            entity.HasIndex(e => e.Sector);
            entity.HasIndex(e => e.CreatedAt);
        });

        modelBuilder.Entity<TradeProposalEntity>(entity =>
        {
            entity.HasIndex(e => e.Isin);
            entity.HasIndex(e => e.ExpiresAt);
        });
    }
}