@sungly/cognito-cli/src/cmds/user/index.js

CLI for all Cognito user life cycle management

import prompt from 'prompt';
import Promise from "bluebird";

import config from '../../config';
import { userService } from '../../services';
import {logger} from '../../util';

class UserCmd {
    createUser() {
        config.requiredAttributeList.push('password');

        const requiredAttributes = config.requiredAttributeList.map((attr) => {
            return {
                name: attr,
                required: true,
            };
        });

        prompt.get(requiredAttributes, async (err, result) => {
            let username = result.username ? result.username : result.email;
            const password = result.password;

            delete result.password;

            const attributes = Object.keys(result).map((key) => {
                return {
                    Name: key,
                    Value: result[key],
                };
            });

            await userService.createUser({
                username,
                password,
                attributes,
            });
        });
    }

    confirmUserSignup() {
        const requiredAttributes = [
            {
                name: 'username',
                required: true,
            },
            {
                name: 'code',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { username, code } = result;

            await userService.confirmSignUp({
                username,
                code,
            });
        });
    }

    resendConfirmationCode() {
        const requiredAttributes = [
            {
                name: 'username',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { username } = result;

            const res = await userService.resendConfirmationCode({
                username,
            });

            logger.info(res);
        });
    }

    forgotPassword() {
        const requiredAttributes = [
            {
                name: 'username',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { username } = result;

            await userService.forgotPassword({
                username,
            });
        });
    }

    setUserPassword() {
        const requiredAttributes = [
            {
                name: 'username',
                required: true,
            },
            {
                name: 'password',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { username, password } = result;

            await userService.setUserPassword({
                username,
                newPassword: password,
            });
        });
    }

    getUserProfile() {
        const requiredAttributes = [
            {
                name: 'username',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { username } = result;

            await userService.getUserProfile({
                username,
            });
        });
    }

    verifyUserEmail() {
        const requiredAttributes = [
            {
                name: 'email',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { email } = result;

            await userService.verifyUserEmail({
                username: email,
            });
        });
    }

    disableUser() {
        const requiredAttributes = [
            {
                name: 'username',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { username } = result;

            await userService.disableUser({
                username,
            });
        });
    }

    enableUser() {
        const requiredAttributes = [
            {
                name: 'username',
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            const { username } = result;

            await userService.enableUser({
                username,
            });
        });
    }

    async batchDeleteUsers() {
        const requiredAttributes = [
            {
                name: 'Confirmation',
                description: `Are you sure you want to delete? yes/no`,
                pattern: /^(?:yes|Yes|no|No)$/,
                required: true,
            },
        ];

        prompt.get(requiredAttributes, async (err, result) => {
            if (result.Confirmation.toLowerCase() === 'yes') {
                let users;

                do {
                    const results = await userService.listUsers();
                    users = results.Users.map(users => users.Username);
                    logger.info(`Number of users found ${users.length}`);

                    await Promise.map(users, async username => {
                        await userService.deleteUsers(username)
                    }, {concurrency: 3});

                    logger.info(`Successfully deleted ${users.length}`);
                } while (users.length !== 0);
            }
        });
    }
}

export default UserCmd;