All files / utils currentUser.js

100% Statements 24/24
100% Branches 8/8
100% Functions 6/6
100% Lines 22/22

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          370x 370x     60x 60x 55x 248x   1x   55x 55x   4x 4x                   266x 266x 266x 1x 1x 1x   266x                   634x   613x           29x     584x    
import { useMutation, useQuery, useQueryClient } from "react-query";
import axios from "axios";
import { useNavigate } from "react-router-dom";
 
export function useCurrentUser() {
  let rolesList = ["ERROR_GETTING_ROLES"];
  return useQuery(
    "current user",
    async () => {
      try {
        const response = await axios.get("/api/currentUser");
        try {
          rolesList = response.data.roles.map((r) => r.authority);
        } catch (e) {
          console.error("Error getting roles: ", e);
        }
        response.data = { ...response.data, rolesList: rolesList };
        return { loggedIn: true, root: response.data };
      } catch (e) {
        console.error("Error invoking axios.get: ", e);
        return { loggedIn: false, root: null };
      }
    },
    {
      initialData: { loggedIn: false, root: null, initialData: true },
    },
  );
}
 
export function useLogout() {
  const queryClient = useQueryClient();
  const navigate = useNavigate();
  const mutation = useMutation(async () => {
    await axios.post("/logout");
    await queryClient.resetQueries("current user", { exact: true });
    navigate("/");
  });
  return mutation;
}
 
export function hasRole(currentUser, role) {
  // The following hack is because there is some bug in terms of the
  // shape of the data returned by useCurrentUser.  Is there a separate
  // data level, or not?
 
  // We will file an issue to track that down and then remove this hack
 
  if (currentUser == null) return false;
 
  if (
    "data" in currentUser &&
    "root" in currentUser.data &&
    currentUser.data.root != null &&
    "rolesList" in currentUser.data.root
  ) {
    return currentUser.data.root.rolesList.includes(role);
  }
 
  return currentUser.root?.rolesList?.includes(role);
}