clear_cognito_pool

clear_cognito_pool

#!/usr/bin/env node

const { USER_POOL_ID } = process.env;
if (!USER_POOL_ID) throw new Error("Missing USER_POOL_ID");

const AWS = require("aws-sdk");
const cognito = new AWS.CognitoIdentityServiceProvider();

const attribute => ({ Attributes }, name) =>
  Attributes.filter(a => a.Name === name)[0].Value;

const shouldDelete = user =>
  attribute(user, "email").indexOf("dev+") > -1;

(async function() {
  while (true) {
    const { Users } = await cognito.listUsers({ UserPoolId: USER_POOL_ID }).promise();
    let deletedUsers = 0;
    for (const user of Users) {
      if (shouldDelete(user)) {
        await cognito.adminDeleteUser({
          UserPoolId: USER_POOL_ID,
          Username: user.Username
        }).promise();
        deletedUsers++;
      }
    }
    if (deletedUsers === 0) break;
  }
})();