serverless/test/utils/cognito.js

'use strict';

const awsLog = require('log').get('aws');
const awsRequest = require('@serverless/test/aws-request');

function createUserPool(name, config = {}) {
  const params = Object.assign({}, { PoolName: name }, config);
  return awsRequest('CognitoIdentityServiceProvider', 'createUserPool', params);
}

function createUserPoolClient(name, userPoolId) {
  const params = {
    ClientName: name,
    UserPoolId: userPoolId,
    ExplicitAuthFlows: ['USER_PASSWORD_AUTH'],
  };
  return awsRequest('CognitoIdentityServiceProvider', 'createUserPoolClient', params);
}

function deleteUserPool(name) {
  return findUserPoolByName(name).then(pool =>
    awsRequest('CognitoIdentityServiceProvider', 'deleteUserPool', { UserPoolId: pool.Id })
  );
}

function deleteUserPoolById(poolId) {
  return awsRequest('CognitoIdentityServiceProvider', 'deleteUserPool', { UserPoolId: poolId });
}

function findUserPoolByName(name) {
  awsLog.debug('find cognito user pool by name %s', name);

  const params = {
    MaxResults: 60,
  };

  const pools = [];
  function recursiveFind(nextToken) {
    if (nextToken) params.NextToken = nextToken;
    return awsRequest('CognitoIdentityServiceProvider', 'listUserPools', params).then(result => {
      pools.push(...result.UserPools.filter(pool => pool.Name === name));
      if (result.NextToken) return recursiveFind(result.NextToken);
      switch (pools.length) {
        case 0:
          return null;
        case 1:
          return pools[0];
        default:
          throw new Error(`Found more than one pool named '${name}'`);
      }
    });
  }

  return recursiveFind();
}

function findUserPools() {
  const params = { MaxResults: 60 };

  const pools = [];
  function recursiveFind(nextToken) {
    if (nextToken) params.NextToken = nextToken;
    return awsRequest('CognitoIdentityServiceProvider', 'listUserPools', params).then(result => {
      pools.push(...result.UserPools.filter(pool => pool.Name.includes(' CUP ')));
      if (result.NextToken) return recursiveFind(result.NextToken);
      return null;
    });
  }

  return recursiveFind().then(() => pools);
}

function describeUserPool(userPoolId) {
  return awsRequest('CognitoIdentityServiceProvider', 'describeUserPool', {
    UserPoolId: userPoolId,
  }).then(result => {
    awsLog.debug('cognito.describeUserPool %s %j', userPoolId, result);
    return result;
  });
}

function createUser(userPoolId, username, password) {
  const params = {
    UserPoolId: userPoolId,
    Username: username,
    TemporaryPassword: password,
  };
  return awsRequest('CognitoIdentityServiceProvider', 'adminCreateUser', params);
}

function setUserPassword(userPoolId, username, password) {
  const params = {
    UserPoolId: userPoolId,
    Username: username,
    Password: password,
    Permanent: true,
  };
  return awsRequest('CognitoIdentityServiceProvider', 'adminSetUserPassword', params);
}

function initiateAuth(clientId, username, password) {
  const params = {
    ClientId: clientId,
    AuthFlow: 'USER_PASSWORD_AUTH',
    AuthParameters: {
      USERNAME: username,
      PASSWORD: password,
    },
  };
  return awsRequest('CognitoIdentityServiceProvider', 'initiateAuth', params);
}

module.exports = {
  createUserPool,
  deleteUserPool,
  deleteUserPoolById,
  findUserPoolByName,
  findUserPools,
  describeUserPool,
  createUserPoolClient,
  createUser,
  setUserPassword,
  initiateAuth,
};