import { api } from "../trpc";
import { PostParams } from "../../singleton/DataHolder";

export function useVisitor() {
	const utils = api.useUtils();
	const postParams = PostParams.getInstance();

	const create = api.visitor.create.useMutation({
		onSuccess() {
			utils.visitor.invalidate();
			// utils.post.invalidate();
		},
	});
	/**
	 * 通用的乐观更新mutation工厂函数
	 * @param updateFn 更新数据的具体逻辑函数
	 * @returns 封装后的mutation配置对象
	 */
	const createOptimisticMutation = (
		updateFn: (item: any, variables: any) => any
	) => ({
		//在请求发送前执行本地数据预更新
		onMutate: async (variables: any) => {
			const previousDataList: any[] = [];
			// 动态生成参数列表，包括星标和其他参数

			const paramsList = postParams.getItems();
			console.log("paramsList.length", paramsList.length);
			// 遍历所有参数列表，执行乐观更新
			for (const params of paramsList) {
				// 取消可能的并发请求
				await utils.post.findManyWithCursor.cancel();
				// 获取并保存当前数据
				const previousData =
					utils.post.findManyWithCursor.getInfiniteData({
						...params,
					});
				previousDataList.push(previousData);
				// 执行乐观更新
				utils.post.findManyWithCursor.setInfiniteData(
					{
						...params,
					},
					(oldData) => {
						if (!oldData) return oldData;
						return {
							...oldData,
							pages: oldData.pages.map((page) => ({
								...page,
								items: (page.items as any).map((item) =>
									item.id === variables?.postId
										? updateFn(item, variables)
										: item
								),
							})),
						};
					}
				);
			}

			return { previousDataList };
		},
		// 错误处理：数据回滚
		onError: (_err: any, _variables: any, context: any) => {
			const paramsList = postParams.getItems();
			paramsList.forEach((params, index) => {
				if (context?.previousDataList?.[index]) {
					utils.post.findManyWithCursor.setInfiniteData(
						{ ...params },
						context.previousDataList[index]
					);
				}
			});
		},
		// 成功后的缓存失效
		onSuccess: async (_: any, variables: any) => {
			await Promise.all([
				utils.visitor.invalidate(),
				utils.post.findFirst.invalidate({
					where: {
						id: (variables as any)?.postId,
					},
				}),
				utils.post.findManyWithCursor.invalidate(),
			]);
		},
	});
	// 定义具体的mutation
	const read = api.visitor.create.useMutation(
		createOptimisticMutation((item) => ({
			...item,
			views: (item.views || 0) + 1,
			readed: true,
		}))
	);
	const like = api.visitor.create.useMutation(
		createOptimisticMutation((item) => ({
			...item,
			likes: (item.likes || 0) + 1,
			liked: true,
		}))
	);

	const addStar = api.visitor.create.useMutation(
		createOptimisticMutation((item) => ({
			...item,
			star: true,
		}))
	);

	const deleteStar = api.visitor.deleteMany.useMutation(
		createOptimisticMutation((item) => ({
			...item,
			star: false,
		}))
	);

	const deleteMany = api.visitor.deleteMany.useMutation({
		onSuccess() {
			utils.visitor.invalidate();
		},
	});

	const createMany = api.visitor.createMany.useMutation({
		onSuccess() {
			utils.visitor.invalidate();
			utils.message.invalidate();
			utils.post.invalidate();
		},
	});

	return {
		postParams,
		create,
		createMany,
		deleteMany,
		read,
		addStar,
		deleteStar,
		like,
	};
}