using Microsoft.EntityFrameworkCore;
using HyperBooru.Services;

namespace HyperBooru;

enum HBObjectId {
    NsfwTag   = -1,
    IngestTag = -2,
    AdminUser = -3
}

public class HBContext : DbContext {
    public static readonly Guid NsfwTag   = new("EBDAD4F8-455A-4351-8017-1D4854D6FA38");
    public static readonly Guid IngestTag = new("EA212801-5BCC-4C0E-814F-FB9D30DB58BC");

    public DbSet<HBObject>      Objects        { get; set; }
    public DbSet<TagDefinition> TagDefinitions { get; set; }
    public DbSet<Tag>           Tags           { get; set; }
    public DbSet<Media>         Media          { get; set; }
    public DbSet<UploadedFile>  UploadedFiles  { get; set; }
    public DbSet<OcrData>       OcrData        { get; set; }

    // Security-related tables
    public DbSet<HBPrincipal> Principals { get; set; }
    public DbSet<User>        Users      { get; set; }
    public DbSet<Group>       Groups     { get; set; }
    public DbSet<Acl>         Acls       { get; set; }
    public DbSet<AclRule>     AclRules   { get; set; }

    private IConfigService config;

    public HBContext(DbContextOptions<HBContext> options, IConfigService config) : base(options) =>
        this.config = config;

    protected override void OnConfiguring(DbContextOptionsBuilder options) {
        options.UseNpgsql(config.DbConnectionString);

        #if DEBUG
        options.EnableSensitiveDataLogging();
        #endif
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder) {
        // Don't use shared tables for inherited types
        modelBuilder.Entity<HBObject>().ToTable("Objects");
        modelBuilder.Entity<TagDefinition>().ToTable("TagDefinitions");
        modelBuilder.Entity<Tag>().ToTable("Tags");
        modelBuilder.Entity<Media>().ToTable("Media");
        modelBuilder.Entity<UploadedFile>().ToTable("UploadedFiles");

        // Seed internal tag definitions
        // These should NEVER change
        modelBuilder.Entity<TagDefinition>().HasData(new TagDefinition[] {
            new() {
                ObjectId = (int) HBObjectId.NsfwTag,
                Guid     = NsfwTag,
                Source   = TagSource.Internal,
                Name     = "nsfw"
            },
            new() {
                ObjectId = (int) HBObjectId.IngestTag,
                Guid     = IngestTag,
                Source   = TagSource.Internal,
                Name     = "ingest"
            }
        });

        // Seed initial admin user
        modelBuilder.Entity<User>().HasData(new User[] {
            new() {
                ObjectId     = (int) HBObjectId.AdminUser,
                Name         = "admin",
                PasswordHash = UserService.HashPassword("admin")
            }
        });

        // Some complex relationships cannot be inferred and require
        // additional configuration, as seen below.
        modelBuilder.Entity<TagDefinition>()
            .HasMany(e => e.ImplicitTags)
            .WithMany();

        modelBuilder.Entity<Media>()
            .HasOne(m => m.CurrentUploadedFile)
            .WithOne()
            .HasForeignKey<Media>("CurrentUploadedFileId");
    }
}