using System.Linq.Expressions;
using Microsoft.EntityFrameworkCore;
using MilliSim.Common.Entities;

namespace MilliSim.Data;

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

    public DbSet<User> Users => Set<User>();
    public DbSet<Project> Projects => Set<Project>();
    public DbSet<ProjectCategory> ProjectCategories => Set<ProjectCategory>();
    public DbSet<PlatformType> PlatformTypes => Set<PlatformType>();
    public DbSet<FeatureTag> FeatureTags => Set<FeatureTag>();
    public DbSet<ProjectTagRelation> ProjectTagRelations => Set<ProjectTagRelation>();
    public DbSet<ProjectScreenshot> ProjectScreenshots => Set<ProjectScreenshot>();
    public DbSet<ProjectPackage> ProjectPackages => Set<ProjectPackage>();
    public DbSet<Favorite> Favorites => Set<Favorite>();
    public DbSet<ExperienceLog> ExperienceLogs => Set<ExperienceLog>();
    public DbSet<Authorization> Authorizations => Set<Authorization>();
    public DbSet<Banner> Banners => Set<Banner>();
    public DbSet<Partner> Partners => Set<Partner>();
    public DbSet<Consultation> Consultations => Set<Consultation>();
    public DbSet<Notification> Notifications => Set<Notification>();
    public DbSet<NotificationRead> NotificationReads => Set<NotificationRead>();
    public DbSet<OperationLog> OperationLogs => Set<OperationLog>();
    public DbSet<SystemSetting> SystemSettings => Set<SystemSetting>();
    public DbSet<Backup> Backups => Set<Backup>();
    public DbSet<WakeProtocol> WakeProtocols => Set<WakeProtocol>();
    public DbSet<PendingUpload> PendingUploads => Set<PendingUpload>();

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        base.OnModelCreating(modelBuilder);

        // 软删除全局过滤器
        foreach (var entityType in modelBuilder.Model.GetEntityTypes())
        {
            if (typeof(BaseEntity).IsAssignableFrom(entityType.ClrType))
            {
                var parameter = Expression.Parameter(entityType.ClrType, "e");
                var property = Expression.Property(parameter, nameof(BaseEntity.IsDeleted));
                var falseValue = Expression.Constant(false);
                var comparison = Expression.Equal(property, falseValue);
                var lambda = Expression.Lambda(comparison, parameter);
                modelBuilder.Entity(entityType.ClrType).HasQueryFilter((dynamic)lambda);
            }
        }

        // 唯一索引
        modelBuilder.Entity<User>().HasIndex(u => u.Username).IsUnique();
        modelBuilder.Entity<SystemSetting>().HasIndex(s => s.Key).IsUnique();

        // PendingUpload 索引（用于垃圾资源清理查询）
        modelBuilder.Entity<PendingUpload>().HasIndex(p => p.ObjectKey);
        modelBuilder.Entity<PendingUpload>().HasIndex(p => p.ConsumedAt);

        // 字符集
        modelBuilder.Entity<Project>().Property(p => p.Content).HasColumnType("longtext");

        // 精度
        foreach (var entityType in modelBuilder.Model.GetEntityTypes())
        {
            foreach (var property in entityType.GetProperties())
            {
                if (property.ClrType == typeof(DateTime))
                {
                    property.SetColumnType("datetime");
                }
            }
        }
    }
}