serverless/lib/plugins/aws/customResources/resources/s3/lib/bucket.js

Serverless Framework - Build web, mobile and IoT applications with serverless architectures using AWS Lambda, Azure Functions, Google CloudFunctions & more

'use strict';

const crypto = require('crypto');
const { awsRequest } = require('../../utils');

function generateId(functionName, bucketConfig) {
  const md5 = crypto.createHash('md5').update(JSON.stringify(bucketConfig)).digest('hex');
  return `${functionName}-${md5}`;
}

function createFilter(config) {
  const rules = config.Rules;
  if (rules && rules.length) {
    const FilterRules = rules.map((rule) => {
      const key = Object.keys(rule)[0];
      const Name = key.toLowerCase();
      const Value = rule[key];
      return {
        Name,
        Value,
      };
    });
    return {
      Key: {
        FilterRules,
      },
    };
  }
  return undefined;
}

async function getConfiguration(config) {
  const { bucketName, region } = config;
  const Bucket = bucketName;
  const payload = {
    Bucket,
  };
  return awsRequest(
    { name: 'S3', params: { region } },
    'getBucketNotificationConfiguration',
    payload
  ).then((data) => data);
}

async function updateConfiguration(config) {
  const { lambdaArn, functionName, bucketName, bucketConfigs, region } = config;
  const Bucket = bucketName;

  return getConfiguration(config).then((NotificationConfiguration) => {
    // remove configurations for this specific function
    NotificationConfiguration.LambdaFunctionConfigurations =
      NotificationConfiguration.LambdaFunctionConfigurations.filter(
        (conf) => !conf.Id.startsWith(functionName)
      );

    // add the events to the existing NotificationConfiguration
    bucketConfigs.forEach((bucketConfig) => {
      const Events = [bucketConfig.Event];
      const LambdaFunctionArn = lambdaArn;
      const Id = generateId(functionName, bucketConfig);
      const Filter = createFilter(bucketConfig);
      NotificationConfiguration.LambdaFunctionConfigurations.push({
        Events,
        LambdaFunctionArn,
        Filter,
        Id,
      });
    });

    const payload = {
      Bucket,
      NotificationConfiguration,
    };
    return awsRequest(
      { name: 'S3', params: { region } },
      'putBucketNotificationConfiguration',
      payload
    );
  });
}

async function removeConfiguration(config) {
  const { functionName, bucketName, region } = config;
  const Bucket = bucketName;

  return getConfiguration(config).then((NotificationConfiguration) => {
    // remove configurations for this specific function
    NotificationConfiguration.LambdaFunctionConfigurations =
      NotificationConfiguration.LambdaFunctionConfigurations.filter(
        (conf) => !conf.Id.startsWith(functionName)
      );

    const payload = {
      Bucket,
      NotificationConfiguration,
    };
    return awsRequest(
      { name: 'S3', params: { region } },
      'putBucketNotificationConfiguration',
      payload
    );
  });
}

module.exports = {
  updateConfiguration,
  removeConfiguration,
};