simplify-cli/cognito.js

Simplify CLI - Easy Deployment

require('cross-fetch/polyfill');
var AmazonCognitoIdentity = require('amazon-cognito-identity-js');
var CognitoUserPool = AmazonCognitoIdentity.CognitoUserPool;
const { v4 } = require('uuid');
const ApplicationStorage = require('./storage')

var poolData = {
    UserPoolId: process.env.COGNITO_USERPOOL_ID || 'us-east-1_UAf0SxZ5c',
    ClientId: process.env.COGNITO_CLIENT_ID || '72nrnku0iajiqlahhvkvhqgln2',
    Storage: ApplicationStorage
};
var userPool = process.env.ENABLE_TRACKING_DATA_FOR_ANALYTICS ? new AmazonCognitoIdentity.CognitoUserPool(poolData) : null;

const registerUser = function (name, email, password) {
    var attributeList = [];
    attributeList.push(new AmazonCognitoIdentity.CognitoUserAttribute({
        Name: 'email',
        Value: email,
    }));
    attributeList.push(new AmazonCognitoIdentity.CognitoUserAttribute({
        Name: 'name',
        Value: name,
    }));
    attributeList.push(new AmazonCognitoIdentity.CognitoUserAttribute({
        Name: 'subscription',
        Value: 'Basic',
    }));
    return new Promise(function (resolve, reject) {
        userPool.signUp(v4(), password, attributeList, null, function (err, result) {
            if (err) {
                reject(err.message || JSON.stringify(err));
            } else {
                var cognitoUser = result.user;
                resolve(cognitoUser);
            }
        });
    })
}

const confirmRegistration = function (username, code) {
    var userData = {
        Username: username,
        Pool: userPool,
        Storage: ApplicationStorage
    };
    var cognitoUser = new AmazonCognitoIdentity.CognitoUser(userData);
    return new Promise(function (resolve, reject) {
        cognitoUser.confirmRegistration(code, true, function (err, result) {
            if (err) {
                reject(err.message || JSON.stringify(err));
            } else {
                resolve(result);
            }
        });
    })
}

const getCurrentSession = function () {
    if (process.env.ENABLE_TRACKING_DATA_FOR_ANALYTICS) {
        var cognitoUser = userPool.getCurrentUser();
        if (cognitoUser != null) {
            return new Promise(function (resolve, reject) {
                cognitoUser.getSession(function (err, session) {
                    if (err) {
                        reject(err.message || JSON.stringify(err));
                    } else {
                        if (session.isValid()) {
                            resolve(session)
                        } else {
                            cognitoUser.refreshSession(session.getRefreshToken(), (err, session) => {
                                if (err) {
                                    reject(err.message || JSON.stringify(err));
                                } else {
                                    resolve(session)
                                }
                            })
                        }
                    }
                });
            })
        } else {
            return Promise.reject(`Session is not found`)
        }
    } else {
        return Promise.resolve()
    }
}

const authenticate = function (email, password) {
    var authenticationData = {
        Username: email,
        Password: password,
    };
    var authenticationDetails = new AmazonCognitoIdentity.AuthenticationDetails(
        authenticationData
    );
    var userData = {
        Username: email,
        Pool: userPool,
        Storage: ApplicationStorage
    };
    var cognitoUser = new AmazonCognitoIdentity.CognitoUser(userData);
    return new Promise(function (resolve, reject) {
        cognitoUser.authenticateUser(authenticationDetails, {
            onSuccess: function (result) {
                resolve(result)
            },
            onFailure: function (err) {
                reject(err.message || JSON.stringify(err));
            },
        });
    })
}

const userSignOut = function (username) {
    var cognitoUser = new AmazonCognitoIdentity.CognitoUser({
        Username: username,
        Pool: userPool,
        Storage: ApplicationStorage
    });
    cognitoUser.signOut();
}

module.exports = {
    authenticate,
    registerUser,
    confirmRegistration,
    getCurrentSession,
    userSignOut
}