@keybittech/awayto/app/src/webapp/hooks/useCognitoUser.ts

Deploy a fully-featured application in about 10 minutes that is primed for quick development. Do business, impress a client with a quick demo, finish your poc with time to spare; all easily achievable with Awayto.

import { useEffect, useState } from 'react';
import { fromCognitoIdentityPool } from '@aws-sdk/credential-provider-cognito-identity';
import { CognitoIdentityClient } from "@aws-sdk/client-cognito-identity";

import { CognitoUser, CognitoUserPool, ILoginActionTypes, Region, UserPoolId, ClientId, IdentityPoolId } from 'awayto';
import { useAct } from 'awayto-hooks';

const { AUTH_SUCCESS, AUTH_DENIAL } = ILoginActionTypes;

/**
 * Use this hook to get access to cognito functionality once the user has logged in, or to check if the user is logged in.
 * 
 * ```
 * import { useCognitoUser } from 'awayto';
 * 
 * const cognitoUser = useCognitoUser();
 * 
 * cognitoUser.getSession();
 * ```
 * 
 * @category Hooks
 */
export function useCognitoUser(): CognitoUser {
  const act = useAct();
  const [cognitoUser, setCognitoUser] = useState<CognitoUser>({ signInUserSession: undefined } as CognitoUser);

  useEffect(() => {
    async function getCognitoUser() {
      if (!UserPoolId || !ClientId)
        throw new Error('Configuration error: userPoolId missing during getCognitoUser.');

      const pool = new CognitoUserPool({ UserPoolId, ClientId });
      const user = pool.getCurrentUser();

      if (!user) {
        act(AUTH_DENIAL, { isLoggedIn: false });
        return undefined;
      }

      await user.getSession();

      act(AUTH_SUCCESS, { isLoggedIn: true, username: user.getUsername() });

      user.isLoggedIn = true;
      user.attributes = await user.getUserAttributes();

      if (IdentityPoolId && user.signInUserSession) {
        const credentials = fromCognitoIdentityPool({
          identityPoolId: IdentityPoolId,
          client: new CognitoIdentityClient({ region: Region }),
          logins: {
            [`cognito-idp.${Region}.amazonaws.com/${UserPoolId}`]: user.signInUserSession.idToken.token
          }
        })
        user.credentials = await credentials();
      }

      setCognitoUser(user)
    }
    void getCognitoUser();
  }, [])

  return cognitoUser;
}