import { isFunction } from 'lodash';
import { reactive, toRaw } from 'vue';

// 分页钩子函数
interface Options {
	page?: number;
	size?: number;
	fetchFun: (_arg: any) => Promise<any>;
	params?: Record<any, any>;
	firstLoading?: boolean;
	beforeRequest?(params: Record<any, any>): Record<any, any>;
	afterRequest?(res: Record<any, any>): void;
}

export function usePaging(options: Options) {
	const { page = 1, size = 15, fetchFun, params = {}, firstLoading = false, beforeRequest, afterRequest } = options;
	// 记录分页初始参数
	const paramsInit: Record<any, any> = Object.assign({}, toRaw(params));
	// 分页数据
	const pager = reactive({
		current: page,
		size,
		loading: firstLoading,
		count: 0,
		total: 0,
		lists: [] as any[],
		extend: {} as Record<any, any>,
	});
	// 请求分页接口
	const getLists = () => {
		pager.loading = true;
		let requestParams = params;
		if (isFunction(beforeRequest)) {
			requestParams = beforeRequest(params);
		}
		return fetchFun({
			current: pager.current,
			size: pager.size,
			...requestParams,
		})
			.then(({ data }) => {
				pager.count = data?.total;
				pager.total = data?.total;
				pager.lists = data?.records;
				pager.extend = data?.extend;
				if (isFunction(afterRequest)) {
					afterRequest(data);
				}
				return Promise.resolve(data);
			})
			.catch((err: any) => {
				return Promise.reject(err);
			})
			.finally(() => {
				pager.loading = false;
			});
	};
	// 重置为第一页
	const resetPage = () => {
		pager.current = 1;
		getLists();
	};
	// 重置参数
	const resetParams = () => {
		Object.keys(paramsInit).forEach((item) => {
			params[item] = paramsInit[item];
		});
		getLists();
	};
	return {
		pager,
		getLists,
		resetParams,
		resetPage,
	};
}