using MareSynchronosShared.Models;
using Microsoft.EntityFrameworkCore;

namespace MareSynchronosShared.Data;

public class MareDbContext : DbContext
{
#if DEBUG
    public MareDbContext() { }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        if (optionsBuilder.IsConfigured)
        {
            base.OnConfiguring(optionsBuilder);
            return;
        }

        optionsBuilder.UseNpgsql("Host=localhost;Port=5432;Database=mare;Username=postgres", builder =>
        {
            builder.MigrationsHistoryTable("_efmigrationshistory", "public");
            builder.MigrationsAssembly("MareSynchronosShared");
        }).UseSnakeCaseNamingConvention();
        optionsBuilder.EnableThreadSafetyChecks(false);

        base.OnConfiguring(optionsBuilder);
    }
#endif

    public MareDbContext(DbContextOptions<MareDbContext> options) : base(options)
    {
    }

    public DbSet<User> Users { get; set; }
    public DbSet<FileCache> Files { get; set; }
    public DbSet<ClientPair> ClientPairs { get; set; }
    public DbSet<ForbiddenUploadEntry> ForbiddenUploadEntries { get; set; }
    public DbSet<Banned> BannedUsers { get; set; }
    public DbSet<Auth> Auth { get; set; }
    public DbSet<LodeStoneAuth> LodeStoneAuth { get; set; }
    public DbSet<BannedRegistrations> BannedRegistrations { get; set; }
    public DbSet<Group> Groups { get; set; }
    public DbSet<GroupPair> GroupPairs { get; set; }
    public DbSet<GroupBan> GroupBans { get; set; }
    public DbSet<GroupTempInvite> GroupTempInvites { get; set; }


    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<Auth>().ToTable("auth");
        modelBuilder.Entity<User>().ToTable("users");
        modelBuilder.Entity<FileCache>().ToTable("file_caches");
        modelBuilder.Entity<FileCache>().HasIndex(c => c.UploaderUID);
        modelBuilder.Entity<ClientPair>().ToTable("client_pairs");
        modelBuilder.Entity<ClientPair>().HasKey(u => new { u.UserUID, u.OtherUserUID });
        modelBuilder.Entity<ClientPair>().HasIndex(c => c.UserUID);
        modelBuilder.Entity<ClientPair>().HasIndex(c => c.OtherUserUID);
        modelBuilder.Entity<ForbiddenUploadEntry>().ToTable("forbidden_upload_entries");
        modelBuilder.Entity<Banned>().ToTable("banned_users");
        modelBuilder.Entity<LodeStoneAuth>().ToTable("lodestone_auth");
        modelBuilder.Entity<BannedRegistrations>().ToTable("banned_registrations");
        modelBuilder.Entity<Group>().ToTable("groups");
        modelBuilder.Entity<Group>().HasIndex(c => c.OwnerUID);
        modelBuilder.Entity<GroupPair>().ToTable("group_pairs");
        modelBuilder.Entity<GroupPair>().HasKey(u => new { u.GroupGID, u.GroupUserUID });
        modelBuilder.Entity<GroupPair>().HasIndex(c => c.GroupUserUID);
        modelBuilder.Entity<GroupPair>().HasIndex(c => c.GroupGID);
        modelBuilder.Entity<GroupBan>().ToTable("group_bans");
        modelBuilder.Entity<GroupBan>().HasKey(u => new { u.GroupGID, u.BannedUserUID });
        modelBuilder.Entity<GroupBan>().HasIndex(c => c.BannedUserUID);
        modelBuilder.Entity<GroupBan>().HasIndex(c => c.GroupGID);
        modelBuilder.Entity<GroupTempInvite>().ToTable("group_temp_invites");
        modelBuilder.Entity<GroupTempInvite>().HasKey(u => new { u.GroupGID, u.Invite });
        modelBuilder.Entity<GroupTempInvite>().HasIndex(c => c.GroupGID);
        modelBuilder.Entity<GroupTempInvite>().HasIndex(c => c.Invite);
    }
}