accounts-frontend/packages/app/services/authFlow/PasswordState.ts

import logger from 'app/services/logger';
import { getCredentials } from 'app/components/auth/reducer';
import { getActiveAccount, getAvailableAccounts } from 'app/components/accounts/reducer';

import AbstractState from './AbstractState';
import ChooseAccountState from './ChooseAccountState';
import CompleteState from './CompleteState';
import ForgotPasswordState from './ForgotPasswordState';
import LoginState from './LoginState';
import MfaState from './MfaState';
import { AuthContext } from './AuthFlow';

export default class PasswordState extends AbstractState {
    enter(context: AuthContext): Promise<void> | void {
        const { login, isTotpRequired } = getCredentials(context.getState());

        if (isTotpRequired) {
            context.setState(new MfaState());
        } else if (login) {
            context.navigate('/password');
        } else {
            context.setState(new CompleteState());
        }
    }

    resolve(
        context: AuthContext,
        {
            password,
            rememberMe,
        }: {
            password: string;
            rememberMe: boolean;
        },
    ): Promise<void> | void {
        const { login, returnUrl } = getCredentials(context.getState());

        return context
            .run('login', {
                password,
                rememberMe,
                login,
            })
            .then(() => {
                const { isTotpRequired } = getCredentials(context.getState());

                if (isTotpRequired) {
                    return context.setState(new MfaState());
                }

                if (returnUrl) {
                    context.navigate(returnUrl);

                    return;
                }

                return context.setState(new CompleteState());
            })
            .catch((err = {}) => err.errors || logger.warn('Error logging in', err));
    }

    reject(context: AuthContext): void {
        context.setState(new ForgotPasswordState());
    }

    goBack(context: AuthContext): void {
        const state = context.getState();
        const { isRelogin } = getCredentials(state);

        if (isRelogin) {
            const availableAccounts = getAvailableAccounts(state);
            const accountToRemove = getActiveAccount(state);

            if (availableAccounts.length === 1 || !accountToRemove) {
                context.run('logout');
                context.run('setLogin', null);
                context.setState(new LoginState());
            } else {
                const accountToReplace = availableAccounts.find(({ id }) => id !== accountToRemove.id);

                context.run('activateAccount', accountToReplace);
                context.run('removeAccount', accountToRemove);
                context.setState(new ChooseAccountState());
            }
        } else {
            context.run('setLogin', null);
            context.setState(new LoginState());
        }
    }
}