using ZB.MOM.WW.CBDD.Bson;
using ZB.MOM.WW.CBDD.Core;
using ZB.MOM.WW.CBDD.Core.Collections;
using ZB.MOM.WW.CBDD.Core.Compression;
using ZB.MOM.WW.CBDD.Core.Indexing;
using ZB.MOM.WW.CBDD.Core.Metadata;
using ZB.MOM.WW.CBDD.Core.Storage;

namespace ZB.MOM.WW.CBDD.Shared;

/// <summary>
/// Test context with manual collection initialization
/// (Source Generator will automate this in the future)
/// </summary>
public partial class TestDbContext : DocumentDbContext
{
    public DocumentCollection<ObjectId, AnnotatedUser> AnnotatedUsers { get; set; } = null!;
    public DocumentCollection<OrderId, Order> Orders { get; set; } = null!;
    public DocumentCollection<ObjectId, TestDocument> TestDocuments { get; set; } = null!;
    public DocumentCollection<ObjectId, OrderDocument> OrderDocuments { get; set; } = null!;
    public DocumentCollection<ObjectId, ComplexDocument> ComplexDocuments { get; set; } = null!;
    public DocumentCollection<ObjectId, User> Users { get; set; } = null!;
    public DocumentCollection<ObjectId, ComplexUser> ComplexUsers { get; set; } = null!;
    public DocumentCollection<int, AutoInitEntity> AutoInitEntities { get; set; } = null!;
    public DocumentCollection<int, Person> People { get; set; } = null!;
    public DocumentCollection<ObjectId, PersonV2> PeopleV2 { get; set; } = null!;
    public DocumentCollection<int, Product> Products { get; set; } = null!;
    public DocumentCollection<int, IntEntity> IntEntities { get; set; } = null!;
    public DocumentCollection<string, StringEntity> StringEntities { get; set; } = null!;
    public DocumentCollection<Guid, GuidEntity> GuidEntities { get; set; } = null!;
    public DocumentCollection<string, CustomKeyEntity> CustomKeyEntities { get; set; } = null!;
    public DocumentCollection<int, AsyncDoc> AsyncDocs { get; set; } = null!;
    public DocumentCollection<int, SchemaUser> SchemaUsers { get; set; } = null!;
    public DocumentCollection<ObjectId, VectorEntity> VectorItems { get; set; } = null!;
    public DocumentCollection<ObjectId, GeoEntity> GeoItems { get; set; } = null!;

    // Source Generator Feature Tests
    public DocumentCollection<ObjectId, DerivedEntity> DerivedEntities { get; set; } = null!;
    public DocumentCollection<ObjectId, EntityWithComputedProperties> ComputedPropertyEntities { get; set; } = null!;
    public DocumentCollection<ObjectId, EntityWithAdvancedCollections> AdvancedCollectionEntities { get; set; } = null!;
    public DocumentCollection<ObjectId, EntityWithPrivateSetters> PrivateSetterEntities { get; set; } = null!;
    public DocumentCollection<ObjectId, EntityWithInitSetters> InitSetterEntities { get; set; } = null!;

    // Circular Reference Tests
    public DocumentCollection<ObjectId, Employee> Employees { get; set; } = null!;
    public DocumentCollection<ObjectId, CategoryRef> CategoryRefs { get; set; } = null!;
    public DocumentCollection<ObjectId, ProductRef> ProductRefs { get; set; } = null!;

    // Nullable String Id Test (UuidEntity scenario with inheritance)
    public DocumentCollection<string, MockCounter> MockCounters { get; set; } = null!;

    // Temporal Types Test (DateTimeOffset, TimeSpan, DateOnly, TimeOnly)
    public DocumentCollection<ObjectId, TemporalEntity> TemporalEntities { get; set; } = null!;

    public TestDbContext(string databasePath)
        : this(databasePath, PageFileConfig.Default, CompressionOptions.Default)
    {
    }

    public TestDbContext(string databasePath, CompressionOptions compressionOptions)
        : this(databasePath, PageFileConfig.Default, compressionOptions)
    {
    }

    public TestDbContext(string databasePath, PageFileConfig pageFileConfig)
        : this(databasePath, pageFileConfig, CompressionOptions.Default)
    {
    }

    public TestDbContext(
        string databasePath,
        PageFileConfig pageFileConfig,
        CompressionOptions? compressionOptions,
        MaintenanceOptions? maintenanceOptions = null)
        : base(databasePath, pageFileConfig, compressionOptions, maintenanceOptions)
    {
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<AnnotatedUser>();
        modelBuilder.Entity<User>().ToCollection("users");
        modelBuilder.Entity<ComplexUser>().ToCollection("complex_users");
        modelBuilder.Entity<AutoInitEntity>().ToCollection("auto_init_entities");
        modelBuilder.Entity<Person>().ToCollection("people_collection").HasIndex(p => p.Age);
        modelBuilder.Entity<PersonV2>().ToCollection("peoplev2_collection").HasIndex(p => p.Age);
        modelBuilder.Entity<Product>().ToCollection("products_collection").HasIndex(p => p.Price);
        modelBuilder.Entity<IntEntity>().HasKey(e => e.Id);
        modelBuilder.Entity<StringEntity>().HasKey(e => e.Id);
        modelBuilder.Entity<GuidEntity>().HasKey(e => e.Id);
        modelBuilder.Entity<CustomKeyEntity>().HasKey(e => e.Code);
        modelBuilder.Entity<AsyncDoc>().ToCollection("async_docs");
        modelBuilder.Entity<SchemaUser>().ToCollection("schema_users").HasKey(e => e.Id);
        modelBuilder.Entity<TestDocument>();
        modelBuilder.Entity<OrderDocument>();
        modelBuilder.Entity<ComplexDocument>();

        modelBuilder.Entity<VectorEntity>()
            .ToCollection("vector_items")
            .HasVectorIndex(x => x.Embedding, dimensions: 3, metric: VectorMetric.L2, name: "idx_vector");

        modelBuilder.Entity<GeoEntity>()
            .ToCollection("geo_items")
            .HasSpatialIndex(x => x.Location, name: "idx_spatial");

        modelBuilder.Entity<Order>()
            .HasKey(x => x.Id)
            .HasConversion<OrderIdConverter>();

        // Source Generator Feature Tests
        modelBuilder.Entity<DerivedEntity>().ToCollection("derived_entities");
        modelBuilder.Entity<EntityWithComputedProperties>().ToCollection("computed_property_entities");
        modelBuilder.Entity<EntityWithAdvancedCollections>().ToCollection("advanced_collection_entities");
        modelBuilder.Entity<EntityWithPrivateSetters>().ToCollection("private_setter_entities");
        modelBuilder.Entity<EntityWithInitSetters>().ToCollection("init_setter_entities");

        // Circular Reference Tests
        modelBuilder.Entity<Employee>().ToCollection("employees");
        modelBuilder.Entity<CategoryRef>().ToCollection("category_refs");
        modelBuilder.Entity<ProductRef>().ToCollection("product_refs");

        // Nullable String Id Test (UuidEntity scenario)
        modelBuilder.Entity<MockCounter>().ToCollection("mock_counters").HasKey(e => e.Id);

        // Temporal Types Test
        modelBuilder.Entity<TemporalEntity>().ToCollection("temporal_entities").HasKey(e => e.Id);
    }

    public void ForceCheckpoint()
    {
        Engine.Checkpoint();
    }

    public StorageEngine Storage => Engine;
}