using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Threading;
using System.Threading.Tasks;
using Dpz.Core.Entity.Base;
using MongoDB.Driver;

namespace Dpz.Core.MongodbAccess;

public interface IRepository
{
    IMongoDatabase Database { get; }
}

public interface IRepository<T> : IRepository
    where T : IBaseEntity
{
    /// <summary>
    /// 获取 该实体Mongodb的集合
    /// </summary>
    IMongoCollection<T> Collection { get; }

    /// <summary>
    /// Queryable
    /// </summary>
    IQueryable<T> MongodbQueryable { get; }

    /// <summary>
    /// 查询数据
    /// </summary>
    /// <param name="predicate"></param>
    /// <returns></returns>
    IQueryable<T> SearchFor(Expression<Func<T, bool>> predicate);

    /// <summary>
    /// 数据查询
    /// </summary>
    /// <param name="filter"></param>
    /// <returns></returns>
    IFindFluent<T, T> SearchFor(FilterDefinition<T> filter);

    /// <summary>
    ///
    /// </summary>
    /// <param name="filter"></param>
    /// <param name="options"></param>
    /// <returns></returns>
    IFindFluent<T, T> SearchFor(FilterDefinition<T> filter, FindOptions options);

    /// <summary>
    /// 异步 数据查询
    /// </summary>
    /// <param name="filter"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    IAsyncEnumerable<T> SearchForAsync(
        FilterDefinition<T> filter,
        CancellationToken cancellationToken = default
    );

    /// <summary>
    /// 异步 数据查询
    /// </summary>
    /// <param name="filter"></param>
    /// <param name="options"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    IAsyncEnumerable<T> SearchForAsync(
        FilterDefinition<T> filter,
        FindOptions<T> options,
        CancellationToken cancellationToken = default
    );

    #region 异步方法

    /// <summary>
    /// (异步)根据 Id 获取单条记录，不存在将会返回null
    /// </summary>
    /// <param name="id"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task<T?> FindAsync(object id, CancellationToken cancellationToken = default);

    /// <summary>
    /// 插入数据
    /// </summary>
    /// <param name="entity"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task InsertAsync(T entity, CancellationToken cancellationToken = default);

    /// <summary>
    /// 插入数据
    /// </summary>
    /// <param name="source"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task InsertAsync(IReadOnlyCollection<T> source, CancellationToken cancellationToken = default);

    /// <summary>
    /// 删除数据
    /// </summary>
    /// <param name="filter"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task<DeleteResult> DeleteAsync(
        Expression<Func<T, bool>> filter,
        CancellationToken cancellationToken = default
    );

    /// <summary>
    /// 删除数据
    /// </summary>
    /// <param name="filter"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task<DeleteResult> DeleteAsync(
        FilterDefinition<T> filter,
        CancellationToken cancellationToken = default
    );

    /// <summary>
    /// 根据ID删除
    /// </summary>
    /// <param name="id"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task<DeleteResult> DeleteAsync(object id, CancellationToken cancellationToken = default);

    /// <summary>
    /// 根据查询条件更新数据
    /// </summary>
    /// <param name="predicate"></param>
    /// <param name="update"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task<UpdateResult> UpdateAsync(
        Expression<Func<T, bool>> predicate,
        UpdateDefinition<T> update,
        CancellationToken cancellationToken = default
    );

    /// <summary>
    /// 根据实体修改数据
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="entity"></param>
    /// <param name="cancellationToken"></param>
    /// <exception cref="ArgumentException">If the entity no exists property 'id',then will throw exception.</exception>
    Task<ReplaceOneResult> UpdateAsync(T entity, CancellationToken cancellationToken = default);

    /// <summary>
    /// 批量修改实体
    /// </summary>
    /// <param name="entities"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    Task<BulkWriteResult<T>> UpdateAsync(
        IEnumerable<T> entities,
        CancellationToken cancellationToken = default
    );

    #endregion
}