AWS Lambda functions

index-1.js

const AWS = require("aws-sdk");
const { randomize } = require("./randomize.js");

const documentClient = new AWS.DynamoDB.DocumentClient();

exports.handler = async event => {
  const { url } = JSON.parse(event.body);
  const id = randomize(6);
  const params = {
    TableName: // dynamoDB,
    Item: {
      id,
      url,
      timestamp: Date.now()
    }
  };
  
  try {
    const data = await documentClient.put(params).promise();
    const response = {
      statusCode: 200,
      headers: {
        "Access-Control-Allow-Origin": "*"
      },
      body: JSON.stringify({id})
    };
    return response;
  } catch (e) {
    return {
      statusCode: 500,
      headers: {
        "Access-Control-Allow-Origin": "*"
      },
      body: JSON.stringify(e)
    };
  }
};

// {"body": "{\"url\": \"https://twitter.com\"}"}

index-2.js

const AWS = require("aws-sdk");

const documentClient = new AWS.DynamoDB.DocumentClient();

exports.handler = async (event) => {
    const id = event.pathParameters.id;
    
    if(id) {
        try {
            const params = {
                TableName:  // dynamoDB,
                Key: {
                    id: id
                }
            }
            const data = await documentClient.get(params).promise();
            if (data.Item) {
               return({
                    statusCode: 302,
                    headers: {
                        Location: data.Item.url
                    }
                }) 
            }
        } catch(e) {
            return({
                statusCode: 302,
                headers: {
                    Location: "https://slit.link/error"
                }
            })
        }
    }
    
    return({
        statusCode: 302,
        headers: {
            Location: "https://slit.link/lnf"
        }
    })
};