All files / packages/hooks/src useDateQueries.ts

100% Statements 15/15
100% Branches 18/18
100% Functions 4/4
100% Lines 14/14

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                1x       1x                                 1x       17x 17x 17x   17x                     1x   4x   4x         4x 4x   4x 4x                    
import { useCallback, useMemo } from "react";
 
import { API_ENDPOINTS } from "@repo/constants/endpoints";
 
import { QUERY_CACHE_TIME } from "./constants";
import { useApiQueryClients } from "./useApiClients";
import { type HttpReadableClient, type RequestGuard, useApiDataQuery } from "./useApiQuery";
 
export const DATE_ENDPOINTS = {
  HOLIDAYS: API_ENDPOINTS.HOLIDAYS,
};
 
export const DATE_QUERY_KEYS = {
  HOLIDAYS: ["holidays"] as const,
};
 
export interface SharedHolidays {
  holidays: string[];
  makeUpWorkdays: string[];
}
 
interface UseGetHolidaysOptions {
  httpClient?: HttpReadableClient;
  requestGuard?: RequestGuard;
  enabled?: boolean;
  staleTime?: number;
  gcTime?: number;
}
 
export const useGetHolidays = <THolidays = SharedHolidays>(
  year: number,
  options?: UseGetHolidaysOptions,
) => {
  const { mainHttp, requestGuard } = useApiQueryClients();
  const resolvedHttpClient = options?.httpClient ?? mainHttp;
  const resolvedRequestGuard = options?.requestGuard ?? requestGuard;
 
  return useApiDataQuery<THolidays>({
    queryKey: [...DATE_QUERY_KEYS.HOLIDAYS, year],
    endpoint: API_ENDPOINTS.HOLIDAYS_BY_YEAR(year),
    httpClient: resolvedHttpClient,
    requestGuard: resolvedRequestGuard,
    enabled: options?.enabled,
    staleTime: options?.staleTime ?? QUERY_CACHE_TIME,
    gcTime: options?.gcTime ?? QUERY_CACHE_TIME,
  });
};
 
export const useSharedHolidayCalendar = (year: number, options?: UseGetHolidaysOptions) => {
  // Do not trigger global unauthorized navigation for optional holiday lookup.
  const passthroughGuard = useCallback(async <T>(requestFn: () => Promise<T>) => requestFn(), []);
 
  const resolvedOptions = {
    ...options,
    requestGuard: options?.requestGuard ?? passthroughGuard,
  };
 
  const { data: currentYearHolidays } = useGetHolidays(year, resolvedOptions);
  const { data: nextYearHolidays } = useGetHolidays(year + 1, resolvedOptions);
 
  return useMemo(
    () => ({
      holidays: [...(currentYearHolidays?.holidays ?? []), ...(nextYearHolidays?.holidays ?? [])],
      makeUpWorkdays: [
        ...(currentYearHolidays?.makeUpWorkdays ?? []),
        ...(nextYearHolidays?.makeUpWorkdays ?? []),
      ],
    }),
    [currentYearHolidays, nextYearHolidays],
  );
};