index.js

const AWS = require('aws-sdk');
AWS.config.update( {
  region: 'us-east-2'
});
const dynamodb = new AWS.DynamoDB.DocumentClient();
const dynamodbTableName = 'contact-list';
const healthPath = '/health';
const contactPath = '/contact';
const contactsPath = '/contacts';

exports.handler = async function(event) {
  console.log('Request event: ', event);
  let response;
  switch(true) {
    case event.httpMethod === 'GET' && event.path === healthPath:
      response = buildResponse(200);
      break;
    case event.httpMethod === 'GET' && event.path === contactPath:
      response = await getContact(event.queryStringParameters.contactId);
      break;
    case event.httpMethod === 'GET' && event.path === contactsPath:
      response = await getContacts();
      break;
    case event.httpMethod === 'POST' && event.path === contactPath:
      response = await saveContact(JSON.parse(event.body));
      break;
    case event.httpMethod === 'PATCH' && event.path === contactPath:
      const requestBody = JSON.parse(event.body);
      response = await modifyContact(requestBody.contactId, requestBody.updateKey, requestBody.updateValue);
      break;
    case event.httpMethod === 'DELETE' && event.path === contactPath:
      response = await deleteContact(JSON.parse(event.body).contactId);
      break;
    default:
      response = buildResponse(404, '404 Not Found');
  }
  return response;
}

// Get a single contact from DynamoDB table
async function getContact(contactId) {
  const params = {
    TableName: dynamodbTableName,
    Key: {
      'contactId': contactId
    }
  }
  return await dynamodb.get(params).promise().then((response) => {
    return buildResponse(200, response.Item);
  }, (error) => {
    console.error('Error getting contact: ', error);
  });
}

// Get all contacts from DynamoDB table
async function getContacts() {
  const params = {
    TableName: dynamodbTableName
  }
  const allContacts = await scanDynamoRecords(params, []);
  const body = {
    contacts: allContacts
  }
  return buildResponse(200, body);
}

async function scanDynamoRecords(scanParams, itemArray) {
  try {
    const dynamoData = await dynamodb.scan(scanParams).promise();
    itemArray = itemArray.concat(dynamoData.Items);
    if (dynamoData.LastEvaluatedKey) {
      scanParams.ExclusiveStartkey = dynamoData.LastEvaluatedKey;
      return await scanDynamoRecords(scanParams, itemArray);
    }
    return itemArray;
  } catch(error) {
    console.error('Error getting contacts: ', error);
  }
}

// Add a new contact to DynamoDB table
async function saveContact(requestBody) {
  const params = {
    TableName: dynamodbTableName,
    Item: requestBody
  }
  return await dynamodb.put(params).promise().then(() => {
    const body = {
      Operation: 'SAVE',
      Message: 'SUCCESS',
      Item: requestBody
    }
    return buildResponse(200, body);
  }, (error) => {
    console.error('Error saving contact: ', error);
  })
}

// Modify/update a single contact in DynamoDB table
async function modifyContact(contactId, updateKey, updateValue) {
  const params = {
    TableName: dynamodbTableName,
    Key: {
      'contactId': contactId
    },
    UpdateExpression: `set ${updateKey} = :value`,
    ExpressionAttributeValues: {
      ':value': updateValue
    },
    ReturnValues: 'UPDATED_NEW'
  }
  return await dynamodb.update(params).promise().then((response) => {
    const body = {
      Operation: 'UPDATE',
      Message: 'SUCCESS',
      UpdatedAttributes: response
    }
    return buildResponse(200, body);
  }, (error) => {
    console.error('Error modifying contact: ', error);
  })
}

// Delete a single contact from DynamoDB table
async function deleteContact(contactId) {
  const params = {
    TableName: dynamodbTableName,
    Key: {
      'contactId': contactId
    },
    ReturnValues: 'ALL_OLD'
  }
  return await dynamodb.delete(params).promise().then((response) => {
    const body = {
      Operation: 'DELETE',
      Message: 'SUCCESS',
      Item: response
    }
    return buildResponse(200, body);
  }, (error) => {
    console.error('Error deleting contact: ', error);
  })
}

function buildResponse(statusCode, body) {
  return {
    statusCode: statusCode,
    headers: {
      'Content-Type': 'application/json'
    },
    body: JSON.stringify(body)
  }
}