using Masuit.Tools.Core;
using Microsoft.EntityFrameworkCore;

namespace Masuit.Tools.Models;

public static partial class IQueryableExt
{
	/// <summary>
	/// 生成分页集合
	/// </summary>
	/// <typeparam name="T"></typeparam>
	/// <param name="query"></param>
	/// <param name="page">当前页</param>
	/// <param name="size">页大小</param>
	/// <returns></returns>
	public static async Task<PagedList<T>> ToPagedListAsync<T>(this IQueryable<T> query, int page, int size)
	{
		var totalCount = await query.CountAsync();
		if (1L * page * size > totalCount)
		{
			page = (int)Math.Ceiling(totalCount / (size * 1.0));
		}

		if (page <= 0)
		{
			page = 1;
		}

		var list = await query.Skip(size * (page - 1)).Take(size).ToListAsync();
		return new PagedList<T>(list, page, size, totalCount);
	}

	/// <summary>
	/// 生成分页集合
	/// </summary>
	/// <typeparam name="T"></typeparam>
	/// <param name="query"></param>
	/// <param name="page">当前页</param>
	/// <param name="size">页大小</param>
	/// <returns></returns>
	public static async Task<PagedList<T>> ToPagedListNoLockAsync<T>(this IQueryable<T> query, int page, int size)
	{
		var totalCount = await query.CountAsync();
		if (1L * page * size > totalCount)
		{
			page = (int)Math.Ceiling(totalCount / (size * 1.0));
		}

		if (page <= 0)
		{
			page = 1;
		}

		var list = await query.Skip(size * (page - 1)).Take(size).ToListWithNoLockAsync();
		return new PagedList<T>(list, page, size, totalCount);
	}

	/// <summary>
	/// 生成分页集合
	/// </summary>
	/// <typeparam name="T"></typeparam>
	/// <param name="query"></param>
	/// <param name="page">当前页</param>
	/// <param name="size">页大小</param>
	/// <returns></returns>
	public static PagedList<T> ToPagedList<T>(this IQueryable<T> query, int page, int size)
	{
		var totalCount = query.Count();
		if (1L * page * size > totalCount)
		{
			page = (int)Math.Ceiling(totalCount / (size * 1.0));
		}

		if (page <= 0)
		{
			page = 1;
		}

		var list = query.Skip(size * (page - 1)).Take(size).ToList();
		return new PagedList<T>(list, page, size, totalCount);
	}

	/// <summary>
	/// 生成分页集合
	/// </summary>
	/// <typeparam name="T"></typeparam>
	/// <param name="query"></param>
	/// <param name="page">当前页</param>
	/// <param name="size">页大小</param>
	/// <returns></returns>
	public static PagedList<T> ToPagedListNoLock<T>(this IQueryable<T> query, int page, int size)
	{
		var totalCount = query.Count();
		if (1L * page * size > totalCount)
		{
			page = (int)Math.Ceiling(totalCount / (size * 1.0));
		}

		if (page <= 0)
		{
			page = 1;
		}

		var list = query.Skip(size * (page - 1)).Take(size).ToListWithNoLock();
		return new PagedList<T>(list, page, size, totalCount);
	}

	/// <summary>
	/// 生成分页集合
	/// </summary>
	/// <typeparam name="T"></typeparam>
	/// <param name="query"></param>
	/// <param name="page">当前页</param>
	/// <param name="size">页大小</param>
	/// <returns></returns>
	public static PagedList<T> ToPagedList<T>(this IEnumerable<T> query, int page, int size)
	{
		var totalCount = query.Count();
		if (1L * page * size > totalCount)
		{
			page = (int)Math.Ceiling(totalCount / (size * 1.0));
		}

		if (page <= 0)
		{
			page = 1;
		}

		var list = query.Skip(size * (page - 1)).Take(size).ToList();
		return new PagedList<T>(list, page, size, totalCount);
	}
}