using System;
using System.Collections.Generic;
using System.Runtime.CompilerServices;
using System.Threading;
using System.Threading.Tasks;
using MongoDB.Bson;
using MongoDB.Driver;

namespace Dpz.Core.MongodbAccess;

public class RepositoryBase(string? connectionString) : IRepositoryBase
{
    private readonly AnyAccess _access = string.IsNullOrWhiteSpace(connectionString)
        ? throw new ArgumentNullException(nameof(connectionString))
        : new AnyAccess(connectionString);

    public IMongoDatabase Database => _access.Database;

    public async Task<List<string>> GetAllCollectionAsync()
    {
        using var cursor = await Database.ListCollectionNamesAsync();
        var list = await cursor.ToListAsync();
        return list;
    }

    public async IAsyncEnumerable<BsonDocument> SearchForAsync(
        string collectionName,
        FilterDefinition<BsonDocument>? filter = null,
        FindOptions<BsonDocument>? options = null,
        [EnumeratorCancellation] CancellationToken cancellationToken = default
    )
    {
        filter ??= Builders<BsonDocument>.Filter.Empty;
        var result = await Database
            .GetCollection<BsonDocument>(collectionName)
            .FindAsync(filter, options, cancellationToken);
        while (await result.MoveNextAsync(cancellationToken))
        {
            foreach (var item in result.Current)
            {
                yield return item;
            }
        }
    }

    public async Task InsertAsync(string collectionName, ICollection<BsonDocument> source)
    {
        await Database.GetCollection<BsonDocument>(collectionName).InsertManyAsync(source);
    }

    public async Task DeleteAllAsync(string collectionName)
    {
        await Database
            .GetCollection<BsonDocument>(collectionName)
            .DeleteManyAsync(Builders<BsonDocument>.Filter.Empty);
    }

    private class AnyAccess(string? connectionString) : MongodbAccess(connectionString);
}