using Microsoft.EntityFrameworkCore;
using Svrnty.Sample.Data.Entities;

namespace Svrnty.Sample.Data;

/// <summary>
/// Database context for AI agent system with conversation history and business data
/// </summary>
public class AgentDbContext : DbContext
{
    public AgentDbContext(DbContextOptions<AgentDbContext> options) : base(options)
    {
    }

    public DbSet<Conversation> Conversations => Set<Conversation>();
    public DbSet<Revenue> Revenues => Set<Revenue>();
    public DbSet<Customer> Customers => Set<Customer>();

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

        // Configure Conversation entity
        modelBuilder.Entity<Conversation>(entity =>
        {
            entity.HasKey(e => e.Id);
            entity.HasIndex(e => e.CreatedAt).HasDatabaseName("idx_conversations_created");
            entity.HasIndex(e => e.UpdatedAt).HasDatabaseName("idx_conversations_updated");

            entity.Property(e => e.MessagesJson)
                .HasColumnType("jsonb")
                .IsRequired()
                .HasDefaultValue("[]");
        });

        // Configure Revenue entity
        modelBuilder.Entity<Revenue>(entity =>
        {
            entity.HasKey(e => e.Id);
            entity.HasIndex(e => new { e.Month, e.Year })
                .HasDatabaseName("idx_revenue_month")
                .IsUnique();

            entity.Property(e => e.Amount)
                .HasPrecision(18, 2);
        });

        // Configure Customer entity
        modelBuilder.Entity<Customer>(entity =>
        {
            entity.HasKey(e => e.Id);
            entity.HasIndex(e => e.State).HasDatabaseName("idx_customers_state");
            entity.HasIndex(e => e.Tier).HasDatabaseName("idx_customers_tier");
            entity.HasIndex(e => new { e.State, e.Tier })
                .HasDatabaseName("idx_customers_state_tier");
        });
    }
}