gost-aws.js

const { SESClient, SendEmailCommand } = require('@aws-sdk/client-ses');
const { S3Client } = require('@aws-sdk/client-s3');
const { SQSClient } = require('@aws-sdk/client-sqs');
const { getSignedUrl: awsGetSignedUrl } = require('@aws-sdk/s3-request-presigner');
/*
----------------------------------------------------------
                        AWS S3
----------------------------------------------------------
*/

function getS3Client() {
    let s3;
    if (process.env.LOCALSTACK_HOSTNAME) {
        /*
            1. Make sure the local environment has awslocal installed.
            2. Use the commands to create a bucket to test with.
                - awslocal s3api create-bucket --bucket arpa-audit-reports --region us-west-2 --create-bucket-configuration '{"LocationConstraint": "us-west-2"}'
            3. Access bucket resource metadata through the following URL.
                - awslocal s3api list-buckets
                - awslocal s3api list-objects --bucket arpa-audit-reports
        */
        console.log('------------ USING LOCALSTACK ------------');
        const endpoint = `http://${process.env.LOCALSTACK_HOSTNAME}:${process.env.EDGE_PORT || 4566}`;
        console.log(`endpoint: ${endpoint}`);
        s3 = new S3Client({
            endpoint,
            forcePathStyle: true,
            region: process.env.AWS_DEFAULT_REGION,
        });
    } else {
        s3 = new S3Client();
    }
    return s3;
}

/**
 *  This function is a wrapper around the getSignedUrl function from the @aws-sdk/s3-request-presigner package.
 *  Exists to organize the imports and to make it easier to mock in tests.
 */
async function getSignedUrl(s3, command, options) {
    return awsGetSignedUrl(s3, command, options);
}

/*
----------------------------------------------------------
                        AWS SES
----------------------------------------------------------
*/

function getSESClient() {
    const sesOptions = {};

    if (process.env.LOCALSTACK_HOSTNAME) {
        sesOptions.endpoint = `http://${process.env.LOCALSTACK_HOSTNAME}:${process.env.EDGE_PORT || 4566}`;
        sesOptions.region = process.env.AWS_DEFAULT_REGION;
    }

    return new SESClient(sesOptions);
}

async function sendEmail(message) {
    if (process.env.SUPPRESS_EMAIL) return;
    if (!process.env.NOTIFICATIONS_EMAIL) throw new Error('NOTIFICATIONS_EMAIL is not set');

    const transport = module.exports.getSESClient();
    const source = message.fromName
        ? `"${message.fromName}" <${process.env.NOTIFICATIONS_EMAIL}>`
        : process.env.NOTIFICATIONS_EMAIL;
    const params = {
        Destination: {
            ToAddresses: [message.toAddress],
        },
        Source: source,
        Message: {
            Subject: {
                Charset: 'UTF-8',
                Data: message.subject,
            },
            Body: {
                Html: {
                    Charset: 'UTF-8',
                    Data: message.body,
                },
                Text: {
                    Charset: 'UTF-8',
                    Data: message.text,
                },
            },
        },
        Tags: message.tags.map((tag) => {
            // Tags must be strings of format 'name=value'
            const match = tag.match(/(?<name>\w+)=(?<value>\w+)/);
            if (!match) {
                return null;
            }

            return {
                Name: match.groups.name,
                Value: match.groups.value,
            };
        }).filter((tagObj) => !!tagObj),
    };
    if (message.ccAddress) {
        params.Destination.CcAddresses = [message.ccAddress];
    }
    if (process.env.SES_CONFIGURATION_SET_DEFAULT) {
        params.ConfigurationSetName = process.env.SES_CONFIGURATION_SET_DEFAULT;
    }
    const command = new SendEmailCommand(params);
    try {
        const data = await transport.send(command);
        console.log('Success sending SES email:', JSON.stringify(data));
    } catch (err) {
        console.error('Error sending SES email:', err, err.stack);
        throw err;
    }
}

/*
----------------------------------------------------------
                        AWS SQS
----------------------------------------------------------
*/

function getSQSClient() {
    let sqs;
    if (process.env.LOCALSTACK_HOSTNAME) {
        console.log('------------ USING LOCALSTACK FOR SQS ------------');
        const endpoint = `http://${process.env.LOCALSTACK_HOSTNAME}:${process.env.EDGE_PORT || 4566}`;
        sqs = new SQSClient({ endpoint, region: process.env.AWS_DEFAULT_REGION });
    } else {
        sqs = new SQSClient();
    }
    return sqs;
}

module.exports = {
    getS3Client,
    getSignedUrl,
    getSESClient,
    sendEmail,
    getSQSClient,
};