Function usePoll

packages/shared/src/hooks/usePoll.tsx:24–143

({ post }: { post: Post })

import { postLogEvent } from '../lib/feed';

const usePoll = ({ post }: { post: Post }) => {
  const { user } = useAuthContext();
  const { logEvent } = useLogContext();
  const { queryKey } = useActiveFeedContext();
  const queryClient = useQueryClient();
  const { toggleGroup, getGroupStatus } = useNotificationSettings();
  const { displayToast } = useToastNotification();

  const { mutate, isPending: isCastingVote } = useMutation({
    mutationFn: (optionId: string) => votePoll({ postId: post.id, optionId }),
    onMutate: (optionId: string) => {
      const adsQueryKey = queryKey ? [RequestKey.Ads, ...queryKey] : null;
      const currentFeedData =
        queryClient.getQueryData<InfiniteData<FeedData>>(queryKey);
      const currentAdsData =
        queryClient.getQueryData<InfiniteData<Ad>>(adsQueryKey);

      const updatedOptions = post.pollOptions.map((option) => {
        if (option.id === optionId) {
          return { ...option, numVotes: option.numVotes + 1 };
        }
        return option;
      });

      const postUpdate = {
        numPollVotes: post.numPollVotes + 1,
        pollOptions: updatedOptions,
        userState: {
          ...post.userState,
          pollOption: { id: optionId },
        },
      };

      updatePostCache(queryClient, post.id, postUpdate);

      if (currentFeedData) {
        const updateFeed = updateCachedPagePost(queryKey, queryClient);
        const feedData =
          queryClient.getQueryData<InfiniteData<FeedData>>(queryKey);

        const { pageIndex, index } = findIndexOfPostInData(
          feedData,
          post.id,
          false,
        );

        if (pageIndex > -1 && index > -1) {
          const currentPost = getFeedApiItemPost(
            feedData.pages[pageIndex].page.edges[index].node,
          );

          if (currentPost) {
            updateFeed(pageIndex, index, {
              ...currentPost,
              ...postUpdate,
            });
          }
        }