import { useMutation, useQueryClient } from "@tanstack/react-query";
import { toast } from "sonner";
import { authClient } from "@/lib/auth-client";
import { userKeys } from "./keys";

interface UpdateUserParams {
	name?: string;
	image?: string;
}

export async function updateUser(params: UpdateUserParams) {
	const { data, error } = await authClient.updateUser(params);
	if (error) throw new Error(error.message);

	return data;
}
export type UpdateUserData = Awaited<ReturnType<typeof updateUser>>;

export const useUpdateUserMutation = () => {
	const queryClient = useQueryClient();

	return useMutation({
		mutationFn: updateUser,
		onSuccess: async () => {
			await queryClient.invalidateQueries({
				queryKey: userKeys.session(),
			});
			toast.success("User updated successfully!");
		},
		onError: (error: any) => {
			toast.error(error.message || "Failed to update user");
		},
	});
};