Function useGetShortUrl

packages/shared/src/hooks/utils/useGetShortUrl.ts:43–101

({
  query,
}: UseGetShortUrl = {})

};

export const useGetShortUrl = ({
  query,
}: UseGetShortUrl = {}): UseGetShortUrlResult => {
  const { user, isAuthReady } = useAuthContext();
  const queryClient = useQueryClient();
  const getProps = useCallback(
    (url: string, cid?: ReferralCampaignKey) =>
      getShortLinkProps(url, cid, user),
    [user],
  );

  const queryShortUrl = async (url: string) => {
    const res = await gqlClient.request(GET_SHORT_URL_QUERY, { url });
    return res.getShortUrl;
  };

  const getShortUrl = useCallback(
    async (url: string, cid?: ReferralCampaignKey) => {
      if (!url || !isAuthReady || !user) {
        return url;
      }

      const { trackedUrl, queryKey } = getProps(url, cid);

      try {
        return queryClient.fetchQuery({
          queryKey,
          queryFn: () => queryShortUrl(trackedUrl),
          staleTime: Infinity,
        });
      } catch (err) {
        return trackedUrl;
      }
    },
    [isAuthReady, user, getProps, queryClient],
  );

  const getTrackedUrl = useCallback(
    (url: string, cid?: ReferralCampaignKey) => {
      const { trackedUrl } = getProps(url, cid);
      return trackedUrl;
    },
    [getProps],
  );

  const isEnabled = query?.enabled ?? true;
  const { queryKey, trackedUrl } = query
    ? getProps(query.url, query.cid)
    : { queryKey: [], trackedUrl: '' };
  const { data: shareLink, isPending } = useQuery({
    queryKey,
    queryFn: () => queryShortUrl(trackedUrl),
    ...disabledRefetch,
    staleTime: Infinity,
    enabled: !!query?.url && isEnabled && !!user,
  });

  return { getShortUrl, getTrackedUrl, shareLink, isLoading: isPending };