All files / app/features/connections/friends useRespondToFriendRequest.ts

83.33% Statements 20/24
0% Branches 0/4
100% Functions 5/5
83.33% Lines 20/24

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 733x   3x 3x 3x                   3x 6x           6x   2x     2x 2x   2x         2x                                     2x     1x 1x         1x 1x 1x 1x         6x    
import { friendRequestKey, userKey } from "features/queryKeys";
import { Empty } from "google-protobuf/google/protobuf/empty_pb";
import { FriendRequest, User } from "proto/api_pb";
import { useMutation, useQueryClient } from "react-query";
import { service } from "service";
 
import { SetMutationError } from ".";
 
interface RespondToFriendRequestVariables {
  accept: boolean;
  friendRequest: FriendRequest.AsObject;
  setMutationError: SetMutationError;
}
 
export default function useRespondToFriendRequest() {
  const queryClient = useQueryClient();
  const {
    mutate: respondToFriendRequest,
    isLoading,
    isSuccess,
    reset,
  } = useMutation<Empty, Error, RespondToFriendRequestVariables>(
    ({ friendRequest, accept }) =>
      service.api.respondFriendRequest(friendRequest.friendRequestId, accept),
    {
      onMutate: async ({ setMutationError, friendRequest, accept }) => {
        setMutationError("");
        await queryClient.cancelQueries(friendRequestKey("received"));
 
        const cachedUser = queryClient.getQueryData<User.AsObject>([
          "user",
          friendRequest.userId,
        ]);
 
        Iif (cachedUser) {
          if (accept === true) {
            queryClient.setQueryData<User.AsObject>(
              userKey(friendRequest.userId),
              {
                ...cachedUser,
                friends: User.FriendshipStatus.FRIENDS,
              }
            );
          } else {
            queryClient.setQueryData<User.AsObject>(
              userKey(friendRequest.userId),
              {
                ...cachedUser,
                friends: User.FriendshipStatus.NOT_FRIENDS,
              }
            );
          }
        }
        return cachedUser;
      },
      onError: (error, { setMutationError, friendRequest }, cachedUser) => {
        setMutationError(error.message);
        Iif (cachedUser) {
          queryClient.setQueryData(userKey(friendRequest.userId), cachedUser);
        }
      },
      onSuccess: (_, { friendRequest }) => {
        queryClient.invalidateQueries("friendIds");
        queryClient.invalidateQueries(friendRequestKey("received"));
        queryClient.invalidateQueries(userKey(friendRequest.userId));
        queryClient.invalidateQueries("ping");
      },
    }
  );
 
  return { isLoading, isSuccess, reset, respondToFriendRequest };
}