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

namespace Dpz.Core.MongodbAccess;

public class Repository<T> : IRepository<T> where T : IBaseEntity, new()
{
    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 IMongoQueryable<T> MongodbQueryable => _access.MongoQueryable;

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

    public IMongoQueryable<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 async IAsyncEnumerable<T> SearchForAsync(FilterDefinition<T> filter)
    {
        var result = await Collection.FindAsync(filter);
        while (await result.MoveNextAsync())
        {
            foreach (var item in result.Current)
            {
                yield return item;
            }
        }
    }

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


    public async Task InsertAsync(params T[] source)
    {
        if(source.Any())
            await Collection.InsertManyAsync(source);
    }

    public async Task InsertAsync(IReadOnlyCollection<T> source)
    {
        if(source.Any())
            await Collection.InsertManyAsync(source);
    }


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

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

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

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

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

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

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

        return await Collection.BulkWriteAsync(writes);
    }

    public IMongoDatabase Database => _access.Database;
}