All files / app/features/badges hooks.ts

50% Statements 9/18
0% Branches 0/2
28.57% Functions 2/7
50% Lines 8/16

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 414x       4x 4x   33x 33x 21x 21x                     33x                                      
import { badgesKey, badgeUsersKey } from "features/queryKeys";
import { RpcError } from "grpc-web";
import { ListBadgeUsersRes } from "proto/api_pb";
import { Badge } from "proto/resources_pb";
import { useInfiniteQuery, useQuery } from "react-query";
import { service } from "service";
 
export const useBadges = () => {
  const { data, ...rest } = useQuery(badgesKey, async () => {
    const result = await service.resources.getBadges();
    return result.badgesList.reduce(
      (badgesResult, badge) => {
        badgesResult.badges[badge.id] = badge;
        return badgesResult;
      },
      {
        badges: {} as { [id: string]: Badge.AsObject },
      },
    );
  });
 
  return {
    badges: data?.badges,
    ...rest,
  };
};
 
export function useBadgeUsers(badgeId: string) {
  const query = useInfiniteQuery<ListBadgeUsersRes.AsObject, RpcError>({
    queryKey: badgeUsersKey({ badgeId }),
    queryFn: ({ pageParam }) =>
      service.api.listBadgeUsers({ badgeId, pageToken: pageParam }),
    getNextPageParam: (lastPage) => lastPage.nextPageToken || undefined,
  });
  const badgeUserIds = query.data?.pages.flatMap((res) => res.userIdsList);
  return {
    ...query,
    badgeUserIds,
  };
}