using System.Linq.Expressions;
using SqlSugar;

namespace Entity.Base
{
    /// <summary>
    /// 分页泛型集合
    /// </summary>
    /// <typeparam name="TEntity"></typeparam>
    public class SqlSugarPagedList<TEntity>
    {
        /// <summary>
        /// 页码
        /// </summary>
        public int Page { get; set; }

        /// <summary>
        /// 页容量
        /// </summary>
        public int PageSize { get; set; }

        /// <summary>
        /// 总条数
        /// </summary>
        public int Total { get; set; }

        /// <summary>
        /// 总页数
        /// </summary>
        public int TotalPages { get; set; }

        /// <summary>
        /// 当前页集合
        /// </summary>
        public IEnumerable<TEntity> Items { get; set; }

        /// <summary>
        /// 是否有上一页
        /// </summary>
        public bool HasPrevPage { get; set; }

        /// <summary>
        /// 是否有下一页
        /// </summary>
        public bool HasNextPage { get; set; }
    }

    /// <summary>
    /// 分页拓展类
    /// </summary>
    public static class SqlSugarPagedExtensions
    {
        /// <summary>
        /// 分页拓展
        /// </summary>
        /// <param name="query"><see cref="ISugarQueryable{T}"/>对象</param>
        /// <param name="pageIndex">当前页码，从1开始</param>
        /// <param name="pageSize">页码容量</param>
        /// <param name="expression">查询结果 Select 表达式</param>
        /// <returns></returns>
        public static SqlSugarPagedList<TResult> ToPagedList<TEntity, TResult>(this ISugarQueryable<TEntity> query,
            int pageIndex, int pageSize,
            Expression<Func<TEntity, TResult>> expression)
        {
            var total = 0;
            var items = query.ToPageList(pageIndex, pageSize, ref total, expression);
            return CreateSqlSugarPagedList(items, total, pageIndex, pageSize);
        }

        /// <summary>
        /// 分页拓展
        /// </summary>
        /// <param name="query"><see cref="ISugarQueryable{TEntity}"/>对象</param>
        /// <param name="pageIndex">当前页码，从1开始</param>
        /// <param name="pageSize">页码容量</param>
        /// <returns></returns>
        public static SqlSugarPagedList<TEntity> ToPagedList<TEntity>(this ISugarQueryable<TEntity> query,
            int pageIndex, int pageSize)
        {
            var total = 0;
            var items = query.ToPageList(pageIndex, pageSize, ref total);
            return CreateSqlSugarPagedList(items, total, pageIndex, pageSize);
        }

        /// <summary>
        /// 分页拓展
        /// </summary>
        /// <param name="query"><see cref="ISugarQueryable{TEntity}"/>对象</param>
        /// <param name="pageIndex">当前页码，从1开始</param>
        /// <param name="pageSize">页码容量</param>
        /// <param name="expression">查询结果 Select 表达式</param>
        /// <returns></returns>
        public static async Task<SqlSugarPagedList<TResult>> ToPagedListAsync<TEntity, TResult>(
            this ISugarQueryable<TEntity> query, int pageIndex, int pageSize,
            Expression<Func<TEntity, TResult>> expression)
        {
            RefAsync<int> total = 0;
            var items = await query.ToPageListAsync(pageIndex, pageSize, total, expression);
            return CreateSqlSugarPagedList(items, total, pageIndex, pageSize);
        }

        /// <summary>
        /// 分页拓展
        /// </summary>
        /// <param name="query"><see cref="ISugarQueryable{TEntity}"/>对象</param>
        /// <param name="pageIndex">当前页码，从1开始</param>
        /// <param name="pageSize">页码容量</param>
        /// <returns></returns>
        public static async Task<SqlSugarPagedList<TEntity>> ToPagedListAsync<TEntity>(
            this ISugarQueryable<TEntity> query, int pageIndex, int pageSize)
        {
            RefAsync<int> total = 0;
            var items = await query.ToPageListAsync(pageIndex, pageSize, total);
            return CreateSqlSugarPagedList(items, total, pageIndex, pageSize);
        }

        /// <summary>
        /// 分页拓展
        /// </summary>
        /// <param name="list">集合对象</param>
        /// <param name="pageIndex">当前页码，从1开始</param>
        /// <param name="pageSize">页码容量</param>
        /// <returns></returns>
        public static SqlSugarPagedList<TEntity> ToPagedList<TEntity>(this IEnumerable<TEntity> list, int pageIndex,
            int pageSize)
        {
            var total = list.Count();
            var items = list.Skip((pageIndex - 1) * pageSize).Take(pageSize).ToList();
            return CreateSqlSugarPagedList(items, total, pageIndex, pageSize);
        }

        /// <summary>
        /// 创建 <see cref="SqlSugarPagedList{TEntity}"/> 对象
        /// </summary>
        /// <typeparam name="TEntity"></typeparam>
        /// <param name="items">分页内容的对象集合</param>
        /// <param name="total">总条数</param>
        /// <param name="pageIndex">当前页码，从1开始</param>
        /// <param name="pageSize">页码容量</param>
        /// <returns></returns>
        public static SqlSugarPagedList<TEntity> CreateSqlSugarPagedList<TEntity>(IEnumerable<TEntity> items, int total,
            int pageIndex, int pageSize)
        {
            var totalPages = pageSize > 0 ? (int)Math.Ceiling(total / (double)pageSize) : 0;
            return new SqlSugarPagedList<TEntity>
            {
                Page = pageIndex,
                PageSize = pageSize,
                Items = items,
                Total = total,
                TotalPages = totalPages,
                HasNextPage = pageIndex < totalPages,
                HasPrevPage = pageIndex - 1 > 0
            };
        }
    }
}