using MongoDB.Bson;
using MongoDB.Bson.Serialization.Attributes;
using MongoDB.Bson.Serialization.IdGenerators;
using MongoDB.Driver;

namespace MongoChecks;

internal class Program
{
    private static async Task Main(string[] args)
    {
        MongoClient client = new("mongodb://localhost:27017/?replicaSet=rs0");
        var db = client.GetDatabase("MongoTests");
        var collection = db.GetCollection<Document>("Docs");

        Document template = new Document
        {
            Values = [4, 2, 6],
            SubDocuments = [
                new(){ Value = 4 },
                new(){ Value = 2 },
                new(){ Value = 6 },
                ]
        };
        Document doc = template with { };
        Document doc2 = template with { };

        await collection.InsertOneAsync(doc);

        await collection.UpdateOneAsync(
            dbDoc => dbDoc.Id == doc.Id,
            Builders<Document>.Update.PushEach(
                doc => doc.Values,
                [1, 5, 3],
                sort: Builders<int>.Sort.Ascending(val => val)));

        var dbDocs = await collection.FindAsync(
            dbDoc => dbDoc.Id == doc.Id);

        var res = await dbDocs.ToListAsync();
        Console.WriteLine("Push-sorted values: " + string.Join(", ", res.Single().Values));

        await collection.UpdateOneAsync(
            dbDoc => dbDoc.Id == doc.Id,
            Builders<Document>.Update.PushEach(
                doc => doc.SubDocuments,
                [
                    new SubDocument() { Value = 1},
                    new SubDocument() { Value = 5},
                    new SubDocument() { Value = 3}],
                sort: Builders<SubDocument>.Sort.Ascending(subDoc => subDoc.Value)));

        dbDocs = await collection.FindAsync(
            dbDoc => dbDoc.Id == doc.Id);

        res = await dbDocs.ToListAsync();
        Console.WriteLine("Push sorted sub-docs: " + string.Join(", ", res.Single().SubDocuments));

        await collection.InsertOneAsync(doc2);

        await collection.UpdateOneAsync(
            dbDoc => dbDoc.Id == doc2.Id,
            Builders<Document>.Update.PushEach(
                doc => doc.Values,
                [],
                sort: Builders<int>.Sort.Ascending(val => val)));

        dbDocs = await collection.FindAsync(
            dbDoc => dbDoc.Id == doc2.Id);

        res = await dbDocs.ToListAsync();
        Console.WriteLine("Sorted values: " + string.Join(", ", res.Single().Values));

        await collection.UpdateOneAsync(
            dbDoc => dbDoc.Id == doc2.Id,
            Builders<Document>.Update.PushEach(
                doc => doc.SubDocuments,
                [],
                sort: Builders<SubDocument>.Sort.Ascending(val => val.Value)));

        dbDocs = await collection.FindAsync(
            dbDoc => dbDoc.Id == doc2.Id);

        res = await dbDocs.ToListAsync();
        Console.WriteLine("Sorted sub-docs: " + string.Join(", ", res.Single().SubDocuments));
    }
}

internal record Document
{
    [BsonId(IdGenerator = typeof(ObjectIdGenerator))]
    public ObjectId Id { get; set; }    public List<int> Values { get; set; } = [];
    public List<SubDocument> SubDocuments { get; set; } = [];
}

internal record SubDocument
{
    public int Value { get; set; }
}