using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Microsoft.AspNetCore.Identity.EntityFrameworkCore;
using Microsoft.AspNetCore.Identity;
using X1.Domain.Entities;
using X1.Domain.Entities.Logging;
using X1.Domain.Entities.Device;
using X1.Domain.Entities.NetworkProfile;
using X1.Domain.Entities.Terminal;
using X1.Domain.Entities.TestCase;

namespace X1.Infrastructure.Context;

/// <summary>
/// 应用程序数据库上下文
/// 继承自 IdentityDbContext 以支持身份认证
/// </summary>
public class AppDbContext : IdentityDbContext<AppUser, AppRole, string>
{
    private readonly ILogger<AppDbContext> _logger;

    /// <summary>
    /// 用户角色关系
    /// </summary>
    public new DbSet<UserRole> UserRoles { get; set; } = null!;

    /// <summary>
    /// 权限集合
    /// </summary>
    public DbSet<Permission> Permissions { get; set; } = null!;

    /// <summary>
    /// 角色权限关联集合
    /// </summary>
    public DbSet<RolePermission> RolePermissions { get; set; } = null!;

    /// <summary>
    /// 用户登录日志
    /// </summary>
    public DbSet<LoginLog> LoginLogs { get; set; } = null!;

    /// <summary>
    /// 协议日志集合
    /// </summary>
    public DbSet<ProtocolLog> ProtocolLogs { get; set; } = null!;

    /// <summary>
    /// 蜂窝设备集合
    /// </summary>
    public DbSet<CellularDevice> CellularDevices { get; set; } = null!;

    /// <summary>
    /// 协议版本集合
    /// </summary>
    public DbSet<ProtocolVersion> ProtocolVersions { get; set; } = null!;

    /// <summary>
    /// 蜂窝设备运行时状态集合
    /// </summary>
    public DbSet<CellularDeviceRuntime> CellularDeviceRuntimes { get; set; } = null!;

    /// <summary>
    /// RAN配置集合
    /// </summary>
    public DbSet<RAN_Configuration> RAN_Configurations { get; set; } = null!;

    /// <summary>
    /// 核心网配置集合
    /// </summary>
    public DbSet<CoreNetworkConfig> CoreNetworkConfigs { get; set; } = null!;

    /// <summary>
    /// IMS配置集合
    /// </summary>
    public DbSet<IMS_Configuration> IMS_Configurations { get; set; } = null!;

    /// <summary>
    /// 网络栈配置集合
    /// </summary>
    public DbSet<NetworkStackConfig> NetworkStackConfigs { get; set; } = null!;

    /// <summary>
    /// 栈与核心网/IMS绑定关系集合
    /// </summary>
    public DbSet<Stack_CoreIMS_Binding> Stack_CoreIMS_Bindings { get; set; } = null!;

    /// <summary>
    /// 终端服务集合
    /// </summary>
    public DbSet<TerminalService> TerminalServices { get; set; } = null!;

    /// <summary>
    /// 终端设备集合
    /// </summary>
    public DbSet<TerminalDevice> TerminalDevices { get; set; } = null!;

    /// <summary>
    /// ADB操作集合
    /// </summary>
    public DbSet<AdbOperation> AdbOperations { get; set; } = null!;

    /// <summary>
    /// AT操作集合
    /// </summary>
    public DbSet<AtOperation> AtOperations { get; set; } = null!;

    /// <summary>
    /// 用例步骤配置集合
    /// </summary>
    public DbSet<CaseStepConfig> CaseStepConfigs { get; set; } = null!;

    /// <summary>
    /// 测试用例流程集合
    /// </summary>
    public DbSet<TestCaseFlow> TestCaseFlows { get; set; } = null!;

    /// <summary>
    /// 测试用例节点集合
    /// </summary>
    public DbSet<TestCaseNode> TestCaseNodes { get; set; } = null!;

    /// <summary>
    /// 测试用例连线集合
    /// </summary>
    public DbSet<TestCaseEdge> TestCaseEdges { get; set; } = null!;

    /// <summary>
    /// IMSI注册记录集合
    /// </summary>
    public DbSet<ImsiRegistrationRecord> ImsiRegistrationRecords { get; set; } = null!;

    /// <summary>
    /// 测试场景集合
    /// </summary>
    public DbSet<TestScenario> TestScenarios { get; set; } = null!;

    /// <summary>
    /// 场景测试用例集合
    /// </summary>
    public DbSet<ScenarioTestCase> ScenarioTestCases { get; set; } = null!;

    /// <summary>
    /// 初始化数据库上下文
    /// </summary>
    /// <param name="options">数据库上下文选项</param>
    /// <param name="logger">日志记录器</param>
    public AppDbContext(DbContextOptions<AppDbContext> options, ILogger<AppDbContext> logger) 
        : base(options)
    {
        _logger = logger;
    }

    /// <summary>
    /// 配置实体模型
    /// </summary>
    /// <param name="modelBuilder">模型构建器</param>
    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);
        
        // 应用所有的实体配置
        modelBuilder.ApplyConfigurationsFromAssembly(typeof(AppDbContext).Assembly);

        // 忽略其他 Identity 相关的实体
        modelBuilder.Ignore<IdentityUserLogin<string>>();
        modelBuilder.Ignore<IdentityRoleClaim<string>>();
        modelBuilder.Ignore<IdentityUserClaim<string>>();
        modelBuilder.Ignore<IdentityUserToken<string>>();
        modelBuilder.Ignore<IdentityUserRole<string>>(); // 使用自定义的 UserRole 实体
       
    }
}