using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Runtime.CompilerServices;
using System.Threading;
using System.Threading.Tasks;
using Dpz.Core.Entity.Base;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using MongoDB.Driver;

namespace Dpz.Core.MongodbAccess;

public sealed class Repository<T> : IRepository<T>
    where T : IBaseEntity
{
    private readonly MongodbAccess<T> _access;

    public Repository(string? connectionString)
    {
        _access = new MongodbAccess<T>(connectionString);
    }

    [ActivatorUtilitiesConstructor]
    public Repository(IConfiguration configuration)
    {
        var connectionString = configuration.GetConnectionString("mongodb");
        _access = new MongodbAccess<T>(connectionString);
    }

    public Repository(IConfiguration configuration, string collectionName)
    {
        var connectionString = configuration.GetConnectionString("mongodb");
        _access = string.IsNullOrEmpty(collectionName)
            ? new MongodbAccess<T>(connectionString)
            : new MongodbAccess<T>(connectionString, collectionName);
    }

    public Repository(string? connectionString, string collectionName)
    {
        _access = string.IsNullOrEmpty(collectionName)
            ? new MongodbAccess<T>(connectionString)
            : new MongodbAccess<T>(connectionString, collectionName);
    }

    public IQueryable<T> MongodbQueryable => _access.MongoQueryable;

    public IMongoCollection<T> Collection => _access.Collection;

    public IQueryable<T> SearchFor(Expression<Func<T, bool>> predicate)
    {
        return _access.MongoQueryable.Where(predicate);
    }

    public IFindFluent<T, T> SearchFor(FilterDefinition<T> filter)
    {
        return Collection.Find(filter);
    }

    public IFindFluent<T, T> SearchFor(FilterDefinition<T> filter, FindOptions options)
    {
        return Collection.Find(filter, options);
    }

    public async IAsyncEnumerable<T> SearchForAsync(
        FilterDefinition<T> filter,
        [EnumeratorCancellation] CancellationToken cancellationToken = default
    )
    {
        var result = await Collection.FindAsync(filter, cancellationToken: cancellationToken);
        while (await result.MoveNextAsync(cancellationToken))
        {
            foreach (var item in result.Current)
            {
                yield return item;
            }
        }
    }

    public async IAsyncEnumerable<T> SearchForAsync(
        FilterDefinition<T> filter,
        FindOptions<T> options,
        [EnumeratorCancellation] CancellationToken cancellationToken = default
    )
    {
        var result = await Collection.FindAsync(filter, options, cancellationToken);
        while (await result.MoveNextAsync(cancellationToken))
        {
            foreach (var item in result.Current)
            {
                yield return item;
            }
        }
    }

    public async Task<T?> FindAsync(object id, CancellationToken cancellationToken = default)
    {
        var filter = MongodbExtensions.GetIdPropertyFilter<T>(id);
        return await (
            await Collection.FindAsync(filter, cancellationToken: cancellationToken)
        ).SingleOrDefaultAsync(cancellationToken: cancellationToken);
    }

    public async Task InsertAsync(T entity, CancellationToken cancellationToken = default)
    {
        await Collection.InsertOneAsync(entity, cancellationToken: cancellationToken);
    }

    public async Task InsertAsync(
        IReadOnlyCollection<T> source,
        CancellationToken cancellationToken = default
    )
    {
        if (source.Count > 0)
        {
            await Collection.InsertManyAsync(source, cancellationToken: cancellationToken);
        }
    }

    public async Task<DeleteResult> DeleteAsync(
        Expression<Func<T, bool>> filter,
        CancellationToken cancellationToken = default
    )
    {
        var result = await Collection.DeleteManyAsync(filter, cancellationToken);
        return result;
    }

    public async Task<DeleteResult> DeleteAsync(
        FilterDefinition<T> filter,
        CancellationToken cancellationToken = default
    )
    {
        var result = await Collection.DeleteManyAsync(filter, cancellationToken);
        return result;
    }

    public async Task<DeleteResult> DeleteAsync(
        object id,
        CancellationToken cancellationToken = default
    )
    {
        var filter = MongodbExtensions.GetIdPropertyFilter<T>(id);
        return await Collection.DeleteOneAsync(filter, cancellationToken);
    }

    public async Task<UpdateResult> UpdateAsync(
        Expression<Func<T, bool>> predicate,
        UpdateDefinition<T> update,
        CancellationToken cancellationToken = default
    )
    {
        var result = await Collection.UpdateManyAsync(
            predicate,
            update,
            cancellationToken: cancellationToken
        );
        return result;
    }

    public async Task<ReplaceOneResult> UpdateAsync(
        T entity,
        CancellationToken cancellationToken = default
    )
    {
        var filter = MongodbExtensions.GetIdPropertyFilter(entity);

        var result = await Collection.ReplaceOneAsync(
            filter,
            entity,
            cancellationToken: cancellationToken
        );
        return result;
    }

    public async Task<BulkWriteResult<T>> UpdateAsync(
        IEnumerable<T> entities,
        CancellationToken cancellationToken = default
    )
    {
        var writes = entities
            .Select(x => new ReplaceOneModel<T>(MongodbExtensions.GetIdPropertyFilter(x), x))
            .ToList();

        return await Collection.BulkWriteAsync(writes, cancellationToken: cancellationToken);
    }

    public IMongoDatabase Database => _access.Database;
}