All files / react-app/src/hooks useAvailableStates.ts

100% Statements 14/14
100% Branches 12/12
100% Functions 7/7
100% Lines 11/11

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                          24x 24x 24x 560x         20x 560x               10x 560x 560x 112x       560x      
import { useMemo } from "react";
import { FULL_CENSUS_STATES, STATE_ROLES } from "shared-types";
import { UserRole } from "shared-types/events/legacy-user";
 
import { StateAccess } from "@/api";
 
import { useStateAccessMap } from "../api/useStateAccessMap";
import { useFeatureFlag } from "./useFeatureFlag";
 
export function useAvailableStates(
  roleToRequest: UserRole,
  stateAccessList: StateAccess[] | undefined,
) {
  const isNewUserRoleDisplay = useFeatureFlag("SHOW_USER_ROLE_UPDATE");
  const stateAccessMap = useStateAccessMap(stateAccessList ?? []);
  return useMemo(() => {
    const formatOption = ({ label, value }: (typeof FULL_CENSUS_STATES)[number]) => ({
      label: isNewUserRoleDisplay ? `${label}, ${value}` : label,
      value,
    });
 
    if (!stateAccessList) {
      return FULL_CENSUS_STATES.map(({ label, value }) => ({
        label: isNewUserRoleDisplay ? `${label}, ${value}` : label,
        value,
      }));
    }
 
    // Check if user has all possible roles for this state (Feature Flag ON)
    // Show states user doesn't have access to for role being requested (Feature Flag OFF)
    const statesToRequest = (stateCode: string) => {
      const activeStates = stateAccessMap[stateCode] ?? new Set<UserRole>();
      return isNewUserRoleDisplay
        ? !STATE_ROLES.every((role) => activeStates.has(role))
        : !activeStates.has(roleToRequest);
    };
 
    return FULL_CENSUS_STATES.filter(({ value }) => statesToRequest(value)).map(formatOption);
  }, [roleToRequest, stateAccessList, stateAccessMap, isNewUserRoleDisplay]);
}