using Microsoft.EntityFrameworkCore;
using StreakBot.Data.Entities;

namespace StreakBot.Data;

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

    public DbSet<Streak> Streaks => Set<Streak>();
    public DbSet<Channel> Channels => Set<Channel>();

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<Streak>(entity =>
        {
            entity.HasKey(e => e.Id);
            entity.Property(e => e.CreatedDate).IsRequired();
            entity.Property(e => e.User1Id).IsRequired();
            entity.Property(e => e.User2Id).IsRequired();
            entity.Property(e => e.ServerId).IsRequired();
            entity.Property(e => e.LastResetCheck).IsRequired();
            entity.Property(e => e.ReminderSent).IsRequired();
        });

        modelBuilder.Entity<Channel>(entity =>
        {
            entity.HasKey(e => e.Id);
            entity.Property(e => e.ServerId).IsRequired();
            entity.Property(e => e.ChannelId).IsRequired();
            entity.Property(e => e.ChannelType).IsRequired();
        });
    }
}