bcgov/citz-imb-staff-purchasing-reimbursement

/* eslint-disable camelcase */
import { useContext, useMemo } from 'react';

import { AuthContext } from '../KeycloakProvider';
import decodeJWT from '../utils/decodeJWT';
import AuthActionType from './authActions';
import Constants from '../../constants/Constants';

const { SET_TOKEN } = AuthActionType;

/**
 * A custom hook that provides authentication-related functionality to other components.
 * @returns {Object} - An object containing authentication-related functions
 * and the current authentication state.
 */
const useAuthService = () => {
  // Get the authentication state and dispatch function from the authentication context.
  const { state, dispatch } = useContext<any>(AuthContext);

  // Use useMemo to memoize the returned object and prevent unnecessary re-renders.
  return useMemo(() => {
    const getLoginURL = () => `${Constants.BACKEND_URL}/oauth/login`;
    const getLogoutURL = () => `${Constants.BACKEND_URL}/oauth/logout`;

    // Sets the user information in the authentication state using a JWT token.
    const setUserInfo = (token: string) => {
      const decodedToken = decodeJWT(token);
      dispatch({
        type: SET_TOKEN,
        payload: { accessToken: token, userInfo: decodedToken },
      });
    };

    // Get a new access token using the refresh token.
    const refreshAccessToken = async () => {
      const response = await fetch(`${Constants.BACKEND_URL}/oauth/token`, {
        method: 'POST',
        credentials: 'include',
      });

      if (response.ok) {
        const { access_token } = await response.json();
        const decodedToken = decodeJWT(access_token);
        dispatch({
          type: SET_TOKEN,
          payload: { accessToken: access_token, userInfo: decodedToken },
        });
      }
    };

    return {
      getLoginURL,
      getLogoutURL,
      setUserInfo,
      refreshAccessToken,
      state,
    };
  }, [state]);
};

export default useAuthService;