import type { SubscriptionWithRoom } from '@rocket.chat/ui-contexts';
import { useEndpoint, useToastMessageDispatch, useUserId } from '@rocket.chat/ui-contexts';
import { useMutation, useQueryClient } from '@tanstack/react-query';
import { useTranslation } from 'react-i18next';

import { toggleFavoriteRoom } from '../../../lib/mutationEffects/room';
import { subscriptionsQueryKeys } from '../../../lib/queryKeys';

export const useToggleFavoriteMutation = () => {
	const { t } = useTranslation();
	const dispatchToastMessage = useToastMessageDispatch();
	const uid = useUserId();

	const toggleFavorite = useEndpoint('POST', '/v1/rooms.favorite');
	const queryClient = useQueryClient();

	return useMutation({
		mutationFn: async ({ roomId, favorite }: { roomId: string; favorite: boolean; roomName: string }) => {
			await toggleFavorite({ roomId, favorite });
		},
		onMutate: ({ roomId, favorite }) => {
			queryClient.setQueryData<SubscriptionWithRoom | null>(subscriptionsQueryKeys.subscription(roomId), (subscription) =>
				subscription
					? {
							...subscription,
							f: favorite,
						}
					: undefined,
			);
		},
		onSuccess: (_data, { roomId, favorite, roomName }) => {
			toggleFavoriteRoom(roomId, favorite, uid);
			dispatchToastMessage({
				type: 'success',
				message: favorite
					? t('__roomName__was_added_to_favorites', { roomName })
					: t('__roomName__was_removed_from_favorites', { roomName }),
			});
		},
		onError: (error) => {
			dispatchToastMessage({ type: 'error', message: error });
		},
		onSettled: (_data, _error, { roomId }) => {
			queryClient.invalidateQueries({ queryKey: subscriptionsQueryKeys.subscription(roomId) });
		},
	});
};