accounts-frontend/src/services/authFlow/PasswordState.js

// @flow
import logger from 'services/logger';
import { getCredentials } from 'components/auth/reducer';
import { getActiveAccount, getAvailableAccounts } from '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 type { AuthContext } from './AuthFlow';

export default class PasswordState extends AbstractState {
    enter(context: AuthContext) {
        const {login} = getCredentials(context.getState());

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

    resolve(
        context: AuthContext,
        {
            password,
            rememberMe
        }: {
            password: string,
            rememberMe: bool
        }
    ) {
        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) {
        context.setState(new ForgotPasswordState());
    }

    goBack(context: AuthContext) {
        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());
        }
    }
}