aws-cloudfront-invalidator/src/cloudfront.js

A CLI tool that invalidates CloudFront distributions based on the S3 origin

process.env.AWS_SDK_LOAD_CONFIG = true

const aws = require('aws-sdk')
const region = aws.config.region
const cloudfront = new aws.CloudFront()

/**
 * @async
 * @method listDistributions A wrapper around Cf SDK listDistributions 
 * @return {Promise<{object}>} returns an object containing information about
 * all the distributions associated with the user account
 */
const listDistributions = () => (
  new Promise((resolve, reject) => {
    cloudfront.listDistributions({}, (err, data) => 
      err ? reject(err) : resolve(data))
  })
)

/**
 * @method getDistributionId takes the object outcome of @method listDistributions
 * and checks against the bucket name @param bucket
 * and returns the distributionId ID of that bucket
 * @param {object} list list distributions object to iterate over
 * @param {string} bucket name of the bucket to check against
 * @return {string} Id of the bucket
 */
const getDistributionId = (list, region, bucket) =>
{
  //Look for region specific bucket created by CloudFormation script
  let matchingItem = list.DistributionList.Items.find(e => 
      e.Origins.Items.find(z => z.DomainName === `${bucket}.s3-website-${region}.amazonaws.com`))

  if(!matchingItem) {
    //Look for manually created bucket without region in origin
    matchingItem = list.DistributionList.Items.find(e => 
        e.Origins.Items.find(z => z.DomainName === `${bucket}.s3.amazonaws.com`))
  }

  if (!matchingItem) {
    throw new Error(`No match found on the provided bucket ${bucket}`)
  }

  console.log(`Found distribution ${matchingItem.Id} for bucket ${bucket}.`)

  return matchingItem.Id
}

/**
 * @async
 * @method invalidateBucket invalidates the distributions of the bucket 
 * calls @method listDistributions and @method getDistributionId to retrieve 
 * the Id and CF SDK @method createInvalidation to create an invalidation
 * @param {string} bucket name of the bucket to invalidate
 * @return {Promise<{object}>} data object of @method createInvalidation
 */
const invalidateBucket = bucket => (
  new Promise((resolve, reject) => {
    listDistributions()
      .then(r => {
        cloudfront.createInvalidation({
          DistributionId: getDistributionId(r, region, bucket),
          InvalidationBatch: {
            CallerReference: Date.now().toString(),
            Paths: {
              Quantity: 1,
              Items: ['/*']
            }
          }
        }, (err, data) => err ? reject(err) : resolve(data))
      })
  })
)

module.exports.invalidateBucket = invalidateBucket