import AWS from "aws-sdk";

const TableName = process.env.tableName;

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

export async function main(event) {

const messageData = JSON.parse(event.body).data;

const { stage, domainName } = event.requestContext;

// Get all the connections

const connections = await dynamoDb

.scan({ TableName, ProjectionExpression: "id" })

.promise();

const apiG = new AWS.ApiGatewayManagementApi({

endpoint: `${domainName}/${stage}`,

});

const postToConnection = async function ({ id }) {

try {

// Send the message to the given client

await apiG

.postToConnection({ ConnectionId: id, Data: messageData })

.promise();

} catch (e) {

if (e.statusCode === 410) {

// Remove stale connections

await dynamoDb.delete({ TableName, Key: { id } }).promise();

}

}

};

// Iterate through all the connections

await Promise.all(connections.Items.map(postToConnection));

return { statusCode: 200, body: "Message sent" };

}