using AmtScanner.Api.Models;
using Microsoft.EntityFrameworkCore;

namespace AmtScanner.Api.Data;

public class AppDbContext : DbContext
{
    public AppDbContext(DbContextOptions<AppDbContext> options) : base(options)
    {
    }
    
    public DbSet<AmtDevice> AmtDevices { get; set; }
    public DbSet<AmtCredential> AmtCredentials { get; set; }
    public DbSet<HardwareInfo> HardwareInfos { get; set; }
    public DbSet<MemoryModule> MemoryModules { get; set; }
    public DbSet<StorageDevice> StorageDevices { get; set; }
    public DbSet<WindowsCredential> WindowsCredentials { get; set; }
    public DbSet<RemoteAccessToken> RemoteAccessTokens { get; set; }
    
    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);
        
        // 限制索引字段长度以适应 MySQL
        modelBuilder.Entity<AmtDevice>()
            .Property(d => d.IpAddress)
            .HasMaxLength(50);
        
        modelBuilder.Entity<AmtDevice>()
            .HasIndex(d => d.IpAddress)
            .IsUnique();
        
        modelBuilder.Entity<AmtCredential>()
            .Property(c => c.Name)
            .HasMaxLength(200);
        
        modelBuilder.Entity<AmtCredential>()
            .HasIndex(c => c.Name);
        
        // HardwareInfo 配置
        modelBuilder.Entity<HardwareInfo>()
            .HasOne(h => h.Device)
            .WithMany()
            .HasForeignKey(h => h.DeviceId)
            .OnDelete(DeleteBehavior.Cascade);
        
        modelBuilder.Entity<HardwareInfo>()
            .HasIndex(h => h.DeviceId);
        
        modelBuilder.Entity<HardwareInfo>()
            .HasIndex(h => h.LastUpdated);
        
        // MemoryModule 配置
        modelBuilder.Entity<MemoryModule>()
            .HasOne(m => m.HardwareInfo)
            .WithMany(h => h.MemoryModules)
            .HasForeignKey(m => m.HardwareInfoId)
            .OnDelete(DeleteBehavior.Cascade);
        
        // StorageDevice 配置
        modelBuilder.Entity<StorageDevice>()
            .HasOne(s => s.HardwareInfo)
            .WithMany(h => h.StorageDevices)
            .HasForeignKey(s => s.HardwareInfoId)
            .OnDelete(DeleteBehavior.Cascade);
        
        // WindowsCredential 配置
        modelBuilder.Entity<WindowsCredential>()
            .Property(w => w.Name)
            .HasMaxLength(200);
        
        modelBuilder.Entity<WindowsCredential>()
            .HasIndex(w => w.Name);
        
        // RemoteAccessToken 配置
        modelBuilder.Entity<RemoteAccessToken>()
            .Property(t => t.Token)
            .HasMaxLength(64);
        
        modelBuilder.Entity<RemoteAccessToken>()
            .HasIndex(t => t.Token)
            .IsUnique();
        
        modelBuilder.Entity<RemoteAccessToken>()
            .HasOne(t => t.Device)
            .WithMany()
            .HasForeignKey(t => t.DeviceId)
            .OnDelete(DeleteBehavior.Cascade);
        
        modelBuilder.Entity<RemoteAccessToken>()
            .HasOne(t => t.WindowsCredential)
            .WithMany()
            .HasForeignKey(t => t.WindowsCredentialId)
            .OnDelete(DeleteBehavior.SetNull);
    }
}